adamc@1268
|
1 #include "config.h"
|
adamc@1268
|
2
|
adamc@116
|
3 #include <stdio.h>
|
adamc@116
|
4 #include <string.h>
|
adamc@502
|
5 #include <stdlib.h>
|
adamc@116
|
6 #include <sys/types.h>
|
adamc@116
|
7 #include <sys/socket.h>
|
adamc@116
|
8 #include <netinet/in.h>
|
adamc@472
|
9 #include <unistd.h>
|
adamc@502
|
10 #include <signal.h>
|
adamc@856
|
11 #include <stdarg.h>
|
adamc@116
|
12
|
adamc@138
|
13 #include <pthread.h>
|
adamc@138
|
14
|
adamc@244
|
15 #include "urweb.h"
|
adamc@853
|
16 #include "request.h"
|
adamc@859
|
17 #include "queue.h"
|
adamc@117
|
18
|
adamc@1094
|
19 extern uw_app uw_application;
|
adamc@1094
|
20
|
adamc@311
|
21 int uw_backlog = 10;
|
adamc@116
|
22
|
adamc@854
|
23 static char *get_header(void *data, const char *h) {
|
adamc@854
|
24 char *s = data;
|
adamc@854
|
25 int len = strlen(h);
|
adamc@854
|
26 char *p;
|
adamc@854
|
27
|
adamc@1134
|
28 while ((p = strchr(s, ':'))) {
|
adamc@854
|
29 if (p - s == len && !strncasecmp(s, h, len)) {
|
adamc@854
|
30 return p + 2;
|
adamc@854
|
31 } else {
|
adamc@854
|
32 if ((s = strchr(p, 0)) && s[1] != 0)
|
adamc@854
|
33 s += 2;
|
adamc@854
|
34 else
|
adamc@854
|
35 return NULL;
|
adamc@854
|
36 }
|
adamc@854
|
37 }
|
adamc@854
|
38
|
adamc@854
|
39 return NULL;
|
adamc@854
|
40 }
|
adamc@854
|
41
|
adamc@856
|
42 static void on_success(uw_context ctx) {
|
adamc@856
|
43 uw_write_header(ctx, "HTTP/1.1 200 OK\r\n");
|
adamc@856
|
44 }
|
adamc@856
|
45
|
adamc@856
|
46 static void on_failure(uw_context ctx) {
|
adamc@856
|
47 uw_write_header(ctx, "HTTP/1.1 500 Internal Server Error\r\n");
|
adamc@856
|
48 }
|
adamc@856
|
49
|
adamc@856
|
50 static void log_error(void *data, const char *fmt, ...) {
|
adamc@856
|
51 va_list ap;
|
adamc@856
|
52 va_start(ap, fmt);
|
adamc@856
|
53
|
adamc@856
|
54 vfprintf(stderr, fmt, ap);
|
adamc@856
|
55 }
|
adamc@856
|
56
|
adamc@856
|
57 static void log_debug(void *data, const char *fmt, ...) {
|
adamc@856
|
58 va_list ap;
|
adamc@856
|
59 va_start(ap, fmt);
|
adamc@856
|
60
|
adamc@856
|
61 vprintf(fmt, ap);
|
adamc@856
|
62 }
|
adamc@856
|
63
|
adamc@853
|
64 static void *worker(void *data) {
|
adamc@853
|
65 int me = *(int *)data;
|
adamc@1094
|
66 uw_context ctx = uw_request_new_context(&uw_application, NULL, log_error, log_debug);
|
adamc@853
|
67 size_t buf_size = 2;
|
adamc@853
|
68 char *buf = malloc(buf_size);
|
adamc@853
|
69 uw_request_context rc = uw_new_request_context();
|
adamc@698
|
70
|
adamc@272
|
71 while (1) {
|
adamc@853
|
72 char *back = buf;
|
adamc@859
|
73 int sock = uw_dequeue();
|
adamc@138
|
74
|
adamc@138
|
75 printf("Handling connection with thread #%d.\n", me);
|
adamc@138
|
76
|
adamc@138
|
77 while (1) {
|
adamc@737
|
78 int r;
|
adamc@854
|
79 char *method, *path, *query_string, *headers, *body, *s, *s2;
|
adamc@737
|
80
|
adamc@742
|
81 if (back - buf == buf_size - 1) {
|
adamc@737
|
82 char *new_buf;
|
adamc@737
|
83 buf_size *= 2;
|
adamc@737
|
84 new_buf = realloc(buf, buf_size);
|
adamc@737
|
85 back = new_buf + (back - buf);
|
adamc@737
|
86 buf = new_buf;
|
adamc@737
|
87 }
|
adamc@737
|
88
|
adamc@742
|
89 r = recv(sock, back, buf_size - 1 - (back - buf), 0);
|
adamc@138
|
90
|
adamc@138
|
91 if (r < 0) {
|
adamc@138
|
92 fprintf(stderr, "Recv failed\n");
|
adamc@138
|
93 break;
|
adamc@138
|
94 }
|
adamc@138
|
95
|
adamc@138
|
96 if (r == 0) {
|
adamc@138
|
97 printf("Connection closed.\n");
|
adamc@138
|
98 break;
|
adamc@138
|
99 }
|
adamc@138
|
100
|
adamc@138
|
101 back += r;
|
adamc@138
|
102 *back = 0;
|
adamc@730
|
103
|
adamc@854
|
104 if ((body = strstr(buf, "\r\n\r\n"))) {
|
adamc@853
|
105 request_result rr;
|
adamc@138
|
106
|
adamc@854
|
107 body[0] = body[1] = 0;
|
adamc@854
|
108 body += 4;
|
adamc@854
|
109
|
adamc@854
|
110 if ((s = strcasestr(buf, "\r\nContent-Length: ")) && s < body) {
|
adamc@853
|
111 int clen;
|
adamc@737
|
112
|
adamc@854
|
113 if (sscanf(s + 18, "%d\r\n", &clen) != 1) {
|
adamc@853
|
114 fprintf(stderr, "Malformed Content-Length header\n");
|
adamc@167
|
115 break;
|
adamc@167
|
116 }
|
adamc@167
|
117
|
adamc@854
|
118 while (back - body < clen) {
|
adamc@854
|
119 if (back - buf == buf_size - 1) {
|
adamc@854
|
120 char *new_buf;
|
adamc@854
|
121 buf_size *= 2;
|
adamc@854
|
122 new_buf = realloc(buf, buf_size);
|
adamc@854
|
123
|
adamc@854
|
124 back = new_buf + (back - buf);
|
adamc@854
|
125 body = new_buf + (body - buf);
|
adamc@854
|
126 s = new_buf + (s - buf);
|
adamc@854
|
127
|
adamc@854
|
128 buf = new_buf;
|
adamc@854
|
129 }
|
adamc@854
|
130
|
adamc@854
|
131 r = recv(sock, back, buf_size - 1 - (back - buf), 0);
|
adamc@854
|
132
|
adamc@854
|
133 if (r < 0) {
|
adamc@854
|
134 fprintf(stderr, "Recv failed\n");
|
adamc@854
|
135 close(sock);
|
adamc@854
|
136 goto done;
|
adamc@854
|
137 }
|
adamc@854
|
138
|
adamc@854
|
139 if (r == 0) {
|
adamc@854
|
140 fprintf(stderr, "Connection closed.\n");
|
adamc@854
|
141 close(sock);
|
adamc@854
|
142 goto done;
|
adamc@854
|
143 }
|
adamc@854
|
144
|
adamc@854
|
145 back += r;
|
adamc@854
|
146 *back = 0;
|
adamc@854
|
147 }
|
adamc@167
|
148 }
|
adamc@138
|
149
|
adamc@1119
|
150 body[-4] = '\r';
|
adamc@1119
|
151 body[-3] = '\n';
|
adamc@1119
|
152
|
adamc@854
|
153 if (!(s = strstr(buf, "\r\n"))) {
|
adamc@854
|
154 fprintf(stderr, "No newline in request\n");
|
adamc@854
|
155 close(sock);
|
adamc@854
|
156 goto done;
|
adamc@854
|
157 }
|
adamc@854
|
158
|
adamc@1119
|
159 body[-4] = body[-3] = 0;
|
adamc@1119
|
160
|
adamc@854
|
161 *s = 0;
|
adamc@854
|
162 headers = s + 2;
|
adamc@854
|
163 method = s = buf;
|
adamc@854
|
164
|
adamc@854
|
165 if (!strsep(&s, " ")) {
|
adamc@854
|
166 fprintf(stderr, "No first space in HTTP command\n");
|
adamc@854
|
167 close(sock);
|
adamc@854
|
168 goto done;
|
adamc@854
|
169 }
|
adamc@854
|
170 path = s;
|
adamc@854
|
171
|
adamc@1134
|
172 if ((s = strchr(path, ' ')))
|
adamc@854
|
173 *s = 0;
|
adamc@854
|
174
|
adamc@1134
|
175 if ((s = strchr(path, '?'))) {
|
adamc@854
|
176 *s = 0;
|
adamc@854
|
177 query_string = s+1;
|
adamc@854
|
178 }
|
adamc@854
|
179 else
|
adamc@854
|
180 query_string = NULL;
|
adamc@854
|
181
|
adamc@854
|
182 s = headers;
|
adamc@1134
|
183 while ((s2 = strchr(s, '\r'))) {
|
adamc@854
|
184 s = s2;
|
adamc@854
|
185
|
adamc@854
|
186 if (s[1] == 0)
|
adamc@854
|
187 break;
|
adamc@854
|
188
|
adamc@854
|
189 *s = 0;
|
adamc@854
|
190 s += 2;
|
adamc@854
|
191 }
|
adamc@854
|
192
|
adamc@854
|
193 uw_set_headers(ctx, get_header, headers);
|
adamc@854
|
194
|
adamc@856
|
195 rr = uw_request(rc, ctx, method, path, query_string, body, back - body,
|
adamc@856
|
196 on_success, on_failure,
|
adamc@856
|
197 NULL, log_error, log_debug,
|
adamc@863
|
198 sock, uw_really_send, close);
|
adamc@311
|
199 uw_send(ctx, sock);
|
adamc@138
|
200
|
adamc@853
|
201 if (rr == SERVED || rr == FAILED)
|
adamc@853
|
202 close(sock);
|
adamc@853
|
203 else if (rr != KEEP_OPEN)
|
adamc@853
|
204 fprintf(stderr, "Illegal uw_request return code: %d\n", rr);
|
adamc@853
|
205
|
adamc@138
|
206 break;
|
adamc@138
|
207 }
|
adamc@116
|
208 }
|
adamc@116
|
209
|
adamc@854
|
210 done:
|
adamc@311
|
211 uw_reset(ctx);
|
adamc@116
|
212 }
|
adamc@1138
|
213
|
adamc@1138
|
214 return NULL;
|
adamc@116
|
215 }
|
adamc@102
|
216
|
adamc@477
|
217 static void help(char *cmd) {
|
adamc@477
|
218 printf("Usage: %s [-p <port>] [-t <thread-count>]\n", cmd);
|
adamc@477
|
219 }
|
adamc@477
|
220
|
adamc@502
|
221 static void sigint(int signum) {
|
adamc@502
|
222 printf("Exiting....\n");
|
adamc@502
|
223 exit(0);
|
adamc@502
|
224 }
|
adamc@502
|
225
|
adamc@1094
|
226 static loggers ls = {&uw_application, NULL, log_error, log_debug};
|
adamc@856
|
227
|
adamc@138
|
228 int main(int argc, char *argv[]) {
|
adamc@116
|
229 // The skeleton for this function comes from Beej's sockets tutorial.
|
adamc@138
|
230 int sockfd; // listen on sock_fd
|
adamc@116
|
231 struct sockaddr_in my_addr;
|
adamc@116
|
232 struct sockaddr_in their_addr; // connector's address information
|
adamc@1134
|
233 socklen_t sin_size;
|
adamc@1134
|
234 int yes = 1, uw_port = 8080, nthreads = 1, i, *names, opt;
|
adamc@502
|
235
|
adamc@502
|
236 signal(SIGINT, sigint);
|
adamc@505
|
237 signal(SIGPIPE, SIG_IGN);
|
adamc@505
|
238
|
adamc@477
|
239 while ((opt = getopt(argc, argv, "hp:t:")) != -1) {
|
adamc@472
|
240 switch (opt) {
|
adamc@472
|
241 case '?':
|
adamc@472
|
242 fprintf(stderr, "Unknown command-line option");
|
adamc@477
|
243 help(argv[0]);
|
adamc@472
|
244 return 1;
|
adamc@138
|
245
|
adamc@477
|
246 case 'h':
|
adamc@477
|
247 help(argv[0]);
|
adamc@477
|
248 return 0;
|
adamc@477
|
249
|
adamc@472
|
250 case 'p':
|
adamc@472
|
251 uw_port = atoi(optarg);
|
adamc@472
|
252 if (uw_port <= 0) {
|
adamc@472
|
253 fprintf(stderr, "Invalid port number\n");
|
adamc@477
|
254 help(argv[0]);
|
adamc@472
|
255 return 1;
|
adamc@472
|
256 }
|
adamc@472
|
257 break;
|
adamc@472
|
258
|
adamc@472
|
259 case 't':
|
adamc@472
|
260 nthreads = atoi(optarg);
|
adamc@472
|
261 if (nthreads <= 0) {
|
adamc@472
|
262 fprintf(stderr, "Invalid thread count\n");
|
adamc@477
|
263 help(argv[0]);
|
adamc@472
|
264 return 1;
|
adamc@472
|
265 }
|
adamc@472
|
266 break;
|
adamc@472
|
267
|
adamc@472
|
268 default:
|
adamc@472
|
269 fprintf(stderr, "Unexpected getopt() behavior\n");
|
adamc@472
|
270 return 1;
|
adamc@472
|
271 }
|
adamc@138
|
272 }
|
adamc@138
|
273
|
adamc@1094
|
274 uw_request_init(&uw_application, NULL, log_error, log_debug);
|
adamc@687
|
275
|
adamc@138
|
276 names = calloc(nthreads, sizeof(int));
|
adamc@116
|
277
|
adamc@116
|
278 sockfd = socket(PF_INET, SOCK_STREAM, 0); // do some error checking!
|
adamc@116
|
279
|
adamc@116
|
280 if (sockfd < 0) {
|
adamc@116
|
281 fprintf(stderr, "Listener socket creation failed\n");
|
adamc@116
|
282 return 1;
|
adamc@116
|
283 }
|
adamc@116
|
284
|
adamc@116
|
285 if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) < 0) {
|
adamc@116
|
286 fprintf(stderr, "Listener socket option setting failed\n");
|
adamc@116
|
287 return 1;
|
adamc@116
|
288 }
|
adamc@116
|
289
|
adamc@116
|
290 my_addr.sin_family = AF_INET; // host byte order
|
adamc@311
|
291 my_addr.sin_port = htons(uw_port); // short, network byte order
|
adamc@116
|
292 my_addr.sin_addr.s_addr = INADDR_ANY; // auto-fill with my IP
|
adamc@116
|
293 memset(my_addr.sin_zero, '\0', sizeof my_addr.sin_zero);
|
adamc@116
|
294
|
adamc@116
|
295 if (bind(sockfd, (struct sockaddr *)&my_addr, sizeof my_addr) < 0) {
|
adamc@116
|
296 fprintf(stderr, "Listener socket bind failed\n");
|
adamc@116
|
297 return 1;
|
adamc@116
|
298 }
|
adamc@116
|
299
|
adamc@311
|
300 if (listen(sockfd, uw_backlog) < 0) {
|
adamc@116
|
301 fprintf(stderr, "Socket listen failed\n");
|
adamc@116
|
302 return 1;
|
adamc@116
|
303 }
|
adamc@116
|
304
|
adamc@116
|
305 sin_size = sizeof their_addr;
|
adamc@116
|
306
|
adamc@311
|
307 printf("Listening on port %d....\n", uw_port);
|
adamc@116
|
308
|
adamc@667
|
309 {
|
adamc@667
|
310 pthread_t thread;
|
adamc@667
|
311
|
adamc@856
|
312 if (pthread_create(&thread, NULL, client_pruner, &ls)) {
|
adamc@667
|
313 fprintf(stderr, "Error creating pruner thread\n");
|
adamc@667
|
314 return 1;
|
adamc@667
|
315 }
|
adamc@667
|
316 }
|
adamc@667
|
317
|
adamc@138
|
318 for (i = 0; i < nthreads; ++i) {
|
adamc@138
|
319 pthread_t thread;
|
adamc@138
|
320 names[i] = i;
|
adamc@138
|
321 if (pthread_create(&thread, NULL, worker, &names[i])) {
|
adamc@138
|
322 fprintf(stderr, "Error creating worker thread #%d\n", i);
|
adamc@138
|
323 return 1;
|
adamc@138
|
324 }
|
adamc@138
|
325 }
|
adamc@138
|
326
|
adamc@116
|
327 while (1) {
|
adamc@138
|
328 int new_fd = accept(sockfd, (struct sockaddr *)&their_addr, &sin_size);
|
adamc@116
|
329
|
adamc@116
|
330 if (new_fd < 0) {
|
adamc@116
|
331 fprintf(stderr, "Socket accept failed\n");
|
adamc@116
|
332 return 1;
|
adamc@116
|
333 }
|
adamc@116
|
334
|
adamc@116
|
335 printf("Accepted connection.\n");
|
adamc@138
|
336
|
adamc@859
|
337 uw_enqueue(new_fd);
|
adamc@116
|
338 }
|
adamc@102
|
339 }
|
adamc@1121
|
340
|
adamc@1121
|
341 void *uw_init_client_data() {
|
adamc@1121
|
342 return NULL;
|
adamc@1121
|
343 }
|
adamc@1121
|
344
|
adamc@1121
|
345 void uw_free_client_data(void *data) {
|
adamc@1121
|
346 }
|
adamc@1121
|
347
|
adamc@1121
|
348 void uw_copy_client_data(void *dst, void *src) {
|
adamc@1121
|
349 }
|
adamc@1121
|
350
|
adamc@1121
|
351 void uw_do_expunge(uw_context ctx, uw_Basis_client cli, void *data) {
|
adamc@1121
|
352 if (uw_get_app(ctx)->db_begin(ctx))
|
adamc@1121
|
353 uw_error(ctx, FATAL, "Error running SQL BEGIN");
|
adamc@1121
|
354 uw_get_app(ctx)->expunger(ctx, cli);
|
adamc@1121
|
355 if (uw_get_app(ctx)->db_commit(ctx))
|
adamc@1121
|
356 uw_error(ctx, FATAL, "Error running SQL COMMIT");
|
adamc@1121
|
357 }
|
adamc@1121
|
358
|
adamc@1121
|
359 void uw_post_expunge(uw_context ctx, void *data) {
|
adamc@1121
|
360 }
|
adam@1320
|
361
|
adam@1320
|
362 int uw_supports_direct_status = 1;
|