blob: 717f6c0257ccb2f4853697e869557ee1110bdb19 [file] [log] [blame]
Andy Green58eaa742011-03-07 17:54:06 +00001/*
Andy Greena0da8a82010-11-08 17:12:19 +00002 * libwebsockets - small server side websockets and web server implementation
Andy Green8f037e42010-12-19 22:13:26 +00003 *
Andy Greena0da8a82010-11-08 17:12:19 +00004 * Copyright (C) 2010 Andy Green <andy@warmcat.com>
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation:
9 * version 2.1 of the License.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
19 * MA 02110-1301 USA
Andy Green05a0a7b2010-10-31 17:51:39 +000020 */
21
Andy Green7c212cc2010-11-08 20:20:42 +000022#include "private-libwebsockets.h"
Andy Greenff95d7a2010-10-28 22:36:01 +010023
Peter Hinz56885f32011-03-02 22:03:47 +000024#ifdef WIN32
25
26#else
27#include <ifaddrs.h>
Andy Green7627af52011-03-09 15:13:52 +000028#include <sys/un.h>
Peter Hinz56885f32011-03-02 22:03:47 +000029#endif
Andy Green2e24da02011-03-05 16:12:04 +000030
31#ifdef LWS_OPENSSL_SUPPORT
32int openssl_websocket_private_data_index;
33#endif
34
Andy Greenbe93fef2011-02-14 20:25:43 +000035/*
36 * In-place str to lower case
37 */
38
39static void
40strtolower(char *s)
41{
42 while (*s) {
43 *s = tolower(*s);
44 s++;
45 }
46}
47
Andy Green0d338332011-02-12 11:57:43 +000048/* file descriptor hash management */
49
50struct libwebsocket *
Peter Hinz56885f32011-03-02 22:03:47 +000051wsi_from_fd(struct libwebsocket_context *context, int fd)
Andy Green0d338332011-02-12 11:57:43 +000052{
53 int h = LWS_FD_HASH(fd);
54 int n = 0;
55
Peter Hinz56885f32011-03-02 22:03:47 +000056 for (n = 0; n < context->fd_hashtable[h].length; n++)
57 if (context->fd_hashtable[h].wsi[n]->sock == fd)
58 return context->fd_hashtable[h].wsi[n];
Andy Green0d338332011-02-12 11:57:43 +000059
60 return NULL;
61}
62
63int
Peter Hinz56885f32011-03-02 22:03:47 +000064insert_wsi(struct libwebsocket_context *context, struct libwebsocket *wsi)
Andy Green0d338332011-02-12 11:57:43 +000065{
66 int h = LWS_FD_HASH(wsi->sock);
67
Peter Hinz56885f32011-03-02 22:03:47 +000068 if (context->fd_hashtable[h].length == MAX_CLIENTS - 1) {
Andy Green0d338332011-02-12 11:57:43 +000069 fprintf(stderr, "hash table overflow\n");
70 return 1;
71 }
72
Peter Hinz56885f32011-03-02 22:03:47 +000073 context->fd_hashtable[h].wsi[context->fd_hashtable[h].length++] = wsi;
Andy Green0d338332011-02-12 11:57:43 +000074
75 return 0;
76}
77
78int
Peter Hinz56885f32011-03-02 22:03:47 +000079delete_from_fd(struct libwebsocket_context *context, int fd)
Andy Green0d338332011-02-12 11:57:43 +000080{
81 int h = LWS_FD_HASH(fd);
82 int n = 0;
83
Peter Hinz56885f32011-03-02 22:03:47 +000084 for (n = 0; n < context->fd_hashtable[h].length; n++)
85 if (context->fd_hashtable[h].wsi[n]->sock == fd) {
86 while (n < context->fd_hashtable[h].length) {
87 context->fd_hashtable[h].wsi[n] =
88 context->fd_hashtable[h].wsi[n + 1];
Andy Green0d338332011-02-12 11:57:43 +000089 n++;
90 }
Peter Hinz56885f32011-03-02 22:03:47 +000091 context->fd_hashtable[h].length--;
Andy Green0d338332011-02-12 11:57:43 +000092
93 return 0;
94 }
95
96 fprintf(stderr, "Failed to find fd %d requested for "
97 "delete in hashtable\n", fd);
98 return 1;
99}
100
Andy Green1f9bf522011-02-14 21:14:37 +0000101#ifdef LWS_OPENSSL_SUPPORT
102static void
103libwebsockets_decode_ssl_error(void)
104{
105 char buf[256];
106 u_long err;
107
108 while ((err = ERR_get_error()) != 0) {
109 ERR_error_string_n(err, buf, sizeof(buf));
110 fprintf(stderr, "*** %s\n", buf);
111 }
112}
113#endif
Andy Green0d338332011-02-12 11:57:43 +0000114
Andy Green32375b72011-02-19 08:32:53 +0000115
116static int
117interface_to_sa(const char* ifname, struct sockaddr_in *addr, size_t addrlen)
118{
119 int rc = -1;
Peter Hinz56885f32011-03-02 22:03:47 +0000120#ifdef WIN32
121 // TODO
122#else
Andy Green32375b72011-02-19 08:32:53 +0000123 struct ifaddrs *ifr;
124 struct ifaddrs *ifc;
125 struct sockaddr_in *sin;
126
127 getifaddrs(&ifr);
128 for (ifc = ifr; ifc != NULL; ifc = ifc->ifa_next) {
129 if (strcmp(ifc->ifa_name, ifname))
130 continue;
131 if (ifc->ifa_addr == NULL)
132 continue;
133 sin = (struct sockaddr_in *)ifc->ifa_addr;
134 if (sin->sin_family != AF_INET)
135 continue;
136 memcpy(addr, sin, addrlen);
137 rc = 0;
138 }
139
140 freeifaddrs(ifr);
Peter Hinz56885f32011-03-02 22:03:47 +0000141#endif
Andy Green32375b72011-02-19 08:32:53 +0000142 return rc;
143}
144
Andy Green8f037e42010-12-19 22:13:26 +0000145void
Peter Hinz56885f32011-03-02 22:03:47 +0000146libwebsocket_close_and_free_session(struct libwebsocket_context *context,
Andy Green687b0182011-02-26 11:04:01 +0000147 struct libwebsocket *wsi, enum lws_close_status reason)
Andy Green251f6fa2010-11-03 11:13:06 +0000148{
Andy Greenb45993c2010-12-18 15:13:50 +0000149 int n;
Andy Green62c54d22011-02-14 09:14:25 +0000150 int old_state;
Andy Green5e1fa172011-02-10 09:07:05 +0000151 unsigned char buf[LWS_SEND_BUFFER_PRE_PADDING + 2 +
152 LWS_SEND_BUFFER_POST_PADDING];
Andy Greenc44159f2011-03-07 07:08:18 +0000153 int ret;
154 int m;
155 struct lws_tokens eff_buf;
Andy Greenb45993c2010-12-18 15:13:50 +0000156
Andy Green4b6fbe12011-02-14 08:03:48 +0000157 if (!wsi)
Andy Greenb45993c2010-12-18 15:13:50 +0000158 return;
159
Andy Green62c54d22011-02-14 09:14:25 +0000160 old_state = wsi->state;
Andy Green251f6fa2010-11-03 11:13:06 +0000161
Andy Green62c54d22011-02-14 09:14:25 +0000162 if (old_state == WSI_STATE_DEAD_SOCKET)
Andy Green5e1fa172011-02-10 09:07:05 +0000163 return;
164
Andy Greenda527df2011-03-07 07:08:12 +0000165 wsi->close_reason = reason;
166
167 /*
Andy Greenc44159f2011-03-07 07:08:18 +0000168 * flush any tx pending from extensions, since we may send close packet
169 * if there are problems with send, just nuke the connection
170 */
171
172 ret = 1;
173 while (ret == 1) {
174
175 /* default to nobody has more to spill */
176
177 ret = 0;
178 eff_buf.token = NULL;
179 eff_buf.token_len = 0;
180
181 /* show every extension the new incoming data */
182
183 for (n = 0; n < wsi->count_active_extensions; n++) {
184 m = wsi->active_extensions[n]->callback(
185 wsi->protocol->owning_server, wsi,
186 LWS_EXT_CALLBACK_FLUSH_PENDING_TX,
187 wsi->active_extensions_user[n], &eff_buf, 0);
188 if (m < 0) {
189 fprintf(stderr, "Extension reports "
190 "fatal error\n");
191 goto just_kill_connection;
192 }
193 if (m)
194 /*
195 * at least one extension told us he has more
196 * to spill, so we will go around again after
197 */
198 ret = 1;
199 }
200
201 /* assuming they left us something to send, send it */
202
203 if (eff_buf.token_len)
204 if (lws_issue_raw(wsi, (unsigned char *)eff_buf.token,
205 eff_buf.token_len))
206 goto just_kill_connection;
207 }
208
209 /*
Andy Greenda527df2011-03-07 07:08:12 +0000210 * signal we are closing, libsocket_write will
211 * add any necessary version-specific stuff. If the write fails,
212 * no worries we are closing anyway. If we didn't initiate this
213 * close, then our state has been changed to
214 * WSI_STATE_RETURNED_CLOSE_ALREADY and we will skip this.
215 *
216 * Likewise if it's a second call to close this connection after we
217 * sent the close indication to the peer already, we are in state
218 * WSI_STATE_AWAITING_CLOSE_ACK and will skip doing this a second time.
219 */
220
221 if (old_state == WSI_STATE_ESTABLISHED &&
222 reason != LWS_CLOSE_STATUS_NOSTATUS) {
223 n = libwebsocket_write(wsi, &buf[LWS_SEND_BUFFER_PRE_PADDING],
224 0, LWS_WRITE_CLOSE);
225 if (!n) {
226 /*
227 * we have sent a nice protocol level indication we
228 * now wish to close, we should not send anything more
229 */
230
231 wsi->state = WSI_STATE_AWAITING_CLOSE_ACK;
232
233 /* and we should wait for a reply for a bit */
234
235 libwebsocket_set_timeout(wsi,
236 PENDING_TIMEOUT_CLOSE_ACK, 5);
237
238 fprintf(stderr, "sent close indication, awaiting ack\n");
239
240 return;
241 }
242
243 /* else, the send failed and we should just hang up */
244 }
245
Andy Greenc44159f2011-03-07 07:08:18 +0000246just_kill_connection:
Andy Greenda527df2011-03-07 07:08:12 +0000247 /*
248 * we won't be servicing or receiving anything further from this guy
249 * remove this fd from wsi mapping hashtable
250 */
Andy Green4b6fbe12011-02-14 08:03:48 +0000251
Peter Hinz56885f32011-03-02 22:03:47 +0000252 delete_from_fd(context, wsi->sock);
Andy Green4b6fbe12011-02-14 08:03:48 +0000253
254 /* delete it from the internal poll list if still present */
255
Peter Hinz56885f32011-03-02 22:03:47 +0000256 for (n = 0; n < context->fds_count; n++) {
257 if (context->fds[n].fd != wsi->sock)
Andy Green4b6fbe12011-02-14 08:03:48 +0000258 continue;
Peter Hinz56885f32011-03-02 22:03:47 +0000259 while (n < context->fds_count - 1) {
260 context->fds[n] = context->fds[n + 1];
Andy Green4b6fbe12011-02-14 08:03:48 +0000261 n++;
262 }
Peter Hinz56885f32011-03-02 22:03:47 +0000263 context->fds_count--;
Andy Green4b6fbe12011-02-14 08:03:48 +0000264 /* we only have to deal with one */
Peter Hinz56885f32011-03-02 22:03:47 +0000265 n = context->fds_count;
Andy Green4b6fbe12011-02-14 08:03:48 +0000266 }
267
268 /* remove also from external POLL support via protocol 0 */
269
Peter Hinz56885f32011-03-02 22:03:47 +0000270 context->protocols[0].callback(context, wsi,
Andy Green4b6fbe12011-02-14 08:03:48 +0000271 LWS_CALLBACK_DEL_POLL_FD, (void *)(long)wsi->sock, NULL, 0);
272
Andy Green251f6fa2010-11-03 11:13:06 +0000273 wsi->state = WSI_STATE_DEAD_SOCKET;
274
Andy Green4b6fbe12011-02-14 08:03:48 +0000275 /* tell the user it's all over for this guy */
276
Andy Greend4302732011-02-28 07:45:29 +0000277 if (wsi->protocol && wsi->protocol->callback &&
278 old_state == WSI_STATE_ESTABLISHED)
Peter Hinz56885f32011-03-02 22:03:47 +0000279 wsi->protocol->callback(context, wsi, LWS_CALLBACK_CLOSED,
Andy Greene77ddd82010-11-13 10:03:47 +0000280 wsi->user_space, NULL, 0);
Andy Green251f6fa2010-11-03 11:13:06 +0000281
Andy Greenef660a92011-03-06 10:29:38 +0000282 /* deallocate any active extension contexts */
283
284 for (n = 0; n < wsi->count_active_extensions; n++) {
285 if (!wsi->active_extensions[n]->callback)
286 continue;
287
288 wsi->active_extensions[n]->callback(context, wsi,
289 LWS_EXT_CALLBACK_DESTROY,
290 wsi->active_extensions_user[n], NULL, 0);
291
292 free(wsi->active_extensions_user[n]);
293 }
294
295 /* free up his parsing allocations */
Andy Green4b6fbe12011-02-14 08:03:48 +0000296
Andy Green251f6fa2010-11-03 11:13:06 +0000297 for (n = 0; n < WSI_TOKEN_COUNT; n++)
298 if (wsi->utf8_token[n].token)
299 free(wsi->utf8_token[n].token);
300
Andy Green0ca6a172010-12-19 20:50:01 +0000301/* fprintf(stderr, "closing fd=%d\n", wsi->sock); */
Andy Green251f6fa2010-11-03 11:13:06 +0000302
Andy Green3faa9c72010-11-08 17:03:03 +0000303#ifdef LWS_OPENSSL_SUPPORT
Andy Green90c7cbc2011-01-27 06:26:52 +0000304 if (wsi->ssl) {
Andy Green3faa9c72010-11-08 17:03:03 +0000305 n = SSL_get_fd(wsi->ssl);
306 SSL_shutdown(wsi->ssl);
Peter Hinz56885f32011-03-02 22:03:47 +0000307#ifdef WIN32
308 closesocket(n);
309#else
Andy Green3faa9c72010-11-08 17:03:03 +0000310 close(n);
Peter Hinz56885f32011-03-02 22:03:47 +0000311#endif
Andy Green3faa9c72010-11-08 17:03:03 +0000312 SSL_free(wsi->ssl);
313 } else {
314#endif
315 shutdown(wsi->sock, SHUT_RDWR);
Peter Hinz56885f32011-03-02 22:03:47 +0000316#ifdef WIN32
317 closesocket(wsi->sock);
318#else
Andy Green3faa9c72010-11-08 17:03:03 +0000319 close(wsi->sock);
Peter Hinz56885f32011-03-02 22:03:47 +0000320#endif
Andy Green3faa9c72010-11-08 17:03:03 +0000321#ifdef LWS_OPENSSL_SUPPORT
322 }
323#endif
Andy Green4f3943a2010-11-12 10:44:16 +0000324 if (wsi->user_space)
325 free(wsi->user_space);
326
Andy Green251f6fa2010-11-03 11:13:06 +0000327 free(wsi);
328}
329
Andy Green07034092011-02-13 08:37:12 +0000330/**
Andy Greenf7ee5492011-02-13 09:04:21 +0000331 * libwebsockets_hangup_on_client() - Server calls to terminate client
332 * connection
Peter Hinz56885f32011-03-02 22:03:47 +0000333 * @context: libwebsockets context
Andy Greenf7ee5492011-02-13 09:04:21 +0000334 * @fd: Connection socket descriptor
335 */
336
337void
Peter Hinz56885f32011-03-02 22:03:47 +0000338libwebsockets_hangup_on_client(struct libwebsocket_context *context, int fd)
Andy Greenf7ee5492011-02-13 09:04:21 +0000339{
Peter Hinz56885f32011-03-02 22:03:47 +0000340 struct libwebsocket *wsi = wsi_from_fd(context, fd);
Andy Greenf7ee5492011-02-13 09:04:21 +0000341
342 if (wsi == NULL)
343 return;
344
Peter Hinz56885f32011-03-02 22:03:47 +0000345 libwebsocket_close_and_free_session(context, wsi,
Andy Green6da560c2011-02-26 11:06:27 +0000346 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenf7ee5492011-02-13 09:04:21 +0000347}
348
349
350/**
Andy Green07034092011-02-13 08:37:12 +0000351 * libwebsockets_get_peer_addresses() - Get client address information
352 * @fd: Connection socket descriptor
353 * @name: Buffer to take client address name
354 * @name_len: Length of client address name buffer
355 * @rip: Buffer to take client address IP qotted quad
356 * @rip_len: Length of client address IP buffer
357 *
358 * This function fills in @name and @rip with the name and IP of
359 * the client connected with socket descriptor @fd. Names may be
360 * truncated if there is not enough room. If either cannot be
361 * determined, they will be returned as valid zero-length strings.
362 */
363
364void
365libwebsockets_get_peer_addresses(int fd, char *name, int name_len,
366 char *rip, int rip_len)
367{
368 unsigned int len;
369 struct sockaddr_in sin;
370 struct hostent *host;
371 struct hostent *host1;
372 char ip[128];
Andy Greenf92def72011-03-09 15:02:20 +0000373 unsigned char *p;
Andy Green07034092011-02-13 08:37:12 +0000374 int n;
Andy Green7627af52011-03-09 15:13:52 +0000375 struct sockaddr_un *un;
Andy Green07034092011-02-13 08:37:12 +0000376
377 rip[0] = '\0';
378 name[0] = '\0';
379
380 len = sizeof sin;
381 if (getpeername(fd, (struct sockaddr *) &sin, &len) < 0) {
382 perror("getpeername");
383 return;
384 }
385
386 host = gethostbyaddr((char *) &sin.sin_addr, sizeof sin.sin_addr,
387 AF_INET);
388 if (host == NULL) {
389 perror("gethostbyaddr");
390 return;
391 }
392
393 strncpy(name, host->h_name, name_len);
394 name[name_len - 1] = '\0';
395
396 host1 = gethostbyname(host->h_name);
397 if (host1 == NULL)
398 return;
Andy Greenf92def72011-03-09 15:02:20 +0000399 p = (unsigned char *)host1;
Andy Green07034092011-02-13 08:37:12 +0000400 n = 0;
401 while (p != NULL) {
Andy Greenf92def72011-03-09 15:02:20 +0000402 p = (unsigned char *)host1->h_addr_list[n++];
Andy Green07034092011-02-13 08:37:12 +0000403 if (p == NULL)
404 continue;
Peter Hinzbb45a902011-03-10 18:14:01 +0000405 if ((host1->h_addrtype != AF_INET)
406#ifdef AF_LOCAL
407 && (host1->h_addrtype != AF_LOCAL)
408#endif
409 )
Andy Green07034092011-02-13 08:37:12 +0000410 continue;
411
Andy Green7627af52011-03-09 15:13:52 +0000412 if (host1->h_addrtype == AF_INET)
413 sprintf(ip, "%u.%u.%u.%u", p[0], p[1], p[2], p[3]);
Peter Hinzbb45a902011-03-10 18:14:01 +0000414#ifdef AF_LOCAL
Andy Green7627af52011-03-09 15:13:52 +0000415 else {
416 un = (struct sockaddr_un *)p;
417 strncpy(ip, un->sun_path, sizeof(ip) -1);
418 ip[sizeof(ip) - 1] = '\0';
419 }
Peter Hinzbb45a902011-03-10 18:14:01 +0000420#endif
Andy Green07034092011-02-13 08:37:12 +0000421 p = NULL;
422 strncpy(rip, ip, rip_len);
423 rip[rip_len - 1] = '\0';
424 }
425}
Andy Green9f990342011-02-12 11:57:45 +0000426
Peter Hinz56885f32011-03-02 22:03:47 +0000427int libwebsockets_get_random(struct libwebsocket_context *context,
428 void *buf, int len)
429{
430 int n;
431 char *p = buf;
432
433#ifdef WIN32
434 for (n = 0; n < len; n++)
435 p[n] = (unsigned char)rand();
436#else
437 n = read(context->fd_random, p, len);
438#endif
439
440 return n;
441}
442
Andy Green2836c642011-03-07 20:47:41 +0000443unsigned char *
444libwebsockets_SHA1(const unsigned char *d, size_t n, unsigned char *md)
445{
446 return SHA1(d, n, md);
447}
448
Andy Greeneeaacb32011-03-01 20:44:24 +0000449void libwebsockets_00_spaceout(char *key, int spaces, int seed)
450{
451 char *p;
Peter Hinz56885f32011-03-02 22:03:47 +0000452
Andy Greeneeaacb32011-03-01 20:44:24 +0000453 key++;
454 while (spaces--) {
455 if (*key && (seed & 1))
456 key++;
457 seed >>= 1;
Peter Hinz56885f32011-03-02 22:03:47 +0000458
Andy Greeneeaacb32011-03-01 20:44:24 +0000459 p = key + strlen(key);
460 while (p >= key) {
461 p[1] = p[0];
462 p--;
463 }
464 *key++ = ' ';
465 }
466}
467
468void libwebsockets_00_spam(char *key, int count, int seed)
469{
470 char *p;
471
472 key++;
473 while (count--) {
474
475 if (*key && (seed & 1))
476 key++;
477 seed >>= 1;
478
479 p = key + strlen(key);
480 while (p >= key) {
481 p[1] = p[0];
482 p--;
483 }
484 *key++ = 0x21 + ((seed & 0xffff) % 15);
485 /* 4 would use it up too fast.. not like it matters */
486 seed >>= 1;
487 }
488}
489
Andy Green95a7b5d2011-03-06 10:29:39 +0000490int lws_send_pipe_choked(struct libwebsocket *wsi)
491{
492 struct pollfd fds;
493
494 fds.fd = wsi->sock;
495 fds.events = POLLOUT;
496 fds.revents = 0;
497
498 if (poll(&fds, 1, 0) != 1)
499 return 1;
500
501 if ((fds.revents & POLLOUT) == 0)
502 return 1;
503
504 /* okay to send another packet without blocking */
505
506 return 0;
507}
508
Andy Green3b84c002011-03-06 13:14:42 +0000509static int
510lws_handle_POLLOUT_event(struct libwebsocket_context *context,
511 struct libwebsocket *wsi, struct pollfd *pollfd)
512{
513 struct lws_tokens eff_buf;
514 int n;
515 int ret;
516 int m;
517
518 if (!wsi->extension_data_pending)
519 goto user_service;
520
521 /*
522 * check in on the active extensions, see if they
523 * had pending stuff to spill... they need to get the
524 * first look-in otherwise sequence will be disordered
525 *
526 * NULL, zero-length eff_buf means just spill pending
527 */
528
529 ret = 1;
530 while (ret == 1) {
531
532 /* default to nobody has more to spill */
533
534 ret = 0;
535 eff_buf.token = NULL;
536 eff_buf.token_len = 0;
537
538 /* give every extension a chance to spill */
539
540 for (n = 0; n < wsi->count_active_extensions; n++) {
541 m = wsi->active_extensions[n]->callback(
542 wsi->protocol->owning_server, wsi,
543 LWS_EXT_CALLBACK_PACKET_TX_PRESEND,
544 wsi->active_extensions_user[n], &eff_buf, 0);
545 if (m < 0) {
546 fprintf(stderr, "extension reports fatal error\n");
547 return -1;
548 }
549 if (m)
550 /*
551 * at least one extension told us he has more
552 * to spill, so we will go around again after
553 */
554 ret = 1;
555 }
556
557 /* assuming they gave us something to send, send it */
558
559 if (eff_buf.token_len) {
560 if (lws_issue_raw(wsi, (unsigned char *)eff_buf.token,
561 eff_buf.token_len))
562 return -1;
563 } else
564 continue;
565
566 /* no extension has more to spill */
567
568 if (!ret)
569 continue;
570
571 /*
572 * There's more to spill from an extension, but we just sent
573 * something... did that leave the pipe choked?
574 */
575
576 if (!lws_send_pipe_choked(wsi))
577 /* no we could add more */
578 continue;
579
580 fprintf(stderr, "choked in POLLOUT service\n");
581
582 /*
583 * Yes, he's choked. Leave the POLLOUT masked on so we will
584 * come back here when he is unchoked. Don't call the user
585 * callback to enforce ordering of spilling, he'll get called
586 * when we come back here and there's nothing more to spill.
587 */
588
589 return 0;
590 }
591
592 wsi->extension_data_pending = 0;
593
594user_service:
595 /* one shot */
596
597 pollfd->events &= ~POLLOUT;
598
599 /* external POLL support via protocol 0 */
600 context->protocols[0].callback(context, wsi,
601 LWS_CALLBACK_CLEAR_MODE_POLL_FD,
602 (void *)(long)wsi->sock, NULL, POLLOUT);
603
Andy Green9e4c2b62011-03-07 20:47:39 +0000604 if (wsi->mode == LWS_CONNMODE_WS_CLIENT)
605 n = LWS_CALLBACK_CLIENT_WRITEABLE;
606 else
607 n = LWS_CALLBACK_SERVER_WRITEABLE;
608
609 wsi->protocol->callback(context, wsi, n, wsi->user_space, NULL, 0);
Andy Green3b84c002011-03-06 13:14:42 +0000610
611 return 0;
612}
613
614
615
Andy Green9f990342011-02-12 11:57:45 +0000616/**
617 * libwebsocket_service_fd() - Service polled socket with something waiting
Peter Hinz56885f32011-03-02 22:03:47 +0000618 * @context: Websocket context
Andy Green9f990342011-02-12 11:57:45 +0000619 * @pollfd: The pollfd entry describing the socket fd and which events
620 * happened.
621 *
622 * This function closes any active connections and then frees the
623 * context. After calling this, any further use of the context is
624 * undefined.
625 */
626
627int
Peter Hinz56885f32011-03-02 22:03:47 +0000628libwebsocket_service_fd(struct libwebsocket_context *context,
Andy Green0d338332011-02-12 11:57:43 +0000629 struct pollfd *pollfd)
Andy Greenb45993c2010-12-18 15:13:50 +0000630{
Andy Green3b84c002011-03-06 13:14:42 +0000631 unsigned char buf[LWS_SEND_BUFFER_PRE_PADDING + 1 + MAX_BROADCAST_PAYLOAD +
Andy Greenb45993c2010-12-18 15:13:50 +0000632 LWS_SEND_BUFFER_POST_PADDING];
Andy Greena71eafc2011-02-14 17:59:43 +0000633 struct libwebsocket *wsi;
Andy Green0d338332011-02-12 11:57:43 +0000634 struct libwebsocket *new_wsi;
Andy Greenb45993c2010-12-18 15:13:50 +0000635 int n;
Andy Green0d338332011-02-12 11:57:43 +0000636 int m;
Andy Greenb45993c2010-12-18 15:13:50 +0000637 size_t len;
Andy Green0d338332011-02-12 11:57:43 +0000638 int accept_fd;
639 unsigned int clilen;
640 struct sockaddr_in cli_addr;
Andy Greena71eafc2011-02-14 17:59:43 +0000641 struct timeval tv;
Andy Greenbe93fef2011-02-14 20:25:43 +0000642 static const char magic_websocket_guid[] =
643 "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
644 static const char magic_websocket_04_masking_guid[] =
645 "61AC5F19-FBBA-4540-B96F-6561F1AB40A8";
646 char hash[20];
647 char pkt[1024];
648 char *p = &pkt[0];
649 const char *pc;
Andy Green2366b1c2011-03-06 13:15:31 +0000650 const char *c;
651 int more = 1;
Andy Greenbe93fef2011-02-14 20:25:43 +0000652 int okay = 0;
Andy Green2366b1c2011-03-06 13:15:31 +0000653 char ext_name[128];
Andy Green98a717c2011-03-06 13:14:15 +0000654 struct lws_tokens eff_buf;
Andy Greenc6517fa2011-03-06 13:15:29 +0000655 int ext_count = 0;
656 struct libwebsocket_extension *ext;
Andy Green6c939552011-03-08 08:56:57 +0000657 int opt = 1;
Andy Greenc6517fa2011-03-06 13:15:29 +0000658
Andy Greenbe93fef2011-02-14 20:25:43 +0000659#ifdef LWS_OPENSSL_SUPPORT
660 char ssl_err_buf[512];
661#endif
Andy Greena71eafc2011-02-14 17:59:43 +0000662 /*
663 * you can call us with pollfd = NULL to just allow the once-per-second
664 * global timeout checks; if less than a second since the last check
665 * it returns immediately then.
666 */
667
668 gettimeofday(&tv, NULL);
669
Peter Hinz56885f32011-03-02 22:03:47 +0000670 if (context->last_timeout_check_s != tv.tv_sec) {
671 context->last_timeout_check_s = tv.tv_sec;
Andy Greena71eafc2011-02-14 17:59:43 +0000672
673 /* global timeout check once per second */
674
Peter Hinz56885f32011-03-02 22:03:47 +0000675 for (n = 0; n < context->fds_count; n++) {
676 wsi = wsi_from_fd(context, context->fds[n].fd);
Andy Greena71eafc2011-02-14 17:59:43 +0000677 if (!wsi->pending_timeout)
678 continue;
679
680 /*
681 * if we went beyond the allowed time, kill the
682 * connection
683 */
684
Andy Greenda527df2011-03-07 07:08:12 +0000685 if (tv.tv_sec > wsi->pending_timeout_limit) {
686 fprintf(stderr, "TIMEDOUT WAITING\n");
Peter Hinz56885f32011-03-02 22:03:47 +0000687 libwebsocket_close_and_free_session(context,
688 wsi, LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenda527df2011-03-07 07:08:12 +0000689 }
Andy Greena71eafc2011-02-14 17:59:43 +0000690 }
691 }
692
693 /* just here for timeout management? */
694
695 if (pollfd == NULL)
696 return 0;
697
698 /* no, here to service a socket descriptor */
699
Peter Hinz56885f32011-03-02 22:03:47 +0000700 wsi = wsi_from_fd(context, pollfd->fd);
Andy Greenb45993c2010-12-18 15:13:50 +0000701
Andy Green0d338332011-02-12 11:57:43 +0000702 if (wsi == NULL)
703 return 1;
Andy Green8f037e42010-12-19 22:13:26 +0000704
Andy Green0d338332011-02-12 11:57:43 +0000705 switch (wsi->mode) {
706 case LWS_CONNMODE_SERVER_LISTENER:
707
708 /* pollin means a client has connected to us then */
709
710 if (!pollfd->revents & POLLIN)
711 break;
712
713 /* listen socket got an unencrypted connection... */
714
715 clilen = sizeof(cli_addr);
716 accept_fd = accept(pollfd->fd, (struct sockaddr *)&cli_addr,
717 &clilen);
718 if (accept_fd < 0) {
719 fprintf(stderr, "ERROR on accept");
720 break;
721 }
722
Andy Green6c939552011-03-08 08:56:57 +0000723 /* Disable Nagle */
724 opt = 1;
725 setsockopt(accept_fd, SOL_TCP, TCP_NODELAY, &opt, sizeof(opt));
726
Peter Hinz56885f32011-03-02 22:03:47 +0000727 if (context->fds_count >= MAX_CLIENTS) {
Andy Green3221f922011-02-12 13:14:11 +0000728 fprintf(stderr, "too busy to accept new client\n");
Peter Hinz56885f32011-03-02 22:03:47 +0000729#ifdef WIN32
730 closesocket(accept_fd);
731#else
Andy Green0d338332011-02-12 11:57:43 +0000732 close(accept_fd);
Peter Hinz56885f32011-03-02 22:03:47 +0000733#endif
Andy Green0d338332011-02-12 11:57:43 +0000734 break;
735 }
736
Andy Green07034092011-02-13 08:37:12 +0000737 /*
738 * look at who we connected to and give user code a chance
739 * to reject based on client IP. There's no protocol selected
740 * yet so we issue this to protocols[0]
741 */
742
Peter Hinz56885f32011-03-02 22:03:47 +0000743 if ((context->protocols[0].callback)(context, wsi,
Andy Green07034092011-02-13 08:37:12 +0000744 LWS_CALLBACK_FILTER_NETWORK_CONNECTION,
745 (void*)(long)accept_fd, NULL, 0)) {
746 fprintf(stderr, "Callback denied network connection\n");
Peter Hinz56885f32011-03-02 22:03:47 +0000747#ifdef WIN32
748 closesocket(accept_fd);
749#else
Andy Green07034092011-02-13 08:37:12 +0000750 close(accept_fd);
Peter Hinz56885f32011-03-02 22:03:47 +0000751#endif
Andy Green07034092011-02-13 08:37:12 +0000752 break;
753 }
754
Andy Green0d338332011-02-12 11:57:43 +0000755 /* accepting connection to main listener */
756
757 new_wsi = malloc(sizeof(struct libwebsocket));
758 if (new_wsi == NULL) {
759 fprintf(stderr, "Out of memory for new connection\n");
760 break;
761 }
762
763 memset(new_wsi, 0, sizeof (struct libwebsocket));
764 new_wsi->sock = accept_fd;
Andy Greend6e09112011-03-05 16:12:15 +0000765 new_wsi->count_active_extensions = 0;
Andy Greena71eafc2011-02-14 17:59:43 +0000766 new_wsi->pending_timeout = NO_PENDING_TIMEOUT;
Andy Green0d338332011-02-12 11:57:43 +0000767
768#ifdef LWS_OPENSSL_SUPPORT
769 new_wsi->ssl = NULL;
Andy Green0d338332011-02-12 11:57:43 +0000770
Peter Hinz56885f32011-03-02 22:03:47 +0000771 if (context->use_ssl) {
Andy Green0d338332011-02-12 11:57:43 +0000772
Peter Hinz56885f32011-03-02 22:03:47 +0000773 new_wsi->ssl = SSL_new(context->ssl_ctx);
Andy Green0d338332011-02-12 11:57:43 +0000774 if (new_wsi->ssl == NULL) {
775 fprintf(stderr, "SSL_new failed: %s\n",
776 ERR_error_string(SSL_get_error(
777 new_wsi->ssl, 0), NULL));
Andy Green1f9bf522011-02-14 21:14:37 +0000778 libwebsockets_decode_ssl_error();
Andy Green0d338332011-02-12 11:57:43 +0000779 free(new_wsi);
780 break;
781 }
782
783 SSL_set_fd(new_wsi->ssl, accept_fd);
784
785 n = SSL_accept(new_wsi->ssl);
786 if (n != 1) {
787 /*
788 * browsers seem to probe with various
789 * ssl params which fail then retry
790 * and succeed
791 */
792 debug("SSL_accept failed skt %u: %s\n",
793 pollfd->fd,
794 ERR_error_string(SSL_get_error(
795 new_wsi->ssl, n), NULL));
796 SSL_free(
797 new_wsi->ssl);
798 free(new_wsi);
799 break;
800 }
Andy Greenc6bf2c22011-02-20 11:10:47 +0000801
Andy Green0d338332011-02-12 11:57:43 +0000802 debug("accepted new SSL conn "
803 "port %u on fd=%d SSL ver %s\n",
804 ntohs(cli_addr.sin_port), accept_fd,
805 SSL_get_version(new_wsi->ssl));
806
807 } else
808#endif
809 debug("accepted new conn port %u on fd=%d\n",
810 ntohs(cli_addr.sin_port), accept_fd);
811
812 /* intialize the instance struct */
813
814 new_wsi->state = WSI_STATE_HTTP;
815 new_wsi->name_buffer_pos = 0;
816 new_wsi->mode = LWS_CONNMODE_WS_SERVING;
817
818 for (n = 0; n < WSI_TOKEN_COUNT; n++) {
819 new_wsi->utf8_token[n].token = NULL;
820 new_wsi->utf8_token[n].token_len = 0;
821 }
822
823 /*
824 * these can only be set once the protocol is known
825 * we set an unestablished connection's protocol pointer
826 * to the start of the supported list, so it can look
827 * for matching ones during the handshake
828 */
Peter Hinz56885f32011-03-02 22:03:47 +0000829 new_wsi->protocol = context->protocols;
Andy Green0d338332011-02-12 11:57:43 +0000830 new_wsi->user_space = NULL;
831
832 /*
833 * Default protocol is 76 / 00
834 * After 76, there's a header specified to inform which
835 * draft the client wants, when that's seen we modify
836 * the individual connection's spec revision accordingly
837 */
838 new_wsi->ietf_spec_revision = 0;
839
Peter Hinz56885f32011-03-02 22:03:47 +0000840 insert_wsi(context, new_wsi);
Andy Green0d338332011-02-12 11:57:43 +0000841
Andy Green0d338332011-02-12 11:57:43 +0000842 /*
843 * make sure NO events are seen yet on this new socket
844 * (otherwise we inherit old fds[client].revents from
845 * previous socket there and die mysteriously! )
846 */
Peter Hinz56885f32011-03-02 22:03:47 +0000847 context->fds[context->fds_count].revents = 0;
Andy Green0d338332011-02-12 11:57:43 +0000848
Peter Hinz56885f32011-03-02 22:03:47 +0000849 context->fds[context->fds_count].events = POLLIN;
850 context->fds[context->fds_count++].fd = accept_fd;
Andy Green0d338332011-02-12 11:57:43 +0000851
Andy Green3221f922011-02-12 13:14:11 +0000852 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +0000853 context->protocols[0].callback(context, new_wsi,
Andy Green3221f922011-02-12 13:14:11 +0000854 LWS_CALLBACK_ADD_POLL_FD,
855 (void *)(long)accept_fd, NULL, POLLIN);
856
Andy Green0d338332011-02-12 11:57:43 +0000857 break;
858
859 case LWS_CONNMODE_BROADCAST_PROXY_LISTENER:
860
861 /* as we are listening, POLLIN means accept() is needed */
862
863 if (!pollfd->revents & POLLIN)
864 break;
865
866 /* listen socket got an unencrypted connection... */
867
868 clilen = sizeof(cli_addr);
869 accept_fd = accept(pollfd->fd, (struct sockaddr *)&cli_addr,
870 &clilen);
871 if (accept_fd < 0) {
872 fprintf(stderr, "ERROR on accept");
873 break;
874 }
875
Peter Hinz56885f32011-03-02 22:03:47 +0000876 if (context->fds_count >= MAX_CLIENTS) {
Andy Green3221f922011-02-12 13:14:11 +0000877 fprintf(stderr, "too busy to accept new broadcast "
878 "proxy client\n");
Peter Hinz56885f32011-03-02 22:03:47 +0000879#ifdef WIN32
880 closesocket(accept_fd);
881#else
Andy Green0d338332011-02-12 11:57:43 +0000882 close(accept_fd);
Peter Hinz56885f32011-03-02 22:03:47 +0000883#endif
Andy Green0d338332011-02-12 11:57:43 +0000884 break;
885 }
886
887 /* create a dummy wsi for the connection and add it */
888
889 new_wsi = malloc(sizeof(struct libwebsocket));
890 memset(new_wsi, 0, sizeof (struct libwebsocket));
891 new_wsi->sock = accept_fd;
892 new_wsi->mode = LWS_CONNMODE_BROADCAST_PROXY;
893 new_wsi->state = WSI_STATE_ESTABLISHED;
Andy Greend6e09112011-03-05 16:12:15 +0000894 new_wsi->count_active_extensions = 0;
Andy Green0d338332011-02-12 11:57:43 +0000895 /* note which protocol we are proxying */
896 new_wsi->protocol_index_for_broadcast_proxy =
897 wsi->protocol_index_for_broadcast_proxy;
Peter Hinz56885f32011-03-02 22:03:47 +0000898 insert_wsi(context, new_wsi);
Andy Green0d338332011-02-12 11:57:43 +0000899
900 /* add connected socket to internal poll array */
901
Peter Hinz56885f32011-03-02 22:03:47 +0000902 context->fds[context->fds_count].revents = 0;
903 context->fds[context->fds_count].events = POLLIN;
904 context->fds[context->fds_count++].fd = accept_fd;
Andy Green0d338332011-02-12 11:57:43 +0000905
Andy Green3221f922011-02-12 13:14:11 +0000906 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +0000907 context->protocols[0].callback(context, new_wsi,
Andy Green3221f922011-02-12 13:14:11 +0000908 LWS_CALLBACK_ADD_POLL_FD,
909 (void *)(long)accept_fd, NULL, POLLIN);
910
Andy Green0d338332011-02-12 11:57:43 +0000911 break;
912
913 case LWS_CONNMODE_BROADCAST_PROXY:
Andy Green8f037e42010-12-19 22:13:26 +0000914
Andy Greenb45993c2010-12-18 15:13:50 +0000915 /* handle session socket closed */
Andy Green8f037e42010-12-19 22:13:26 +0000916
Andy Green0d338332011-02-12 11:57:43 +0000917 if (pollfd->revents & (POLLERR | POLLHUP)) {
Andy Green8f037e42010-12-19 22:13:26 +0000918
Andy Green0d338332011-02-12 11:57:43 +0000919 debug("Session Socket %p (fd=%d) dead\n",
Timothy J Fontaineb86d64e2011-02-14 17:55:27 +0000920 (void *)wsi, pollfd->fd);
Andy Greenb45993c2010-12-18 15:13:50 +0000921
Peter Hinz56885f32011-03-02 22:03:47 +0000922 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +0000923 LWS_CLOSE_STATUS_NORMAL);
Andy Green4b6fbe12011-02-14 08:03:48 +0000924 return 1;
Andy Greenb45993c2010-12-18 15:13:50 +0000925 }
Andy Green8f037e42010-12-19 22:13:26 +0000926
Andy Green3b84c002011-03-06 13:14:42 +0000927 /*
928 * either extension code with stuff to spill, or the user code,
929 * requested a callback when it was OK to write
930 */
Andy Green90c7cbc2011-01-27 06:26:52 +0000931
Andy Green3b84c002011-03-06 13:14:42 +0000932 if (pollfd->revents & POLLOUT)
933 if (lws_handle_POLLOUT_event(context, wsi, pollfd) < 0) {
934 libwebsocket_close_and_free_session(context, wsi,
935 LWS_CLOSE_STATUS_NORMAL);
936 return 1;
937 }
Andy Green90c7cbc2011-01-27 06:26:52 +0000938
Andy Greenb45993c2010-12-18 15:13:50 +0000939 /* any incoming data ready? */
940
Andy Green0d338332011-02-12 11:57:43 +0000941 if (!(pollfd->revents & POLLIN))
942 break;
Andy Greenb45993c2010-12-18 15:13:50 +0000943
Andy Green0d338332011-02-12 11:57:43 +0000944 /* get the issued broadcast payload from the socket */
Andy Greenb45993c2010-12-18 15:13:50 +0000945
Andy Green0d338332011-02-12 11:57:43 +0000946 len = read(pollfd->fd, buf + LWS_SEND_BUFFER_PRE_PADDING,
947 MAX_BROADCAST_PAYLOAD);
948 if (len < 0) {
949 fprintf(stderr, "Error reading broadcast payload\n");
Andy Green4b6fbe12011-02-14 08:03:48 +0000950 break;
Andy Green0d338332011-02-12 11:57:43 +0000951 }
Andy Greenb45993c2010-12-18 15:13:50 +0000952
Andy Green0d338332011-02-12 11:57:43 +0000953 /* broadcast it to all guys with this protocol index */
Andy Green8f037e42010-12-19 22:13:26 +0000954
Andy Green0d338332011-02-12 11:57:43 +0000955 for (n = 0; n < FD_HASHTABLE_MODULUS; n++) {
Andy Green8f037e42010-12-19 22:13:26 +0000956
Peter Hinz56885f32011-03-02 22:03:47 +0000957 for (m = 0; m < context->fd_hashtable[n].length; m++) {
Andy Greenb45993c2010-12-18 15:13:50 +0000958
Peter Hinz56885f32011-03-02 22:03:47 +0000959 new_wsi = context->fd_hashtable[n].wsi[m];
Andy Greenb45993c2010-12-18 15:13:50 +0000960
Andy Green0d338332011-02-12 11:57:43 +0000961 /* only to clients we are serving to */
Andy Greenb45993c2010-12-18 15:13:50 +0000962
Andy Green0d338332011-02-12 11:57:43 +0000963 if (new_wsi->mode != LWS_CONNMODE_WS_SERVING)
Andy Greenb45993c2010-12-18 15:13:50 +0000964 continue;
965
966 /*
967 * never broadcast to non-established
968 * connection
969 */
970
Andy Green0d338332011-02-12 11:57:43 +0000971 if (new_wsi->state != WSI_STATE_ESTABLISHED)
Andy Green4739e5c2011-01-22 12:51:57 +0000972 continue;
973
Andy Greenb45993c2010-12-18 15:13:50 +0000974 /*
975 * only broadcast to connections using
976 * the requested protocol
977 */
978
Andy Green0d338332011-02-12 11:57:43 +0000979 if (new_wsi->protocol->protocol_index !=
980 wsi->protocol_index_for_broadcast_proxy)
Andy Greenb45993c2010-12-18 15:13:50 +0000981 continue;
982
Andy Green8f037e42010-12-19 22:13:26 +0000983 /* broadcast it to this connection */
984
Peter Hinz56885f32011-03-02 22:03:47 +0000985 new_wsi->protocol->callback(context, new_wsi,
Andy Green8f037e42010-12-19 22:13:26 +0000986 LWS_CALLBACK_BROADCAST,
Andy Green0d338332011-02-12 11:57:43 +0000987 new_wsi->user_space,
Andy Green0ca6a172010-12-19 20:50:01 +0000988 buf + LWS_SEND_BUFFER_PRE_PADDING, len);
Andy Greenb45993c2010-12-18 15:13:50 +0000989 }
Andy Green0d338332011-02-12 11:57:43 +0000990 }
991 break;
Andy Greenb45993c2010-12-18 15:13:50 +0000992
Andy Greenbe93fef2011-02-14 20:25:43 +0000993 case LWS_CONNMODE_WS_CLIENT_WAITING_PROXY_REPLY:
994
995 /* handle proxy hung up on us */
996
997 if (pollfd->revents & (POLLERR | POLLHUP)) {
998
999 fprintf(stderr, "Proxy connection %p (fd=%d) dead\n",
1000 (void *)wsi, pollfd->fd);
1001
Peter Hinz56885f32011-03-02 22:03:47 +00001002 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001003 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001004 return 1;
1005 }
1006
1007 n = recv(wsi->sock, pkt, sizeof pkt, 0);
1008 if (n < 0) {
Peter Hinz56885f32011-03-02 22:03:47 +00001009 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001010 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001011 fprintf(stderr, "ERROR reading from proxy socket\n");
1012 return 1;
1013 }
1014
1015 pkt[13] = '\0';
1016 if (strcmp(pkt, "HTTP/1.0 200 ") != 0) {
Peter Hinz56885f32011-03-02 22:03:47 +00001017 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001018 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001019 fprintf(stderr, "ERROR from proxy: %s\n", pkt);
1020 return 1;
1021 }
1022
1023 /* clear his proxy connection timeout */
1024
1025 libwebsocket_set_timeout(wsi, NO_PENDING_TIMEOUT, 0);
1026
1027 /* fallthru */
1028
1029 case LWS_CONNMODE_WS_CLIENT_ISSUE_HANDSHAKE:
1030
1031 #ifdef LWS_OPENSSL_SUPPORT
1032 if (wsi->use_ssl) {
1033
Peter Hinz56885f32011-03-02 22:03:47 +00001034 wsi->ssl = SSL_new(context->ssl_client_ctx);
1035 wsi->client_bio = BIO_new_socket(wsi->sock,
1036 BIO_NOCLOSE);
Andy Greenbe93fef2011-02-14 20:25:43 +00001037 SSL_set_bio(wsi->ssl, wsi->client_bio, wsi->client_bio);
1038
Andy Green6901cb32011-02-21 08:06:47 +00001039 SSL_set_ex_data(wsi->ssl,
Andy Green2e24da02011-03-05 16:12:04 +00001040 openssl_websocket_private_data_index,
Peter Hinz56885f32011-03-02 22:03:47 +00001041 context);
Andy Green6901cb32011-02-21 08:06:47 +00001042
Andy Greenbe93fef2011-02-14 20:25:43 +00001043 if (SSL_connect(wsi->ssl) <= 0) {
1044 fprintf(stderr, "SSL connect error %s\n",
Andy Green687b0182011-02-26 11:04:01 +00001045 ERR_error_string(ERR_get_error(),
1046 ssl_err_buf));
Peter Hinz56885f32011-03-02 22:03:47 +00001047 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001048 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001049 return 1;
1050 }
1051
1052 n = SSL_get_verify_result(wsi->ssl);
Andy Green2e24da02011-03-05 16:12:04 +00001053 if ((n != X509_V_OK) && (
Andy Green687b0182011-02-26 11:04:01 +00001054 n != X509_V_ERR_DEPTH_ZERO_SELF_SIGNED_CERT ||
1055 wsi->use_ssl != 2)) {
Andy Greenbe93fef2011-02-14 20:25:43 +00001056
Andy Green687b0182011-02-26 11:04:01 +00001057 fprintf(stderr, "server's cert didn't "
1058 "look good %d\n", n);
Peter Hinz56885f32011-03-02 22:03:47 +00001059 libwebsocket_close_and_free_session(context,
1060 wsi, LWS_CLOSE_STATUS_NOSTATUS);
Andy Green687b0182011-02-26 11:04:01 +00001061 return 1;
Andy Greenbe93fef2011-02-14 20:25:43 +00001062 }
1063 } else {
1064 wsi->ssl = NULL;
1065 #endif
1066
1067
1068 #ifdef LWS_OPENSSL_SUPPORT
1069 }
1070 #endif
1071
1072 /*
1073 * create the random key
1074 */
1075
Peter Hinz56885f32011-03-02 22:03:47 +00001076 n = libwebsockets_get_random(context, hash, 16);
Andy Greenbe93fef2011-02-14 20:25:43 +00001077 if (n != 16) {
1078 fprintf(stderr, "Unable to read from random dev %s\n",
1079 SYSTEM_RANDOM_FILEPATH);
1080 free(wsi->c_path);
1081 free(wsi->c_host);
Andy Green08d33922011-02-26 10:22:49 +00001082 if (wsi->c_origin)
1083 free(wsi->c_origin);
Andy Greenbe93fef2011-02-14 20:25:43 +00001084 if (wsi->c_protocol)
1085 free(wsi->c_protocol);
Peter Hinz56885f32011-03-02 22:03:47 +00001086 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001087 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001088 return 1;
1089 }
1090
1091 lws_b64_encode_string(hash, 16, wsi->key_b64,
1092 sizeof wsi->key_b64);
1093
1094 /*
Andy Greeneeaacb32011-03-01 20:44:24 +00001095 * 00 example client handshake
1096 *
1097 * GET /socket.io/websocket HTTP/1.1
1098 * Upgrade: WebSocket
1099 * Connection: Upgrade
1100 * Host: 127.0.0.1:9999
1101 * Origin: http://127.0.0.1
1102 * Sec-WebSocket-Key1: 1 0 2#0W 9 89 7 92 ^
1103 * Sec-WebSocket-Key2: 7 7Y 4328 B2v[8(z1
1104 * Cookie: socketio=websocket
1105 *
1106 * (Á®Ä0¶†≥
1107 *
Andy Greenbe93fef2011-02-14 20:25:43 +00001108 * 04 example client handshake
1109 *
1110 * GET /chat HTTP/1.1
1111 * Host: server.example.com
1112 * Upgrade: websocket
1113 * Connection: Upgrade
1114 * Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==
1115 * Sec-WebSocket-Origin: http://example.com
1116 * Sec-WebSocket-Protocol: chat, superchat
1117 * Sec-WebSocket-Version: 4
1118 */
1119
Andy Green08d33922011-02-26 10:22:49 +00001120 p += sprintf(p, "GET %s HTTP/1.1\x0d\x0a", wsi->c_path);
Andy Greeneeaacb32011-03-01 20:44:24 +00001121
1122 if (wsi->ietf_spec_revision == 0) {
1123 unsigned char spaces_1, spaces_2;
1124 unsigned int max_1, max_2;
1125 unsigned int num_1, num_2;
1126 unsigned long product_1, product_2;
1127 char key_1[40];
1128 char key_2[40];
1129 unsigned int seed;
1130 unsigned int count;
1131 char challenge[16];
1132
Peter Hinz56885f32011-03-02 22:03:47 +00001133 libwebsockets_get_random(context, &spaces_1,
1134 sizeof(char));
1135 libwebsockets_get_random(context, &spaces_2,
1136 sizeof(char));
1137
Andy Greeneeaacb32011-03-01 20:44:24 +00001138 spaces_1 = (spaces_1 % 12) + 1;
1139 spaces_2 = (spaces_2 % 12) + 1;
Peter Hinz56885f32011-03-02 22:03:47 +00001140
Andy Greeneeaacb32011-03-01 20:44:24 +00001141 max_1 = 4294967295 / spaces_1;
1142 max_2 = 4294967295 / spaces_2;
1143
Peter Hinz56885f32011-03-02 22:03:47 +00001144 libwebsockets_get_random(context, &num_1, sizeof(int));
1145 libwebsockets_get_random(context, &num_2, sizeof(int));
1146
Andy Greeneeaacb32011-03-01 20:44:24 +00001147 num_1 = (num_1 % max_1);
1148 num_2 = (num_2 % max_2);
Peter Hinz56885f32011-03-02 22:03:47 +00001149
Andy Greeneeaacb32011-03-01 20:44:24 +00001150 challenge[0] = num_1 >> 24;
1151 challenge[1] = num_1 >> 16;
1152 challenge[2] = num_1 >> 8;
1153 challenge[3] = num_1;
1154 challenge[4] = num_2 >> 24;
1155 challenge[5] = num_2 >> 16;
1156 challenge[6] = num_2 >> 8;
1157 challenge[7] = num_2;
Peter Hinz56885f32011-03-02 22:03:47 +00001158
Andy Greeneeaacb32011-03-01 20:44:24 +00001159 product_1 = num_1 * spaces_1;
1160 product_2 = num_2 * spaces_2;
Peter Hinz56885f32011-03-02 22:03:47 +00001161
Andy Greeneeaacb32011-03-01 20:44:24 +00001162 sprintf(key_1, "%lu", product_1);
1163 sprintf(key_2, "%lu", product_2);
1164
Peter Hinz56885f32011-03-02 22:03:47 +00001165 libwebsockets_get_random(context, &seed, sizeof(int));
1166 libwebsockets_get_random(context, &count, sizeof(int));
1167
Andy Greeneeaacb32011-03-01 20:44:24 +00001168 libwebsockets_00_spam(key_1, (count % 12) + 1, seed);
Peter Hinz56885f32011-03-02 22:03:47 +00001169
1170 libwebsockets_get_random(context, &seed, sizeof(int));
1171 libwebsockets_get_random(context, &count, sizeof(int));
1172
Andy Greeneeaacb32011-03-01 20:44:24 +00001173 libwebsockets_00_spam(key_2, (count % 12) + 1, seed);
Peter Hinz56885f32011-03-02 22:03:47 +00001174
1175 libwebsockets_get_random(context, &seed, sizeof(int));
1176
Andy Greeneeaacb32011-03-01 20:44:24 +00001177 libwebsockets_00_spaceout(key_1, spaces_1, seed);
1178 libwebsockets_00_spaceout(key_2, spaces_2, seed >> 16);
Peter Hinz56885f32011-03-02 22:03:47 +00001179
Andy Greeneeaacb32011-03-01 20:44:24 +00001180 p += sprintf(p, "Upgrade: websocket\x0d\x0a"
1181 "Connection: Upgrade\x0d\x0aHost: %s\x0d\x0a",
Peter Hinz56885f32011-03-02 22:03:47 +00001182 wsi->c_host);
Andy Greeneeaacb32011-03-01 20:44:24 +00001183 if (wsi->c_origin)
1184 p += sprintf(p, "Origin: %s\x0d\x0a",
Peter Hinz56885f32011-03-02 22:03:47 +00001185 wsi->c_origin);
1186
Andy Greeneeaacb32011-03-01 20:44:24 +00001187 if (wsi->c_protocol)
Peter Hinz56885f32011-03-02 22:03:47 +00001188 p += sprintf(p, "Sec-WebSocket-Protocol: %s"
1189 "\x0d\x0a", wsi->c_protocol);
1190
Andy Greeneeaacb32011-03-01 20:44:24 +00001191 p += sprintf(p, "Sec-WebSocket-Key1: %s\x0d\x0a",
Peter Hinz56885f32011-03-02 22:03:47 +00001192 key_1);
Andy Greeneeaacb32011-03-01 20:44:24 +00001193 p += sprintf(p, "Sec-WebSocket-Key2: %s\x0d\x0a",
Peter Hinz56885f32011-03-02 22:03:47 +00001194 key_2);
Andy Greeneeaacb32011-03-01 20:44:24 +00001195
Andy Green385e7ad2011-03-01 21:06:02 +00001196 /* give userland a chance to append, eg, cookies */
Peter Hinz56885f32011-03-02 22:03:47 +00001197
1198 context->protocols[0].callback(context, wsi,
1199 LWS_CALLBACK_CLIENT_APPEND_HANDSHAKE_HEADER,
Andy Green385e7ad2011-03-01 21:06:02 +00001200 NULL, &p, (pkt + sizeof(pkt)) - p - 12);
1201
Andy Greeneeaacb32011-03-01 20:44:24 +00001202 p += sprintf(p, "\x0d\x0a");
Patrick McManus4bf91d72011-03-09 07:18:28 +00001203
1204 if (libwebsockets_get_random(context, p, 8) != 8)
1205 return -1;
Andy Greeneeaacb32011-03-01 20:44:24 +00001206 memcpy(&challenge[8], p, 8);
1207 p += 8;
Peter Hinz56885f32011-03-02 22:03:47 +00001208
Andy Greeneeaacb32011-03-01 20:44:24 +00001209 /* precompute what we want to see from the server */
Peter Hinz56885f32011-03-02 22:03:47 +00001210
Andy Greeneeaacb32011-03-01 20:44:24 +00001211 MD5((unsigned char *)challenge, 16,
1212 (unsigned char *)wsi->initial_handshake_hash_base64);
Peter Hinz56885f32011-03-02 22:03:47 +00001213
Andy Greeneeaacb32011-03-01 20:44:24 +00001214 goto issue_hdr;
1215 }
1216
Andy Green08d33922011-02-26 10:22:49 +00001217 p += sprintf(p, "Host: %s\x0d\x0a", wsi->c_host);
1218 p += sprintf(p, "Upgrade: websocket\x0d\x0a");
1219 p += sprintf(p, "Connection: Upgrade\x0d\x0a"
Andy Greenbe93fef2011-02-14 20:25:43 +00001220 "Sec-WebSocket-Key: ");
Andy Green08d33922011-02-26 10:22:49 +00001221 strcpy(p, wsi->key_b64);
1222 p += strlen(wsi->key_b64);
1223 p += sprintf(p, "\x0d\x0a");
1224 if (wsi->c_origin)
1225 p += sprintf(p, "Sec-WebSocket-Origin: %s\x0d\x0a",
Andy Greenbe93fef2011-02-14 20:25:43 +00001226 wsi->c_origin);
Andy Green08d33922011-02-26 10:22:49 +00001227 if (wsi->c_protocol)
Andy Greenbe93fef2011-02-14 20:25:43 +00001228 p += sprintf(p, "Sec-WebSocket-Protocol: %s\x0d\x0a",
1229 wsi->c_protocol);
Andy Greenc6517fa2011-03-06 13:15:29 +00001230
1231 /* tell the server what extensions we could support */
1232
1233 p += sprintf(p, "Sec-WebSocket-Extensions: ");
1234
1235 ext =context->extensions;
1236 while (ext && ext->callback) {
1237
1238 n = 0;
1239 n = context->protocols[0].callback(context, wsi,
1240 LWS_CALLBACK_CLIENT_CONFIRM_EXTENSION_SUPPORTED,
1241 wsi->user_space, (char *)ext->name, 0);
1242
1243 /*
1244 * zero return from callback means
1245 * go ahead and allow the extension,
1246 * it's what we get if the callback is
1247 * unhandled
1248 */
1249
1250 if (n) {
1251 ext++;
1252 continue;
1253 }
1254
1255 /* apply it */
1256
1257 if (ext_count)
1258 *p++ = ',';
Patrick McManus4bf91d72011-03-09 07:18:28 +00001259 p += sprintf(p, "%s", ext->name);
Andy Greenc6517fa2011-03-06 13:15:29 +00001260 ext_count++;
1261
1262 ext++;
1263 }
1264
1265 p += sprintf(p, "\x0d\x0a");
1266
1267 if (wsi->ietf_spec_revision)
1268 p += sprintf(p, "Sec-WebSocket-Version: %d\x0d\x0a",
1269 wsi->ietf_spec_revision);
1270
Andy Green385e7ad2011-03-01 21:06:02 +00001271 /* give userland a chance to append, eg, cookies */
Peter Hinz56885f32011-03-02 22:03:47 +00001272
1273 context->protocols[0].callback(context, wsi,
1274 LWS_CALLBACK_CLIENT_APPEND_HANDSHAKE_HEADER,
1275 NULL, &p, (pkt + sizeof(pkt)) - p - 12);
1276
Andy Green385e7ad2011-03-01 21:06:02 +00001277 p += sprintf(p, "\x0d\x0a");
1278
Andy Greenbe93fef2011-02-14 20:25:43 +00001279 /* prepare the expected server accept response */
1280
1281 strcpy((char *)buf, wsi->key_b64);
1282 strcpy((char *)&buf[strlen((char *)buf)], magic_websocket_guid);
1283
1284 SHA1(buf, strlen((char *)buf), (unsigned char *)hash);
1285
1286 lws_b64_encode_string(hash, 20,
1287 wsi->initial_handshake_hash_base64,
1288 sizeof wsi->initial_handshake_hash_base64);
Peter Hinz56885f32011-03-02 22:03:47 +00001289
Andy Greeneeaacb32011-03-01 20:44:24 +00001290issue_hdr:
Peter Hinz56885f32011-03-02 22:03:47 +00001291
Andy Greeneeaacb32011-03-01 20:44:24 +00001292 /* done with these now */
Peter Hinz56885f32011-03-02 22:03:47 +00001293
Andy Greeneeaacb32011-03-01 20:44:24 +00001294 free(wsi->c_path);
1295 free(wsi->c_host);
1296 if (wsi->c_origin)
1297 free(wsi->c_origin);
1298
Andy Greenbe93fef2011-02-14 20:25:43 +00001299 /* send our request to the server */
1300
1301 #ifdef LWS_OPENSSL_SUPPORT
1302 if (wsi->use_ssl)
1303 n = SSL_write(wsi->ssl, pkt, p - pkt);
1304 else
1305 #endif
1306 n = send(wsi->sock, pkt, p - pkt, 0);
1307
1308 if (n < 0) {
1309 fprintf(stderr, "ERROR writing to client socket\n");
Peter Hinz56885f32011-03-02 22:03:47 +00001310 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001311 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001312 return 1;
1313 }
1314
1315 wsi->parser_state = WSI_TOKEN_NAME_PART;
1316 wsi->mode = LWS_CONNMODE_WS_CLIENT_WAITING_SERVER_REPLY;
1317 libwebsocket_set_timeout(wsi,
1318 PENDING_TIMEOUT_AWAITING_SERVER_RESPONSE, 5);
1319
1320 break;
1321
1322 case LWS_CONNMODE_WS_CLIENT_WAITING_SERVER_REPLY:
1323
1324 /* handle server hung up on us */
1325
1326 if (pollfd->revents & (POLLERR | POLLHUP)) {
1327
1328 fprintf(stderr, "Server connection %p (fd=%d) dead\n",
1329 (void *)wsi, pollfd->fd);
1330
1331 goto bail3;
1332 }
1333
1334
1335 /* interpret the server response */
1336
1337 /*
1338 * HTTP/1.1 101 Switching Protocols
1339 * Upgrade: websocket
1340 * Connection: Upgrade
1341 * Sec-WebSocket-Accept: me89jWimTRKTWwrS3aRrL53YZSo=
1342 * Sec-WebSocket-Nonce: AQIDBAUGBwgJCgsMDQ4PEC==
1343 * Sec-WebSocket-Protocol: chat
1344 */
1345
1346 #ifdef LWS_OPENSSL_SUPPORT
1347 if (wsi->use_ssl)
1348 len = SSL_read(wsi->ssl, pkt, sizeof pkt);
1349 else
1350 #endif
1351 len = recv(wsi->sock, pkt, sizeof pkt, 0);
1352
1353 if (len < 0) {
1354 fprintf(stderr,
1355 "libwebsocket_client_handshake read error\n");
1356 goto bail3;
1357 }
1358
1359 p = pkt;
1360 for (n = 0; n < len; n++)
1361 libwebsocket_parse(wsi, *p++);
1362
1363 if (wsi->parser_state != WSI_PARSING_COMPLETE) {
1364 fprintf(stderr, "libwebsocket_client_handshake "
Peter Hinz56885f32011-03-02 22:03:47 +00001365 "server response failed parsing\n");
Andy Greenbe93fef2011-02-14 20:25:43 +00001366 goto bail3;
1367 }
1368
1369 /*
Andy Greeneeaacb32011-03-01 20:44:24 +00001370 * 00 / 76 -->
1371 *
1372 * HTTP/1.1 101 WebSocket Protocol Handshake
1373 * Upgrade: WebSocket
1374 * Connection: Upgrade
1375 * Sec-WebSocket-Origin: http://127.0.0.1
1376 * Sec-WebSocket-Location: ws://127.0.0.1:9999/socket.io/websocket
1377 *
1378 * xxxxxxxxxxxxxxxx
1379 */
Peter Hinz56885f32011-03-02 22:03:47 +00001380
Andy Greeneeaacb32011-03-01 20:44:24 +00001381 if (wsi->ietf_spec_revision == 0) {
1382 if (!wsi->utf8_token[WSI_TOKEN_HTTP].token_len ||
Peter Hinz56885f32011-03-02 22:03:47 +00001383 !wsi->utf8_token[WSI_TOKEN_UPGRADE].token_len ||
1384 !wsi->utf8_token[WSI_TOKEN_CHALLENGE].token_len ||
1385 !wsi->utf8_token[WSI_TOKEN_CONNECTION].token_len ||
1386 (!wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len &&
1387 wsi->c_protocol != NULL)) {
Andy Greeneeaacb32011-03-01 20:44:24 +00001388 fprintf(stderr, "libwebsocket_client_handshake "
Peter Hinz56885f32011-03-02 22:03:47 +00001389 "missing required header(s)\n");
Andy Greeneeaacb32011-03-01 20:44:24 +00001390 pkt[len] = '\0';
1391 fprintf(stderr, "%s", pkt);
1392 goto bail3;
1393 }
Andy Greeneeaacb32011-03-01 20:44:24 +00001394
Peter Hinz56885f32011-03-02 22:03:47 +00001395 strtolower(wsi->utf8_token[WSI_TOKEN_HTTP].token);
1396 if (strcmp(wsi->utf8_token[WSI_TOKEN_HTTP].token,
1397 "101 websocket protocol handshake")) {
Andy Greeneeaacb32011-03-01 20:44:24 +00001398 fprintf(stderr, "libwebsocket_client_handshake "
Peter Hinz56885f32011-03-02 22:03:47 +00001399 "server sent bad HTTP response '%s'\n",
1400 wsi->utf8_token[WSI_TOKEN_HTTP].token);
1401 goto bail3;
1402 }
1403
1404 if (wsi->utf8_token[WSI_TOKEN_CHALLENGE].token_len <
1405 16) {
1406 fprintf(stderr, "libwebsocket_client_handshake "
1407 "challenge reply too short %d\n",
1408 wsi->utf8_token[
1409 WSI_TOKEN_CHALLENGE].token_len);
Andy Greeneeaacb32011-03-01 20:44:24 +00001410 pkt[len] = '\0';
1411 fprintf(stderr, "%s", pkt);
1412 goto bail3;
Peter Hinz56885f32011-03-02 22:03:47 +00001413
Andy Greeneeaacb32011-03-01 20:44:24 +00001414 }
Peter Hinz56885f32011-03-02 22:03:47 +00001415
Andy Greeneeaacb32011-03-01 20:44:24 +00001416 goto select_protocol;
1417 }
Peter Hinz56885f32011-03-02 22:03:47 +00001418
Andy Greeneeaacb32011-03-01 20:44:24 +00001419 /*
Andy Greenbe93fef2011-02-14 20:25:43 +00001420 * well, what the server sent looked reasonable for syntax.
1421 * Now let's confirm it sent all the necessary headers
1422 */
1423
1424 if (!wsi->utf8_token[WSI_TOKEN_HTTP].token_len ||
1425 !wsi->utf8_token[WSI_TOKEN_UPGRADE].token_len ||
1426 !wsi->utf8_token[WSI_TOKEN_CONNECTION].token_len ||
1427 !wsi->utf8_token[WSI_TOKEN_ACCEPT].token_len ||
Andy Green4eaa86b2011-02-26 11:17:48 +00001428 (!wsi->utf8_token[WSI_TOKEN_NONCE].token_len &&
1429 wsi->ietf_spec_revision == 4) ||
Andy Greenbe93fef2011-02-14 20:25:43 +00001430 (!wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len &&
1431 wsi->c_protocol != NULL)) {
1432 fprintf(stderr, "libwebsocket_client_handshake "
Andy Green687b0182011-02-26 11:04:01 +00001433 "missing required header(s)\n");
Andy Greenbe93fef2011-02-14 20:25:43 +00001434 pkt[len] = '\0';
1435 fprintf(stderr, "%s", pkt);
1436 goto bail3;
1437 }
1438
1439 /*
1440 * Everything seems to be there, now take a closer look at what
1441 * is in each header
1442 */
1443
1444 strtolower(wsi->utf8_token[WSI_TOKEN_HTTP].token);
1445 if (strcmp(wsi->utf8_token[WSI_TOKEN_HTTP].token,
1446 "101 switching protocols")) {
1447 fprintf(stderr, "libwebsocket_client_handshake "
1448 "server sent bad HTTP response '%s'\n",
1449 wsi->utf8_token[WSI_TOKEN_HTTP].token);
1450 goto bail3;
1451 }
1452
1453 strtolower(wsi->utf8_token[WSI_TOKEN_UPGRADE].token);
1454 if (strcmp(wsi->utf8_token[WSI_TOKEN_UPGRADE].token,
1455 "websocket")) {
1456 fprintf(stderr, "libwebsocket_client_handshake server "
1457 "sent bad Upgrade header '%s'\n",
1458 wsi->utf8_token[WSI_TOKEN_UPGRADE].token);
1459 goto bail3;
1460 }
1461
1462 strtolower(wsi->utf8_token[WSI_TOKEN_CONNECTION].token);
1463 if (strcmp(wsi->utf8_token[WSI_TOKEN_CONNECTION].token,
1464 "upgrade")) {
1465 fprintf(stderr, "libwebsocket_client_handshake server "
1466 "sent bad Connection hdr '%s'\n",
1467 wsi->utf8_token[WSI_TOKEN_CONNECTION].token);
1468 goto bail3;
1469 }
1470
Andy Greeneeaacb32011-03-01 20:44:24 +00001471select_protocol:
Andy Greenbe93fef2011-02-14 20:25:43 +00001472 pc = wsi->c_protocol;
1473
1474 /*
1475 * confirm the protocol the server wants to talk was in the list
1476 * of protocols we offered
1477 */
1478
1479 if (!wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len) {
1480
1481 /*
1482 * no protocol name to work from,
1483 * default to first protocol
1484 */
Peter Hinz56885f32011-03-02 22:03:47 +00001485 wsi->protocol = &context->protocols[0];
Andy Greenbe93fef2011-02-14 20:25:43 +00001486
1487 free(wsi->c_protocol);
1488
1489 goto check_accept;
1490 }
1491
1492 while (*pc && !okay) {
1493 if ((!strncmp(pc,
1494 wsi->utf8_token[WSI_TOKEN_PROTOCOL].token,
1495 wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len)) &&
1496 (pc[wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len] == ',' ||
1497 pc[wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len] == '\0')) {
1498 okay = 1;
1499 continue;
1500 }
1501 while (*pc && *pc != ',')
1502 pc++;
1503 while (*pc && *pc != ' ')
1504 pc++;
1505 }
1506
1507 /* done with him now */
1508
1509 if (wsi->c_protocol)
1510 free(wsi->c_protocol);
1511
1512
1513 if (!okay) {
1514 fprintf(stderr, "libwebsocket_client_handshake server "
1515 "sent bad protocol '%s'\n",
1516 wsi->utf8_token[WSI_TOKEN_PROTOCOL].token);
1517 goto bail2;
1518 }
1519
1520 /*
1521 * identify the selected protocol struct and set it
1522 */
1523 n = 0;
1524 wsi->protocol = NULL;
Peter Hinz56885f32011-03-02 22:03:47 +00001525 while (context->protocols[n].callback) {
Andy Greenbe93fef2011-02-14 20:25:43 +00001526 if (strcmp(wsi->utf8_token[WSI_TOKEN_PROTOCOL].token,
Peter Hinz56885f32011-03-02 22:03:47 +00001527 context->protocols[n].name) == 0)
1528 wsi->protocol = &context->protocols[n];
Andy Greenbe93fef2011-02-14 20:25:43 +00001529 n++;
1530 }
1531
1532 if (wsi->protocol == NULL) {
1533 fprintf(stderr, "libwebsocket_client_handshake server "
1534 "requested protocol '%s', which we "
1535 "said we supported but we don't!\n",
1536 wsi->utf8_token[WSI_TOKEN_PROTOCOL].token);
1537 goto bail2;
1538 }
1539
Andy Green2366b1c2011-03-06 13:15:31 +00001540
1541 /* instantiate the accepted extensions */
1542
1543 if (!wsi->utf8_token[WSI_TOKEN_EXTENSIONS].token_len)
1544 goto check_accept;
1545
1546 /*
1547 * break down the list of server accepted extensions
1548 * and go through matching them or identifying bogons
1549 */
1550
1551 c = wsi->utf8_token[WSI_TOKEN_EXTENSIONS].token;
1552 n = 0;
1553 while (more) {
1554
1555 if (*c && *c != ',') {
1556 ext_name[n] = *c++;
1557 if (n < sizeof(ext_name) - 1)
1558 n++;
1559 continue;
1560 }
1561 ext_name[n] = '\0';
1562 if (!*c)
1563 more = 0;
1564
1565 /* check we actually support it */
1566
1567 n = 0;
1568 ext = wsi->protocol->owning_server->extensions;
1569 while (ext && ext->callback) {
1570
1571 if (strcmp(ext_name, ext->name)) {
1572 ext++;
1573 continue;
1574 }
1575
1576 n = 1;
1577
1578 /* instantiate the extension on this conn */
1579
1580 wsi->active_extensions_user[
1581 wsi->count_active_extensions] =
1582 malloc(ext->per_session_data_size);
1583 wsi->active_extensions[
1584 wsi->count_active_extensions] = ext;
1585
1586 /* allow him to construct his context */
1587
1588 ext->callback(wsi->protocol->owning_server,
1589 wsi, LWS_EXT_CALLBACK_CLIENT_CONSTRUCT,
1590 wsi->active_extensions_user[
1591 wsi->count_active_extensions], NULL, 0);
1592
1593 wsi->count_active_extensions++;
1594
1595 ext++;
1596 }
1597
1598 if (n == 0) {
1599 fprintf(stderr, "Server said we should use"
1600 "an unknown extension '%s'!\n", ext_name);
1601 goto bail2;
1602 }
1603
1604 n = 0;
1605 }
1606
1607
Andy Greenbe93fef2011-02-14 20:25:43 +00001608 check_accept:
Andy Greeneeaacb32011-03-01 20:44:24 +00001609
1610 if (wsi->ietf_spec_revision == 0) {
Peter Hinz56885f32011-03-02 22:03:47 +00001611
Andy Greeneeaacb32011-03-01 20:44:24 +00001612 if (memcmp(wsi->initial_handshake_hash_base64,
Peter Hinz56885f32011-03-02 22:03:47 +00001613 wsi->utf8_token[WSI_TOKEN_CHALLENGE].token, 16)) {
Andy Greeneeaacb32011-03-01 20:44:24 +00001614 fprintf(stderr, "libwebsocket_client_handshake "
Peter Hinz56885f32011-03-02 22:03:47 +00001615 "failed 00 challenge compare\n");
1616 pkt[len] = '\0';
1617 fprintf(stderr, "%s", pkt);
1618 goto bail2;
Andy Greeneeaacb32011-03-01 20:44:24 +00001619 }
Peter Hinz56885f32011-03-02 22:03:47 +00001620
Andy Greeneeaacb32011-03-01 20:44:24 +00001621 goto accept_ok;
1622 }
Peter Hinz56885f32011-03-02 22:03:47 +00001623
Andy Greenbe93fef2011-02-14 20:25:43 +00001624 /*
1625 * Confirm his accept token is the one we precomputed
1626 */
1627
1628 if (strcmp(wsi->utf8_token[WSI_TOKEN_ACCEPT].token,
1629 wsi->initial_handshake_hash_base64)) {
1630 fprintf(stderr, "libwebsocket_client_handshake server "
1631 "sent bad ACCEPT '%s' vs computed '%s'\n",
1632 wsi->utf8_token[WSI_TOKEN_ACCEPT].token,
1633 wsi->initial_handshake_hash_base64);
1634 goto bail2;
1635 }
1636
Andy Green4eaa86b2011-02-26 11:17:48 +00001637 if (wsi->ietf_spec_revision == 4) {
1638 /*
1639 * Calculate the 04 masking key to use when
1640 * sending data to server
1641 */
Andy Greenbe93fef2011-02-14 20:25:43 +00001642
Andy Green4eaa86b2011-02-26 11:17:48 +00001643 strcpy((char *)buf, wsi->key_b64);
1644 p = (char *)buf + strlen(wsi->key_b64);
1645 strcpy(p, wsi->utf8_token[WSI_TOKEN_NONCE].token);
1646 p += wsi->utf8_token[WSI_TOKEN_NONCE].token_len;
1647 strcpy(p, magic_websocket_04_masking_guid);
1648 SHA1(buf, strlen((char *)buf), wsi->masking_key_04);
1649 }
Andy Greeneeaacb32011-03-01 20:44:24 +00001650accept_ok:
1651
Andy Greenbe93fef2011-02-14 20:25:43 +00001652 /* allocate the per-connection user memory (if any) */
1653
1654 if (wsi->protocol->per_session_data_size) {
1655 wsi->user_space = malloc(
1656 wsi->protocol->per_session_data_size);
1657 if (wsi->user_space == NULL) {
1658 fprintf(stderr, "Out of memory for "
1659 "conn user space\n");
1660 goto bail2;
1661 }
1662 } else
1663 wsi->user_space = NULL;
1664
1665 /* clear his proxy connection timeout */
1666
1667 libwebsocket_set_timeout(wsi, NO_PENDING_TIMEOUT, 0);
1668
1669 /* mark him as being alive */
1670
1671 wsi->state = WSI_STATE_ESTABLISHED;
1672 wsi->mode = LWS_CONNMODE_WS_CLIENT;
1673
1674 fprintf(stderr, "handshake OK for protocol %s\n",
1675 wsi->protocol->name);
1676
1677 /* call him back to inform him he is up */
1678
Peter Hinz56885f32011-03-02 22:03:47 +00001679 wsi->protocol->callback(context, wsi,
Andy Greenbe93fef2011-02-14 20:25:43 +00001680 LWS_CALLBACK_CLIENT_ESTABLISHED,
1681 wsi->user_space,
1682 NULL, 0);
1683
1684 break;
1685
1686bail3:
1687 if (wsi->c_protocol)
1688 free(wsi->c_protocol);
1689
1690bail2:
Peter Hinz56885f32011-03-02 22:03:47 +00001691 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001692 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001693 return 1;
1694
1695
Andy Green0d338332011-02-12 11:57:43 +00001696 case LWS_CONNMODE_WS_SERVING:
1697 case LWS_CONNMODE_WS_CLIENT:
1698
1699 /* handle session socket closed */
1700
1701 if (pollfd->revents & (POLLERR | POLLHUP)) {
1702
Andy Green62c54d22011-02-14 09:14:25 +00001703 fprintf(stderr, "Session Socket %p (fd=%d) dead\n",
Andy Green0d338332011-02-12 11:57:43 +00001704 (void *)wsi, pollfd->fd);
1705
Peter Hinz56885f32011-03-02 22:03:47 +00001706 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001707 LWS_CLOSE_STATUS_NOSTATUS);
Andy Green4b6fbe12011-02-14 08:03:48 +00001708 return 1;
Andy Greenb45993c2010-12-18 15:13:50 +00001709 }
1710
Andy Green0d338332011-02-12 11:57:43 +00001711 /* the guy requested a callback when it was OK to write */
1712
Andy Greenda527df2011-03-07 07:08:12 +00001713 if ((pollfd->revents & POLLOUT) &&
1714 wsi->state == WSI_STATE_ESTABLISHED)
1715 if (lws_handle_POLLOUT_event(context, wsi,
1716 pollfd) < 0) {
1717 libwebsocket_close_and_free_session(
1718 context, wsi, LWS_CLOSE_STATUS_NORMAL);
Andy Green3b84c002011-03-06 13:14:42 +00001719 return 1;
1720 }
Andy Green0d338332011-02-12 11:57:43 +00001721
Andy Green0d338332011-02-12 11:57:43 +00001722
1723 /* any incoming data ready? */
1724
1725 if (!(pollfd->revents & POLLIN))
1726 break;
1727
Andy Greenb45993c2010-12-18 15:13:50 +00001728#ifdef LWS_OPENSSL_SUPPORT
Andy Green0d338332011-02-12 11:57:43 +00001729 if (wsi->ssl)
Andy Green98a717c2011-03-06 13:14:15 +00001730 eff_buf.token_len = SSL_read(wsi->ssl, buf, sizeof buf);
Andy Greenb45993c2010-12-18 15:13:50 +00001731 else
1732#endif
Andy Green98a717c2011-03-06 13:14:15 +00001733 eff_buf.token_len =
1734 recv(pollfd->fd, buf, sizeof buf, 0);
Andy Greenb45993c2010-12-18 15:13:50 +00001735
Andy Green98a717c2011-03-06 13:14:15 +00001736 if (eff_buf.token_len < 0) {
1737 fprintf(stderr, "Socket read returned %d\n",
1738 eff_buf.token_len);
Andy Green4b6fbe12011-02-14 08:03:48 +00001739 break;
Andy Greenb45993c2010-12-18 15:13:50 +00001740 }
Andy Green98a717c2011-03-06 13:14:15 +00001741 if (!eff_buf.token_len) {
Peter Hinz56885f32011-03-02 22:03:47 +00001742 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001743 LWS_CLOSE_STATUS_NOSTATUS);
Andy Green4b6fbe12011-02-14 08:03:48 +00001744 return 1;
Andy Greenb45993c2010-12-18 15:13:50 +00001745 }
1746
Andy Green98a717c2011-03-06 13:14:15 +00001747 /*
1748 * give any active extensions a chance to munge the buffer
1749 * before parse. We pass in a pointer to an lws_tokens struct
1750 * prepared with the default buffer and content length that's in
1751 * there. Rather than rewrite the default buffer, extensions
1752 * that expect to grow the buffer can adapt .token to
1753 * point to their own per-connection buffer in the extension
1754 * user allocation. By default with no extensions or no
1755 * extension callback handling, just the normal input buffer is
1756 * used then so it is efficient.
1757 */
Andy Greenb45993c2010-12-18 15:13:50 +00001758
Andy Green98a717c2011-03-06 13:14:15 +00001759 eff_buf.token = (char *)buf;
Andy Greenb45993c2010-12-18 15:13:50 +00001760
Andy Green98a717c2011-03-06 13:14:15 +00001761 more = 1;
1762 while (more) {
Andy Green0d338332011-02-12 11:57:43 +00001763
Andy Green98a717c2011-03-06 13:14:15 +00001764 more = 0;
1765
1766 for (n = 0; n < wsi->count_active_extensions; n++) {
1767 m = wsi->active_extensions[n]->callback(context, wsi,
1768 LWS_EXT_CALLBACK_PACKET_RX_PREPARSE,
1769 wsi->active_extensions_user[n], &eff_buf, 0);
1770 if (m < 0) {
1771 fprintf(stderr, "Extension reports fatal error\n");
1772 libwebsocket_close_and_free_session(context, wsi,
1773 LWS_CLOSE_STATUS_NOSTATUS);
1774 return 1;
1775 }
1776 if (m)
1777 more = 1;
1778 }
1779
1780 /* service incoming data */
1781
1782 if (eff_buf.token_len) {
1783 n = libwebsocket_read(context, wsi,
1784 (unsigned char *)eff_buf.token, eff_buf.token_len);
1785 if (n < 0)
1786 /* we closed wsi */
1787 return 1;
1788 }
1789
1790 eff_buf.token = NULL;
1791 eff_buf.token_len = 0;
1792 }
1793 break;
Andy Greenb45993c2010-12-18 15:13:50 +00001794 }
1795
1796 return 0;
1797}
1798
Andy Green0d338332011-02-12 11:57:43 +00001799
Andy Green6964bb52011-01-23 16:50:33 +00001800/**
1801 * libwebsocket_context_destroy() - Destroy the websocket context
Peter Hinz56885f32011-03-02 22:03:47 +00001802 * @context: Websocket context
Andy Green6964bb52011-01-23 16:50:33 +00001803 *
1804 * This function closes any active connections and then frees the
1805 * context. After calling this, any further use of the context is
1806 * undefined.
1807 */
1808void
Peter Hinz56885f32011-03-02 22:03:47 +00001809libwebsocket_context_destroy(struct libwebsocket_context *context)
Andy Green6964bb52011-01-23 16:50:33 +00001810{
Andy Green0d338332011-02-12 11:57:43 +00001811 int n;
1812 int m;
1813 struct libwebsocket *wsi;
Andy Green6964bb52011-01-23 16:50:33 +00001814
Andy Green4b6fbe12011-02-14 08:03:48 +00001815 for (n = 0; n < FD_HASHTABLE_MODULUS; n++)
Peter Hinz56885f32011-03-02 22:03:47 +00001816 for (m = 0; m < context->fd_hashtable[n].length; m++) {
1817 wsi = context->fd_hashtable[n].wsi[m];
1818 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001819 LWS_CLOSE_STATUS_GOINGAWAY);
Andy Greenf3d3b402011-02-09 07:16:34 +00001820 }
Andy Green6964bb52011-01-23 16:50:33 +00001821
Peter Hinz56885f32011-03-02 22:03:47 +00001822#ifdef WIN32
1823#else
1824 close(context->fd_random);
Andy Green6964bb52011-01-23 16:50:33 +00001825#endif
1826
Peter Hinz56885f32011-03-02 22:03:47 +00001827#ifdef LWS_OPENSSL_SUPPORT
1828 if (context->ssl_ctx)
1829 SSL_CTX_free(context->ssl_ctx);
1830 if (context->ssl_client_ctx)
1831 SSL_CTX_free(context->ssl_client_ctx);
1832#endif
1833
1834 free(context);
1835
1836#ifdef WIN32
1837 WSACleanup();
1838#endif
Andy Green6964bb52011-01-23 16:50:33 +00001839}
1840
1841/**
1842 * libwebsocket_service() - Service any pending websocket activity
Peter Hinz56885f32011-03-02 22:03:47 +00001843 * @context: Websocket context
Andy Green6964bb52011-01-23 16:50:33 +00001844 * @timeout_ms: Timeout for poll; 0 means return immediately if nothing needed
1845 * service otherwise block and service immediately, returning
1846 * after the timeout if nothing needed service.
1847 *
1848 * This function deals with any pending websocket traffic, for three
1849 * kinds of event. It handles these events on both server and client
1850 * types of connection the same.
1851 *
1852 * 1) Accept new connections to our context's server
1853 *
1854 * 2) Perform pending broadcast writes initiated from other forked
1855 * processes (effectively serializing asynchronous broadcasts)
1856 *
1857 * 3) Call the receive callback for incoming frame data received by
1858 * server or client connections.
1859 *
1860 * You need to call this service function periodically to all the above
1861 * functions to happen; if your application is single-threaded you can
1862 * just call it in your main event loop.
1863 *
1864 * Alternatively you can fork a new process that asynchronously handles
1865 * calling this service in a loop. In that case you are happy if this
1866 * call blocks your thread until it needs to take care of something and
1867 * would call it with a large nonzero timeout. Your loop then takes no
1868 * CPU while there is nothing happening.
1869 *
1870 * If you are calling it in a single-threaded app, you don't want it to
1871 * wait around blocking other things in your loop from happening, so you
1872 * would call it with a timeout_ms of 0, so it returns immediately if
1873 * nothing is pending, or as soon as it services whatever was pending.
1874 */
1875
Andy Greenb45993c2010-12-18 15:13:50 +00001876
Andy Greene92cd172011-01-19 13:11:55 +00001877int
Peter Hinz56885f32011-03-02 22:03:47 +00001878libwebsocket_service(struct libwebsocket_context *context, int timeout_ms)
Andy Greene92cd172011-01-19 13:11:55 +00001879{
1880 int n;
Andy Greene92cd172011-01-19 13:11:55 +00001881
1882 /* stay dead once we are dead */
1883
Peter Hinz56885f32011-03-02 22:03:47 +00001884 if (context == NULL)
Andy Greene92cd172011-01-19 13:11:55 +00001885 return 1;
1886
Andy Green0d338332011-02-12 11:57:43 +00001887 /* wait for something to need service */
Andy Green4739e5c2011-01-22 12:51:57 +00001888
Peter Hinz56885f32011-03-02 22:03:47 +00001889 n = poll(context->fds, context->fds_count, timeout_ms);
Andy Green3221f922011-02-12 13:14:11 +00001890 if (n == 0) /* poll timeout */
1891 return 0;
Andy Greene92cd172011-01-19 13:11:55 +00001892
Andy Green62c54d22011-02-14 09:14:25 +00001893 if (n < 0) {
Andy Green5e1fa172011-02-10 09:07:05 +00001894 /*
Andy Greene92cd172011-01-19 13:11:55 +00001895 fprintf(stderr, "Listen Socket dead\n");
Andy Green5e1fa172011-02-10 09:07:05 +00001896 */
Andy Green0d338332011-02-12 11:57:43 +00001897 return 1;
Andy Greene92cd172011-01-19 13:11:55 +00001898 }
Andy Greene92cd172011-01-19 13:11:55 +00001899
1900 /* handle accept on listening socket? */
1901
Peter Hinz56885f32011-03-02 22:03:47 +00001902 for (n = 0; n < context->fds_count; n++)
1903 if (context->fds[n].revents)
1904 libwebsocket_service_fd(context, &context->fds[n]);
Andy Greene92cd172011-01-19 13:11:55 +00001905
1906 return 0;
Andy Greene92cd172011-01-19 13:11:55 +00001907}
1908
Andy Green90c7cbc2011-01-27 06:26:52 +00001909/**
1910 * libwebsocket_callback_on_writable() - Request a callback when this socket
1911 * becomes able to be written to without
1912 * blocking
Andy Green32375b72011-02-19 08:32:53 +00001913 *
Peter Hinz56885f32011-03-02 22:03:47 +00001914 * @context: libwebsockets context
Andy Green90c7cbc2011-01-27 06:26:52 +00001915 * @wsi: Websocket connection instance to get callback for
1916 */
1917
1918int
Peter Hinz56885f32011-03-02 22:03:47 +00001919libwebsocket_callback_on_writable(struct libwebsocket_context *context,
Andy Green62c54d22011-02-14 09:14:25 +00001920 struct libwebsocket *wsi)
Andy Green90c7cbc2011-01-27 06:26:52 +00001921{
Andy Green90c7cbc2011-01-27 06:26:52 +00001922 int n;
1923
Peter Hinz56885f32011-03-02 22:03:47 +00001924 for (n = 0; n < context->fds_count; n++)
1925 if (context->fds[n].fd == wsi->sock) {
1926 context->fds[n].events |= POLLOUT;
1927 n = context->fds_count;
Andy Green90c7cbc2011-01-27 06:26:52 +00001928 }
1929
Andy Green3221f922011-02-12 13:14:11 +00001930 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00001931 context->protocols[0].callback(context, wsi,
Andy Green3221f922011-02-12 13:14:11 +00001932 LWS_CALLBACK_SET_MODE_POLL_FD,
1933 (void *)(long)wsi->sock, NULL, POLLOUT);
1934
Andy Green90c7cbc2011-01-27 06:26:52 +00001935 return 1;
1936}
1937
1938/**
1939 * libwebsocket_callback_on_writable_all_protocol() - Request a callback for
1940 * all connections using the given protocol when it
1941 * becomes possible to write to each socket without
1942 * blocking in turn.
1943 *
1944 * @protocol: Protocol whose connections will get callbacks
1945 */
1946
1947int
1948libwebsocket_callback_on_writable_all_protocol(
1949 const struct libwebsocket_protocols *protocol)
1950{
Peter Hinz56885f32011-03-02 22:03:47 +00001951 struct libwebsocket_context *context = protocol->owning_server;
Andy Green90c7cbc2011-01-27 06:26:52 +00001952 int n;
Andy Green0d338332011-02-12 11:57:43 +00001953 int m;
1954 struct libwebsocket *wsi;
Andy Green90c7cbc2011-01-27 06:26:52 +00001955
Andy Green0d338332011-02-12 11:57:43 +00001956 for (n = 0; n < FD_HASHTABLE_MODULUS; n++) {
1957
Peter Hinz56885f32011-03-02 22:03:47 +00001958 for (m = 0; m < context->fd_hashtable[n].length; m++) {
Andy Green0d338332011-02-12 11:57:43 +00001959
Peter Hinz56885f32011-03-02 22:03:47 +00001960 wsi = context->fd_hashtable[n].wsi[m];
Andy Green0d338332011-02-12 11:57:43 +00001961
1962 if (wsi->protocol == protocol)
Peter Hinz56885f32011-03-02 22:03:47 +00001963 libwebsocket_callback_on_writable(context, wsi);
Andy Green0d338332011-02-12 11:57:43 +00001964 }
1965 }
Andy Green90c7cbc2011-01-27 06:26:52 +00001966
1967 return 0;
1968}
1969
Andy Greenbe93fef2011-02-14 20:25:43 +00001970/**
1971 * libwebsocket_set_timeout() - marks the wsi as subject to a timeout
1972 *
1973 * You will not need this unless you are doing something special
1974 *
1975 * @wsi: Websocket connection instance
1976 * @reason: timeout reason
1977 * @secs: how many seconds
1978 */
1979
1980void
1981libwebsocket_set_timeout(struct libwebsocket *wsi,
1982 enum pending_timeout reason, int secs)
1983{
1984 struct timeval tv;
1985
1986 gettimeofday(&tv, NULL);
1987
1988 wsi->pending_timeout_limit = tv.tv_sec + secs;
1989 wsi->pending_timeout = reason;
1990}
1991
Andy Greena6cbece2011-01-27 20:06:03 +00001992
1993/**
1994 * libwebsocket_get_socket_fd() - returns the socket file descriptor
1995 *
1996 * You will not need this unless you are doing something special
1997 *
1998 * @wsi: Websocket connection instance
1999 */
2000
2001int
2002libwebsocket_get_socket_fd(struct libwebsocket *wsi)
2003{
2004 return wsi->sock;
2005}
2006
Andy Green90c7cbc2011-01-27 06:26:52 +00002007/**
2008 * libwebsocket_rx_flow_control() - Enable and disable socket servicing for
2009 * receieved packets.
2010 *
2011 * If the output side of a server process becomes choked, this allows flow
2012 * control for the input side.
2013 *
2014 * @wsi: Websocket connection instance to get callback for
2015 * @enable: 0 = disable read servicing for this connection, 1 = enable
2016 */
2017
2018int
2019libwebsocket_rx_flow_control(struct libwebsocket *wsi, int enable)
2020{
Peter Hinz56885f32011-03-02 22:03:47 +00002021 struct libwebsocket_context *context = wsi->protocol->owning_server;
Andy Green90c7cbc2011-01-27 06:26:52 +00002022 int n;
2023
Peter Hinz56885f32011-03-02 22:03:47 +00002024 for (n = 0; n < context->fds_count; n++)
2025 if (context->fds[n].fd == wsi->sock) {
Andy Green90c7cbc2011-01-27 06:26:52 +00002026 if (enable)
Peter Hinz56885f32011-03-02 22:03:47 +00002027 context->fds[n].events |= POLLIN;
Andy Green90c7cbc2011-01-27 06:26:52 +00002028 else
Peter Hinz56885f32011-03-02 22:03:47 +00002029 context->fds[n].events &= ~POLLIN;
Andy Green90c7cbc2011-01-27 06:26:52 +00002030
2031 return 0;
2032 }
2033
Andy Green3221f922011-02-12 13:14:11 +00002034 if (enable)
2035 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00002036 context->protocols[0].callback(context, wsi,
Andy Green3221f922011-02-12 13:14:11 +00002037 LWS_CALLBACK_SET_MODE_POLL_FD,
2038 (void *)(long)wsi->sock, NULL, POLLIN);
2039 else
2040 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00002041 context->protocols[0].callback(context, wsi,
Andy Green3221f922011-02-12 13:14:11 +00002042 LWS_CALLBACK_CLEAR_MODE_POLL_FD,
2043 (void *)(long)wsi->sock, NULL, POLLIN);
2044
2045
Andy Green90c7cbc2011-01-27 06:26:52 +00002046 fprintf(stderr, "libwebsocket_callback_on_writable "
2047 "unable to find socket\n");
2048 return 1;
2049}
2050
Andy Green2ac5a6f2011-01-28 10:00:18 +00002051/**
2052 * libwebsocket_canonical_hostname() - returns this host's hostname
2053 *
2054 * This is typically used by client code to fill in the host parameter
2055 * when making a client connection. You can only call it after the context
2056 * has been created.
2057 *
Peter Hinz56885f32011-03-02 22:03:47 +00002058 * @context: Websocket context
Andy Green2ac5a6f2011-01-28 10:00:18 +00002059 */
2060
2061
2062extern const char *
Peter Hinz56885f32011-03-02 22:03:47 +00002063libwebsocket_canonical_hostname(struct libwebsocket_context *context)
Andy Green2ac5a6f2011-01-28 10:00:18 +00002064{
Peter Hinz56885f32011-03-02 22:03:47 +00002065 return (const char *)context->canonical_hostname;
Andy Green2ac5a6f2011-01-28 10:00:18 +00002066}
2067
2068
Andy Green90c7cbc2011-01-27 06:26:52 +00002069static void sigpipe_handler(int x)
2070{
2071}
2072
Andy Green6901cb32011-02-21 08:06:47 +00002073#ifdef LWS_OPENSSL_SUPPORT
2074static int
2075OpenSSL_verify_callback(int preverify_ok, X509_STORE_CTX *x509_ctx)
2076{
2077
2078 SSL *ssl;
2079 int n;
Andy Green2e24da02011-03-05 16:12:04 +00002080 struct libwebsocket_context *context;
Andy Green6901cb32011-02-21 08:06:47 +00002081
2082 ssl = X509_STORE_CTX_get_ex_data(x509_ctx,
2083 SSL_get_ex_data_X509_STORE_CTX_idx());
2084
2085 /*
Andy Green2e24da02011-03-05 16:12:04 +00002086 * !!! nasty openssl requires the index to come as a library-scope
2087 * static
Andy Green6901cb32011-02-21 08:06:47 +00002088 */
Andy Green2e24da02011-03-05 16:12:04 +00002089 context = SSL_get_ex_data(ssl, openssl_websocket_private_data_index);
Andy Green6901cb32011-02-21 08:06:47 +00002090
Peter Hinz56885f32011-03-02 22:03:47 +00002091 n = context->protocols[0].callback(NULL, NULL,
Andy Green6901cb32011-02-21 08:06:47 +00002092 LWS_CALLBACK_OPENSSL_PERFORM_CLIENT_CERT_VERIFICATION,
2093 x509_ctx, ssl, preverify_ok);
2094
2095 /* convert return code from 0 = OK to 1 = OK */
2096
2097 if (!n)
2098 n = 1;
2099 else
2100 n = 0;
2101
2102 return n;
2103}
2104#endif
2105
Andy Greenb45993c2010-12-18 15:13:50 +00002106
Andy Greenab990e42010-10-31 12:42:52 +00002107/**
Andy Green4739e5c2011-01-22 12:51:57 +00002108 * libwebsocket_create_context() - Create the websocket handler
2109 * @port: Port to listen on... you can use 0 to suppress listening on
Andy Green6964bb52011-01-23 16:50:33 +00002110 * any port, that's what you want if you are not running a
2111 * websocket server at all but just using it as a client
Peter Hinz56885f32011-03-02 22:03:47 +00002112 * @interf: NULL to bind the listen socket to all interfaces, or the
Andy Green32375b72011-02-19 08:32:53 +00002113 * interface name, eg, "eth2"
Andy Green4f3943a2010-11-12 10:44:16 +00002114 * @protocols: Array of structures listing supported protocols and a protocol-
Andy Green8f037e42010-12-19 22:13:26 +00002115 * specific callback for each one. The list is ended with an
2116 * entry that has a NULL callback pointer.
Andy Green6964bb52011-01-23 16:50:33 +00002117 * It's not const because we write the owning_server member
Andy Greenc5114822011-03-06 10:29:35 +00002118 * @extensions: NULL or array of libwebsocket_extension structs listing the
2119 * extensions this context supports
Andy Green3faa9c72010-11-08 17:03:03 +00002120 * @ssl_cert_filepath: If libwebsockets was compiled to use ssl, and you want
Andy Green8f037e42010-12-19 22:13:26 +00002121 * to listen using SSL, set to the filepath to fetch the
2122 * server cert from, otherwise NULL for unencrypted
Andy Green3faa9c72010-11-08 17:03:03 +00002123 * @ssl_private_key_filepath: filepath to private key if wanting SSL mode,
Andy Green8f037e42010-12-19 22:13:26 +00002124 * else ignored
Andy Green3faa9c72010-11-08 17:03:03 +00002125 * @gid: group id to change to after setting listen socket, or -1.
2126 * @uid: user id to change to after setting listen socket, or -1.
Andy Greenbfb051f2011-02-09 08:49:14 +00002127 * @options: 0, or LWS_SERVER_OPTION_DEFEAT_CLIENT_MASK
Andy Green05464c62010-11-12 10:44:18 +00002128 *
Andy Green8f037e42010-12-19 22:13:26 +00002129 * This function creates the listening socket and takes care
2130 * of all initialization in one step.
2131 *
Andy Greene92cd172011-01-19 13:11:55 +00002132 * After initialization, it returns a struct libwebsocket_context * that
2133 * represents this server. After calling, user code needs to take care
2134 * of calling libwebsocket_service() with the context pointer to get the
2135 * server's sockets serviced. This can be done in the same process context
2136 * or a forked process, or another thread,
Andy Green05464c62010-11-12 10:44:18 +00002137 *
Andy Green8f037e42010-12-19 22:13:26 +00002138 * The protocol callback functions are called for a handful of events
2139 * including http requests coming in, websocket connections becoming
2140 * established, and data arriving; it's also called periodically to allow
2141 * async transmission.
2142 *
2143 * HTTP requests are sent always to the FIRST protocol in @protocol, since
2144 * at that time websocket protocol has not been negotiated. Other
2145 * protocols after the first one never see any HTTP callack activity.
2146 *
2147 * The server created is a simple http server by default; part of the
2148 * websocket standard is upgrading this http connection to a websocket one.
2149 *
2150 * This allows the same server to provide files like scripts and favicon /
2151 * images or whatever over http and dynamic data over websockets all in
2152 * one place; they're all handled in the user callback.
Andy Greenab990e42010-10-31 12:42:52 +00002153 */
Andy Green4ea60062010-10-30 12:15:07 +01002154
Andy Greene92cd172011-01-19 13:11:55 +00002155struct libwebsocket_context *
Peter Hinz56885f32011-03-02 22:03:47 +00002156libwebsocket_create_context(int port, const char *interf,
Andy Greenb45993c2010-12-18 15:13:50 +00002157 struct libwebsocket_protocols *protocols,
Andy Greend6e09112011-03-05 16:12:15 +00002158 struct libwebsocket_extension *extensions,
Andy Green8f037e42010-12-19 22:13:26 +00002159 const char *ssl_cert_filepath,
2160 const char *ssl_private_key_filepath,
Andy Green8014b292011-01-30 20:57:25 +00002161 int gid, int uid, unsigned int options)
Andy Greenff95d7a2010-10-28 22:36:01 +01002162{
2163 int n;
Andy Green4739e5c2011-01-22 12:51:57 +00002164 int sockfd = 0;
Andy Green251f6fa2010-11-03 11:13:06 +00002165 int fd;
Andy Greenff95d7a2010-10-28 22:36:01 +01002166 struct sockaddr_in serv_addr, cli_addr;
Andy Green251f6fa2010-11-03 11:13:06 +00002167 int opt = 1;
Peter Hinz56885f32011-03-02 22:03:47 +00002168 struct libwebsocket_context *context = NULL;
Andy Greenb45993c2010-12-18 15:13:50 +00002169 unsigned int slen;
Andy Green9659f372011-01-27 22:01:43 +00002170 char *p;
Andy Green2ac5a6f2011-01-28 10:00:18 +00002171 char hostname[1024];
Andy Green42f69142011-01-30 08:10:02 +00002172 struct hostent *he;
Andy Green0d338332011-02-12 11:57:43 +00002173 struct libwebsocket *wsi;
Andy Greenff95d7a2010-10-28 22:36:01 +01002174
Andy Green3faa9c72010-11-08 17:03:03 +00002175#ifdef LWS_OPENSSL_SUPPORT
Andy Greenf2f54d52010-11-15 22:08:00 +00002176 SSL_METHOD *method;
Andy Green3faa9c72010-11-08 17:03:03 +00002177 char ssl_err_buf[512];
Andy Green3faa9c72010-11-08 17:03:03 +00002178#endif
2179
Peter Hinz56885f32011-03-02 22:03:47 +00002180#ifdef _WIN32
2181 {
2182 WORD wVersionRequested;
2183 WSADATA wsaData;
2184 int err;
2185
2186 /* Use the MAKEWORD(lowbyte, highbyte) macro from Windef.h */
2187 wVersionRequested = MAKEWORD(2, 2);
2188
2189 err = WSAStartup(wVersionRequested, &wsaData);
2190 if (err != 0) {
2191 /* Tell the user that we could not find a usable */
2192 /* Winsock DLL. */
2193 fprintf(stderr, "WSAStartup failed with error: %d\n",
2194 err);
2195 return NULL;
2196 }
2197 }
2198#endif
2199
2200
2201 context = malloc(sizeof(struct libwebsocket_context));
2202 if (!context) {
Andy Green90c7cbc2011-01-27 06:26:52 +00002203 fprintf(stderr, "No memory for websocket context\n");
2204 return NULL;
2205 }
Peter Hinz56885f32011-03-02 22:03:47 +00002206 context->protocols = protocols;
2207 context->listen_port = port;
2208 context->http_proxy_port = 0;
2209 context->http_proxy_address[0] = '\0';
2210 context->options = options;
2211 context->fds_count = 0;
Andy Greend6e09112011-03-05 16:12:15 +00002212 context->extensions = extensions;
Andy Green9659f372011-01-27 22:01:43 +00002213
Peter Hinz56885f32011-03-02 22:03:47 +00002214#ifdef WIN32
2215 context->fd_random = 0;
2216#else
2217 context->fd_random = open(SYSTEM_RANDOM_FILEPATH, O_RDONLY);
2218 if (context->fd_random < 0) {
Andy Green44eee682011-02-10 09:32:24 +00002219 fprintf(stderr, "Unable to open random device %s %d\n",
Peter Hinz56885f32011-03-02 22:03:47 +00002220 SYSTEM_RANDOM_FILEPATH, context->fd_random);
Andy Green44eee682011-02-10 09:32:24 +00002221 return NULL;
2222 }
Peter Hinz56885f32011-03-02 22:03:47 +00002223#endif
Andy Green44eee682011-02-10 09:32:24 +00002224
Peter Hinz56885f32011-03-02 22:03:47 +00002225#ifdef LWS_OPENSSL_SUPPORT
2226 context->use_ssl = 0;
2227 context->ssl_ctx = NULL;
2228 context->ssl_client_ctx = NULL;
Andy Green2e24da02011-03-05 16:12:04 +00002229 openssl_websocket_private_data_index = 0;
Peter Hinz56885f32011-03-02 22:03:47 +00002230#endif
Andy Green2ac5a6f2011-01-28 10:00:18 +00002231 /* find canonical hostname */
2232
2233 hostname[(sizeof hostname) - 1] = '\0';
2234 gethostname(hostname, (sizeof hostname) - 1);
2235 he = gethostbyname(hostname);
Darin Willitsc19456f2011-02-14 17:52:39 +00002236 if (he) {
Peter Hinz56885f32011-03-02 22:03:47 +00002237 strncpy(context->canonical_hostname, he->h_name,
2238 sizeof context->canonical_hostname - 1);
2239 context->canonical_hostname[
2240 sizeof context->canonical_hostname - 1] = '\0';
Darin Willitsc19456f2011-02-14 17:52:39 +00002241 } else
Peter Hinz56885f32011-03-02 22:03:47 +00002242 strncpy(context->canonical_hostname, hostname,
2243 sizeof context->canonical_hostname - 1);
Andy Green2ac5a6f2011-01-28 10:00:18 +00002244
Andy Green9659f372011-01-27 22:01:43 +00002245 /* split the proxy ads:port if given */
2246
2247 p = getenv("http_proxy");
2248 if (p) {
Peter Hinz56885f32011-03-02 22:03:47 +00002249 strncpy(context->http_proxy_address, p,
2250 sizeof context->http_proxy_address - 1);
2251 context->http_proxy_address[
2252 sizeof context->http_proxy_address - 1] = '\0';
Andy Green9659f372011-01-27 22:01:43 +00002253
Peter Hinz56885f32011-03-02 22:03:47 +00002254 p = strchr(context->http_proxy_address, ':');
Andy Green9659f372011-01-27 22:01:43 +00002255 if (p == NULL) {
2256 fprintf(stderr, "http_proxy needs to be ads:port\n");
2257 return NULL;
2258 }
2259 *p = '\0';
Peter Hinz56885f32011-03-02 22:03:47 +00002260 context->http_proxy_port = atoi(p + 1);
Andy Green9659f372011-01-27 22:01:43 +00002261
2262 fprintf(stderr, "Using proxy %s:%u\n",
Peter Hinz56885f32011-03-02 22:03:47 +00002263 context->http_proxy_address,
2264 context->http_proxy_port);
Andy Green9659f372011-01-27 22:01:43 +00002265 }
Andy Green90c7cbc2011-01-27 06:26:52 +00002266
2267 if (port) {
2268
Andy Green3faa9c72010-11-08 17:03:03 +00002269#ifdef LWS_OPENSSL_SUPPORT
Peter Hinz56885f32011-03-02 22:03:47 +00002270 context->use_ssl = ssl_cert_filepath != NULL &&
Andy Green90c7cbc2011-01-27 06:26:52 +00002271 ssl_private_key_filepath != NULL;
Peter Hinz56885f32011-03-02 22:03:47 +00002272 if (context->use_ssl)
Andy Green90c7cbc2011-01-27 06:26:52 +00002273 fprintf(stderr, " Compiled with SSL support, "
2274 "using it\n");
2275 else
2276 fprintf(stderr, " Compiled with SSL support, "
2277 "not using it\n");
Andy Green3faa9c72010-11-08 17:03:03 +00002278
Andy Green90c7cbc2011-01-27 06:26:52 +00002279#else
2280 if (ssl_cert_filepath != NULL &&
2281 ssl_private_key_filepath != NULL) {
2282 fprintf(stderr, " Not compiled for OpenSSl support!\n");
Andy Greene92cd172011-01-19 13:11:55 +00002283 return NULL;
Andy Green3faa9c72010-11-08 17:03:03 +00002284 }
Andy Green90c7cbc2011-01-27 06:26:52 +00002285 fprintf(stderr, " Compiled without SSL support, "
2286 "serving unencrypted\n");
2287#endif
2288 }
2289
2290 /* ignore SIGPIPE */
Peter Hinz56885f32011-03-02 22:03:47 +00002291#ifdef WIN32
2292#else
Andy Green90c7cbc2011-01-27 06:26:52 +00002293 signal(SIGPIPE, sigpipe_handler);
Peter Hinz56885f32011-03-02 22:03:47 +00002294#endif
Andy Green90c7cbc2011-01-27 06:26:52 +00002295
2296
2297#ifdef LWS_OPENSSL_SUPPORT
2298
2299 /* basic openssl init */
2300
2301 SSL_library_init();
2302
2303 OpenSSL_add_all_algorithms();
2304 SSL_load_error_strings();
2305
Andy Green2e24da02011-03-05 16:12:04 +00002306 openssl_websocket_private_data_index =
Andy Green6901cb32011-02-21 08:06:47 +00002307 SSL_get_ex_new_index(0, "libwebsockets", NULL, NULL, NULL);
2308
Andy Green90c7cbc2011-01-27 06:26:52 +00002309 /*
2310 * Firefox insists on SSLv23 not SSLv3
2311 * Konq disables SSLv2 by default now, SSLv23 works
2312 */
2313
2314 method = (SSL_METHOD *)SSLv23_server_method();
2315 if (!method) {
2316 fprintf(stderr, "problem creating ssl method: %s\n",
2317 ERR_error_string(ERR_get_error(), ssl_err_buf));
2318 return NULL;
2319 }
Peter Hinz56885f32011-03-02 22:03:47 +00002320 context->ssl_ctx = SSL_CTX_new(method); /* create context */
2321 if (!context->ssl_ctx) {
Andy Green90c7cbc2011-01-27 06:26:52 +00002322 fprintf(stderr, "problem creating ssl context: %s\n",
2323 ERR_error_string(ERR_get_error(), ssl_err_buf));
2324 return NULL;
2325 }
2326
2327 /* client context */
Peter Hinz56885f32011-03-02 22:03:47 +00002328 if (port == CONTEXT_PORT_NO_LISTEN)
2329 {
2330 method = (SSL_METHOD *)SSLv23_client_method();
2331 if (!method) {
2332 fprintf(stderr, "problem creating ssl method: %s\n",
2333 ERR_error_string(ERR_get_error(), ssl_err_buf));
2334 return NULL;
2335 }
2336 /* create context */
2337 context->ssl_client_ctx = SSL_CTX_new(method);
2338 if (!context->ssl_client_ctx) {
2339 fprintf(stderr, "problem creating ssl context: %s\n",
2340 ERR_error_string(ERR_get_error(), ssl_err_buf));
2341 return NULL;
2342 }
Andy Green90c7cbc2011-01-27 06:26:52 +00002343
Peter Hinz56885f32011-03-02 22:03:47 +00002344 /* openssl init for cert verification (for client sockets) */
Andy Green90c7cbc2011-01-27 06:26:52 +00002345
Peter Hinz56885f32011-03-02 22:03:47 +00002346 if (!SSL_CTX_load_verify_locations(
2347 context->ssl_client_ctx, NULL,
2348 LWS_OPENSSL_CLIENT_CERTS))
2349 fprintf(stderr,
2350 "Unable to load SSL Client certs from %s "
2351 "(set by --with-client-cert-dir= in configure) -- "
2352 " client ssl isn't going to work",
Andy Green90c7cbc2011-01-27 06:26:52 +00002353 LWS_OPENSSL_CLIENT_CERTS);
Peter Hinz56885f32011-03-02 22:03:47 +00002354
2355 /*
2356 * callback allowing user code to load extra verification certs
2357 * helping the client to verify server identity
2358 */
2359
2360 context->protocols[0].callback(context, NULL,
2361 LWS_CALLBACK_OPENSSL_LOAD_EXTRA_CLIENT_VERIFY_CERTS,
2362 context->ssl_client_ctx, NULL, 0);
Andy Green90c7cbc2011-01-27 06:26:52 +00002363 }
Andy Greenc6bf2c22011-02-20 11:10:47 +00002364 /* as a server, are we requiring clients to identify themselves? */
2365
2366 if (options & LWS_SERVER_OPTION_REQUIRE_VALID_OPENSSL_CLIENT_CERT) {
2367
2368 /* absolutely require the client cert */
2369
Peter Hinz56885f32011-03-02 22:03:47 +00002370 SSL_CTX_set_verify(context->ssl_ctx,
Andy Green6901cb32011-02-21 08:06:47 +00002371 SSL_VERIFY_PEER | SSL_VERIFY_FAIL_IF_NO_PEER_CERT,
2372 OpenSSL_verify_callback);
Andy Greenc6bf2c22011-02-20 11:10:47 +00002373
2374 /*
2375 * give user code a chance to load certs into the server
2376 * allowing it to verify incoming client certs
2377 */
2378
Peter Hinz56885f32011-03-02 22:03:47 +00002379 context->protocols[0].callback(context, NULL,
Andy Greenc6bf2c22011-02-20 11:10:47 +00002380 LWS_CALLBACK_OPENSSL_LOAD_EXTRA_SERVER_VERIFY_CERTS,
Peter Hinz56885f32011-03-02 22:03:47 +00002381 context->ssl_ctx, NULL, 0);
Andy Greenc6bf2c22011-02-20 11:10:47 +00002382 }
2383
Peter Hinz56885f32011-03-02 22:03:47 +00002384 if (context->use_ssl) {
Andy Green90c7cbc2011-01-27 06:26:52 +00002385
2386 /* openssl init for server sockets */
2387
Andy Green3faa9c72010-11-08 17:03:03 +00002388 /* set the local certificate from CertFile */
Peter Hinz56885f32011-03-02 22:03:47 +00002389 n = SSL_CTX_use_certificate_file(context->ssl_ctx,
Andy Green3faa9c72010-11-08 17:03:03 +00002390 ssl_cert_filepath, SSL_FILETYPE_PEM);
2391 if (n != 1) {
2392 fprintf(stderr, "problem getting cert '%s': %s\n",
2393 ssl_cert_filepath,
2394 ERR_error_string(ERR_get_error(), ssl_err_buf));
Andy Greene92cd172011-01-19 13:11:55 +00002395 return NULL;
Andy Green3faa9c72010-11-08 17:03:03 +00002396 }
2397 /* set the private key from KeyFile */
Peter Hinz56885f32011-03-02 22:03:47 +00002398 if (SSL_CTX_use_PrivateKey_file(context->ssl_ctx,
2399 ssl_private_key_filepath, SSL_FILETYPE_PEM) != 1) {
Andy Green018d8eb2010-11-08 21:04:23 +00002400 fprintf(stderr, "ssl problem getting key '%s': %s\n",
2401 ssl_private_key_filepath,
2402 ERR_error_string(ERR_get_error(), ssl_err_buf));
Andy Greene92cd172011-01-19 13:11:55 +00002403 return NULL;
Andy Green3faa9c72010-11-08 17:03:03 +00002404 }
2405 /* verify private key */
Peter Hinz56885f32011-03-02 22:03:47 +00002406 if (!SSL_CTX_check_private_key(context->ssl_ctx)) {
Andy Green018d8eb2010-11-08 21:04:23 +00002407 fprintf(stderr, "Private SSL key doesn't match cert\n");
Andy Greene92cd172011-01-19 13:11:55 +00002408 return NULL;
Andy Green3faa9c72010-11-08 17:03:03 +00002409 }
2410
2411 /* SSL is happy and has a cert it's content with */
2412 }
2413#endif
Andy Greenb45993c2010-12-18 15:13:50 +00002414
Andy Greendf736162011-01-18 15:39:02 +00002415 /* selftest */
2416
2417 if (lws_b64_selftest())
Andy Greene92cd172011-01-19 13:11:55 +00002418 return NULL;
Andy Greendf736162011-01-18 15:39:02 +00002419
Andy Green0d338332011-02-12 11:57:43 +00002420 /* fd hashtable init */
2421
2422 for (n = 0; n < FD_HASHTABLE_MODULUS; n++)
Peter Hinz56885f32011-03-02 22:03:47 +00002423 context->fd_hashtable[n].length = 0;
Andy Green0d338332011-02-12 11:57:43 +00002424
Andy Greenb45993c2010-12-18 15:13:50 +00002425 /* set up our external listening socket we serve on */
Andy Green8f037e42010-12-19 22:13:26 +00002426
Andy Green4739e5c2011-01-22 12:51:57 +00002427 if (port) {
Andy Green8f037e42010-12-19 22:13:26 +00002428
Andy Green4739e5c2011-01-22 12:51:57 +00002429 sockfd = socket(AF_INET, SOCK_STREAM, 0);
2430 if (sockfd < 0) {
2431 fprintf(stderr, "ERROR opening socket");
2432 return NULL;
2433 }
Andy Green775c0dd2010-10-29 14:15:22 +01002434
Andy Green4739e5c2011-01-22 12:51:57 +00002435 /* allow us to restart even if old sockets in TIME_WAIT */
2436 setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
Andy Greene77ddd82010-11-13 10:03:47 +00002437
Andy Green6c939552011-03-08 08:56:57 +00002438
2439 /* Disable Nagle */
2440 opt = 1;
2441 setsockopt(sockfd, SOL_TCP, TCP_NODELAY, &opt, sizeof(opt));
2442
Andy Green4739e5c2011-01-22 12:51:57 +00002443 bzero((char *) &serv_addr, sizeof(serv_addr));
2444 serv_addr.sin_family = AF_INET;
Peter Hinz56885f32011-03-02 22:03:47 +00002445 if (interf == NULL)
Andy Green32375b72011-02-19 08:32:53 +00002446 serv_addr.sin_addr.s_addr = INADDR_ANY;
2447 else
Peter Hinz56885f32011-03-02 22:03:47 +00002448 interface_to_sa(interf, &serv_addr,
Andy Green32375b72011-02-19 08:32:53 +00002449 sizeof(serv_addr));
Andy Green4739e5c2011-01-22 12:51:57 +00002450 serv_addr.sin_port = htons(port);
2451
2452 n = bind(sockfd, (struct sockaddr *) &serv_addr,
2453 sizeof(serv_addr));
2454 if (n < 0) {
2455 fprintf(stderr, "ERROR on binding to port %d (%d %d)\n",
Andy Green8f037e42010-12-19 22:13:26 +00002456 port, n, errno);
Andy Green4739e5c2011-01-22 12:51:57 +00002457 return NULL;
2458 }
Andy Green0d338332011-02-12 11:57:43 +00002459
2460 wsi = malloc(sizeof(struct libwebsocket));
2461 memset(wsi, 0, sizeof (struct libwebsocket));
2462 wsi->sock = sockfd;
Andy Greend6e09112011-03-05 16:12:15 +00002463 wsi->count_active_extensions = 0;
Andy Green0d338332011-02-12 11:57:43 +00002464 wsi->mode = LWS_CONNMODE_SERVER_LISTENER;
Peter Hinz56885f32011-03-02 22:03:47 +00002465 insert_wsi(context, wsi);
Andy Green0d338332011-02-12 11:57:43 +00002466
2467 listen(sockfd, 5);
2468 fprintf(stderr, " Listening on port %d\n", port);
2469
2470 /* list in the internal poll array */
2471
Peter Hinz56885f32011-03-02 22:03:47 +00002472 context->fds[context->fds_count].fd = sockfd;
2473 context->fds[context->fds_count++].events = POLLIN;
Andy Green3221f922011-02-12 13:14:11 +00002474
2475 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00002476 context->protocols[0].callback(context, wsi,
Andy Green3221f922011-02-12 13:14:11 +00002477 LWS_CALLBACK_ADD_POLL_FD,
2478 (void *)(long)sockfd, NULL, POLLIN);
2479
Andy Green8f037e42010-12-19 22:13:26 +00002480 }
Andy Greenb45993c2010-12-18 15:13:50 +00002481
Andy Greene77ddd82010-11-13 10:03:47 +00002482 /* drop any root privs for this process */
Peter Hinz56885f32011-03-02 22:03:47 +00002483#ifdef WIN32
2484#else
Andy Green3faa9c72010-11-08 17:03:03 +00002485 if (gid != -1)
2486 if (setgid(gid))
2487 fprintf(stderr, "setgid: %s\n", strerror(errno));
2488 if (uid != -1)
2489 if (setuid(uid))
2490 fprintf(stderr, "setuid: %s\n", strerror(errno));
Peter Hinz56885f32011-03-02 22:03:47 +00002491#endif
Andy Greenb45993c2010-12-18 15:13:50 +00002492
2493 /* set up our internal broadcast trigger sockets per-protocol */
2494
Peter Hinz56885f32011-03-02 22:03:47 +00002495 for (context->count_protocols = 0;
2496 protocols[context->count_protocols].callback;
2497 context->count_protocols++) {
2498 protocols[context->count_protocols].owning_server = context;
2499 protocols[context->count_protocols].protocol_index =
2500 context->count_protocols;
Andy Greenb45993c2010-12-18 15:13:50 +00002501
2502 fd = socket(AF_INET, SOCK_STREAM, 0);
2503 if (fd < 0) {
2504 fprintf(stderr, "ERROR opening socket");
Andy Greene92cd172011-01-19 13:11:55 +00002505 return NULL;
Andy Greenb45993c2010-12-18 15:13:50 +00002506 }
Andy Green8f037e42010-12-19 22:13:26 +00002507
Andy Greenb45993c2010-12-18 15:13:50 +00002508 /* allow us to restart even if old sockets in TIME_WAIT */
2509 setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
2510
2511 bzero((char *) &serv_addr, sizeof(serv_addr));
2512 serv_addr.sin_family = AF_INET;
2513 serv_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
2514 serv_addr.sin_port = 0; /* pick the port for us */
2515
2516 n = bind(fd, (struct sockaddr *) &serv_addr, sizeof(serv_addr));
2517 if (n < 0) {
Andy Green8f037e42010-12-19 22:13:26 +00002518 fprintf(stderr, "ERROR on binding to port %d (%d %d)\n",
Andy Greenb45993c2010-12-18 15:13:50 +00002519 port, n, errno);
Andy Greene92cd172011-01-19 13:11:55 +00002520 return NULL;
Andy Greenb45993c2010-12-18 15:13:50 +00002521 }
2522
2523 slen = sizeof cli_addr;
2524 n = getsockname(fd, (struct sockaddr *)&cli_addr, &slen);
2525 if (n < 0) {
2526 fprintf(stderr, "getsockname failed\n");
Andy Greene92cd172011-01-19 13:11:55 +00002527 return NULL;
Andy Greenb45993c2010-12-18 15:13:50 +00002528 }
Peter Hinz56885f32011-03-02 22:03:47 +00002529 protocols[context->count_protocols].broadcast_socket_port =
Andy Greenb45993c2010-12-18 15:13:50 +00002530 ntohs(cli_addr.sin_port);
2531 listen(fd, 5);
2532
2533 debug(" Protocol %s broadcast socket %d\n",
Peter Hinz56885f32011-03-02 22:03:47 +00002534 protocols[context->count_protocols].name,
Andy Greenb45993c2010-12-18 15:13:50 +00002535 ntohs(cli_addr.sin_port));
2536
Andy Green0d338332011-02-12 11:57:43 +00002537 /* dummy wsi per broadcast proxy socket */
2538
2539 wsi = malloc(sizeof(struct libwebsocket));
2540 memset(wsi, 0, sizeof (struct libwebsocket));
2541 wsi->sock = fd;
2542 wsi->mode = LWS_CONNMODE_BROADCAST_PROXY_LISTENER;
Andy Greend6e09112011-03-05 16:12:15 +00002543 wsi->count_active_extensions = 0;
Andy Green0d338332011-02-12 11:57:43 +00002544 /* note which protocol we are proxying */
Peter Hinz56885f32011-03-02 22:03:47 +00002545 wsi->protocol_index_for_broadcast_proxy =
2546 context->count_protocols;
2547 insert_wsi(context, wsi);
Andy Green0d338332011-02-12 11:57:43 +00002548
2549 /* list in internal poll array */
2550
Peter Hinz56885f32011-03-02 22:03:47 +00002551 context->fds[context->fds_count].fd = fd;
2552 context->fds[context->fds_count].events = POLLIN;
2553 context->fds[context->fds_count].revents = 0;
2554 context->fds_count++;
Andy Green3221f922011-02-12 13:14:11 +00002555
2556 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00002557 context->protocols[0].callback(context, wsi,
Andy Green3221f922011-02-12 13:14:11 +00002558 LWS_CALLBACK_ADD_POLL_FD,
2559 (void *)(long)fd, NULL, POLLIN);
Andy Greenb45993c2010-12-18 15:13:50 +00002560 }
2561
Peter Hinz56885f32011-03-02 22:03:47 +00002562 return context;
Andy Greene92cd172011-01-19 13:11:55 +00002563}
Andy Greenb45993c2010-12-18 15:13:50 +00002564
Andy Green4739e5c2011-01-22 12:51:57 +00002565
Andy Greened11a022011-01-20 10:23:50 +00002566#ifndef LWS_NO_FORK
2567
Andy Greene92cd172011-01-19 13:11:55 +00002568/**
2569 * libwebsockets_fork_service_loop() - Optional helper function forks off
2570 * a process for the websocket server loop.
Andy Green6964bb52011-01-23 16:50:33 +00002571 * You don't have to use this but if not, you
2572 * have to make sure you are calling
2573 * libwebsocket_service periodically to service
2574 * the websocket traffic
Peter Hinz56885f32011-03-02 22:03:47 +00002575 * @context: server context returned by creation function
Andy Greene92cd172011-01-19 13:11:55 +00002576 */
Andy Greenb45993c2010-12-18 15:13:50 +00002577
Andy Greene92cd172011-01-19 13:11:55 +00002578int
Peter Hinz56885f32011-03-02 22:03:47 +00002579libwebsockets_fork_service_loop(struct libwebsocket_context *context)
Andy Greene92cd172011-01-19 13:11:55 +00002580{
Andy Greene92cd172011-01-19 13:11:55 +00002581 int fd;
2582 struct sockaddr_in cli_addr;
2583 int n;
Andy Green3221f922011-02-12 13:14:11 +00002584 int p;
Andy Greenb45993c2010-12-18 15:13:50 +00002585
Andy Greened11a022011-01-20 10:23:50 +00002586 n = fork();
2587 if (n < 0)
2588 return n;
2589
2590 if (!n) {
2591
2592 /* main process context */
2593
Andy Green3221f922011-02-12 13:14:11 +00002594 /*
2595 * set up the proxy sockets to allow broadcast from
2596 * service process context
2597 */
2598
Peter Hinz56885f32011-03-02 22:03:47 +00002599 for (p = 0; p < context->count_protocols; p++) {
Andy Greened11a022011-01-20 10:23:50 +00002600 fd = socket(AF_INET, SOCK_STREAM, 0);
2601 if (fd < 0) {
2602 fprintf(stderr, "Unable to create socket\n");
2603 return -1;
2604 }
2605 cli_addr.sin_family = AF_INET;
2606 cli_addr.sin_port = htons(
Peter Hinz56885f32011-03-02 22:03:47 +00002607 context->protocols[p].broadcast_socket_port);
Andy Greened11a022011-01-20 10:23:50 +00002608 cli_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
2609 n = connect(fd, (struct sockaddr *)&cli_addr,
2610 sizeof cli_addr);
2611 if (n < 0) {
2612 fprintf(stderr, "Unable to connect to "
2613 "broadcast socket %d, %s\n",
Andy Green3221f922011-02-12 13:14:11 +00002614 n, strerror(errno));
Andy Greened11a022011-01-20 10:23:50 +00002615 return -1;
2616 }
2617
Peter Hinz56885f32011-03-02 22:03:47 +00002618 context->protocols[p].broadcast_socket_user_fd = fd;
Andy Greened11a022011-01-20 10:23:50 +00002619 }
2620
Andy Greene92cd172011-01-19 13:11:55 +00002621 return 0;
Andy Greenb45993c2010-12-18 15:13:50 +00002622 }
2623
2624 /* we want a SIGHUP when our parent goes down */
2625 prctl(PR_SET_PDEATHSIG, SIGHUP);
2626
2627 /* in this forked process, sit and service websocket connections */
Andy Green8f037e42010-12-19 22:13:26 +00002628
Andy Greene92cd172011-01-19 13:11:55 +00002629 while (1)
Peter Hinz56885f32011-03-02 22:03:47 +00002630 if (libwebsocket_service(context, 1000))
Andy Greene92cd172011-01-19 13:11:55 +00002631 return -1;
Andy Green8f037e42010-12-19 22:13:26 +00002632
Andy Green251f6fa2010-11-03 11:13:06 +00002633 return 0;
Andy Greenff95d7a2010-10-28 22:36:01 +01002634}
2635
Andy Greened11a022011-01-20 10:23:50 +00002636#endif
2637
Andy Greenb45993c2010-12-18 15:13:50 +00002638/**
2639 * libwebsockets_get_protocol() - Returns a protocol pointer from a websocket
Andy Green8f037e42010-12-19 22:13:26 +00002640 * connection.
Andy Greenb45993c2010-12-18 15:13:50 +00002641 * @wsi: pointer to struct websocket you want to know the protocol of
2642 *
Andy Green8f037e42010-12-19 22:13:26 +00002643 *
2644 * This is useful to get the protocol to broadcast back to from inside
Andy Greenb45993c2010-12-18 15:13:50 +00002645 * the callback.
2646 */
Andy Greenab990e42010-10-31 12:42:52 +00002647
Andy Greenb45993c2010-12-18 15:13:50 +00002648const struct libwebsocket_protocols *
2649libwebsockets_get_protocol(struct libwebsocket *wsi)
2650{
2651 return wsi->protocol;
2652}
2653
2654/**
Andy Greene92cd172011-01-19 13:11:55 +00002655 * libwebsockets_broadcast() - Sends a buffer to the callback for all active
Andy Green8f037e42010-12-19 22:13:26 +00002656 * connections of the given protocol.
Andy Greenb45993c2010-12-18 15:13:50 +00002657 * @protocol: pointer to the protocol you will broadcast to all members of
2658 * @buf: buffer containing the data to be broadcase. NOTE: this has to be
Andy Green8f037e42010-12-19 22:13:26 +00002659 * allocated with LWS_SEND_BUFFER_PRE_PADDING valid bytes before
2660 * the pointer and LWS_SEND_BUFFER_POST_PADDING afterwards in the
2661 * case you are calling this function from callback context.
Andy Greenb45993c2010-12-18 15:13:50 +00002662 * @len: length of payload data in buf, starting from buf.
Andy Green8f037e42010-12-19 22:13:26 +00002663 *
2664 * This function allows bulk sending of a packet to every connection using
Andy Greenb45993c2010-12-18 15:13:50 +00002665 * the given protocol. It does not send the data directly; instead it calls
2666 * the callback with a reason type of LWS_CALLBACK_BROADCAST. If the callback
2667 * wants to actually send the data for that connection, the callback itself
2668 * should call libwebsocket_write().
2669 *
2670 * libwebsockets_broadcast() can be called from another fork context without
2671 * having to take any care about data visibility between the processes, it'll
2672 * "just work".
2673 */
2674
2675
2676int
Andy Green8f037e42010-12-19 22:13:26 +00002677libwebsockets_broadcast(const struct libwebsocket_protocols *protocol,
Andy Greenb45993c2010-12-18 15:13:50 +00002678 unsigned char *buf, size_t len)
2679{
Peter Hinz56885f32011-03-02 22:03:47 +00002680 struct libwebsocket_context *context = protocol->owning_server;
Andy Greenb45993c2010-12-18 15:13:50 +00002681 int n;
Andy Green0d338332011-02-12 11:57:43 +00002682 int m;
2683 struct libwebsocket * wsi;
Andy Greenb45993c2010-12-18 15:13:50 +00002684
2685 if (!protocol->broadcast_socket_user_fd) {
2686 /*
Andy Greene92cd172011-01-19 13:11:55 +00002687 * We are either running unforked / flat, or we are being
2688 * called from poll thread context
Andy Greenb45993c2010-12-18 15:13:50 +00002689 * eg, from a callback. In that case don't use sockets for
2690 * broadcast IPC (since we can't open a socket connection to
2691 * a socket listening on our own thread) but directly do the
2692 * send action.
2693 *
2694 * Locking is not needed because we are by definition being
2695 * called in the poll thread context and are serialized.
2696 */
2697
Andy Green0d338332011-02-12 11:57:43 +00002698 for (n = 0; n < FD_HASHTABLE_MODULUS; n++) {
Andy Greenb45993c2010-12-18 15:13:50 +00002699
Peter Hinz56885f32011-03-02 22:03:47 +00002700 for (m = 0; m < context->fd_hashtable[n].length; m++) {
Andy Greenb45993c2010-12-18 15:13:50 +00002701
Peter Hinz56885f32011-03-02 22:03:47 +00002702 wsi = context->fd_hashtable[n].wsi[m];
Andy Greenb45993c2010-12-18 15:13:50 +00002703
Andy Green0d338332011-02-12 11:57:43 +00002704 if (wsi->mode != LWS_CONNMODE_WS_SERVING)
2705 continue;
Andy Greenb45993c2010-12-18 15:13:50 +00002706
Andy Green0d338332011-02-12 11:57:43 +00002707 /*
2708 * never broadcast to
2709 * non-established connections
2710 */
2711 if (wsi->state != WSI_STATE_ESTABLISHED)
2712 continue;
2713
2714 /* only broadcast to guys using
2715 * requested protocol
2716 */
2717 if (wsi->protocol != protocol)
2718 continue;
2719
Peter Hinz56885f32011-03-02 22:03:47 +00002720 wsi->protocol->callback(context, wsi,
Andy Green8f037e42010-12-19 22:13:26 +00002721 LWS_CALLBACK_BROADCAST,
Andy Green0d338332011-02-12 11:57:43 +00002722 wsi->user_space,
Andy Greenb45993c2010-12-18 15:13:50 +00002723 buf, len);
Andy Green0d338332011-02-12 11:57:43 +00002724 }
Andy Greenb45993c2010-12-18 15:13:50 +00002725 }
2726
2727 return 0;
2728 }
2729
Andy Green0ca6a172010-12-19 20:50:01 +00002730 /*
2731 * We're being called from a different process context than the server
2732 * loop. Instead of broadcasting directly, we send our
2733 * payload on a socket to do the IPC; the server process will serialize
2734 * the broadcast action in its main poll() loop.
2735 *
2736 * There's one broadcast socket listening for each protocol supported
2737 * set up when the websocket server initializes
2738 */
2739
Andy Green6964bb52011-01-23 16:50:33 +00002740 n = send(protocol->broadcast_socket_user_fd, buf, len, MSG_NOSIGNAL);
Andy Greenb45993c2010-12-18 15:13:50 +00002741
2742 return n;
2743}
Andy Green82c3d542011-03-07 21:16:31 +00002744
2745int
2746libwebsocket_is_final_fragment(struct libwebsocket *wsi)
2747{
2748 return wsi->final;
2749}