-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathserver.c
More file actions
109 lines (80 loc) · 2.56 KB
/
server.c
File metadata and controls
109 lines (80 loc) · 2.56 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <uv.h>
uv_loop_t *loop;
void alloc_buffer(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf) {
*buf = uv_buf_init((char*) malloc(suggested_size), suggested_size);
}
typedef struct {
uv_write_t req;
uv_buf_t buf; // data payload
} write_req_t;
void free_write_req(uv_write_t *req) {
// coerce to write_req_t
write_req_t *wr = (write_req_t *) req;
// free the buffer space, allocated by libuv
free( wr->buf.base);
// and the request
free( wr);
}
void on_write(uv_write_t *req, int status) {
// callback on write completion to client...
fprintf( stderr, "on_write..\n");
if (status == -1) {
fprintf(stderr, "Write error!\n");
}
free_write_req( req);
}
void on_read(uv_stream_t *client, ssize_t nread, const uv_buf_t *buf) {
// callback on read completion from client....
fprintf( stderr, "on_read... %ld, %ld\n", nread, buf->len);
if (nread == -1) {
fprintf(stderr, "Read error!\n");
uv_close((uv_handle_t*)client, NULL);
return;
}
// we echo the data which was just read.
// allocate a write request incl buffer , to be freed...
write_req_t *req = (write_req_t *) malloc( sizeof( write_req_t));
// initialize buffer, data is shared by reference and thus not yet freed here...
// allocation done by libuv, to be freed when write is done
req->buf = uv_buf_init( buf->base, nread);
uv_write( (uv_write_t *) req, client, &(req->buf), 1, on_write);
}
void on_new_connection(uv_stream_t *server, int status) {
// callback on listen() to new connection.
fprintf( stderr, "on_new_connection...\n");
if (status == -1) {
return;
}
uv_tcp_t *client = (uv_tcp_t*) malloc(sizeof(uv_tcp_t));
uv_tcp_init(loop, client);
// this seems to be synchronous, one wonders...
if (uv_accept(server, (uv_stream_t*) client) == 0) {
uv_read_start((uv_stream_t*) client, alloc_buffer, on_read);
}
else {
fprintf( stderr, "failure on uv_accept...\n");
uv_close((uv_handle_t*) client, NULL);
}
}
#define PORT 7000
// #define HOST "127.0.0.1"
#define HOST "0.0.0.0"
int main() {
loop = uv_default_loop();
uv_tcp_t server;
uv_tcp_init(loop, &server);
struct sockaddr_in bind_addr;
uv_ip4_addr(HOST, PORT, &bind_addr);
uv_tcp_bind(&server, (const struct sockaddr *) &bind_addr, 0);
int r = uv_listen((uv_stream_t*) &server, 128, on_new_connection);
if (r) {
fprintf(stderr, "Listen error!\n");
return 1;
} else {
fprintf( stderr, "server listening...\n");
}
return uv_run(loop, UV_RUN_DEFAULT);
}