mirror of
https://github.com/warmcat/libwebsockets.git
synced 2025-03-16 00:00:07 +01:00
5c0b0450f2
HTTP server protocols have had for a while LWS_CALLBACK_HTTP_DROP/BIND_PROTOCOL callbacks that mark when a wsi is attched to a protocol and detached. It turns out this is generally useful for everything to know when a wsi is joining a protocol and definitively completely finished with a protocol. Particularly with client wsi where you provided the userdata externally, this makes a clear point to free() it on the protocol binding being dropped. This patch adds protocol bind / unbind callbacks to the role definition and lets them operate on all roles. For the various roles HTTP server: LWS_CALLBACK_HTTP_BIND/DROP_PROTOCOL as before HTTP client: LWS_CALLBACK_CLIENT_HTTP_BIND/DROP_PROTOCOL ws server: LWS_CALLBACK_WS_SERVER_BIND/DROP_PROTOCOL ws client: LWS_CALLBACK_WS_CLIENT_BIND/DROP_PROTOCOL raw file: LWS_CALLBACK_RAW_FILE_BIND/DROP_PROTOCOL raw skt: LWS_CALLBACK_RAW_SKT_BIND/DROP_PROTOCOL
122 lines
3.3 KiB
C
122 lines
3.3 KiB
C
/*
|
|
* libwebsockets - small server side websockets and web server implementation
|
|
*
|
|
* Copyright (C) 2010-2018 Andy Green <andy@warmcat.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation:
|
|
* version 2.1 of the License.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
* MA 02110-1301 USA
|
|
*/
|
|
|
|
#include <core/private.h>
|
|
|
|
static int
|
|
rops_handle_POLLIN_cgi(struct lws_context_per_thread *pt, struct lws *wsi,
|
|
struct lws_pollfd *pollfd)
|
|
{
|
|
struct lws_cgi_args args;
|
|
|
|
assert(wsi->role_ops == &role_ops_cgi);
|
|
|
|
if (wsi->cgi_channel >= LWS_STDOUT &&
|
|
!(pollfd->revents & pollfd->events & LWS_POLLIN))
|
|
return LWS_HPI_RET_HANDLED;
|
|
|
|
if (wsi->cgi_channel == LWS_STDIN &&
|
|
!(pollfd->revents & pollfd->events & LWS_POLLOUT))
|
|
return LWS_HPI_RET_HANDLED;
|
|
|
|
if (wsi->cgi_channel == LWS_STDIN &&
|
|
lws_change_pollfd(wsi, LWS_POLLOUT, 0)) {
|
|
lwsl_info("failed at set pollfd\n");
|
|
return LWS_HPI_RET_WSI_ALREADY_DIED;
|
|
}
|
|
|
|
args.ch = wsi->cgi_channel;
|
|
args.stdwsi = &wsi->parent->http.cgi->stdwsi[0];
|
|
args.hdr_state = wsi->hdr_state;
|
|
|
|
lwsl_debug("CGI LWS_STDOUT %p wsistate 0x%x\n",
|
|
wsi->parent, wsi->wsistate);
|
|
|
|
if (user_callback_handle_rxflow(wsi->parent->protocol->callback,
|
|
wsi->parent, LWS_CALLBACK_CGI,
|
|
wsi->parent->user_space,
|
|
(void *)&args, 0))
|
|
return 1;
|
|
|
|
return LWS_HPI_RET_HANDLED;
|
|
}
|
|
|
|
static int
|
|
rops_handle_POLLOUT_cgi(struct lws *wsi)
|
|
{
|
|
return LWS_HP_RET_USER_SERVICE;
|
|
}
|
|
|
|
static int
|
|
rops_periodic_checks_cgi(struct lws_context *context, int tsi, time_t now)
|
|
{
|
|
struct lws_context_per_thread *pt = &context->pt[tsi];
|
|
|
|
lws_cgi_kill_terminated(pt);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
rops_destroy_role_cgi(struct lws *wsi)
|
|
{
|
|
#if defined(LWS_WITH_ZLIB)
|
|
if (!wsi->http.cgi)
|
|
return 0;
|
|
if (!wsi->http.cgi->gzip_init)
|
|
return 0;
|
|
|
|
inflateEnd(&wsi->http.cgi->inflate);
|
|
wsi->http.cgi->gzip_init = 0;
|
|
#endif
|
|
|
|
return 0;
|
|
}
|
|
|
|
struct lws_role_ops role_ops_cgi = {
|
|
/* role name */ "cgi",
|
|
/* alpn id */ NULL,
|
|
/* check_upgrades */ NULL,
|
|
/* init_context */ NULL,
|
|
/* init_vhost */ NULL,
|
|
/* destroy_vhost */ NULL,
|
|
/* periodic_checks */ rops_periodic_checks_cgi,
|
|
/* service_flag_pending */ NULL,
|
|
/* handle_POLLIN */ rops_handle_POLLIN_cgi,
|
|
/* handle_POLLOUT */ rops_handle_POLLOUT_cgi,
|
|
/* perform_user_POLLOUT */ NULL,
|
|
/* callback_on_writable */ NULL,
|
|
/* tx_credit */ NULL,
|
|
/* write_role_protocol */ NULL,
|
|
/* encapsulation_parent */ NULL,
|
|
/* alpn_negotiated */ NULL,
|
|
/* close_via_role_protocol */ NULL,
|
|
/* close_role */ NULL,
|
|
/* close_kill_connection */ NULL,
|
|
/* destroy_role */ rops_destroy_role_cgi,
|
|
/* adoption_bind */ NULL,
|
|
/* client_bind */ NULL,
|
|
/* writeable cb clnt, srv */ { 0, 0 },
|
|
/* close cb clnt, srv */ { 0, 0 },
|
|
/* protocol_bind_cb c,s */ { 0, 0 },
|
|
/* protocol_unbind_cb c,s */ { 0, 0 },
|
|
/* file_handle */ 0,
|
|
};
|