uv_link_t/example/src/main.c

89 lines
2.0 KiB
C

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "uv.h"
#include "uv_link_t.h"
/* Declaration of `middle_methods` */
#include "middle.h"
#define CHECK(V) if ((V) != 0) abort()
#define CHECK_ALLOC(V) if ((V) == NULL) abort()
typedef struct client_s client_t;
static uv_tcp_t server;
struct client_s {
uv_tcp_t tcp;
uv_link_source_t source;
uv_link_t middle;
uv_link_observer_t observer;
};
static void close_cb(uv_link_t* link) {
client_t* client;
client = link->data;
free(client);
}
static void read_cb(uv_link_observer_t* observer,
ssize_t nread,
const uv_buf_t* buf) {
client_t* client;
client = observer->data;
if (nread < 0) {
fprintf(stderr, "error or close\n");
uv_link_close((uv_link_t*) observer, close_cb);
return;
}
fprintf(stderr, "read \"%.*s\"\n", (int) nread, buf->base);
}
static void connection_cb(uv_stream_t* s, int status) {
client_t* client;
CHECK_ALLOC(client = malloc(sizeof(*client)));
CHECK(uv_tcp_init(uv_default_loop(), &client->tcp));
CHECK(uv_accept(s, (uv_stream_t*) &client->tcp));
CHECK(uv_link_source_init(&client->source, (uv_stream_t*) &client->tcp));
CHECK(uv_link_init(&client->middle, &middle_methods));
CHECK(uv_link_chain((uv_link_t*) &client->source, &client->middle));
CHECK(uv_link_observer_init(&client->observer));
CHECK(uv_link_chain(&client->middle, (uv_link_t*) &client->observer));
client->observer.observer_read_cb = read_cb;
client->observer.data = client;
CHECK(uv_link_read_start((uv_link_t*) &client->observer));
}
int main() {
static const int kBacklog = 128;
uv_loop_t* loop;
struct sockaddr_in addr;
loop = uv_default_loop();
CHECK(uv_tcp_init(loop, &server));
CHECK(uv_ip4_addr("0.0.0.0", 9000, &addr));
CHECK(uv_tcp_bind(&server, (struct sockaddr*) &addr, 0));
fprintf(stderr, "Listening on 0.0.0.0:9000\n");
CHECK(uv_listen((uv_stream_t*) &server, kBacklog, connection_cb));
CHECK(uv_run(loop, UV_RUN_DEFAULT));
return 0;
}