使用libuv写的tcp server

来源:互联网 发布:自己印t恤 知乎 编辑:程序博客网 时间:2024/05/19 11:20

运行环境:OSX 10

首次接触libuv,根据官方文档写的一个demo,记录一下

server:

#include <stdio.h>#include <stdlib.h>#include <string.h>#include <uv.h>#define DEFAULT_PORT 7000#define DEFAULT_BACKLOG 128uv_loop_t *loop;struct sockaddr_in addr;void alloc_buffer(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf) {    buf->base = (char*) malloc(suggested_size);    buf->len = suggested_size;}void echo_write(uv_write_t *req, int status) {    if (status) {        fprintf(stderr, "Write error %s\n", uv_strerror(status));    }    free(req);}void echo_read(uv_stream_t *client, ssize_t nread, const uv_buf_t *buf) {    if (nread < 0) {        if (nread != UV_EOF)            fprintf(stderr, "Read error %s\n", uv_err_name(nread));        uv_close((uv_handle_t*) client, NULL);    } else if (nread > 0) {        /*uv_write_t *req = (uv_write_t *) malloc(sizeof(uv_write_t));        uv_buf_t wrbuf = uv_buf_init(buf->base, nread);        uv_write(req, client, &wrbuf, 1, echo_write);*/    }        printf("echo_read:%s\r\n",buf->base);    if (buf->base)        free(buf->base);}void on_new_connection(uv_stream_t *server, int status) {    if (status < 0) {        fprintf(stderr, "New connection error %s\n", uv_strerror(status));        // error!        return;    }        uv_tcp_t *client = (uv_tcp_t*) malloc(sizeof(uv_tcp_t));    uv_tcp_init(loop, client);    if (uv_accept(server, (uv_stream_t*) client) == 0) {        uv_read_start((uv_stream_t*) client, alloc_buffer, echo_read);    }    else {        uv_close((uv_handle_t*) client, NULL);    }    printf("on new connection, status:%d\r\n", status);}int main() {    printf("buliding tcp\n");    loop = uv_default_loop();        uv_tcp_t server;    uv_tcp_init(loop, &server);        printf("port%d", DEFAULT_PORT);        uv_ip4_addr("0.0.0.0", DEFAULT_PORT, &addr);        uv_tcp_bind(&server, (const struct sockaddr*)&addr, 0);    printf("listening %d\n", DEFAULT_PORT);    int r = uv_listen((uv_stream_t*) &server, DEFAULT_BACKLOG, on_new_connection);    if (r) {        fprintf(stderr, "Listen error %s\n", uv_strerror(r));        return 1;    }    return uv_run(loop, UV_RUN_DEFAULT);}


1 0