ESP32C3实现TCP客户端和本地服务器
/*本地TCP服务器端和客户端*/
#include <stdio.h>
#include <string.h>
#include "freertos/FreeRTOS.h"
#include "freertos/task.h"
#include "esp_wifi.h"
#include "esp_netif.h"
#include "nvs_flash.h"
#include "wifi_provisioning/manager.h"
#include "wifi_provisioning/scheme_softap.h"
#include "socket.h"
/*宏定义*/
#define ERASE_NVS 0
// 清除储存的wifi密码->0: 沿用上次的WIFI信息进行连接; 1: 清除WIFI信息,使用SOFTAP一键配网进行新链接
#define TCP_SERRVER_OR_CLIENT 1
// 0: 服务器; 1: 客户端
/*函数声明*/
void WIFI_CALLBACK_FUN(void *event_handler_arg, esp_event_base_t event_base, int32_t event_id, void *event_data); // 事件队列回调函数
void wifi_connect(void); // WIFI连接处理函数
/*全局变量*/
/***************************************************************************************************************/
/*MAIN函数*/
void app_main(void)
{
wifi_connect(); // 连接wifi
#if TCP_SERRVER_OR_CLIENT != 0 // TCP客户端
/*1. 创建套接字*/
int sock_fd = socket(AF_INET, SOCK_STREAM, 0); // 创建套接字
if (sock_fd < 0)
goto error;
/*2. 连接服务器*/
struct sockaddr_in srvr_info =
{
.sin_family = AF_INET,
.sin_addr.s_addr = inet_addr("192.168.137.156"),
.sin_port = htons(65530)};
int ret = connect(sock_fd, (struct sockaddr *)&srvr_info, sizeof(srvr_info));
if (ret != 0)
{
printf("ERRORNO: %d\n", errno);
for (int i = 0; i < 10; i++)
{
printf("正在尝试第%d次连接服务器...\n", i);
if (0 == connect(sock_fd, (struct sockaddr *)&srvr_info, sizeof(srvr_info)))
break;
vTaskDelay(100);
if (i == 9)
{
printf("connect函数出错,服务器无法连接!\n");
goto error;
}
}
}
/*3. 发送数据给服务器*/
int snd_cnt = 0;
char *snd_dt = "Hello Server.";
while (1)
{
snd_cnt++;
send(sock_fd, snd_dt, strlen(snd_dt), 0);
vTaskDelay(200);
printf("第%d次发送数据到服务器!\n", snd_cnt);
}
shutdown(sock_fd, 0);
close(sock_fd);
#else // TCP服务器
/* 1. 创建套接字 */
int fd = socket(AF_INET, SOCK_STREAM, 0); // IPV4, TCP
if (fd < 0)
goto error;
/* 1.x 进行连接中断处理设置 */
// int opt_val = 1;
// setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &opt_val, sizeof(int));//服务器出现等待时间超时确保server能够重启成功
/* 2. 绑定IP和端口号 */
struct sockaddr_in bind_addr =
{
.sin_family = AF_INET,
.sin_addr.s_addr = htonl(inet_addr("0.0.0.0")), // 0x00 00 00 00UL表示不对报文源地址进行验证
.sin_port = htons(3333)};
int ret = bind(fd, (struct sockaddr *)&bind_addr, sizeof(struct sockaddr)); // 绑定fd, IP和端口
if (ret < 0)
goto error;
/* 3. 监听 */
ret = listen(fd, 1); // 套接字fd,允许连接数量1
if (ret < 0)
goto error;
/* 4. 等待客户端连接 */
// u32_t ip_info = inet_addr("192.168.137.156"); //将点分IP转换为整型
struct sockaddr_in client_addr = {0}; // 为accept连接的客户端保存IP
u32_t client_addr_len = sizeof(client_addr);
int clnt_fd = -1;
for (;;)
{
clnt_fd = accept(fd, (struct sockaddr *)&client_addr, &client_addr_len); // 监听等待客户端连接
if (clnt_fd >= 0) // 客户端连接成功
{
// printf("客户端连接成功, 客户端的IP地址为:%s\n",inet_ntoa(ip_info));
printf("客户端连接成功,客户端的IP地址为:%s\n", inet_ntoa(client_addr.sin_addr.s_addr));
break;
}
}
/* 5. 连接成功,进行数据发送 */
int len = 0;
char rev_dat[1024] = {0};
int rcv_cnt = 0;
while (1)
{
rcv_cnt++;
len = recv(clnt_fd, rev_dat, 1023, 0); // 阻塞接收数据,接收到数据后跳出recv函数
if (len > 0) // 接收到数据
{
printf("第%d次接收到IP为%s的数据为: %s, 数据长度为:%d\n", rcv_cnt, inet_ntoa(client_addr.sin_addr.s_addr), rev_dat, len);
strncpy(rev_dat + strlen(rev_dat), "\0", 1); // 在字符末位增加换行符
send(clnt_fd, rev_dat, strlen(rev_dat) + 1, 0); // 将接收到的数据发送回客户端
}
else if (len == 0) // 连接中断
{
printf("连接中断!正在等待新连接...\n");
close(clnt_fd);
rcv_cnt = 0;
for (;;)
{
clnt_fd = accept(fd, (struct sockaddr *)&client_addr, &client_addr_len); // 监听等待新的客户端连接
if (clnt_fd > 0)
{
printf("新链接已经建立!\n");
break;
}
}
}
else
{
printf("退出!\n");
break;
}
vTaskDelay(1); // 防止看门狗复位
printf("标记\n");
}
shutdown(clnt_fd, 0); // 用于任何类型的 套接口禁止接收、禁止发送或禁止收发
close(clnt_fd);
close(fd);
#endif
error:
while (1)
;
}
/**********************************************************************************************/
/*回调函数*/
void WIFI_CALLBACK_FUN(void *event_handler_arg, esp_event_base_t eb, int32_t ei, void *event_data)
{
if (eb == WIFI_EVENT)
{
if (ei == WIFI_EVENT_STA_START)
{
esp_wifi_connect(); // 开始连接WIFI
printf("正在连接WIFI!\n");
}
if (ei == WIFI_EVENT_STA_DISCONNECTED)
{
printf("自动连接失败,正在启动SOFTAP配网服务!\n");
const char *pop = "abcd1234";
const char *wifi_name = "SOFTAP_PROV_MGR_PROB";
const char *wifi_key = "12345678";
wifi_prov_mgr_start_provisioning(WIFI_PROV_SECURITY_1, pop, wifi_name, wifi_key);
wifi_prov_mgr_wait();
wifi_prov_mgr_deinit();
printf("Provisioning服务结束,连接WIFI成功!\n");
}
}
if (eb == IP_EVENT && ei == IP_EVENT_STA_GOT_IP)
{
printf("WIFI连接成功!\n");
ip_event_got_ip_t *sta_ip = (ip_event_got_ip_t *)event_data;
printf("本机IP地址为: " IPSTR "\n", IP2STR(&sta_ip->ip_info.ip));
}
}
/*连接WIFI*/
void wifi_connect(void)
{
nvs_flash_init();
#if ERASE_NVS != 0
nvs_flash_erase();
nvs_flash_init();
#endif
esp_event_loop_create_default(); // 创建默认事件队列
esp_event_handler_instance_register(IP_EVENT, IP_EVENT_STA_GOT_IP, WIFI_CALLBACK_FUN, NULL, NULL);
esp_netif_init(); // 初始化网卡
esp_netif_create_default_wifi_ap(); // 创建默认网卡ap
esp_netif_create_default_wifi_sta(); // 创建默认网卡sta
wifi_init_config_t esp_wifi_init_config = WIFI_INIT_CONFIG_DEFAULT();
esp_wifi_init(&esp_wifi_init_config); // 初始化WIFI功能
/*softAP配网*/
wifi_prov_mgr_config_t wifi_prov_mgr_config_t =
{
.scheme = wifi_prov_scheme_softap,
.app_event_handler = WIFI_PROV_EVENT_HANDLER_NONE};
wifi_prov_mgr_init(wifi_prov_mgr_config_t);
bool wifi_prov_mgr_diag = false;
wifi_prov_mgr_is_provisioned(&wifi_prov_mgr_diag);
if (wifi_prov_mgr_diag == false)
{
const char *pop = "abcd1234";
const char *wifi_name = "SOFTAP_PROV_MGR_PROB";
const char *wifi_key = "12345678";
wifi_prov_mgr_start_provisioning(WIFI_PROV_SECURITY_1, pop, wifi_name, wifi_key);
wifi_prov_mgr_wait();
wifi_prov_mgr_deinit();
printf("Provisioning服务结束,连接WIFI成功!\n");
}
else
{
esp_event_handler_instance_register(WIFI_EVENT, WIFI_EVENT_STA_START, WIFI_CALLBACK_FUN, NULL, NULL); // 注册WIFI开启事件
esp_event_handler_instance_register(WIFI_EVENT, WIFI_EVENT_STA_CONNECTED, WIFI_CALLBACK_FUN, NULL, NULL); // 注册WIFI连接事件
esp_event_handler_instance_register(WIFI_EVENT, WIFI_EVENT_STA_DISCONNECTED, WIFI_CALLBACK_FUN, NULL, NULL); // 注册WIFI连接失败事件
esp_wifi_start();
}
esp_wifi_set_ps(WIFI_PS_NONE); // 关闭WIFI的省电模式
}