您的位置:首页 > 编程语言 > C语言/C++

Visual Studio 2019使用C语言进行websocket编程

2021-03-03 22:26 1521 查看

一直在写C#代码好多年不写C语言代码了,记录一下之前某个项目里用C写的一个websocket服务,用C的优势是写的东西体积小性能高,但是写业务的话还得用C#、Java之类的语言,不然会折腾死人。。。

 

用Visual Studio新建一个C++(因为不能直接建C语言项目)项目,我演示就创建一个控制台项目。项目创建完后首先要添加socket编程需要的依赖库ws2_32.lib,添加方式如下图

也可以在代码文件里添加这句代码:#pragma comment(lib,"Ws2_32.lib"),接着添加b64、cJSON、sha1依赖库。

添加完成后就可以开始写代码了,说句题外话,Visual Studio写C语言最好把SDL检查也关掉。

新建一个wsserver.h头文件,头文件相关定义代码如下

#pragma once
#include <WinSock2.h>
#include <stdint.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include <windef.h>
#include <stdlib.h>
#include "sha1.h"
#include "b64.h"
#include "cJSON.h"

typedef enum FrameType
{
frameType_continuation,
frameType_text,
frameType_binary,
frameType_connectionClose,
frameType_ping,
frameType_pong

ad8
} FrameType;

typedef enum FrameState
{
frameState_init,            // 未读取任何字节
frameState_firstByte,       // 已读取首字节FIN、RSV、opcode
frameState_mask,            // 已读取掩码
frameState_7bitLength,      // 已读取7bit长度
frameState_16bitLengthWait, // 等待读取16bit长度
frameState_63bitLengthWait, // 等待读取63bit长度
frameState_16bitLength,     // 已读取16bit长度
frameState_63bitLength,        // 已读取63bit长度
frameState_maskingKey,      // 已读取Masking-key
frameState_readingData,     // 正在读取载荷数据
frameState_success,         // 读取完毕
frameState_failure          // 读取错误
} FrameState;

typedef struct WsFrame
{
FrameState state;
bool FIN;
FrameType frameType;
uint8_t  mask[4];
unsigned char* buff;   // 数据存放的空间
uint64_t buffSize;      // 当前申请的buff大小
uint64_t handledLen;    // 已处理的帧长度
uint64_t headerLen;     // 帧头长度 只有在state为'已读取掩码'及之后才有意义
uint64_t payloadLen;    // 载荷长度 只有在state为'已读取xbit长度'后才有意义
struct WsFrame* next;   // 下一帧的指针
} WsFrame;

void initWsFrameStruct(WsFrame* wsFrame);
char* convertToWebSocketFrame(const char* data, FrameType type, size_t len, size_t* newLen);
int readWebSocketFrameStream(WsFrame* wsFrame, const char* buff, int len);
void freeWebSocketFrame(WsFrame* wsFrame);
int wsShakeHands(const char* recvBuff, int recvLen, SOCKET socket, const char* path);
int wsFrameSend(SOCKET socket, const char* buff, int len, FrameType type);
void wsFrameSendToAll(const char* buff, int len, FrameType type);
int serverStart(const char* address, u_short port, const char* path);
void serverStop(void);

 

新建一个wsserver.c代码文件,我们一步一步的来实现这些方法。

首先是serverStart方法,顾名思义,启动ws服务,第一个参数是地址(一般传本机IP),第二个参数是要监听的端口,第三个参数是路径,完整代码如下

#include "wsserver.h"
#include <time.h>
#define _CRT_SECURE_NO_WARNINGS
#define _WINSOCK_DEPRECATED_NO_WARNINGS

typedef enum
{
socketProtocol,
websocketProtocol
} Protocol;

typedef struct
{
Protocol protocol;
SOCKET socket;
WsFrame wsFrame;
} Client;

#define MAX_CLIENT_NUM FD_SETSIZE
static struct
{
int    total;
Client clients[MAX_CLIENT_NUM];
} clientSockets;

