2 * libwebsockets - small server side websockets and web server implementation
4 * Copyright (C) 2010-2014 Andy Green <andy@warmcat.com>
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.
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.
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,
22 #include "private-libwebsockets.h"
25 insert_wsi_socket_into_fds(struct libwebsocket_context *context,
26 struct libwebsocket *wsi)
28 struct libwebsocket_pollargs pa = { wsi->sock, LWS_POLLIN, 0 };
30 if (context->fds_count >= context->max_fds) {
31 lwsl_err("Too many fds (%d)\n", context->max_fds);
35 if (wsi->sock >= context->max_fds) {
36 lwsl_err("Socket fd %d is too high (%d)\n",
37 wsi->sock, context->max_fds);
42 assert(wsi->sock >= 0);
44 lwsl_info("insert_wsi_socket_into_fds: wsi=%p, sock=%d, fds pos=%d\n",
45 wsi, wsi->sock, context->fds_count);
47 context->protocols[0].callback(context, wsi,
48 LWS_CALLBACK_LOCK_POLL,
49 wsi->user_space, (void *) &pa, 0);
51 context->lws_lookup[wsi->sock] = wsi;
52 wsi->position_in_fds_table = context->fds_count;
53 context->fds[context->fds_count].fd = wsi->sock;
54 context->fds[context->fds_count].events = LWS_POLLIN;
56 lws_plat_insert_socket_into_fds(context, wsi);
58 /* external POLL support via protocol 0 */
59 context->protocols[0].callback(context, wsi,
60 LWS_CALLBACK_ADD_POLL_FD,
61 wsi->user_space, (void *) &pa, 0);
63 context->protocols[0].callback(context, wsi,
64 LWS_CALLBACK_UNLOCK_POLL,
65 wsi->user_space, (void *)&pa, 0);
71 remove_wsi_socket_from_fds(struct libwebsocket_context *context,
72 struct libwebsocket *wsi)
75 struct libwebsocket_pollargs pa = { wsi->sock, 0, 0 };
77 lws_libev_io(context, wsi, LWS_EV_STOP | LWS_EV_READ | LWS_EV_WRITE);
79 if (!--context->fds_count) {
80 context->protocols[0].callback(context, wsi,
81 LWS_CALLBACK_LOCK_POLL,
82 wsi->user_space, (void *) &pa, 0);
86 if (wsi->sock > context->max_fds) {
87 lwsl_err("Socket fd %d too high (%d)\n",
88 wsi->sock, context->max_fds);
92 lwsl_info("%s: wsi=%p, sock=%d, fds pos=%d\n", __func__,
93 wsi, wsi->sock, wsi->position_in_fds_table);
95 context->protocols[0].callback(context, wsi,
96 LWS_CALLBACK_LOCK_POLL,
97 wsi->user_space, (void *)&pa, 0);
99 m = wsi->position_in_fds_table; /* replace the contents for this */
101 /* have the last guy take up the vacant slot */
102 context->fds[m] = context->fds[context->fds_count];
104 lws_plat_delete_socket_from_fds(context, wsi, m);
107 * end guy's fds_lookup entry remains unchanged
108 * (still same fd pointing to same wsi)
110 /* end guy's "position in fds table" changed */
111 context->lws_lookup[context->fds[context->fds_count].fd]->
112 position_in_fds_table = m;
113 /* deletion guy's lws_lookup entry needs nuking */
114 context->lws_lookup[wsi->sock] = NULL;
115 /* removed wsi has no position any more */
116 wsi->position_in_fds_table = -1;
119 /* remove also from external POLL support via protocol 0 */
121 context->protocols[0].callback(context, wsi,
122 LWS_CALLBACK_DEL_POLL_FD, wsi->user_space,
125 context->protocols[0].callback(context, wsi,
126 LWS_CALLBACK_UNLOCK_POLL,
127 wsi->user_space, (void *) &pa, 0);
132 lws_change_pollfd(struct libwebsocket *wsi, int _and, int _or)
134 struct libwebsocket_context *context = wsi->protocol->owning_server;
137 struct libwebsocket_pollfd *pfd;
138 struct libwebsocket_pollargs pa;
140 pfd = &context->fds[wsi->position_in_fds_table];
143 context->protocols[0].callback(context, wsi,
144 LWS_CALLBACK_LOCK_POLL, wsi->user_space, (void *) &pa, 0);
146 pa.prev_events = pfd->events;
147 pa.events = pfd->events = (pfd->events & ~_and) | _or;
149 context->protocols[0].callback(context, wsi,
150 LWS_CALLBACK_CHANGE_MODE_POLL_FD,
151 wsi->user_space, (void *) &pa, 0);
154 * if we changed something in this pollfd...
155 * ... and we're running in a different thread context
156 * than the service thread...
157 * ... and the service thread is waiting ...
158 * then cancel it to force a restart with our changed events
160 if (pa.prev_events != pa.events) {
162 if (lws_plat_change_pollfd(context, wsi, pfd)) {
163 lwsl_info("%s failed\n", __func__);
167 sampled_tid = context->service_tid;
169 tid = context->protocols[0].callback(context, NULL,
170 LWS_CALLBACK_GET_THREAD_ID, NULL, NULL, 0);
171 if (tid != sampled_tid)
172 libwebsocket_cancel_service(context);
176 context->protocols[0].callback(context, wsi,
177 LWS_CALLBACK_UNLOCK_POLL, wsi->user_space, (void *) &pa, 0);
184 * libwebsocket_callback_on_writable() - Request a callback when this socket
185 * becomes able to be written to without
188 * @context: libwebsockets context
189 * @wsi: Websocket connection instance to get callback for
193 libwebsocket_callback_on_writable(struct libwebsocket_context *context,
194 struct libwebsocket *wsi)
197 struct libwebsocket *network_wsi, *wsi2;
200 lwsl_info("%s: %p\n", __func__, wsi);
202 if (wsi->mode != LWS_CONNMODE_HTTP2_SERVING)
205 if (wsi->u.http2.requested_POLLOUT) {
206 lwsl_info("already pending writable\n");
210 network_wsi = lws_http2_get_network_wsi(wsi);
211 already = network_wsi->u.http2.requested_POLLOUT;
213 /* mark everybody above him as requesting pollout */
217 wsi2->u.http2.requested_POLLOUT = 1;
218 lwsl_info("mark %p pending writable\n", wsi2);
219 wsi2 = wsi2->u.http2.parent_wsi;
222 /* for network action, act only on the network wsi */
230 if (lws_ext_callback_for_each_active(wsi,
231 LWS_EXT_CALLBACK_REQUEST_ON_WRITEABLE, NULL, 0))
234 if (wsi->position_in_fds_table < 0) {
235 lwsl_err("%s: failed to find socket %d\n", __func__, wsi->sock);
239 if (lws_change_pollfd(wsi, 0, LWS_POLLOUT))
242 lws_libev_io(context, wsi, LWS_EV_START | LWS_EV_WRITE);
248 * libwebsocket_callback_on_writable_all_protocol() - Request a callback for
249 * all connections using the given protocol when it
250 * becomes possible to write to each socket without
253 * @protocol: Protocol whose connections will get callbacks
257 libwebsocket_callback_on_writable_all_protocol(
258 const struct libwebsocket_protocols *protocol)
260 struct libwebsocket_context *context = protocol->owning_server;
262 struct libwebsocket *wsi;
264 for (n = 0; n < context->fds_count; n++) {
265 wsi = context->lws_lookup[context->fds[n].fd];
268 if (wsi->protocol == protocol)
269 libwebsocket_callback_on_writable(context, wsi);