sockets_stresstest.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726
  1. /**
  2. * @file
  3. * Sockets stresstest
  4. *
  5. * This file uses the lwIP socket API to do stress tests that should test the
  6. * stability when used in many different situations, with many concurrent
  7. * sockets making concurrent transfers in different manners.
  8. *
  9. * - test rely on loopback sockets for now, so netif drivers are not tested
  10. * - all enabled functions shall be used
  11. * - parallelism of the tests depend on enough resources being available
  12. * (configure your lwipopts.h settings high enough)
  13. * - test should also be able to run in a real target
  14. *
  15. * TODO:
  16. * - full duplex
  17. * - add asserts about internal socket/netconn/pcb state?
  18. */
  19. /*
  20. * Copyright (c) 2017 Simon Goldschmidt
  21. * All rights reserved.
  22. *
  23. * Redistribution and use in source and binary forms, with or without modification,
  24. * are permitted provided that the following conditions are met:
  25. *
  26. * 1. Redistributions of source code must retain the above copyright notice,
  27. * this list of conditions and the following disclaimer.
  28. * 2. Redistributions in binary form must reproduce the above copyright notice,
  29. * this list of conditions and the following disclaimer in the documentation
  30. * and/or other materials provided with the distribution.
  31. * 3. The name of the author may not be used to endorse or promote products
  32. * derived from this software without specific prior written permission.
  33. *
  34. * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
  35. * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  36. * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
  37. * SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  38. * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
  39. * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  40. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  41. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
  42. * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
  43. * OF SUCH DAMAGE.
  44. *
  45. * This file is part of the lwIP TCP/IP stack.
  46. *
  47. * Author: Simon Goldschmidt <goldsimon@gmx.de>
  48. *
  49. */
  50. #include "lwip/opt.h"
  51. #include "sockets_stresstest.h"
  52. #include "lwip/sockets.h"
  53. #include "lwip/sys.h"
  54. #include "lwip/mem.h"
  55. #include <stdio.h>
  56. #include <string.h>
  57. #if LWIP_SOCKET && LWIP_IPV4 /* this uses IPv4 loopback sockets, currently */
  58. #ifndef TEST_SOCKETS_STRESS
  59. #define TEST_SOCKETS_STRESS LWIP_DBG_OFF
  60. #endif
  61. #define TEST_TIME_SECONDS 10
  62. #define TEST_TXRX_BUFSIZE (TCP_MSS * 2)
  63. #define TEST_MAX_RXWAIT_MS 50
  64. #define TEST_MAX_CONNECTIONS 50
  65. #define TEST_SOCK_READABLE 0x01
  66. #define TEST_SOCK_WRITABLE 0x02
  67. #define TEST_SOCK_ERR 0x04
  68. #define TEST_MODE_SELECT 0x01
  69. #define TEST_MODE_POLL 0x02
  70. #define TEST_MODE_NONBLOCKING 0x04
  71. #define TEST_MODE_WAIT 0x08
  72. #define TEST_MODE_RECVTIMEO 0x10
  73. #define TEST_MODE_SLEEP 0x20
  74. static int sockets_stresstest_numthreads;
  75. struct test_settings {
  76. struct sockaddr_storage addr;
  77. int start_client;
  78. int loop_cnt;
  79. };
  80. struct sockets_stresstest_fullduplex {
  81. int s;
  82. volatile int closed;
  83. };
  84. static void
  85. fill_test_data(void *buf, size_t buf_len_bytes)
  86. {
  87. u8_t *p = (u8_t*)buf;
  88. u16_t i, chk;
  89. LWIP_ASSERT("buffer too short", buf_len_bytes >= 4);
  90. LWIP_ASSERT("buffer too big", buf_len_bytes <= 0xFFFF);
  91. /* store the total number of bytes */
  92. p[0] = (u8_t)(buf_len_bytes >> 8);
  93. p[1] = (u8_t)buf_len_bytes;
  94. /* fill buffer with random */
  95. chk = 0;
  96. for (i = 4; i < buf_len_bytes; i++) {
  97. u8_t rnd = (u8_t)LWIP_RAND();
  98. p[i] = rnd;
  99. chk += rnd;
  100. }
  101. /* store checksum */
  102. p[2] = (u8_t)(chk >> 8);
  103. p[3] = (u8_t)chk;
  104. }
  105. static size_t
  106. check_test_data(const void *buf, size_t buf_len_bytes)
  107. {
  108. u8_t *p = (u8_t*)buf;
  109. u16_t i, chk, chk_rx, len_rx;
  110. LWIP_ASSERT("buffer too short", buf_len_bytes >= 4);
  111. len_rx = (((u16_t)p[0]) << 8) | p[1];
  112. LWIP_ASSERT("len too short", len_rx >= 4);
  113. if (len_rx > buf_len_bytes) {
  114. /* not all data received in this segment */
  115. LWIP_DEBUGF(TEST_SOCKETS_STRESS | LWIP_DBG_TRACE, (0x10009788, "check-\n"));
  116. return buf_len_bytes;
  117. }
  118. chk_rx = (((u16_t)p[2]) << 8) | p[3];
  119. /* calculate received checksum */
  120. chk = 0;
  121. for (i = 4; i < len_rx; i++) {
  122. chk += p[i];
  123. }
  124. LWIP_ASSERT("invalid checksum", chk == chk_rx);
  125. if (len_rx < buf_len_bytes) {
  126. size_t data_left = buf_len_bytes - len_rx;
  127. memmove(p, &p[len_rx], data_left);
  128. return data_left;
  129. }
  130. /* if we come here, we received exactly one chunk
  131. -> next offset is 0 */
  132. return 0;
  133. }
  134. static size_t
  135. recv_and_check_data_return_offset(int s, char *rxbuf, size_t rxbufsize, size_t rxoff, int *closed, const char *dbg)
  136. {
  137. ssize_t ret;
  138. ret = lwip_read(s, &rxbuf[rxoff], rxbufsize - rxoff);
  139. if (ret == 0) {
  140. *closed = 1;
  141. return rxoff;
  142. }
  143. *closed = 0;
  144. LWIP_DEBUGFS(TEST_SOCKETS_STRESS | LWIP_DBG_TRACE, (OSI_LOGPAR_SII, 0x10009789, "net: %s %d rx %d\n", dbg, s, (int)ret));
  145. if (ret == -1) {
  146. /* TODO: for this to work, 'errno' has to support multithreading... */
  147. int err = errno;
  148. if (err == ENOTCONN) {
  149. *closed = 1;
  150. return 0;
  151. }
  152. LWIP_ASSERT("err == 0", err == 0);
  153. }
  154. LWIP_ASSERT("ret > 0", ret > 0);
  155. return check_test_data(rxbuf, rxoff + ret);
  156. }
  157. #if LWIP_SOCKET_SELECT
  158. static int
  159. sockets_stresstest_wait_readable_select(int s, int timeout_ms)
  160. {
  161. int ret;
  162. struct timeval tv;
  163. fd_set fs_r;
  164. fd_set fs_w;
  165. fd_set fs_e;
  166. FD_ZERO(&fs_r);
  167. FD_ZERO(&fs_w);
  168. FD_ZERO(&fs_e);
  169. FD_SET(s, &fs_r);
  170. FD_SET(s, &fs_e);
  171. tv.tv_sec = timeout_ms / 1000;
  172. tv.tv_usec = (timeout_ms - (tv.tv_sec * 1000)) * 1000;
  173. ret = lwip_select(s + 1, &fs_r, &fs_w, &fs_e, &tv);
  174. LWIP_ASSERT("select error", ret >= 0);
  175. if (ret) {
  176. /* convert poll flags to our flags */
  177. ret = 0;
  178. if (FD_ISSET(s, &fs_r)) {
  179. ret |= TEST_SOCK_READABLE;
  180. }
  181. if (FD_ISSET(s, &fs_w)) {
  182. ret |= TEST_SOCK_WRITABLE;
  183. }
  184. if (FD_ISSET(s, &fs_e)) {
  185. ret |= TEST_SOCK_ERR;
  186. }
  187. return ret;
  188. }
  189. return 0;
  190. }
  191. #endif
  192. #if LWIP_SOCKET_POLL
  193. static int
  194. sockets_stresstest_wait_readable_poll(int s, int timeout_ms)
  195. {
  196. int ret;
  197. struct pollfd pfd;
  198. pfd.fd = s;
  199. pfd.revents = 0;
  200. pfd.events = POLLIN | POLLERR;
  201. ret = lwip_poll(&pfd, 1, timeout_ms);
  202. if (ret) {
  203. /* convert poll flags to our flags */
  204. ret = 0;
  205. if (pfd.revents & POLLIN) {
  206. ret |= TEST_SOCK_READABLE;
  207. }
  208. if (pfd.revents & POLLOUT) {
  209. ret |= TEST_SOCK_WRITABLE;
  210. }
  211. if (pfd.revents & POLLERR) {
  212. ret |= TEST_SOCK_ERR;
  213. }
  214. return ret;
  215. }
  216. return 0;
  217. }
  218. #endif
  219. #if LWIP_SO_RCVTIMEO
  220. static int
  221. sockets_stresstest_wait_readable_recvtimeo(int s, int timeout_ms)
  222. {
  223. int ret;
  224. char buf;
  225. #if LWIP_SO_SNDRCVTIMEO_NONSTANDARD
  226. int opt_on = timeout_ms;
  227. int opt_off = 0;
  228. #else
  229. struct timeval opt_on, opt_off;
  230. opt_on.tv_sec = timeout_ms / 1000;
  231. opt_on.tv_usec = (timeout_ms - (opt_on.tv_sec * 1000)) * 1000;
  232. opt_off.tv_sec = 0;
  233. opt_off.tv_usec = 0;
  234. #endif
  235. /* enable receive timeout */
  236. ret = lwip_setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, &opt_on, sizeof(opt_on));
  237. LWIP_ASSERT("setsockopt error", ret == 0);
  238. /* peek for one byte with timeout */
  239. ret = lwip_recv(s, &buf, 1, MSG_PEEK);
  240. /* disable receive timeout */
  241. ret = lwip_setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, &opt_off, sizeof(opt_off));
  242. LWIP_ASSERT("setsockopt error", ret == 0);
  243. if (ret == 1) {
  244. return TEST_SOCK_READABLE;
  245. }
  246. if (ret == 0) {
  247. return 0;
  248. }
  249. if (ret == -1) {
  250. return TEST_SOCK_ERR;
  251. }
  252. LWIP_ASSERT("invalid return value", 0);
  253. return TEST_SOCK_ERR;
  254. }
  255. #endif
  256. static int
  257. sockets_stresstest_wait_readable_wait_peek(int s, int timeout_ms)
  258. {
  259. int ret;
  260. char buf;
  261. LWIP_UNUSED_ARG(timeout_ms); /* cannot time out here */
  262. /* peek for one byte */
  263. ret = lwip_recv(s, &buf, 1, MSG_PEEK);
  264. if (ret == 1) {
  265. return TEST_SOCK_READABLE;
  266. }
  267. if (ret == 0) {
  268. return 0;
  269. }
  270. if (ret == -1) {
  271. return TEST_SOCK_ERR;
  272. }
  273. LWIP_ASSERT("invalid return value", 0);
  274. return TEST_SOCK_ERR;
  275. }
  276. static int
  277. sockets_stresstest_wait_readable_nonblock(int s, int timeout_ms)
  278. {
  279. int ret;
  280. char buf;
  281. u32_t wait_until = sys_now() + timeout_ms;
  282. while(sys_now() < wait_until) {
  283. /* peek for one byte */
  284. ret = lwip_recv(s, &buf, 1, MSG_PEEK | MSG_DONTWAIT);
  285. if (ret == 1) {
  286. return TEST_SOCK_READABLE;
  287. }
  288. if (ret == -1) {
  289. /* TODO: for this to work, 'errno' has to support multithreading... */
  290. int err = errno;
  291. if (err != EWOULDBLOCK) {
  292. return TEST_SOCK_ERR;
  293. }
  294. }
  295. /* TODO: sleep? */
  296. }
  297. return 0;
  298. }
  299. static int sockets_stresstest_rand_mode(int allow_wait, int allow_rx)
  300. {
  301. u32_t random_value = LWIP_RAND();
  302. #if LWIP_SOCKET_SELECT
  303. if (random_value & TEST_MODE_SELECT) {
  304. return TEST_MODE_SELECT;
  305. }
  306. #endif
  307. #if LWIP_SOCKET_POLL
  308. if (random_value & TEST_MODE_POLL) {
  309. return TEST_MODE_POLL;
  310. }
  311. #endif
  312. if (!allow_rx) {
  313. return TEST_MODE_SLEEP;
  314. }
  315. #if LWIP_SO_RCVTIMEO
  316. if (random_value & TEST_MODE_RECVTIMEO) {
  317. return TEST_MODE_RECVTIMEO;
  318. }
  319. #endif
  320. if (allow_wait) {
  321. if (random_value & TEST_MODE_RECVTIMEO) {
  322. return TEST_MODE_RECVTIMEO;
  323. }
  324. }
  325. return TEST_MODE_NONBLOCKING;
  326. }
  327. static int
  328. sockets_stresstest_wait_readable(int mode, int s, int timeout_ms)
  329. {
  330. switch(mode)
  331. {
  332. #if LWIP_SOCKET_SELECT
  333. case TEST_MODE_SELECT:
  334. return sockets_stresstest_wait_readable_select(s, timeout_ms);
  335. #endif
  336. #if LWIP_SOCKET_POLL
  337. case TEST_MODE_POLL:
  338. return sockets_stresstest_wait_readable_poll(s, timeout_ms);
  339. #endif
  340. #if LWIP_SO_RCVTIMEO
  341. case TEST_MODE_RECVTIMEO:
  342. return sockets_stresstest_wait_readable_recvtimeo(s, timeout_ms);
  343. #endif
  344. case TEST_MODE_WAIT:
  345. return sockets_stresstest_wait_readable_wait_peek(s, timeout_ms);
  346. case TEST_MODE_NONBLOCKING:
  347. return sockets_stresstest_wait_readable_nonblock(s, timeout_ms);
  348. case TEST_MODE_SLEEP:
  349. {
  350. sys_msleep(timeout_ms);
  351. return 1;
  352. }
  353. default:
  354. LWIP_ASSERT("invalid mode", 0);
  355. break;
  356. }
  357. return 0;
  358. }
  359. #if LWIP_NETCONN_FULLDUPLEX
  360. static void
  361. sockets_stresstest_conn_client_r(void *arg)
  362. {
  363. struct sockets_stresstest_fullduplex *fd = (struct sockets_stresstest_fullduplex *)arg;
  364. int s = fd->s;
  365. size_t rxoff = 0;
  366. char rxbuf[TEST_TXRX_BUFSIZE];
  367. while (1) {
  368. int closed;
  369. if (fd->closed) {
  370. break;
  371. }
  372. rxoff = recv_and_check_data_return_offset(s, rxbuf, sizeof(rxbuf), rxoff, &closed, "cli");
  373. if (fd->closed) {
  374. break;
  375. }
  376. if (closed) {
  377. lwip_close(s);
  378. break;
  379. }
  380. }
  381. SYS_ARCH_DEC(sockets_stresstest_numthreads, 1);
  382. LWIP_ASSERT("", sockets_stresstest_numthreads >= 0);
  383. }
  384. #endif
  385. static void
  386. sockets_stresstest_conn_client(void *arg)
  387. {
  388. struct sockaddr_storage addr;
  389. struct sockaddr_in *addr_in;
  390. int s, ret;
  391. char txbuf[TEST_TXRX_BUFSIZE];
  392. char rxbuf[TEST_TXRX_BUFSIZE];
  393. size_t rxoff = 0;
  394. u32_t max_time = sys_now() + (TEST_TIME_SECONDS * 1000);
  395. int do_rx = 1;
  396. struct sockets_stresstest_fullduplex *data = NULL;
  397. memcpy(&addr, arg, sizeof(addr));
  398. LWIP_ASSERT("", addr.ss_family == AF_INET);
  399. addr_in = (struct sockaddr_in *)&addr;
  400. addr_in->sin_addr.s_addr = inet_addr("127.0.0.1");
  401. /* sleep a random time between 1 and 2 seconds */
  402. sys_msleep(1000 + (LWIP_RAND() % 1000));
  403. /* connect to the server */
  404. s = lwip_socket(addr.ss_family, SOCK_STREAM, 0);
  405. LWIP_ASSERT("s >= 0", s >= 0);
  406. #if LWIP_NETCONN_FULLDUPLEX
  407. if (LWIP_RAND() & 1) {
  408. sys_thread_t t;
  409. data = (struct sockets_stresstest_fullduplex*)mem_malloc(sizeof(struct sockets_stresstest_fullduplex));
  410. LWIP_ASSERT("data != NULL", data != 0);
  411. SYS_ARCH_INC(sockets_stresstest_numthreads, 1);
  412. data->s = s;
  413. data->closed = 0;
  414. t = sys_thread_new("sockets_stresstest_conn_client_r", sockets_stresstest_conn_client_r, data, 0, 0);
  415. LWIP_ASSERT("thread != NULL", t != 0);
  416. do_rx = 0;
  417. }
  418. #endif
  419. /* @todo: nonblocking connect? */
  420. ret = lwip_connect(s, (struct sockaddr *)&addr, sizeof(struct sockaddr_storage));
  421. LWIP_ASSERT("ret == 0", ret == 0);
  422. while (sys_now() < max_time) {
  423. int closed;
  424. int mode = sockets_stresstest_rand_mode(0, do_rx);
  425. int timeout_ms = LWIP_RAND() % TEST_MAX_RXWAIT_MS;
  426. ret = sockets_stresstest_wait_readable(mode, s, timeout_ms);
  427. if (ret) {
  428. if (do_rx) {
  429. /* read some */
  430. LWIP_ASSERT("readable", ret == TEST_SOCK_READABLE);
  431. rxoff = recv_and_check_data_return_offset(s, rxbuf, sizeof(rxbuf), rxoff, &closed, "cli");
  432. LWIP_ASSERT("client got closed", !closed);
  433. }
  434. } else {
  435. /* timeout, send some */
  436. size_t send_len = (LWIP_RAND() % (sizeof(txbuf) - 4)) + 4;
  437. fill_test_data(txbuf, send_len);
  438. LWIP_DEBUGF(TEST_SOCKETS_STRESS | LWIP_DBG_TRACE, (0x1000978a, "cli %d tx %d\n", s, (int)send_len));
  439. ret = lwip_write(s, txbuf, send_len);
  440. if (ret == -1) {
  441. /* TODO: for this to work, 'errno' has to support multithreading... */
  442. int err = errno;
  443. LWIP_ASSERT("err == 0", err == 0);
  444. }
  445. LWIP_ASSERT("ret == send_len", ret == (int)send_len);
  446. }
  447. }
  448. if (data) {
  449. data->closed = 1;
  450. }
  451. ret = lwip_close(s);
  452. LWIP_ASSERT("ret == 0", ret == 0);
  453. SYS_ARCH_DEC(sockets_stresstest_numthreads, 1);
  454. LWIP_ASSERT("", sockets_stresstest_numthreads >= 0);
  455. }
  456. static void
  457. sockets_stresstest_conn_server(void *arg)
  458. {
  459. int s, ret;
  460. char txbuf[TEST_TXRX_BUFSIZE];
  461. char rxbuf[TEST_TXRX_BUFSIZE];
  462. size_t rxoff = 0;
  463. s = (int)arg;
  464. while (1) {
  465. int closed;
  466. int mode = sockets_stresstest_rand_mode(1, 1);
  467. int timeout_ms = LWIP_RAND() % TEST_MAX_RXWAIT_MS;
  468. ret = sockets_stresstest_wait_readable(mode, s, timeout_ms);
  469. if (ret) {
  470. if (ret & TEST_SOCK_ERR) {
  471. /* closed? */
  472. break;
  473. }
  474. /* read some */
  475. LWIP_ASSERT("readable", ret == TEST_SOCK_READABLE);
  476. rxoff = recv_and_check_data_return_offset(s, rxbuf, sizeof(rxbuf), rxoff, &closed, "srv");
  477. if (closed) {
  478. break;
  479. }
  480. } else {
  481. /* timeout, send some */
  482. size_t send_len = (LWIP_RAND() % (sizeof(txbuf) - 4)) + 4;
  483. fill_test_data(txbuf, send_len);
  484. LWIP_DEBUGF(TEST_SOCKETS_STRESS | LWIP_DBG_TRACE, (0x1000978b, "srv %d tx %d\n", s, (int)send_len));
  485. ret = lwip_write(s, txbuf, send_len);
  486. if (ret == -1) {
  487. /* TODO: for this to work, 'errno' has to support multithreading... */
  488. int err = errno;
  489. if (err == ECONNRESET) {
  490. break;
  491. }
  492. if (err == ENOTCONN) {
  493. break;
  494. }
  495. LWIP_ASSERT("unknown error", 0);
  496. }
  497. LWIP_ASSERT("ret == send_len", ret == (int)send_len);
  498. }
  499. }
  500. ret = lwip_close(s);
  501. LWIP_ASSERT("ret == 0", ret == 0);
  502. SYS_ARCH_DEC(sockets_stresstest_numthreads, 1);
  503. LWIP_ASSERT("", sockets_stresstest_numthreads >= 0);
  504. }
  505. static int
  506. sockets_stresstest_start_clients(const struct sockaddr_storage *remote_addr)
  507. {
  508. /* limit the number of connections */
  509. const int max_connections = LWIP_MIN(TEST_MAX_CONNECTIONS, MEMP_NUM_TCP_PCB/3);
  510. int i;
  511. for (i = 0; i < max_connections; i++) {
  512. sys_thread_t t;
  513. SYS_ARCH_INC(sockets_stresstest_numthreads, 1);
  514. t = sys_thread_new("sockets_stresstest_conn_client", sockets_stresstest_conn_client, (void*)remote_addr, 0, 0);
  515. LWIP_ASSERT("thread != NULL", t != 0);
  516. }
  517. return max_connections;
  518. }
  519. static void
  520. sockets_stresstest_listener(void *arg)
  521. {
  522. int slisten;
  523. int ret;
  524. struct sockaddr_storage addr;
  525. socklen_t addr_len;
  526. struct test_settings *settings = (struct test_settings *)arg;
  527. int num_clients, num_servers = 0;
  528. slisten = lwip_socket(AF_INET, SOCK_STREAM, 0);
  529. LWIP_ASSERT("slisten >= 0", slisten >= 0);
  530. memcpy(&addr, &settings->addr, sizeof(struct sockaddr_storage));
  531. ret = lwip_bind(slisten, (struct sockaddr *)&addr, sizeof(addr));
  532. LWIP_ASSERT("ret == 0", ret == 0);
  533. ret = lwip_listen(slisten, 0);
  534. LWIP_ASSERT("ret == 0", ret == 0);
  535. addr_len = sizeof(addr);
  536. ret = lwip_getsockname(slisten, (struct sockaddr *)&addr, &addr_len);
  537. LWIP_ASSERT("ret == 0", ret == 0);
  538. num_clients = sockets_stresstest_start_clients(&addr);
  539. while (num_servers < num_clients) {
  540. struct sockaddr_storage aclient;
  541. socklen_t aclient_len = sizeof(aclient);
  542. int sclient = lwip_accept(slisten, (struct sockaddr *)&aclient, &aclient_len);
  543. #if 1
  544. /* using server threads */
  545. {
  546. sys_thread_t t;
  547. SYS_ARCH_INC(sockets_stresstest_numthreads, 1);
  548. num_servers++;
  549. t = sys_thread_new("sockets_stresstest_conn_server", sockets_stresstest_conn_server, (void*)sclient, 0, 0);
  550. LWIP_ASSERT("thread != NULL", t != 0);
  551. }
  552. #else
  553. /* using server select */
  554. #endif
  555. }
  556. LWIP_DEBUGF(TEST_SOCKETS_STRESS | LWIP_DBG_STATE, (0x1000978c, "sockets_stresstest_listener: all %d connections established\n", num_clients));
  557. /* accepted all clients */
  558. while (sockets_stresstest_numthreads > 0) {
  559. sys_msleep(1);
  560. }
  561. ret = lwip_close(slisten);
  562. LWIP_ASSERT("ret == 0", ret == 0);
  563. LWIP_DEBUGF(TEST_SOCKETS_STRESS |LWIP_DBG_STATE, (0x1000978d, "sockets_stresstest_listener: done\n"));
  564. }
  565. static void
  566. sockets_stresstest_listener_loop(void *arg)
  567. {
  568. int i;
  569. struct test_settings *settings = (struct test_settings *)arg;
  570. if (settings->loop_cnt) {
  571. for (i = 0; i < settings->loop_cnt; i++) {
  572. LWIP_DEBUGF(TEST_SOCKETS_STRESS |LWIP_DBG_STATE, (0x1000978e, "sockets_stresstest_listener_loop: iteration %d\n", i));
  573. sockets_stresstest_listener(arg);
  574. sys_msleep(2);
  575. }
  576. LWIP_DEBUGF(TEST_SOCKETS_STRESS |LWIP_DBG_STATE, (0x1000978f, "sockets_stresstest_listener_loop: done\n"));
  577. } else {
  578. for (i = 0; ; i++) {
  579. LWIP_DEBUGF(TEST_SOCKETS_STRESS |LWIP_DBG_STATE, (0x1000978e, "sockets_stresstest_listener_loop: iteration %d\n", i));
  580. sockets_stresstest_listener(arg);
  581. sys_msleep(2);
  582. }
  583. }
  584. }
  585. void
  586. sockets_stresstest_init_loopback(int addr_family)
  587. {
  588. sys_thread_t t;
  589. struct test_settings *settings = (struct test_settings *)mem_malloc(sizeof(struct test_settings));
  590. LWIP_ASSERT("OOM", settings != NULL);
  591. memset(settings, 0, sizeof(struct test_settings));
  592. #if LWIP_IPV4 && LWIP_IPV6
  593. LWIP_ASSERT("invalid addr_family", (addr_family == AF_INET) || (addr_family == AF_INET6));
  594. #endif
  595. settings->addr.ss_family = (sa_family_t)addr_family;
  596. LWIP_UNUSED_ARG(addr_family);
  597. settings->start_client = 1;
  598. t = sys_thread_new("sockets_stresstest_listener_loop", sockets_stresstest_listener_loop, settings, 0, 0);
  599. LWIP_ASSERT("thread != NULL", t != 0);
  600. }
  601. void
  602. sockets_stresstest_init_server(int addr_family, u16_t server_port)
  603. {
  604. sys_thread_t t;
  605. struct test_settings *settings = (struct test_settings *)mem_malloc(sizeof(struct test_settings));
  606. LWIP_ASSERT("OOM", settings != NULL);
  607. memset(settings, 0, sizeof(struct test_settings));
  608. #if LWIP_IPV4 && LWIP_IPV6
  609. LWIP_ASSERT("invalid addr_family", (addr_family == AF_INET) || (addr_family == AF_INET6));
  610. settings->addr.ss_family = (sa_family_t)addr_family;
  611. #endif
  612. LWIP_UNUSED_ARG(addr_family);
  613. ((struct sockaddr_in *)(&settings->addr))->sin_port = server_port;
  614. t = sys_thread_new("sockets_stresstest_listener", sockets_stresstest_listener, settings, 0, 0);
  615. LWIP_ASSERT("thread != NULL", t != 0);
  616. }
  617. void
  618. sockets_stresstest_init_client(const char *remote_ip, u16_t remote_port)
  619. {
  620. #if LWIP_IPV4
  621. ip4_addr_t ip4;
  622. #endif
  623. #if LWIP_IPV6
  624. ip6_addr_t ip6;
  625. #endif
  626. struct sockaddr_storage *addr = (struct sockaddr_storage *)mem_malloc(sizeof(struct sockaddr_storage));
  627. LWIP_ASSERT("OOM", addr != NULL);
  628. memset(addr, 0, sizeof(struct test_settings));
  629. #if LWIP_IPV4
  630. if (ip4addr_aton(remote_ip, &ip4)) {
  631. addr->ss_family = AF_INET;
  632. ((struct sockaddr_in *)addr)->sin_addr.s_addr = ip4_addr_get_u32(&ip4);
  633. }
  634. #endif
  635. #if LWIP_IPV4 && LWIP_IPV6
  636. else
  637. #endif
  638. #if LWIP_IPV6
  639. if (ip6addr_aton(remote_ip, &ip6)) {
  640. addr->ss_family = AF_INET6;
  641. /* todo: copy ipv6 address */
  642. }
  643. #endif
  644. ((struct sockaddr_in *)addr)->sin_port = remote_port;
  645. sockets_stresstest_start_clients(addr);
  646. }
  647. #endif /* LWIP_SOCKET && LWIP_IPV4 */