static SOCKET serverSocket;

// 打印日志
void printLog(const char* type, const char* format, ...)
{
char buff[512] = { 0 };
va_list arg;
va_start(arg, format);
vsnprintf(buff, sizeof(buff) - 1, format, arg);
va_end(arg);
char rbuf[512] = { 0 };
time_t log_time = time(NULL);
struct tm* tm_log = localtime(&log_time);
printf("[%04d-%02d-%02d %02d:%02d:%02d] ", tm_log->tm_year + 1900, tm_log->tm_mon + 1, tm_log->tm_mday, tm_log->tm_hour, tm_log->tm_min, tm_log->tm_sec);
snprintf(rbuf, 512, "%s->%s\n", type, buff);
printf(rbuf);
}
char* UTF8ToGBK(const char* str)
{
// GB18030代码页
const int CODE_PAGE = 54936;
int n = MultiByteToWideChar(CP_UTF8, 0, str, -1, NULL, 0);
wchar_t u16str[10000];
MultiByteToWideChar(CP_UTF8, 0, str, -1, u16str, n);

n = WideCharToMultiByte(CODE_PAGE, 0, u16str, -1, NULL, 0, NULL, NULL);
char* gbstr = malloc(n + 1);
WideCharToMultiByte(CODE_PAGE, 0, u16str, -1, gbstr, n, NULL, NULL);
return gbstr;
}
char* GBKToUTF8(const char* str)
{
const int CODE_PAGE = 54936;
int n = MultiByteToWideChar(CODE_PAGE, 0, str, -1, NULL, 0);
wchar_t u16str[10000];
MultiByteToWideChar(CODE_PAGE, 0, str, -1, u16str, n);

n = WideCharToMultiByte(CP_UTF8, 0, u16str, -1, NULL, 0, NULL, NULL);
char* u8str = malloc(n + 1);
WideCharToMultiByte(CP_UTF8, 0, u16str, -1, u8str, n, NULL, NULL);
return u8str;
}

int serverStart(const char* address, u_short port, const char* path)
{
// 调用 WSAStartup() 函数进行初始化,并指明要使用的版本号。
WSADATA wsaData;
// WSAStartup 函数启动进程使用 Winsock DLL。
int iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
if (iResult != 0)
{
printLog("ServerStart", "WSAStartup failed");
return -1;
}

struct sockaddr_in sockAddr;
// ZeroMemory 宏 等价于 memset((buf),0,(BUF_SIZE))
ZeroMemory(&sockAddr, sizeof(sockAddr));
sockAddr.sin_family = PF_INET; // 等价于 AF_INET  TCP UDP etc..
sockAddr.sin_addr.s_addr = inet_addr(address);
sockAddr.sin_port = htons(port);
// 构建一个socket对象
serverSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);

if (serverSocket == INVALID_SOCKET)
{
printLog("ServerStart","Error at socket(): %d", WSAGetLastError());
WSACleanup();
return -1;
}
// 给socket绑定地址
if (bind(serverSocket, (SOCKADDR*)&sockAddr, sizeof(SOCKADDR)) == SOCKET_ERROR)
{
printLog("ServerStart", "Bind failed with error: %d", WSAGetLastError());
closesocket(serverSocket);
WSACleanup();
return -1;
}
// 开始启动监听
if (listen(serverSocket, SOMAXCONN) == SOCKET_ERROR)
{
printLog("ServerStart", "Listen failed with error: %d", WSAGetLastError());
closesocket(serverSocket);
WSACleanup();
return -1;
}

clientSockets.total = 0;

DWORD dwThreadId;
// 创建线程开始接收socket数据
HANDLE hHandle = CreateThread(NULL, 0, (void*)receiveComingData, (PVOID)path, 0, &dwThreadId);

return 0;
}

serverStart方法中最后创建线程开始接收socket数据的方法receiveComingData代码

