c/libuv
1
0
Fork 0
mirror of https://github.com/libuv/libuv synced 2025-03-28 21:13:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Echo server gracefully dies on 'Q'

Browse Source 
Allows for better clean up during tests.
This commit is contained in:
Ryan Dahl 2011-05-27 02:01:25 -07:00
parent e28c0bb60b
commit 982a620f1c
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
test/echo-server.c
View File

@ -31,6 +31,7 @@ typedef struct {
} write_req_t;
static int server_closed;
static uv_handle_t server;
@ -58,11 +59,13 @@ static void after_write(uv_req_t* req, int status) {
static void after_shutdown(uv_req_t* req, int status) {
uv_close(req->handle); /* XXX Is the uv_close necessary??? */
free(req);
}
static void after_read(uv_handle_t* handle, int nread, uv_buf_t buf) {
int i;
write_req_t *wr;
uv_req_t* req;
@ -87,6 +90,16 @@ static void after_read(uv_handle_t* handle, int nread, uv_buf_t buf) {
return;
}
/* Scan for the letter Q which signals that we should quit. */
if (!server_closed) {
for (i = 0; i < nread; i++) {
if (buf.base[i] == 'Q') {
uv_close(&server);
server_closed = 1;
}
}
}
wr = (write_req_t*) malloc(sizeof *wr);
uv_req_init(&wr->req, handle, after_write);
@ -151,11 +164,6 @@ static int echo_start(int port) {
}
static int echo_stop() {
return uv_close(&server);
}
static uv_buf_t echo_alloc(uv_handle_t* handle, size_t suggested_size) {
uv_buf_t buf;
buf.base = (char*) malloc(suggested_size);
@ -169,7 +177,6 @@ HELPER_IMPL(echo_server) {
if (echo_start(TEST_PORT))
return 1;
fprintf(stderr, "Listening!\n");
uv_run();
return 0;
}