golden hour

/opt/alt/php71/usr/include/php/ext/swoole/include

File/Folder	Size	Actions
swoole.h	19.74 KB
swoole_api.h	2.76 KB
swoole_asm_context.h	1.47 KB
swoole_async.h	2.13 KB
swoole_atomic.h	2.24 KB
swoole_base64.h	1.29 KB
swoole_buffer.h	2.02 KB
swoole_c_api.h	1.89 KB
swoole_channel.h	2.36 KB
swoole_client.h	7.85 KB
swoole_config.h	9.26 KB
swoole_coroutine.h	6.25 KB
swoole_coroutine_c_api.h	3.7 KB
swoole_coroutine_channel.h	4.04 KB
swoole_coroutine_context.h	2.39 KB
swoole_coroutine_socket.h	17.16 KB
swoole_coroutine_system.h	3.05 KB
swoole_dtls.h	2.38 KB
swoole_error.h	4.72 KB
swoole_file.h	4.04 KB
swoole_file_hook.h	1.95 KB
swoole_hash.h	9.69 KB
swoole_heap.h	1.79 KB
swoole_http.h	4.63 KB
swoole_http2.h	8 KB
swoole_lock.h	2.62 KB
swoole_log.h	15.53 KB
swoole_lru_cache.h	3.2 KB
swoole_memory.h	2.57 KB
swoole_mime_type.h	1.44 KB
swoole_mqtt.h	2.11 KB
swoole_msg_queue.h	1.84 KB
swoole_pipe.h	2.64 KB
swoole_process_pool.h	5.82 KB
swoole_protocol.h	5 KB
swoole_proxy.h	2.07 KB
swoole_reactor.h	10.47 KB
swoole_redis.h	2.18 KB
swoole_server.h	32.83 KB
swoole_signal.h	1.55 KB
swoole_socket.h	15.48 KB
swoole_socket_hook.h	2.12 KB
swoole_ssl.h	3.58 KB
swoole_static_handler.h	2.8 KB
swoole_string.h	6.87 KB
swoole_table.h	5.74 KB
swoole_timer.h	4.08 KB
swoole_util.h	3.59 KB
swoole_version.h	2.21 KB
swoole_websocket.h	5.07 KB
swoole_wheel_timer.h	2.48 KB

Edit: swoole_websocket.h

/*
  +----------------------------------------------------------------------+
  | Swoole                                                               |
  +----------------------------------------------------------------------+
  | This source file is subject to version 2.0 of the Apache license,    |
  | that is bundled with this package in the file LICENSE, and is        |
  | available through the world-wide-web at the following url:           |
  | http://www.apache.org/licenses/LICENSE-2.0.html                      |
  | If you did not receive a copy of the Apache2.0 license and are unable|
  | to obtain it through the world-wide-web, please send a note to       |
  | license@php.net so we can mail you a copy immediately.               |
  +----------------------------------------------------------------------+
  | Author: Tianfeng Han  <mikan.tenny@gmail.com>                        |
  +----------------------------------------------------------------------+
*/

#pragma once

#include "swoole_http.h"

#define SW_WEBSOCKET_SEC_KEY_LEN 16
#define SW_WEBSOCKET_GUID "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"
#define SW_WEBSOCKET_HEADER_LEN 2
#define SW_WEBSOCKET_MASK_LEN 4
#define SW_WEBSOCKET_MASK_DATA "258E"
#define SW_WEBSOCKET_EXT16_MAX_LEN 0xFFFF
#define SW_WEBSOCKET_EXT16_LENGTH 0x7E
#define SW_WEBSOCKET_EXT64_LENGTH 0x7F
#define SW_WEBSOCKET_CLOSE_CODE_LEN 2
#define SW_WEBSOCKET_CLOSE_REASON_MAX_LEN 125
#define SW_WEBSOCKET_OPCODE_MAX WEBSOCKET_OPCODE_PONG

enum swWebsocket_status {
    WEBSOCKET_STATUS_NONE = 0,
    WEBSOCKET_STATUS_CONNECTION = 1,
    WEBSOCKET_STATUS_HANDSHAKE = 2,
    WEBSOCKET_STATUS_ACTIVE = 3,
    WEBSOCKET_STATUS_CLOSING = 4,
};

enum swWebSocket_frame_flag {
    SW_WEBSOCKET_FLAG_FIN = 1 << 0, /* BC: must be 1 */
    SW_WEBSOCKET_FLAG_COMPRESS = 1 << 1,
    // readonly for user
    SW_WEBSOCKET_FLAG_RSV1 = 1 << 2,
    SW_WEBSOCKET_FLAG_RSV2 = 1 << 3,
    SW_WEBSOCKET_FLAG_RSV3 = 1 << 4,
    SW_WEBSOCKET_FLAG_MASK = 1 << 5,
    // for encoder/decoder
    SW_WEBSOCKET_FLAG_ENCODE_HEADER_ONLY = 1 << 6,
};