void receiveComingData(const char* path)
{

#define RECV_BUFLEN 0X40000
char recvbuf[RECV_BUFLEN];
int iResult;

int ret;
fd_set fdread;
struct timeval tv = { 1, 0 };

receivingDataLoop:

FD_ZERO(&fdread); // 清空socket集合
FD_SET(serverSocket, &fdread); // 设置socket数据读取集合
for (int i = 0; i < clientSockets.total; i++)
{
FD_SET(clientSockets.clients[i].socket, &fdread);
}
// 检查socket是否有数据可读
ret = select(0, &fdread, NULL, NULL, &tv);

if (ret == 0)
{
goto receivingDataLoop;     // select的等待时间到达,开始下一轮等待
}
// 检查socket是否在这个集合里
if (FD_ISSET(serverSocket, &fdread))
{
acceptConnect(); // 处理socket连接
}

for (int i = 0; i < clientSockets.total; i++)
{
Client* client = &clientSockets.clients[i];
if (!FD_ISSET(client->socket, &fdread))
{
continue;
}
// 接收数据
iResult = recv(client->socket, recvbuf, RECV_BUFLEN, 0);

if (iResult > 0)
{
printLog("receiveComingData", "Bytes received: %d", iResult);
// 协议升级
if (client->protocol == socketProtocol)
{
int result = wsShakeHands(recvbuf, iResult, client->socket, path);
if (result != 0)
{
removeClient(i--);
}
else
{
client->protocol = websocketProtocol;
initWsFrameStruct(&client->wsFrame); // 初始化ws帧结构
}
}
// WebSocket通信
else if (client->protocol == websocketProtocol)
{
int result = wsClientDataHandle(recvbuf, iResult, client);
if (result == -1)
{
removeClient(i--);
}
}
}
else
{
if (iResult == 0)
{
// 客户端礼貌的关闭连接
printLog("receiveComingData", "Connection closing...");
}
else
{
// 客户端异常关闭连接等情况
printLog("receiveComingData", "Recv failed: %d", WSAGetLastError());
}
removeClient(i--);
}
}
goto receivingDataLoop;
}

receiveComingData方法里处理socket协议升级的代码

// 不区分大小写的比较字符串,相等返回true
bool stricasecmp(const char* a, const char* b)
{
do
{
if (*a == '\0' && *b == '\0')
return true;
} while (tolower(*a++) == tolower(*b++));

return false;
}

