blob: b30dcf949b27e0dd5104e0efa35bfae4ae596a82 [file] [log] [blame]
Dave Wallace543852a2017-08-03 02:11:34 -04001/*
2 * Copyright (c) 2017 Cisco and/or its affiliates.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at:
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#include <unistd.h>
17#include <errno.h>
18#include <sys/types.h>
19#include <sys/socket.h>
20#include <stdio.h>
21#include <string.h>
22#include <time.h>
23#include <ctype.h>
Dave Wallace5c7cf1c2017-10-24 04:12:18 -040024#include <vcl/sock_test.h>
Dave Wallace35830af2017-10-09 01:43:42 -040025#include <sys/stat.h>
26#include <fcntl.h>
Dave Wallace543852a2017-08-03 02:11:34 -040027
Dave Wallacef7f809c2017-10-03 01:48:42 -040028#define SOCK_SERVER_USE_EPOLL 1
Dave Wallace35830af2017-10-09 01:43:42 -040029#define VPPCOM_SESSION_ATTR_UNIT_TEST 0
Dave Wallacef7f809c2017-10-03 01:48:42 -040030
31#if SOCK_SERVER_USE_EPOLL
32#include <sys/epoll.h>
33#endif
34
Dave Wallace35830af2017-10-09 01:43:42 -040035#ifdef VCL_TEST
36#if VPPCOM_SESSION_ATTR_UNIT_TEST
37#define BUFLEN sizeof (uint64_t) * 16
38uint64_t buffer[16];
39uint32_t buflen = BUFLEN;
40uint32_t *flags = (uint32_t *) buffer;
41#endif
42#endif
43
Dave Wallace543852a2017-08-03 02:11:34 -040044typedef struct
45{
46 uint8_t is_alloc;
47 int fd;
48 uint8_t *buf;
49 uint32_t buf_size;
50 sock_test_cfg_t cfg;
51 sock_test_stats_t stats;
52#ifdef VCL_TEST
53 vppcom_endpt_t endpt;
54 uint8_t ip[16];
55#endif
56} sock_server_conn_t;
57
58#define SOCK_SERVER_MAX_TEST_CONN 10
Dave Wallacef7f809c2017-10-03 01:48:42 -040059#define SOCK_SERVER_MAX_EPOLL_EVENTS 10
Dave Wallace543852a2017-08-03 02:11:34 -040060typedef struct
61{
62 int listen_fd;
Dave Wallacef7f809c2017-10-03 01:48:42 -040063#if SOCK_SERVER_USE_EPOLL
64 int epfd;
65 struct epoll_event listen_ev;
66 struct epoll_event wait_events[SOCK_SERVER_MAX_EPOLL_EVENTS];
67#endif
Dave Wallace543852a2017-08-03 02:11:34 -040068 size_t num_conn;
69 size_t conn_pool_size;
70 sock_server_conn_t *conn_pool;
71 int nfds;
72 fd_set rd_fdset;
73 fd_set wr_fdset;
74 struct timeval timeout;
75} sock_server_main_t;
76
77sock_server_main_t sock_server_main;
78
Dave Wallacef7f809c2017-10-03 01:48:42 -040079#if ! SOCK_SERVER_USE_EPOLL
Dave Wallace543852a2017-08-03 02:11:34 -040080static inline int
81get_nfds (void)
82{
83 sock_server_main_t *ssm = &sock_server_main;
84 int i, nfds;
85
86 for (nfds = i = 0; i < FD_SETSIZE; i++)
87 {
88 if (FD_ISSET (i, &ssm->rd_fdset) || FD_ISSET (i, &ssm->wr_fdset))
89 nfds = i + 1;
90 }
91 return nfds;
92}
93
94static inline void
95conn_fdset_set (sock_server_conn_t * conn, fd_set * fdset)
96{
97 sock_server_main_t *ssm = &sock_server_main;
98
99 FD_SET (conn->fd, fdset);
100 ssm->nfds = get_nfds ();
101}
102
103static inline void
104conn_fdset_clr (sock_server_conn_t * conn, fd_set * fdset)
105{
106 sock_server_main_t *ssm = &sock_server_main;
107
108 FD_CLR (conn->fd, fdset);
109 ssm->nfds = get_nfds ();
110}
Dave Wallacef7f809c2017-10-03 01:48:42 -0400111#endif
Dave Wallace543852a2017-08-03 02:11:34 -0400112
113static inline void
114conn_pool_expand (size_t expand_size)
115{
116 sock_server_main_t *ssm = &sock_server_main;
117 sock_server_conn_t *conn_pool;
118 size_t new_size = ssm->conn_pool_size + expand_size;
119 int i;
120
121 conn_pool = realloc (ssm->conn_pool, new_size * sizeof (*ssm->conn_pool));
122 if (conn_pool)
123 {
124 for (i = ssm->conn_pool_size; i < new_size; i++)
125 {
126 sock_server_conn_t *conn = &conn_pool[i];
127 memset (conn, 0, sizeof (*conn));
128 sock_test_cfg_init (&conn->cfg);
129 sock_test_buf_alloc (&conn->cfg, 1 /* is_rxbuf */ ,
130 &conn->buf, &conn->buf_size);
131 conn->cfg.txbuf_size = conn->cfg.rxbuf_size;
132 }
133
134 ssm->conn_pool = conn_pool;
135 ssm->conn_pool_size = new_size;
136 }
137 else
138 {
139 int errno_val = errno;
140 perror ("ERROR in conn_pool_expand()");
141 fprintf (stderr, "ERROR: Memory allocation failed (errno = %d)!\n",
142 errno_val);
143 }
144}
145
146static inline sock_server_conn_t *
147conn_pool_alloc (void)
148{
149 sock_server_main_t *ssm = &sock_server_main;
150 int i;
151
152 for (i = 0; i < ssm->conn_pool_size; i++)
153 {
154 if (!ssm->conn_pool[i].is_alloc)
155 {
156#ifdef VCL_TEST
157 ssm->conn_pool[i].endpt.ip = ssm->conn_pool[i].ip;
158#endif
159 ssm->conn_pool[i].is_alloc = 1;
160 return (&ssm->conn_pool[i]);
161 }
162 }
163
164 return 0;
165}
166
167static inline void
168conn_pool_free (sock_server_conn_t * conn)
169{
Dave Wallacef7f809c2017-10-03 01:48:42 -0400170#if ! SOCK_SERVER_USE_EPOLL
Dave Wallace543852a2017-08-03 02:11:34 -0400171 sock_server_main_t *ssm = &sock_server_main;
172
173 conn_fdset_clr (conn, &ssm->rd_fdset);
174 conn_fdset_clr (conn, &ssm->wr_fdset);
Dave Wallacef7f809c2017-10-03 01:48:42 -0400175#endif
Dave Wallace543852a2017-08-03 02:11:34 -0400176 conn->fd = 0;
177 conn->is_alloc = 0;
178}
179
180static inline void
181sync_config_and_reply (sock_server_conn_t * conn, sock_test_cfg_t * rx_cfg)
182{
183 conn->cfg = *rx_cfg;
184 sock_test_buf_alloc (&conn->cfg, 1 /* is_rxbuf */ ,
185 &conn->buf, &conn->buf_size);
186 conn->cfg.txbuf_size = conn->cfg.rxbuf_size;
187
188 if (conn->cfg.verbose)
189 {
190 printf ("\nSERVER (fd %d): Replying to cfg message!\n", conn->fd);
191 sock_test_cfg_dump (&conn->cfg, 0 /* is_client */ );
192 }
193 (void) sock_test_write (conn->fd, (uint8_t *) & conn->cfg,
194 sizeof (conn->cfg), NULL, conn->cfg.verbose);
195}
196
197static void
198stream_test_server_start_stop (sock_server_conn_t * conn,
199 sock_test_cfg_t * rx_cfg)
200{
201 sock_server_main_t *ssm = &sock_server_main;
202 int client_fd = conn->fd;
203 sock_test_t test = rx_cfg->test;
204
205 if (rx_cfg->ctrl_handle == conn->fd)
206 {
207 int i;
208 clock_gettime (CLOCK_REALTIME, &conn->stats.stop);
209
210 for (i = 0; i < ssm->conn_pool_size; i++)
211 {
212 sock_server_conn_t *tc = &ssm->conn_pool[i];
213
214 if (tc->cfg.ctrl_handle == conn->fd)
215 {
216 sock_test_stats_accumulate (&conn->stats, &tc->stats);
217
218 if (conn->cfg.verbose)
219 {
220 static char buf[64];
221
222 sprintf (buf, "SERVER (fd %d) RESULTS", tc->fd);
223 sock_test_stats_dump (buf, &tc->stats, 1 /* show_rx */ ,
224 test == SOCK_TEST_TYPE_BI
225 /* show tx */ ,
226 conn->cfg.verbose);
227 }
228 }
229 }
230
231 sock_test_stats_dump ("SERVER RESULTS", &conn->stats, 1 /* show_rx */ ,
232 (test == SOCK_TEST_TYPE_BI) /* show_tx */ ,
233 conn->cfg.verbose);
234 sock_test_cfg_dump (&conn->cfg, 0 /* is_client */ );
235 if (conn->cfg.verbose)
236 {
237 printf (" sock server main\n"
238 SOCK_TEST_SEPARATOR_STRING
239 " buf: %p\n"
240 " buf size: %u (0x%08x)\n"
241 SOCK_TEST_SEPARATOR_STRING,
242 conn->buf, conn->buf_size, conn->buf_size);
243 }
244
245 sync_config_and_reply (conn, rx_cfg);
246 printf ("\nSERVER (fd %d): %s-directional Stream Test Complete!\n"
247 SOCK_TEST_BANNER_STRING "\n", conn->fd,
248 test == SOCK_TEST_TYPE_BI ? "Bi" : "Uni");
249 }
250 else
251 {
252 printf ("\n" SOCK_TEST_BANNER_STRING
253 "SERVER (fd %d): %s-directional Stream Test!\n"
254 " Sending client the test cfg to start streaming data...\n",
255 client_fd, test == SOCK_TEST_TYPE_BI ? "Bi" : "Uni");
256
257 rx_cfg->ctrl_handle = (rx_cfg->ctrl_handle == ~0) ? conn->fd :
258 rx_cfg->ctrl_handle;
259
260 sync_config_and_reply (conn, rx_cfg);
261
262 /* read the 1st chunk, record start time */
263 memset (&conn->stats, 0, sizeof (conn->stats));
264 clock_gettime (CLOCK_REALTIME, &conn->stats.start);
265 }
266}
267
268
269static inline void
270stream_test_server (sock_server_conn_t * conn, int rx_bytes)
271{
272 int client_fd = conn->fd;
273 sock_test_t test = conn->cfg.test;
274
275 if (test == SOCK_TEST_TYPE_BI)
276 (void) sock_test_write (client_fd, conn->buf, rx_bytes, &conn->stats,
277 conn->cfg.verbose);
278
279 if (conn->stats.rx_bytes >= conn->cfg.total_bytes)
280 {
281 clock_gettime (CLOCK_REALTIME, &conn->stats.stop);
282 }
283}
284
285static inline void
286new_client (void)
287{
288 sock_server_main_t *ssm = &sock_server_main;
289 int client_fd;
290 sock_server_conn_t *conn;
291
292 if (ssm->conn_pool_size < (ssm->num_conn + SOCK_SERVER_MAX_TEST_CONN + 1))
293 conn_pool_expand (SOCK_SERVER_MAX_TEST_CONN + 1);
294
295 conn = conn_pool_alloc ();
296 if (!conn)
297 {
298 fprintf (stderr, "\nERROR: No free connections!\n");
299 return;
300 }
301
302#ifdef VCL_TEST
Dave Wallace227867f2017-11-13 21:21:53 -0500303 client_fd = vppcom_session_accept (ssm->listen_fd, &conn->endpt, 0,
Dave Wallace543852a2017-08-03 02:11:34 -0400304 -1.0 /* wait forever */ );
Dave Wallacef7f809c2017-10-03 01:48:42 -0400305 if (client_fd < 0)
306 errno = -client_fd;
Dave Wallace59179392017-11-07 02:20:07 -0500307#elif HAVE_ACCEPT4
308 client_fd = accept4 (ssm->listen_fd, (struct sockaddr *) NULL, NULL, NULL);
Dave Wallace543852a2017-08-03 02:11:34 -0400309#else
310 client_fd = accept (ssm->listen_fd, (struct sockaddr *) NULL, NULL);
311#endif
312 if (client_fd < 0)
313 {
314 int errno_val;
315 errno_val = errno;
Dave Wallacef7f809c2017-10-03 01:48:42 -0400316 perror ("ERROR in new_client()");
Dave Wallace543852a2017-08-03 02:11:34 -0400317 fprintf (stderr, "ERROR: accept failed (errno = %d)!\n", errno_val);
Dave Wallaceee45d412017-11-24 21:44:06 -0500318 return;
Dave Wallace543852a2017-08-03 02:11:34 -0400319 }
320
321 printf ("SERVER: Got a connection -- fd = %d (0x%08x)!\n",
322 client_fd, client_fd);
323
324 conn->fd = client_fd;
Dave Wallacef7f809c2017-10-03 01:48:42 -0400325
326#if ! SOCK_SERVER_USE_EPOLL
Dave Wallace543852a2017-08-03 02:11:34 -0400327 conn_fdset_set (conn, &ssm->rd_fdset);
Dave Wallacef7f809c2017-10-03 01:48:42 -0400328 ssm->nfds++;
329#else
330 {
331 struct epoll_event ev;
332 int rv;
333
334 ev.events = EPOLLIN;
335 ev.data.u64 = conn - ssm->conn_pool;
336#ifdef VCL_TEST
337 rv = vppcom_epoll_ctl (ssm->epfd, EPOLL_CTL_ADD, client_fd, &ev);
338 if (rv)
339 errno = -rv;
340#else
341 rv = epoll_ctl (ssm->epfd, EPOLL_CTL_ADD, client_fd, &ev);
342#endif
343 if (rv < 0)
344 {
345 int errno_val;
346 errno_val = errno;
347 perror ("ERROR in new_client()");
348 fprintf (stderr, "ERROR: epoll_ctl failed (errno = %d)!\n",
349 errno_val);
350 }
351 else
352 ssm->nfds++;
353 }
354#endif
Dave Wallace543852a2017-08-03 02:11:34 -0400355}
356
357int
358main (int argc, char **argv)
359{
360 sock_server_main_t *ssm = &sock_server_main;
361 int client_fd, rv, main_rv = 0;
362 int tx_bytes, rx_bytes, nbytes;
363 sock_server_conn_t *conn;
364 sock_test_cfg_t *rx_cfg;
365 uint32_t xtra = 0;
366 uint64_t xtra_bytes = 0;
367 struct sockaddr_in servaddr;
368 int errno_val;
369 int v, i;
370 uint16_t port = SOCK_TEST_SERVER_PORT;
Dave Wallacef7f809c2017-10-03 01:48:42 -0400371#if ! SOCK_SERVER_USE_EPOLL
Dave Wallace543852a2017-08-03 02:11:34 -0400372 fd_set _rfdset, *rfdset = &_rfdset;
Dave Wallacef7f809c2017-10-03 01:48:42 -0400373#endif
Dave Wallace543852a2017-08-03 02:11:34 -0400374#ifdef VCL_TEST
375 vppcom_endpt_t endpt;
376#else
Dave Wallacef7f809c2017-10-03 01:48:42 -0400377#if ! SOCK_SERVER_USE_EPOLL
Dave Wallace543852a2017-08-03 02:11:34 -0400378 fd_set _wfdset, *wfdset = &_wfdset;
379#endif
Dave Wallacef7f809c2017-10-03 01:48:42 -0400380#endif
Dave Wallace543852a2017-08-03 02:11:34 -0400381
382 if ((argc == 2) && (sscanf (argv[1], "%d", &v) == 1))
383 port = (uint16_t) v;
384
385 conn_pool_expand (SOCK_SERVER_MAX_TEST_CONN + 1);
386
387#ifdef VCL_TEST
388 rv = vppcom_app_create ("vcl_test_server");
389 if (rv)
390 {
391 errno = -rv;
392 ssm->listen_fd = -1;
393 }
394 else
395 {
396 ssm->listen_fd =
397 vppcom_session_create (VPPCOM_VRF_DEFAULT, VPPCOM_PROTO_TCP,
398 0 /* is_nonblocking */ );
399 }
400#else
401 ssm->listen_fd = socket (AF_INET, SOCK_STREAM, 0);
402#endif
403 if (ssm->listen_fd < 0)
404 {
405 errno_val = errno;
406 perror ("ERROR in main()");
407 fprintf (stderr, "ERROR: socket() failed (errno = %d)!\n", errno_val);
408 return ssm->listen_fd;
409 }
410
411 memset (&servaddr, 0, sizeof (servaddr));
412
413 servaddr.sin_family = AF_INET;
Dave Wallace33e002b2017-09-06 01:20:02 -0400414 servaddr.sin_addr.s_addr = htonl (INADDR_ANY);
Dave Wallace543852a2017-08-03 02:11:34 -0400415 servaddr.sin_port = htons (port);
416
417#ifdef VCL_TEST
418 endpt.vrf = VPPCOM_VRF_DEFAULT;
419 endpt.is_ip4 = (servaddr.sin_family == AF_INET);
420 endpt.ip = (uint8_t *) & servaddr.sin_addr;
421 endpt.port = (uint16_t) servaddr.sin_port;
422
423 rv = vppcom_session_bind (ssm->listen_fd, &endpt);
424 if (rv)
425 {
426 errno = -rv;
427 rv = -1;
428 }
Dave Wallace35830af2017-10-09 01:43:42 -0400429
430#if VPPCOM_SESSION_ATTR_UNIT_TEST
431 buflen = BUFLEN;
432 if (vppcom_session_attr (ssm->listen_fd, VPPCOM_ATTR_GET_FLAGS,
433 buffer, &buflen) != VPPCOM_OK)
434 printf ("\nGET_FLAGS0: Oh no, Mr. Biiiiiiiiiiiilllllll ! ! ! !\n");
435 buflen = BUFLEN;
436 *flags = O_RDWR | O_NONBLOCK;
437 if (vppcom_session_attr (ssm->listen_fd, VPPCOM_ATTR_SET_FLAGS,
438 buffer, &buflen) != VPPCOM_OK)
439 printf ("\nSET_FLAGS1: Oh no, Mr. Biiiiiiiiiiiilllllll ! ! ! !\n");
440 buflen = BUFLEN;
441 if (vppcom_session_attr (ssm->listen_fd, VPPCOM_ATTR_GET_FLAGS,
442 buffer, &buflen) != VPPCOM_OK)
443 printf ("\nGET_FLAGS1:Oh no, Mr. Biiiiiiiiiiiilllllll ! ! ! !\n");
444 *flags = O_RDWR;
445 buflen = BUFLEN;
446 if (vppcom_session_attr (ssm->listen_fd, VPPCOM_ATTR_SET_FLAGS,
447 buffer, &buflen) != VPPCOM_OK)
448 printf ("\nSET_FLAGS2: Oh no, Mr. Biiiiiiiiiiiilllllll ! ! ! !\n");
449 buflen = BUFLEN;
450 if (vppcom_session_attr (ssm->listen_fd, VPPCOM_ATTR_GET_FLAGS,
451 buffer, &buflen) != VPPCOM_OK)
452 printf ("\nGET_FLAGS2:Oh no, Mr. Biiiiiiiiiiiilllllll ! ! ! !\n");
Dave Wallace35830af2017-10-09 01:43:42 -0400453#endif
Dave Wallace543852a2017-08-03 02:11:34 -0400454#else
455 rv =
456 bind (ssm->listen_fd, (struct sockaddr *) &servaddr, sizeof (servaddr));
457#endif
458 if (rv < 0)
459 {
460 errno_val = errno;
461 perror ("ERROR in main()");
462 fprintf (stderr, "ERROR: bind failed (errno = %d)!\n", errno_val);
463 return rv;
464 }
465
466#ifdef VCL_TEST
467 rv = vppcom_session_listen (ssm->listen_fd, 10);
468 if (rv)
469 {
470 errno = -rv;
471 rv = -1;
472 }
473#else
474 rv = listen (ssm->listen_fd, 10);
475#endif
476 if (rv < 0)
477 {
478 errno_val = errno;
479 perror ("ERROR in main()");
480 fprintf (stderr, "ERROR: listen failed (errno = %d)!\n", errno_val);
481 return rv;
482 }
483
Dave Wallacef7f809c2017-10-03 01:48:42 -0400484 printf ("\nSERVER: Waiting for a client to connect on port %d...\n", port);
485
486#if ! SOCK_SERVER_USE_EPOLL
487
Dave Wallace543852a2017-08-03 02:11:34 -0400488 FD_ZERO (&ssm->wr_fdset);
489 FD_ZERO (&ssm->rd_fdset);
490
491 FD_SET (ssm->listen_fd, &ssm->rd_fdset);
492 ssm->nfds = ssm->listen_fd + 1;
493
Dave Wallacef7f809c2017-10-03 01:48:42 -0400494#else
495#ifdef VCL_TEST
496 ssm->epfd = vppcom_epoll_create ();
497 if (ssm->epfd < 0)
498 errno = -ssm->epfd;
499#else
500 ssm->epfd = epoll_create (1);
501#endif
502 if (ssm->epfd < 0)
503 {
504 errno_val = errno;
505 perror ("ERROR in main()");
506 fprintf (stderr, "ERROR: epoll_create failed (errno = %d)!\n",
507 errno_val);
508 return ssm->epfd;
509 }
510
511 ssm->listen_ev.events = EPOLLIN;
512 ssm->listen_ev.data.u32 = ~0;
513#ifdef VCL_TEST
514 rv = vppcom_epoll_ctl (ssm->epfd, EPOLL_CTL_ADD, ssm->listen_fd,
515 &ssm->listen_ev);
516 if (rv < 0)
517 errno = -rv;
518#else
519 rv = epoll_ctl (ssm->epfd, EPOLL_CTL_ADD, ssm->listen_fd, &ssm->listen_ev);
520#endif
521 if (rv < 0)
522 {
523 errno_val = errno;
524 perror ("ERROR in main()");
525 fprintf (stderr, "ERROR: epoll_ctl failed (errno = %d)!\n", errno_val);
526 return rv;
527 }
528#endif
Dave Wallace543852a2017-08-03 02:11:34 -0400529
530 while (1)
531 {
Dave Wallacef7f809c2017-10-03 01:48:42 -0400532#if ! SOCK_SERVER_USE_EPOLL
Dave Wallace543852a2017-08-03 02:11:34 -0400533 _rfdset = ssm->rd_fdset;
534
535#ifdef VCL_TEST
536 rv = vppcom_select (ssm->nfds, (uint64_t *) rfdset, NULL, NULL, 0);
537#else
538 {
539 struct timeval timeout;
540 timeout = ssm->timeout;
541 _wfdset = ssm->wr_fdset;
542 rv = select (ssm->nfds, rfdset, wfdset, NULL, &timeout);
543 }
544#endif
545 if (rv < 0)
546 {
547 perror ("select()");
548 fprintf (stderr, "\nERROR: select() failed -- aborting!\n");
549 main_rv = -1;
550 goto done;
551 }
552 else if (rv == 0)
553 continue;
554
555 if (FD_ISSET (ssm->listen_fd, rfdset))
556 new_client ();
557
558 for (i = 0; i < ssm->conn_pool_size; i++)
559 {
560 if (!ssm->conn_pool[i].is_alloc)
561 continue;
562
563 conn = &ssm->conn_pool[i];
Dave Wallacef7f809c2017-10-03 01:48:42 -0400564#else
565 int num_ev;
566#ifdef VCL_TEST
567 num_ev = vppcom_epoll_wait (ssm->epfd, ssm->wait_events,
568 SOCK_SERVER_MAX_EPOLL_EVENTS, 60.0);
Dave Wallace0fd6ad62017-10-20 13:23:40 -0400569 if (num_ev < 0)
570 errno = -num_ev;
Dave Wallacef7f809c2017-10-03 01:48:42 -0400571#else
572 num_ev = epoll_wait (ssm->epfd, ssm->wait_events,
573 SOCK_SERVER_MAX_EPOLL_EVENTS, 60000);
574#endif
575 if (num_ev < 0)
576 {
577 perror ("epoll_wait()");
578 fprintf (stderr, "\nERROR: epoll_wait() failed -- aborting!\n");
579 main_rv = -1;
580 goto done;
581 }
582 if (num_ev == 0)
583 {
584 fprintf (stderr, "\nepoll_wait() timeout!\n");
585 continue;
586 }
587 for (i = 0; i < num_ev; i++)
588 {
589 if (ssm->wait_events[i].data.u32 == ~0)
590 {
591 new_client ();
592 continue;
593 }
594 conn = &ssm->conn_pool[ssm->wait_events[i].data.u32];
595#endif
Dave Wallace543852a2017-08-03 02:11:34 -0400596 client_fd = conn->fd;
597
Dave Wallacef7f809c2017-10-03 01:48:42 -0400598#if ! SOCK_SERVER_USE_EPOLL
Dave Wallace543852a2017-08-03 02:11:34 -0400599 if (FD_ISSET (client_fd, rfdset))
Dave Wallacef7f809c2017-10-03 01:48:42 -0400600#else
601 if (EPOLLIN & ssm->wait_events[i].events)
602#endif
Dave Wallace543852a2017-08-03 02:11:34 -0400603 {
Dave Wallace35830af2017-10-09 01:43:42 -0400604#ifdef VCL_TEST
605#if VPPCOM_SESSION_ATTR_UNIT_TEST
Dave Wallace9c5161a2017-12-14 21:36:31 -0500606 {
607 vppcom_endpt_t ep;
608 uint8_t addr[16];
609
610 ep.ip = addr;
611 buflen = BUFLEN;
612 if (vppcom_session_attr (client_fd, VPPCOM_ATTR_GET_NREAD,
613 buffer, &buflen) < VPPCOM_OK)
614 printf ("\nNREAD: Oh no, Mr. "
615 "Biiiiiiiiiiiilllllll ! ! ! !\n");
616 buflen = sizeof (ep);
617 if (vppcom_session_attr (client_fd,
618 VPPCOM_ATTR_GET_PEER_ADDR,
619 &ep, &buflen) != VPPCOM_OK)
620 printf ("\nGET_PEER_ADDR: Oh no, Mr. "
621 "Biiiiiiiiiiiilllllll ! ! ! !\n");
622 buflen = sizeof (ep);
623 if (vppcom_session_attr (client_fd, VPPCOM_ATTR_GET_LCL_ADDR,
624 &ep, &buflen) != VPPCOM_OK)
625 printf ("\nGET_LCL_ADDR: Oh no, Mr. "
626 "Biiiiiiiiiiiilllllll ! ! ! !\n");
627 }
Dave Wallace35830af2017-10-09 01:43:42 -0400628#endif
629#endif
Dave Wallace543852a2017-08-03 02:11:34 -0400630 rx_bytes = sock_test_read (client_fd, conn->buf,
631 conn->buf_size, &conn->stats);
632 if (rx_bytes > 0)
633 {
634 rx_cfg = (sock_test_cfg_t *) conn->buf;
635 if (rx_cfg->magic == SOCK_TEST_CFG_CTRL_MAGIC)
636 {
637 if (rx_cfg->verbose)
638 {
639 printf ("SERVER (fd %d): Received a cfg message!\n",
640 client_fd);
641 sock_test_cfg_dump (rx_cfg, 0 /* is_client */ );
642 }
643
644 if (rx_bytes != sizeof (*rx_cfg))
645 {
646 printf ("SERVER (fd %d): Invalid cfg message "
647 "size (%d)!\n Should be %lu bytes.\n",
648 client_fd, rx_bytes, sizeof (*rx_cfg));
649 conn->cfg.rxbuf_size = 0;
650 conn->cfg.num_writes = 0;
651 if (conn->cfg.verbose)
652 {
653 printf ("SERVER (fd %d): Replying to "
654 "cfg message!\n", client_fd);
655 sock_test_cfg_dump (rx_cfg, 0 /* is_client */ );
656 }
657 sock_test_write (client_fd, (uint8_t *) & conn->cfg,
658 sizeof (conn->cfg), NULL,
659 conn->cfg.verbose);
660 continue;
661 }
662
663 switch (rx_cfg->test)
664 {
665 case SOCK_TEST_TYPE_NONE:
666 case SOCK_TEST_TYPE_ECHO:
667 sync_config_and_reply (conn, rx_cfg);
668 break;
669
670 case SOCK_TEST_TYPE_BI:
671 case SOCK_TEST_TYPE_UNI:
672 stream_test_server_start_stop (conn, rx_cfg);
673 break;
674
675 case SOCK_TEST_TYPE_EXIT:
676 printf ("SERVER: Have a great day, "
677 "connection %d!\n", client_fd);
678#ifdef VCL_TEST
679 vppcom_session_close (client_fd);
680#else
681 close (client_fd);
682#endif
683 conn_pool_free (conn);
Dave Wallaceee45d412017-11-24 21:44:06 -0500684 printf ("SERVER: Closed client fd %d\n", client_fd);
Dave Wallacef7f809c2017-10-03 01:48:42 -0400685#if ! SOCK_SERVER_USE_EPOLL
Dave Wallace543852a2017-08-03 02:11:34 -0400686 if (ssm->nfds == (ssm->listen_fd + 1))
Dave Wallacef7f809c2017-10-03 01:48:42 -0400687#else
688 ssm->nfds--;
689 if (!ssm->nfds)
690#endif
Dave Wallace543852a2017-08-03 02:11:34 -0400691 {
692 printf ("SERVER: All client connections "
693 "closed.\n\nSERVER: "
694 "May the force be with you!\n\n");
695 goto done;
696 }
697 break;
698
699 default:
700 fprintf (stderr, "ERROR: Unknown test type!\n");
701 sock_test_cfg_dump (rx_cfg, 0 /* is_client */ );
702 break;
703 }
704 continue;
705 }
706
707 else if ((conn->cfg.test == SOCK_TEST_TYPE_UNI) ||
708 (conn->cfg.test == SOCK_TEST_TYPE_BI))
709 {
710 stream_test_server (conn, rx_bytes);
711 continue;
712 }
713
Chris Luke879ace32017-09-26 13:15:16 -0400714 else if (isascii (conn->buf[0]))
Chris Lukeab7b8d92017-09-07 07:40:13 -0400715 {
716 // If it looks vaguely like a string, make sure it's terminated
717 ((char *) conn->buf)[rx_bytes <
718 conn->buf_size ? rx_bytes :
719 conn->buf_size - 1] = 0;
Dave Wallacef7f809c2017-10-03 01:48:42 -0400720 printf ("SERVER (fd %d): RX (%d bytes) - '%s'\n",
Chris Lukeab7b8d92017-09-07 07:40:13 -0400721 conn->fd, rx_bytes, conn->buf);
722 }
Dave Wallace543852a2017-08-03 02:11:34 -0400723 }
724 else // rx_bytes < 0
725 {
726 if (errno == ECONNRESET)
727 {
728 printf ("\nSERVER: Connection reset by remote peer.\n"
729 " Y'all have a great day now!\n\n");
730 break;
731 }
732 else
733 continue;
734 }
735
Chris Luke879ace32017-09-26 13:15:16 -0400736 if (isascii (conn->buf[0]))
Dave Wallace543852a2017-08-03 02:11:34 -0400737 {
Chris Luke879ace32017-09-26 13:15:16 -0400738 // If it looks vaguely like a string, make sure it's terminated
739 ((char *) conn->buf)[rx_bytes <
740 conn->buf_size ? rx_bytes :
741 conn->buf_size - 1] = 0;
Dave Wallace543852a2017-08-03 02:11:34 -0400742 if (xtra)
743 fprintf (stderr,
744 "ERROR: FIFO not drained in previous test!\n"
745 " extra chunks %u (0x%x)\n"
746 " extra bytes %lu (0x%lx)\n",
747 xtra, xtra, xtra_bytes, xtra_bytes);
748
749 xtra = 0;
750 xtra_bytes = 0;
751
752 if (conn->cfg.verbose)
753 printf ("SERVER (fd %d): Echoing back\n", client_fd);
754
755 nbytes = strlen ((const char *) conn->buf) + 1;
756
757 tx_bytes = sock_test_write (client_fd, conn->buf,
758 nbytes, &conn->stats,
759 conn->cfg.verbose);
760 if (tx_bytes >= 0)
761 printf ("SERVER (fd %d): TX (%d bytes) - '%s'\n",
762 conn->fd, tx_bytes, conn->buf);
763 }
764
765 else // Extraneous read data from non-echo tests???
766 {
767 xtra++;
768 xtra_bytes += rx_bytes;
769 }
770 }
771 }
772 }
773
774done:
775#ifdef VCL_TEST
776 vppcom_session_close (ssm->listen_fd);
777 vppcom_app_destroy ();
778#else
779 close (ssm->listen_fd);
780#endif
781 if (ssm->conn_pool)
782 free (ssm->conn_pool);
783
784 return main_rv;
785}
786
787/*
788 * fd.io coding-style-patch-verification: ON
789 *
790 * Local Variables:
791 * eval: (c-set-style "gnu")
792 * End:
793 */