1 /*
2 * Copyright (c) 2006-2021, RT-Thread Development Team
3 *
4 * SPDX-License-Identifier: Apache-2.0
5 *
6 * Change Logs:
7 * Date Author Notes
8 * 2021-05-15 Sherman the first version
9 * 2021-08-04 Sherman Adapted to new version of rt-link API
10 */
11
12 #include <stdlib.h>
13 #include <string.h>
14 #include <rtthread.h>
15 #include <rtlink.h>
16
17 #define DBG_ENABLE
18 #define DBG_TAG "rtlink_exam"
19 #define DBG_LVL DBG_INFO
20 #include <rtdbg.h>
21
22 #define TEST_CONTEXT "This message is sent by RT-Link"
23
24 enum
25 {
26 NONE_TEST = 0,
27 SHORT_FRAME_TEST,
28 LONG_FRAME_TEST
29 };
30
31 void rt_link_speed_test(void *paremeter);
32 static rt_uint8_t speed_test_type = NONE_TEST;
33 static struct rt_link_service serv_socket;
34 static struct rt_link_service serv_wifi;
35
send_cb(struct rt_link_service * service,void * buffer)36 static void send_cb(struct rt_link_service *service, void *buffer)
37 {
38 LOG_I("send_cb: service (%d) buffer (0x%p) err(%d)", service->service, buffer, service->err);
39 }
40
recv_cb(struct rt_link_service * service,void * data,rt_size_t size)41 static void recv_cb(struct rt_link_service *service, void *data, rt_size_t size)
42 {
43 LOG_I("service (%d) size (%d) data(0x%p)", service->service, size, data);
44
45 if (size)
46 {
47 rt_free(data);
48 }
49 }
50
rt_link_speed_test(void * paremeter)51 static void rt_link_speed_test(void *paremeter)
52 {
53 int ret;
54 rt_uint8_t *send_buf, *data;
55 rt_size_t bufflen = 0;
56 rt_size_t sentlen = 0;
57 rt_size_t count = 0;
58 rt_tick_t tick1, tick2;
59 rt_size_t total = 0;
60 rt_uint32_t integer, decimal;
61
62 if (speed_test_type == SHORT_FRAME_TEST)
63 {
64 bufflen = RT_LINK_MAX_DATA_LENGTH;
65 }
66 else
67 {
68 bufflen = RT_LINK_MAX_DATA_LENGTH * RT_LINK_FRAMES_MAX;
69 }
70
71 send_buf = rt_malloc(bufflen);
72 if (send_buf != RT_NULL)
73 {
74 data = send_buf;
75 for (count = 0; count < bufflen; count++)
76 {
77 *data++ = (count % 93 + 33);
78 }
79 }
80 else
81 {
82 rt_kprintf("speed of send buffer malloc failed.");
83 return;
84 }
85
86 tick1 = rt_tick_get();
87 while (speed_test_type)
88 {
89 ret = rt_link_send(&serv_socket, send_buf, bufflen);
90 if (ret > 0)
91 {
92 sentlen += ret;
93 }
94 else
95 {
96 LOG_W("send err %d", ret);
97 }
98
99 tick2 = rt_tick_get();
100 if (tick2 - tick1 >= RT_TICK_PER_SECOND)
101 {
102 total = sentlen * RT_TICK_PER_SECOND / 125 / (tick2 - tick1);
103 integer = total / 1000;
104 decimal = total % 1000;
105 LOG_I("%d.%03d0 Mbps!", integer, decimal);
106 sentlen = 0;
107 tick1 = tick2;
108 }
109 }
110 rt_free(send_buf);
111 LOG_W("speed test end, type %d", speed_test_type);
112 }
113
create_thead_to_test_speed(rt_uint8_t mutil_num)114 static void create_thead_to_test_speed(rt_uint8_t mutil_num)
115 {
116 rt_uint8_t i = 0;
117
118 LOG_D("Speed test type [%02d], mutil [%02d]", speed_test_type, mutil_num);
119 for (i = 0; i < mutil_num; i++)
120 {
121 rt_thread_t tid;
122 char tid_name[RT_NAME_MAX + 1] = {0};
123
124 rt_snprintf(tid_name, sizeof(tid_name), "lny_s%03d", i + 1);
125 tid = rt_thread_create(tid_name, rt_link_speed_test, RT_NULL, 1024, 20, 10);
126 if (tid)
127 {
128 rt_thread_startup(tid);
129 LOG_I("Speed test thread[%s] startup", tid_name);
130 }
131 }
132 }
133
rtlink_exsend(int argc,char ** argv)134 static int rtlink_exsend(int argc, char **argv)
135 {
136 char *receive = RT_NULL;
137 rt_size_t length = 0;
138 rt_uint16_t count = 0;
139
140 if (argc == 1)
141 {
142 receive = rt_malloc(sizeof(TEST_CONTEXT));
143 rt_memcpy(receive, TEST_CONTEXT, sizeof(TEST_CONTEXT) - 1);
144 length = rt_link_send(&serv_socket, receive, sizeof(TEST_CONTEXT) - 1);
145 LOG_I("send data length: %d.", length);
146 rt_free(receive);
147 }
148 else if (argc >= 3)
149 {
150 if (strcmp(argv[1], "-l") == 0)
151 {
152 receive = rt_malloc(atoi((const char *)argv[2]));
153 for (count = 0; count < atoi((const char *)argv[2]); count++)
154 {
155 *receive++ = (count % 93 + 33);
156 }
157 length = rt_link_send(&serv_socket, receive - atoi((const char *)argv[2]), atoi((const char *)argv[2]));
158 rt_free(receive - atoi((const char *)argv[2]));
159
160 LOG_I("send data length: %d.", length);
161 }
162 else if (strcmp(argv[1], "-s") == 0)
163 {
164 if (speed_test_type == NONE_TEST)
165 {
166 rt_uint8_t mutil_num = 1;
167 if (argc > 3)
168 {
169 mutil_num = atoi((const char *)argv[3]);
170 }
171
172 if (strncmp(argv[2], "-s", rt_strlen(argv[2])) == 0)
173 {
174 speed_test_type = SHORT_FRAME_TEST;
175 }
176 else if (strncmp(argv[2], "-l", rt_strlen(argv[2])) == 0)
177 {
178 speed_test_type = LONG_FRAME_TEST;
179 }
180 create_thead_to_test_speed(mutil_num);
181 }
182 else
183 {
184 speed_test_type = NONE_TEST;
185 LOG_I("set NONE_TEST");
186 }
187 }
188 else
189 {
190 LOG_E("Invalid parameter.");
191 }
192 }
193 return 0;
194 }
195 MSH_CMD_EXPORT(rtlink_exsend, rt link layer send test);
196
rtlink_exinit(void)197 int rtlink_exinit(void)
198 {
199 serv_socket.service = RT_LINK_SERVICE_SOCKET;
200 serv_socket.timeout_tx = RT_WAITING_FOREVER;
201 serv_socket.flag = RT_LINK_FLAG_ACK | RT_LINK_FLAG_CRC;
202 serv_socket.recv_cb = recv_cb;
203 serv_socket.send_cb = send_cb;
204 rt_link_service_attach(&serv_socket);
205
206 serv_wifi.service = RT_LINK_SERVICE_WIFI;
207 serv_wifi.timeout_tx = RT_WAITING_FOREVER;
208 serv_wifi.flag = RT_NULL;
209 serv_wifi.recv_cb = recv_cb;
210 serv_wifi.send_cb = send_cb;
211 rt_link_service_attach(&serv_wifi);
212 return RT_EOK;
213 }
214 MSH_CMD_EXPORT(rtlink_exinit, rt link example init);
215