// 不区分大小写的比较字符串,n个字符内(包括n)相等返回true
bool strnicasecmp(const char* a, const char* b, unsigned n)
{
do
{
if (n-- == 0 || (*a == '\0' && *b == '\0'))
return true;
} while (tolower(*a++) == tolower(*b++));

return false;
}
int getSecWebSocketAcceptKey(const char* key, char* b64buff, int len)
{
SHA1_CTX ctx;
unsigned char hash[20], buff[512];
if (strlen(key) > 256)
{
return -1;
}
sprintf(buff, "%s%s", key, "258EAFA5-E914-47DA-95CA-C5AB0DC85B11");
SHA1Init(&ctx);
SHA1Update(&ctx, buff, strlen(buff));
SHA1Final(hash, &ctx);

const char* base64 = b64_encode(hash, sizeof(hash));
strncpy(b64buff, base64, len - 1);
b64buff[len - 1] = '\0';
free((void*)base64);

return 0;
}
// 校验WebSocket握手的HTTP头,失败返回NULL,校验成功顺带返回Sec-WebSocket-Key,记得free
char* verifyHandshakeHeaders(const char* str, size_t len)
{
char* secKey = NULL;
char a[1024], b[1024];
bool connection, upgrade, version, key;
connection = upgrade = version = key = false;

if (strcmp(str + len - 4, "\r\n\r\n") != 0)
{
printLog("verifyHandshakeHeaders","HTTP header does not end with '\\r\\n\\r\\n'");
return NULL;
}

const char* cur1 = strstr(str, "\r\n") + 2;
const char* cur2;

while ((cur2 = strstr(cur1, "\r\n")) != cur1)
{
cur2 += 2;  // 跳过\r\n
const char* colon = strchr(cur1, ':');
if (colon == NULL || colon >= cur2)
{
printLog("verifyHandshakeHeaders", "Unexpected HTTP header");
break;
}

if (sscanf(cur1, "%[^:]:%s", a, b) != 2)
{
printLog("verifyHandshakeHeaders", "HTTP header parsing failed");
break;
}

if (stricasecmp(a, "connection"))
{
connection = true;
}
else if (stricasecmp(a, "upgrade"))
{
if (!stricasecmp(b, "websocket"))
{
printLog("verifyHandshakeHeaders", "Unexpected value '%s' of Upgrade filed", b);
break;
}
upgrade = true;

}
else if (stricasecmp(a, "Sec-WebSocket-Version"))
{
if (!stricasecmp(b, "13"))
{
printLog("verifyHandshakeHeaders","Unexpected value '%s' of Sec-WebSocket-Version filed", b);
break;
}
version = true;
}
else if (stricasecmp(a, "Sec-WebSocket-Key"))
{
if (!key)
{
key = true;
secKey = malloc(strlen(b) + 1);
strcpy(secKey, b);
}
}
cur1 = cur2;
}

if (!(connection && upgrade && version && key))
{
printLog("verifyHandshakeHeaders", "Missing necessary fields");
if (key) free((void*)secKey); // 释放申请的内存
return NULL;
}
return secKey;
}

int wsShakeHands(const char* recvBuff, int recvLen, SOCKET socket, const char* path)
{
#define RECV_BUFLEN 0X40001
#define HTTP_MAXLEN 1536
#define HTTP_400 "HTTP/1.1 400 Bad Request\r\n\r\n"

// HTTP握手包太长
if (recvLen > HTTP_MAXLEN)
{
send(socket, HTTP_400, strlen(HTTP_400), 0);
printLog("wsShakeHands","Request too long");
return -1;
}

// 注:recvBuff不以'\0'结尾
char resText[HTTP_MAXLEN + 1];
memcpy(resText, recvBuff, recvLen);
resText[recvLen] = '\0';

char requestLine[512];
sprintf(requestLine, "GET %s%s HTTP/1.1\r\n", (strlen(path) == 0 || path[0] != '/') ? "/" : "", path);

// 注:路径部分也被不区分大小写的比较
if (!strnicasecmp(resText, requestLine, strlen(requestLine)))
{
send(socket, HTTP_400, strlen(HTTP_400), 0);
printLog("wsShakeHands","Unexpected request line");
printLog("wsShakeHands", resText);
return -1;
}

const char* secKey = verifyHandshakeHeaders(resText, recvLen);

if (!secKey)
{
send(socket, HTTP_400, strlen(HTTP_400), 0);
return -1;
}

// 获取Sec-WebSocket-Accept
char acptBuff[128];
getSecWebSocketAcceptKey(secKey, acptBuff, sizeof(acptBuff));
printLog("wsShakeHands","Sec-WebSocket-Key is '%s'", secKey);
printLog("wsShakeHands", "Sec-WebSocket-Accept is '%s'", acptBuff);
free((void*)secKey);   // 释放secKey

// 协议升级
char resBuff[256];

// 注:当前的CORS设置可能会导致安全问题
// 注:响应中没有包含Sec-Websocket-Protocol头,代表不接受任何客户端请求的ws扩展
const char resHeader[] =
"HTTP/1.1 101 ojbk\r\n"
"Connection: Upgrade\r\n"
"Upgrade: websocket\r\n"
"Sec-WebSocket-Accept: %s\r\n"
"Access-Control-Allow-Origin: *\r\n"
"\r\n"
;

int resLen = sprintf(resBuff, resHeader, acptBuff);

// Send data to the client
int iSendResult = send(socket, resBuff, resLen, 0);

if (iSendResult == SOCKET_ERROR)
{
return -1;
}

printLog("wsShakeHands","Bytes sent: %d", iSendResult);
printLog("wsShakeHands","WebSocket handshake succeeded");

return 0;
}

