websocket底层处理粘包_Websocket消息帧粘包，拆包及处理方法

最新推荐文章于 2024-01-04 10:17:41 发布

weixin_39945475

最新推荐文章于 2024-01-04 10:17:41 发布

阅读量984

点赞数

文章标签： websocket底层处理粘包

本文链接：https://blog.csdn.net/weixin_39945475/article/details/112034886

版权

本文介绍了WebSocket协议在处理粘包、拆包问题时的方法，详细讲解了如何构建和解析WebSocket消息帧。通过示例代码展示了如何实现连接服务器、心跳检测、数据发送和接收，并提供了相关函数如`webSocket_send`和`webSocket_recv`的实现。内容包括基础的网络控制、数据编码和解码，以及基于套接字的通信技巧。

摘要由CSDN通过智能技术生成

#include "websocket_common.h"

#include

#include // 使用 malloc, calloc等动态分配内存方法

#include // 获取系统时间

#include

#include // 非阻塞

#include

#include // inet_addr()

#include // close()

#include // 文件IO操作

#include //

#include

#include // gethostbyname, gethostbyname2, gethostbyname_r, gethostbyname_r2

#include

#include // SIOCSIFADDR

//==============================================================================================

//======================================== 设置和工具部分 =======================================

//==============================================================================================

// 连接服务器

#define WEBSOCKET_LOGIN_CONNECT_TIMEOUT 1000 // 登录连接超时设置 1000ms

#define WEBSOCKET_LOGIN_RESPOND_TIMEOUT (1000 + WEBSOCKET_LOGIN_CONNECT_TIMEOUT) // 登录等待回应超时设置 1000ms

// 发收

// 生成握手key的长度

#define WEBSOCKET_SHAKE_KEY_LEN 16

//==================== delay ms ====================

void webSocket_delayms(unsigned int ms)

{

struct timeval tim;

tim.tv_sec = ms/1000;

tim.tv_usec = (ms%1000)*1000;

select(0, NULL, NULL, NULL, &tim);

}

//-------------------- IP控制 --------------------

int netCheck_setIP(char *devName, char *ip)

{

struct ifreq temp;

struct sockaddr_in *addr;

int fd, ret;

strcpy(temp.ifr_name, devName);

if((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)

return -1;

addr = (struct sockaddr_in *)&(temp.ifr_addr);

addr->sin_family = AF_INET;

addr->sin_addr.s_addr = inet_addr(ip);

ret = ioctl(fd, SIOCSIFADDR, &temp);

close(fd);

if(ret < 0)

return -1;

return 0;

}

void netCheck_getIP(char *devName, char *ip)

{

struct ifreq temp;

struct sockaddr_in *addr;

int fd, ret;

strcpy(temp.ifr_name, devName);

if((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)

return;

ret = ioctl(fd, SIOCGIFADDR, &temp);

close(fd);

if(ret < 0)

return;

addr = (struct sockaddr_in *)&(temp.ifr_addr);

strcpy(ip, inet_ntoa(addr->sin_addr));

// return ip;

}

//==================== 域名转IP ====================

typedef struct{

pthread_t thread_id;

char ip[256];

bool result;

bool actionEnd;

}GetHostName_Struct;

void *websocket_getHost_fun(void *arge)

{

int ret;

//int i;

char buf[1024];

struct hostent host_body, *host = NULL;

struct in_addr **addr_list;

GetHostName_Struct *gs = (GetHostName_Struct *)arge;

/* 此类方法不可重入! 即使关闭线程

if((host = gethostbyname(gs->ip)) == NULL)

//if((host = gethostbyname2(gs->ip, AF_INET)) == NULL)

{

gs->actionEnd = true;

return NULL;

}*/

if(gethostbyname_r(gs->ip, &host_body, buf, sizeof(buf), &host, &ret))

{

gs->actionEnd = true;

return NULL;

}

if(host == NULL)

{

gs->actionEnd = true;

return NULL;

}

addr_list = (struct in_addr **)host->h_addr_list;

//printf("ip name : %s\r\nip list : ", host->h_name);

//for(i = 0; addr_list[i] != NULL; i++) printf("%s, ", inet_ntoa(*addr_list[i])); printf("\r\n");

if(addr_list[0] == NULL)

{

gs->actionEnd = true;

return NULL;

}

memset(gs->ip, 0, sizeof(gs->ip));

strcpy(gs->ip, (char *)(inet_ntoa(*addr_list[0])));

gs->result = true;

gs->actionEnd = true;

return NULL;

}

int websocket_getIpByHostName(char *hostName, char *backIp)

{

int i, timeOut = 1;

GetHostName_Struct gs;

if(hostName == NULL)

return -1;

else if(strlen(hostName) < 1)

return -1;

//----- 开线程从域名获取IP -----

memset(&gs, 0, sizeof(GetHostName_Struct));

strcpy(gs.ip, hostName);

gs.result = false;

gs.actionEnd = false;

if (pthread_create(&gs.thread_id, NULL, (void *)websocket_getHost_fun, &gs) < 0)

return -1;

i = 0;

while(!gs.actionEnd)

{

if(++i > 10)

{

i = 0;

if(++timeOut > 1000)

break;

}

webSocket_delayms(1000);// 1ms延时

}

// pthread_cancel(gs.thread_id);

pthread_join(gs.thread_id, NULL);

if(!gs.result)

return -timeOut;

//----- 开线程从域名获取IP -----

memset(backIp, 0, strlen(backIp));

strcpy(backIp, gs.ip);

return timeOut;

}

//==================== 加密方法BASE64 ====================

//base64编/解码用的基础字符集

const char websocket_base64char[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";

/*******************************************************************************

* 名称: websocket_base64_encode

* 功能: ascii编码为base64格式

* 形参: bindata : ascii字符串输入

* base64 : base64字符串输出

* binlength : bindata的长度

* 返回: base64字符串长度

* 说明: 无

******************************************************************************/

int websocket_base64_encode( const unsigned char *bindata, char *base64, int binlength)

{

int i, j;

unsigned char current;

for ( i = 0, j = 0 ; i < binlength ; i += 3 )

{

current = (bindata[i] >> 2) ;

current &= (unsigned char)0x3F;