HarmonyOS网络编程 --TCP服务端开发_homary os tcp

参考网上连老师的TCP自己整理一套TCP服务端的开发，整体思路大概如下：
1、TCP服务首先需要连接WiFi，采用的是STA模式，在编写TCP服务端代码之前已经对WiFi功能进行了学习，代码也编写好了，所以直接封装一下拿来使用就OK！！代码如下，在applications/sample/wifi-iot/app/下新建wifi_demo文件夹，然后新建wifi_sta_connect.h文件和wifi_sta_connect.c文件：

wifi_sta_connect.h

#ifndef _WIFI_STA_CONNECT_H
#define _WIFI_STA_CONNECT_H

        
extern int g_connected;
//WIFI sta模式相关功能函数
void PrintLinkedInfo(WifiLinkedInfo* info);
void OnWifiConnectionChanged(int state, WifiLinkedInfo* info);
void OnWifiScanStateChanged(int state, int size);

#endif
1
2
3
4
5
6
7
8
9
10
11

wifi_sta_connect.c

#include <stdio.h>
#include <string.h>
#include <unistd.h>

#include "ohos_init.h"
#include "cmsis_os2.h"
#include "wifi_device.h"

#include "lwip/netifapi.h"
#include "lwip/api_shell.h"
#include "wifi_sta_connect.h"