receiveComingData方法里处理websocket数据的方法wsClientDataHandle代码

int readWebSocketFrameStream(WsFrame* wsFrame, const char* buff, int len)
{
if (wsFrame->buff == NULL)
{
wsFrame->buff = malloc(len);
wsFrame->buffSize = len;
memcpy(wsFrame->buff, buff, len);
}
else
{
char* copyStartAddr;
int requiedLen = wsFrame->buffSize + len;
wsFrame->buff = realloc((void*)wsFrame->buff, requiedLen);
copyStartAddr = wsFrame->buff + wsFrame->buffSize;
wsFrame->buffSize = requiedLen;
memcpy(copyStartAddr, buff, len);
}

// 消耗的数据量
int consumed = 0;

stateTransitionBegin:

switch (wsFrame->state)
{
case frameState_init:

if (wsFrame->buffSize < 1)
{
return consumed;
}

wsFrame->FIN = !!(wsFrame->buff[0] & 0X80);

// RSV位不全为0,存在扩展协议,服务器不处理扩展协议
if ((wsFrame->buff[0] & 0X70) != 0)
{
wsFrame->state = frameState_failure;
break;
}

int opcode = wsFrame->buff[0] & 0X0F;

if (opcode == 0X0)
{
wsFrame->frameType = frameType_continuation;
}
else if (opcode == 0X1)
{
wsFrame->frameType = frameType_text;
}
else if (opcode == 0X2)
{
wsFrame->frameType = frameType_binary;
}
else if (opcode == 0X8)
{
wsFrame->frameType = frameType_connectionClose;
}
else if (opcode == 0X9)
{
wsFrame->frameType = frameType_ping;
}
else if (opcode == 0XA)
{
wsFrame->frameType = frameType_pong;
}
else
{
wsFrame->state = frameState_failure;
break;
}

consumed += 1;
wsFrame->handledLen += 1;
wsFrame->headerLen += 1;
wsFrame->state = frameState_firstByte;

break;

case frameState_firstByte:

if (wsFrame->buffSize < 2)
{
return consumed;
}

// 标准规定客户端传入帧的掩码位必须不为0
if ((wsFrame->buff[1] & 0X80) == 0)
{
wsFrame->state = frameState_failure;
}

wsFrame->state = frameState_mask;

break;

case frameState_mask:

if (wsFrame->buffSize < 2)
{
return consumed;
}

uint8_t payloadLen = wsFrame->buff[1] & 0X7F;

// frame-payload-length-7
if (payloadLen < 126)
{
wsFrame->payloadLen = payloadLen;
wsFrame->state = frameState_7bitLength;
}
else if (payloadLen == 126)
{
wsFrame->state = frameState_16bitLengthWait;
}
else if (payloadLen == 127)
{
wsFrame->state = frameState_63bitLengthWait;
}

consumed += 1;
wsFrame->headerLen += 1;
wsFrame->handledLen += 1;

break;

case frameState_7bitLength:

// 2字节共有字段 + 0字节附加长度字段 + 4字节掩码
if (wsFrame->buffSize < 6)
{
return consumed;
}

for (int i = 0; i < 4; i++)
{
wsFrame->mask[i] = wsFrame->buff[i + 2];
}

consumed += 4;
wsFrame->headerLen += 4;
wsFrame->handledLen += 4;
wsFrame->state = frameState_maskingKey;
break;

case frameState_16bitLengthWait:

if (wsFrame->buffSize < 4)
{
return consumed;
}

wsFrame->payloadLen = ((uint16_t)wsFrame->buff[2] << 8) + (uint16_t)wsFrame->buff[3];
consumed += 2;
wsFrame->headerLen += 2;
wsFrame->handledLen += 2;
wsFrame->state = frameState_16bitLength;

break;

case frameState_63bitLengthWait:

if (wsFrame->buffSize < 10)
{
return consumed;
}
unsigned char* recvBuff = wsFrame->buff;

// 注:标准规定64位时最高bit必须为0,这里未作处理
56c

wsFrame->payloadLen =
((uint64_t)recvBuff[2] << (8 * 7)) +
((uint64_t)recvBuff[3] << (8 * 6)) +
((uint64_t)recvBuff[4] << (8 * 5)) +
((uint64_t)recvBuff[5] << (8 * 4)) +
((uint64_t)recvBuff[6] << (8 * 3)) +
((uint64_t)recvBuff[7] << (8 * 2)) +
((uint64_t)recvBuff[8] << (8 * 1)) +
((uint64_t)recvBuff[9] << (8 * 0));

consumed += 8;
wsFrame->headerLen += 8;
wsFrame->handledLen += 8;
wsFrame->state = frameState_63bitLength;

break;

case frameState_16bitLength:

3ff8
// 2字节共有字段 + 2字节附加长度字段 + 4字节掩码

if (wsFrame->buffSize < 8)
{
return consumed;
}

for (int i = 0; i < 4; i++)
{
wsFrame->mask[i] = wsFrame->buff[i + 4];
}

consumed += 4;
wsFrame->headerLen += 4;
wsFrame->handledLen += 4;

wsFrame->state = frameState_maskingKey;

break;

case frameState_63bitLength:

// 2字节共有字段 + 8字节附加长度字段 + 4字节掩码

if (wsFrame->buffSize < 14)
{
return consumed;
}

for (int i = 0; i < 4; i++)
{
wsFrame->mask[i] = wsFrame->buff[i + 10];
}

consumed += 4;
wsFrame->headerLen += 4;
wsFrame->handledLen += 4;

wsFrame->state = frameState_maskingKey;

break;

case frameState_maskingKey:

wsFrame->state = frameState_readingData;

break;

case frameState_readingData:

;    // case第一个语句不能是变量声明
uint64_t total = wsFrame->payloadLen + wsFrame->headerLen;

// 注意 buff的长度可能大于帧总长度
// 因为TCP是面向字节流的,buff中有可能包含下一帧的数据
// 所以读取时要根据帧头和载荷长度来判断最多读多少数据
if (wsFrame->buffSize >= total)
{
consumed += total - wsFrame->handledLen;
wsFrame->handledLen = total;
wsFrame->state = frameState_success;
}
else
{
consumed += wsFrame->buffSize - wsFrame->handledLen;
wsFrame->handledLen = wsFrame->buffSize;
return consumed;
}

break;

case frameState_success:

return consumed;

break;
case frameState_failure:

return consumed;
break;
}

goto stateTransitionBegin;

return consumed;
}

