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
110
111
112
113
114
115
116
117
118
|
#include <stdio.h>
#include <sys/epoll.h>
#include <unistd.h>
#include <lua.h>
#include "plop.h"
/// Prints program usage to standard error.
/// \param name Name of the executable from argv
static void usage(const char * const name)
{
dprintf(2,
"Usage: %s [-p PORT] [-h] [HANDLER]\n"
"Starts plop server listening on PORT and serving HANDLER.\n"
"HANDLER defaults to '" PLOP_DEFAULT_HANDLER "'.\n\n"
" -p\tstart listening on PORT (default: 8080)\n"
" -h\tprints this help message\n",
name);
}
/// Standard entry point for the program.
/// \param argc Argument count
/// \param argv Argument array
/// \return Error code
int main(int argc, char ** argv)
{
const char * service = "8080";
int opt;
while (-1 != (opt = getopt(argc, argv, "p:h")))
{
switch (opt)
{
case 'p':
{
service = optarg;
break;
}
case 'h':
{
usage(argv[0]);
return 0;
}
default:
{
usage(argv[0]);
return 1; // TODO: Extend error handling in main().
}
}
}
if (optind < argc)
{
plop.handler = argv[optind];
}
plop.L = plop_initialize_lua();
if (NULL == plop.L)
{
return 9;
}
if (LUA_OK != plop_load_handler(plop.L, plop.handler))
{
return 2;
}
plop.efd = epoll_create1(0);
if (-1 == plop.efd)
{
return 3;
}
struct epoll_event e;
e.events = EPOLLIN;
e.data.ptr = NULL; // TODO: Consider putting server's Lua state in here?
const int server = open_server(NULL, service); // TODO: Check server's fd before ctl?
if (-1 == epoll_ctl(plop.efd, EPOLL_CTL_ADD, server, &e))
{
return 4;
}
static const int MAX_EVENTS = 20;
struct epoll_event events[MAX_EVENTS];
while (1)
{
int evc = epoll_wait(plop.efd, events, MAX_EVENTS, -1);
if (-1 == evc)
{
return 5;
}
for (int i = 0; i < evc; ++i)
{
if (NULL == events[i].data.ptr)
{
if (-1 == plop_handle_server(plop.L, plop.efd, server))
{
return 6;
}
}
else
{
struct connection * c = (struct connection *) events[i].data.ptr;
if (-1 == plop_handle_client(plop.L, c))
{
return 7;
}
}
}
}
}
|