void PrintLinkedInfo(WifiLinkedInfo* info)
{
    if (!info) return;

    static char macAddress[32] = {0};
    unsigned char* mac = info->bssid;
    snprintf(macAddress, sizeof(macAddress), "%02X:%02X:%02X:%02X:%02X:%02X",
        mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
    printf("bssid: %s, rssi: %d, connState: %d, reason: %d, ssid: %s\r\n",
        macAddress, info->rssi, info->connState, info->disconnectedReason, info->ssid);
}

int g_connected = 0;
void OnWifiConnectionChanged(int state, WifiLinkedInfo* info)
{
    if (!info) return;

    printf("%s %d, state = %d, info = \r\n", __FUNCTION__, __LINE__, state);
    PrintLinkedInfo(info);

    if (state == WIFI_STATE_AVALIABLE) {
        g_connected = 1;
    } else {
        g_connected = 0;
    }
}

void OnWifiScanStateChanged(int state, int size)
{
    printf("%s %d, state = %X, size = %d\r\n", __FUNCTION__, __LINE__, state, size);
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

2、WiFi功能模块封装的相对简单，后期如果需要更多的功能，在模块中添加即可，接下来就是TCP服务端代码的编写，其中需要调用WiFi功能。
在applications/sample/wifi-iot/app下新建socket_demo文件夹，然后添加tcp_server_test.c文件
tcp_server_test.c

#include <stdio.h>
#include <unistd.h>

#include "ohos_init.h"
#include "cmsis_os2.h"
#include "wifi_device.h"
#include "lwip/api_shell.h"

#include "hi_wifi_api.h"
#include "lwip/ip_addr.h"
#include "lwip/netifapi.h"

#include "lwip/sockets.h"
#include "wifi_sta_connect.h"

#define SERVER_PORT_TCP			6666
#define TCP_BACKLOG 10


/* 在sock_fd 进行监听，在 new_fd 接收新的链接 */
int sock_fd, new_fd;

char recvbuf[512];
char *buf = "hello! I'm server!";

int tcp_server_demo(void){
    /* 自己的地址信息 */
	struct sockaddr_in my_addr;
	/*	连接者的地址信息*/
	struct sockaddr_in their_addr;
	int sin_size;

	struct sockaddr_in *cli_addr;

    printf("%s %d \r\n", __FILE__, __LINE__);

    /* 1 、创建socket  */
	if((sock_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
	{
        printf("%s %d \r\n", __FILE__, __LINE__);
		perror("socket is error\r\n");
		exit(1);
	}

    /* 主机字节顺序 */
	/* 协议 */
	my_addr.sin_family = AF_INET;
	my_addr.sin_port = htons(6666);//设置你的端口号
	/* 当前IP 地址写入 */
	my_addr.sin_addr.s_addr = INADDR_ANY;

	/* 将结构体其余的都清零 */
	bzero(&(my_addr.sin_zero), 8);

    printf("%s %d \r\n", __FILE__, __LINE__);
	/* bind 绑定*/
	if(bind(sock_fd, (struct sockaddr *)&my_addr, sizeof(struct sockaddr)) == -1)
	{
        printf("%s %d \r\n", __FILE__, __LINE__);
		perror("bind is error\r\n");
		exit(1);
	}

    printf("%s %d \r\n", __FILE__, __LINE__);
	/* 开始监听 */
	if(listen(sock_fd, TCP_BACKLOG) == -1)
	{
		perror("listen is error\r\n");
		exit(1);
	}

    printf("%s %d \r\n", __FILE__, __LINE__);
	printf("start accept\n");

    /* accept() 循环 */
	while(1)
	{
		sin_size = sizeof(struct sockaddr_in);

        printf("%s %d \r\n", __FILE__, __LINE__);
		if((new_fd = accept(sock_fd, (struct sockaddr *)&their_addr, (socklen_t *)&sin_size)) == -1)
		{
			perror("accept");
			continue;
		}

		cli_addr = malloc(sizeof(struct sockaddr));

		printf("accept addr\r\n");

		if(cli_addr != NULL)
		{
			memcpy(cli_addr, &their_addr, sizeof(struct sockaddr));
		}

		//处理目标
		ssize_t ret;
		

		while(1)
		{
            printf("%s %d \r\n", __FILE__, __LINE__);
			if((ret = recv(new_fd, recvbuf, sizeof(recvbuf), 0)) == -1){
				printf("recv error \r\n");
				return -1;
			}
			printf("recv :\r\n");
			printf("%s", recvbuf);
			printf("\r\n");
			sleep(2);
			if((ret = send(new_fd, buf, strlen(buf) + 1, 0)) == -1)
			{
				perror("send : ");
			}

			sleep(2);
		}

		

		close(new_fd);

        return 0;
	}
}

void start_sta_mode(void){ 
    WifiErrorCode errCode;
    WifiEvent eventListener = {
        .OnWifiConnectionChanged = OnWifiConnectionChanged,  //连接状态改变
        .OnWifiScanStateChanged = OnWifiScanStateChanged     //扫描状态变化
    };
    WifiDeviceConfig apConfig = {};
    int netId = -1;

    osDelay(10);
    errCode = RegisterWifiEvent(&eventListener); //注册wifi事件的回调函数
    printf("RegisterWifiEvent: %d\r\n", errCode);

    // setup your AP params
    strcpy(apConfig.ssid, "*******");  //设置你的wifi名称
    strcpy(apConfig.preSharedKey, "*******");  //设置你的wifi密码
    apConfig.securityType = WIFI_SEC_TYPE_PSK;

    while (1) {
        errCode = EnableWifi();  //打开wifi设备的sta模式，使其可以扫描
        printf("EnableWifi: %d\r\n", errCode);
        osDelay(10);

        errCode = AddDeviceConfig(&apConfig, &netId);  //增加WIFI设备的配置，分配得到一个net id保存在netId中
        printf("AddDeviceConfig: %d\r\n", errCode);

        g_connected = 0;
        errCode = ConnectTo(netId);
        printf("ConnectTo(%d): %d\r\n", netId, errCode);

        while (!g_connected) {
            osDelay(10);
        }
        printf("g_connected: %d\r\n", g_connected);
        osDelay(50);

        // 联网业务开始
        struct netif* iface = netifapi_netif_find("wlan0");
        if (iface) {
            err_t ret = netifapi_dhcp_start(iface);
            printf("netifapi_dhcp_start: %d\r\n", ret);

            osDelay(200); // wait DHCP server give me IP
            ret = netifapi_netif_common(iface, dhcp_clients_info_show, NULL);
            printf("netifapi_netif_common: %d\r\n", ret);
        }

        // 模拟一段时间的联网业务
        // int timeout = 60;
        while (1) {
            // osDelay(100);
            // printf("after %d seconds, I'll disconnect WiFi!\n", timeout);
            printf("ongoing Iot business\n");
            uint32_t result = tcp_server_demo();
            if (result==0)
            {
                /* code */
                printf("I'll disconnect WiFi!\n");
                break;
            }
            
        }

        // 联网业务结束
        err_t ret = netifapi_dhcp_stop(iface);
        printf("netifapi_dhcp_stop: %d\r\n", ret);

        Disconnect(); // disconnect with your AP

        RemoveDevice(netId); // remove AP config

        errCode = DisableWifi();
        printf("DisableWifi: %d\r\n", errCode);
        osDelay(200);
    }
}



void TcpServerTask(void* arg){
    (void) arg;
    start_sta_mode();
    // tcp_server_demo();

}

void TcpServerEntry(void){
    osThreadAttr_t attr;

    attr.name = "TcpServerTask";
    attr.attr_bits = 0U;
    attr.cb_mem = NULL;
    attr.cb_size = 0U;
    attr.stack_mem = NULL;
    attr.stack_size = 10240;
    attr.priority = osPriorityNormal;

    if (osThreadNew(TcpServerTask, NULL, &attr) == NULL) {
        printf("[WifiConnectDemo] Falied to create TcpServerTask!\n");
    }
}

SYS_RUN(TcpServerEntry);
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229

在applications/sample/wifi-iot/app下新建socket_demo文件夹，然后添加BUILD.gn文件，这里需要把封装WiFi功能模块的文件包含进去：
BUILD.gn

static_library("socket_demo") {
    sources = [
        "tcp_server_test.c",
    ]

    include_dirs = [
        "//utils/native/lite/include",
        "//kernel/liteos_m/components/cmsis/2.0",
        "//base/iot_hardware/interfaces/kits/wifiiot_lite",
        "//foundation/communication/interfaces/kits/wifi_lite/wifiservice",
        "//vendor/hisi/hi3861/hi3861/third_party/lwip_sack/include/",
        "//foundation/communication/wifi_lite/interfaces/wifiservice",
        "//applications/sample/wifi-iot/app/wifi_demo",
    ]
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

3、最后修改applications/sample/wifi-iot/app/BUILD.gn文件：

import("//build/lite/config/component/lite_component.gni")

lite_component("app") {

    features = [

        "wifi_demo",

        "socket_demo",

    ]

}
1
2
3
4
5
6
7
8
9
10
11
12
13

4、最后编译烧录即可，自己可以编写一个TCP客户端去连接服务端进行测试，作为学习HarmonyOS的新手，还有许多不足，欢迎大家留言补充，共同学习成长！