int wsFrameSend(SOCKET socket, const char* buff, int len, FrameType type)
{
int newLen;
const char* frame = convertToWebSocketFrame(buff, type, len, &newLen);

int iSendResult = send(socket, frame, newLen, 0);

if (iSendResult == SOCKET_ERROR)
{
printLog("wsFrameSend", 1, "Send failed: %d", WSAGetLastError());
goto wsFrameSendEnd;
}

wsFrameSendEnd:
free((void*)frame);
return iSendResult;
}

// 处理WebSocket帧数据,返回-1代表需要关闭连接
int wsClientDataHandle(const char* recvBuff, int recvLen, Client* client)
{
WsFrame* wsFrame = &client->wsFrame;
if (recvLen == 0)
{
return 0;
}

int consume = readWebSocketFrameStream(wsFrame, recvBuff, recvLen);

if (wsFrame->state == frameState_success)
{
// 暂时不处理多帧数据,遇到多帧数据关闭连接
if (wsFrame->FIN == 0)
{
return -1;
}

// 客户端希望关闭连接
if (wsFrame->frameType == frameType_connectionClose)
{
return -1;
}

// 遇到意料之外的帧类型
if (wsFrame->frameType == frameType_binary ||
wsFrame->frameType == frameType_pong ||
wsFrame->frameType == frameType_continuation
)
{
return -1;
}

uint64_t payloadLen = wsFrame->payloadLen;
u_char* payload = wsFrame->buff + wsFrame->headerLen;

// 解码载荷
for (uint64_t j = 0; j < payloadLen; j++)
{
payload[j] = payload[j] ^ wsFrame->mask[j % 4];
}

int iSendResult = 0;

// 心跳
if (wsFrame->frameType == frameType_ping)
{
wsFrameSend(client->socket, payload, payloadLen, frameType_pong);
}

// 处理文本数据
if (wsFrame->frameType == frameType_text)
{
wsClientTextDataHandle(payload, payloadLen, client->socket);
}
}

// 一个帧接收完成并处理完毕后释放内存
if (wsFrame->state == frameState_success)
{
freeWebSocketFrame(wsFrame);
}

// 解析ws帧出错,释放内存并通知关闭连接
if (wsFrame->state == frameState_failure)
{
freeWebSocketFrame(wsFrame);
return -1;
}

// 传入的数据不止包含当前帧,包含下一帧的数据
if (consume != recvLen)
{
return wsClientDataHandle(recvBuff + consume, recvLen - consume, client);
}

return 0;
}

