384 lines
13 KiB
C
384 lines
13 KiB
C
#include "web_socket/websocket_api.h"
|
||
#include "wifi/wifi_connect.h"
|
||
#include "system/includes.h"
|
||
//#include "zliblite/zlib.h"
|
||
#ifdef WEBSOCKET_MEMHEAP_DEBUG
|
||
#include "mem_leak_test.h"
|
||
#endif
|
||
|
||
#include "app_config.h"
|
||
|
||
#ifdef USE_WEBSOCKET_TEST
|
||
|
||
/**** websoket "ws://" websokets "wss://" *****/
|
||
|
||
//#define OBJ_URL "wss://ws.postman-echo.com/raw" //远程服务器测试
|
||
#define OBJ_URL "wss://openspeech.bytedance.com/api/v2/asr"
|
||
//#define OBJ_URL "ws://121.40.165.18:8800"
|
||
//#define OBJ_URL "ws://www.example.com/socketserver"
|
||
//#define OBJ_URL "wss://127.0.0.1:8888" //本地服务器测试
|
||
//#define OBJ_URL "wss://172.16.23.28:8888" //本地服务器测试
|
||
|
||
static void websockets_callback(u8 *buf, u32 len, u8 type)
|
||
{
|
||
printf("wbs recv msg : %s\n", buf);
|
||
// compress(NULL,NULL,NULL,0);
|
||
}
|
||
|
||
/*******************************************************************************
|
||
* Websocket Client api
|
||
*******************************************************************************/
|
||
static void websockets_client_reg(struct websocket_struct *websockets_info, char mode)
|
||
{
|
||
memset(websockets_info, 0, sizeof(struct websocket_struct));
|
||
websockets_info->_init = websockets_client_socket_init;
|
||
websockets_info->_exit = websockets_client_socket_exit;
|
||
websockets_info->_handshack = webcockets_client_socket_handshack;
|
||
websockets_info->_send = websockets_client_socket_send;
|
||
websockets_info->_recv_thread = websockets_client_socket_recv_thread;
|
||
websockets_info->_heart_thread = websockets_client_socket_heart_thread;
|
||
websockets_info->_recv_cb = websockets_callback;
|
||
websockets_info->_recv = NULL;
|
||
|
||
websockets_info->websocket_mode = mode;
|
||
|
||
#if 0
|
||
//应用层指定申请接收内存大小（申请即可，不用释放，内部会释放），默认不使用
|
||
websockets_info->recv_buf_size = 30 * 1024;
|
||
websockets_info->recv_buf = malloc(websockets_info->recv_buf_size);
|
||
#endif
|
||
}
|
||
|
||
static int websockets_client_init(struct websocket_struct *websockets_info, u8 *url, const char *origin_str, const char *user_agent_str)
|
||
{
|
||
websockets_info->ip_or_url = url;
|
||
websockets_info->origin_str = origin_str;
|
||
websockets_info->user_agent_str = user_agent_str;
|
||
websockets_info->recv_time_out = 1000;
|
||
|
||
//应用层和库的版本检测，结构体不一样则返回出错
|
||
int err = websockets_struct_check(sizeof(struct websocket_struct));
|
||
if (err == FALSE) {
|
||
return err;
|
||
}
|
||
return websockets_info->_init(websockets_info);
|
||
}
|
||
|
||
static int websockets_client_handshack(struct websocket_struct *websockets_info)
|
||
{
|
||
return websockets_info->_handshack(websockets_info);
|
||
}
|
||
|
||
static int websockets_client_send(struct websocket_struct *websockets_info, u8 *buf, int len, char type)
|
||
{
|
||
//SSL加密时一次发送数据不能超过16K，用户需要自己分包
|
||
return websockets_info->_send(websockets_info, buf, len, type);
|
||
}
|
||
|
||
static void websockets_client_exit(struct websocket_struct *websockets_info)
|
||
{
|
||
websockets_info->_exit(websockets_info);
|
||
}
|
||
|
||
|
||
/*******************************************************************************
|
||
* Websocket Client.c
|
||
* Just one example for test
|
||
*******************************************************************************/
|
||
static void websockets_client_main_thread(void *priv)
|
||
{
|
||
int err;
|
||
char mode = WEBSOCKET_MODE;
|
||
#ifdef WIN32
|
||
void *heart_thread_hdl = NULL;
|
||
void *recv_thread_hdl = NULL;
|
||
#endif
|
||
u8 url[256] = OBJ_URL;
|
||
const char *input_str = "1234546";
|
||
const char *origin_str = "openspeech.bytedance.com";
|
||
const char *user_agent_str = "Authorization: Bearer; BNA-XBIrEisvMczcmc-Sm4B3H-PmfqlL";
|
||
|
||
printf(" . ----------------- Client Websocket ------------------\r\n");
|
||
|
||
#ifdef WIN32
|
||
printf(" . Please input the mode for websocket : 1 . websocket , 2 . websockets (ssl)\r\n");
|
||
scanf("%d", &mode);
|
||
if (mode == 2) {
|
||
mode = WEBSOCKETS_MODE;
|
||
printf(" . Please input the URL , for example: wss://localhost:8888\r\n");
|
||
} else {
|
||
mode = WEBSOCKETS_MODE;
|
||
printf(" . Please input the URL , for example: ws://localhost:8888\r\n");
|
||
}
|
||
|
||
scanf("%d", &url);
|
||
#endif
|
||
|
||
/* 0 . malloc buffer */
|
||
struct websocket_struct *websockets_info = malloc(sizeof(struct websocket_struct));
|
||
if (!websockets_info) {
|
||
return;
|
||
}
|
||
/* 1 . register */
|
||
websockets_client_reg(websockets_info, mode);
|
||
|
||
/* 2 . init */
|
||
err = websockets_client_init(websockets_info, url, origin_str, user_agent_str);
|
||
if (FALSE == err) {
|
||
printf(" . ! Cilent websocket init error !!!\r\n");
|
||
goto exit_ws;
|
||
}
|
||
|
||
/* 3 . hanshack */
|
||
err = websockets_client_handshack(websockets_info);
|
||
if (FALSE == err) {
|
||
printf(" . ! Handshake error !!!\r\n");
|
||
goto exit_ws;
|
||
}
|
||
printf(" . Handshake success \r\n");
|
||
|
||
/* 4 . CreateThread */
|
||
#ifdef WIN32
|
||
heart_thread_hdl = CreateThread(NULL, 0, websockets_info->_heart_thread, websockets_info, 0, &websockets_info->ping_thread_id);
|
||
recv_thread_hdl = CreateThread(NULL, 0, websockets_info->_recv_thread, websockets_info, 0, &websockets_info->ping_thread_id);
|
||
#else
|
||
thread_fork("websocket_client_heart", 19, 512, 0,
|
||
&websockets_info->ping_thread_id,
|
||
websockets_info->_heart_thread,
|
||
websockets_info);
|
||
thread_fork("websocket_client_recv", 18, 512, 0,
|
||
&websockets_info->recv_thread_id,
|
||
websockets_info->_recv_thread,
|
||
websockets_info);
|
||
#endif
|
||
websockets_sleep(1000);
|
||
|
||
/* 5 . recv or send data */
|
||
while (1) {
|
||
// err = websockets_client_send(websockets_info, (u8 *)input_str, strlen(input_str), WCT_TXTDATA);
|
||
// if (FALSE == err) {
|
||
// printf(" . ! send err !!!\r\n");
|
||
// goto exit_ws;
|
||
// }
|
||
printf(" . Send massage : %s \r\n", input_str);
|
||
websockets_sleep(3000);
|
||
#ifdef WIN32
|
||
printf(" . Please input the send massage\r\n");
|
||
memset(input_str, 0, sizeof(input_str));
|
||
scanf("%s", input_str);
|
||
#endif
|
||
}
|
||
|
||
exit_ws:
|
||
/* 6 . exit */
|
||
#ifdef WIN32
|
||
if (heart_thread_hdl) {
|
||
TerminateThread(heart_thread_hdl, 0);
|
||
}
|
||
if (recv_thread_hdl) {
|
||
TerminateThread(recv_thread_hdl, 0);
|
||
}
|
||
#else
|
||
if (websockets_info->ping_thread_id) {
|
||
thread_kill(&websockets_info->ping_thread_id, KILL_REQ);
|
||
}
|
||
if (websockets_info->recv_thread_id) {
|
||
thread_kill(&websockets_info->recv_thread_id, KILL_REQ);
|
||
}
|
||
#endif
|
||
websockets_client_exit(websockets_info);
|
||
free(websockets_info);
|
||
}
|
||
|
||
//static void websockets_callback(u8 *buf, u32 len, u8 type)
|
||
//{
|
||
// printf("wbs recv msg : %s\n", buf);
|
||
//}
|
||
//
|
||
///*******************************************************************************
|
||
//* Websocket Client api
|
||
//*******************************************************************************/
|
||
//static void websockets_client_reg(struct websocket_struct *websockets_info, char mode)
|
||
//{
|
||
// memset(websockets_info, 0, sizeof(struct websocket_struct));
|
||
// websockets_info->_init = websockets_client_socket_init;
|
||
// websockets_info->_exit = websockets_client_socket_exit;
|
||
// websockets_info->_handshack = webcockets_client_socket_handshack;
|
||
// websockets_info->_send = websockets_client_socket_send;
|
||
// websockets_info->_recv_thread = websockets_client_socket_recv_thread;
|
||
// websockets_info->_heart_thread = websockets_client_socket_heart_thread;
|
||
// websockets_info->_recv_cb = websockets_callback;
|
||
// websockets_info->_recv = NULL;
|
||
//
|
||
// websockets_info->websocket_mode = mode;
|
||
//#if 0
|
||
// //应用层指定申请接收内存大小（申请即可，不用释放，内部会释放），默认不使用
|
||
// websockets_info->recv_buf_size = 30 * 1024;
|
||
// websockets_info->recv_buf = malloc(websockets_info->recv_buf_size);
|
||
//#endif
|
||
//}
|
||
//
|
||
//static int websockets_client_init(struct websocket_struct *websockets_info, u8 *url, const char *origin_str)
|
||
//{
|
||
// websockets_info->ip_or_url = url;
|
||
// websockets_info->origin_str = origin_str;
|
||
// websockets_info->recv_time_out = 1000;
|
||
//// websockets_info->user_agent_str = LOG_HTTP_HEAD_OPTION;
|
||
//
|
||
// //应用层和库的版本检测，结构体不一样则返回出错
|
||
// int err = websockets_struct_check(sizeof(struct websocket_struct));
|
||
// if (err == FALSE) {
|
||
// return err;
|
||
// }
|
||
// return websockets_info->_init(websockets_info);
|
||
//}
|
||
//
|
||
//static int websockets_client_handshack(struct websocket_struct *websockets_info)
|
||
//{
|
||
// return websockets_info->_handshack(websockets_info);
|
||
//}
|
||
//
|
||
//static int websockets_client_send(struct websocket_struct *websockets_info, u8 *buf, int len, char type)
|
||
//{
|
||
// //SSL加密时一次发送数据不能超过16K，用户需要自己分包
|
||
// return websockets_info->_send(websockets_info, buf, len, type);
|
||
//}
|
||
//
|
||
//static void websockets_client_exit(struct websocket_struct *websockets_info)
|
||
//{
|
||
// websockets_info->_exit(websockets_info);
|
||
//}
|
||
//
|
||
///*******************************************************************************
|
||
//* Websocket Client.c
|
||
//* Just one example for test
|
||
//*******************************************************************************/
|
||
//static void websockets_client_main_thread(void *priv)
|
||
//{
|
||
// int err;
|
||
// char mode = WEBSOCKET_MODE;
|
||
//#ifdef WIN32
|
||
// void *heart_thread_hdl = NULL;
|
||
// void *recv_thread_hdl = NULL;
|
||
//#endif
|
||
// u8 url[256] = OBJ_URL;
|
||
// const char *input_str = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
|
||
// const char *origin_str = "openspeech.bytedance.com/api/v2/asr";
|
||
//
|
||
// printf(" . ----------------- Client Websocket ------------------\r\n");
|
||
//
|
||
//#ifdef WIN32
|
||
// printf(" . Please input the mode for websocket : 1 . websocket , 2 . websockets (ssl)\r\n");
|
||
// scanf("%d", &mode);
|
||
// if (mode == 2) {
|
||
// mode = WEBSOCKETS_MODE;
|
||
// printf(" . Please input the URL , for example: wss://localhost:8888\r\n");
|
||
// } else {
|
||
// mode = WEBSOCKETS_MODE;
|
||
// printf(" . Please input the URL , for example: ws://localhost:8888\r\n");
|
||
// }
|
||
//
|
||
// scanf("%d", &url);
|
||
//#endif
|
||
//
|
||
// /* 0 . malloc buffer */
|
||
// struct websocket_struct *websockets_info = malloc(sizeof(struct websocket_struct));
|
||
// if (!websockets_info) {
|
||
// return;
|
||
// }
|
||
// /* 1 . register */
|
||
// websockets_client_reg(websockets_info, mode);
|
||
//
|
||
// /* 2 . init */
|
||
// err = websockets_client_init(websockets_info, url, origin_str);
|
||
// if (FALSE == err) {
|
||
// printf(" . ! Cilent websocket init error !!!\r\n");
|
||
// goto exit_ws;
|
||
// }
|
||
//
|
||
// /* 3 . hanshack */
|
||
// err = websockets_client_handshack(websockets_info);
|
||
// if (FALSE == err) {
|
||
// printf(" . ! Handshake error !!!\r\n");
|
||
// goto exit_ws;
|
||
// }
|
||
// printf(" . Handshake success \r\n");
|
||
//
|
||
// /* 4 . CreateThread */
|
||
//#ifdef WIN32
|
||
// heart_thread_hdl = CreateThread(NULL, 0, websockets_info->_heart_thread, websockets_info, 0, &websockets_info->ping_thread_id);
|
||
// recv_thread_hdl = CreateThread(NULL, 0, websockets_info->_recv_thread, websockets_info, 0, &websockets_info->ping_thread_id);
|
||
//#else
|
||
// thread_fork("websocket_client_heart", 19, 512, 0,
|
||
// &websockets_info->ping_thread_id,
|
||
// websockets_info->_heart_thread,
|
||
// websockets_info);
|
||
// thread_fork("websocket_client_recv", 18, 512, 0,
|
||
// &websockets_info->recv_thread_id,
|
||
// websockets_info->_recv_thread,
|
||
// websockets_info);
|
||
//#endif
|
||
// websockets_sleep(1000);
|
||
//
|
||
// /* 5 . recv or send data */
|
||
// while (1) {
|
||
// err = websockets_client_send(websockets_info, (u8 *)input_str, strlen(input_str), WCT_TXTDATA);
|
||
// if (FALSE == err) {
|
||
// printf(" . ! send err !!!\r\n");
|
||
// goto exit_ws;
|
||
// }
|
||
// printf(" . Send massage : %s \r\n", input_str);
|
||
// websockets_sleep(3000);
|
||
//#ifdef WIN32
|
||
// printf(" . Please input the send massage\r\n");
|
||
// memset(input_str, 0, sizeof(input_str));
|
||
// scanf("%s", input_str);
|
||
//#endif
|
||
// }
|
||
//
|
||
//exit_ws:
|
||
// /* 6 . exit */
|
||
//#ifdef WIN32
|
||
// TerminateThread(heart_thread_hdl, 0);
|
||
// TerminateThread(recv_thread_hdl, 0);
|
||
//#else
|
||
// thread_kill(&websockets_info->ping_thread_id, KILL_REQ);
|
||
// thread_kill(&websockets_info->recv_thread_id, KILL_REQ);
|
||
//#endif
|
||
// websockets_client_exit(websockets_info);
|
||
// free(websockets_info);
|
||
//}
|
||
|
||
static void websocket_client_thread_create(void)
|
||
{
|
||
thread_fork("websockets_client_main", 15, 512 * 3, 0, 0, websockets_client_main_thread, NULL);
|
||
}
|
||
|
||
static void http_websocket_start(void *priv)
|
||
{
|
||
enum wifi_sta_connect_state state;
|
||
while (1) {
|
||
printf("Connecting to the network...\n");
|
||
state = wifi_get_sta_connect_state();
|
||
if (WIFI_STA_NETWORK_STACK_DHCP_SUCC == state) {
|
||
printf("Network connection is successful!\n");
|
||
break;
|
||
}
|
||
os_time_dly(500);
|
||
}
|
||
|
||
websocket_client_thread_create();
|
||
}
|
||
|
||
//应用程序入口,需要运行在STA模式下
|
||
void c_main(void *priv)
|
||
{
|
||
if (thread_fork("http_websocket_start", 10, 512, 0, NULL, http_websocket_start, NULL) != OS_NO_ERR) {
|
||
printf("thread fork fail\n");
|
||
}
|
||
}
|
||
|
||
late_initcall(c_main);
|
||
|
||
#endif//USE_WEBSOCKET_TEST
|