enum swWebSocket_frame_union_flag {
    SW_WEBSOCKET_FLAGS_ALL = /* used to prevent overflow  */
    SW_WEBSOCKET_FLAG_FIN | SW_WEBSOCKET_FLAG_RSV1 | SW_WEBSOCKET_FLAG_RSV2 | SW_WEBSOCKET_FLAG_RSV3 |
    SW_WEBSOCKET_FLAG_MASK | SW_WEBSOCKET_FLAG_COMPRESS
};

struct swWebSocket_frame_header {
    /**
     * fin:1 rsv1:1 rsv2:1 rsv3:1 opcode:4
     */
    uchar OPCODE : 4;
    uchar RSV3 : 1;
    uchar RSV2 : 1;
    uchar RSV1 : 1;
    uchar FIN : 1;
    uchar LENGTH : 7;
    uchar MASK : 1;
};

struct swWebSocket_frame {
    swWebSocket_frame_header header;
    char mask_key[SW_WEBSOCKET_MASK_LEN];
    uint16_t header_length;
    size_t payload_length;
    char *payload;
};

#define WEBSOCKET_VERSION 13

enum swWebsocket_opcode {
    WEBSOCKET_OPCODE_CONTINUATION = 0x0,
    WEBSOCKET_OPCODE_TEXT = 0x1,
    WEBSOCKET_OPCODE_BINARY = 0x2,
    WEBSOCKET_OPCODE_CLOSE = 0x8,
    WEBSOCKET_OPCODE_PING = 0x9,
    WEBSOCKET_OPCODE_PONG = 0xa,
};

enum swWebsocket_close_reason {
    WEBSOCKET_CLOSE_NORMAL = 1000,
    WEBSOCKET_CLOSE_GOING_AWAY = 1001,
    WEBSOCKET_CLOSE_PROTOCOL_ERROR = 1002,
    WEBSOCKET_CLOSE_DATA_ERROR = 1003,
    WEBSOCKET_CLOSE_STATUS_ERROR = 1005,
    WEBSOCKET_CLOSE_ABNORMAL = 1006,
    WEBSOCKET_CLOSE_MESSAGE_ERROR = 1007,
    WEBSOCKET_CLOSE_POLICY_ERROR = 1008,
    WEBSOCKET_CLOSE_MESSAGE_TOO_BIG = 1009,
    WEBSOCKET_CLOSE_EXTENSION_MISSING = 1010,
    WEBSOCKET_CLOSE_SERVER_ERROR = 1011,
    WEBSOCKET_CLOSE_TLS = 1015,
};

static inline uchar swWebSocket_get_flags(swWebSocket_frame *frame) {
    uchar flags = 0;
    if (frame->header.FIN) {
        flags |= SW_WEBSOCKET_FLAG_FIN;
    }
    if (frame->header.RSV1) {
        flags |= SW_WEBSOCKET_FLAG_RSV1;
    }
    if (frame->header.RSV2) {
        flags |= SW_WEBSOCKET_FLAG_RSV2;
    }
    if (frame->header.RSV3) {
        flags |= SW_WEBSOCKET_FLAG_RSV3;
    }
    if (frame->header.MASK) {
        flags |= SW_WEBSOCKET_FLAG_MASK;
    }
    return flags;
}

static inline uchar swWebSocket_set_flags(uchar fin, uchar mask, uchar rsv1, uchar rsv2, uchar rsv3) {
    uchar flags = 0;
    if (fin) {
        flags |= SW_WEBSOCKET_FLAG_FIN;
    }
    if (mask) {
        flags |= SW_WEBSOCKET_FLAG_MASK;
    }
    if (rsv1) {
        flags |= SW_WEBSOCKET_FLAG_RSV1;
    }
    if (rsv2) {
        flags |= SW_WEBSOCKET_FLAG_RSV2;
    }
    if (rsv3) {
        flags |= SW_WEBSOCKET_FLAG_RSV3;
    }
    return flags;
}

bool swWebSocket_encode(swString *buffer, const char *data, size_t length, char opcode, uint8_t flags);
bool swWebSocket_decode(swWebSocket_frame *frame, char *data, size_t length);
int swWebSocket_pack_close_frame(swString *buffer, int code, char *reason, size_t length, uint8_t flags);
void swWebSocket_print_frame(swWebSocket_frame *frame);

inline bool swWebSocket_decode(swWebSocket_frame *frame, swString *str) {
    return swWebSocket_decode(frame, str->str, str->length);
}

ssize_t swWebSocket_get_package_length(swProtocol *protocol, swSocket *conn, const char *data, uint32_t length);
int swWebSocket_dispatch_frame(swProtocol *protocol, swSocket *conn, const char *data, uint32_t length);