Created
September 11, 2011 07:47
-
-
Save creationix/1209301 to your computer and use it in GitHub Desktop.
libuv webserver updated to latest libuv
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#include <stdio.h> | |
#include <stdlib.h> | |
#include <assert.h> | |
#include "uv.h" | |
#include "http_parser.h" | |
static uv_stream_t server; | |
static http_parser_settings settings; | |
static uv_buf_t refbuf; | |
#define RESPONSE \ | |
"HTTP/1.1 200 OK\r\n" \ | |
"Content-Type: text/plain\r\n" \ | |
"Content-Length: 12\r\n" \ | |
"\r\n" \ | |
"hello world\n" | |
typedef struct { | |
uv_tcp_t handle; | |
http_parser parser; | |
uv_write_t write_req; | |
} client_t; | |
void on_close(uv_handle_t* handle) { | |
free(handle); | |
// printf("disconnected\n"); | |
} | |
uv_buf_t on_alloc(uv_handle_t* handle, size_t suggested_size) { | |
uv_buf_t buf; | |
buf.base = malloc(suggested_size); | |
buf.len = suggested_size; | |
return buf; | |
} | |
void on_read(uv_stream_t* stream, ssize_t nread, uv_buf_t buf) { | |
client_t* client = stream->data; | |
size_t parsed; | |
if (nread >= 0) { | |
parsed = http_parser_execute(&client->parser, &settings, buf.base, nread); | |
if (parsed < nread) { | |
uv_close((uv_handle_t*)stream, on_close); | |
fprintf(stderr, "parse error\n"); | |
} | |
} else { | |
uv_err_t err = uv_last_error(uv_default_loop()); | |
if (err.code == UV_EOF) { | |
uv_close((uv_handle_t*)stream, on_close); | |
} else { | |
fprintf(stderr, "read: %s\n", uv_strerror(err)); | |
} | |
} | |
free(buf.base); | |
} | |
void on_connection(uv_stream_t* server_handle, int status) { | |
assert(server_handle == &server); | |
// printf("connected\n"); | |
client_t* client = malloc(sizeof(client_t)); | |
uv_tcp_init(uv_default_loop(), &client->handle); | |
client->handle.data = client; | |
client->parser.data = client; | |
int r = uv_accept(&server, (uv_stream_t*)&client->handle); | |
if (r) { | |
uv_err_t err = uv_last_error(uv_default_loop()); | |
fprintf(stderr, "accept: %s\n", uv_strerror(err)); | |
exit(-1); | |
} | |
http_parser_init(&client->parser, HTTP_REQUEST); | |
uv_read_start((uv_stream_t*)&client->handle, on_alloc, on_read); | |
} | |
void after_write(uv_write_t* req, int status) { | |
uv_close((uv_handle_t*)req->handle, on_close); | |
} | |
int on_headers_complete(http_parser* parser) { | |
client_t* client = parser->data; | |
// printf("http message!\n"); | |
uv_write(&client->write_req, (uv_stream_t*)&client->handle, &refbuf, 1, after_write); | |
return 1; | |
} | |
int main() { | |
uv_init(); | |
refbuf.base = RESPONSE; | |
refbuf.len = sizeof(RESPONSE); | |
settings.on_headers_complete = on_headers_complete; | |
uv_tcp_init(uv_default_loop(), (uv_tcp_t*)&server); | |
struct sockaddr_in address = uv_ip4_addr("0.0.0.0", 8080); | |
int r = uv_tcp_bind((uv_tcp_t*)&server, address); | |
if (r) { | |
uv_err_t err = uv_last_error(uv_default_loop()); | |
fprintf(stderr, "bind: %s\n", uv_strerror(err)); | |
return -1; | |
} | |
r = uv_listen(&server, 128, on_connection); | |
if (r) { | |
uv_err_t err = uv_last_error(uv_default_loop()); | |
fprintf(stderr, "listen: %s\n", uv_strerror(err)); | |
return -1; | |
} | |
// Block in the main loop | |
uv_run(uv_default_loop()); | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment