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
|
#include <uvh.h>
#define MAX_CHUNK 100
struct chunker
{
FILE *fp;
};
int stream_cb(char **chunk, void *data)
{
printf("%s\n", __FUNCTION__);
struct chunker *chunker = (struct chunker *) data;
if (!chunker->fp)
{
free(chunker);
return 0;
}
*chunk = malloc(MAX_CHUNK);
size_t read = fread(*chunk, 1, MAX_CHUNK, chunker->fp);
if (read < MAX_CHUNK || feof(chunker->fp))
{
fclose(chunker->fp);
chunker->fp = NULL;
}
if (read == 0)
{
free(chunker);
}
return read;
}
int request_handler(struct uvh_request *req)
{
printf("%s\n", __FUNCTION__);
FILE *fp = fopen(req->url.path, "r");
if (!fp)
{
uvh_request_write_status(req, HTTP_NOT_FOUND);
uvh_request_end(req);
return 0;
}
struct chunker *chunker = calloc(1, sizeof(*chunker));
chunker->fp = fp;
uvh_request_stream(req, stream_cb, chunker);
return 0;
}
void ctrlc_handler(uv_signal_t *handle, int signum)
{
struct uvh_server *server = handle->data;
uvh_server_stop(server);
uv_close((uv_handle_t *) handle, NULL);
(void) signum;
}
int main()
{
uv_signal_t sig;
struct uvh_server *server = uvh_server_init(uv_default_loop(),
NULL, &request_handler);
if (!server)
goto error;
if (uvh_server_listen(server, "127.0.0.1", 9898))
goto error;
uv_signal_init(uv_default_loop(), &sig);
sig.data = server;
uv_signal_start(&sig, &ctrlc_handler, SIGINT);
uv_run(uv_default_loop(), UV_RUN_DEFAULT);
printf("done\n");
uvh_server_free(server);
return 0;
error:
fprintf(stderr, "error: %s\n",
uv_strerror(uv_last_error(uv_default_loop())));
return 1;
}
|
