260 lines
9.4 KiB
C
260 lines
9.4 KiB
C
/*
|
|
* libwebsockets - small server side websockets and web server implementation
|
|
*
|
|
* Copyright (C) 2010 - 2019 Andy Green <andy@warmcat.com>
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to
|
|
* deal in the Software without restriction, including without limitation the
|
|
* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
|
|
* sell copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
|
* IN THE SOFTWARE.
|
|
*/
|
|
|
|
/*! \defgroup sending-data Sending data
|
|
|
|
APIs related to writing data on a connection
|
|
*/
|
|
//@{
|
|
#define LWS_WRITE_RAW LWS_WRITE_HTTP
|
|
|
|
/*
|
|
* NOTE: These public enums are part of the abi. If you want to add one,
|
|
* add it at where specified so existing users are unaffected.
|
|
*/
|
|
enum lws_write_protocol {
|
|
LWS_WRITE_TEXT = 0,
|
|
/**< Send a ws TEXT message,the pointer must have LWS_PRE valid
|
|
* memory behind it.
|
|
*
|
|
* The receiver expects only valid utf-8 in the payload */
|
|
LWS_WRITE_BINARY = 1,
|
|
/**< Send a ws BINARY message, the pointer must have LWS_PRE valid
|
|
* memory behind it.
|
|
*
|
|
* Any sequence of bytes is valid */
|
|
LWS_WRITE_CONTINUATION = 2,
|
|
/**< Continue a previous ws message, the pointer must have LWS_PRE valid
|
|
* memory behind it */
|
|
LWS_WRITE_HTTP = 3,
|
|
/**< Send HTTP content */
|
|
|
|
/* LWS_WRITE_CLOSE is handled by lws_close_reason() */
|
|
LWS_WRITE_PING = 5,
|
|
LWS_WRITE_PONG = 6,
|
|
|
|
/* Same as write_http but we know this write ends the transaction */
|
|
LWS_WRITE_HTTP_FINAL = 7,
|
|
|
|
/* HTTP2 */
|
|
|
|
LWS_WRITE_HTTP_HEADERS = 8,
|
|
/**< Send http headers (http2 encodes this payload and LWS_WRITE_HTTP
|
|
* payload differently, http 1.x links also handle this correctly. so
|
|
* to be compatible with both in the future,header response part should
|
|
* be sent using this regardless of http version expected)
|
|
*/
|
|
LWS_WRITE_HTTP_HEADERS_CONTINUATION = 9,
|
|
/**< Continuation of http/2 headers
|
|
*/
|
|
|
|
/****** add new things just above ---^ ******/
|
|
|
|
/* flags */
|
|
|
|
LWS_WRITE_BUFLIST = 0x20,
|
|
/**< Don't actually write it... stick it on the output buflist and
|
|
* write it as soon as possible. Useful if you learn you have to
|
|
* write something, have the data to write to hand but the timing is
|
|
* unrelated as to whether the connection is writable or not, and were
|
|
* otherwise going to have to allocate a temp buffer and write it
|
|
* later anyway */
|
|
|
|
LWS_WRITE_NO_FIN = 0x40,
|
|
/**< This part of the message is not the end of the message */
|
|
|
|
LWS_WRITE_H2_STREAM_END = 0x80,
|
|
/**< Flag indicates this packet should go out with STREAM_END if h2
|
|
* STREAM_END is allowed on DATA or HEADERS.
|
|
*/
|
|
|
|
LWS_WRITE_CLIENT_IGNORE_XOR_MASK = 0x80
|
|
/**< client packet payload goes out on wire unmunged
|
|
* only useful for security tests since normal servers cannot
|
|
* decode the content if used */
|
|
};
|
|
|
|
/* used with LWS_CALLBACK_CHILD_WRITE_VIA_PARENT */
|
|
|
|
struct lws_write_passthru {
|
|
struct lws *wsi;
|
|
unsigned char *buf;
|
|
size_t len;
|
|
enum lws_write_protocol wp;
|
|
};
|
|
|
|
|
|
/**
|
|
* lws_write() - Apply protocol then write data to client
|
|
*
|
|
* \param wsi: Websocket instance (available from user callback)
|
|
* \param buf: The data to send. For data being sent on a websocket
|
|
* connection (ie, not default http), this buffer MUST have
|
|
* LWS_PRE bytes valid BEFORE the pointer.
|
|
* This is so the protocol header data can be added in-situ.
|
|
* \param len: Count of the data bytes in the payload starting from buf
|
|
* \param protocol: Use LWS_WRITE_HTTP to reply to an http connection, and one
|
|
* of LWS_WRITE_BINARY or LWS_WRITE_TEXT to send appropriate
|
|
* data on a websockets connection. Remember to allow the extra
|
|
* bytes before and after buf if LWS_WRITE_BINARY or LWS_WRITE_TEXT
|
|
* are used.
|
|
*
|
|
* This function provides the way to issue data back to the client, for any
|
|
* role (h1, h2, ws, raw, etc). It can only be called from the WRITEABLE
|
|
* callback.
|
|
*
|
|
* IMPORTANT NOTICE!
|
|
*
|
|
* When sending with ws protocol
|
|
*
|
|
* LWS_WRITE_TEXT,
|
|
* LWS_WRITE_BINARY,
|
|
* LWS_WRITE_CONTINUATION,
|
|
* LWS_WRITE_PING,
|
|
* LWS_WRITE_PONG,
|
|
*
|
|
* or sending on http/2... the send buffer has to have LWS_PRE bytes valid
|
|
* BEFORE the buffer pointer you pass to lws_write(). Since you'll probably
|
|
* want to use http/2 before too long, it's wise to just always do this with
|
|
* lws_write buffers... LWS_PRE is typically 16 bytes it's not going to hurt
|
|
* usually.
|
|
*
|
|
* start of alloc ptr passed to lws_write end of allocation
|
|
* | | |
|
|
* v <-- LWS_PRE bytes --> v v
|
|
* [---------------- allocated memory ---------------]
|
|
* (for lws use) [====== user buffer ======]
|
|
*
|
|
* This allows us to add protocol info before the data, and send as one packet
|
|
* on the network without payload copying, for maximum efficiency.
|
|
*
|
|
* So for example you need this kind of code to use lws_write with a
|
|
* 128-byte payload
|
|
*
|
|
* char buf[LWS_PRE + 128];
|
|
*
|
|
* // fill your part of the buffer... for example here it's all zeros
|
|
* memset(&buf[LWS_PRE], 0, 128);
|
|
*
|
|
* if (lws_write(wsi, &buf[LWS_PRE], 128, LWS_WRITE_TEXT) < 128) {
|
|
* ... the connection is dead ...
|
|
* return -1;
|
|
* }
|
|
*
|
|
* LWS_PRE is currently 16, which covers ws and h2 frame headers, and is
|
|
* compatible with 32 and 64-bit alignment requirements.
|
|
*
|
|
* (LWS_SEND_BUFFER_POST_PADDING is deprecated, it's now 0 and can be left off.)
|
|
*
|
|
* Return may be -1 is the write failed in a way indicating that the connection
|
|
* has ended already, in which case you can close your side, or a positive
|
|
* number that is at least the number of bytes requested to send (under some
|
|
* encapsulation scenarios, it can indicate more than you asked was sent).
|
|
*
|
|
* The recommended test of the return is less than what you asked indicates
|
|
* the connection has failed.
|
|
*
|
|
* Truncated Writes
|
|
* ================
|
|
*
|
|
* The OS may not accept everything you asked to write on the connection.
|
|
*
|
|
* Posix defines POLLOUT indication from poll() to show that the connection
|
|
* will accept more write data, but it doesn't specifiy how much. It may just
|
|
* accept one byte of whatever you wanted to send.
|
|
*
|
|
* LWS will buffer the remainder automatically, and send it out autonomously.
|
|
*
|
|
* During that time, WRITABLE callbacks to user code will be suppressed and
|
|
* instead used internally. After it completes, it will send an extra WRITEABLE
|
|
* callback to the user code, in case any request was missed. So it is possible
|
|
* to receive unasked-for WRITEABLE callbacks, the user code should have enough
|
|
* state to know if it wants to write anything and just return if not.
|
|
*
|
|
* This is to handle corner cases where unexpectedly the OS refuses what we
|
|
* usually expect it to accept. It's not recommended as the way to randomly
|
|
* send huge payloads, since it is being copied on to heap and is inefficient.
|
|
*
|
|
* Huge payloads should instead be sent in fragments that are around 2 x mtu,
|
|
* which is almost always directly accepted by the OS. To simplify this for
|
|
* ws fragments, there is a helper lws_write_ws_flags() below that simplifies
|
|
* selecting the correct flags to give lws_write() for each fragment.
|
|
*
|
|
* In the case of RFC8441 ws-over-h2, you cannot send ws fragments larger than
|
|
* the max h2 frame size, typically 16KB, but should further restrict it to
|
|
* the same ~2 x mtu limit mentioned above.
|
|
*/
|
|
LWS_VISIBLE LWS_EXTERN int
|
|
lws_write(struct lws *wsi, unsigned char *buf, size_t len,
|
|
enum lws_write_protocol protocol);
|
|
|
|
/* helper for case where buffer may be const */
|
|
#define lws_write_http(wsi, buf, len) \
|
|
lws_write(wsi, (unsigned char *)(buf), len, LWS_WRITE_HTTP)
|
|
|
|
/**
|
|
* lws_write_ws_flags() - Helper for multi-frame ws message flags
|
|
*
|
|
* \param initial: the lws_write flag to use for the start fragment, eg,
|
|
* LWS_WRITE_TEXT
|
|
* \param is_start: nonzero if this is the first fragment of the message
|
|
* \param is_end: nonzero if this is the last fragment of the message
|
|
*
|
|
* Returns the correct LWS_WRITE_ flag to use for each fragment of a message
|
|
* in turn.
|
|
*/
|
|
static LWS_INLINE int
|
|
lws_write_ws_flags(int initial, int is_start, int is_end)
|
|
{
|
|
int r;
|
|
|
|
if (is_start)
|
|
r = initial;
|
|
else
|
|
r = LWS_WRITE_CONTINUATION;
|
|
|
|
if (!is_end)
|
|
r |= LWS_WRITE_NO_FIN;
|
|
|
|
return r;
|
|
}
|
|
|
|
/**
|
|
* lws_raw_transaction_completed() - Helper for flushing before close
|
|
*
|
|
* \param wsi: the struct lws to operate on
|
|
*
|
|
* Returns -1 if the wsi can close now. However if there is buffered, unsent
|
|
* data, the wsi is marked as to be closed when the output buffer data is
|
|
* drained, and it returns 0.
|
|
*
|
|
* For raw cases where the transaction completed without failure,
|
|
* `return lws_raw_transaction_completed(wsi)` should better be used than
|
|
* return -1.
|
|
*/
|
|
LWS_VISIBLE LWS_EXTERN int LWS_WARN_UNUSED_RESULT
|
|
lws_raw_transaction_completed(struct lws *wsi);
|
|
|
|
///@}
|