2 * Copyright (c) 2017-2021 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:
7 * http://www.apache.org/licenses/LICENSE-2.0
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.
16 #ifndef __vcl_test_h__
17 #define __vcl_test_h__
24 #include <vcl/vppcom.h>
26 #define vtfail(_fn, _rv) \
29 perror ("ERROR when calling " _fn); \
30 fprintf (stderr, "\nERROR: " _fn " failed (errno = %d)!\n", -_rv); \
34 #define vterr(_fn, _rv) \
37 fprintf (stderr, "\nERROR: " _fn " failed (errno = %d)!\n", -_rv); \
40 #define vtwrn(_fmt, _args...) \
41 fprintf (stderr, "\nERROR: " _fmt "\n", ##_args) \
43 #define vtinf(_fmt, _args...) \
44 fprintf (stdout, "vt<w%u>: " _fmt "\n", __wrk_index, ##_args)
46 #define vt_atomic_add(_ptr, _val) \
47 __atomic_fetch_add (_ptr, _val, __ATOMIC_RELEASE)
49 #define VCL_TEST_TOKEN_HELP "#H"
50 #define VCL_TEST_TOKEN_EXIT "#X"
51 #define VCL_TEST_TOKEN_VERBOSE "#V"
52 #define VCL_TEST_TOKEN_TXBUF_SIZE "#T:"
53 #define VCL_TEST_TOKEN_NUM_TEST_SESS "#I:"
54 #define VCL_TEST_TOKEN_NUM_WRITES "#N:"
55 #define VCL_TEST_TOKEN_RXBUF_SIZE "#R:"
56 #define VCL_TEST_TOKEN_SHOW_CFG "#C"
57 #define VCL_TEST_TOKEN_RUN_UNI "#U"
58 #define VCL_TEST_TOKEN_RUN_BI "#B"
60 #define VCL_TEST_SERVER_PORT 22000
61 #define VCL_TEST_LOCALHOST_IPADDR "127.0.0.1"
63 #define VCL_TEST_CFG_CTRL_MAGIC 0xfeedface
64 #define VCL_TEST_CFG_NUM_WRITES_DEF 1000000
65 #define VCL_TEST_CFG_TXBUF_SIZE_DEF 8192
66 #define VCL_TEST_CFG_RXBUF_SIZE_DEF (64*VCL_TEST_CFG_TXBUF_SIZE_DEF)
67 #define VCL_TEST_CFG_BUF_SIZE_MIN 128
68 #define VCL_TEST_CFG_MAX_TEST_SESS ((uint32_t) 1e6)
69 #define VCL_TEST_CFG_MAX_SELECT_SESS 512
70 #define VCL_TEST_CFG_INIT_TEST_SESS 512
71 #define VCL_TEST_CFG_MAX_EPOLL_EVENTS 16
73 #define VCL_TEST_CTRL_LISTENER (~0 - 1)
74 #define VCL_TEST_DATA_LISTENER (~0)
75 #define VCL_TEST_DELAY_DISCONNECT 1
76 #define VCL_TEST_SEPARATOR_STRING \
77 " -----------------------------\n"
85 VCL_TEST_TYPE_EXIT_CLIENT,
95 typedef struct __attribute__ ((packed))
101 uint32_t ctrl_handle;
102 uint32_t num_test_sessions;
103 uint32_t num_test_sessions_perq;
104 uint32_t num_test_qsessions;
106 uint32_t address_ip6;
107 uint32_t transport_udp;
111 uint64_t total_bytes;
124 struct timespec start;
125 struct timespec stop;
128 typedef struct vcl_test_session
131 uint8_t is_alloc : 1;
133 uint8_t noblk_connect : 1;
135 int (*read) (struct vcl_test_session *ts, void *buf, uint32_t buflen);
136 int (*write) (struct vcl_test_session *ts, void *buf, uint32_t buflen);
142 vcl_test_stats_t stats;
143 vcl_test_stats_t old_stats;
145 struct vcl_test_session *next;
146 vppcom_endpt_t endpt;
148 vppcom_data_segment_t ds[2];
150 } vcl_test_session_t;
152 static __thread int __wrk_index = 0;
155 vcl_test_worker_index (void)
162 int (*init) (vcl_test_cfg_t *cfg);
163 int (*open) (vcl_test_session_t *ts, vppcom_endpt_t *endpt);
164 int (*listen) (vcl_test_session_t *ts, vppcom_endpt_t *endpt);
165 int (*accept) (int listen_fd, vcl_test_session_t *ts);
166 int (*close) (vcl_test_session_t *ts);
167 } vcl_test_proto_vft_t;
171 vcl_test_session_t *qsessions;
172 uint32_t n_qsessions;
178 const vcl_test_proto_vft_t *protos[VPPCOM_PROTO_SRTP + 1];
179 uint32_t ckpair_index;
184 extern vcl_test_main_t vcl_test_main;
186 #define VCL_TEST_REGISTER_PROTO(proto, vft) \
187 static void __attribute__ ((constructor)) vcl_test_init_##proto (void) \
189 vcl_test_main.protos[proto] = &vft; \
193 vcl_test_stats_accumulate (vcl_test_stats_t * accum, vcl_test_stats_t * incr)
195 accum->rx_xacts += incr->rx_xacts;
196 accum->rx_bytes += incr->rx_bytes;
197 accum->rx_eagain += incr->rx_eagain;
198 accum->rx_incomp += incr->rx_incomp;
199 accum->tx_xacts += incr->tx_xacts;
200 accum->tx_bytes += incr->tx_bytes;
201 accum->tx_eagain += incr->tx_eagain;
202 accum->tx_incomp += incr->tx_incomp;
206 vcl_test_cfg_init (vcl_test_cfg_t * cfg)
208 cfg->magic = VCL_TEST_CFG_CTRL_MAGIC;
209 cfg->test = VCL_TEST_TYPE_UNI;
210 cfg->ctrl_handle = ~0;
211 cfg->num_test_sessions = 1;
212 cfg->num_test_sessions_perq = 1;
214 cfg->rxbuf_size = VCL_TEST_CFG_RXBUF_SIZE_DEF;
215 cfg->num_writes = VCL_TEST_CFG_NUM_WRITES_DEF;
216 cfg->txbuf_size = VCL_TEST_CFG_TXBUF_SIZE_DEF;
217 cfg->total_bytes = cfg->num_writes * cfg->txbuf_size;
221 vcl_test_cfg_verify (vcl_test_cfg_t * cfg, vcl_test_cfg_t * valid_cfg)
223 /* Note: txbuf & rxbuf on server are the same buffer,
224 * so txbuf_size is not included in this check.
226 return ((cfg->magic == valid_cfg->magic)
227 && (cfg->test == valid_cfg->test)
228 && (cfg->verbose == valid_cfg->verbose)
229 && (cfg->rxbuf_size == valid_cfg->rxbuf_size)
230 && (cfg->num_writes == valid_cfg->num_writes)
231 && (cfg->total_bytes == valid_cfg->total_bytes));
235 vcl_test_buf_alloc (vcl_test_cfg_t * cfg, uint8_t is_rxbuf, uint8_t ** buf,
238 uint32_t alloc_size = is_rxbuf ? cfg->rxbuf_size : cfg->txbuf_size;
239 uint8_t *lb = realloc (*buf, (size_t) alloc_size);
244 cfg->rxbuf_size = *bufsize = alloc_size;
246 cfg->txbuf_size = *bufsize = alloc_size;
252 vtwrn ("realloc failed. using buffer size %d instead of %u",
253 *bufsize, alloc_size);
258 vcl_test_session_buf_alloc (vcl_test_session_t *ts)
260 ts->rxbuf_size = ts->cfg.rxbuf_size;
261 ts->txbuf_size = ts->cfg.txbuf_size;
262 vcl_test_buf_alloc (&ts->cfg, 0 /* is_rxbuf */, (uint8_t **) &ts->txbuf,
264 vcl_test_buf_alloc (&ts->cfg, 1 /* is_rxbuf */, (uint8_t **) &ts->rxbuf,
269 vcl_test_session_buf_free (vcl_test_session_t *ts)
278 vcl_test_type_str (vcl_test_t t)
282 case VCL_TEST_TYPE_NONE:
285 case VCL_TEST_TYPE_ECHO:
288 case VCL_TEST_TYPE_UNI:
291 case VCL_TEST_TYPE_BI:
294 case VCL_TEST_TYPE_EXIT:
303 vcl_test_cfg_dump (vcl_test_cfg_t * cfg, uint8_t is_client)
307 printf (" test config (%p):\n"
308 VCL_TEST_SEPARATOR_STRING
311 "%-5s test: %s (%d)\n"
312 " ctrl handle: %d (0x%x)\n"
313 "%-5s num test sockets: %u (0x%08x)\n"
314 "%-5s verbose: %s (%d)\n"
315 "%-5s rxbuf size: %lu (0x%08lx)\n"
316 "%-5s txbuf size: %lu (0x%08lx)\n"
317 "%-5s num writes: %lu (0x%08lx)\n"
318 " client tx bytes: %lu (0x%08lx)\n"
319 VCL_TEST_SEPARATOR_STRING,
320 (void *) cfg, cfg->magic, cfg->seq_num,
321 is_client && (cfg->test == VCL_TEST_TYPE_UNI) ?
322 "'" VCL_TEST_TOKEN_RUN_UNI "'" :
323 is_client && (cfg->test == VCL_TEST_TYPE_BI) ?
324 "'" VCL_TEST_TOKEN_RUN_BI "'" : spc,
325 vcl_test_type_str (cfg->test), cfg->test,
326 cfg->ctrl_handle, cfg->ctrl_handle,
327 is_client ? "'" VCL_TEST_TOKEN_NUM_TEST_SESS "'" : spc,
328 cfg->num_test_sessions, cfg->num_test_sessions,
329 is_client ? "'" VCL_TEST_TOKEN_VERBOSE "'" : spc,
330 cfg->verbose ? "on" : "off", cfg->verbose,
331 is_client ? "'" VCL_TEST_TOKEN_RXBUF_SIZE "'" : spc,
332 cfg->rxbuf_size, cfg->rxbuf_size,
333 is_client ? "'" VCL_TEST_TOKEN_TXBUF_SIZE "'" : spc,
334 cfg->txbuf_size, cfg->txbuf_size,
335 is_client ? "'" VCL_TEST_TOKEN_NUM_WRITES "'" : spc,
336 cfg->num_writes, cfg->num_writes,
337 cfg->total_bytes, cfg->total_bytes);
341 vcl_test_stats_dump (char *header, vcl_test_stats_t * stats,
342 uint8_t show_rx, uint8_t show_tx, uint8_t verbose)
344 struct timespec diff;
345 double duration, rate;
346 uint64_t total_bytes;
348 if ((stats->stop.tv_nsec - stats->start.tv_nsec) < 0)
350 diff.tv_sec = stats->stop.tv_sec - stats->start.tv_sec - 1;
351 diff.tv_nsec = stats->stop.tv_nsec - stats->start.tv_nsec + 1e9;
355 diff.tv_sec = stats->stop.tv_sec - stats->start.tv_sec;
356 diff.tv_nsec = stats->stop.tv_nsec - stats->start.tv_nsec;
358 duration = (double) diff.tv_sec + (1e-9 * diff.tv_nsec);
360 total_bytes = stats->tx_bytes + stats->rx_bytes;
361 rate = (double) total_bytes *8 / duration / 1e9;
362 printf ("\n%s: Streamed %lu bytes\n"
363 " in %lf seconds (%lf Gbps %s-duplex)!\n",
364 header, total_bytes, duration, rate,
365 (show_rx && show_tx) ? "full" : "half");
369 printf (VCL_TEST_SEPARATOR_STRING
370 " tx stats (0x%p):\n"
371 VCL_TEST_SEPARATOR_STRING
372 " writes: %lu (0x%08lx)\n"
373 " tx bytes: %lu (0x%08lx)\n"
374 " tx eagain: %u (0x%08x)\n"
375 " tx incomplete: %u (0x%08x)\n",
376 (void *) stats, stats->tx_xacts, stats->tx_xacts,
377 stats->tx_bytes, stats->tx_bytes,
378 stats->tx_eagain, stats->tx_eagain,
379 stats->tx_incomp, stats->tx_incomp);
383 printf (VCL_TEST_SEPARATOR_STRING
384 " rx stats (0x%p):\n"
385 VCL_TEST_SEPARATOR_STRING
386 " reads: %lu (0x%08lx)\n"
387 " rx bytes: %lu (0x%08lx)\n"
388 " rx eagain: %u (0x%08x)\n"
389 " rx incomplete: %u (0x%08x)\n",
390 (void *) stats, stats->rx_xacts, stats->rx_xacts,
391 stats->rx_bytes, stats->rx_bytes,
392 stats->rx_eagain, stats->rx_eagain,
393 stats->rx_incomp, stats->rx_incomp);
396 printf (" start.tv_sec: %ld\n"
397 " start.tv_nsec: %ld\n"
398 " stop.tv_sec: %ld\n"
399 " stop.tv_nsec: %ld\n",
400 stats->start.tv_sec, stats->start.tv_nsec,
401 stats->stop.tv_sec, stats->stop.tv_nsec);
403 printf (VCL_TEST_SEPARATOR_STRING);
407 vcl_test_time_diff (struct timespec *old, struct timespec *new)
410 if ((new->tv_nsec - old->tv_nsec) < 0)
412 sec = new->tv_sec - old->tv_sec - 1;
413 nsec = new->tv_nsec - old->tv_nsec + 1e9;
417 sec = new->tv_sec - old->tv_sec;
418 nsec = new->tv_nsec - old->tv_nsec;
420 return (double) sec + (1e-9 * nsec);
424 vcl_test_stats_dump_inc (vcl_test_session_t *ts, int is_rx)
426 vcl_test_stats_t *old, *new;
427 double duration, rate;
428 uint64_t total_bytes;
431 old = &ts->old_stats;
433 duration = vcl_test_time_diff (&old->stop, &new->stop);
437 total_bytes = new->rx_bytes - old->rx_bytes;
438 dir_str = "Received";
442 total_bytes = new->tx_bytes - old->tx_bytes;
446 rate = (double) total_bytes * 8 / duration / 1e9;
447 printf ("%d: %s %lu Mbytes in %.2lf seconds %.2lf Gbps\n", ts->fd, dir_str,
448 (uint64_t) (total_bytes / 1e6), duration, rate);
452 vcl_comp_tspec (struct timespec *a, struct timespec *b)
454 if (a->tv_sec < b->tv_sec)
456 else if (a->tv_sec > b->tv_sec)
458 else if (a->tv_nsec < b->tv_nsec)
460 else if (a->tv_nsec > b->tv_nsec)
467 vcl_test_read (vcl_test_session_t *ts, void *buf, uint32_t nbytes)
469 vcl_test_stats_t *stats = &ts->stats;
470 int rv, rx_bytes = 0;
475 rv = vppcom_session_read (ts->fd, buf, nbytes);
479 if (errno == EAGAIN || errno == EWOULDBLOCK)
485 vterr ("vppcom_session_read()", -errno);
495 stats->rx_bytes += rx_bytes;
501 vcl_test_read_ds (vcl_test_session_t *ts)
503 vcl_test_stats_t *stats = &ts->stats;
509 rx_bytes = vppcom_session_read_segments (ts->fd, ts->ds, 2, ~0);
516 if ((rx_bytes == 0) ||
518 && ((errno == EAGAIN) || (errno == EWOULDBLOCK))))
521 while ((rx_bytes == 0) ||
522 ((rx_bytes < 0) && ((errno == EAGAIN) || (errno == EWOULDBLOCK))));
526 vterr ("vppcom_session_read()", -errno);
529 stats->rx_bytes += rx_bytes;
535 vcl_test_write (vcl_test_session_t *ts, void *buf, uint32_t nbytes)
537 int tx_bytes = 0, nbytes_left = nbytes, rv;
538 vcl_test_stats_t *stats = &ts->stats;
543 rv = vppcom_session_write (ts->fd, buf, nbytes_left);
547 if ((errno == EAGAIN || errno == EWOULDBLOCK))
553 nbytes_left = nbytes_left - rv;
555 if (rv < nbytes_left)
558 while (tx_bytes != nbytes);
562 vterr ("vpcom_session_write", -errno);
565 stats->tx_bytes += tx_bytes;
575 printf ("CLIENT: Test configuration commands:"
576 INDENT VCL_TEST_TOKEN_HELP
577 "\t\t\tDisplay help."
578 INDENT VCL_TEST_TOKEN_EXIT
579 "\t\t\tExit test client & server."
580 INDENT VCL_TEST_TOKEN_SHOW_CFG
581 "\t\t\tShow the current test cfg."
582 INDENT VCL_TEST_TOKEN_RUN_UNI
583 "\t\t\tRun the Uni-directional test."
584 INDENT VCL_TEST_TOKEN_RUN_BI
585 "\t\t\tRun the Bi-directional test."
586 INDENT VCL_TEST_TOKEN_VERBOSE
587 "\t\t\tToggle verbose setting."
588 INDENT VCL_TEST_TOKEN_RXBUF_SIZE
589 "<rxbuf size>\tRx buffer size (bytes)."
590 INDENT VCL_TEST_TOKEN_TXBUF_SIZE
591 "<txbuf size>\tTx buffer size (bytes)."
592 INDENT VCL_TEST_TOKEN_NUM_WRITES
593 "<# of writes>\tNumber of txbuf writes to server." "\n");
596 #endif /* __vcl_test_h__ */
599 * fd.io coding-style-patch-verification: ON
602 * eval: (c-set-style "gnu")