处理文本数据的方法wsClientTextDataHandle代码

// 处理数据,也可以写在别处做回调函数
void wsClientTextDataHandle(const char* payload, uint64_t payloadLen, SOCKET socket)
{
const char* parseEnd;
cJSON* json = cJSON_ParseWithOpts(payload, &parseEnd, 0);

if (json == NULL)
{
const char* error_ptr = cJSON_GetErrorPtr();
if (error_ptr != NULL)
{
printLog("jsonParse",  "Error before: %d", error_ptr - payload);
}
return;
}

const cJSON* j_msg = cJSON_GetObjectItemCaseSensitive(json, "msg");
const cJSON_bool e_msg = cJSON_IsString(j_msg);
const char* v_msg = e_msg ? j_msg->valuestring : NULL;

char* gbkText = UTF8ToGBK(v_msg);
sendJSON(socket, "send", GBKToUTF8(gbkText));
free((void*)gbkText);

}

给客户端发送消息的代码

int wsFrameSend(SOCKET socket, const char* buff, int len, FrameType type)
{
int newLen;
const char* frame = convertToWebSocketFrame(buff, type, len, &newLen);

int iSendResult = send(socket, frame, newLen, 0);

if (iSendResult == SOCKET_ERROR)
{
printLog("wsFrameSend", 1, "Send failed: %d", WSAGetLastError());
goto wsFrameSendEnd;
}

printLog("wsFrameSend","Bytes sent: %d", iSendResult);

wsFrameSendEnd:
free((void*)frame);
return iSendResult;
}
void sendJSON(SOCKET socket, const char* event, const char* data)
{
cJSON* root = cJSON_CreateObject();
cJSON_AddItemToObject(root, "event", cJSON_CreateString(event));
cJSON_AddItemToObject(root, "data", cJSON_CreateString(data));

const char* jsonStr = cJSON_PrintUnformatted(root);
wsFrameSend(socket, jsonStr, strlen(jsonStr), frameType_text);

cJSON_Delete(root);
free((void*)jsonStr);
}

到这里主要的核心代码就写完了,接下来测试一下

int main()
{
int result = serverStart("127.0.0.1", 1024, "/");

if (result != 0)
{
MessageBoxA(NULL, "wsserver start failed", "WebSocket Plugin", MB_OK | MB_ICONERROR);
}
else
{
//fileLog("websocket server startup success");
}
system("pause");
return 0;
}

测试成功 如图所示

 

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: