mirror of
https://github.com/warmcat/libwebsockets.git
synced 2025-03-16 00:00:07 +01:00

https://github.com/warmcat/libwebsockets/issues/1625 "dead bodies" that were sent but not processed by lws as server will clog up and destroy transaction tracking if repeated POSTs with keepalive are sent to nonexistant paths. This patch introduces a DISCARD_BODY state that follows BODY except the payload is not signalled to the protocol callback. Calling transaction_completed() with pending body makes lws enter DISCARD_BODY and retry transaction completed only after the pending body is exhausted.
335 lines
11 KiB
C
335 lines
11 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
|
|
*
|
|
* This is included from core/private.h
|
|
*/
|
|
|
|
typedef uint32_t lws_wsi_state_t;
|
|
|
|
/*
|
|
* The wsi->role_ops pointer decides almost everything about what role the wsi
|
|
* will play, h2, raw, ws, etc.
|
|
*
|
|
* However there are a few additional flags needed that vary, such as if the
|
|
* role is a client or server side, if it has that concept. And the connection
|
|
* fulfilling the role, has a separate dynamic state.
|
|
*
|
|
* 31 16 15 0
|
|
* [ role flags ] [ state ]
|
|
*
|
|
* The role flags part is generally invariant for the lifetime of the wsi,
|
|
* although it can change if the connection role itself does, eg, if the
|
|
* connection upgrades from H1 -> WS1 the role flags may be changed at that
|
|
* point.
|
|
*
|
|
* The state part reflects the dynamic connection state, and the states are
|
|
* reused between roles.
|
|
*
|
|
* None of the internal role or state representations are made available outside
|
|
* of lws internals. Even for lws internals, if you add stuff here, please keep
|
|
* the constants inside this header only by adding necessary helpers here and
|
|
* use the helpers in the actual code. This is to ease any future refactors.
|
|
*
|
|
* Notice LWSIFR_ENCAP means we have a parent wsi that actually carries our
|
|
* data as a stream inside a different protocol.
|
|
*/
|
|
|
|
#define _RS 16
|
|
|
|
#define LWSIFR_CLIENT (0x1000 << _RS) /* client side */
|
|
#define LWSIFR_SERVER (0x2000 << _RS) /* server side */
|
|
|
|
#define LWSIFR_P_ENCAP_H2 (0x0100 << _RS) /* we are encapsulated by h2 */
|
|
|
|
enum lwsi_role {
|
|
LWSI_ROLE_MASK = (0xffff << _RS),
|
|
LWSI_ROLE_ENCAP_MASK = (0x0f00 << _RS),
|
|
};
|
|
|
|
#define lwsi_role(wsi) (wsi->wsistate & LWSI_ROLE_MASK)
|
|
#if !defined (_DEBUG)
|
|
#define lwsi_set_role(wsi, role) wsi->wsistate = \
|
|
(wsi->wsistate & (~LWSI_ROLE_MASK)) | role
|
|
#else
|
|
void lwsi_set_role(struct lws *wsi, lws_wsi_state_t role);
|
|
#endif
|
|
|
|
#define lwsi_role_client(wsi) (!!(wsi->wsistate & LWSIFR_CLIENT))
|
|
#define lwsi_role_server(wsi) (!!(wsi->wsistate & LWSIFR_SERVER))
|
|
#define lwsi_role_h2_ENCAPSULATION(wsi) \
|
|
((wsi->wsistate & LWSI_ROLE_ENCAP_MASK) == LWSIFR_P_ENCAP_H2)
|
|
|
|
/* Pollout wants a callback in this state */
|
|
#define LWSIFS_POCB (0x100)
|
|
/* Before any protocol connection was established */
|
|
#define LWSIFS_NOT_EST (0x200)
|
|
|
|
enum lwsi_state {
|
|
|
|
/* Phase 1: pre-transport */
|
|
|
|
LRS_UNCONNECTED = LWSIFS_NOT_EST | 0,
|
|
LRS_WAITING_CONNECT = LWSIFS_NOT_EST | 1,
|
|
|
|
/* Phase 2: establishing intermediaries on top of transport */
|
|
|
|
LRS_WAITING_PROXY_REPLY = LWSIFS_NOT_EST | 2,
|
|
LRS_WAITING_SSL = LWSIFS_NOT_EST | 3,
|
|
LRS_WAITING_SOCKS_GREETING_REPLY = LWSIFS_NOT_EST | 4,
|
|
LRS_WAITING_SOCKS_CONNECT_REPLY = LWSIFS_NOT_EST | 5,
|
|
LRS_WAITING_SOCKS_AUTH_REPLY = LWSIFS_NOT_EST | 6,
|
|
|
|
/* Phase 3: establishing tls tunnel */
|
|
|
|
LRS_SSL_INIT = LWSIFS_NOT_EST | 7,
|
|
LRS_SSL_ACK_PENDING = LWSIFS_NOT_EST | 8,
|
|
LRS_PRE_WS_SERVING_ACCEPT = LWSIFS_NOT_EST | 9,
|
|
|
|
/* Phase 4: connected */
|
|
|
|
LRS_WAITING_SERVER_REPLY = LWSIFS_NOT_EST | 10,
|
|
LRS_H2_AWAIT_PREFACE = LWSIFS_NOT_EST | 11,
|
|
LRS_H2_AWAIT_SETTINGS = LWSIFS_NOT_EST |
|
|
LWSIFS_POCB | 12,
|
|
|
|
/* Phase 5: protocol logically established */
|
|
|
|
LRS_H2_CLIENT_SEND_SETTINGS = LWSIFS_POCB | 13,
|
|
LRS_H2_WAITING_TO_SEND_HEADERS = LWSIFS_POCB | 14,
|
|
LRS_DEFERRING_ACTION = LWSIFS_POCB | 15,
|
|
LRS_IDLING = 16,
|
|
LRS_H1C_ISSUE_HANDSHAKE = 17,
|
|
LRS_H1C_ISSUE_HANDSHAKE2 = 18,
|
|
LRS_ISSUE_HTTP_BODY = 19,
|
|
LRS_ISSUING_FILE = 20,
|
|
LRS_HEADERS = 21,
|
|
LRS_BODY = 22,
|
|
LRS_DISCARD_BODY = 31,
|
|
LRS_ESTABLISHED = LWSIFS_POCB | 23,
|
|
/* we are established, but we have embarked on serving a single
|
|
* transaction. Other transaction input may be pending, but we will
|
|
* not service it while we are busy dealing with the current
|
|
* transaction.
|
|
*
|
|
* When we complete the current transaction, we would reset our state
|
|
* back to ESTABLISHED and start to process the next transaction.
|
|
*/
|
|
LRS_DOING_TRANSACTION = LWSIFS_POCB | 24,
|
|
|
|
/* Phase 6: finishing */
|
|
|
|
LRS_WAITING_TO_SEND_CLOSE = LWSIFS_POCB | 25,
|
|
LRS_RETURNED_CLOSE = LWSIFS_POCB | 26,
|
|
LRS_AWAITING_CLOSE_ACK = LWSIFS_POCB | 27,
|
|
LRS_FLUSHING_BEFORE_CLOSE = LWSIFS_POCB | 28,
|
|
LRS_SHUTDOWN = 29,
|
|
|
|
/* Phase 7: dead */
|
|
|
|
LRS_DEAD_SOCKET = 30,
|
|
|
|
LRS_MASK = 0xffff
|
|
};
|
|
|
|
#define lwsi_state(wsi) ((enum lwsi_state)(wsi->wsistate & LRS_MASK))
|
|
#define lwsi_state_PRE_CLOSE(wsi) \
|
|
((enum lwsi_state)(wsi->wsistate_pre_close & LRS_MASK))
|
|
#define lwsi_state_est(wsi) (!(wsi->wsistate & LWSIFS_NOT_EST))
|
|
#define lwsi_state_est_PRE_CLOSE(wsi) \
|
|
(!(wsi->wsistate_pre_close & LWSIFS_NOT_EST))
|
|
#define lwsi_state_can_handle_POLLOUT(wsi) (wsi->wsistate & LWSIFS_POCB)
|
|
#if !defined (_DEBUG)
|
|
#define lwsi_set_state(wsi, lrs) wsi->wsistate = \
|
|
(wsi->wsistate & (~LRS_MASK)) | lrs
|
|
#else
|
|
void lwsi_set_state(struct lws *wsi, lws_wsi_state_t lrs);
|
|
#endif
|
|
|
|
#define _LWS_ADOPT_FINISH (1 << 24)
|
|
|
|
/*
|
|
* internal role-specific ops
|
|
*/
|
|
struct lws_context_per_thread;
|
|
struct lws_role_ops {
|
|
const char *name;
|
|
const char *alpn;
|
|
/*
|
|
* After http headers have parsed, this is the last chance for a role
|
|
* to upgrade the connection to something else using the headers.
|
|
* ws-over-h2 is upgraded from h2 like this.
|
|
*/
|
|
int (*check_upgrades)(struct lws *wsi);
|
|
/* role-specific context init during context creation */
|
|
int (*init_context)(struct lws_context *context,
|
|
const struct lws_context_creation_info *info);
|
|
/* role-specific per-vhost init during vhost creation */
|
|
int (*init_vhost)(struct lws_vhost *vh,
|
|
const struct lws_context_creation_info *info);
|
|
/* role-specific per-vhost destructor during vhost destroy */
|
|
int (*destroy_vhost)(struct lws_vhost *vh);
|
|
/* generic 1Hz callback for the role itself */
|
|
int (*periodic_checks)(struct lws_context *context, int tsi,
|
|
time_t now);
|
|
/* chance for the role to force POLLIN without network activity */
|
|
int (*service_flag_pending)(struct lws_context *context, int tsi);
|
|
/* an fd using this role has POLLIN signalled */
|
|
int (*handle_POLLIN)(struct lws_context_per_thread *pt, struct lws *wsi,
|
|
struct lws_pollfd *pollfd);
|
|
/* an fd using the role wanted a POLLOUT callback and now has it */
|
|
int (*handle_POLLOUT)(struct lws *wsi);
|
|
/* perform user pollout */
|
|
int (*perform_user_POLLOUT)(struct lws *wsi);
|
|
/* do effective callback on writeable */
|
|
int (*callback_on_writable)(struct lws *wsi);
|
|
/* connection-specific tx credit in bytes */
|
|
lws_fileofs_t (*tx_credit)(struct lws *wsi);
|
|
/* role-specific write formatting */
|
|
int (*write_role_protocol)(struct lws *wsi, unsigned char *buf,
|
|
size_t len, enum lws_write_protocol *wp);
|
|
|
|
/* get encapsulation parent */
|
|
struct lws * (*encapsulation_parent)(struct lws *wsi);
|
|
|
|
/* role-specific destructor */
|
|
int (*alpn_negotiated)(struct lws *wsi, const char *alpn);
|
|
|
|
/* chance for the role to handle close in the protocol */
|
|
int (*close_via_role_protocol)(struct lws *wsi,
|
|
enum lws_close_status reason);
|
|
/* role-specific close processing */
|
|
int (*close_role)(struct lws_context_per_thread *pt, struct lws *wsi);
|
|
/* role-specific connection close processing */
|
|
int (*close_kill_connection)(struct lws *wsi,
|
|
enum lws_close_status reason);
|
|
/* role-specific destructor */
|
|
int (*destroy_role)(struct lws *wsi);
|
|
|
|
/* role-specific socket-adopt */
|
|
int (*adoption_bind)(struct lws *wsi, int type, const char *prot);
|
|
/* role-specific client-bind:
|
|
* ret 1 = bound, 0 = not bound, -1 = fail out
|
|
* i may be NULL, indicating client_bind is being called after
|
|
* a successful bind earlier, to finalize the binding. In that
|
|
* case ret 0 = OK, 1 = fail, wsi needs freeing, -1 = fail, wsi freed */
|
|
int (*client_bind)(struct lws *wsi,
|
|
const struct lws_client_connect_info *i);
|
|
|
|
/*
|
|
* the callback reasons for adoption for client, server
|
|
* (just client applies if no concept of client or server)
|
|
*/
|
|
uint16_t adoption_cb[2];
|
|
/*
|
|
* the callback reasons for adoption for client, server
|
|
* (just client applies if no concept of client or server)
|
|
*/
|
|
uint16_t rx_cb[2];
|
|
/*
|
|
* the callback reasons for WRITEABLE for client, server
|
|
* (just client applies if no concept of client or server)
|
|
*/
|
|
uint16_t writeable_cb[2];
|
|
/*
|
|
* the callback reasons for CLOSE for client, server
|
|
* (just client applies if no concept of client or server)
|
|
*/
|
|
uint16_t close_cb[2];
|
|
/*
|
|
* the callback reasons for protocol bind for client, server
|
|
* (just client applies if no concept of client or server)
|
|
*/
|
|
uint16_t protocol_bind_cb[2];
|
|
/*
|
|
* the callback reasons for protocol unbind for client, server
|
|
* (just client applies if no concept of client or server)
|
|
*/
|
|
uint16_t protocol_unbind_cb[2];
|
|
|
|
unsigned int file_handle:1; /* role operates on files not sockets */
|
|
};
|
|
|
|
/* core roles */
|
|
extern struct lws_role_ops role_ops_raw_skt, role_ops_raw_file, role_ops_listen,
|
|
role_ops_pipe;
|
|
|
|
/* bring in role private declarations */
|
|
|
|
#if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)
|
|
#include "roles/http/private.h"
|
|
#else
|
|
#define lwsi_role_http(wsi) (0)
|
|
#endif
|
|
|
|
#if defined(LWS_ROLE_H1)
|
|
#include "roles/h1/private.h"
|
|
#else
|
|
#define lwsi_role_h1(wsi) (0)
|
|
#endif
|
|
|
|
#if defined(LWS_ROLE_H2)
|
|
#include "roles/h2/private.h"
|
|
#else
|
|
#define lwsi_role_h2(wsi) (0)
|
|
#endif
|
|
|
|
#if defined(LWS_ROLE_WS)
|
|
#include "roles/ws/private.h"
|
|
#else
|
|
#define lwsi_role_ws(wsi) (0)
|
|
#endif
|
|
|
|
#if defined(LWS_ROLE_CGI)
|
|
#include "roles/cgi/private.h"
|
|
#else
|
|
#define lwsi_role_cgi(wsi) (0)
|
|
#endif
|
|
|
|
#if defined(LWS_ROLE_DBUS)
|
|
#include "roles/dbus/private.h"
|
|
#else
|
|
#define lwsi_role_dbus(wsi) (0)
|
|
#endif
|
|
|
|
#if defined(LWS_ROLE_RAW_PROXY)
|
|
#include "roles/raw-proxy/private.h"
|
|
#else
|
|
#define lwsi_role_raw_proxy(wsi) (0)
|
|
#endif
|
|
|
|
enum {
|
|
LWS_HP_RET_BAIL_OK,
|
|
LWS_HP_RET_BAIL_DIE,
|
|
LWS_HP_RET_USER_SERVICE,
|
|
|
|
LWS_HPI_RET_WSI_ALREADY_DIED, /* we closed it */
|
|
LWS_HPI_RET_HANDLED, /* no probs */
|
|
LWS_HPI_RET_PLEASE_CLOSE_ME, /* close it for us */
|
|
|
|
LWS_UPG_RET_DONE,
|
|
LWS_UPG_RET_CONTINUE,
|
|
LWS_UPG_RET_BAIL
|
|
};
|
|
|
|
int
|
|
lws_role_call_adoption_bind(struct lws *wsi, int type, const char *prot);
|
|
|
|
struct lws *
|
|
lws_client_connect_3(struct lws *wsi, struct lws *wsi_piggyback, ssize_t plen);
|