11_tcp_ping_pong.c File Reference

#include <arpa/inet.h>
#include <fcntl.h>
#include <netinet/in.h>
#include <stdint.h>
#include <stdio.h>
#include <sys/socket.h>
#include <unistd.h>
#include <serene/rt/fiber/io.h>
#include <serene/runtime.h>

Include dependency graph for 11_tcp_ping_pong.c:

Go to the source code of this file.

Data Structures
struct	server_args_t
struct	client_args_t

Macros
#define	ROUNDS   4000

Functions
static void	set_nonblock (int fd)
static srn_fiber_result_t	server (srn_context_t *ctx, void *arg)
static srn_fiber_result_t	client (srn_context_t *ctx, void *arg)
int	main (void)

Macro Definition Documentation

ROUNDS

#define ROUNDS   4000

Definition at line 43 of file 11_tcp_ping_pong.c.

Function Documentation

client()

srn_fiber_result_t client ( srn_context_t * ctx, void * arg )

static

Definition at line 85 of file 11_tcp_ping_pong.c.

                                                                {
  (void)ctx;
  client_args_t *c = arg;
  int fd           = socket(AF_INET, SOCK_STREAM, 0);
  if (fd < 0) {
    perror("socket");
    return nullptr;
  }
  set_nonblock(fd);
 
  struct sockaddr_in to = {.sin_family = AF_INET, .sin_port = c->port};
  to.sin_addr.s_addr    = htonl(INADDR_LOOPBACK);
 
  srn_error_t *err = srn_fiber_connect(fd, (struct sockaddr *)&to, sizeof(to));
  if (err != nullptr) {
    printf("client: connect failed: %s\n", err->msg);
    close(fd);
    return nullptr;
  }
 
  // Send a ping, wait for the pong, then send the next one.
  for (int i = 0; i < ROUNDS; i++) {
    (void)srn_fiber_write(fd, "ping", 4, -1);
    char buf[16]           = {0};
    srn_io_size_result_t r = srn_fiber_read(fd, buf, sizeof(buf) - 1, -1);
    if (HAS_ERROR(r) || r.count == 0) {
      printf("client: peer closed after %d rounds\n", i);
      break;
    }
    buf[r.count] = '\0';
    printf("client: round %d sent \"ping\", got \"%s\"\n", i, buf);
  }
 
  close(fd);
  return nullptr;
}

Here is the call graph for this function:

main()

int main

(

void

)

Definition at line 122 of file 11_tcp_ping_pong.c.

               {
  SERENE_RUNTIME_INIT_WITH_SCHED(engine, sched, nullptr);
  srn_context_t *ctx = srn_context_make(engine);
 
  int lst              = socket(AF_INET, SOCK_STREAM, 0);
  struct sockaddr_in a = {.sin_family = AF_INET};
  a.sin_addr.s_addr    = htonl(INADDR_LOOPBACK);
  if (lst < 0 || bind(lst, (struct sockaddr *)&a, sizeof(a)) != 0 || listen(lst, 4) != 0) {
    perror("listener setup");
    return 1;
  }
  set_nonblock(lst);
  socklen_t al = sizeof(a);
  if (getsockname(lst, (struct sockaddr *)&a, &al) != 0) {
    perror("getsockname");
    return 1;
  }
 
  server_args_t s = {lst};
  client_args_t c = {a.sin_port};
  (void)srn_fiber_spawn(ctx, server, &s);
  (void)srn_fiber_spawn(ctx, client, &c);
 
  srn_sched_run(sched, 1);
  close(lst);
 
  srn_context_release(ctx);
  SERENE_RUNTIME_SHUTDOWN(engine);
  return 0;
}

Here is the call graph for this function:

server()

srn_fiber_result_t server ( srn_context_t * ctx, void * arg )

static

Definition at line 58 of file 11_tcp_ping_pong.c.

                                                                {
  (void)ctx;
  server_args_t *s         = arg;
  srn_io_accept_result_t a = srn_fiber_accept(s->listener, nullptr, 0);
  if (HAS_ERROR(a)) {
    printf("server: accept failed: %s\n", a.maybe_error->msg);
    return nullptr;
  }
  int conn = a.fd;
 
  // Read a request, write a response, repeat until the peer closes.
  for (int i = 0; i < ROUNDS; i++) {
    char buf[16]           = {0};
    srn_io_size_result_t r = srn_fiber_read(conn, buf, sizeof(buf) - 1, -1);
    if (HAS_ERROR(r) || r.count == 0) {
      printf("server: peer closed after %d rounds\n", i);
      break;
    }
    buf[r.count] = '\0';
    printf("server: round %d got \"%s\", replying \"pong\"\n", i, buf);
    (void)srn_fiber_write(conn, "pong", 4, -1);
  }
 
  close(conn);
  return nullptr;
}

Here is the call graph for this function:

set_nonblock()

void set_nonblock ( int fd )

static

Definition at line 53 of file 11_tcp_ping_pong.c.

                                 {
  int fl = fcntl(fd, F_GETFL, 0);
  (void)fcntl(fd, F_SETFL, fl | O_NONBLOCK);
}

Here is the caller graph for this function: