blob: 0021eb449fa63cb23a98c98bc4b795d5b97829ac [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 Greena41314f2011-05-23 10:00:03 +0100156 struct libwebsocket_extension *ext;
Andy Greenb45993c2010-12-18 15:13:50 +0000157
Andy Green4b6fbe12011-02-14 08:03:48 +0000158 if (!wsi)
Andy Greenb45993c2010-12-18 15:13:50 +0000159 return;
160
Andy Green62c54d22011-02-14 09:14:25 +0000161 old_state = wsi->state;
Andy Green251f6fa2010-11-03 11:13:06 +0000162
Andy Green62c54d22011-02-14 09:14:25 +0000163 if (old_state == WSI_STATE_DEAD_SOCKET)
Andy Green5e1fa172011-02-10 09:07:05 +0000164 return;
165
Andy Greenda527df2011-03-07 07:08:12 +0000166 wsi->close_reason = reason;
167
168 /*
Andy Green68b45042011-05-25 21:41:57 +0100169 * are his extensions okay with him closing? Eg he might be a mux
170 * parent and just his ch1 aspect is closing?
171 */
172
173
174 for (n = 0; n < wsi->count_active_extensions; n++) {
175 if (!wsi->active_extensions[n]->callback)
176 continue;
177
178 m = wsi->active_extensions[n]->callback(context,
179 wsi->active_extensions[n], wsi,
180 LWS_EXT_CALLBACK_CHECK_OK_TO_REALLY_CLOSE,
181 wsi->active_extensions_user[n], NULL, 0);
182
183 /*
184 * if somebody vetoed actually closing him at this time....
185 * up to the extension to track the attempted close, let's
186 * just bail
187 */
188
189 if (m) {
190 fprintf(stderr, "extension vetoed close\n");
191 return;
192 }
193 }
194
195
196
197 /*
Andy Greenc44159f2011-03-07 07:08:18 +0000198 * flush any tx pending from extensions, since we may send close packet
199 * if there are problems with send, just nuke the connection
200 */
201
202 ret = 1;
203 while (ret == 1) {
204
205 /* default to nobody has more to spill */
206
207 ret = 0;
208 eff_buf.token = NULL;
209 eff_buf.token_len = 0;
210
211 /* show every extension the new incoming data */
212
213 for (n = 0; n < wsi->count_active_extensions; n++) {
214 m = wsi->active_extensions[n]->callback(
Andy Green46c2ea02011-03-22 09:04:01 +0000215 wsi->protocol->owning_server,
216 wsi->active_extensions[n], wsi,
Andy Greenc44159f2011-03-07 07:08:18 +0000217 LWS_EXT_CALLBACK_FLUSH_PENDING_TX,
218 wsi->active_extensions_user[n], &eff_buf, 0);
219 if (m < 0) {
220 fprintf(stderr, "Extension reports "
221 "fatal error\n");
222 goto just_kill_connection;
223 }
224 if (m)
225 /*
226 * at least one extension told us he has more
227 * to spill, so we will go around again after
228 */
229 ret = 1;
230 }
231
232 /* assuming they left us something to send, send it */
233
234 if (eff_buf.token_len)
235 if (lws_issue_raw(wsi, (unsigned char *)eff_buf.token,
236 eff_buf.token_len))
237 goto just_kill_connection;
238 }
239
240 /*
Andy Greenda527df2011-03-07 07:08:12 +0000241 * signal we are closing, libsocket_write will
242 * add any necessary version-specific stuff. If the write fails,
243 * no worries we are closing anyway. If we didn't initiate this
244 * close, then our state has been changed to
245 * WSI_STATE_RETURNED_CLOSE_ALREADY and we will skip this.
246 *
247 * Likewise if it's a second call to close this connection after we
248 * sent the close indication to the peer already, we are in state
249 * WSI_STATE_AWAITING_CLOSE_ACK and will skip doing this a second time.
250 */
251
252 if (old_state == WSI_STATE_ESTABLISHED &&
253 reason != LWS_CLOSE_STATUS_NOSTATUS) {
Andy Green66a16f32011-05-24 22:07:45 +0100254
255 fprintf(stderr, "sending close indication...\n");
256
Andy Greenda527df2011-03-07 07:08:12 +0000257 n = libwebsocket_write(wsi, &buf[LWS_SEND_BUFFER_PRE_PADDING],
258 0, LWS_WRITE_CLOSE);
259 if (!n) {
260 /*
261 * we have sent a nice protocol level indication we
262 * now wish to close, we should not send anything more
263 */
264
265 wsi->state = WSI_STATE_AWAITING_CLOSE_ACK;
266
267 /* and we should wait for a reply for a bit */
268
269 libwebsocket_set_timeout(wsi,
270 PENDING_TIMEOUT_CLOSE_ACK, 5);
271
272 fprintf(stderr, "sent close indication, awaiting ack\n");
273
274 return;
275 }
276
277 /* else, the send failed and we should just hang up */
278 }
279
Andy Greenc44159f2011-03-07 07:08:18 +0000280just_kill_connection:
Andy Green66a16f32011-05-24 22:07:45 +0100281
282 fprintf(stderr, "libwebsocket_close_and_free_session: just_kill_connection\n");
283
Andy Greenda527df2011-03-07 07:08:12 +0000284 /*
285 * we won't be servicing or receiving anything further from this guy
286 * remove this fd from wsi mapping hashtable
287 */
Andy Green4b6fbe12011-02-14 08:03:48 +0000288
Andy Greena41314f2011-05-23 10:00:03 +0100289 if (wsi->sock)
290 delete_from_fd(context, wsi->sock);
Andy Green4b6fbe12011-02-14 08:03:48 +0000291
292 /* delete it from the internal poll list if still present */
293
Peter Hinz56885f32011-03-02 22:03:47 +0000294 for (n = 0; n < context->fds_count; n++) {
295 if (context->fds[n].fd != wsi->sock)
Andy Green4b6fbe12011-02-14 08:03:48 +0000296 continue;
Peter Hinz56885f32011-03-02 22:03:47 +0000297 while (n < context->fds_count - 1) {
298 context->fds[n] = context->fds[n + 1];
Andy Green4b6fbe12011-02-14 08:03:48 +0000299 n++;
300 }
Peter Hinz56885f32011-03-02 22:03:47 +0000301 context->fds_count--;
Andy Green4b6fbe12011-02-14 08:03:48 +0000302 /* we only have to deal with one */
Peter Hinz56885f32011-03-02 22:03:47 +0000303 n = context->fds_count;
Andy Green4b6fbe12011-02-14 08:03:48 +0000304 }
305
306 /* remove also from external POLL support via protocol 0 */
Andy Greena41314f2011-05-23 10:00:03 +0100307 if (wsi->sock)
308 context->protocols[0].callback(context, wsi,
Andy Green4b6fbe12011-02-14 08:03:48 +0000309 LWS_CALLBACK_DEL_POLL_FD, (void *)(long)wsi->sock, NULL, 0);
310
Andy Green251f6fa2010-11-03 11:13:06 +0000311 wsi->state = WSI_STATE_DEAD_SOCKET;
312
Andy Green4b6fbe12011-02-14 08:03:48 +0000313 /* tell the user it's all over for this guy */
314
Andy Greend4302732011-02-28 07:45:29 +0000315 if (wsi->protocol && wsi->protocol->callback &&
Andy Green66a16f32011-05-24 22:07:45 +0100316 ((old_state == WSI_STATE_ESTABLISHED) ||
317 (old_state == WSI_STATE_RETURNED_CLOSE_ALREADY) ||
318 (old_state == WSI_STATE_AWAITING_CLOSE_ACK))) {
319 fprintf(stderr, "calling back CLOSED\n");
Peter Hinz56885f32011-03-02 22:03:47 +0000320 wsi->protocol->callback(context, wsi, LWS_CALLBACK_CLOSED,
Andy Greene77ddd82010-11-13 10:03:47 +0000321 wsi->user_space, NULL, 0);
Andy Green66a16f32011-05-24 22:07:45 +0100322 } else {
323 fprintf(stderr, "not calling back closed due to old_state=%d\n", old_state);
324 }
Andy Green251f6fa2010-11-03 11:13:06 +0000325
Andy Greenef660a92011-03-06 10:29:38 +0000326 /* deallocate any active extension contexts */
327
328 for (n = 0; n < wsi->count_active_extensions; n++) {
329 if (!wsi->active_extensions[n]->callback)
330 continue;
331
Andy Green46c2ea02011-03-22 09:04:01 +0000332 wsi->active_extensions[n]->callback(context,
333 wsi->active_extensions[n], wsi,
334 LWS_EXT_CALLBACK_DESTROY,
335 wsi->active_extensions_user[n], NULL, 0);
Andy Greenef660a92011-03-06 10:29:38 +0000336
337 free(wsi->active_extensions_user[n]);
338 }
339
Andy Greena41314f2011-05-23 10:00:03 +0100340 /*
341 * inform all extensions in case they tracked this guy out of band
342 * even though not active on him specifically
343 */
344
345 ext = context->extensions;
346 while (ext && ext->callback) {
347 ext->callback(context, ext, wsi,
348 LWS_EXT_CALLBACK_DESTROY_ANY_WSI_CLOSING,
349 NULL, NULL, 0);
350 ext++;
351 }
352
Andy Greenef660a92011-03-06 10:29:38 +0000353 /* free up his parsing allocations */
Andy Green4b6fbe12011-02-14 08:03:48 +0000354
Andy Green251f6fa2010-11-03 11:13:06 +0000355 for (n = 0; n < WSI_TOKEN_COUNT; n++)
356 if (wsi->utf8_token[n].token)
357 free(wsi->utf8_token[n].token);
358
Andy Greena41314f2011-05-23 10:00:03 +0100359 if (wsi->c_address)
360 free(wsi->c_address);
361
Andy Green0ca6a172010-12-19 20:50:01 +0000362/* fprintf(stderr, "closing fd=%d\n", wsi->sock); */
Andy Green251f6fa2010-11-03 11:13:06 +0000363
Andy Green3faa9c72010-11-08 17:03:03 +0000364#ifdef LWS_OPENSSL_SUPPORT
Andy Green90c7cbc2011-01-27 06:26:52 +0000365 if (wsi->ssl) {
Andy Green3faa9c72010-11-08 17:03:03 +0000366 n = SSL_get_fd(wsi->ssl);
367 SSL_shutdown(wsi->ssl);
Peter Hinz56885f32011-03-02 22:03:47 +0000368#ifdef WIN32
369 closesocket(n);
370#else
Andy Green3faa9c72010-11-08 17:03:03 +0000371 close(n);
Peter Hinz56885f32011-03-02 22:03:47 +0000372#endif
Andy Green3faa9c72010-11-08 17:03:03 +0000373 SSL_free(wsi->ssl);
374 } else {
375#endif
376 shutdown(wsi->sock, SHUT_RDWR);
Peter Hinz56885f32011-03-02 22:03:47 +0000377#ifdef WIN32
Andy Green66a16f32011-05-24 22:07:45 +0100378 if (wsi->sock)
379 closesocket(wsi->sock);
Peter Hinz56885f32011-03-02 22:03:47 +0000380#else
Andy Green66a16f32011-05-24 22:07:45 +0100381 if (wsi->sock)
382 close(wsi->sock);
Peter Hinz56885f32011-03-02 22:03:47 +0000383#endif
Andy Green3faa9c72010-11-08 17:03:03 +0000384#ifdef LWS_OPENSSL_SUPPORT
385 }
386#endif
Andy Green4f3943a2010-11-12 10:44:16 +0000387 if (wsi->user_space)
388 free(wsi->user_space);
389
Andy Green251f6fa2010-11-03 11:13:06 +0000390 free(wsi);
391}
392
Andy Green07034092011-02-13 08:37:12 +0000393/**
Andy Greenf7ee5492011-02-13 09:04:21 +0000394 * libwebsockets_hangup_on_client() - Server calls to terminate client
395 * connection
Peter Hinz56885f32011-03-02 22:03:47 +0000396 * @context: libwebsockets context
Andy Greenf7ee5492011-02-13 09:04:21 +0000397 * @fd: Connection socket descriptor
398 */
399
400void
Peter Hinz56885f32011-03-02 22:03:47 +0000401libwebsockets_hangup_on_client(struct libwebsocket_context *context, int fd)
Andy Greenf7ee5492011-02-13 09:04:21 +0000402{
Peter Hinz56885f32011-03-02 22:03:47 +0000403 struct libwebsocket *wsi = wsi_from_fd(context, fd);
Andy Greenf7ee5492011-02-13 09:04:21 +0000404
405 if (wsi == NULL)
406 return;
407
Peter Hinz56885f32011-03-02 22:03:47 +0000408 libwebsocket_close_and_free_session(context, wsi,
Andy Green6da560c2011-02-26 11:06:27 +0000409 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenf7ee5492011-02-13 09:04:21 +0000410}
411
412
413/**
Andy Green07034092011-02-13 08:37:12 +0000414 * libwebsockets_get_peer_addresses() - Get client address information
415 * @fd: Connection socket descriptor
416 * @name: Buffer to take client address name
417 * @name_len: Length of client address name buffer
418 * @rip: Buffer to take client address IP qotted quad
419 * @rip_len: Length of client address IP buffer
420 *
421 * This function fills in @name and @rip with the name and IP of
422 * the client connected with socket descriptor @fd. Names may be
423 * truncated if there is not enough room. If either cannot be
424 * determined, they will be returned as valid zero-length strings.
425 */
426
427void
428libwebsockets_get_peer_addresses(int fd, char *name, int name_len,
429 char *rip, int rip_len)
430{
431 unsigned int len;
432 struct sockaddr_in sin;
433 struct hostent *host;
434 struct hostent *host1;
435 char ip[128];
Andy Greenf92def72011-03-09 15:02:20 +0000436 unsigned char *p;
Andy Green07034092011-02-13 08:37:12 +0000437 int n;
Andy Green7627af52011-03-09 15:13:52 +0000438 struct sockaddr_un *un;
Andy Green07034092011-02-13 08:37:12 +0000439
440 rip[0] = '\0';
441 name[0] = '\0';
442
443 len = sizeof sin;
444 if (getpeername(fd, (struct sockaddr *) &sin, &len) < 0) {
445 perror("getpeername");
446 return;
447 }
448
449 host = gethostbyaddr((char *) &sin.sin_addr, sizeof sin.sin_addr,
450 AF_INET);
451 if (host == NULL) {
452 perror("gethostbyaddr");
453 return;
454 }
455
456 strncpy(name, host->h_name, name_len);
457 name[name_len - 1] = '\0';
458
459 host1 = gethostbyname(host->h_name);
460 if (host1 == NULL)
461 return;
Andy Greenf92def72011-03-09 15:02:20 +0000462 p = (unsigned char *)host1;
Andy Green07034092011-02-13 08:37:12 +0000463 n = 0;
464 while (p != NULL) {
Andy Greenf92def72011-03-09 15:02:20 +0000465 p = (unsigned char *)host1->h_addr_list[n++];
Andy Green07034092011-02-13 08:37:12 +0000466 if (p == NULL)
467 continue;
Peter Hinzbb45a902011-03-10 18:14:01 +0000468 if ((host1->h_addrtype != AF_INET)
469#ifdef AF_LOCAL
470 && (host1->h_addrtype != AF_LOCAL)
471#endif
472 )
Andy Green07034092011-02-13 08:37:12 +0000473 continue;
474
Andy Green7627af52011-03-09 15:13:52 +0000475 if (host1->h_addrtype == AF_INET)
476 sprintf(ip, "%u.%u.%u.%u", p[0], p[1], p[2], p[3]);
Peter Hinzbb45a902011-03-10 18:14:01 +0000477#ifdef AF_LOCAL
Andy Green7627af52011-03-09 15:13:52 +0000478 else {
479 un = (struct sockaddr_un *)p;
480 strncpy(ip, un->sun_path, sizeof(ip) -1);
481 ip[sizeof(ip) - 1] = '\0';
482 }
Peter Hinzbb45a902011-03-10 18:14:01 +0000483#endif
Andy Green07034092011-02-13 08:37:12 +0000484 p = NULL;
485 strncpy(rip, ip, rip_len);
486 rip[rip_len - 1] = '\0';
487 }
488}
Andy Green9f990342011-02-12 11:57:45 +0000489
Peter Hinz56885f32011-03-02 22:03:47 +0000490int libwebsockets_get_random(struct libwebsocket_context *context,
491 void *buf, int len)
492{
493 int n;
494 char *p = buf;
495
496#ifdef WIN32
497 for (n = 0; n < len; n++)
498 p[n] = (unsigned char)rand();
499#else
500 n = read(context->fd_random, p, len);
501#endif
502
503 return n;
504}
505
Andy Green2836c642011-03-07 20:47:41 +0000506unsigned char *
507libwebsockets_SHA1(const unsigned char *d, size_t n, unsigned char *md)
508{
509 return SHA1(d, n, md);
510}
511
Andy Greeneeaacb32011-03-01 20:44:24 +0000512void libwebsockets_00_spaceout(char *key, int spaces, int seed)
513{
514 char *p;
Peter Hinz56885f32011-03-02 22:03:47 +0000515
Andy Greeneeaacb32011-03-01 20:44:24 +0000516 key++;
517 while (spaces--) {
518 if (*key && (seed & 1))
519 key++;
520 seed >>= 1;
Peter Hinz56885f32011-03-02 22:03:47 +0000521
Andy Greeneeaacb32011-03-01 20:44:24 +0000522 p = key + strlen(key);
523 while (p >= key) {
524 p[1] = p[0];
525 p--;
526 }
527 *key++ = ' ';
528 }
529}
530
531void libwebsockets_00_spam(char *key, int count, int seed)
532{
533 char *p;
534
535 key++;
536 while (count--) {
537
538 if (*key && (seed & 1))
539 key++;
540 seed >>= 1;
541
542 p = key + strlen(key);
543 while (p >= key) {
544 p[1] = p[0];
545 p--;
546 }
547 *key++ = 0x21 + ((seed & 0xffff) % 15);
548 /* 4 would use it up too fast.. not like it matters */
549 seed >>= 1;
550 }
551}
552
Andy Green95a7b5d2011-03-06 10:29:39 +0000553int lws_send_pipe_choked(struct libwebsocket *wsi)
554{
555 struct pollfd fds;
556
557 fds.fd = wsi->sock;
558 fds.events = POLLOUT;
559 fds.revents = 0;
560
561 if (poll(&fds, 1, 0) != 1)
562 return 1;
563
564 if ((fds.revents & POLLOUT) == 0)
565 return 1;
566
567 /* okay to send another packet without blocking */
568
569 return 0;
570}
571
Andy Greena41314f2011-05-23 10:00:03 +0100572int
Andy Green3b84c002011-03-06 13:14:42 +0000573lws_handle_POLLOUT_event(struct libwebsocket_context *context,
574 struct libwebsocket *wsi, struct pollfd *pollfd)
575{
576 struct lws_tokens eff_buf;
577 int n;
578 int ret;
579 int m;
Andy Greena41314f2011-05-23 10:00:03 +0100580 int handled = 0;
Andy Green3b84c002011-03-06 13:14:42 +0000581
Andy Greena41314f2011-05-23 10:00:03 +0100582 for (n = 0; n < wsi->count_active_extensions; n++) {
583 if (!wsi->active_extensions[n]->callback)
584 continue;
585
586 m = wsi->active_extensions[n]->callback(context,
587 wsi->active_extensions[n], wsi,
588 LWS_EXT_CALLBACK_IS_WRITEABLE,
589 wsi->active_extensions_user[n], NULL, 0);
590 if (m > handled)
591 handled = m;
592 }
593
594 if (handled == 1)
595 goto notify_action;
596
597 if (!wsi->extension_data_pending || handled == 2)
Andy Green3b84c002011-03-06 13:14:42 +0000598 goto user_service;
599
600 /*
601 * check in on the active extensions, see if they
602 * had pending stuff to spill... they need to get the
603 * first look-in otherwise sequence will be disordered
604 *
605 * NULL, zero-length eff_buf means just spill pending
606 */
607
608 ret = 1;
609 while (ret == 1) {
610
611 /* default to nobody has more to spill */
612
613 ret = 0;
614 eff_buf.token = NULL;
615 eff_buf.token_len = 0;
616
617 /* give every extension a chance to spill */
618
619 for (n = 0; n < wsi->count_active_extensions; n++) {
620 m = wsi->active_extensions[n]->callback(
Andy Green46c2ea02011-03-22 09:04:01 +0000621 wsi->protocol->owning_server,
622 wsi->active_extensions[n], wsi,
Andy Green3b84c002011-03-06 13:14:42 +0000623 LWS_EXT_CALLBACK_PACKET_TX_PRESEND,
624 wsi->active_extensions_user[n], &eff_buf, 0);
625 if (m < 0) {
626 fprintf(stderr, "extension reports fatal error\n");
627 return -1;
628 }
629 if (m)
630 /*
631 * at least one extension told us he has more
632 * to spill, so we will go around again after
633 */
634 ret = 1;
635 }
636
637 /* assuming they gave us something to send, send it */
638
639 if (eff_buf.token_len) {
640 if (lws_issue_raw(wsi, (unsigned char *)eff_buf.token,
641 eff_buf.token_len))
642 return -1;
643 } else
644 continue;
645
646 /* no extension has more to spill */
647
648 if (!ret)
649 continue;
650
651 /*
652 * There's more to spill from an extension, but we just sent
653 * something... did that leave the pipe choked?
654 */
655
656 if (!lws_send_pipe_choked(wsi))
657 /* no we could add more */
658 continue;
659
660 fprintf(stderr, "choked in POLLOUT service\n");
661
662 /*
663 * Yes, he's choked. Leave the POLLOUT masked on so we will
664 * come back here when he is unchoked. Don't call the user
665 * callback to enforce ordering of spilling, he'll get called
666 * when we come back here and there's nothing more to spill.
667 */
668
669 return 0;
670 }
671
672 wsi->extension_data_pending = 0;
673
674user_service:
675 /* one shot */
676
Andy Greena41314f2011-05-23 10:00:03 +0100677 if (pollfd) {
678 pollfd->events &= ~POLLOUT;
Andy Green3b84c002011-03-06 13:14:42 +0000679
Andy Greena41314f2011-05-23 10:00:03 +0100680 /* external POLL support via protocol 0 */
681 context->protocols[0].callback(context, wsi,
682 LWS_CALLBACK_CLEAR_MODE_POLL_FD,
683 (void *)(long)wsi->sock, NULL, POLLOUT);
684 }
685
686notify_action:
Andy Green3b84c002011-03-06 13:14:42 +0000687
Andy Green9e4c2b62011-03-07 20:47:39 +0000688 if (wsi->mode == LWS_CONNMODE_WS_CLIENT)
689 n = LWS_CALLBACK_CLIENT_WRITEABLE;
690 else
691 n = LWS_CALLBACK_SERVER_WRITEABLE;
692
693 wsi->protocol->callback(context, wsi, n, wsi->user_space, NULL, 0);
Andy Green3b84c002011-03-06 13:14:42 +0000694
695 return 0;
696}
697
698
699
Andy Greena41314f2011-05-23 10:00:03 +0100700void
701libwebsocket_service_timeout_check(struct libwebsocket_context *context,
702 struct libwebsocket *wsi, unsigned int sec)
703{
704 int n;
705
706 /*
707 * if extensions want in on it (eg, we are a mux parent)
708 * give them a chance to service child timeouts
709 */
710
711 for (n = 0; n < wsi->count_active_extensions; n++)
712 wsi->active_extensions[n]->callback(
713 context, wsi->active_extensions[n],
714 wsi, LWS_EXT_CALLBACK_1HZ,
715 wsi->active_extensions_user[n], NULL, sec);
716
717 if (!wsi->pending_timeout)
718 return;
719
720 /*
721 * if we went beyond the allowed time, kill the
722 * connection
723 */
724
725 if (sec > wsi->pending_timeout_limit) {
726 fprintf(stderr, "TIMEDOUT WAITING\n");
727 libwebsocket_close_and_free_session(context,
728 wsi, LWS_CLOSE_STATUS_NOSTATUS);
729 }
730}
731
732struct libwebsocket *
733libwebsocket_create_new_server_wsi(struct libwebsocket_context *context)
734{
735 struct libwebsocket *new_wsi;
736 int n;
737
738 new_wsi = malloc(sizeof(struct libwebsocket));
739 if (new_wsi == NULL) {
740 fprintf(stderr, "Out of memory for new connection\n");
741 return NULL;
742 }
743
744 memset(new_wsi, 0, sizeof (struct libwebsocket));
745 new_wsi->count_active_extensions = 0;
746 new_wsi->pending_timeout = NO_PENDING_TIMEOUT;
747
748 /* intialize the instance struct */
749
750 new_wsi->state = WSI_STATE_HTTP;
751 new_wsi->name_buffer_pos = 0;
752 new_wsi->mode = LWS_CONNMODE_WS_SERVING;
753
754 for (n = 0; n < WSI_TOKEN_COUNT; n++) {
755 new_wsi->utf8_token[n].token = NULL;
756 new_wsi->utf8_token[n].token_len = 0;
757 }
758
759 /*
760 * these can only be set once the protocol is known
761 * we set an unestablished connection's protocol pointer
762 * to the start of the supported list, so it can look
763 * for matching ones during the handshake
764 */
765 new_wsi->protocol = context->protocols;
766 new_wsi->user_space = NULL;
767
768 /*
769 * Default protocol is 76 / 00
770 * After 76, there's a header specified to inform which
771 * draft the client wants, when that's seen we modify
772 * the individual connection's spec revision accordingly
773 */
774 new_wsi->ietf_spec_revision = 0;
775
776 return new_wsi;
777}
778
779char *
780libwebsockets_generate_client_handshake(struct libwebsocket_context *context,
781 struct libwebsocket *wsi, char *pkt)
782{
783 char hash[20];
784 char *p = pkt;
785 int n;
786 struct libwebsocket_extension *ext;
787 int ext_count = 0;
788 unsigned char buf[LWS_SEND_BUFFER_PRE_PADDING + 1 + MAX_BROADCAST_PAYLOAD +
789 LWS_SEND_BUFFER_POST_PADDING];
790 static const char magic_websocket_guid[] =
791 "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
792
793 /*
794 * create the random key
795 */
796
797 n = libwebsockets_get_random(context, hash, 16);
798 if (n != 16) {
799 fprintf(stderr, "Unable to read from random dev %s\n",
800 SYSTEM_RANDOM_FILEPATH);
801 free(wsi->c_path);
802 free(wsi->c_host);
803 if (wsi->c_origin)
804 free(wsi->c_origin);
805 if (wsi->c_protocol)
806 free(wsi->c_protocol);
807 libwebsocket_close_and_free_session(context, wsi,
808 LWS_CLOSE_STATUS_NOSTATUS);
809 return NULL;
810 }
811
812 lws_b64_encode_string(hash, 16, wsi->key_b64,
813 sizeof wsi->key_b64);
814
815 /*
816 * 00 example client handshake
817 *
818 * GET /socket.io/websocket HTTP/1.1
819 * Upgrade: WebSocket
820 * Connection: Upgrade
821 * Host: 127.0.0.1:9999
822 * Origin: http://127.0.0.1
823 * Sec-WebSocket-Key1: 1 0 2#0W 9 89 7 92 ^
824 * Sec-WebSocket-Key2: 7 7Y 4328 B2v[8(z1
825 * Cookie: socketio=websocket
826 *
827 * (Á®Ä0¶†≥
828 *
829 * 04 example client handshake
830 *
831 * GET /chat HTTP/1.1
832 * Host: server.example.com
833 * Upgrade: websocket
834 * Connection: Upgrade
835 * Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==
836 * Sec-WebSocket-Origin: http://example.com
837 * Sec-WebSocket-Protocol: chat, superchat
838 * Sec-WebSocket-Version: 4
839 */
840
841 p += sprintf(p, "GET %s HTTP/1.1\x0d\x0a", wsi->c_path);
842
843 if (wsi->ietf_spec_revision == 0) {
844 unsigned char spaces_1, spaces_2;
845 unsigned int max_1, max_2;
846 unsigned int num_1, num_2;
847 unsigned long product_1, product_2;
848 char key_1[40];
849 char key_2[40];
850 unsigned int seed;
851 unsigned int count;
852 char challenge[16];
853
854 libwebsockets_get_random(context, &spaces_1,
855 sizeof(char));
856 libwebsockets_get_random(context, &spaces_2,
857 sizeof(char));
858
859 spaces_1 = (spaces_1 % 12) + 1;
860 spaces_2 = (spaces_2 % 12) + 1;
861
862 max_1 = 4294967295 / spaces_1;
863 max_2 = 4294967295 / spaces_2;
864
865 libwebsockets_get_random(context, &num_1, sizeof(int));
866 libwebsockets_get_random(context, &num_2, sizeof(int));
867
868 num_1 = (num_1 % max_1);
869 num_2 = (num_2 % max_2);
870
871 challenge[0] = num_1 >> 24;
872 challenge[1] = num_1 >> 16;
873 challenge[2] = num_1 >> 8;
874 challenge[3] = num_1;
875 challenge[4] = num_2 >> 24;
876 challenge[5] = num_2 >> 16;
877 challenge[6] = num_2 >> 8;
878 challenge[7] = num_2;
879
880 product_1 = num_1 * spaces_1;
881 product_2 = num_2 * spaces_2;
882
883 sprintf(key_1, "%lu", product_1);
884 sprintf(key_2, "%lu", product_2);
885
886 libwebsockets_get_random(context, &seed, sizeof(int));
887 libwebsockets_get_random(context, &count, sizeof(int));
888
889 libwebsockets_00_spam(key_1, (count % 12) + 1, seed);
890
891 libwebsockets_get_random(context, &seed, sizeof(int));
892 libwebsockets_get_random(context, &count, sizeof(int));
893
894 libwebsockets_00_spam(key_2, (count % 12) + 1, seed);
895
896 libwebsockets_get_random(context, &seed, sizeof(int));
897
898 libwebsockets_00_spaceout(key_1, spaces_1, seed);
899 libwebsockets_00_spaceout(key_2, spaces_2, seed >> 16);
900
901 p += sprintf(p, "Upgrade: WebSocket\x0d\x0a"
902 "Connection: Upgrade\x0d\x0aHost: %s\x0d\x0a",
903 wsi->c_host);
904 if (wsi->c_origin)
905 p += sprintf(p, "Origin: %s\x0d\x0a",
906 wsi->c_origin);
907
908 if (wsi->c_protocol)
909 p += sprintf(p, "Sec-WebSocket-Protocol: %s"
910 "\x0d\x0a", wsi->c_protocol);
911
912 p += sprintf(p, "Sec-WebSocket-Key1: %s\x0d\x0a",
913 key_1);
914 p += sprintf(p, "Sec-WebSocket-Key2: %s\x0d\x0a",
915 key_2);
916
917 /* give userland a chance to append, eg, cookies */
918
919 context->protocols[0].callback(context, wsi,
920 LWS_CALLBACK_CLIENT_APPEND_HANDSHAKE_HEADER,
921 NULL, &p, (pkt + sizeof(pkt)) - p - 12);
922
923 p += sprintf(p, "\x0d\x0a");
924
925 if (libwebsockets_get_random(context, p, 8) != 8)
926 return NULL;
927 memcpy(&challenge[8], p, 8);
928 p += 8;
929
930 /* precompute what we want to see from the server */
931
932 MD5((unsigned char *)challenge, 16,
933 (unsigned char *)wsi->initial_handshake_hash_base64);
934
935 goto issue_hdr;
936 }
937
938 p += sprintf(p, "Host: %s\x0d\x0a", wsi->c_host);
939 p += sprintf(p, "Upgrade: websocket\x0d\x0a");
940 p += sprintf(p, "Connection: Upgrade\x0d\x0a"
941 "Sec-WebSocket-Key: ");
942 strcpy(p, wsi->key_b64);
943 p += strlen(wsi->key_b64);
944 p += sprintf(p, "\x0d\x0a");
945 if (wsi->c_origin)
946 p += sprintf(p, "Sec-WebSocket-Origin: %s\x0d\x0a",
947 wsi->c_origin);
948 if (wsi->c_protocol)
949 p += sprintf(p, "Sec-WebSocket-Protocol: %s\x0d\x0a",
950 wsi->c_protocol);
951
952 /* tell the server what extensions we could support */
953
954 p += sprintf(p, "Sec-WebSocket-Extensions: ");
955
956 ext =context->extensions;
957 while (ext && ext->callback) {
958
959 n = 0;
960 n = context->protocols[0].callback(context, wsi,
961 LWS_CALLBACK_CLIENT_CONFIRM_EXTENSION_SUPPORTED,
962 wsi->user_space, (char *)ext->name, 0);
963
964 /*
965 * zero return from callback means
966 * go ahead and allow the extension,
967 * it's what we get if the callback is
968 * unhandled
969 */
970
971 if (n) {
972 ext++;
973 continue;
974 }
975
976 /* apply it */
977
978 if (ext_count)
979 *p++ = ',';
980 p += sprintf(p, "%s", ext->name);
981 ext_count++;
982
983 ext++;
984 }
985
986 p += sprintf(p, "\x0d\x0a");
987
988 if (wsi->ietf_spec_revision)
989 p += sprintf(p, "Sec-WebSocket-Version: %d\x0d\x0a",
990 wsi->ietf_spec_revision);
991
992 /* give userland a chance to append, eg, cookies */
993
994 context->protocols[0].callback(context, wsi,
995 LWS_CALLBACK_CLIENT_APPEND_HANDSHAKE_HEADER,
996 NULL, &p, (pkt + sizeof(pkt)) - p - 12);
997
998 p += sprintf(p, "\x0d\x0a");
999
1000 /* prepare the expected server accept response */
1001
1002 strcpy((char *)buf, wsi->key_b64);
1003 strcpy((char *)&buf[strlen((char *)buf)], magic_websocket_guid);
1004
1005 SHA1(buf, strlen((char *)buf), (unsigned char *)hash);
1006
1007 lws_b64_encode_string(hash, 20,
1008 wsi->initial_handshake_hash_base64,
1009 sizeof wsi->initial_handshake_hash_base64);
1010
1011issue_hdr:
1012
1013 /* done with these now */
1014
1015 free(wsi->c_path);
1016 free(wsi->c_host);
1017 if (wsi->c_origin)
1018 free(wsi->c_origin);
1019
1020 return p;
1021}
1022
1023int
1024lws_client_interpret_server_handshake(struct libwebsocket_context *context,
1025 struct libwebsocket *wsi)
1026{
1027 unsigned char buf[LWS_SEND_BUFFER_PRE_PADDING + 1 + MAX_BROADCAST_PAYLOAD +
1028 LWS_SEND_BUFFER_POST_PADDING];
1029 char pkt[1024];
1030 char *p = &pkt[0];
1031 const char *pc;
1032 const char *c;
1033 int more = 1;
1034 int okay = 0;
1035 char ext_name[128];
1036 struct libwebsocket_extension *ext;
1037 void *v;
1038 int len;
1039 int n;
1040 static const char magic_websocket_04_masking_guid[] =
1041 "61AC5F19-FBBA-4540-B96F-6561F1AB40A8";
1042
1043 /*
1044 * 00 / 76 -->
1045 *
1046 * HTTP/1.1 101 WebSocket Protocol Handshake
1047 * Upgrade: WebSocket
1048 * Connection: Upgrade
1049 * Sec-WebSocket-Origin: http://127.0.0.1
1050 * Sec-WebSocket-Location: ws://127.0.0.1:9999/socket.io/websocket
1051 *
1052 * xxxxxxxxxxxxxxxx
1053 */
1054
1055 if (wsi->ietf_spec_revision == 0) {
1056 if (!wsi->utf8_token[WSI_TOKEN_HTTP].token_len ||
1057 !wsi->utf8_token[WSI_TOKEN_UPGRADE].token_len ||
1058 !wsi->utf8_token[WSI_TOKEN_CHALLENGE].token_len ||
1059 !wsi->utf8_token[WSI_TOKEN_CONNECTION].token_len ||
1060 (!wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len &&
1061 wsi->c_protocol != NULL)) {
1062 fprintf(stderr, "libwebsocket_client_handshake "
1063 "missing required header(s)\n");
1064 pkt[len] = '\0';
1065 fprintf(stderr, "%s", pkt);
1066 goto bail3;
1067 }
1068
1069 strtolower(wsi->utf8_token[WSI_TOKEN_HTTP].token);
1070 if (strcmp(wsi->utf8_token[WSI_TOKEN_HTTP].token,
1071 "101 websocket protocol handshake")) {
1072 fprintf(stderr, "libwebsocket_client_handshake "
1073 "server sent bad HTTP response '%s'\n",
1074 wsi->utf8_token[WSI_TOKEN_HTTP].token);
1075 goto bail3;
1076 }
1077
1078 if (wsi->utf8_token[WSI_TOKEN_CHALLENGE].token_len <
1079 16) {
1080 fprintf(stderr, "libwebsocket_client_handshake "
1081 "challenge reply too short %d\n",
1082 wsi->utf8_token[
1083 WSI_TOKEN_CHALLENGE].token_len);
1084 pkt[len] = '\0';
1085 fprintf(stderr, "%s", pkt);
1086 goto bail3;
1087
1088 }
1089
1090 goto select_protocol;
1091 }
1092
1093 /*
1094 * well, what the server sent looked reasonable for syntax.
1095 * Now let's confirm it sent all the necessary headers
1096 */
1097#if 0
1098 fprintf(stderr, "WSI_TOKEN_HTTP: %d\n", wsi->utf8_token[WSI_TOKEN_HTTP].token_len);
1099 fprintf(stderr, "WSI_TOKEN_UPGRADE: %d\n", wsi->utf8_token[WSI_TOKEN_UPGRADE].token_len);
1100 fprintf(stderr, "WSI_TOKEN_CONNECTION: %d\n", wsi->utf8_token[WSI_TOKEN_CONNECTION].token_len);
1101 fprintf(stderr, "WSI_TOKEN_ACCEPT: %d\n", wsi->utf8_token[WSI_TOKEN_ACCEPT].token_len);
1102 fprintf(stderr, "WSI_TOKEN_NONCE: %d\n", wsi->utf8_token[WSI_TOKEN_NONCE].token_len);
1103 fprintf(stderr, "WSI_TOKEN_PROTOCOL: %d\n", wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len);
1104#endif
1105 if (!wsi->utf8_token[WSI_TOKEN_HTTP].token_len ||
1106 !wsi->utf8_token[WSI_TOKEN_UPGRADE].token_len ||
1107 !wsi->utf8_token[WSI_TOKEN_CONNECTION].token_len ||
1108 !wsi->utf8_token[WSI_TOKEN_ACCEPT].token_len ||
1109 (!wsi->utf8_token[WSI_TOKEN_NONCE].token_len &&
1110 wsi->ietf_spec_revision == 4) ||
1111 (!wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len &&
1112 wsi->c_protocol != NULL)) {
1113 fprintf(stderr, "libwebsocket_client_handshake "
1114 "missing required header(s)\n");
1115 pkt[len] = '\0';
1116 fprintf(stderr, "%s", pkt);
1117 goto bail3;
1118 }
1119
1120 /*
1121 * Everything seems to be there, now take a closer look at what
1122 * is in each header
1123 */
1124
1125 strtolower(wsi->utf8_token[WSI_TOKEN_HTTP].token);
1126 if (strcmp(wsi->utf8_token[WSI_TOKEN_HTTP].token,
1127 "101 switching protocols")) {
1128 fprintf(stderr, "libwebsocket_client_handshake "
1129 "server sent bad HTTP response '%s'\n",
1130 wsi->utf8_token[WSI_TOKEN_HTTP].token);
1131 goto bail3;
1132 }
1133
1134 strtolower(wsi->utf8_token[WSI_TOKEN_UPGRADE].token);
1135 if (strcmp(wsi->utf8_token[WSI_TOKEN_UPGRADE].token,
1136 "websocket")) {
1137 fprintf(stderr, "libwebsocket_client_handshake server "
1138 "sent bad Upgrade header '%s'\n",
1139 wsi->utf8_token[WSI_TOKEN_UPGRADE].token);
1140 goto bail3;
1141 }
1142
1143 strtolower(wsi->utf8_token[WSI_TOKEN_CONNECTION].token);
1144 if (strcmp(wsi->utf8_token[WSI_TOKEN_CONNECTION].token,
1145 "upgrade")) {
1146 fprintf(stderr, "libwebsocket_client_handshake server "
1147 "sent bad Connection hdr '%s'\n",
1148 wsi->utf8_token[WSI_TOKEN_CONNECTION].token);
1149 goto bail3;
1150 }
1151
1152select_protocol:
1153 pc = wsi->c_protocol;
1154 if (pc == NULL)
1155 fprintf(stderr, "lws_client_interpret_server_handshake: NULL c_protocol\n");
1156 else
1157 fprintf(stderr, "lws_client_interpret_server_handshake: cPprotocol='%s'\n", pc);
1158
1159 /*
1160 * confirm the protocol the server wants to talk was in the list
1161 * of protocols we offered
1162 */
1163
1164 if (!wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len) {
1165
1166 fprintf(stderr, "lws_client_interpret_server_handshake WSI_TOKEN_PROTOCOL is null\n");
1167 /*
1168 * no protocol name to work from,
1169 * default to first protocol
1170 */
1171 wsi->protocol = &context->protocols[0];
1172
1173 free(wsi->c_protocol);
1174
1175 goto check_accept;
1176 }
1177
1178 while (*pc && !okay) {
1179 if ((!strncmp(pc,
1180 wsi->utf8_token[WSI_TOKEN_PROTOCOL].token,
1181 wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len)) &&
1182 (pc[wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len] == ',' ||
1183 pc[wsi->utf8_token[WSI_TOKEN_PROTOCOL].token_len] == '\0')) {
1184 okay = 1;
1185 continue;
1186 }
1187 while (*pc && *pc != ',')
1188 pc++;
1189 while (*pc && *pc != ' ')
1190 pc++;
1191 }
1192
1193 /* done with him now */
1194
1195 if (wsi->c_protocol)
1196 free(wsi->c_protocol);
1197
1198
1199 if (!okay) {
1200 fprintf(stderr, "libwebsocket_client_handshake server "
1201 "sent bad protocol '%s'\n",
1202 wsi->utf8_token[WSI_TOKEN_PROTOCOL].token);
1203 goto bail2;
1204 }
1205
1206 /*
1207 * identify the selected protocol struct and set it
1208 */
1209 n = 0;
1210 wsi->protocol = NULL;
1211 while (context->protocols[n].callback) {
1212 if (strcmp(wsi->utf8_token[WSI_TOKEN_PROTOCOL].token,
1213 context->protocols[n].name) == 0)
1214 wsi->protocol = &context->protocols[n];
1215 n++;
1216 }
1217
1218 if (wsi->protocol == NULL) {
1219 fprintf(stderr, "libwebsocket_client_handshake server "
1220 "requested protocol '%s', which we "
1221 "said we supported but we don't!\n",
1222 wsi->utf8_token[WSI_TOKEN_PROTOCOL].token);
1223 goto bail2;
1224 }
1225
1226
1227 /* instantiate the accepted extensions */
1228
1229 if (!wsi->utf8_token[WSI_TOKEN_EXTENSIONS].token_len) {
1230 fprintf(stderr, "no client extenstions allowed by server \n");
1231 goto check_accept;
1232 }
1233
1234 /*
1235 * break down the list of server accepted extensions
1236 * and go through matching them or identifying bogons
1237 */
1238
1239 c = wsi->utf8_token[WSI_TOKEN_EXTENSIONS].token;
1240 n = 0;
1241 while (more) {
1242
1243 if (*c && (*c != ',' && *c != ' ' && *c != '\t')) {
1244 ext_name[n] = *c++;
1245 if (n < sizeof(ext_name) - 1)
1246 n++;
1247 continue;
1248 }
1249 ext_name[n] = '\0';
1250 if (!*c)
1251 more = 0;
1252 else {
1253 c++;
1254 if (!n)
1255 continue;
1256 }
1257
1258 /* check we actually support it */
1259
1260 fprintf(stderr, "checking client ext %s\n", ext_name);
1261
1262 n = 0;
1263 ext = wsi->protocol->owning_server->extensions;
1264 while (ext && ext->callback) {
1265
1266 if (strcmp(ext_name, ext->name)) {
1267 ext++;
1268 continue;
1269 }
1270
1271 n = 1;
1272
1273 fprintf(stderr, "instantiating client ext %s\n", ext_name);
1274
1275 /* instantiate the extension on this conn */
1276
1277 wsi->active_extensions_user[
1278 wsi->count_active_extensions] =
1279 malloc(ext->per_session_data_size);
Andy Greenf6652412011-05-25 20:46:18 +01001280 memset(wsi->active_extensions_user[
1281 wsi->count_active_extensions], 0,
1282 ext->per_session_data_size);
Andy Greena41314f2011-05-23 10:00:03 +01001283 wsi->active_extensions[
1284 wsi->count_active_extensions] = ext;
1285
1286 /* allow him to construct his context */
1287
1288 ext->callback(wsi->protocol->owning_server,
1289 ext, wsi,
1290 LWS_EXT_CALLBACK_CLIENT_CONSTRUCT,
1291 wsi->active_extensions_user[
1292 wsi->count_active_extensions],
1293 NULL, 0);
1294
1295 wsi->count_active_extensions++;
1296
1297 ext++;
1298 }
1299
1300 if (n == 0) {
1301 fprintf(stderr, "Server said we should use"
1302 "an unknown extension '%s'!\n", ext_name);
1303 goto bail2;
1304 }
1305
1306 n = 0;
1307 }
1308
1309
1310check_accept:
1311
1312 if (wsi->ietf_spec_revision == 0) {
1313
1314 if (memcmp(wsi->initial_handshake_hash_base64,
1315 wsi->utf8_token[WSI_TOKEN_CHALLENGE].token, 16)) {
1316 fprintf(stderr, "libwebsocket_client_handshake "
1317 "failed 00 challenge compare\n");
1318 pkt[len] = '\0';
1319 fprintf(stderr, "%s", pkt);
1320 goto bail2;
1321 }
1322
1323 goto accept_ok;
1324 }
1325
1326 /*
1327 * Confirm his accept token is the one we precomputed
1328 */
1329
1330 if (strcmp(wsi->utf8_token[WSI_TOKEN_ACCEPT].token,
1331 wsi->initial_handshake_hash_base64)) {
1332 fprintf(stderr, "libwebsocket_client_handshake server "
1333 "sent bad ACCEPT '%s' vs computed '%s'\n",
1334 wsi->utf8_token[WSI_TOKEN_ACCEPT].token,
1335 wsi->initial_handshake_hash_base64);
1336 goto bail2;
1337 }
1338
1339 if (wsi->ietf_spec_revision == 4) {
1340 /*
1341 * Calculate the 04 masking key to use when
1342 * sending data to server
1343 */
1344
1345 strcpy((char *)buf, wsi->key_b64);
1346 p = (char *)buf + strlen(wsi->key_b64);
1347 strcpy(p, wsi->utf8_token[WSI_TOKEN_NONCE].token);
1348 p += wsi->utf8_token[WSI_TOKEN_NONCE].token_len;
1349 strcpy(p, magic_websocket_04_masking_guid);
1350 SHA1(buf, strlen((char *)buf), wsi->masking_key_04);
1351 }
1352 accept_ok:
1353
1354 /* allocate the per-connection user memory (if any) */
1355
1356 if (wsi->protocol->per_session_data_size) {
1357 wsi->user_space = malloc(
1358 wsi->protocol->per_session_data_size);
1359 if (wsi->user_space == NULL) {
1360 fprintf(stderr, "Out of memory for "
1361 "conn user space\n");
1362 goto bail2;
1363 }
1364 } else
1365 wsi->user_space = NULL;
1366
1367 /* clear his proxy connection timeout */
1368
1369 libwebsocket_set_timeout(wsi, NO_PENDING_TIMEOUT, 0);
1370
1371 /* mark him as being alive */
1372
1373 wsi->state = WSI_STATE_ESTABLISHED;
1374 wsi->mode = LWS_CONNMODE_WS_CLIENT;
1375
1376 fprintf(stderr, "handshake OK for protocol %s\n",
1377 wsi->protocol->name);
1378
1379 /* call him back to inform him he is up */
1380
1381 wsi->protocol->callback(context, wsi,
1382 LWS_CALLBACK_CLIENT_ESTABLISHED,
1383 wsi->user_space,
1384 NULL, 0);
1385
1386 /*
1387 * inform all extensions, not just active ones since they
1388 * already know
1389 */
1390
1391 ext = context->extensions;
1392
1393 while (ext && ext->callback) {
1394 v = NULL;
1395 for (n = 0; n < wsi->count_active_extensions; n++)
1396 if (wsi->active_extensions[n] == ext)
1397 v = wsi->active_extensions_user[n];
1398
1399 ext->callback(context, ext, wsi,
1400 LWS_EXT_CALLBACK_ANY_WSI_ESTABLISHED, v, NULL, 0);
1401 ext++;
1402 }
1403
1404 return 0;
1405
1406bail3:
1407 if (wsi->c_protocol)
1408 free(wsi->c_protocol);
1409
1410bail2:
1411 libwebsocket_close_and_free_session(context, wsi,
1412 LWS_CLOSE_STATUS_NOSTATUS);
1413 return 1;
1414}
1415
1416
1417
Andy Green9f990342011-02-12 11:57:45 +00001418/**
1419 * libwebsocket_service_fd() - Service polled socket with something waiting
Peter Hinz56885f32011-03-02 22:03:47 +00001420 * @context: Websocket context
Andy Green9f990342011-02-12 11:57:45 +00001421 * @pollfd: The pollfd entry describing the socket fd and which events
1422 * happened.
1423 *
1424 * This function closes any active connections and then frees the
1425 * context. After calling this, any further use of the context is
1426 * undefined.
1427 */
1428
1429int
Peter Hinz56885f32011-03-02 22:03:47 +00001430libwebsocket_service_fd(struct libwebsocket_context *context,
Andy Green0d338332011-02-12 11:57:43 +00001431 struct pollfd *pollfd)
Andy Greenb45993c2010-12-18 15:13:50 +00001432{
Andy Green3b84c002011-03-06 13:14:42 +00001433 unsigned char buf[LWS_SEND_BUFFER_PRE_PADDING + 1 + MAX_BROADCAST_PAYLOAD +
Andy Greenb45993c2010-12-18 15:13:50 +00001434 LWS_SEND_BUFFER_POST_PADDING];
Andy Greena71eafc2011-02-14 17:59:43 +00001435 struct libwebsocket *wsi;
Andy Green0d338332011-02-12 11:57:43 +00001436 struct libwebsocket *new_wsi;
Andy Greenb45993c2010-12-18 15:13:50 +00001437 int n;
Andy Green0d338332011-02-12 11:57:43 +00001438 int m;
Andy Greenb45993c2010-12-18 15:13:50 +00001439 size_t len;
Andy Green0d338332011-02-12 11:57:43 +00001440 int accept_fd;
1441 unsigned int clilen;
1442 struct sockaddr_in cli_addr;
Andy Greena71eafc2011-02-14 17:59:43 +00001443 struct timeval tv;
Andy Greenbe93fef2011-02-14 20:25:43 +00001444 char pkt[1024];
1445 char *p = &pkt[0];
Andy Green2366b1c2011-03-06 13:15:31 +00001446 int more = 1;
Andy Green98a717c2011-03-06 13:14:15 +00001447 struct lws_tokens eff_buf;
Andy Green6c939552011-03-08 08:56:57 +00001448 int opt = 1;
Andy Greenc6517fa2011-03-06 13:15:29 +00001449
Andy Greenbe93fef2011-02-14 20:25:43 +00001450#ifdef LWS_OPENSSL_SUPPORT
1451 char ssl_err_buf[512];
1452#endif
Andy Greena71eafc2011-02-14 17:59:43 +00001453 /*
1454 * you can call us with pollfd = NULL to just allow the once-per-second
1455 * global timeout checks; if less than a second since the last check
1456 * it returns immediately then.
1457 */
1458
1459 gettimeofday(&tv, NULL);
1460
Peter Hinz56885f32011-03-02 22:03:47 +00001461 if (context->last_timeout_check_s != tv.tv_sec) {
1462 context->last_timeout_check_s = tv.tv_sec;
Andy Greena71eafc2011-02-14 17:59:43 +00001463
1464 /* global timeout check once per second */
1465
Peter Hinz56885f32011-03-02 22:03:47 +00001466 for (n = 0; n < context->fds_count; n++) {
1467 wsi = wsi_from_fd(context, context->fds[n].fd);
Andy Greena71eafc2011-02-14 17:59:43 +00001468
Andy Greena41314f2011-05-23 10:00:03 +01001469 libwebsocket_service_timeout_check(context, wsi,
1470 tv.tv_sec);
Andy Greena71eafc2011-02-14 17:59:43 +00001471 }
1472 }
1473
1474 /* just here for timeout management? */
1475
1476 if (pollfd == NULL)
1477 return 0;
1478
1479 /* no, here to service a socket descriptor */
1480
Peter Hinz56885f32011-03-02 22:03:47 +00001481 wsi = wsi_from_fd(context, pollfd->fd);
Andy Greenb45993c2010-12-18 15:13:50 +00001482
Andy Green0d338332011-02-12 11:57:43 +00001483 if (wsi == NULL)
1484 return 1;
Andy Green8f037e42010-12-19 22:13:26 +00001485
Andy Green0d338332011-02-12 11:57:43 +00001486 switch (wsi->mode) {
1487 case LWS_CONNMODE_SERVER_LISTENER:
1488
1489 /* pollin means a client has connected to us then */
1490
1491 if (!pollfd->revents & POLLIN)
1492 break;
1493
1494 /* listen socket got an unencrypted connection... */
1495
1496 clilen = sizeof(cli_addr);
1497 accept_fd = accept(pollfd->fd, (struct sockaddr *)&cli_addr,
1498 &clilen);
1499 if (accept_fd < 0) {
1500 fprintf(stderr, "ERROR on accept");
1501 break;
1502 }
1503
Andy Green6c939552011-03-08 08:56:57 +00001504 /* Disable Nagle */
1505 opt = 1;
Pavel Borzenkov71ea5002011-04-15 13:16:32 +01001506 setsockopt(accept_fd, IPPROTO_TCP, TCP_NODELAY, &opt,
1507 sizeof(opt));
Andy Green6c939552011-03-08 08:56:57 +00001508
Peter Hinz56885f32011-03-02 22:03:47 +00001509 if (context->fds_count >= MAX_CLIENTS) {
Andy Green3221f922011-02-12 13:14:11 +00001510 fprintf(stderr, "too busy to accept new client\n");
Peter Hinz56885f32011-03-02 22:03:47 +00001511#ifdef WIN32
1512 closesocket(accept_fd);
1513#else
Andy Green0d338332011-02-12 11:57:43 +00001514 close(accept_fd);
Peter Hinz56885f32011-03-02 22:03:47 +00001515#endif
Andy Green0d338332011-02-12 11:57:43 +00001516 break;
1517 }
1518
Andy Green07034092011-02-13 08:37:12 +00001519 /*
1520 * look at who we connected to and give user code a chance
1521 * to reject based on client IP. There's no protocol selected
1522 * yet so we issue this to protocols[0]
1523 */
1524
Peter Hinz56885f32011-03-02 22:03:47 +00001525 if ((context->protocols[0].callback)(context, wsi,
Andy Green07034092011-02-13 08:37:12 +00001526 LWS_CALLBACK_FILTER_NETWORK_CONNECTION,
1527 (void*)(long)accept_fd, NULL, 0)) {
1528 fprintf(stderr, "Callback denied network connection\n");
Peter Hinz56885f32011-03-02 22:03:47 +00001529#ifdef WIN32
1530 closesocket(accept_fd);
1531#else
Andy Green07034092011-02-13 08:37:12 +00001532 close(accept_fd);
Peter Hinz56885f32011-03-02 22:03:47 +00001533#endif
Andy Green07034092011-02-13 08:37:12 +00001534 break;
1535 }
1536
Andy Green0d338332011-02-12 11:57:43 +00001537 /* accepting connection to main listener */
1538
Andy Greena41314f2011-05-23 10:00:03 +01001539 new_wsi = libwebsocket_create_new_server_wsi(context);
1540 if (new_wsi == NULL)
Andy Green0d338332011-02-12 11:57:43 +00001541 break;
Andy Green0d338332011-02-12 11:57:43 +00001542
Andy Green0d338332011-02-12 11:57:43 +00001543 new_wsi->sock = accept_fd;
Andy Greena41314f2011-05-23 10:00:03 +01001544
Andy Green0d338332011-02-12 11:57:43 +00001545
1546#ifdef LWS_OPENSSL_SUPPORT
1547 new_wsi->ssl = NULL;
Andy Green0d338332011-02-12 11:57:43 +00001548
Peter Hinz56885f32011-03-02 22:03:47 +00001549 if (context->use_ssl) {
Andy Green0d338332011-02-12 11:57:43 +00001550
Peter Hinz56885f32011-03-02 22:03:47 +00001551 new_wsi->ssl = SSL_new(context->ssl_ctx);
Andy Green0d338332011-02-12 11:57:43 +00001552 if (new_wsi->ssl == NULL) {
1553 fprintf(stderr, "SSL_new failed: %s\n",
1554 ERR_error_string(SSL_get_error(
1555 new_wsi->ssl, 0), NULL));
Andy Green1f9bf522011-02-14 21:14:37 +00001556 libwebsockets_decode_ssl_error();
Andy Green0d338332011-02-12 11:57:43 +00001557 free(new_wsi);
1558 break;
1559 }
1560
1561 SSL_set_fd(new_wsi->ssl, accept_fd);
1562
1563 n = SSL_accept(new_wsi->ssl);
1564 if (n != 1) {
1565 /*
1566 * browsers seem to probe with various
1567 * ssl params which fail then retry
1568 * and succeed
1569 */
1570 debug("SSL_accept failed skt %u: %s\n",
1571 pollfd->fd,
1572 ERR_error_string(SSL_get_error(
1573 new_wsi->ssl, n), NULL));
1574 SSL_free(
1575 new_wsi->ssl);
1576 free(new_wsi);
1577 break;
1578 }
Andy Greenc6bf2c22011-02-20 11:10:47 +00001579
Andy Green0d338332011-02-12 11:57:43 +00001580 debug("accepted new SSL conn "
1581 "port %u on fd=%d SSL ver %s\n",
1582 ntohs(cli_addr.sin_port), accept_fd,
1583 SSL_get_version(new_wsi->ssl));
1584
1585 } else
1586#endif
1587 debug("accepted new conn port %u on fd=%d\n",
1588 ntohs(cli_addr.sin_port), accept_fd);
1589
Peter Hinz56885f32011-03-02 22:03:47 +00001590 insert_wsi(context, new_wsi);
Andy Green0d338332011-02-12 11:57:43 +00001591
Andy Green0d338332011-02-12 11:57:43 +00001592 /*
1593 * make sure NO events are seen yet on this new socket
1594 * (otherwise we inherit old fds[client].revents from
1595 * previous socket there and die mysteriously! )
1596 */
Peter Hinz56885f32011-03-02 22:03:47 +00001597 context->fds[context->fds_count].revents = 0;
Andy Green0d338332011-02-12 11:57:43 +00001598
Peter Hinz56885f32011-03-02 22:03:47 +00001599 context->fds[context->fds_count].events = POLLIN;
1600 context->fds[context->fds_count++].fd = accept_fd;
Andy Green0d338332011-02-12 11:57:43 +00001601
Andy Green3221f922011-02-12 13:14:11 +00001602 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00001603 context->protocols[0].callback(context, new_wsi,
Andy Green3221f922011-02-12 13:14:11 +00001604 LWS_CALLBACK_ADD_POLL_FD,
1605 (void *)(long)accept_fd, NULL, POLLIN);
1606
Andy Green0d338332011-02-12 11:57:43 +00001607 break;
1608
1609 case LWS_CONNMODE_BROADCAST_PROXY_LISTENER:
1610
1611 /* as we are listening, POLLIN means accept() is needed */
1612
1613 if (!pollfd->revents & POLLIN)
1614 break;
1615
1616 /* listen socket got an unencrypted connection... */
1617
1618 clilen = sizeof(cli_addr);
1619 accept_fd = accept(pollfd->fd, (struct sockaddr *)&cli_addr,
1620 &clilen);
1621 if (accept_fd < 0) {
1622 fprintf(stderr, "ERROR on accept");
1623 break;
1624 }
1625
Peter Hinz56885f32011-03-02 22:03:47 +00001626 if (context->fds_count >= MAX_CLIENTS) {
Andy Green3221f922011-02-12 13:14:11 +00001627 fprintf(stderr, "too busy to accept new broadcast "
1628 "proxy client\n");
Peter Hinz56885f32011-03-02 22:03:47 +00001629#ifdef WIN32
1630 closesocket(accept_fd);
1631#else
Andy Green0d338332011-02-12 11:57:43 +00001632 close(accept_fd);
Peter Hinz56885f32011-03-02 22:03:47 +00001633#endif
Andy Green0d338332011-02-12 11:57:43 +00001634 break;
1635 }
1636
1637 /* create a dummy wsi for the connection and add it */
1638
1639 new_wsi = malloc(sizeof(struct libwebsocket));
1640 memset(new_wsi, 0, sizeof (struct libwebsocket));
1641 new_wsi->sock = accept_fd;
1642 new_wsi->mode = LWS_CONNMODE_BROADCAST_PROXY;
1643 new_wsi->state = WSI_STATE_ESTABLISHED;
Andy Greend6e09112011-03-05 16:12:15 +00001644 new_wsi->count_active_extensions = 0;
Andy Green0d338332011-02-12 11:57:43 +00001645 /* note which protocol we are proxying */
1646 new_wsi->protocol_index_for_broadcast_proxy =
1647 wsi->protocol_index_for_broadcast_proxy;
Peter Hinz56885f32011-03-02 22:03:47 +00001648 insert_wsi(context, new_wsi);
Andy Green0d338332011-02-12 11:57:43 +00001649
1650 /* add connected socket to internal poll array */
1651
Peter Hinz56885f32011-03-02 22:03:47 +00001652 context->fds[context->fds_count].revents = 0;
1653 context->fds[context->fds_count].events = POLLIN;
1654 context->fds[context->fds_count++].fd = accept_fd;
Andy Green0d338332011-02-12 11:57:43 +00001655
Andy Green3221f922011-02-12 13:14:11 +00001656 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00001657 context->protocols[0].callback(context, new_wsi,
Andy Green3221f922011-02-12 13:14:11 +00001658 LWS_CALLBACK_ADD_POLL_FD,
1659 (void *)(long)accept_fd, NULL, POLLIN);
1660
Andy Green0d338332011-02-12 11:57:43 +00001661 break;
1662
1663 case LWS_CONNMODE_BROADCAST_PROXY:
Andy Green8f037e42010-12-19 22:13:26 +00001664
Andy Greenb45993c2010-12-18 15:13:50 +00001665 /* handle session socket closed */
Andy Green8f037e42010-12-19 22:13:26 +00001666
Andy Green0d338332011-02-12 11:57:43 +00001667 if (pollfd->revents & (POLLERR | POLLHUP)) {
Andy Green8f037e42010-12-19 22:13:26 +00001668
Andy Green0d338332011-02-12 11:57:43 +00001669 debug("Session Socket %p (fd=%d) dead\n",
Timothy J Fontaineb86d64e2011-02-14 17:55:27 +00001670 (void *)wsi, pollfd->fd);
Andy Greenb45993c2010-12-18 15:13:50 +00001671
Peter Hinz56885f32011-03-02 22:03:47 +00001672 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001673 LWS_CLOSE_STATUS_NORMAL);
Andy Green4b6fbe12011-02-14 08:03:48 +00001674 return 1;
Andy Greenb45993c2010-12-18 15:13:50 +00001675 }
Andy Green8f037e42010-12-19 22:13:26 +00001676
Andy Green3b84c002011-03-06 13:14:42 +00001677 /*
1678 * either extension code with stuff to spill, or the user code,
1679 * requested a callback when it was OK to write
1680 */
Andy Green90c7cbc2011-01-27 06:26:52 +00001681
Andy Green3b84c002011-03-06 13:14:42 +00001682 if (pollfd->revents & POLLOUT)
1683 if (lws_handle_POLLOUT_event(context, wsi, pollfd) < 0) {
1684 libwebsocket_close_and_free_session(context, wsi,
1685 LWS_CLOSE_STATUS_NORMAL);
1686 return 1;
1687 }
Andy Green90c7cbc2011-01-27 06:26:52 +00001688
Andy Greenb45993c2010-12-18 15:13:50 +00001689 /* any incoming data ready? */
1690
Andy Green0d338332011-02-12 11:57:43 +00001691 if (!(pollfd->revents & POLLIN))
1692 break;
Andy Greenb45993c2010-12-18 15:13:50 +00001693
Andy Green0d338332011-02-12 11:57:43 +00001694 /* get the issued broadcast payload from the socket */
Andy Greenb45993c2010-12-18 15:13:50 +00001695
Andy Green0d338332011-02-12 11:57:43 +00001696 len = read(pollfd->fd, buf + LWS_SEND_BUFFER_PRE_PADDING,
1697 MAX_BROADCAST_PAYLOAD);
1698 if (len < 0) {
1699 fprintf(stderr, "Error reading broadcast payload\n");
Andy Green4b6fbe12011-02-14 08:03:48 +00001700 break;
Andy Green0d338332011-02-12 11:57:43 +00001701 }
Andy Greenb45993c2010-12-18 15:13:50 +00001702
Andy Green0d338332011-02-12 11:57:43 +00001703 /* broadcast it to all guys with this protocol index */
Andy Green8f037e42010-12-19 22:13:26 +00001704
Andy Green0d338332011-02-12 11:57:43 +00001705 for (n = 0; n < FD_HASHTABLE_MODULUS; n++) {
Andy Green8f037e42010-12-19 22:13:26 +00001706
Peter Hinz56885f32011-03-02 22:03:47 +00001707 for (m = 0; m < context->fd_hashtable[n].length; m++) {
Andy Greenb45993c2010-12-18 15:13:50 +00001708
Peter Hinz56885f32011-03-02 22:03:47 +00001709 new_wsi = context->fd_hashtable[n].wsi[m];
Andy Greenb45993c2010-12-18 15:13:50 +00001710
Andy Green0d338332011-02-12 11:57:43 +00001711 /* only to clients we are serving to */
Andy Greenb45993c2010-12-18 15:13:50 +00001712
Andy Green0d338332011-02-12 11:57:43 +00001713 if (new_wsi->mode != LWS_CONNMODE_WS_SERVING)
Andy Greenb45993c2010-12-18 15:13:50 +00001714 continue;
1715
1716 /*
1717 * never broadcast to non-established
1718 * connection
1719 */
1720
Andy Green0d338332011-02-12 11:57:43 +00001721 if (new_wsi->state != WSI_STATE_ESTABLISHED)
Andy Green4739e5c2011-01-22 12:51:57 +00001722 continue;
1723
Andy Greenb45993c2010-12-18 15:13:50 +00001724 /*
1725 * only broadcast to connections using
1726 * the requested protocol
1727 */
1728
Andy Green0d338332011-02-12 11:57:43 +00001729 if (new_wsi->protocol->protocol_index !=
1730 wsi->protocol_index_for_broadcast_proxy)
Andy Greenb45993c2010-12-18 15:13:50 +00001731 continue;
1732
Andy Green8f037e42010-12-19 22:13:26 +00001733 /* broadcast it to this connection */
1734
Peter Hinz56885f32011-03-02 22:03:47 +00001735 new_wsi->protocol->callback(context, new_wsi,
Andy Green8f037e42010-12-19 22:13:26 +00001736 LWS_CALLBACK_BROADCAST,
Andy Green0d338332011-02-12 11:57:43 +00001737 new_wsi->user_space,
Andy Green0ca6a172010-12-19 20:50:01 +00001738 buf + LWS_SEND_BUFFER_PRE_PADDING, len);
Andy Greenb45993c2010-12-18 15:13:50 +00001739 }
Andy Green0d338332011-02-12 11:57:43 +00001740 }
1741 break;
Andy Greenb45993c2010-12-18 15:13:50 +00001742
Andy Greenbe93fef2011-02-14 20:25:43 +00001743 case LWS_CONNMODE_WS_CLIENT_WAITING_PROXY_REPLY:
1744
1745 /* handle proxy hung up on us */
1746
1747 if (pollfd->revents & (POLLERR | POLLHUP)) {
1748
1749 fprintf(stderr, "Proxy connection %p (fd=%d) dead\n",
1750 (void *)wsi, pollfd->fd);
1751
Peter Hinz56885f32011-03-02 22:03:47 +00001752 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001753 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001754 return 1;
1755 }
1756
Andy Green72c34322011-04-16 10:46:21 +01001757 n = recv(wsi->sock, pkt, sizeof pkt, 0);
Andy Greenbe93fef2011-02-14 20:25:43 +00001758 if (n < 0) {
Peter Hinz56885f32011-03-02 22:03:47 +00001759 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001760 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001761 fprintf(stderr, "ERROR reading from proxy socket\n");
1762 return 1;
1763 }
1764
1765 pkt[13] = '\0';
1766 if (strcmp(pkt, "HTTP/1.0 200 ") != 0) {
Peter Hinz56885f32011-03-02 22:03:47 +00001767 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001768 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001769 fprintf(stderr, "ERROR from proxy: %s\n", pkt);
1770 return 1;
1771 }
1772
1773 /* clear his proxy connection timeout */
1774
1775 libwebsocket_set_timeout(wsi, NO_PENDING_TIMEOUT, 0);
1776
1777 /* fallthru */
1778
1779 case LWS_CONNMODE_WS_CLIENT_ISSUE_HANDSHAKE:
1780
1781 #ifdef LWS_OPENSSL_SUPPORT
1782 if (wsi->use_ssl) {
1783
Peter Hinz56885f32011-03-02 22:03:47 +00001784 wsi->ssl = SSL_new(context->ssl_client_ctx);
1785 wsi->client_bio = BIO_new_socket(wsi->sock,
1786 BIO_NOCLOSE);
Andy Greenbe93fef2011-02-14 20:25:43 +00001787 SSL_set_bio(wsi->ssl, wsi->client_bio, wsi->client_bio);
1788
Andy Green6901cb32011-02-21 08:06:47 +00001789 SSL_set_ex_data(wsi->ssl,
Andy Green2e24da02011-03-05 16:12:04 +00001790 openssl_websocket_private_data_index,
Peter Hinz56885f32011-03-02 22:03:47 +00001791 context);
Andy Green6901cb32011-02-21 08:06:47 +00001792
Andy Greenbe93fef2011-02-14 20:25:43 +00001793 if (SSL_connect(wsi->ssl) <= 0) {
1794 fprintf(stderr, "SSL connect error %s\n",
Andy Green687b0182011-02-26 11:04:01 +00001795 ERR_error_string(ERR_get_error(),
1796 ssl_err_buf));
Peter Hinz56885f32011-03-02 22:03:47 +00001797 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001798 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001799 return 1;
1800 }
1801
1802 n = SSL_get_verify_result(wsi->ssl);
Andy Green2e24da02011-03-05 16:12:04 +00001803 if ((n != X509_V_OK) && (
Andy Green687b0182011-02-26 11:04:01 +00001804 n != X509_V_ERR_DEPTH_ZERO_SELF_SIGNED_CERT ||
1805 wsi->use_ssl != 2)) {
Andy Greenbe93fef2011-02-14 20:25:43 +00001806
Andy Green687b0182011-02-26 11:04:01 +00001807 fprintf(stderr, "server's cert didn't "
1808 "look good %d\n", n);
Peter Hinz56885f32011-03-02 22:03:47 +00001809 libwebsocket_close_and_free_session(context,
1810 wsi, LWS_CLOSE_STATUS_NOSTATUS);
Andy Green687b0182011-02-26 11:04:01 +00001811 return 1;
Andy Greenbe93fef2011-02-14 20:25:43 +00001812 }
1813 } else {
1814 wsi->ssl = NULL;
1815 #endif
1816
1817
1818 #ifdef LWS_OPENSSL_SUPPORT
1819 }
1820 #endif
1821
Andy Greena41314f2011-05-23 10:00:03 +01001822 p = libwebsockets_generate_client_handshake(context, wsi, p);
1823 if (p ==NULL)
Andy Greenbe93fef2011-02-14 20:25:43 +00001824 return 1;
Andy Greeneeaacb32011-03-01 20:44:24 +00001825
Andy Greenbe93fef2011-02-14 20:25:43 +00001826 /* send our request to the server */
1827
1828 #ifdef LWS_OPENSSL_SUPPORT
1829 if (wsi->use_ssl)
1830 n = SSL_write(wsi->ssl, pkt, p - pkt);
1831 else
1832 #endif
1833 n = send(wsi->sock, pkt, p - pkt, 0);
1834
1835 if (n < 0) {
1836 fprintf(stderr, "ERROR writing to client socket\n");
Peter Hinz56885f32011-03-02 22:03:47 +00001837 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001838 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001839 return 1;
1840 }
1841
1842 wsi->parser_state = WSI_TOKEN_NAME_PART;
1843 wsi->mode = LWS_CONNMODE_WS_CLIENT_WAITING_SERVER_REPLY;
1844 libwebsocket_set_timeout(wsi,
1845 PENDING_TIMEOUT_AWAITING_SERVER_RESPONSE, 5);
1846
1847 break;
1848
1849 case LWS_CONNMODE_WS_CLIENT_WAITING_SERVER_REPLY:
1850
1851 /* handle server hung up on us */
1852
1853 if (pollfd->revents & (POLLERR | POLLHUP)) {
1854
1855 fprintf(stderr, "Server connection %p (fd=%d) dead\n",
1856 (void *)wsi, pollfd->fd);
1857
1858 goto bail3;
1859 }
1860
1861
1862 /* interpret the server response */
1863
1864 /*
1865 * HTTP/1.1 101 Switching Protocols
1866 * Upgrade: websocket
1867 * Connection: Upgrade
1868 * Sec-WebSocket-Accept: me89jWimTRKTWwrS3aRrL53YZSo=
1869 * Sec-WebSocket-Nonce: AQIDBAUGBwgJCgsMDQ4PEC==
1870 * Sec-WebSocket-Protocol: chat
1871 */
1872
1873 #ifdef LWS_OPENSSL_SUPPORT
1874 if (wsi->use_ssl)
1875 len = SSL_read(wsi->ssl, pkt, sizeof pkt);
1876 else
1877 #endif
Andy Green72c34322011-04-16 10:46:21 +01001878 len = recv(wsi->sock, pkt, sizeof pkt, 0);
Andy Greenbe93fef2011-02-14 20:25:43 +00001879
1880 if (len < 0) {
1881 fprintf(stderr,
1882 "libwebsocket_client_handshake read error\n");
1883 goto bail3;
1884 }
1885
1886 p = pkt;
1887 for (n = 0; n < len; n++)
1888 libwebsocket_parse(wsi, *p++);
1889
Andy Green27a0b912011-04-16 10:54:28 +01001890 /*
1891 * may be coming in multiple packets, there is a 5-second
1892 * libwebsocket timeout still active here too, so if parsing did
1893 * not complete just wait for next packet coming in this state
1894 */
1895
1896 if (wsi->parser_state != WSI_PARSING_COMPLETE)
1897 break;
Andy Greenbe93fef2011-02-14 20:25:43 +00001898
Andy Greena41314f2011-05-23 10:00:03 +01001899 return lws_client_interpret_server_handshake(context, wsi);
Andy Greenbe93fef2011-02-14 20:25:43 +00001900
1901bail3:
1902 if (wsi->c_protocol)
1903 free(wsi->c_protocol);
Peter Hinz56885f32011-03-02 22:03:47 +00001904 libwebsocket_close_and_free_session(context, wsi,
Andy Greena41314f2011-05-23 10:00:03 +01001905 LWS_CLOSE_STATUS_NOSTATUS);
Andy Greenbe93fef2011-02-14 20:25:43 +00001906 return 1;
Andy Greena41314f2011-05-23 10:00:03 +01001907
1908 case LWS_CONNMODE_WS_CLIENT_WAITING_EXTENSION_CONNECT:
1909 fprintf(stderr, "LWS_CONNMODE_WS_CLIENT_WAITING_EXTENSION_CONNECT\n");
1910 break;
1911
1912 case LWS_CONNMODE_WS_CLIENT_PENDING_CANDIDATE_CHILD:
1913 fprintf(stderr, "LWS_CONNMODE_WS_CLIENT_PENDING_CANDIDATE_CHILD\n");
1914 break;
1915
Andy Greenbe93fef2011-02-14 20:25:43 +00001916
Andy Green0d338332011-02-12 11:57:43 +00001917 case LWS_CONNMODE_WS_SERVING:
1918 case LWS_CONNMODE_WS_CLIENT:
1919
1920 /* handle session socket closed */
1921
1922 if (pollfd->revents & (POLLERR | POLLHUP)) {
1923
Andy Green62c54d22011-02-14 09:14:25 +00001924 fprintf(stderr, "Session Socket %p (fd=%d) dead\n",
Andy Green0d338332011-02-12 11:57:43 +00001925 (void *)wsi, pollfd->fd);
1926
Peter Hinz56885f32011-03-02 22:03:47 +00001927 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001928 LWS_CLOSE_STATUS_NOSTATUS);
Andy Green4b6fbe12011-02-14 08:03:48 +00001929 return 1;
Andy Greenb45993c2010-12-18 15:13:50 +00001930 }
1931
Andy Green0d338332011-02-12 11:57:43 +00001932 /* the guy requested a callback when it was OK to write */
1933
Andy Greenda527df2011-03-07 07:08:12 +00001934 if ((pollfd->revents & POLLOUT) &&
1935 wsi->state == WSI_STATE_ESTABLISHED)
1936 if (lws_handle_POLLOUT_event(context, wsi,
1937 pollfd) < 0) {
1938 libwebsocket_close_and_free_session(
1939 context, wsi, LWS_CLOSE_STATUS_NORMAL);
Andy Green3b84c002011-03-06 13:14:42 +00001940 return 1;
1941 }
Andy Green0d338332011-02-12 11:57:43 +00001942
Andy Green0d338332011-02-12 11:57:43 +00001943
1944 /* any incoming data ready? */
1945
1946 if (!(pollfd->revents & POLLIN))
1947 break;
1948
Andy Greenb45993c2010-12-18 15:13:50 +00001949#ifdef LWS_OPENSSL_SUPPORT
Andy Green0d338332011-02-12 11:57:43 +00001950 if (wsi->ssl)
Andy Green98a717c2011-03-06 13:14:15 +00001951 eff_buf.token_len = SSL_read(wsi->ssl, buf, sizeof buf);
Andy Greenb45993c2010-12-18 15:13:50 +00001952 else
1953#endif
Andy Green98a717c2011-03-06 13:14:15 +00001954 eff_buf.token_len =
Andy Green72c34322011-04-16 10:46:21 +01001955 recv(pollfd->fd, buf, sizeof buf, 0);
Andy Greenb45993c2010-12-18 15:13:50 +00001956
Andy Green98a717c2011-03-06 13:14:15 +00001957 if (eff_buf.token_len < 0) {
1958 fprintf(stderr, "Socket read returned %d\n",
1959 eff_buf.token_len);
Andy Green4b6fbe12011-02-14 08:03:48 +00001960 break;
Andy Greenb45993c2010-12-18 15:13:50 +00001961 }
Andy Green98a717c2011-03-06 13:14:15 +00001962 if (!eff_buf.token_len) {
Peter Hinz56885f32011-03-02 22:03:47 +00001963 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00001964 LWS_CLOSE_STATUS_NOSTATUS);
Andy Green4b6fbe12011-02-14 08:03:48 +00001965 return 1;
Andy Greenb45993c2010-12-18 15:13:50 +00001966 }
1967
Andy Green98a717c2011-03-06 13:14:15 +00001968 /*
1969 * give any active extensions a chance to munge the buffer
1970 * before parse. We pass in a pointer to an lws_tokens struct
1971 * prepared with the default buffer and content length that's in
1972 * there. Rather than rewrite the default buffer, extensions
1973 * that expect to grow the buffer can adapt .token to
1974 * point to their own per-connection buffer in the extension
1975 * user allocation. By default with no extensions or no
1976 * extension callback handling, just the normal input buffer is
1977 * used then so it is efficient.
1978 */
Andy Greenb45993c2010-12-18 15:13:50 +00001979
Andy Green98a717c2011-03-06 13:14:15 +00001980 eff_buf.token = (char *)buf;
Andy Greenb45993c2010-12-18 15:13:50 +00001981
Andy Green98a717c2011-03-06 13:14:15 +00001982 more = 1;
1983 while (more) {
Andy Green0d338332011-02-12 11:57:43 +00001984
Andy Green98a717c2011-03-06 13:14:15 +00001985 more = 0;
1986
1987 for (n = 0; n < wsi->count_active_extensions; n++) {
Andy Green46c2ea02011-03-22 09:04:01 +00001988 m = wsi->active_extensions[n]->callback(context,
1989 wsi->active_extensions[n], wsi,
Andy Green98a717c2011-03-06 13:14:15 +00001990 LWS_EXT_CALLBACK_PACKET_RX_PREPARSE,
Andy Green46c2ea02011-03-22 09:04:01 +00001991 wsi->active_extensions_user[n],
1992 &eff_buf, 0);
Andy Green98a717c2011-03-06 13:14:15 +00001993 if (m < 0) {
1994 fprintf(stderr, "Extension reports fatal error\n");
1995 libwebsocket_close_and_free_session(context, wsi,
1996 LWS_CLOSE_STATUS_NOSTATUS);
1997 return 1;
1998 }
1999 if (m)
2000 more = 1;
2001 }
2002
2003 /* service incoming data */
2004
2005 if (eff_buf.token_len) {
2006 n = libwebsocket_read(context, wsi,
2007 (unsigned char *)eff_buf.token, eff_buf.token_len);
2008 if (n < 0)
2009 /* we closed wsi */
2010 return 1;
2011 }
2012
2013 eff_buf.token = NULL;
2014 eff_buf.token_len = 0;
2015 }
2016 break;
Andy Greenb45993c2010-12-18 15:13:50 +00002017 }
2018
2019 return 0;
2020}
2021
Andy Green0d338332011-02-12 11:57:43 +00002022
Andy Green6964bb52011-01-23 16:50:33 +00002023/**
2024 * libwebsocket_context_destroy() - Destroy the websocket context
Peter Hinz56885f32011-03-02 22:03:47 +00002025 * @context: Websocket context
Andy Green6964bb52011-01-23 16:50:33 +00002026 *
2027 * This function closes any active connections and then frees the
2028 * context. After calling this, any further use of the context is
2029 * undefined.
2030 */
2031void
Peter Hinz56885f32011-03-02 22:03:47 +00002032libwebsocket_context_destroy(struct libwebsocket_context *context)
Andy Green6964bb52011-01-23 16:50:33 +00002033{
Andy Green0d338332011-02-12 11:57:43 +00002034 int n;
2035 int m;
2036 struct libwebsocket *wsi;
Andy Greena41314f2011-05-23 10:00:03 +01002037 struct libwebsocket_extension *ext;
Andy Green6964bb52011-01-23 16:50:33 +00002038
Andy Green4b6fbe12011-02-14 08:03:48 +00002039 for (n = 0; n < FD_HASHTABLE_MODULUS; n++)
Peter Hinz56885f32011-03-02 22:03:47 +00002040 for (m = 0; m < context->fd_hashtable[n].length; m++) {
2041 wsi = context->fd_hashtable[n].wsi[m];
2042 libwebsocket_close_and_free_session(context, wsi,
Andy Green687b0182011-02-26 11:04:01 +00002043 LWS_CLOSE_STATUS_GOINGAWAY);
Andy Greenf3d3b402011-02-09 07:16:34 +00002044 }
Andy Green6964bb52011-01-23 16:50:33 +00002045
Andy Greena41314f2011-05-23 10:00:03 +01002046 /*
2047 * give all extensions a chance to clean up any per-context
2048 * allocations they might have made
2049 */
2050
2051 ext = context->extensions;
2052 m = LWS_EXT_CALLBACK_CLIENT_CONTEXT_DESTRUCT;
2053 if (context->listen_port)
2054 m = LWS_EXT_CALLBACK_SERVER_CONTEXT_DESTRUCT;
2055 while (ext->callback) {
2056 ext->callback(context, ext, NULL, m, NULL, NULL, 0);
2057 ext++;
2058 }
2059
Peter Hinz56885f32011-03-02 22:03:47 +00002060#ifdef WIN32
2061#else
2062 close(context->fd_random);
Andy Green6964bb52011-01-23 16:50:33 +00002063#endif
2064
Peter Hinz56885f32011-03-02 22:03:47 +00002065#ifdef LWS_OPENSSL_SUPPORT
2066 if (context->ssl_ctx)
2067 SSL_CTX_free(context->ssl_ctx);
2068 if (context->ssl_client_ctx)
2069 SSL_CTX_free(context->ssl_client_ctx);
2070#endif
2071
2072 free(context);
2073
2074#ifdef WIN32
2075 WSACleanup();
2076#endif
Andy Green6964bb52011-01-23 16:50:33 +00002077}
2078
2079/**
2080 * libwebsocket_service() - Service any pending websocket activity
Peter Hinz56885f32011-03-02 22:03:47 +00002081 * @context: Websocket context
Andy Green6964bb52011-01-23 16:50:33 +00002082 * @timeout_ms: Timeout for poll; 0 means return immediately if nothing needed
2083 * service otherwise block and service immediately, returning
2084 * after the timeout if nothing needed service.
2085 *
2086 * This function deals with any pending websocket traffic, for three
2087 * kinds of event. It handles these events on both server and client
2088 * types of connection the same.
2089 *
2090 * 1) Accept new connections to our context's server
2091 *
2092 * 2) Perform pending broadcast writes initiated from other forked
2093 * processes (effectively serializing asynchronous broadcasts)
2094 *
2095 * 3) Call the receive callback for incoming frame data received by
2096 * server or client connections.
2097 *
2098 * You need to call this service function periodically to all the above
2099 * functions to happen; if your application is single-threaded you can
2100 * just call it in your main event loop.
2101 *
2102 * Alternatively you can fork a new process that asynchronously handles
2103 * calling this service in a loop. In that case you are happy if this
2104 * call blocks your thread until it needs to take care of something and
2105 * would call it with a large nonzero timeout. Your loop then takes no
2106 * CPU while there is nothing happening.
2107 *
2108 * If you are calling it in a single-threaded app, you don't want it to
2109 * wait around blocking other things in your loop from happening, so you
2110 * would call it with a timeout_ms of 0, so it returns immediately if
2111 * nothing is pending, or as soon as it services whatever was pending.
2112 */
2113
Andy Greenb45993c2010-12-18 15:13:50 +00002114
Andy Greene92cd172011-01-19 13:11:55 +00002115int
Peter Hinz56885f32011-03-02 22:03:47 +00002116libwebsocket_service(struct libwebsocket_context *context, int timeout_ms)
Andy Greene92cd172011-01-19 13:11:55 +00002117{
2118 int n;
Andy Greene92cd172011-01-19 13:11:55 +00002119
2120 /* stay dead once we are dead */
2121
Peter Hinz56885f32011-03-02 22:03:47 +00002122 if (context == NULL)
Andy Greene92cd172011-01-19 13:11:55 +00002123 return 1;
2124
Andy Green0d338332011-02-12 11:57:43 +00002125 /* wait for something to need service */
Andy Green4739e5c2011-01-22 12:51:57 +00002126
Peter Hinz56885f32011-03-02 22:03:47 +00002127 n = poll(context->fds, context->fds_count, timeout_ms);
Andy Green3221f922011-02-12 13:14:11 +00002128 if (n == 0) /* poll timeout */
2129 return 0;
Andy Greene92cd172011-01-19 13:11:55 +00002130
Andy Green62c54d22011-02-14 09:14:25 +00002131 if (n < 0) {
Andy Green5e1fa172011-02-10 09:07:05 +00002132 /*
Andy Greene92cd172011-01-19 13:11:55 +00002133 fprintf(stderr, "Listen Socket dead\n");
Andy Green5e1fa172011-02-10 09:07:05 +00002134 */
Andy Green0d338332011-02-12 11:57:43 +00002135 return 1;
Andy Greene92cd172011-01-19 13:11:55 +00002136 }
Andy Greene92cd172011-01-19 13:11:55 +00002137
2138 /* handle accept on listening socket? */
2139
Peter Hinz56885f32011-03-02 22:03:47 +00002140 for (n = 0; n < context->fds_count; n++)
2141 if (context->fds[n].revents)
2142 libwebsocket_service_fd(context, &context->fds[n]);
Andy Greene92cd172011-01-19 13:11:55 +00002143
2144 return 0;
Andy Greene92cd172011-01-19 13:11:55 +00002145}
2146
Andy Greena41314f2011-05-23 10:00:03 +01002147int
2148lws_any_extension_handled(struct libwebsocket_context *context,
2149 struct libwebsocket *wsi,
2150 enum libwebsocket_extension_callback_reasons r,
2151 void *v, size_t len)
2152{
2153 int n;
2154 int handled = 0;
2155
2156 /* maybe an extension will take care of it for us */
2157
2158 for (n = 0; n < wsi->count_active_extensions && !handled; n++) {
2159 if (!wsi->active_extensions[n]->callback)
2160 continue;
2161
2162 handled |= wsi->active_extensions[n]->callback(context,
2163 wsi->active_extensions[n], wsi,
2164 r, wsi->active_extensions_user[n], v, len);
2165 }
2166
2167 return handled;
2168}
2169
2170
2171void *
2172lws_get_extension_user_matching_ext(struct libwebsocket *wsi,
2173 struct libwebsocket_extension * ext)
2174{
2175 int n = 0;
2176
Andy Green68b45042011-05-25 21:41:57 +01002177 if (wsi == NULL)
2178 return NULL;
2179
Andy Greena41314f2011-05-23 10:00:03 +01002180 while (n < wsi->count_active_extensions) {
2181 if (wsi->active_extensions[n] != ext) {
2182 n++;
2183 continue;
2184 }
2185 return wsi->active_extensions_user[n];
2186 }
2187
2188 return NULL;
2189}
2190
Andy Green90c7cbc2011-01-27 06:26:52 +00002191/**
2192 * libwebsocket_callback_on_writable() - Request a callback when this socket
2193 * becomes able to be written to without
2194 * blocking
Andy Green32375b72011-02-19 08:32:53 +00002195 *
Peter Hinz56885f32011-03-02 22:03:47 +00002196 * @context: libwebsockets context
Andy Green90c7cbc2011-01-27 06:26:52 +00002197 * @wsi: Websocket connection instance to get callback for
2198 */
2199
2200int
Peter Hinz56885f32011-03-02 22:03:47 +00002201libwebsocket_callback_on_writable(struct libwebsocket_context *context,
Andy Green62c54d22011-02-14 09:14:25 +00002202 struct libwebsocket *wsi)
Andy Green90c7cbc2011-01-27 06:26:52 +00002203{
Andy Green90c7cbc2011-01-27 06:26:52 +00002204 int n;
Andy Greena41314f2011-05-23 10:00:03 +01002205 int handled = 0;
2206
2207 /* maybe an extension will take care of it for us */
2208
2209 for (n = 0; n < wsi->count_active_extensions; n++) {
2210 if (!wsi->active_extensions[n]->callback)
2211 continue;
2212
2213 handled |= wsi->active_extensions[n]->callback(context,
2214 wsi->active_extensions[n], wsi,
2215 LWS_EXT_CALLBACK_REQUEST_ON_WRITEABLE,
2216 wsi->active_extensions_user[n], NULL, 0);
2217 }
2218
2219 if (handled)
2220 return 1;
Andy Green90c7cbc2011-01-27 06:26:52 +00002221
Peter Hinz56885f32011-03-02 22:03:47 +00002222 for (n = 0; n < context->fds_count; n++)
2223 if (context->fds[n].fd == wsi->sock) {
2224 context->fds[n].events |= POLLOUT;
Andy Greena41314f2011-05-23 10:00:03 +01002225 n = context->fds_count + 1;
Andy Green90c7cbc2011-01-27 06:26:52 +00002226 }
2227
Andy Greena41314f2011-05-23 10:00:03 +01002228 if (n == context->fds_count)
2229 fprintf(stderr, "libwebsocket_callback_on_writable: failed to find socket %d\n", wsi->sock);
2230
Andy Green3221f922011-02-12 13:14:11 +00002231 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00002232 context->protocols[0].callback(context, wsi,
Andy Green3221f922011-02-12 13:14:11 +00002233 LWS_CALLBACK_SET_MODE_POLL_FD,
2234 (void *)(long)wsi->sock, NULL, POLLOUT);
2235
Andy Green90c7cbc2011-01-27 06:26:52 +00002236 return 1;
2237}
2238
2239/**
2240 * libwebsocket_callback_on_writable_all_protocol() - Request a callback for
2241 * all connections using the given protocol when it
2242 * becomes possible to write to each socket without
2243 * blocking in turn.
2244 *
2245 * @protocol: Protocol whose connections will get callbacks
2246 */
2247
2248int
2249libwebsocket_callback_on_writable_all_protocol(
2250 const struct libwebsocket_protocols *protocol)
2251{
Peter Hinz56885f32011-03-02 22:03:47 +00002252 struct libwebsocket_context *context = protocol->owning_server;
Andy Green90c7cbc2011-01-27 06:26:52 +00002253 int n;
Andy Green0d338332011-02-12 11:57:43 +00002254 int m;
2255 struct libwebsocket *wsi;
Andy Green90c7cbc2011-01-27 06:26:52 +00002256
Andy Green0d338332011-02-12 11:57:43 +00002257 for (n = 0; n < FD_HASHTABLE_MODULUS; n++) {
2258
Peter Hinz56885f32011-03-02 22:03:47 +00002259 for (m = 0; m < context->fd_hashtable[n].length; m++) {
Andy Green0d338332011-02-12 11:57:43 +00002260
Peter Hinz56885f32011-03-02 22:03:47 +00002261 wsi = context->fd_hashtable[n].wsi[m];
Andy Green0d338332011-02-12 11:57:43 +00002262
2263 if (wsi->protocol == protocol)
Peter Hinz56885f32011-03-02 22:03:47 +00002264 libwebsocket_callback_on_writable(context, wsi);
Andy Green0d338332011-02-12 11:57:43 +00002265 }
2266 }
Andy Green90c7cbc2011-01-27 06:26:52 +00002267
2268 return 0;
2269}
2270
Andy Greenbe93fef2011-02-14 20:25:43 +00002271/**
2272 * libwebsocket_set_timeout() - marks the wsi as subject to a timeout
2273 *
2274 * You will not need this unless you are doing something special
2275 *
2276 * @wsi: Websocket connection instance
2277 * @reason: timeout reason
2278 * @secs: how many seconds
2279 */
2280
2281void
2282libwebsocket_set_timeout(struct libwebsocket *wsi,
2283 enum pending_timeout reason, int secs)
2284{
2285 struct timeval tv;
2286
2287 gettimeofday(&tv, NULL);
2288
2289 wsi->pending_timeout_limit = tv.tv_sec + secs;
2290 wsi->pending_timeout = reason;
2291}
2292
Andy Greena6cbece2011-01-27 20:06:03 +00002293
2294/**
2295 * libwebsocket_get_socket_fd() - returns the socket file descriptor
2296 *
2297 * You will not need this unless you are doing something special
2298 *
2299 * @wsi: Websocket connection instance
2300 */
2301
2302int
2303libwebsocket_get_socket_fd(struct libwebsocket *wsi)
2304{
2305 return wsi->sock;
2306}
2307
Andy Green90c7cbc2011-01-27 06:26:52 +00002308/**
2309 * libwebsocket_rx_flow_control() - Enable and disable socket servicing for
2310 * receieved packets.
2311 *
2312 * If the output side of a server process becomes choked, this allows flow
2313 * control for the input side.
2314 *
2315 * @wsi: Websocket connection instance to get callback for
2316 * @enable: 0 = disable read servicing for this connection, 1 = enable
2317 */
2318
2319int
2320libwebsocket_rx_flow_control(struct libwebsocket *wsi, int enable)
2321{
Peter Hinz56885f32011-03-02 22:03:47 +00002322 struct libwebsocket_context *context = wsi->protocol->owning_server;
Andy Green90c7cbc2011-01-27 06:26:52 +00002323 int n;
2324
Peter Hinz56885f32011-03-02 22:03:47 +00002325 for (n = 0; n < context->fds_count; n++)
2326 if (context->fds[n].fd == wsi->sock) {
Andy Green90c7cbc2011-01-27 06:26:52 +00002327 if (enable)
Peter Hinz56885f32011-03-02 22:03:47 +00002328 context->fds[n].events |= POLLIN;
Andy Green90c7cbc2011-01-27 06:26:52 +00002329 else
Peter Hinz56885f32011-03-02 22:03:47 +00002330 context->fds[n].events &= ~POLLIN;
Andy Green90c7cbc2011-01-27 06:26:52 +00002331
2332 return 0;
2333 }
2334
Andy Green3221f922011-02-12 13:14:11 +00002335 if (enable)
2336 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00002337 context->protocols[0].callback(context, wsi,
Andy Green3221f922011-02-12 13:14:11 +00002338 LWS_CALLBACK_SET_MODE_POLL_FD,
2339 (void *)(long)wsi->sock, NULL, POLLIN);
2340 else
2341 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00002342 context->protocols[0].callback(context, wsi,
Andy Green3221f922011-02-12 13:14:11 +00002343 LWS_CALLBACK_CLEAR_MODE_POLL_FD,
2344 (void *)(long)wsi->sock, NULL, POLLIN);
2345
Andy Greena41314f2011-05-23 10:00:03 +01002346#if 0
2347 fprintf(stderr, "libwebsocket_rx_flow_control "
Andy Green90c7cbc2011-01-27 06:26:52 +00002348 "unable to find socket\n");
Andy Greena41314f2011-05-23 10:00:03 +01002349#endif
Andy Green90c7cbc2011-01-27 06:26:52 +00002350 return 1;
2351}
2352
Andy Green2ac5a6f2011-01-28 10:00:18 +00002353/**
2354 * libwebsocket_canonical_hostname() - returns this host's hostname
2355 *
2356 * This is typically used by client code to fill in the host parameter
2357 * when making a client connection. You can only call it after the context
2358 * has been created.
2359 *
Peter Hinz56885f32011-03-02 22:03:47 +00002360 * @context: Websocket context
Andy Green2ac5a6f2011-01-28 10:00:18 +00002361 */
2362
2363
2364extern const char *
Peter Hinz56885f32011-03-02 22:03:47 +00002365libwebsocket_canonical_hostname(struct libwebsocket_context *context)
Andy Green2ac5a6f2011-01-28 10:00:18 +00002366{
Peter Hinz56885f32011-03-02 22:03:47 +00002367 return (const char *)context->canonical_hostname;
Andy Green2ac5a6f2011-01-28 10:00:18 +00002368}
2369
2370
Andy Green90c7cbc2011-01-27 06:26:52 +00002371static void sigpipe_handler(int x)
2372{
2373}
2374
Andy Green6901cb32011-02-21 08:06:47 +00002375#ifdef LWS_OPENSSL_SUPPORT
2376static int
2377OpenSSL_verify_callback(int preverify_ok, X509_STORE_CTX *x509_ctx)
2378{
2379
2380 SSL *ssl;
2381 int n;
Andy Green2e24da02011-03-05 16:12:04 +00002382 struct libwebsocket_context *context;
Andy Green6901cb32011-02-21 08:06:47 +00002383
2384 ssl = X509_STORE_CTX_get_ex_data(x509_ctx,
2385 SSL_get_ex_data_X509_STORE_CTX_idx());
2386
2387 /*
Andy Green2e24da02011-03-05 16:12:04 +00002388 * !!! nasty openssl requires the index to come as a library-scope
2389 * static
Andy Green6901cb32011-02-21 08:06:47 +00002390 */
Andy Green2e24da02011-03-05 16:12:04 +00002391 context = SSL_get_ex_data(ssl, openssl_websocket_private_data_index);
Andy Green6901cb32011-02-21 08:06:47 +00002392
Peter Hinz56885f32011-03-02 22:03:47 +00002393 n = context->protocols[0].callback(NULL, NULL,
Andy Green6901cb32011-02-21 08:06:47 +00002394 LWS_CALLBACK_OPENSSL_PERFORM_CLIENT_CERT_VERIFICATION,
2395 x509_ctx, ssl, preverify_ok);
2396
2397 /* convert return code from 0 = OK to 1 = OK */
2398
2399 if (!n)
2400 n = 1;
2401 else
2402 n = 0;
2403
2404 return n;
2405}
2406#endif
2407
Andy Greenb45993c2010-12-18 15:13:50 +00002408
Andy Greenab990e42010-10-31 12:42:52 +00002409/**
Andy Green4739e5c2011-01-22 12:51:57 +00002410 * libwebsocket_create_context() - Create the websocket handler
2411 * @port: Port to listen on... you can use 0 to suppress listening on
Andy Green6964bb52011-01-23 16:50:33 +00002412 * any port, that's what you want if you are not running a
2413 * websocket server at all but just using it as a client
Peter Hinz56885f32011-03-02 22:03:47 +00002414 * @interf: NULL to bind the listen socket to all interfaces, or the
Andy Green32375b72011-02-19 08:32:53 +00002415 * interface name, eg, "eth2"
Andy Green4f3943a2010-11-12 10:44:16 +00002416 * @protocols: Array of structures listing supported protocols and a protocol-
Andy Green8f037e42010-12-19 22:13:26 +00002417 * specific callback for each one. The list is ended with an
2418 * entry that has a NULL callback pointer.
Andy Green6964bb52011-01-23 16:50:33 +00002419 * It's not const because we write the owning_server member
Andy Greenc5114822011-03-06 10:29:35 +00002420 * @extensions: NULL or array of libwebsocket_extension structs listing the
2421 * extensions this context supports
Andy Green3faa9c72010-11-08 17:03:03 +00002422 * @ssl_cert_filepath: If libwebsockets was compiled to use ssl, and you want
Andy Green8f037e42010-12-19 22:13:26 +00002423 * to listen using SSL, set to the filepath to fetch the
2424 * server cert from, otherwise NULL for unencrypted
Andy Green3faa9c72010-11-08 17:03:03 +00002425 * @ssl_private_key_filepath: filepath to private key if wanting SSL mode,
Andy Green8f037e42010-12-19 22:13:26 +00002426 * else ignored
Andy Green3faa9c72010-11-08 17:03:03 +00002427 * @gid: group id to change to after setting listen socket, or -1.
2428 * @uid: user id to change to after setting listen socket, or -1.
Andy Greenbfb051f2011-02-09 08:49:14 +00002429 * @options: 0, or LWS_SERVER_OPTION_DEFEAT_CLIENT_MASK
Andy Green05464c62010-11-12 10:44:18 +00002430 *
Andy Green8f037e42010-12-19 22:13:26 +00002431 * This function creates the listening socket and takes care
2432 * of all initialization in one step.
2433 *
Andy Greene92cd172011-01-19 13:11:55 +00002434 * After initialization, it returns a struct libwebsocket_context * that
2435 * represents this server. After calling, user code needs to take care
2436 * of calling libwebsocket_service() with the context pointer to get the
2437 * server's sockets serviced. This can be done in the same process context
2438 * or a forked process, or another thread,
Andy Green05464c62010-11-12 10:44:18 +00002439 *
Andy Green8f037e42010-12-19 22:13:26 +00002440 * The protocol callback functions are called for a handful of events
2441 * including http requests coming in, websocket connections becoming
2442 * established, and data arriving; it's also called periodically to allow
2443 * async transmission.
2444 *
2445 * HTTP requests are sent always to the FIRST protocol in @protocol, since
2446 * at that time websocket protocol has not been negotiated. Other
2447 * protocols after the first one never see any HTTP callack activity.
2448 *
2449 * The server created is a simple http server by default; part of the
2450 * websocket standard is upgrading this http connection to a websocket one.
2451 *
2452 * This allows the same server to provide files like scripts and favicon /
2453 * images or whatever over http and dynamic data over websockets all in
2454 * one place; they're all handled in the user callback.
Andy Greenab990e42010-10-31 12:42:52 +00002455 */
Andy Green4ea60062010-10-30 12:15:07 +01002456
Andy Greene92cd172011-01-19 13:11:55 +00002457struct libwebsocket_context *
Peter Hinz56885f32011-03-02 22:03:47 +00002458libwebsocket_create_context(int port, const char *interf,
Andy Greenb45993c2010-12-18 15:13:50 +00002459 struct libwebsocket_protocols *protocols,
Andy Greend6e09112011-03-05 16:12:15 +00002460 struct libwebsocket_extension *extensions,
Andy Green8f037e42010-12-19 22:13:26 +00002461 const char *ssl_cert_filepath,
2462 const char *ssl_private_key_filepath,
Andy Green8014b292011-01-30 20:57:25 +00002463 int gid, int uid, unsigned int options)
Andy Greenff95d7a2010-10-28 22:36:01 +01002464{
2465 int n;
Andy Greena41314f2011-05-23 10:00:03 +01002466 int m;
Andy Green4739e5c2011-01-22 12:51:57 +00002467 int sockfd = 0;
Andy Green251f6fa2010-11-03 11:13:06 +00002468 int fd;
Andy Greenff95d7a2010-10-28 22:36:01 +01002469 struct sockaddr_in serv_addr, cli_addr;
Andy Green251f6fa2010-11-03 11:13:06 +00002470 int opt = 1;
Peter Hinz56885f32011-03-02 22:03:47 +00002471 struct libwebsocket_context *context = NULL;
Andy Greenb45993c2010-12-18 15:13:50 +00002472 unsigned int slen;
Andy Green9659f372011-01-27 22:01:43 +00002473 char *p;
Andy Green2ac5a6f2011-01-28 10:00:18 +00002474 char hostname[1024];
Andy Green42f69142011-01-30 08:10:02 +00002475 struct hostent *he;
Andy Green0d338332011-02-12 11:57:43 +00002476 struct libwebsocket *wsi;
Andy Greenff95d7a2010-10-28 22:36:01 +01002477
Andy Green3faa9c72010-11-08 17:03:03 +00002478#ifdef LWS_OPENSSL_SUPPORT
Andy Greenf2f54d52010-11-15 22:08:00 +00002479 SSL_METHOD *method;
Andy Green3faa9c72010-11-08 17:03:03 +00002480 char ssl_err_buf[512];
Andy Green3faa9c72010-11-08 17:03:03 +00002481#endif
2482
Peter Hinz56885f32011-03-02 22:03:47 +00002483#ifdef _WIN32
2484 {
2485 WORD wVersionRequested;
2486 WSADATA wsaData;
2487 int err;
2488
2489 /* Use the MAKEWORD(lowbyte, highbyte) macro from Windef.h */
2490 wVersionRequested = MAKEWORD(2, 2);
2491
2492 err = WSAStartup(wVersionRequested, &wsaData);
2493 if (err != 0) {
2494 /* Tell the user that we could not find a usable */
2495 /* Winsock DLL. */
2496 fprintf(stderr, "WSAStartup failed with error: %d\n",
2497 err);
2498 return NULL;
2499 }
2500 }
2501#endif
2502
2503
2504 context = malloc(sizeof(struct libwebsocket_context));
2505 if (!context) {
Andy Green90c7cbc2011-01-27 06:26:52 +00002506 fprintf(stderr, "No memory for websocket context\n");
2507 return NULL;
2508 }
Peter Hinz56885f32011-03-02 22:03:47 +00002509 context->protocols = protocols;
2510 context->listen_port = port;
2511 context->http_proxy_port = 0;
2512 context->http_proxy_address[0] = '\0';
2513 context->options = options;
2514 context->fds_count = 0;
Andy Greend6e09112011-03-05 16:12:15 +00002515 context->extensions = extensions;
Andy Green9659f372011-01-27 22:01:43 +00002516
Peter Hinz56885f32011-03-02 22:03:47 +00002517#ifdef WIN32
2518 context->fd_random = 0;
2519#else
2520 context->fd_random = open(SYSTEM_RANDOM_FILEPATH, O_RDONLY);
2521 if (context->fd_random < 0) {
Andy Green44eee682011-02-10 09:32:24 +00002522 fprintf(stderr, "Unable to open random device %s %d\n",
Peter Hinz56885f32011-03-02 22:03:47 +00002523 SYSTEM_RANDOM_FILEPATH, context->fd_random);
Andy Green44eee682011-02-10 09:32:24 +00002524 return NULL;
2525 }
Peter Hinz56885f32011-03-02 22:03:47 +00002526#endif
Andy Green44eee682011-02-10 09:32:24 +00002527
Peter Hinz56885f32011-03-02 22:03:47 +00002528#ifdef LWS_OPENSSL_SUPPORT
2529 context->use_ssl = 0;
2530 context->ssl_ctx = NULL;
2531 context->ssl_client_ctx = NULL;
Andy Green2e24da02011-03-05 16:12:04 +00002532 openssl_websocket_private_data_index = 0;
Peter Hinz56885f32011-03-02 22:03:47 +00002533#endif
Andy Green2ac5a6f2011-01-28 10:00:18 +00002534 /* find canonical hostname */
2535
2536 hostname[(sizeof hostname) - 1] = '\0';
2537 gethostname(hostname, (sizeof hostname) - 1);
2538 he = gethostbyname(hostname);
Darin Willitsc19456f2011-02-14 17:52:39 +00002539 if (he) {
Peter Hinz56885f32011-03-02 22:03:47 +00002540 strncpy(context->canonical_hostname, he->h_name,
2541 sizeof context->canonical_hostname - 1);
2542 context->canonical_hostname[
2543 sizeof context->canonical_hostname - 1] = '\0';
Darin Willitsc19456f2011-02-14 17:52:39 +00002544 } else
Peter Hinz56885f32011-03-02 22:03:47 +00002545 strncpy(context->canonical_hostname, hostname,
2546 sizeof context->canonical_hostname - 1);
Andy Green2ac5a6f2011-01-28 10:00:18 +00002547
Andy Green9659f372011-01-27 22:01:43 +00002548 /* split the proxy ads:port if given */
2549
2550 p = getenv("http_proxy");
2551 if (p) {
Peter Hinz56885f32011-03-02 22:03:47 +00002552 strncpy(context->http_proxy_address, p,
2553 sizeof context->http_proxy_address - 1);
2554 context->http_proxy_address[
2555 sizeof context->http_proxy_address - 1] = '\0';
Andy Green9659f372011-01-27 22:01:43 +00002556
Peter Hinz56885f32011-03-02 22:03:47 +00002557 p = strchr(context->http_proxy_address, ':');
Andy Green9659f372011-01-27 22:01:43 +00002558 if (p == NULL) {
2559 fprintf(stderr, "http_proxy needs to be ads:port\n");
2560 return NULL;
2561 }
2562 *p = '\0';
Peter Hinz56885f32011-03-02 22:03:47 +00002563 context->http_proxy_port = atoi(p + 1);
Andy Green9659f372011-01-27 22:01:43 +00002564
2565 fprintf(stderr, "Using proxy %s:%u\n",
Peter Hinz56885f32011-03-02 22:03:47 +00002566 context->http_proxy_address,
2567 context->http_proxy_port);
Andy Green9659f372011-01-27 22:01:43 +00002568 }
Andy Green90c7cbc2011-01-27 06:26:52 +00002569
2570 if (port) {
2571
Andy Green3faa9c72010-11-08 17:03:03 +00002572#ifdef LWS_OPENSSL_SUPPORT
Peter Hinz56885f32011-03-02 22:03:47 +00002573 context->use_ssl = ssl_cert_filepath != NULL &&
Andy Green90c7cbc2011-01-27 06:26:52 +00002574 ssl_private_key_filepath != NULL;
Peter Hinz56885f32011-03-02 22:03:47 +00002575 if (context->use_ssl)
Andy Green90c7cbc2011-01-27 06:26:52 +00002576 fprintf(stderr, " Compiled with SSL support, "
2577 "using it\n");
2578 else
2579 fprintf(stderr, " Compiled with SSL support, "
2580 "not using it\n");
Andy Green3faa9c72010-11-08 17:03:03 +00002581
Andy Green90c7cbc2011-01-27 06:26:52 +00002582#else
2583 if (ssl_cert_filepath != NULL &&
2584 ssl_private_key_filepath != NULL) {
2585 fprintf(stderr, " Not compiled for OpenSSl support!\n");
Andy Greene92cd172011-01-19 13:11:55 +00002586 return NULL;
Andy Green3faa9c72010-11-08 17:03:03 +00002587 }
Andy Green90c7cbc2011-01-27 06:26:52 +00002588 fprintf(stderr, " Compiled without SSL support, "
2589 "serving unencrypted\n");
2590#endif
2591 }
2592
2593 /* ignore SIGPIPE */
Peter Hinz56885f32011-03-02 22:03:47 +00002594#ifdef WIN32
2595#else
Andy Green90c7cbc2011-01-27 06:26:52 +00002596 signal(SIGPIPE, sigpipe_handler);
Peter Hinz56885f32011-03-02 22:03:47 +00002597#endif
Andy Green90c7cbc2011-01-27 06:26:52 +00002598
2599
2600#ifdef LWS_OPENSSL_SUPPORT
2601
2602 /* basic openssl init */
2603
2604 SSL_library_init();
2605
2606 OpenSSL_add_all_algorithms();
2607 SSL_load_error_strings();
2608
Andy Green2e24da02011-03-05 16:12:04 +00002609 openssl_websocket_private_data_index =
Andy Green6901cb32011-02-21 08:06:47 +00002610 SSL_get_ex_new_index(0, "libwebsockets", NULL, NULL, NULL);
2611
Andy Green90c7cbc2011-01-27 06:26:52 +00002612 /*
2613 * Firefox insists on SSLv23 not SSLv3
2614 * Konq disables SSLv2 by default now, SSLv23 works
2615 */
2616
2617 method = (SSL_METHOD *)SSLv23_server_method();
2618 if (!method) {
2619 fprintf(stderr, "problem creating ssl method: %s\n",
2620 ERR_error_string(ERR_get_error(), ssl_err_buf));
2621 return NULL;
2622 }
Peter Hinz56885f32011-03-02 22:03:47 +00002623 context->ssl_ctx = SSL_CTX_new(method); /* create context */
2624 if (!context->ssl_ctx) {
Andy Green90c7cbc2011-01-27 06:26:52 +00002625 fprintf(stderr, "problem creating ssl context: %s\n",
2626 ERR_error_string(ERR_get_error(), ssl_err_buf));
2627 return NULL;
2628 }
2629
2630 /* client context */
Peter Hinz56885f32011-03-02 22:03:47 +00002631 if (port == CONTEXT_PORT_NO_LISTEN)
2632 {
2633 method = (SSL_METHOD *)SSLv23_client_method();
2634 if (!method) {
2635 fprintf(stderr, "problem creating ssl method: %s\n",
2636 ERR_error_string(ERR_get_error(), ssl_err_buf));
2637 return NULL;
2638 }
2639 /* create context */
2640 context->ssl_client_ctx = SSL_CTX_new(method);
2641 if (!context->ssl_client_ctx) {
2642 fprintf(stderr, "problem creating ssl context: %s\n",
2643 ERR_error_string(ERR_get_error(), ssl_err_buf));
2644 return NULL;
2645 }
Andy Green90c7cbc2011-01-27 06:26:52 +00002646
Peter Hinz56885f32011-03-02 22:03:47 +00002647 /* openssl init for cert verification (for client sockets) */
Andy Green90c7cbc2011-01-27 06:26:52 +00002648
Peter Hinz56885f32011-03-02 22:03:47 +00002649 if (!SSL_CTX_load_verify_locations(
2650 context->ssl_client_ctx, NULL,
2651 LWS_OPENSSL_CLIENT_CERTS))
2652 fprintf(stderr,
2653 "Unable to load SSL Client certs from %s "
2654 "(set by --with-client-cert-dir= in configure) -- "
2655 " client ssl isn't going to work",
Andy Green90c7cbc2011-01-27 06:26:52 +00002656 LWS_OPENSSL_CLIENT_CERTS);
Peter Hinz56885f32011-03-02 22:03:47 +00002657
2658 /*
2659 * callback allowing user code to load extra verification certs
2660 * helping the client to verify server identity
2661 */
2662
2663 context->protocols[0].callback(context, NULL,
2664 LWS_CALLBACK_OPENSSL_LOAD_EXTRA_CLIENT_VERIFY_CERTS,
2665 context->ssl_client_ctx, NULL, 0);
Andy Green90c7cbc2011-01-27 06:26:52 +00002666 }
Andy Greenc6bf2c22011-02-20 11:10:47 +00002667 /* as a server, are we requiring clients to identify themselves? */
2668
2669 if (options & LWS_SERVER_OPTION_REQUIRE_VALID_OPENSSL_CLIENT_CERT) {
2670
2671 /* absolutely require the client cert */
2672
Peter Hinz56885f32011-03-02 22:03:47 +00002673 SSL_CTX_set_verify(context->ssl_ctx,
Andy Green6901cb32011-02-21 08:06:47 +00002674 SSL_VERIFY_PEER | SSL_VERIFY_FAIL_IF_NO_PEER_CERT,
2675 OpenSSL_verify_callback);
Andy Greenc6bf2c22011-02-20 11:10:47 +00002676
2677 /*
2678 * give user code a chance to load certs into the server
2679 * allowing it to verify incoming client certs
2680 */
2681
Peter Hinz56885f32011-03-02 22:03:47 +00002682 context->protocols[0].callback(context, NULL,
Andy Greenc6bf2c22011-02-20 11:10:47 +00002683 LWS_CALLBACK_OPENSSL_LOAD_EXTRA_SERVER_VERIFY_CERTS,
Peter Hinz56885f32011-03-02 22:03:47 +00002684 context->ssl_ctx, NULL, 0);
Andy Greenc6bf2c22011-02-20 11:10:47 +00002685 }
2686
Peter Hinz56885f32011-03-02 22:03:47 +00002687 if (context->use_ssl) {
Andy Green90c7cbc2011-01-27 06:26:52 +00002688
2689 /* openssl init for server sockets */
2690
Andy Green3faa9c72010-11-08 17:03:03 +00002691 /* set the local certificate from CertFile */
Peter Hinz56885f32011-03-02 22:03:47 +00002692 n = SSL_CTX_use_certificate_file(context->ssl_ctx,
Andy Green3faa9c72010-11-08 17:03:03 +00002693 ssl_cert_filepath, SSL_FILETYPE_PEM);
2694 if (n != 1) {
2695 fprintf(stderr, "problem getting cert '%s': %s\n",
2696 ssl_cert_filepath,
2697 ERR_error_string(ERR_get_error(), ssl_err_buf));
Andy Greene92cd172011-01-19 13:11:55 +00002698 return NULL;
Andy Green3faa9c72010-11-08 17:03:03 +00002699 }
2700 /* set the private key from KeyFile */
Peter Hinz56885f32011-03-02 22:03:47 +00002701 if (SSL_CTX_use_PrivateKey_file(context->ssl_ctx,
2702 ssl_private_key_filepath, SSL_FILETYPE_PEM) != 1) {
Andy Green018d8eb2010-11-08 21:04:23 +00002703 fprintf(stderr, "ssl problem getting key '%s': %s\n",
2704 ssl_private_key_filepath,
2705 ERR_error_string(ERR_get_error(), ssl_err_buf));
Andy Greene92cd172011-01-19 13:11:55 +00002706 return NULL;
Andy Green3faa9c72010-11-08 17:03:03 +00002707 }
2708 /* verify private key */
Peter Hinz56885f32011-03-02 22:03:47 +00002709 if (!SSL_CTX_check_private_key(context->ssl_ctx)) {
Andy Green018d8eb2010-11-08 21:04:23 +00002710 fprintf(stderr, "Private SSL key doesn't match cert\n");
Andy Greene92cd172011-01-19 13:11:55 +00002711 return NULL;
Andy Green3faa9c72010-11-08 17:03:03 +00002712 }
2713
2714 /* SSL is happy and has a cert it's content with */
2715 }
2716#endif
Andy Greenb45993c2010-12-18 15:13:50 +00002717
Andy Greendf736162011-01-18 15:39:02 +00002718 /* selftest */
2719
2720 if (lws_b64_selftest())
Andy Greene92cd172011-01-19 13:11:55 +00002721 return NULL;
Andy Greendf736162011-01-18 15:39:02 +00002722
Andy Green0d338332011-02-12 11:57:43 +00002723 /* fd hashtable init */
2724
2725 for (n = 0; n < FD_HASHTABLE_MODULUS; n++)
Peter Hinz56885f32011-03-02 22:03:47 +00002726 context->fd_hashtable[n].length = 0;
Andy Green0d338332011-02-12 11:57:43 +00002727
Andy Greenb45993c2010-12-18 15:13:50 +00002728 /* set up our external listening socket we serve on */
Andy Green8f037e42010-12-19 22:13:26 +00002729
Andy Green4739e5c2011-01-22 12:51:57 +00002730 if (port) {
Andy Green8f037e42010-12-19 22:13:26 +00002731
Andy Green4739e5c2011-01-22 12:51:57 +00002732 sockfd = socket(AF_INET, SOCK_STREAM, 0);
2733 if (sockfd < 0) {
2734 fprintf(stderr, "ERROR opening socket");
2735 return NULL;
2736 }
Andy Green775c0dd2010-10-29 14:15:22 +01002737
Andy Green4739e5c2011-01-22 12:51:57 +00002738 /* allow us to restart even if old sockets in TIME_WAIT */
2739 setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
Andy Greene77ddd82010-11-13 10:03:47 +00002740
Andy Green6c939552011-03-08 08:56:57 +00002741
2742 /* Disable Nagle */
2743 opt = 1;
Pavel Borzenkov71ea5002011-04-15 13:16:32 +01002744 setsockopt(sockfd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof(opt));
Andy Green6c939552011-03-08 08:56:57 +00002745
Andy Green4739e5c2011-01-22 12:51:57 +00002746 bzero((char *) &serv_addr, sizeof(serv_addr));
2747 serv_addr.sin_family = AF_INET;
Peter Hinz56885f32011-03-02 22:03:47 +00002748 if (interf == NULL)
Andy Green32375b72011-02-19 08:32:53 +00002749 serv_addr.sin_addr.s_addr = INADDR_ANY;
2750 else
Peter Hinz56885f32011-03-02 22:03:47 +00002751 interface_to_sa(interf, &serv_addr,
Andy Green32375b72011-02-19 08:32:53 +00002752 sizeof(serv_addr));
Andy Green4739e5c2011-01-22 12:51:57 +00002753 serv_addr.sin_port = htons(port);
2754
2755 n = bind(sockfd, (struct sockaddr *) &serv_addr,
2756 sizeof(serv_addr));
2757 if (n < 0) {
2758 fprintf(stderr, "ERROR on binding to port %d (%d %d)\n",
Andy Green8f037e42010-12-19 22:13:26 +00002759 port, n, errno);
Andy Green4739e5c2011-01-22 12:51:57 +00002760 return NULL;
2761 }
Andy Green0d338332011-02-12 11:57:43 +00002762
2763 wsi = malloc(sizeof(struct libwebsocket));
2764 memset(wsi, 0, sizeof (struct libwebsocket));
2765 wsi->sock = sockfd;
Andy Greend6e09112011-03-05 16:12:15 +00002766 wsi->count_active_extensions = 0;
Andy Green0d338332011-02-12 11:57:43 +00002767 wsi->mode = LWS_CONNMODE_SERVER_LISTENER;
Peter Hinz56885f32011-03-02 22:03:47 +00002768 insert_wsi(context, wsi);
Andy Green0d338332011-02-12 11:57:43 +00002769
2770 listen(sockfd, 5);
2771 fprintf(stderr, " Listening on port %d\n", port);
2772
2773 /* list in the internal poll array */
2774
Peter Hinz56885f32011-03-02 22:03:47 +00002775 context->fds[context->fds_count].fd = sockfd;
2776 context->fds[context->fds_count++].events = POLLIN;
Andy Green3221f922011-02-12 13:14:11 +00002777
2778 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00002779 context->protocols[0].callback(context, wsi,
Andy Green3221f922011-02-12 13:14:11 +00002780 LWS_CALLBACK_ADD_POLL_FD,
2781 (void *)(long)sockfd, NULL, POLLIN);
2782
Andy Green8f037e42010-12-19 22:13:26 +00002783 }
Andy Greenb45993c2010-12-18 15:13:50 +00002784
Andy Greene77ddd82010-11-13 10:03:47 +00002785 /* drop any root privs for this process */
Peter Hinz56885f32011-03-02 22:03:47 +00002786#ifdef WIN32
2787#else
Andy Green3faa9c72010-11-08 17:03:03 +00002788 if (gid != -1)
2789 if (setgid(gid))
2790 fprintf(stderr, "setgid: %s\n", strerror(errno));
2791 if (uid != -1)
2792 if (setuid(uid))
2793 fprintf(stderr, "setuid: %s\n", strerror(errno));
Peter Hinz56885f32011-03-02 22:03:47 +00002794#endif
Andy Greenb45993c2010-12-18 15:13:50 +00002795
2796 /* set up our internal broadcast trigger sockets per-protocol */
2797
Peter Hinz56885f32011-03-02 22:03:47 +00002798 for (context->count_protocols = 0;
2799 protocols[context->count_protocols].callback;
2800 context->count_protocols++) {
Andy Green2d1301e2011-05-24 10:14:41 +01002801
2802 fprintf(stderr, " Protocol: %s\n", protocols[context->count_protocols].name);
2803
Peter Hinz56885f32011-03-02 22:03:47 +00002804 protocols[context->count_protocols].owning_server = context;
2805 protocols[context->count_protocols].protocol_index =
2806 context->count_protocols;
Andy Greenb45993c2010-12-18 15:13:50 +00002807
2808 fd = socket(AF_INET, SOCK_STREAM, 0);
2809 if (fd < 0) {
2810 fprintf(stderr, "ERROR opening socket");
Andy Greene92cd172011-01-19 13:11:55 +00002811 return NULL;
Andy Greenb45993c2010-12-18 15:13:50 +00002812 }
Andy Green8f037e42010-12-19 22:13:26 +00002813
Andy Greenb45993c2010-12-18 15:13:50 +00002814 /* allow us to restart even if old sockets in TIME_WAIT */
2815 setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
2816
2817 bzero((char *) &serv_addr, sizeof(serv_addr));
2818 serv_addr.sin_family = AF_INET;
2819 serv_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
2820 serv_addr.sin_port = 0; /* pick the port for us */
2821
2822 n = bind(fd, (struct sockaddr *) &serv_addr, sizeof(serv_addr));
2823 if (n < 0) {
Andy Green8f037e42010-12-19 22:13:26 +00002824 fprintf(stderr, "ERROR on binding to port %d (%d %d)\n",
Andy Greenb45993c2010-12-18 15:13:50 +00002825 port, n, errno);
Andy Greene92cd172011-01-19 13:11:55 +00002826 return NULL;
Andy Greenb45993c2010-12-18 15:13:50 +00002827 }
2828
2829 slen = sizeof cli_addr;
2830 n = getsockname(fd, (struct sockaddr *)&cli_addr, &slen);
2831 if (n < 0) {
2832 fprintf(stderr, "getsockname failed\n");
Andy Greene92cd172011-01-19 13:11:55 +00002833 return NULL;
Andy Greenb45993c2010-12-18 15:13:50 +00002834 }
Peter Hinz56885f32011-03-02 22:03:47 +00002835 protocols[context->count_protocols].broadcast_socket_port =
Andy Greenb45993c2010-12-18 15:13:50 +00002836 ntohs(cli_addr.sin_port);
2837 listen(fd, 5);
2838
2839 debug(" Protocol %s broadcast socket %d\n",
Peter Hinz56885f32011-03-02 22:03:47 +00002840 protocols[context->count_protocols].name,
Andy Greenb45993c2010-12-18 15:13:50 +00002841 ntohs(cli_addr.sin_port));
2842
Andy Green0d338332011-02-12 11:57:43 +00002843 /* dummy wsi per broadcast proxy socket */
2844
2845 wsi = malloc(sizeof(struct libwebsocket));
2846 memset(wsi, 0, sizeof (struct libwebsocket));
2847 wsi->sock = fd;
2848 wsi->mode = LWS_CONNMODE_BROADCAST_PROXY_LISTENER;
Andy Greend6e09112011-03-05 16:12:15 +00002849 wsi->count_active_extensions = 0;
Andy Green0d338332011-02-12 11:57:43 +00002850 /* note which protocol we are proxying */
Peter Hinz56885f32011-03-02 22:03:47 +00002851 wsi->protocol_index_for_broadcast_proxy =
2852 context->count_protocols;
2853 insert_wsi(context, wsi);
Andy Green0d338332011-02-12 11:57:43 +00002854
2855 /* list in internal poll array */
2856
Peter Hinz56885f32011-03-02 22:03:47 +00002857 context->fds[context->fds_count].fd = fd;
2858 context->fds[context->fds_count].events = POLLIN;
2859 context->fds[context->fds_count].revents = 0;
2860 context->fds_count++;
Andy Green3221f922011-02-12 13:14:11 +00002861
2862 /* external POLL support via protocol 0 */
Peter Hinz56885f32011-03-02 22:03:47 +00002863 context->protocols[0].callback(context, wsi,
Andy Green3221f922011-02-12 13:14:11 +00002864 LWS_CALLBACK_ADD_POLL_FD,
2865 (void *)(long)fd, NULL, POLLIN);
Andy Greenb45993c2010-12-18 15:13:50 +00002866 }
2867
Andy Greena41314f2011-05-23 10:00:03 +01002868 /*
2869 * give all extensions a chance to create any per-context
2870 * allocations they need
2871 */
2872
2873 m = LWS_EXT_CALLBACK_CLIENT_CONTEXT_CONSTRUCT;
2874 if (port)
2875 m = LWS_EXT_CALLBACK_SERVER_CONTEXT_CONSTRUCT;
2876 while (extensions->callback) {
Andy Green2d1301e2011-05-24 10:14:41 +01002877 fprintf(stderr, " Extension: %s\n", extensions->name);
Andy Greena41314f2011-05-23 10:00:03 +01002878 extensions->callback(context, extensions,
2879 NULL, m, NULL, NULL, 0);
2880 extensions++;
2881 }
2882
Peter Hinz56885f32011-03-02 22:03:47 +00002883 return context;
Andy Greene92cd172011-01-19 13:11:55 +00002884}
Andy Greenb45993c2010-12-18 15:13:50 +00002885
Andy Green4739e5c2011-01-22 12:51:57 +00002886
Andy Greened11a022011-01-20 10:23:50 +00002887#ifndef LWS_NO_FORK
2888
Andy Greene92cd172011-01-19 13:11:55 +00002889/**
2890 * libwebsockets_fork_service_loop() - Optional helper function forks off
2891 * a process for the websocket server loop.
Andy Green6964bb52011-01-23 16:50:33 +00002892 * You don't have to use this but if not, you
2893 * have to make sure you are calling
2894 * libwebsocket_service periodically to service
2895 * the websocket traffic
Peter Hinz56885f32011-03-02 22:03:47 +00002896 * @context: server context returned by creation function
Andy Greene92cd172011-01-19 13:11:55 +00002897 */
Andy Greenb45993c2010-12-18 15:13:50 +00002898
Andy Greene92cd172011-01-19 13:11:55 +00002899int
Peter Hinz56885f32011-03-02 22:03:47 +00002900libwebsockets_fork_service_loop(struct libwebsocket_context *context)
Andy Greene92cd172011-01-19 13:11:55 +00002901{
Andy Greene92cd172011-01-19 13:11:55 +00002902 int fd;
2903 struct sockaddr_in cli_addr;
2904 int n;
Andy Green3221f922011-02-12 13:14:11 +00002905 int p;
Andy Greenb45993c2010-12-18 15:13:50 +00002906
Andy Greened11a022011-01-20 10:23:50 +00002907 n = fork();
2908 if (n < 0)
2909 return n;
2910
2911 if (!n) {
2912
2913 /* main process context */
2914
Andy Green3221f922011-02-12 13:14:11 +00002915 /*
2916 * set up the proxy sockets to allow broadcast from
2917 * service process context
2918 */
2919
Peter Hinz56885f32011-03-02 22:03:47 +00002920 for (p = 0; p < context->count_protocols; p++) {
Andy Greened11a022011-01-20 10:23:50 +00002921 fd = socket(AF_INET, SOCK_STREAM, 0);
2922 if (fd < 0) {
2923 fprintf(stderr, "Unable to create socket\n");
2924 return -1;
2925 }
2926 cli_addr.sin_family = AF_INET;
2927 cli_addr.sin_port = htons(
Peter Hinz56885f32011-03-02 22:03:47 +00002928 context->protocols[p].broadcast_socket_port);
Andy Greened11a022011-01-20 10:23:50 +00002929 cli_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
2930 n = connect(fd, (struct sockaddr *)&cli_addr,
2931 sizeof cli_addr);
2932 if (n < 0) {
2933 fprintf(stderr, "Unable to connect to "
2934 "broadcast socket %d, %s\n",
Andy Green3221f922011-02-12 13:14:11 +00002935 n, strerror(errno));
Andy Greened11a022011-01-20 10:23:50 +00002936 return -1;
2937 }
2938
Peter Hinz56885f32011-03-02 22:03:47 +00002939 context->protocols[p].broadcast_socket_user_fd = fd;
Andy Greened11a022011-01-20 10:23:50 +00002940 }
2941
Andy Greene92cd172011-01-19 13:11:55 +00002942 return 0;
Andy Greenb45993c2010-12-18 15:13:50 +00002943 }
2944
2945 /* we want a SIGHUP when our parent goes down */
2946 prctl(PR_SET_PDEATHSIG, SIGHUP);
2947
2948 /* in this forked process, sit and service websocket connections */
Andy Green8f037e42010-12-19 22:13:26 +00002949
Andy Greene92cd172011-01-19 13:11:55 +00002950 while (1)
Peter Hinz56885f32011-03-02 22:03:47 +00002951 if (libwebsocket_service(context, 1000))
Andy Greene92cd172011-01-19 13:11:55 +00002952 return -1;
Andy Green8f037e42010-12-19 22:13:26 +00002953
Andy Green251f6fa2010-11-03 11:13:06 +00002954 return 0;
Andy Greenff95d7a2010-10-28 22:36:01 +01002955}
2956
Andy Greened11a022011-01-20 10:23:50 +00002957#endif
2958
Andy Greenb45993c2010-12-18 15:13:50 +00002959/**
2960 * libwebsockets_get_protocol() - Returns a protocol pointer from a websocket
Andy Green8f037e42010-12-19 22:13:26 +00002961 * connection.
Andy Greenb45993c2010-12-18 15:13:50 +00002962 * @wsi: pointer to struct websocket you want to know the protocol of
2963 *
Andy Green8f037e42010-12-19 22:13:26 +00002964 *
2965 * This is useful to get the protocol to broadcast back to from inside
Andy Greenb45993c2010-12-18 15:13:50 +00002966 * the callback.
2967 */
Andy Greenab990e42010-10-31 12:42:52 +00002968
Andy Greenb45993c2010-12-18 15:13:50 +00002969const struct libwebsocket_protocols *
2970libwebsockets_get_protocol(struct libwebsocket *wsi)
2971{
2972 return wsi->protocol;
2973}
2974
2975/**
Andy Greene92cd172011-01-19 13:11:55 +00002976 * libwebsockets_broadcast() - Sends a buffer to the callback for all active
Andy Green8f037e42010-12-19 22:13:26 +00002977 * connections of the given protocol.
Andy Greenb45993c2010-12-18 15:13:50 +00002978 * @protocol: pointer to the protocol you will broadcast to all members of
2979 * @buf: buffer containing the data to be broadcase. NOTE: this has to be
Andy Green8f037e42010-12-19 22:13:26 +00002980 * allocated with LWS_SEND_BUFFER_PRE_PADDING valid bytes before
2981 * the pointer and LWS_SEND_BUFFER_POST_PADDING afterwards in the
2982 * case you are calling this function from callback context.
Andy Greenb45993c2010-12-18 15:13:50 +00002983 * @len: length of payload data in buf, starting from buf.
Andy Green8f037e42010-12-19 22:13:26 +00002984 *
2985 * This function allows bulk sending of a packet to every connection using
Andy Greenb45993c2010-12-18 15:13:50 +00002986 * the given protocol. It does not send the data directly; instead it calls
2987 * the callback with a reason type of LWS_CALLBACK_BROADCAST. If the callback
2988 * wants to actually send the data for that connection, the callback itself
2989 * should call libwebsocket_write().
2990 *
2991 * libwebsockets_broadcast() can be called from another fork context without
2992 * having to take any care about data visibility between the processes, it'll
2993 * "just work".
2994 */
2995
2996
2997int
Andy Green8f037e42010-12-19 22:13:26 +00002998libwebsockets_broadcast(const struct libwebsocket_protocols *protocol,
Andy Greenb45993c2010-12-18 15:13:50 +00002999 unsigned char *buf, size_t len)
3000{
Peter Hinz56885f32011-03-02 22:03:47 +00003001 struct libwebsocket_context *context = protocol->owning_server;
Andy Greenb45993c2010-12-18 15:13:50 +00003002 int n;
Andy Green0d338332011-02-12 11:57:43 +00003003 int m;
3004 struct libwebsocket * wsi;
Andy Greenb45993c2010-12-18 15:13:50 +00003005
3006 if (!protocol->broadcast_socket_user_fd) {
3007 /*
Andy Greene92cd172011-01-19 13:11:55 +00003008 * We are either running unforked / flat, or we are being
3009 * called from poll thread context
Andy Greenb45993c2010-12-18 15:13:50 +00003010 * eg, from a callback. In that case don't use sockets for
3011 * broadcast IPC (since we can't open a socket connection to
3012 * a socket listening on our own thread) but directly do the
3013 * send action.
3014 *
3015 * Locking is not needed because we are by definition being
3016 * called in the poll thread context and are serialized.
3017 */
3018
Andy Green0d338332011-02-12 11:57:43 +00003019 for (n = 0; n < FD_HASHTABLE_MODULUS; n++) {
Andy Greenb45993c2010-12-18 15:13:50 +00003020
Peter Hinz56885f32011-03-02 22:03:47 +00003021 for (m = 0; m < context->fd_hashtable[n].length; m++) {
Andy Greenb45993c2010-12-18 15:13:50 +00003022
Peter Hinz56885f32011-03-02 22:03:47 +00003023 wsi = context->fd_hashtable[n].wsi[m];
Andy Greenb45993c2010-12-18 15:13:50 +00003024
Andy Green0d338332011-02-12 11:57:43 +00003025 if (wsi->mode != LWS_CONNMODE_WS_SERVING)
3026 continue;
Andy Greenb45993c2010-12-18 15:13:50 +00003027
Andy Green0d338332011-02-12 11:57:43 +00003028 /*
3029 * never broadcast to
3030 * non-established connections
3031 */
3032 if (wsi->state != WSI_STATE_ESTABLISHED)
3033 continue;
3034
3035 /* only broadcast to guys using
3036 * requested protocol
3037 */
3038 if (wsi->protocol != protocol)
3039 continue;
3040
Peter Hinz56885f32011-03-02 22:03:47 +00003041 wsi->protocol->callback(context, wsi,
Andy Green8f037e42010-12-19 22:13:26 +00003042 LWS_CALLBACK_BROADCAST,
Andy Green0d338332011-02-12 11:57:43 +00003043 wsi->user_space,
Andy Greenb45993c2010-12-18 15:13:50 +00003044 buf, len);
Andy Green0d338332011-02-12 11:57:43 +00003045 }
Andy Greenb45993c2010-12-18 15:13:50 +00003046 }
3047
3048 return 0;
3049 }
3050
Andy Green0ca6a172010-12-19 20:50:01 +00003051 /*
3052 * We're being called from a different process context than the server
3053 * loop. Instead of broadcasting directly, we send our
3054 * payload on a socket to do the IPC; the server process will serialize
3055 * the broadcast action in its main poll() loop.
3056 *
3057 * There's one broadcast socket listening for each protocol supported
3058 * set up when the websocket server initializes
3059 */
3060
Andy Green6964bb52011-01-23 16:50:33 +00003061 n = send(protocol->broadcast_socket_user_fd, buf, len, MSG_NOSIGNAL);
Andy Greenb45993c2010-12-18 15:13:50 +00003062
3063 return n;
3064}
Andy Green82c3d542011-03-07 21:16:31 +00003065
3066int
3067libwebsocket_is_final_fragment(struct libwebsocket *wsi)
3068{
3069 return wsi->final;
3070}