/home/mrjunejune/zenbu: seobeo/s

annotate seobeo/s_websocket.c @ 122:96628cf126a0

[MrJuneJune] Websocket blog.

author	June Park <parkjune1995@gmail.com>
date	Thu, 08 Jan 2026 06:45:51 -0800
parents	7b1719fa918c
children	f236c895604e

rev	line source
120 cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	1 #include "seobeo/seobeo.h"
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	2 #include <time.h>
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	3
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	4 #define SEOBEO_WS_GUID "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	5
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	6 static void Seobeo_WebSocket_Parse_Url(const char url, char p_host, char p_port, char p_path, boolean p_use_tls, Dowa_Arena *p_arena)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	7 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	8 if (!url)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	9 return;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	10
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	11 const char *start = url;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	12 *p_use_tls = FALSE;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	13
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	14 if (strncmp(url, "wss://", 6) == 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	15 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	16 *p_use_tls = TRUE;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	17 start = url + 6;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	18 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	19 else if (strncmp(url, "ws://", 5) == 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	20 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	21 *p_use_tls = FALSE;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	22 start = url + 5;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	23 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	24
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	25 const char *slash = strchr(start, '/');
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	26 const char *colon = strchr(start, ':');
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	27
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	28 if (colon && (!slash \|\| colon < slash))
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	29 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	30 size_t host_len = colon - start;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	31 *p_host = Dowa_Arena_Allocate(p_arena, host_len + 1);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	32 memcpy(*p_host, start, host_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	33 (*p_host)[host_len] = '\0';
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	34
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	35 const char *port_start = colon + 1;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	36 size_t port_len = slash ? (slash - port_start) : strlen(port_start);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	37 *p_port = Dowa_Arena_Allocate(p_arena, port_len + 1);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	38 memcpy(*p_port, port_start, port_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	39 (*p_port)[port_len] = '\0';
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	40 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	41 else
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	42 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	43 size_t host_len = slash ? (slash - start) : strlen(start);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	44 *p_host = Dowa_Arena_Allocate(p_arena, host_len + 1);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	45 memcpy(*p_host, start, host_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	46 (*p_host)[host_len] = '\0';
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	47
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	48 *p_port = Dowa_Arena_Allocate(p_arena, 8);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	49 strcpy(p_port, p_use_tls ? "443" : "80");
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	50 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	51
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	52 if (slash)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	53 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	54 size_t path_len = strlen(slash);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	55 *p_path = Dowa_Arena_Allocate(p_arena, path_len + 1);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	56 strcpy(*p_path, slash);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	57 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	58 else
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	59 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	60 *p_path = Dowa_Arena_Allocate(p_arena, 2);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	61 strcpy(*p_path, "/");
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	62 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	63 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	64
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	65 static void Seobeo_WebSocket_Generate_Key(char *key_out, size_t key_size)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	66 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	67 static const char base64_chars[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	68
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	69 srand((unsigned int)time(NULL) ^ (unsigned int)getpid());
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	70
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	71 uint8 random_bytes[16];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	72 for (int i = 0; i < 16; i++)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	73 random_bytes[i] = (uint8)(rand() % 256);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	74
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	75 int out_idx = 0;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	76 for (int i = 0; i < 16; i += 3)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	77 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	78 uint32 triple = (random_bytes[i] << 16) \|
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	79 (i + 1 < 16 ? random_bytes[i + 1] << 8 : 0) \|
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	80 (i + 2 < 16 ? random_bytes[i + 2] : 0);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	81
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	82 key_out[out_idx++] = base64_chars[(triple >> 18) & 0x3F];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	83 key_out[out_idx++] = base64_chars[(triple >> 12) & 0x3F];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	84 key_out[out_idx++] = base64_chars[(triple >> 6) & 0x3F];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	85 key_out[out_idx++] = base64_chars[triple & 0x3F];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	86 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	87
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	88 key_out[out_idx] = '\0';
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	89 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	90
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	91 Seobeo_WebSocket Seobeo_WebSocket_Connect(const char url)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	92 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	93 Seobeo_WebSocket *p_ws = malloc(sizeof(Seobeo_WebSocket));
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	94 if (!p_ws)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	95 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	96
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	97 memset(p_ws, 0, sizeof(Seobeo_WebSocket));
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	98
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	99 p_ws->p_arena = Dowa_Arena_Create(1024 * 1024);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	100 if (!p_ws->p_arena)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	101 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	102 free(p_ws);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	103 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	104 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	105
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	106 size_t url_len = strlen(url);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	107 p_ws->url = Dowa_Arena_Allocate(p_ws->p_arena, url_len + 1);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	108 strcpy(p_ws->url, url);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	109
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	110 Seobeo_WebSocket_Parse_Url(url, &p_ws->host, &p_ws->port, &p_ws->path, &p_ws->use_tls, p_ws->p_arena);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	111
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	112 p_ws->p_handle = Seobeo_Stream_Handle_Client_Create(p_ws->host, p_ws->port, p_ws->use_tls);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	113 if (!p_ws->p_handle \|\| p_ws->p_handle->socket < 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	114 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	115 Seobeo_Log(SEOBEO_ERROR, "Failed to create socket connection\n");
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	116 if (p_ws->p_handle)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	117 Seobeo_Handle_Destroy(p_ws->p_handle);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	118 Dowa_Arena_Free(p_ws->p_arena);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	119 free(p_ws);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	120 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	121 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	122
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	123 char ws_key[32];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	124 Seobeo_WebSocket_Generate_Key(ws_key, sizeof(ws_key));
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	125
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	126 char handshake[2048];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	127 int handshake_len = snprintf(handshake, sizeof(handshake),
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	128 "GET %s HTTP/1.1\r\n"
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	129 "Host: %s\r\n"
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	130 "Upgrade: websocket\r\n"
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	131 "Connection: Upgrade\r\n"
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	132 "Sec-WebSocket-Key: %s\r\n"
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	133 "Sec-WebSocket-Version: 13\r\n"
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	134 "\r\n",
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	135 p_ws->path, p_ws->host, ws_key);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	136
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	137 Seobeo_Handle_Queue(p_ws->p_handle, (uint8*)handshake, (uint32)handshake_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	138 if (Seobeo_Handle_Flush(p_ws->p_handle) < 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	139 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	140 Seobeo_Log(SEOBEO_ERROR, "Failed to send WebSocket handshake\n");
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	141 Seobeo_Handle_Destroy(p_ws->p_handle);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	142 Dowa_Arena_Free(p_ws->p_arena);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	143 free(p_ws);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	144 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	145 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	146
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	147 while (1)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	148 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	149 int r = Seobeo_Handle_Read(p_ws->p_handle);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	150 if (r < 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	151 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	152 Seobeo_Log(SEOBEO_ERROR, "Failed to read handshake response\n");
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	153 Seobeo_Handle_Destroy(p_ws->p_handle);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	154 Dowa_Arena_Free(p_ws->p_arena);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	155 free(p_ws);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	156 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	157 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	158
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	159 if (p_ws->p_handle->read_buffer_len >= 4 &&
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	160 strstr((char*)p_ws->p_handle->read_buffer, "\r\n\r\n") != NULL)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	161 break;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	162
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	163 if (r == 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	164 continue;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	165 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	166
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	167 char response = (char)p_ws->p_handle->read_buffer;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	168 if (strstr(response, "HTTP/1.1 101") == NULL)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	169 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	170 Seobeo_Log(SEOBEO_ERROR, "WebSocket handshake failed: %s\n", response);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	171 Seobeo_Handle_Destroy(p_ws->p_handle);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	172 Dowa_Arena_Free(p_ws->p_arena);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	173 free(p_ws);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	174 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	175 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	176
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	177 char *end_of_headers = strstr(response, "\r\n\r\n");
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	178 if (end_of_headers)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	179 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	180 size_t header_len = end_of_headers - response + 4;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	181 Seobeo_Handle_Consume(p_ws->p_handle, (uint32)header_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	182 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	183
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	184 p_ws->state = SEOBEO_WS_STATE_OPEN;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	185 p_ws->fragment_capacity = 4096;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	186 p_ws->fragment_buffer = malloc(p_ws->fragment_capacity);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	187
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	188 Seobeo_Log(SEOBEO_INFO, "WebSocket connected to %s\n", url);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	189
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	190 return p_ws;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	191 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	192
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	193 static void Seobeo_WebSocket_Mask_Data(uint8 data, size_t length, const uint8 mask)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	194 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	195 for (size_t i = 0; i < length; i++)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	196 data[i] ^= mask[i % 4];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	197 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	198
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	199 static int32 Seobeo_WebSocket_Send_Frame(Seobeo_WebSocket *p_ws, Seobeo_WebSocket_Opcode opcode,
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	200 const uint8 *payload, size_t payload_length, boolean fin)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	201 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	202 if (!p_ws \|\| p_ws->state != SEOBEO_WS_STATE_OPEN)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	203 return -1;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	204
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	205 uint8 frame[14];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	206 size_t frame_len = 0;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	207
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	208 // Big endian
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	209 frame[0] = (fin ? 0x80 : 0x00) \| (opcode & 0x0F);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	210 frame_len++;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	211
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	212 uint8 mask_key[4];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	213 for (int i = 0; i < 4; i++)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	214 mask_key[i] = (uint8)(rand() % 256);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	215
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	216 // within 1 byte
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	217 if (payload_length < 126)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	218 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	219 frame[1] = 0x80 \| (uint8)payload_length;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	220 frame_len++;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	221 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	222 // from now frame 1 is thrown away just keeping it 1
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	223 // within 4 bytes
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	224 else if (payload_length < MAX_INT_16)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	225 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	226 frame[1] = 0x80 \| 126;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	227 frame[2] = (uint8)((payload_length >> 8) & 0xFF);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	228 frame[3] = (uint8)(payload_length & 0xFF);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	229 frame_len += 3;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	230 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	231 // within 8 bytes
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	232 else
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	233 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	234 frame[1] = 0x80 \| 127;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	235 for (int i = 0; i < 8; i++)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	236 frame[2 + i] = (uint8)((payload_length >> (56 - i * 8)) & 0xFF);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	237 frame_len += 9;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	238 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	239
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	240 memcpy(frame + frame_len, mask_key, 4);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	241 frame_len += 4;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	242
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	243 Seobeo_Handle_Queue(p_ws->p_handle, frame, (uint32)frame_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	244
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	245 if (payload_length > 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	246 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	247 uint8 *masked_payload = malloc(payload_length);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	248 memcpy(masked_payload, payload, payload_length);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	249 Seobeo_WebSocket_Mask_Data(masked_payload, payload_length, mask_key);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	250 Seobeo_Handle_Queue(p_ws->p_handle, masked_payload, (uint32)payload_length);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	251 free(masked_payload);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	252 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	253
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	254 return Seobeo_Handle_Flush(p_ws->p_handle);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	255 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	256
121 7b1719fa918c [Seobeo] Added web socket server. June Park <parkjune1995@gmail.com> parents: 120 diff changeset	257 static int32 Seobeo_WebSocket_Send_Fragmented(Seobeo_WebSocket *p_ws, Seobeo_WebSocket_Opcode opcode,
7b1719fa918c [Seobeo] Added web socket server. June Park <parkjune1995@gmail.com> parents: 120 diff changeset	258 const uint8 *payload, size_t total_length)
120 cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	259 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	260 if (!payload \|\| total_length == 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	261 return -1;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	262
121 7b1719fa918c [Seobeo] Added web socket server. June Park <parkjune1995@gmail.com> parents: 120 diff changeset	263 if (total_length <= MAX_FRAGMENT_SIZE)
120 cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	264 return Seobeo_WebSocket_Send_Frame(p_ws, opcode, payload, total_length, TRUE);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	265
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	266 size_t sent = 0;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	267 int32 result;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	268
121 7b1719fa918c [Seobeo] Added web socket server. June Park <parkjune1995@gmail.com> parents: 120 diff changeset	269 result = Seobeo_WebSocket_Send_Frame(p_ws, opcode, payload, MAX_FRAGMENT_SIZE, FALSE);
120 cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	270 if (result < 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	271 return result;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	272
121 7b1719fa918c [Seobeo] Added web socket server. June Park <parkjune1995@gmail.com> parents: 120 diff changeset	273 sent += MAX_FRAGMENT_SIZE;
120 cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	274
121 7b1719fa918c [Seobeo] Added web socket server. June Park <parkjune1995@gmail.com> parents: 120 diff changeset	275 while (sent + MAX_FRAGMENT_SIZE < total_length)
120 cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	276 {
121 7b1719fa918c [Seobeo] Added web socket server. June Park <parkjune1995@gmail.com> parents: 120 diff changeset	277 result = Seobeo_WebSocket_Send_Frame(p_ws, SEOBEO_WS_OPCODE_CONTINUATION, payload + sent, MAX_FRAGMENT_SIZE, FALSE);
120 cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	278 if (result < 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	279 return result;
121 7b1719fa918c [Seobeo] Added web socket server. June Park <parkjune1995@gmail.com> parents: 120 diff changeset	280 sent += MAX_FRAGMENT_SIZE;
120 cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	281 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	282
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	283 size_t remaining = total_length - sent;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	284 return Seobeo_WebSocket_Send_Frame(p_ws, SEOBEO_WS_OPCODE_CONTINUATION, payload + sent, remaining, TRUE);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	285 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	286
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	287 int32 Seobeo_WebSocket_Send_Text(Seobeo_WebSocket p_ws, const char text)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	288 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	289 if (!text)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	290 return -1;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	291
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	292 return Seobeo_WebSocket_Send_Fragmented(p_ws, SEOBEO_WS_OPCODE_TEXT, (const uint8*)text, strlen(text));
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	293 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	294
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	295 int32 Seobeo_WebSocket_Send_Binary(Seobeo_WebSocket p_ws, const uint8 data, size_t length)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	296 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	297 if (!data)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	298 return -1;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	299
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	300 return Seobeo_WebSocket_Send_Fragmented(p_ws, SEOBEO_WS_OPCODE_BINARY, data, length);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	301 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	302
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	303 int32 Seobeo_WebSocket_Send_Ping(Seobeo_WebSocket p_ws, const char payload)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	304 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	305 size_t len = payload ? strlen(payload) : 0;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	306 return Seobeo_WebSocket_Send_Frame(p_ws, SEOBEO_WS_OPCODE_PING, (const uint8*)payload, len, TRUE);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	307 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	308
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	309 int32 Seobeo_WebSocket_Send_Pong(Seobeo_WebSocket p_ws, const char payload)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	310 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	311 size_t len = payload ? strlen(payload) : 0;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	312 return Seobeo_WebSocket_Send_Frame(p_ws, SEOBEO_WS_OPCODE_PONG, (const uint8*)payload, len, TRUE);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	313 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	314
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	315 Seobeo_WebSocket_Message Seobeo_WebSocket_Receive(Seobeo_WebSocket p_ws)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	316 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	317 if (!p_ws \|\| p_ws->state == SEOBEO_WS_STATE_CLOSED)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	318 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	319
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	320 int r = Seobeo_Handle_Read(p_ws->p_handle);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	321 if (r < 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	322 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	323 Seobeo_Log(SEOBEO_ERROR, "WebSocket read error\n");
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	324 p_ws->state = SEOBEO_WS_STATE_CLOSED;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	325 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	326 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	327
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	328 if (r == -2)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	329 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	330 Seobeo_Log(SEOBEO_INFO, "WebSocket connection closed\n");
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	331 p_ws->state = SEOBEO_WS_STATE_CLOSED;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	332 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	333 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	334
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	335 if (p_ws->p_handle->read_buffer_len < 2)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	336 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	337
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	338 uint8 *buf = p_ws->p_handle->read_buffer;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	339
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	340 uint8 byte1 = buf[0];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	341 uint8 byte2 = buf[1];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	342
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	343 boolean fin = (byte1 & 0x80) != 0;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	344 Seobeo_WebSocket_Opcode opcode = (Seobeo_WebSocket_Opcode)(byte1 & 0x0F);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	345 boolean masked = (byte2 & 0x80) != 0;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	346 uint64 payload_len = byte2 & 0x7F;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	347
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	348 size_t header_len = 2;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	349
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	350 if (payload_len == 126)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	351 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	352 if (p_ws->p_handle->read_buffer_len < 4)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	353 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	354 payload_len = (buf[2] << 8) \| buf[3];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	355 header_len += 2;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	356 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	357 else if (payload_len == 127)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	358 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	359 if (p_ws->p_handle->read_buffer_len < 10)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	360 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	361 payload_len = 0;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	362 for (int i = 0; i < 8; i++)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	363 payload_len = (payload_len << 8) \| buf[2 + i];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	364 header_len += 8;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	365 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	366
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	367 uint8 mask_key[4] = {0};
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	368 if (masked)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	369 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	370 if (p_ws->p_handle->read_buffer_len < header_len + 4)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	371 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	372 memcpy(mask_key, buf + header_len, 4);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	373 header_len += 4;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	374 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	375
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	376 if (p_ws->p_handle->read_buffer_len < header_len + payload_len)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	377 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	378
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	379 uint8 *payload = NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	380 if (payload_len > 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	381 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	382 payload = malloc(payload_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	383 memcpy(payload, buf + header_len, payload_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	384
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	385 if (masked)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	386 Seobeo_WebSocket_Mask_Data(payload, payload_len, mask_key);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	387 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	388
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	389 Seobeo_Handle_Consume(p_ws->p_handle, (uint32)(header_len + payload_len));
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	390
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	391 if (opcode == SEOBEO_WS_OPCODE_PING)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	392 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	393 Seobeo_WebSocket_Send_Pong(p_ws, (char*)payload);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	394 if (payload)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	395 free(payload);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	396 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	397 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	398
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	399 if (opcode == SEOBEO_WS_OPCODE_PONG)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	400 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	401 if (payload)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	402 free(payload);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	403 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	404 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	405
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	406 if (opcode == SEOBEO_WS_OPCODE_CLOSE)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	407 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	408 uint16 close_code = 1000;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	409 if (payload_len >= 2)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	410 close_code = (payload[0] << 8) \| payload[1];
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	411
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	412 Seobeo_Log(SEOBEO_INFO, "WebSocket close received with code %d\n", close_code);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	413 p_ws->state = SEOBEO_WS_STATE_CLOSING;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	414
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	415 Seobeo_WebSocket_Send_Frame(p_ws, SEOBEO_WS_OPCODE_CLOSE, payload, payload_len, TRUE);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	416 p_ws->state = SEOBEO_WS_STATE_CLOSED;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	417
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	418 if (payload)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	419 free(payload);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	420 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	421 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	422
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	423 if (opcode == SEOBEO_WS_OPCODE_CONTINUATION)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	424 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	425 if (p_ws->fragment_length + payload_len > p_ws->fragment_capacity)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	426 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	427 p_ws->fragment_capacity = (p_ws->fragment_length + payload_len) * 2;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	428 p_ws->fragment_buffer = realloc(p_ws->fragment_buffer, p_ws->fragment_capacity);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	429 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	430
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	431 if (payload_len > 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	432 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	433 memcpy(p_ws->fragment_buffer + p_ws->fragment_length, payload, payload_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	434 p_ws->fragment_length += payload_len;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	435 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	436
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	437 if (payload)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	438 free(payload);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	439
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	440 if (!fin)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	441 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	442
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	443 Seobeo_WebSocket_Message *p_msg = malloc(sizeof(Seobeo_WebSocket_Message));
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	444 p_msg->opcode = p_ws->fragment_opcode;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	445 p_msg->data = malloc(p_ws->fragment_length);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	446 memcpy(p_msg->data, p_ws->fragment_buffer, p_ws->fragment_length);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	447 p_msg->length = p_ws->fragment_length;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	448 p_msg->is_final = TRUE;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	449
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	450 p_ws->fragment_length = 0;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	451
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	452 return p_msg;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	453 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	454
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	455 if (!fin)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	456 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	457 p_ws->fragment_opcode = opcode;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	458 p_ws->fragment_length = 0;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	459
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	460 if (payload_len > 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	461 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	462 if (payload_len > p_ws->fragment_capacity)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	463 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	464 p_ws->fragment_capacity = payload_len * 2;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	465 p_ws->fragment_buffer = realloc(p_ws->fragment_buffer, p_ws->fragment_capacity);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	466 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	467
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	468 memcpy(p_ws->fragment_buffer, payload, payload_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	469 p_ws->fragment_length = payload_len;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	470 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	471
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	472 if (payload)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	473 free(payload);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	474
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	475 return NULL;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	476 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	477
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	478 Seobeo_WebSocket_Message *p_msg = malloc(sizeof(Seobeo_WebSocket_Message));
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	479 p_msg->opcode = opcode;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	480 p_msg->data = payload;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	481 p_msg->length = payload_len;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	482 p_msg->is_final = fin;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	483
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	484 return p_msg;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	485 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	486
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	487 void Seobeo_WebSocket_Message_Destroy(Seobeo_WebSocket_Message *p_msg)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	488 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	489 if (!p_msg)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	490 return;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	491
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	492 if (p_msg->data)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	493 free(p_msg->data);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	494
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	495 free(p_msg);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	496 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	497
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	498 int32 Seobeo_WebSocket_Close(Seobeo_WebSocket p_ws, uint16 code, const char reason)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	499 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	500 if (!p_ws \|\| p_ws->state == SEOBEO_WS_STATE_CLOSED)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	501 return -1;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	502
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	503 p_ws->state = SEOBEO_WS_STATE_CLOSING;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	504
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	505 size_t reason_len = reason ? strlen(reason) : 0;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	506 size_t payload_len = 2 + reason_len;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	507 uint8 *payload = malloc(payload_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	508
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	509 payload[0] = (uint8)((code >> 8) & 0xFF);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	510 payload[1] = (uint8)(code & 0xFF);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	511
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	512 if (reason_len > 0)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	513 memcpy(payload + 2, reason, reason_len);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	514
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	515 int32 result = Seobeo_WebSocket_Send_Frame(p_ws, SEOBEO_WS_OPCODE_CLOSE, payload, payload_len, TRUE);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	516
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	517 free(payload);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	518
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	519 p_ws->state = SEOBEO_WS_STATE_CLOSED;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	520
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	521 return result;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	522 }
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	523
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	524 void Seobeo_WebSocket_Destroy(Seobeo_WebSocket *p_ws)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	525 {
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	526 if (!p_ws)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	527 return;
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	528
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	529 if (p_ws->state == SEOBEO_WS_STATE_OPEN)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	530 Seobeo_WebSocket_Close(p_ws, 1000, "Normal closure");
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	531
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	532 if (p_ws->p_handle)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	533 Seobeo_Handle_Destroy(p_ws->p_handle);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	534
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	535 if (p_ws->fragment_buffer)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	536 free(p_ws->fragment_buffer);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	537
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	538 if (p_ws->p_arena)
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	539 Dowa_Arena_Free(p_ws->p_arena);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	540
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	541 free(p_ws);
cbbf78b17cfa [Seobeo][Websocket] Created Web socket client logic. June Park <parkjune1995@gmail.com> parents: diff changeset	542 }

Mercurial

annotate seobeo/s_websocket.c @ 122:96628cf126a0