xref: /iperf/src/iperf_api.c (revision abb24a2e)
1 /*
2  * iperf, Copyright (c) 2014-2022, The Regents of the University of
3  * California, through Lawrence Berkeley National Laboratory (subject
4  * to receipt of any required approvals from the U.S. Dept. of
5  * Energy).  All rights reserved.
6  *
7  * If you have questions about your rights to use or distribute this
8  * software, please contact Berkeley Lab's Technology Transfer
9  * Department at [email protected].
10  *
11  * NOTICE.  This software is owned by the U.S. Department of Energy.
12  * As such, the U.S. Government has been granted for itself and others
13  * acting on its behalf a paid-up, nonexclusive, irrevocable,
14  * worldwide license in the Software to reproduce, prepare derivative
15  * works, and perform publicly and display publicly.  Beginning five
16  * (5) years after the date permission to assert copyright is obtained
17  * from the U.S. Department of Energy, and subject to any subsequent
18  * five (5) year renewals, the U.S. Government is granted for itself
19  * and others acting on its behalf a paid-up, nonexclusive,
20  * irrevocable, worldwide license in the Software to reproduce,
21  * prepare derivative works, distribute copies to the public, perform
22  * publicly and display publicly, and to permit others to do so.
23  *
24  * This code is distributed under a BSD style license, see the LICENSE file
25  * for complete information.
26  */
27 #ifndef _GNU_SOURCE
28 # define _GNU_SOURCE
29 #endif
30 #define __USE_GNU
31 
32 #include "iperf_config.h"
33 
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <string.h>
37 #include <time.h>
38 #include <getopt.h>
39 #include <errno.h>
40 #include <signal.h>
41 #include <unistd.h>
42 #include <assert.h>
43 #include <fcntl.h>
44 #include <sys/socket.h>
45 #include <sys/types.h>
46 #include <netinet/in.h>
47 #include <arpa/inet.h>
48 #include <netdb.h>
49 #ifdef HAVE_STDINT_H
50 #include <stdint.h>
51 #endif
52 #include <sys/time.h>
53 #include <sys/resource.h>
54 #include <sys/mman.h>
55 #include <sys/stat.h>
56 #include <sched.h>
57 #include <setjmp.h>
58 #include <stdarg.h>
59 #include <math.h>
60 
61 #if defined(HAVE_CPUSET_SETAFFINITY)
62 #include <sys/param.h>
63 #include <sys/cpuset.h>
64 #endif /* HAVE_CPUSET_SETAFFINITY */
65 
66 #if defined(__CYGWIN__) || defined(_WIN32) || defined(_WIN64) || defined(__WINDOWS__)
67 #define CPU_SETSIZE __CPU_SETSIZE
68 #endif /* __CYGWIN__, _WIN32, _WIN64, __WINDOWS__ */
69 
70 #if defined(HAVE_SETPROCESSAFFINITYMASK)
71 #include <Windows.h>
72 #endif /* HAVE_SETPROCESSAFFINITYMASK */
73 
74 #include "net.h"
75 #include "iperf.h"
76 #include "iperf_api.h"
77 #include "iperf_udp.h"
78 #include "iperf_tcp.h"
79 #if defined(HAVE_SCTP_H)
80 #include "iperf_sctp.h"
81 #endif /* HAVE_SCTP_H */
82 #include "timer.h"
83 
84 #include "cjson.h"
85 #include "units.h"
86 #include "iperf_util.h"
87 #include "iperf_locale.h"
88 #include "version.h"
89 #if defined(HAVE_SSL)
90 #include <openssl/bio.h>
91 #include <openssl/err.h>
92 #include "iperf_auth.h"
93 #endif /* HAVE_SSL */
94 
95 /* Forwards. */
96 static int send_parameters(struct iperf_test *test);
97 static int get_parameters(struct iperf_test *test);
98 static int send_results(struct iperf_test *test);
99 static int get_results(struct iperf_test *test);
100 static int diskfile_send(struct iperf_stream *sp);
101 static int diskfile_recv(struct iperf_stream *sp);
102 static int JSON_write(int fd, cJSON *json);
103 static void print_interval_results(struct iperf_test *test, struct iperf_stream *sp, cJSON *json_interval_streams);
104 static cJSON *JSON_read(int fd);
105 
106 
107 /*************************** Print usage functions ****************************/
108 
109 void
usage()110 usage()
111 {
112     fputs(usage_shortstr, stderr);
113 }
114 
115 
116 void
usage_long(FILE * f)117 usage_long(FILE *f)
118 {
119     fprintf(f, usage_longstr, DEFAULT_NO_MSG_RCVD_TIMEOUT, UDP_RATE / (1024*1024), DEFAULT_PACING_TIMER, DURATION, DEFAULT_TCP_BLKSIZE / 1024, DEFAULT_UDP_BLKSIZE);
120 }
121 
122 
warning(const char * str)123 void warning(const char *str)
124 {
125     fprintf(stderr, "warning: %s\n", str);
126 }
127 
128 
129 /************** Getter routines for some fields inside iperf_test *************/
130 
131 int
iperf_get_verbose(struct iperf_test * ipt)132 iperf_get_verbose(struct iperf_test *ipt)
133 {
134     return ipt->verbose;
135 }
136 
137 int
iperf_get_control_socket(struct iperf_test * ipt)138 iperf_get_control_socket(struct iperf_test *ipt)
139 {
140     return ipt->ctrl_sck;
141 }
142 
143 int
iperf_get_control_socket_mss(struct iperf_test * ipt)144 iperf_get_control_socket_mss(struct iperf_test *ipt)
145 {
146     return ipt->ctrl_sck_mss;
147 }
148 
149 int
iperf_get_test_omit(struct iperf_test * ipt)150 iperf_get_test_omit(struct iperf_test *ipt)
151 {
152     return ipt->omit;
153 }
154 
155 int
iperf_get_test_duration(struct iperf_test * ipt)156 iperf_get_test_duration(struct iperf_test *ipt)
157 {
158     return ipt->duration;
159 }
160 
161 uint64_t
iperf_get_test_rate(struct iperf_test * ipt)162 iperf_get_test_rate(struct iperf_test *ipt)
163 {
164     return ipt->settings->rate;
165 }
166 
167 uint64_t
iperf_get_test_bitrate_limit(struct iperf_test * ipt)168 iperf_get_test_bitrate_limit(struct iperf_test *ipt)
169 {
170     return ipt->settings->bitrate_limit;
171 }
172 
173 double
iperf_get_test_bitrate_limit_interval(struct iperf_test * ipt)174 iperf_get_test_bitrate_limit_interval(struct iperf_test *ipt)
175 {
176     return ipt->settings->bitrate_limit_interval;
177 }
178 
179 int
iperf_get_test_bitrate_limit_stats_per_interval(struct iperf_test * ipt)180 iperf_get_test_bitrate_limit_stats_per_interval(struct iperf_test *ipt)
181 {
182     return ipt->settings->bitrate_limit_stats_per_interval;
183 }
184 
185 uint64_t
iperf_get_test_fqrate(struct iperf_test * ipt)186 iperf_get_test_fqrate(struct iperf_test *ipt)
187 {
188     return ipt->settings->fqrate;
189 }
190 
191 int
iperf_get_test_pacing_timer(struct iperf_test * ipt)192 iperf_get_test_pacing_timer(struct iperf_test *ipt)
193 {
194     return ipt->settings->pacing_timer;
195 }
196 
197 uint64_t
iperf_get_test_bytes(struct iperf_test * ipt)198 iperf_get_test_bytes(struct iperf_test *ipt)
199 {
200     return (uint64_t) ipt->settings->bytes;
201 }
202 
203 uint64_t
iperf_get_test_blocks(struct iperf_test * ipt)204 iperf_get_test_blocks(struct iperf_test *ipt)
205 {
206     return (uint64_t) ipt->settings->blocks;
207 }
208 
209 int
iperf_get_test_burst(struct iperf_test * ipt)210 iperf_get_test_burst(struct iperf_test *ipt)
211 {
212     return ipt->settings->burst;
213 }
214 
215 char
iperf_get_test_role(struct iperf_test * ipt)216 iperf_get_test_role(struct iperf_test *ipt)
217 {
218     return ipt->role;
219 }
220 
221 int
iperf_get_test_reverse(struct iperf_test * ipt)222 iperf_get_test_reverse(struct iperf_test *ipt)
223 {
224     return ipt->reverse;
225 }
226 
227 int
iperf_get_test_blksize(struct iperf_test * ipt)228 iperf_get_test_blksize(struct iperf_test *ipt)
229 {
230     return ipt->settings->blksize;
231 }
232 
233 FILE *
iperf_get_test_outfile(struct iperf_test * ipt)234 iperf_get_test_outfile (struct iperf_test *ipt)
235 {
236     return ipt->outfile;
237 }
238 
239 int
iperf_get_test_socket_bufsize(struct iperf_test * ipt)240 iperf_get_test_socket_bufsize(struct iperf_test *ipt)
241 {
242     return ipt->settings->socket_bufsize;
243 }
244 
245 double
iperf_get_test_reporter_interval(struct iperf_test * ipt)246 iperf_get_test_reporter_interval(struct iperf_test *ipt)
247 {
248     return ipt->reporter_interval;
249 }
250 
251 double
iperf_get_test_stats_interval(struct iperf_test * ipt)252 iperf_get_test_stats_interval(struct iperf_test *ipt)
253 {
254     return ipt->stats_interval;
255 }
256 
257 int
iperf_get_test_num_streams(struct iperf_test * ipt)258 iperf_get_test_num_streams(struct iperf_test *ipt)
259 {
260     return ipt->num_streams;
261 }
262 
263 int
iperf_get_test_timestamps(struct iperf_test * ipt)264 iperf_get_test_timestamps(struct iperf_test *ipt)
265 {
266     return ipt->timestamps;
267 }
268 
269 const char *
iperf_get_test_timestamp_format(struct iperf_test * ipt)270 iperf_get_test_timestamp_format(struct iperf_test *ipt)
271 {
272     return ipt->timestamp_format;
273 }
274 
275 int
iperf_get_test_repeating_payload(struct iperf_test * ipt)276 iperf_get_test_repeating_payload(struct iperf_test *ipt)
277 {
278     return ipt->repeating_payload;
279 }
280 
281 int
iperf_get_test_bind_port(struct iperf_test * ipt)282 iperf_get_test_bind_port(struct iperf_test *ipt)
283 {
284     return ipt->bind_port;
285 }
286 
287 int
iperf_get_test_server_port(struct iperf_test * ipt)288 iperf_get_test_server_port(struct iperf_test *ipt)
289 {
290     return ipt->server_port;
291 }
292 
293 char*
iperf_get_test_server_hostname(struct iperf_test * ipt)294 iperf_get_test_server_hostname(struct iperf_test *ipt)
295 {
296     return ipt->server_hostname;
297 }
298 
299 char*
iperf_get_test_template(struct iperf_test * ipt)300 iperf_get_test_template(struct iperf_test *ipt)
301 {
302     return ipt->tmp_template;
303 }
304 
305 int
iperf_get_test_protocol_id(struct iperf_test * ipt)306 iperf_get_test_protocol_id(struct iperf_test *ipt)
307 {
308     return ipt->protocol->id;
309 }
310 
311 int
iperf_get_test_json_output(struct iperf_test * ipt)312 iperf_get_test_json_output(struct iperf_test *ipt)
313 {
314     return ipt->json_output;
315 }
316 
317 char *
iperf_get_test_json_output_string(struct iperf_test * ipt)318 iperf_get_test_json_output_string(struct iperf_test *ipt)
319 {
320     return ipt->json_output_string;
321 }
322 
323 int
iperf_get_test_zerocopy(struct iperf_test * ipt)324 iperf_get_test_zerocopy(struct iperf_test *ipt)
325 {
326     return ipt->zerocopy;
327 }
328 
329 int
iperf_get_test_get_server_output(struct iperf_test * ipt)330 iperf_get_test_get_server_output(struct iperf_test *ipt)
331 {
332     return ipt->get_server_output;
333 }
334 
335 char
iperf_get_test_unit_format(struct iperf_test * ipt)336 iperf_get_test_unit_format(struct iperf_test *ipt)
337 {
338     return ipt->settings->unit_format;
339 }
340 
341 char *
iperf_get_test_bind_address(struct iperf_test * ipt)342 iperf_get_test_bind_address(struct iperf_test *ipt)
343 {
344     return ipt->bind_address;
345 }
346 
347 char *
iperf_get_test_bind_dev(struct iperf_test * ipt)348 iperf_get_test_bind_dev(struct iperf_test *ipt)
349 {
350     return ipt->bind_dev;
351 }
352 
353 int
iperf_get_test_udp_counters_64bit(struct iperf_test * ipt)354 iperf_get_test_udp_counters_64bit(struct iperf_test *ipt)
355 {
356     return ipt->udp_counters_64bit;
357 }
358 
359 int
iperf_get_test_one_off(struct iperf_test * ipt)360 iperf_get_test_one_off(struct iperf_test *ipt)
361 {
362     return ipt->one_off;
363 }
364 
365 int
iperf_get_test_tos(struct iperf_test * ipt)366 iperf_get_test_tos(struct iperf_test *ipt)
367 {
368     return ipt->settings->tos;
369 }
370 
371 char *
iperf_get_test_extra_data(struct iperf_test * ipt)372 iperf_get_test_extra_data(struct iperf_test *ipt)
373 {
374     return ipt->extra_data;
375 }
376 
377 static const char iperf_version[] = IPERF_VERSION;
378 char *
iperf_get_iperf_version(void)379 iperf_get_iperf_version(void)
380 {
381     return (char*)iperf_version;
382 }
383 
384 int
iperf_get_test_no_delay(struct iperf_test * ipt)385 iperf_get_test_no_delay(struct iperf_test *ipt)
386 {
387     return ipt->no_delay;
388 }
389 
390 int
iperf_get_test_connect_timeout(struct iperf_test * ipt)391 iperf_get_test_connect_timeout(struct iperf_test *ipt)
392 {
393     return ipt->settings->connect_timeout;
394 }
395 
396 int
iperf_get_test_idle_timeout(struct iperf_test * ipt)397 iperf_get_test_idle_timeout(struct iperf_test *ipt)
398 {
399     return ipt->settings->idle_timeout;
400 }
401 
402 int
iperf_get_dont_fragment(struct iperf_test * ipt)403 iperf_get_dont_fragment(struct iperf_test *ipt)
404 {
405     return ipt->settings->dont_fragment;
406 }
407 
408 struct iperf_time*
iperf_get_test_rcv_timeout(struct iperf_test * ipt)409 iperf_get_test_rcv_timeout(struct iperf_test *ipt)
410 {
411     return &ipt->settings->rcv_timeout;
412 }
413 
414 char*
iperf_get_test_congestion_control(struct iperf_test * ipt)415 iperf_get_test_congestion_control(struct iperf_test* ipt)
416 {
417     return ipt->congestion;
418 }
419 
420 int
iperf_get_test_mss(struct iperf_test * ipt)421 iperf_get_test_mss(struct iperf_test *ipt)
422 {
423     return ipt->settings->mss;
424 }
425 
426 /************** Setter routines for some fields inside iperf_test *************/
427 
428 void
iperf_set_verbose(struct iperf_test * ipt,int verbose)429 iperf_set_verbose(struct iperf_test *ipt, int verbose)
430 {
431     ipt->verbose = verbose;
432 }
433 
434 void
iperf_set_control_socket(struct iperf_test * ipt,int ctrl_sck)435 iperf_set_control_socket(struct iperf_test *ipt, int ctrl_sck)
436 {
437     ipt->ctrl_sck = ctrl_sck;
438 }
439 
440 void
iperf_set_test_omit(struct iperf_test * ipt,int omit)441 iperf_set_test_omit(struct iperf_test *ipt, int omit)
442 {
443     ipt->omit = omit;
444 }
445 
446 void
iperf_set_test_duration(struct iperf_test * ipt,int duration)447 iperf_set_test_duration(struct iperf_test *ipt, int duration)
448 {
449     ipt->duration = duration;
450 }
451 
452 void
iperf_set_test_reporter_interval(struct iperf_test * ipt,double reporter_interval)453 iperf_set_test_reporter_interval(struct iperf_test *ipt, double reporter_interval)
454 {
455     ipt->reporter_interval = reporter_interval;
456 }
457 
458 void
iperf_set_test_stats_interval(struct iperf_test * ipt,double stats_interval)459 iperf_set_test_stats_interval(struct iperf_test *ipt, double stats_interval)
460 {
461     ipt->stats_interval = stats_interval;
462 }
463 
464 void
iperf_set_test_state(struct iperf_test * ipt,signed char state)465 iperf_set_test_state(struct iperf_test *ipt, signed char state)
466 {
467     ipt->state = state;
468 }
469 
470 void
iperf_set_test_blksize(struct iperf_test * ipt,int blksize)471 iperf_set_test_blksize(struct iperf_test *ipt, int blksize)
472 {
473     ipt->settings->blksize = blksize;
474 }
475 
476 void
iperf_set_test_logfile(struct iperf_test * ipt,const char * logfile)477 iperf_set_test_logfile(struct iperf_test *ipt, const char *logfile)
478 {
479     ipt->logfile = strdup(logfile);
480 }
481 
482 void
iperf_set_test_rate(struct iperf_test * ipt,uint64_t rate)483 iperf_set_test_rate(struct iperf_test *ipt, uint64_t rate)
484 {
485     ipt->settings->rate = rate;
486 }
487 
488 void
iperf_set_test_bitrate_limit_maximum(struct iperf_test * ipt,uint64_t total_rate)489 iperf_set_test_bitrate_limit_maximum(struct iperf_test *ipt, uint64_t total_rate)
490 {
491     ipt->settings->bitrate_limit = total_rate;
492 }
493 
494 void
iperf_set_test_bitrate_limit_interval(struct iperf_test * ipt,uint64_t bitrate_limit_interval)495 iperf_set_test_bitrate_limit_interval(struct iperf_test *ipt, uint64_t bitrate_limit_interval)
496 {
497     ipt->settings->bitrate_limit_interval = bitrate_limit_interval;
498 }
499 
500 void
iperf_set_test_bitrate_limit_stats_per_interval(struct iperf_test * ipt,uint64_t bitrate_limit_stats_per_interval)501 iperf_set_test_bitrate_limit_stats_per_interval(struct iperf_test *ipt, uint64_t bitrate_limit_stats_per_interval)
502 {
503     ipt->settings->bitrate_limit_stats_per_interval = bitrate_limit_stats_per_interval;
504 }
505 
506 void
iperf_set_test_fqrate(struct iperf_test * ipt,uint64_t fqrate)507 iperf_set_test_fqrate(struct iperf_test *ipt, uint64_t fqrate)
508 {
509     ipt->settings->fqrate = fqrate;
510 }
511 
512 void
iperf_set_test_pacing_timer(struct iperf_test * ipt,int pacing_timer)513 iperf_set_test_pacing_timer(struct iperf_test *ipt, int pacing_timer)
514 {
515     ipt->settings->pacing_timer = pacing_timer;
516 }
517 
518 void
iperf_set_test_bytes(struct iperf_test * ipt,uint64_t bytes)519 iperf_set_test_bytes(struct iperf_test *ipt, uint64_t bytes)
520 {
521     ipt->settings->bytes = (iperf_size_t) bytes;
522 }
523 
524 void
iperf_set_test_blocks(struct iperf_test * ipt,uint64_t blocks)525 iperf_set_test_blocks(struct iperf_test *ipt, uint64_t blocks)
526 {
527     ipt->settings->blocks = (iperf_size_t) blocks;
528 }
529 
530 void
iperf_set_test_burst(struct iperf_test * ipt,int burst)531 iperf_set_test_burst(struct iperf_test *ipt, int burst)
532 {
533     ipt->settings->burst = burst;
534 }
535 
536 void
iperf_set_test_bind_port(struct iperf_test * ipt,int bind_port)537 iperf_set_test_bind_port(struct iperf_test *ipt, int bind_port)
538 {
539     ipt->bind_port = bind_port;
540 }
541 
542 void
iperf_set_test_server_port(struct iperf_test * ipt,int srv_port)543 iperf_set_test_server_port(struct iperf_test *ipt, int srv_port)
544 {
545     ipt->server_port = srv_port;
546 }
547 
548 void
iperf_set_test_socket_bufsize(struct iperf_test * ipt,int socket_bufsize)549 iperf_set_test_socket_bufsize(struct iperf_test *ipt, int socket_bufsize)
550 {
551     ipt->settings->socket_bufsize = socket_bufsize;
552 }
553 
554 void
iperf_set_test_num_streams(struct iperf_test * ipt,int num_streams)555 iperf_set_test_num_streams(struct iperf_test *ipt, int num_streams)
556 {
557     ipt->num_streams = num_streams;
558 }
559 
560 void
iperf_set_test_repeating_payload(struct iperf_test * ipt,int repeating_payload)561 iperf_set_test_repeating_payload(struct iperf_test *ipt, int repeating_payload)
562 {
563     ipt->repeating_payload = repeating_payload;
564 }
565 
566 void
iperf_set_test_timestamps(struct iperf_test * ipt,int timestamps)567 iperf_set_test_timestamps(struct iperf_test *ipt, int timestamps)
568 {
569     ipt->timestamps = timestamps;
570 }
571 
572 void
iperf_set_test_timestamp_format(struct iperf_test * ipt,const char * tf)573 iperf_set_test_timestamp_format(struct iperf_test *ipt, const char *tf)
574 {
575     ipt->timestamp_format = strdup(tf);
576 }
577 
578 static void
check_sender_has_retransmits(struct iperf_test * ipt)579 check_sender_has_retransmits(struct iperf_test *ipt)
580 {
581     if (ipt->mode != RECEIVER && ipt->protocol->id == Ptcp && has_tcpinfo_retransmits())
582 	ipt->sender_has_retransmits = 1;
583     else
584 	ipt->sender_has_retransmits = 0;
585 }
586 
587 void
iperf_set_test_role(struct iperf_test * ipt,char role)588 iperf_set_test_role(struct iperf_test *ipt, char role)
589 {
590     ipt->role = role;
591     if (!ipt->reverse) {
592         if (ipt->bidirectional)
593             ipt->mode = BIDIRECTIONAL;
594         else if (role == 'c')
595             ipt->mode = SENDER;
596         else if (role == 's')
597             ipt->mode = RECEIVER;
598     } else {
599         if (role == 'c')
600             ipt->mode = RECEIVER;
601         else if (role == 's')
602             ipt->mode = SENDER;
603     }
604     check_sender_has_retransmits(ipt);
605 }
606 
607 void
iperf_set_test_server_hostname(struct iperf_test * ipt,const char * server_hostname)608 iperf_set_test_server_hostname(struct iperf_test *ipt, const char *server_hostname)
609 {
610     ipt->server_hostname = strdup(server_hostname);
611 }
612 
613 void
iperf_set_test_template(struct iperf_test * ipt,const char * tmp_template)614 iperf_set_test_template(struct iperf_test *ipt, const char *tmp_template)
615 {
616     ipt->tmp_template = strdup(tmp_template);
617 }
618 
619 void
iperf_set_test_reverse(struct iperf_test * ipt,int reverse)620 iperf_set_test_reverse(struct iperf_test *ipt, int reverse)
621 {
622     ipt->reverse = reverse;
623     if (!ipt->reverse) {
624         if (ipt->role == 'c')
625             ipt->mode = SENDER;
626         else if (ipt->role == 's')
627             ipt->mode = RECEIVER;
628     } else {
629         if (ipt->role == 'c')
630             ipt->mode = RECEIVER;
631         else if (ipt->role == 's')
632             ipt->mode = SENDER;
633     }
634     check_sender_has_retransmits(ipt);
635 }
636 
637 void
iperf_set_test_json_output(struct iperf_test * ipt,int json_output)638 iperf_set_test_json_output(struct iperf_test *ipt, int json_output)
639 {
640     ipt->json_output = json_output;
641 }
642 
643 int
iperf_has_zerocopy(void)644 iperf_has_zerocopy( void )
645 {
646     return has_sendfile();
647 }
648 
649 void
iperf_set_test_zerocopy(struct iperf_test * ipt,int zerocopy)650 iperf_set_test_zerocopy(struct iperf_test *ipt, int zerocopy)
651 {
652     ipt->zerocopy = (zerocopy && has_sendfile());
653 }
654 
655 void
iperf_set_test_get_server_output(struct iperf_test * ipt,int get_server_output)656 iperf_set_test_get_server_output(struct iperf_test *ipt, int get_server_output)
657 {
658     ipt->get_server_output = get_server_output;
659 }
660 
661 void
iperf_set_test_unit_format(struct iperf_test * ipt,char unit_format)662 iperf_set_test_unit_format(struct iperf_test *ipt, char unit_format)
663 {
664     ipt->settings->unit_format = unit_format;
665 }
666 
667 #if defined(HAVE_SSL)
668 void
iperf_set_test_client_username(struct iperf_test * ipt,const char * client_username)669 iperf_set_test_client_username(struct iperf_test *ipt, const char *client_username)
670 {
671     ipt->settings->client_username = strdup(client_username);
672 }
673 
674 void
iperf_set_test_client_password(struct iperf_test * ipt,const char * client_password)675 iperf_set_test_client_password(struct iperf_test *ipt, const char *client_password)
676 {
677     ipt->settings->client_password = strdup(client_password);
678 }
679 
680 void
iperf_set_test_client_rsa_pubkey(struct iperf_test * ipt,const char * client_rsa_pubkey_base64)681 iperf_set_test_client_rsa_pubkey(struct iperf_test *ipt, const char *client_rsa_pubkey_base64)
682 {
683     ipt->settings->client_rsa_pubkey = load_pubkey_from_base64(client_rsa_pubkey_base64);
684 }
685 
686 void
iperf_set_test_server_authorized_users(struct iperf_test * ipt,const char * server_authorized_users)687 iperf_set_test_server_authorized_users(struct iperf_test *ipt, const char *server_authorized_users)
688 {
689     ipt->server_authorized_users = strdup(server_authorized_users);
690 }
691 
692 void
iperf_set_test_server_skew_threshold(struct iperf_test * ipt,int server_skew_threshold)693 iperf_set_test_server_skew_threshold(struct iperf_test *ipt, int server_skew_threshold)
694 {
695     ipt->server_skew_threshold = server_skew_threshold;
696 }
697 
698 void
iperf_set_test_server_rsa_privkey(struct iperf_test * ipt,const char * server_rsa_privkey_base64)699 iperf_set_test_server_rsa_privkey(struct iperf_test *ipt, const char *server_rsa_privkey_base64)
700 {
701     ipt->server_rsa_private_key = load_privkey_from_base64(server_rsa_privkey_base64);
702 }
703 #endif // HAVE_SSL
704 
705 void
iperf_set_test_bind_address(struct iperf_test * ipt,const char * bnd_address)706 iperf_set_test_bind_address(struct iperf_test *ipt, const char *bnd_address)
707 {
708     ipt->bind_address = strdup(bnd_address);
709 }
710 
711 void
iperf_set_test_bind_dev(struct iperf_test * ipt,char * bnd_dev)712 iperf_set_test_bind_dev(struct iperf_test *ipt, char *bnd_dev)
713 {
714     ipt->bind_dev = strdup(bnd_dev);
715 }
716 
717 void
iperf_set_test_udp_counters_64bit(struct iperf_test * ipt,int udp_counters_64bit)718 iperf_set_test_udp_counters_64bit(struct iperf_test *ipt, int udp_counters_64bit)
719 {
720     ipt->udp_counters_64bit = udp_counters_64bit;
721 }
722 
723 void
iperf_set_test_one_off(struct iperf_test * ipt,int one_off)724 iperf_set_test_one_off(struct iperf_test *ipt, int one_off)
725 {
726     ipt->one_off = one_off;
727 }
728 
729 void
iperf_set_test_tos(struct iperf_test * ipt,int tos)730 iperf_set_test_tos(struct iperf_test *ipt, int tos)
731 {
732     ipt->settings->tos = tos;
733 }
734 
735 void
iperf_set_test_extra_data(struct iperf_test * ipt,const char * dat)736 iperf_set_test_extra_data(struct iperf_test *ipt, const char *dat)
737 {
738     ipt->extra_data = strdup(dat);
739 }
740 
741 void
iperf_set_test_bidirectional(struct iperf_test * ipt,int bidirectional)742 iperf_set_test_bidirectional(struct iperf_test* ipt, int bidirectional)
743 {
744     ipt->bidirectional = bidirectional;
745     if (bidirectional)
746         ipt->mode = BIDIRECTIONAL;
747     else
748         iperf_set_test_reverse(ipt, ipt->reverse);
749 }
750 
751 void
iperf_set_test_no_delay(struct iperf_test * ipt,int no_delay)752 iperf_set_test_no_delay(struct iperf_test* ipt, int no_delay)
753 {
754     ipt->no_delay = no_delay;
755 }
756 
757 void
iperf_set_test_connect_timeout(struct iperf_test * ipt,int ct)758 iperf_set_test_connect_timeout(struct iperf_test* ipt, int ct)
759 {
760     ipt->settings->connect_timeout = ct;
761 }
762 
763 void
iperf_set_test_idle_timeout(struct iperf_test * ipt,int to)764 iperf_set_test_idle_timeout(struct iperf_test* ipt, int to)
765 {
766     ipt->settings->idle_timeout = to;
767 }
768 
769 void
iperf_set_dont_fragment(struct iperf_test * ipt,int dnf)770 iperf_set_dont_fragment(struct iperf_test* ipt, int dnf)
771 {
772     ipt->settings->dont_fragment = dnf;
773 }
774 
775 void
iperf_set_test_rcv_timeout(struct iperf_test * ipt,struct iperf_time * to)776 iperf_set_test_rcv_timeout(struct iperf_test* ipt, struct iperf_time* to)
777 {
778     ipt->settings->rcv_timeout.secs = to->secs;
779     ipt->settings->rcv_timeout.usecs = to->usecs;
780 }
781 
782 void
iperf_set_test_congestion_control(struct iperf_test * ipt,char * cc)783 iperf_set_test_congestion_control(struct iperf_test* ipt, char* cc)
784 {
785     ipt->congestion = strdup(cc);
786 }
787 
788 void
iperf_set_test_mss(struct iperf_test * ipt,int mss)789 iperf_set_test_mss(struct iperf_test *ipt, int mss)
790 {
791     ipt->settings->mss = mss;
792 }
793 
794 /********************** Get/set test protocol structure ***********************/
795 
796 struct protocol *
get_protocol(struct iperf_test * test,int prot_id)797 get_protocol(struct iperf_test *test, int prot_id)
798 {
799     struct protocol *prot;
800 
801     SLIST_FOREACH(prot, &test->protocols, protocols) {
802         if (prot->id == prot_id)
803             break;
804     }
805 
806     if (prot == NULL)
807         i_errno = IEPROTOCOL;
808 
809     return prot;
810 }
811 
812 int
set_protocol(struct iperf_test * test,int prot_id)813 set_protocol(struct iperf_test *test, int prot_id)
814 {
815     struct protocol *prot = NULL;
816 
817     SLIST_FOREACH(prot, &test->protocols, protocols) {
818         if (prot->id == prot_id) {
819             test->protocol = prot;
820 	    check_sender_has_retransmits(test);
821             return 0;
822         }
823     }
824 
825     i_errno = IEPROTOCOL;
826     return -1;
827 }
828 
829 
830 /************************** Iperf callback functions **************************/
831 
832 void
iperf_on_new_stream(struct iperf_stream * sp)833 iperf_on_new_stream(struct iperf_stream *sp)
834 {
835     connect_msg(sp);
836 }
837 
838 void
iperf_on_test_start(struct iperf_test * test)839 iperf_on_test_start(struct iperf_test *test)
840 {
841     if (test->json_output) {
842 	cJSON_AddItemToObject(test->json_start, "test_start", iperf_json_printf("protocol: %s  num_streams: %d  blksize: %d  omit: %d  duration: %d  bytes: %d  blocks: %d  reverse: %d  tos: %d  target_bitrate: %d", test->protocol->name, (int64_t) test->num_streams, (int64_t) test->settings->blksize, (int64_t) test->omit, (int64_t) test->duration, (int64_t) test->settings->bytes, (int64_t) test->settings->blocks, test->reverse?(int64_t)1:(int64_t)0, (int64_t) test->settings->tos, (int64_t) test->settings->rate));
843     } else {
844 	if (test->verbose) {
845 	    if (test->settings->bytes)
846 		iperf_printf(test, test_start_bytes, test->protocol->name, test->num_streams, test->settings->blksize, test->omit, test->settings->bytes, test->settings->tos);
847 	    else if (test->settings->blocks)
848 		iperf_printf(test, test_start_blocks, test->protocol->name, test->num_streams, test->settings->blksize, test->omit, test->settings->blocks, test->settings->tos);
849 	    else
850 		iperf_printf(test, test_start_time, test->protocol->name, test->num_streams, test->settings->blksize, test->omit, test->duration, test->settings->tos);
851 	}
852     }
853 }
854 
855 /* This converts an IPv6 string address from IPv4-mapped format into regular
856 ** old IPv4 format, which is easier on the eyes of network veterans.
857 **
858 ** If the v6 address is not v4-mapped it is left alone.
859 */
860 static void
mapped_v4_to_regular_v4(char * str)861 mapped_v4_to_regular_v4(char *str)
862 {
863     char *prefix = "::ffff:";
864     int prefix_len;
865 
866     prefix_len = strlen(prefix);
867     if (strncmp(str, prefix, prefix_len) == 0) {
868 	int str_len = strlen(str);
869 	memmove(str, str + prefix_len, str_len - prefix_len + 1);
870     }
871 }
872 
873 void
iperf_on_connect(struct iperf_test * test)874 iperf_on_connect(struct iperf_test *test)
875 {
876     time_t now_secs;
877     const char* rfc1123_fmt = "%a, %d %b %Y %H:%M:%S %Z";
878     char now_str[100];
879     char ipr[INET6_ADDRSTRLEN];
880     int port;
881     struct sockaddr_storage sa;
882     struct sockaddr_in *sa_inP;
883     struct sockaddr_in6 *sa_in6P;
884     socklen_t len;
885 
886     now_secs = time((time_t*) 0);
887     (void) strftime(now_str, sizeof(now_str), rfc1123_fmt, gmtime(&now_secs));
888     if (test->json_output)
889 	cJSON_AddItemToObject(test->json_start, "timestamp", iperf_json_printf("time: %s  timesecs: %d", now_str, (int64_t) now_secs));
890     else if (test->verbose)
891 	iperf_printf(test, report_time, now_str);
892 
893     if (test->role == 'c') {
894 	if (test->json_output)
895 	    cJSON_AddItemToObject(test->json_start, "connecting_to", iperf_json_printf("host: %s  port: %d", test->server_hostname, (int64_t) test->server_port));
896 	else {
897 	    iperf_printf(test, report_connecting, test->server_hostname, test->server_port);
898 	    if (test->reverse)
899 		iperf_printf(test, report_reverse, test->server_hostname);
900 	}
901     } else {
902         len = sizeof(sa);
903         getpeername(test->ctrl_sck, (struct sockaddr *) &sa, &len);
904         if (getsockdomain(test->ctrl_sck) == AF_INET) {
905 	    sa_inP = (struct sockaddr_in *) &sa;
906             inet_ntop(AF_INET, &sa_inP->sin_addr, ipr, sizeof(ipr));
907 	    port = ntohs(sa_inP->sin_port);
908         } else {
909 	    sa_in6P = (struct sockaddr_in6 *) &sa;
910             inet_ntop(AF_INET6, &sa_in6P->sin6_addr, ipr, sizeof(ipr));
911 	    port = ntohs(sa_in6P->sin6_port);
912         }
913 	mapped_v4_to_regular_v4(ipr);
914 	if (test->json_output)
915 	    cJSON_AddItemToObject(test->json_start, "accepted_connection", iperf_json_printf("host: %s  port: %d", ipr, (int64_t) port));
916 	else
917 	    iperf_printf(test, report_accepted, ipr, port);
918     }
919     if (test->json_output) {
920 	cJSON_AddStringToObject(test->json_start, "cookie", test->cookie);
921         if (test->protocol->id == SOCK_STREAM) {
922 	    if (test->settings->mss)
923 		cJSON_AddNumberToObject(test->json_start, "tcp_mss", test->settings->mss);
924 	    else {
925 		cJSON_AddNumberToObject(test->json_start, "tcp_mss_default", test->ctrl_sck_mss);
926 	    }
927         }
928         cJSON_AddNumberToObject(test->json_start, "target_bitrate", test->settings->rate);
929     } else if (test->verbose) {
930         iperf_printf(test, report_cookie, test->cookie);
931         if (test->protocol->id == SOCK_STREAM) {
932             if (test->settings->mss)
933                 iperf_printf(test, "      TCP MSS: %d\n", test->settings->mss);
934             else {
935                 iperf_printf(test, "      TCP MSS: %d (default)\n", test->ctrl_sck_mss);
936             }
937         }
938         if (test->settings->rate)
939             iperf_printf(test, "      Target Bitrate: %"PRIu64"\n", test->settings->rate);
940     }
941 }
942 
943 void
iperf_on_test_finish(struct iperf_test * test)944 iperf_on_test_finish(struct iperf_test *test)
945 {
946 }
947 
948 
949 /******************************************************************************/
950 
951 /*
952  * iperf_parse_hostname tries to split apart a string into hostname %
953  * interface parts, which are returned in **p and **p1, if they
954  * exist. If the %interface part is detected, and it's not an IPv6
955  * link local address, then returns 1, else returns 0.
956  *
957  * Modifies the string pointed to by spec in-place due to the use of
958  * strtok(3). The caller should strdup(3) or otherwise copy the string
959  * if an unmodified copy is needed.
960  */
961 int
iperf_parse_hostname(struct iperf_test * test,char * spec,char ** p,char ** p1)962 iperf_parse_hostname(struct iperf_test *test, char *spec, char **p, char **p1) {
963     struct in6_addr ipv6_addr;
964 
965     // Format is <addr>[%<device>]
966     if ((*p = strtok(spec, "%")) != NULL &&
967         (*p1 = strtok(NULL, "%")) != NULL) {
968 
969         /*
970          * If an IPv6 literal for a link-local address, then
971          * tell the caller to leave the "%" in the hostname.
972          */
973         if (inet_pton(AF_INET6, *p, &ipv6_addr) == 1 &&
974             IN6_IS_ADDR_LINKLOCAL(&ipv6_addr)) {
975             if (test->debug) {
976                 iperf_printf(test, "IPv6 link-local address literal detected\n");
977             }
978             return 0;
979         }
980         /*
981          * Other kind of address or FQDN. The interface name after
982          * "%" is a shorthand for --bind-dev.
983          */
984         else {
985             if (test->debug) {
986                 iperf_printf(test, "p %s p1 %s\n", *p, *p1);
987             }
988             return 1;
989         }
990     }
991     else {
992         if (test->debug) {
993             iperf_printf(test, "noparse\n");
994         }
995         return 0;
996     }
997 }
998 
999 int
iperf_parse_arguments(struct iperf_test * test,int argc,char ** argv)1000 iperf_parse_arguments(struct iperf_test *test, int argc, char **argv)
1001 {
1002     static struct option longopts[] =
1003     {
1004         {"port", required_argument, NULL, 'p'},
1005         {"format", required_argument, NULL, 'f'},
1006         {"interval", required_argument, NULL, 'i'},
1007         {"daemon", no_argument, NULL, 'D'},
1008         {"one-off", no_argument, NULL, '1'},
1009         {"verbose", no_argument, NULL, 'V'},
1010         {"json", no_argument, NULL, 'J'},
1011         {"version", no_argument, NULL, 'v'},
1012         {"server", no_argument, NULL, 's'},
1013         {"client", required_argument, NULL, 'c'},
1014         {"udp", no_argument, NULL, 'u'},
1015         {"bitrate", required_argument, NULL, 'b'},
1016         {"bandwidth", required_argument, NULL, 'b'},
1017 	{"server-bitrate-limit", required_argument, NULL, OPT_SERVER_BITRATE_LIMIT},
1018         {"time", required_argument, NULL, 't'},
1019         {"bytes", required_argument, NULL, 'n'},
1020         {"blockcount", required_argument, NULL, 'k'},
1021         {"length", required_argument, NULL, 'l'},
1022         {"parallel", required_argument, NULL, 'P'},
1023         {"reverse", no_argument, NULL, 'R'},
1024         {"bidir", no_argument, NULL, OPT_BIDIRECTIONAL},
1025         {"window", required_argument, NULL, 'w'},
1026         {"bind", required_argument, NULL, 'B'},
1027 #if defined(HAVE_SO_BINDTODEVICE)
1028         {"bind-dev", required_argument, NULL, OPT_BIND_DEV},
1029 #endif /* HAVE_SO_BINDTODEVICE */
1030         {"cport", required_argument, NULL, OPT_CLIENT_PORT},
1031         {"set-mss", required_argument, NULL, 'M'},
1032         {"no-delay", no_argument, NULL, 'N'},
1033         {"version4", no_argument, NULL, '4'},
1034         {"version6", no_argument, NULL, '6'},
1035         {"tos", required_argument, NULL, 'S'},
1036         {"dscp", required_argument, NULL, OPT_DSCP},
1037 	{"extra-data", required_argument, NULL, OPT_EXTRA_DATA},
1038 #if defined(HAVE_FLOWLABEL)
1039         {"flowlabel", required_argument, NULL, 'L'},
1040 #endif /* HAVE_FLOWLABEL */
1041         {"zerocopy", no_argument, NULL, 'Z'},
1042         {"omit", required_argument, NULL, 'O'},
1043         {"file", required_argument, NULL, 'F'},
1044         {"repeating-payload", no_argument, NULL, OPT_REPEATING_PAYLOAD},
1045         {"timestamps", optional_argument, NULL, OPT_TIMESTAMPS},
1046 #if defined(HAVE_CPU_AFFINITY)
1047         {"affinity", required_argument, NULL, 'A'},
1048 #endif /* HAVE_CPU_AFFINITY */
1049         {"title", required_argument, NULL, 'T'},
1050 #if defined(HAVE_TCP_CONGESTION)
1051         {"congestion", required_argument, NULL, 'C'},
1052         {"linux-congestion", required_argument, NULL, 'C'},
1053 #endif /* HAVE_TCP_CONGESTION */
1054 #if defined(HAVE_SCTP_H)
1055         {"sctp", no_argument, NULL, OPT_SCTP},
1056         {"nstreams", required_argument, NULL, OPT_NUMSTREAMS},
1057         {"xbind", required_argument, NULL, 'X'},
1058 #endif
1059 	{"pidfile", required_argument, NULL, 'I'},
1060 	{"logfile", required_argument, NULL, OPT_LOGFILE},
1061 	{"forceflush", no_argument, NULL, OPT_FORCEFLUSH},
1062 	{"get-server-output", no_argument, NULL, OPT_GET_SERVER_OUTPUT},
1063 	{"udp-counters-64bit", no_argument, NULL, OPT_UDP_COUNTERS_64BIT},
1064  	{"no-fq-socket-pacing", no_argument, NULL, OPT_NO_FQ_SOCKET_PACING},
1065 #if defined(HAVE_DONT_FRAGMENT)
1066 	{"dont-fragment", no_argument, NULL, OPT_DONT_FRAGMENT},
1067 #endif /* HAVE_DONT_FRAGMENT */
1068 #if defined(HAVE_SSL)
1069     {"username", required_argument, NULL, OPT_CLIENT_USERNAME},
1070     {"rsa-public-key-path", required_argument, NULL, OPT_CLIENT_RSA_PUBLIC_KEY},
1071     {"rsa-private-key-path", required_argument, NULL, OPT_SERVER_RSA_PRIVATE_KEY},
1072     {"authorized-users-path", required_argument, NULL, OPT_SERVER_AUTHORIZED_USERS},
1073     {"time-skew-threshold", required_argument, NULL, OPT_SERVER_SKEW_THRESHOLD},
1074 #endif /* HAVE_SSL */
1075 	{"fq-rate", required_argument, NULL, OPT_FQ_RATE},
1076 	{"pacing-timer", required_argument, NULL, OPT_PACING_TIMER},
1077 	{"connect-timeout", required_argument, NULL, OPT_CONNECT_TIMEOUT},
1078         {"idle-timeout", required_argument, NULL, OPT_IDLE_TIMEOUT},
1079         {"rcv-timeout", required_argument, NULL, OPT_RCV_TIMEOUT},
1080         {"snd-timeout", required_argument, NULL, OPT_SND_TIMEOUT},
1081         {"debug", optional_argument, NULL, 'd'},
1082         {"help", no_argument, NULL, 'h'},
1083         {NULL, 0, NULL, 0}
1084     };
1085     int flag;
1086     int portno;
1087     int blksize;
1088     int server_flag, client_flag, rate_flag, duration_flag, rcv_timeout_flag, snd_timeout_flag;
1089     char *endptr;
1090 #if defined(HAVE_CPU_AFFINITY)
1091     char* comma;
1092 #endif /* HAVE_CPU_AFFINITY */
1093     char* slash;
1094     char *p, *p1;
1095     struct xbind_entry *xbe;
1096     double farg;
1097     int rcv_timeout_in = 0;
1098 
1099     blksize = 0;
1100     server_flag = client_flag = rate_flag = duration_flag = rcv_timeout_flag = snd_timeout_flag =0;
1101 #if defined(HAVE_SSL)
1102     char *client_username = NULL, *client_rsa_public_key = NULL, *server_rsa_private_key = NULL;
1103 #endif /* HAVE_SSL */
1104 
1105     while ((flag = getopt_long(argc, argv, "p:f:i:D1VJvsc:ub:t:n:k:l:P:Rw:B:M:N46S:L:ZO:F:A:T:C:dI:hX:", longopts, NULL)) != -1) {
1106         switch (flag) {
1107             case 'p':
1108 		portno = atoi(optarg);
1109 		if (portno < 1 || portno > 65535) {
1110 		    i_errno = IEBADPORT;
1111 		    return -1;
1112 		}
1113 		test->server_port = portno;
1114                 break;
1115             case 'f':
1116 		if (!optarg) {
1117 		    i_errno = IEBADFORMAT;
1118 		    return -1;
1119 		}
1120 		test->settings->unit_format = *optarg;
1121 		if (test->settings->unit_format == 'k' ||
1122 		    test->settings->unit_format == 'K' ||
1123 		    test->settings->unit_format == 'm' ||
1124 		    test->settings->unit_format == 'M' ||
1125 		    test->settings->unit_format == 'g' ||
1126 		    test->settings->unit_format == 'G' ||
1127 		    test->settings->unit_format == 't' ||
1128 		    test->settings->unit_format == 'T') {
1129 			break;
1130 		}
1131 		else {
1132 		    i_errno = IEBADFORMAT;
1133 		    return -1;
1134 		}
1135                 break;
1136             case 'i':
1137                 /* XXX: could potentially want separate stat collection and reporting intervals,
1138                    but just set them to be the same for now */
1139                 test->stats_interval = test->reporter_interval = atof(optarg);
1140                 if ((test->stats_interval < MIN_INTERVAL || test->stats_interval > MAX_INTERVAL) && test->stats_interval != 0) {
1141                     i_errno = IEINTERVAL;
1142                     return -1;
1143                 }
1144                 break;
1145             case 'D':
1146 		test->daemon = 1;
1147 		server_flag = 1;
1148 	        break;
1149             case '1':
1150 		test->one_off = 1;
1151 		server_flag = 1;
1152 	        break;
1153             case 'V':
1154                 test->verbose = 1;
1155                 break;
1156             case 'J':
1157                 test->json_output = 1;
1158                 break;
1159             case 'v':
1160                 printf("%s (cJSON %s)\n%s\n%s\n", version, cJSON_Version(), get_system_info(),
1161 		       get_optional_features());
1162                 exit(0);
1163             case 's':
1164                 if (test->role == 'c') {
1165                     i_errno = IESERVCLIENT;
1166                     return -1;
1167                 }
1168 		iperf_set_test_role(test, 's');
1169                 break;
1170             case 'c':
1171                 if (test->role == 's') {
1172                     i_errno = IESERVCLIENT;
1173                     return -1;
1174                 }
1175 		iperf_set_test_role(test, 'c');
1176 		iperf_set_test_server_hostname(test, optarg);
1177 
1178                 if (iperf_parse_hostname(test, optarg, &p, &p1)) {
1179 #if defined(HAVE_SO_BINDTODEVICE)
1180                     /* Get rid of the hostname we saved earlier. */
1181                     free(iperf_get_test_server_hostname(test));
1182                     iperf_set_test_server_hostname(test, p);
1183                     iperf_set_test_bind_dev(test, p1);
1184 #else /* HAVE_SO_BINDTODEVICE */
1185                     i_errno = IEBINDDEVNOSUPPORT;
1186                     return -1;
1187 #endif /* HAVE_SO_BINDTODEVICE */
1188                 }
1189                 break;
1190             case 'u':
1191                 set_protocol(test, Pudp);
1192 		client_flag = 1;
1193                 break;
1194             case OPT_SCTP:
1195 #if defined(HAVE_SCTP_H)
1196                 set_protocol(test, Psctp);
1197                 client_flag = 1;
1198                 break;
1199 #else /* HAVE_SCTP_H */
1200                 i_errno = IEUNIMP;
1201                 return -1;
1202 #endif /* HAVE_SCTP_H */
1203 
1204             case OPT_NUMSTREAMS:
1205 #if defined(linux) || defined(__FreeBSD__)
1206                 test->settings->num_ostreams = unit_atoi(optarg);
1207                 client_flag = 1;
1208 #else /* linux */
1209                 i_errno = IEUNIMP;
1210                 return -1;
1211 #endif /* linux */
1212             case 'b':
1213 		slash = strchr(optarg, '/');
1214 		if (slash) {
1215 		    *slash = '\0';
1216 		    ++slash;
1217 		    test->settings->burst = atoi(slash);
1218 		    if (test->settings->burst <= 0 ||
1219 		        test->settings->burst > MAX_BURST) {
1220 			i_errno = IEBURST;
1221 			return -1;
1222 		    }
1223 		}
1224                 test->settings->rate = unit_atof_rate(optarg);
1225 		rate_flag = 1;
1226 		client_flag = 1;
1227                 break;
1228             case OPT_SERVER_BITRATE_LIMIT:
1229 		slash = strchr(optarg, '/');
1230 		if (slash) {
1231 		    *slash = '\0';
1232 		    ++slash;
1233 		    test->settings->bitrate_limit_interval = atof(slash);
1234 		    if (test->settings->bitrate_limit_interval != 0 &&	/* Using same Max/Min limits as for Stats Interval */
1235 		        (test->settings->bitrate_limit_interval < MIN_INTERVAL || test->settings->bitrate_limit_interval > MAX_INTERVAL) ) {
1236 			i_errno = IETOTALINTERVAL;
1237 			return -1;
1238 		    }
1239 		}
1240 		test->settings->bitrate_limit = unit_atof_rate(optarg);
1241 		server_flag = 1;
1242 	        break;
1243             case 't':
1244                 test->duration = atoi(optarg);
1245                 if (test->duration > MAX_TIME) {
1246                     i_errno = IEDURATION;
1247                     return -1;
1248                 }
1249 		duration_flag = 1;
1250 		client_flag = 1;
1251                 break;
1252             case 'n':
1253                 test->settings->bytes = unit_atoi(optarg);
1254 		client_flag = 1;
1255                 break;
1256             case 'k':
1257                 test->settings->blocks = unit_atoi(optarg);
1258 		client_flag = 1;
1259                 break;
1260             case 'l':
1261                 blksize = unit_atoi(optarg);
1262 		client_flag = 1;
1263                 break;
1264             case 'P':
1265                 test->num_streams = atoi(optarg);
1266                 if (test->num_streams > MAX_STREAMS) {
1267                     i_errno = IENUMSTREAMS;
1268                     return -1;
1269                 }
1270 		client_flag = 1;
1271                 break;
1272             case 'R':
1273                 if (test->bidirectional) {
1274                     i_errno = IEREVERSEBIDIR;
1275                     return -1;
1276                 }
1277 		iperf_set_test_reverse(test, 1);
1278 		client_flag = 1;
1279                 break;
1280             case OPT_BIDIRECTIONAL:
1281                 if (test->reverse) {
1282                     i_errno = IEREVERSEBIDIR;
1283                     return -1;
1284                 }
1285                 iperf_set_test_bidirectional(test, 1);
1286                 client_flag = 1;
1287                 break;
1288             case 'w':
1289                 // XXX: This is a socket buffer, not specific to TCP
1290 		// Do sanity checks as double-precision floating point
1291 		// to avoid possible integer overflows.
1292                 farg = unit_atof(optarg);
1293                 if (farg > (double) MAX_TCP_BUFFER) {
1294                     i_errno = IEBUFSIZE;
1295                     return -1;
1296                 }
1297                 test->settings->socket_bufsize = (int) farg;
1298 		client_flag = 1;
1299                 break;
1300 
1301             case 'B':
1302                 iperf_set_test_bind_address(test, optarg);
1303 
1304                 if (iperf_parse_hostname(test, optarg, &p, &p1)) {
1305 #if defined(HAVE_SO_BINDTODEVICE)
1306                     /* Get rid of the hostname we saved earlier. */
1307                     free(iperf_get_test_server_hostname(test));
1308                     iperf_set_test_server_hostname(test, p);
1309                     iperf_set_test_bind_dev(test, p1);
1310 #else /* HAVE_SO_BINDTODEVICE */
1311                     i_errno = IEBINDDEVNOSUPPORT;
1312                     return -1;
1313 #endif /* HAVE_SO_BINDTODEVICE */
1314                 }
1315                 break;
1316 #if defined (HAVE_SO_BINDTODEVICE)
1317             case OPT_BIND_DEV:
1318                 iperf_set_test_bind_dev(test, optarg);
1319                 break;
1320 #endif /* HAVE_SO_BINDTODEVICE */
1321             case OPT_CLIENT_PORT:
1322 		portno = atoi(optarg);
1323 		if (portno < 1 || portno > 65535) {
1324 		    i_errno = IEBADPORT;
1325 		    return -1;
1326 		}
1327                 test->bind_port = portno;
1328                 break;
1329             case 'M':
1330                 test->settings->mss = atoi(optarg);
1331                 if (test->settings->mss > MAX_MSS) {
1332                     i_errno = IEMSS;
1333                     return -1;
1334                 }
1335 		client_flag = 1;
1336                 break;
1337             case 'N':
1338                 test->no_delay = 1;
1339 		client_flag = 1;
1340                 break;
1341             case '4':
1342                 test->settings->domain = AF_INET;
1343                 break;
1344             case '6':
1345                 test->settings->domain = AF_INET6;
1346                 break;
1347             case 'S':
1348                 test->settings->tos = strtol(optarg, &endptr, 0);
1349 		if (endptr == optarg ||
1350 		    test->settings->tos < 0 ||
1351 		    test->settings->tos > 255) {
1352 		    i_errno = IEBADTOS;
1353 		    return -1;
1354 		}
1355 		client_flag = 1;
1356                 break;
1357 	    case OPT_DSCP:
1358                 test->settings->tos = parse_qos(optarg);
1359 		if(test->settings->tos < 0) {
1360 			i_errno = IEBADTOS;
1361 			return -1;
1362 		}
1363 		client_flag = 1;
1364                 break;
1365 	    case OPT_EXTRA_DATA:
1366 		test->extra_data = strdup(optarg);
1367 		client_flag = 1;
1368 	        break;
1369             case 'L':
1370 #if defined(HAVE_FLOWLABEL)
1371                 test->settings->flowlabel = strtol(optarg, &endptr, 0);
1372 		if (endptr == optarg ||
1373 		    test->settings->flowlabel < 1 || test->settings->flowlabel > 0xfffff) {
1374                     i_errno = IESETFLOW;
1375                     return -1;
1376 		}
1377 		client_flag = 1;
1378 #else /* HAVE_FLOWLABEL */
1379                 i_errno = IEUNIMP;
1380                 return -1;
1381 #endif /* HAVE_FLOWLABEL */
1382                 break;
1383             case 'X':
1384 		xbe = (struct xbind_entry *)malloc(sizeof(struct xbind_entry));
1385                 if (!xbe) {
1386 		    i_errno = IESETSCTPBINDX;
1387                     return -1;
1388                 }
1389 	        memset(xbe, 0, sizeof(*xbe));
1390                 xbe->name = strdup(optarg);
1391                 if (!xbe->name) {
1392 		    i_errno = IESETSCTPBINDX;
1393                     return -1;
1394                 }
1395 		TAILQ_INSERT_TAIL(&test->xbind_addrs, xbe, link);
1396                 break;
1397             case 'Z':
1398                 if (!has_sendfile()) {
1399                     i_errno = IENOSENDFILE;
1400                     return -1;
1401                 }
1402                 test->zerocopy = 1;
1403 		client_flag = 1;
1404                 break;
1405             case OPT_REPEATING_PAYLOAD:
1406                 test->repeating_payload = 1;
1407                 client_flag = 1;
1408                 break;
1409             case OPT_TIMESTAMPS:
1410                 iperf_set_test_timestamps(test, 1);
1411 		if (optarg) {
1412 		    iperf_set_test_timestamp_format(test, optarg);
1413 		}
1414 		else {
1415 		    iperf_set_test_timestamp_format(test, TIMESTAMP_FORMAT);
1416 		}
1417                 break;
1418             case 'O':
1419                 test->omit = atoi(optarg);
1420                 if (test->omit < 0 || test->omit > 60) {
1421                     i_errno = IEOMIT;
1422                     return -1;
1423                 }
1424 		client_flag = 1;
1425                 break;
1426             case 'F':
1427                 test->diskfile_name = optarg;
1428                 break;
1429             case OPT_IDLE_TIMEOUT:
1430                 test->settings->idle_timeout = atoi(optarg);
1431                 if (test->settings->idle_timeout < 1 || test->settings->idle_timeout > MAX_TIME) {
1432                     i_errno = IEIDLETIMEOUT;
1433                     return -1;
1434                 }
1435 		server_flag = 1;
1436 	        break;
1437             case OPT_RCV_TIMEOUT:
1438                 rcv_timeout_in = atoi(optarg);
1439                 if (rcv_timeout_in < MIN_NO_MSG_RCVD_TIMEOUT || rcv_timeout_in > MAX_TIME * SEC_TO_mS) {
1440                     i_errno = IERCVTIMEOUT;
1441                     return -1;
1442                 }
1443                 test->settings->rcv_timeout.secs = rcv_timeout_in / SEC_TO_mS;
1444                 test->settings->rcv_timeout.usecs = (rcv_timeout_in % SEC_TO_mS) * mS_TO_US;
1445                 rcv_timeout_flag = 1;
1446 	        break;
1447 #if defined(HAVE_TCP_USER_TIMEOUT)
1448             case OPT_SND_TIMEOUT:
1449                 test->settings->snd_timeout = atoi(optarg);
1450                 if (test->settings->snd_timeout < 0 || test->settings->snd_timeout > MAX_TIME * SEC_TO_mS) {
1451                     i_errno = IESNDTIMEOUT;
1452                     return -1;
1453                 }
1454                 snd_timeout_flag = 1;
1455 	        break;
1456 #endif /* HAVE_TCP_USER_TIMEOUT */
1457             case 'A':
1458 #if defined(HAVE_CPU_AFFINITY)
1459                 test->affinity = strtol(optarg, &endptr, 0);
1460                 if (endptr == optarg ||
1461 		    test->affinity < 0 || test->affinity > 1024) {
1462                     i_errno = IEAFFINITY;
1463                     return -1;
1464                 }
1465 		comma = strchr(optarg, ',');
1466 		if (comma != NULL) {
1467 		    test->server_affinity = atoi(comma+1);
1468 		    if (test->server_affinity < 0 || test->server_affinity > 1024) {
1469 			i_errno = IEAFFINITY;
1470 			return -1;
1471 		    }
1472 		    client_flag = 1;
1473 		}
1474 #else /* HAVE_CPU_AFFINITY */
1475                 i_errno = IEUNIMP;
1476                 return -1;
1477 #endif /* HAVE_CPU_AFFINITY */
1478                 break;
1479             case 'T':
1480                 test->title = strdup(optarg);
1481 		client_flag = 1;
1482                 break;
1483 	    case 'C':
1484 #if defined(HAVE_TCP_CONGESTION)
1485 		test->congestion = strdup(optarg);
1486 		client_flag = 1;
1487 #else /* HAVE_TCP_CONGESTION */
1488 		i_errno = IEUNIMP;
1489 		return -1;
1490 #endif /* HAVE_TCP_CONGESTION */
1491 		break;
1492 	    case 'd':
1493 		test->debug = 1;
1494                 test->debug_level = DEBUG_LEVEL_MAX;
1495                 if (optarg) {
1496                     test->debug_level = atoi(optarg);
1497                     if (test->debug_level < 0)
1498                         test->debug_level = DEBUG_LEVEL_MAX;
1499                 }
1500 		break;
1501 	    case 'I':
1502 		test->pidfile = strdup(optarg);
1503 	        break;
1504 	    case OPT_LOGFILE:
1505 		test->logfile = strdup(optarg);
1506 		break;
1507 	    case OPT_FORCEFLUSH:
1508 		test->forceflush = 1;
1509 		break;
1510 	    case OPT_GET_SERVER_OUTPUT:
1511 		test->get_server_output = 1;
1512 		client_flag = 1;
1513 		break;
1514 	    case OPT_UDP_COUNTERS_64BIT:
1515 		test->udp_counters_64bit = 1;
1516 		break;
1517 	    case OPT_NO_FQ_SOCKET_PACING:
1518 #if defined(HAVE_SO_MAX_PACING_RATE)
1519 		printf("Warning:  --no-fq-socket-pacing is deprecated\n");
1520 		test->settings->fqrate = 0;
1521 		client_flag = 1;
1522 #else /* HAVE_SO_MAX_PACING_RATE */
1523 		i_errno = IEUNIMP;
1524 		return -1;
1525 #endif
1526 		break;
1527 	    case OPT_FQ_RATE:
1528 #if defined(HAVE_SO_MAX_PACING_RATE)
1529 		test->settings->fqrate = unit_atof_rate(optarg);
1530 		client_flag = 1;
1531 #else /* HAVE_SO_MAX_PACING_RATE */
1532 		i_errno = IEUNIMP;
1533 		return -1;
1534 #endif
1535 		break;
1536 #if defined(HAVE_DONT_FRAGMENT)
1537         case OPT_DONT_FRAGMENT:
1538             test->settings->dont_fragment = 1;
1539             client_flag = 1;
1540             break;
1541 #endif /* HAVE_DONT_FRAGMENT */
1542 #if defined(HAVE_SSL)
1543         case OPT_CLIENT_USERNAME:
1544             client_username = strdup(optarg);
1545             break;
1546         case OPT_CLIENT_RSA_PUBLIC_KEY:
1547             client_rsa_public_key = strdup(optarg);
1548             break;
1549         case OPT_SERVER_RSA_PRIVATE_KEY:
1550             server_rsa_private_key = strdup(optarg);
1551             break;
1552         case OPT_SERVER_AUTHORIZED_USERS:
1553             test->server_authorized_users = strdup(optarg);
1554             break;
1555         case OPT_SERVER_SKEW_THRESHOLD:
1556             test->server_skew_threshold = atoi(optarg);
1557             if(test->server_skew_threshold <= 0){
1558                 i_errno = IESKEWTHRESHOLD;
1559                 return -1;
1560             }
1561             break;
1562 #endif /* HAVE_SSL */
1563 	    case OPT_PACING_TIMER:
1564 		test->settings->pacing_timer = unit_atoi(optarg);
1565 		client_flag = 1;
1566 		break;
1567 	    case OPT_CONNECT_TIMEOUT:
1568 		test->settings->connect_timeout = unit_atoi(optarg);
1569 		client_flag = 1;
1570 		break;
1571 	    case 'h':
1572 		usage_long(stdout);
1573 		exit(0);
1574             default:
1575                 usage_long(stderr);
1576                 exit(1);
1577         }
1578     }
1579 
1580     /* Check flag / role compatibility. */
1581     if (test->role == 'c' && server_flag) {
1582         i_errno = IESERVERONLY;
1583         return -1;
1584     }
1585     if (test->role == 's' && client_flag) {
1586         i_errno = IECLIENTONLY;
1587         return -1;
1588     }
1589 
1590 #if defined(HAVE_SSL)
1591 
1592     if (test->role == 's' && (client_username || client_rsa_public_key)){
1593         i_errno = IECLIENTONLY;
1594         return -1;
1595     } else if (test->role == 'c' && (client_username || client_rsa_public_key) &&
1596         !(client_username && client_rsa_public_key)) {
1597         i_errno = IESETCLIENTAUTH;
1598         return -1;
1599     } else if (test->role == 'c' && (client_username && client_rsa_public_key)){
1600 
1601         char *client_password = NULL;
1602         size_t s;
1603         /* Need to copy env var, so we can do a common free */
1604         if ((client_password = getenv("IPERF3_PASSWORD")) != NULL)
1605              client_password = strdup(client_password);
1606         else if (iperf_getpass(&client_password, &s, stdin) < 0){
1607             i_errno = IESETCLIENTAUTH;
1608             return -1;
1609         }
1610         if (test_load_pubkey_from_file(client_rsa_public_key) < 0){
1611             iperf_err(test, "%s\n", ERR_error_string(ERR_get_error(), NULL));
1612             i_errno = IESETCLIENTAUTH;
1613             return -1;
1614         }
1615 
1616         test->settings->client_username = client_username;
1617         test->settings->client_password = client_password;
1618         test->settings->client_rsa_pubkey = load_pubkey_from_file(client_rsa_public_key);
1619 	free(client_rsa_public_key);
1620 	client_rsa_public_key = NULL;
1621     }
1622 
1623     if (test->role == 'c' && (server_rsa_private_key || test->server_authorized_users)){
1624         i_errno = IESERVERONLY;
1625         return -1;
1626     } else if (test->role == 'c' && (test->server_skew_threshold != 0)){
1627         i_errno = IESERVERONLY;
1628         return -1;
1629     } else if (test->role == 'c' && rcv_timeout_flag && test->mode == SENDER){
1630         i_errno = IERVRSONLYRCVTIMEOUT;
1631         return -1;
1632     } else if (test->role == 's' && (server_rsa_private_key || test->server_authorized_users) &&
1633         !(server_rsa_private_key && test->server_authorized_users)) {
1634          i_errno = IESETSERVERAUTH;
1635         return -1;
1636     } else if (test->role == 's' && server_rsa_private_key) {
1637         test->server_rsa_private_key = load_privkey_from_file(server_rsa_private_key);
1638         if (test->server_rsa_private_key == NULL){
1639             iperf_err(test, "%s\n", ERR_error_string(ERR_get_error(), NULL));
1640             i_errno = IESETSERVERAUTH;
1641             return -1;
1642         }
1643 	    free(server_rsa_private_key);
1644 	    server_rsa_private_key = NULL;
1645 
1646         if(test->server_skew_threshold == 0){
1647             // Set default value for time skew threshold
1648             test->server_skew_threshold=10;
1649         }
1650     }
1651 
1652 #endif //HAVE_SSL
1653     if (blksize == 0) {
1654 	if (test->protocol->id == Pudp)
1655 	    blksize = 0;	/* try to dynamically determine from MSS */
1656 	else if (test->protocol->id == Psctp)
1657 	    blksize = DEFAULT_SCTP_BLKSIZE;
1658 	else
1659 	    blksize = DEFAULT_TCP_BLKSIZE;
1660     }
1661     if ((test->protocol->id != Pudp && blksize <= 0)
1662 	|| blksize > MAX_BLOCKSIZE) {
1663 	i_errno = IEBLOCKSIZE;
1664 	return -1;
1665     }
1666     if (test->protocol->id == Pudp &&
1667 	(blksize > 0 &&
1668 	    (blksize < MIN_UDP_BLOCKSIZE || blksize > MAX_UDP_BLOCKSIZE))) {
1669 	i_errno = IEUDPBLOCKSIZE;
1670 	return -1;
1671     }
1672     test->settings->blksize = blksize;
1673 
1674     if (!rate_flag)
1675 	test->settings->rate = test->protocol->id == Pudp ? UDP_RATE : 0;
1676 
1677     /* if no bytes or blocks specified, nor a duration_flag, and we have -F,
1678     ** get the file-size as the bytes count to be transferred
1679     */
1680     if (test->settings->bytes == 0 &&
1681         test->settings->blocks == 0 &&
1682         ! duration_flag &&
1683         test->diskfile_name != (char*) 0 &&
1684         test->role == 'c'
1685         ){
1686         struct stat st;
1687         if( stat(test->diskfile_name, &st) == 0 ){
1688             iperf_size_t file_bytes = st.st_size;
1689             test->settings->bytes = file_bytes;
1690             if (test->debug)
1691                 printf("End condition set to file-size: %"PRIu64" bytes\n", test->settings->bytes);
1692         }
1693         // if failing to read file stat, it should fallback to default duration mode
1694     }
1695 
1696     if ((test->settings->bytes != 0 || test->settings->blocks != 0) && ! duration_flag)
1697         test->duration = 0;
1698 
1699     /* Disallow specifying multiple test end conditions. The code actually
1700     ** works just fine without this prohibition. As soon as any one of the
1701     ** three possible end conditions is met, the test ends. So this check
1702     ** could be removed if desired.
1703     */
1704     if ((duration_flag && test->settings->bytes != 0) ||
1705         (duration_flag && test->settings->blocks != 0) ||
1706 	(test->settings->bytes != 0 && test->settings->blocks != 0)) {
1707         i_errno = IEENDCONDITIONS;
1708         return -1;
1709     }
1710 
1711     /* For subsequent calls to getopt */
1712 #ifdef __APPLE__
1713     optreset = 1;
1714 #endif
1715     optind = 0;
1716 
1717     if ((test->role != 'c') && (test->role != 's')) {
1718         i_errno = IENOROLE;
1719         return -1;
1720     }
1721 
1722     /* Set Total-rate average interval to multiplicity of State interval */
1723     if (test->settings->bitrate_limit_interval != 0) {
1724 	test->settings->bitrate_limit_stats_per_interval =
1725 	    (test->settings->bitrate_limit_interval <= test->stats_interval ?
1726 	    1 : round(test->settings->bitrate_limit_interval/test->stats_interval) );
1727     }
1728 
1729     /* Show warning if JSON output is used with explicit report format */
1730     if ((test->json_output) && (test->settings->unit_format != 'a')) {
1731         warning("Report format (-f) flag ignored with JSON output (-J)");
1732     }
1733 
1734     /* Show warning if JSON output is used with verbose or debug flags */
1735     if (test->json_output && test->verbose) {
1736         warning("Verbose output (-v) may interfere with JSON output (-J)");
1737     }
1738     if (test->json_output && test->debug) {
1739         warning("Debug output (-d) may interfere with JSON output (-J)");
1740     }
1741 
1742     return 0;
1743 }
1744 
1745 /*
1746  * Open the file specified by test->logfile and set test->outfile to its' FD.
1747  */
iperf_open_logfile(struct iperf_test * test)1748 int iperf_open_logfile(struct iperf_test *test)
1749 {
1750     test->outfile = fopen(test->logfile, "a+");
1751     if (test->outfile == NULL) {
1752         i_errno = IELOGFILE;
1753         return -1;
1754     }
1755 
1756     return 0;
1757 }
1758 
1759 int
iperf_set_send_state(struct iperf_test * test,signed char state)1760 iperf_set_send_state(struct iperf_test *test, signed char state)
1761 {
1762     if (test->ctrl_sck >= 0) {
1763         test->state = state;
1764         if (Nwrite(test->ctrl_sck, (char*) &state, sizeof(state), Ptcp) < 0) {
1765 	    i_errno = IESENDMESSAGE;
1766 	    return -1;
1767         }
1768     }
1769     return 0;
1770 }
1771 
1772 void
iperf_check_throttle(struct iperf_stream * sp,struct iperf_time * nowP)1773 iperf_check_throttle(struct iperf_stream *sp, struct iperf_time *nowP)
1774 {
1775     struct iperf_time temp_time;
1776     double seconds;
1777     uint64_t bits_per_second;
1778 
1779     if (sp->test->done || sp->test->settings->rate == 0)
1780         return;
1781     iperf_time_diff(&sp->result->start_time_fixed, nowP, &temp_time);
1782     seconds = iperf_time_in_secs(&temp_time);
1783     bits_per_second = sp->result->bytes_sent * 8 / seconds;
1784     if (bits_per_second < sp->test->settings->rate) {
1785         sp->green_light = 1;
1786         FD_SET(sp->socket, &sp->test->write_set);
1787     } else {
1788         sp->green_light = 0;
1789         FD_CLR(sp->socket, &sp->test->write_set);
1790     }
1791 }
1792 
1793 /* Verify that average traffic is not greater than the specified limit */
1794 void
iperf_check_total_rate(struct iperf_test * test,iperf_size_t last_interval_bytes_transferred)1795 iperf_check_total_rate(struct iperf_test *test, iperf_size_t last_interval_bytes_transferred)
1796 {
1797     double seconds;
1798     uint64_t bits_per_second;
1799     iperf_size_t total_bytes;
1800     int i;
1801 
1802     if (test->done || test->settings->bitrate_limit == 0)    // Continue only if check should be done
1803         return;
1804 
1805     /* Add last inetrval's transferred bytes to the array */
1806     if (++test->bitrate_limit_last_interval_index >= test->settings->bitrate_limit_stats_per_interval)
1807         test->bitrate_limit_last_interval_index = 0;
1808     test->bitrate_limit_intervals_traffic_bytes[test->bitrate_limit_last_interval_index] = last_interval_bytes_transferred;
1809 
1810     /* Ensure that enough stats periods passed to allow averaging throughput */
1811     test->bitrate_limit_stats_count += 1;
1812     if (test->bitrate_limit_stats_count < test->settings->bitrate_limit_stats_per_interval)
1813         return;
1814 
1815      /* Calculating total bytes traffic to be averaged */
1816     for (total_bytes = 0, i = 0; i < test->settings->bitrate_limit_stats_per_interval; i++) {
1817         total_bytes += test->bitrate_limit_intervals_traffic_bytes[i];
1818     }
1819 
1820     seconds = test->stats_interval * test->settings->bitrate_limit_stats_per_interval;
1821     bits_per_second = total_bytes * 8 / seconds;
1822     if (test->debug) {
1823         iperf_printf(test,"Interval %" PRIu64 " - throughput %" PRIu64 " bps (limit %" PRIu64 ")\n", test->bitrate_limit_stats_count, bits_per_second, test->settings->bitrate_limit);
1824     }
1825 
1826     if (bits_per_second  > test->settings->bitrate_limit) {
1827         if (iperf_get_verbose(test))
1828             iperf_err(test, "Total throughput of %" PRIu64 " bps exceeded %" PRIu64 " bps limit", bits_per_second, test->settings->bitrate_limit);
1829 	test->bitrate_limit_exceeded = 1;
1830     }
1831 }
1832 
1833 int
iperf_send(struct iperf_test * test,fd_set * write_setP)1834 iperf_send(struct iperf_test *test, fd_set *write_setP)
1835 {
1836     register int multisend, r, streams_active;
1837     register struct iperf_stream *sp;
1838     struct iperf_time now;
1839     int no_throttle_check;
1840 
1841     /* Can we do multisend mode? */
1842     if (test->settings->burst != 0)
1843         multisend = test->settings->burst;
1844     else if (test->settings->rate == 0)
1845         multisend = test->multisend;
1846     else
1847         multisend = 1;	/* nope */
1848 
1849     /* Should bitrate throttle be checked for every send */
1850     no_throttle_check = test->settings->rate != 0 && test->settings->burst == 0;
1851 
1852     for (; multisend > 0; --multisend) {
1853 	if (no_throttle_check)
1854 	    iperf_time_now(&now);
1855 	streams_active = 0;
1856 	SLIST_FOREACH(sp, &test->streams, streams) {
1857 	    if ((sp->green_light && sp->sender &&
1858 		 (write_setP == NULL || FD_ISSET(sp->socket, write_setP)))) {
1859         if (multisend > 1 && test->settings->bytes != 0 && test->bytes_sent >= test->settings->bytes)
1860             break;
1861         if (multisend > 1 && test->settings->blocks != 0 && test->blocks_sent >= test->settings->blocks)
1862             break;
1863 		if ((r = sp->snd(sp)) < 0) {
1864 		    if (r == NET_SOFTERROR)
1865 			break;
1866 		    i_errno = IESTREAMWRITE;
1867 		    return r;
1868 		}
1869 		streams_active = 1;
1870 		test->bytes_sent += r;
1871 		if (!sp->pending_size)
1872 		    ++test->blocks_sent;
1873                 if (no_throttle_check)
1874 		    iperf_check_throttle(sp, &now);
1875 	    }
1876 	}
1877 	if (!streams_active)
1878 	    break;
1879     }
1880     if (!no_throttle_check) {   /* Throttle check if was not checked for each send */
1881 	iperf_time_now(&now);
1882 	SLIST_FOREACH(sp, &test->streams, streams)
1883 	    if (sp->sender)
1884 	        iperf_check_throttle(sp, &now);
1885     }
1886     if (write_setP != NULL)
1887 	SLIST_FOREACH(sp, &test->streams, streams)
1888 	    if (FD_ISSET(sp->socket, write_setP))
1889 		FD_CLR(sp->socket, write_setP);
1890 
1891     return 0;
1892 }
1893 
1894 int
iperf_recv(struct iperf_test * test,fd_set * read_setP)1895 iperf_recv(struct iperf_test *test, fd_set *read_setP)
1896 {
1897     int r;
1898     struct iperf_stream *sp;
1899 
1900     SLIST_FOREACH(sp, &test->streams, streams) {
1901 	if (FD_ISSET(sp->socket, read_setP) && !sp->sender) {
1902 	    if ((r = sp->rcv(sp)) < 0) {
1903 		i_errno = IESTREAMREAD;
1904 		return r;
1905 	    }
1906 	    test->bytes_received += r;
1907 	    ++test->blocks_received;
1908 	    FD_CLR(sp->socket, read_setP);
1909 	}
1910     }
1911 
1912     return 0;
1913 }
1914 
1915 int
iperf_init_test(struct iperf_test * test)1916 iperf_init_test(struct iperf_test *test)
1917 {
1918     struct iperf_time now;
1919     struct iperf_stream *sp;
1920 
1921     if (test->protocol->init) {
1922         if (test->protocol->init(test) < 0)
1923             return -1;
1924     }
1925 
1926     /* Init each stream. */
1927     if (iperf_time_now(&now) < 0) {
1928 	i_errno = IEINITTEST;
1929 	return -1;
1930     }
1931     SLIST_FOREACH(sp, &test->streams, streams) {
1932 	sp->result->start_time = sp->result->start_time_fixed = now;
1933     }
1934 
1935     if (test->on_test_start)
1936         test->on_test_start(test);
1937 
1938     return 0;
1939 }
1940 
1941 static void
send_timer_proc(TimerClientData client_data,struct iperf_time * nowP)1942 send_timer_proc(TimerClientData client_data, struct iperf_time *nowP)
1943 {
1944     struct iperf_stream *sp = client_data.p;
1945 
1946     /* All we do here is set or clear the flag saying that this stream may
1947     ** be sent to.  The actual sending gets done in the send proc, after
1948     ** checking the flag.
1949     */
1950     iperf_check_throttle(sp, nowP);
1951 }
1952 
1953 int
iperf_create_send_timers(struct iperf_test * test)1954 iperf_create_send_timers(struct iperf_test * test)
1955 {
1956     struct iperf_time now;
1957     struct iperf_stream *sp;
1958     TimerClientData cd;
1959 
1960     if (iperf_time_now(&now) < 0) {
1961 	i_errno = IEINITTEST;
1962 	return -1;
1963     }
1964     SLIST_FOREACH(sp, &test->streams, streams) {
1965         sp->green_light = 1;
1966 	if (test->settings->rate != 0 && sp->sender) {
1967 	    cd.p = sp;
1968 	    sp->send_timer = tmr_create(NULL, send_timer_proc, cd, test->settings->pacing_timer, 1);
1969 	    if (sp->send_timer == NULL) {
1970 		i_errno = IEINITTEST;
1971 		return -1;
1972 	    }
1973 	}
1974     }
1975     return 0;
1976 }
1977 
1978 #if defined(HAVE_SSL)
test_is_authorized(struct iperf_test * test)1979 int test_is_authorized(struct iperf_test *test){
1980     if ( !(test->server_rsa_private_key && test->server_authorized_users)) {
1981         return 0;
1982     }
1983 
1984     if (test->settings->authtoken){
1985         char *username = NULL, *password = NULL;
1986         time_t ts;
1987         int rc = decode_auth_setting(test->debug, test->settings->authtoken, test->server_rsa_private_key, &username, &password, &ts);
1988 	if (rc) {
1989 	    return -1;
1990 	}
1991         int ret = check_authentication(username, password, ts, test->server_authorized_users, test->server_skew_threshold);
1992         if (ret == 0){
1993             if (test->debug) {
1994               iperf_printf(test, report_authentication_succeeded, username, ts);
1995             }
1996             free(username);
1997             free(password);
1998             return 0;
1999         } else {
2000             if (test->debug) {
2001                 iperf_printf(test, report_authentication_failed, username, ts);
2002             }
2003             free(username);
2004             free(password);
2005             return -1;
2006         }
2007     }
2008     return -1;
2009 }
2010 #endif //HAVE_SSL
2011 
2012 /**
2013  * iperf_exchange_parameters - handles the param_Exchange part for client
2014  *
2015  */
2016 
2017 int
iperf_exchange_parameters(struct iperf_test * test)2018 iperf_exchange_parameters(struct iperf_test *test)
2019 {
2020     int s;
2021     int32_t err;
2022 
2023     if (test->role == 'c') {
2024 
2025         if (send_parameters(test) < 0)
2026             return -1;
2027 
2028     } else {
2029 
2030         if (get_parameters(test) < 0)
2031             return -1;
2032 
2033 #if defined(HAVE_SSL)
2034         if (test_is_authorized(test) < 0){
2035             if (iperf_set_send_state(test, SERVER_ERROR) != 0)
2036                 return -1;
2037             i_errno = IEAUTHTEST;
2038             err = htonl(i_errno);
2039             if (Nwrite(test->ctrl_sck, (char*) &err, sizeof(err), Ptcp) < 0) {
2040                 i_errno = IECTRLWRITE;
2041                 return -1;
2042             }
2043             return -1;
2044         }
2045 #endif //HAVE_SSL
2046 
2047         if ((s = test->protocol->listen(test)) < 0) {
2048 	        if (iperf_set_send_state(test, SERVER_ERROR) != 0)
2049                 return -1;
2050             err = htonl(i_errno);
2051             if (Nwrite(test->ctrl_sck, (char*) &err, sizeof(err), Ptcp) < 0) {
2052                 i_errno = IECTRLWRITE;
2053                 return -1;
2054             }
2055             err = htonl(errno);
2056             if (Nwrite(test->ctrl_sck, (char*) &err, sizeof(err), Ptcp) < 0) {
2057                 i_errno = IECTRLWRITE;
2058                 return -1;
2059             }
2060             return -1;
2061         }
2062 
2063         FD_SET(s, &test->read_set);
2064         test->max_fd = (s > test->max_fd) ? s : test->max_fd;
2065         test->prot_listener = s;
2066 
2067         // Send the control message to create streams and start the test
2068 	if (iperf_set_send_state(test, CREATE_STREAMS) != 0)
2069             return -1;
2070 
2071     }
2072 
2073     return 0;
2074 }
2075 
2076 /*************************************************************/
2077 
2078 int
iperf_exchange_results(struct iperf_test * test)2079 iperf_exchange_results(struct iperf_test *test)
2080 {
2081     if (test->role == 'c') {
2082         /* Send results to server. */
2083 	if (send_results(test) < 0)
2084             return -1;
2085         /* Get server results. */
2086         if (get_results(test) < 0)
2087             return -1;
2088     } else {
2089         /* Get client results. */
2090         if (get_results(test) < 0)
2091             return -1;
2092         /* Send results to client. */
2093 	if (send_results(test) < 0)
2094             return -1;
2095     }
2096     return 0;
2097 }
2098 
2099 /*************************************************************/
2100 
2101 static int
send_parameters(struct iperf_test * test)2102 send_parameters(struct iperf_test *test)
2103 {
2104     int r = 0;
2105     cJSON *j;
2106 
2107     j = cJSON_CreateObject();
2108     if (j == NULL) {
2109 	i_errno = IESENDPARAMS;
2110 	r = -1;
2111     } else {
2112 	if (test->protocol->id == Ptcp)
2113 	    cJSON_AddTrueToObject(j, "tcp");
2114 	else if (test->protocol->id == Pudp)
2115 	    cJSON_AddTrueToObject(j, "udp");
2116         else if (test->protocol->id == Psctp)
2117             cJSON_AddTrueToObject(j, "sctp");
2118 	cJSON_AddNumberToObject(j, "omit", test->omit);
2119 	if (test->server_affinity != -1)
2120 	    cJSON_AddNumberToObject(j, "server_affinity", test->server_affinity);
2121 	cJSON_AddNumberToObject(j, "time", test->duration);
2122 	if (test->settings->bytes)
2123 	    cJSON_AddNumberToObject(j, "num", test->settings->bytes);
2124 	if (test->settings->blocks)
2125 	    cJSON_AddNumberToObject(j, "blockcount", test->settings->blocks);
2126 	if (test->settings->mss)
2127 	    cJSON_AddNumberToObject(j, "MSS", test->settings->mss);
2128 	if (test->no_delay)
2129 	    cJSON_AddTrueToObject(j, "nodelay");
2130 	cJSON_AddNumberToObject(j, "parallel", test->num_streams);
2131 	if (test->reverse)
2132 	    cJSON_AddTrueToObject(j, "reverse");
2133 	if (test->bidirectional)
2134 	            cJSON_AddTrueToObject(j, "bidirectional");
2135 	if (test->settings->socket_bufsize)
2136 	    cJSON_AddNumberToObject(j, "window", test->settings->socket_bufsize);
2137 	if (test->settings->blksize)
2138 	    cJSON_AddNumberToObject(j, "len", test->settings->blksize);
2139 	if (test->settings->rate)
2140 	    cJSON_AddNumberToObject(j, "bandwidth", test->settings->rate);
2141 	if (test->settings->fqrate)
2142 	    cJSON_AddNumberToObject(j, "fqrate", test->settings->fqrate);
2143 	if (test->settings->pacing_timer)
2144 	    cJSON_AddNumberToObject(j, "pacing_timer", test->settings->pacing_timer);
2145 	if (test->settings->burst)
2146 	    cJSON_AddNumberToObject(j, "burst", test->settings->burst);
2147 	if (test->settings->tos)
2148 	    cJSON_AddNumberToObject(j, "TOS", test->settings->tos);
2149 	if (test->settings->flowlabel)
2150 	    cJSON_AddNumberToObject(j, "flowlabel", test->settings->flowlabel);
2151 	if (test->title)
2152 	    cJSON_AddStringToObject(j, "title", test->title);
2153 	if (test->extra_data)
2154 	    cJSON_AddStringToObject(j, "extra_data", test->extra_data);
2155 	if (test->congestion)
2156 	    cJSON_AddStringToObject(j, "congestion", test->congestion);
2157 	if (test->congestion_used)
2158 	    cJSON_AddStringToObject(j, "congestion_used", test->congestion_used);
2159 	if (test->get_server_output)
2160 	    cJSON_AddNumberToObject(j, "get_server_output", iperf_get_test_get_server_output(test));
2161 	if (test->udp_counters_64bit)
2162 	    cJSON_AddNumberToObject(j, "udp_counters_64bit", iperf_get_test_udp_counters_64bit(test));
2163 	if (test->repeating_payload)
2164 	    cJSON_AddNumberToObject(j, "repeating_payload", test->repeating_payload);
2165 	if (test->zerocopy)
2166 	    cJSON_AddNumberToObject(j, "zerocopy", test->zerocopy);
2167 #if defined(HAVE_DONT_FRAGMENT)
2168 	if (test->settings->dont_fragment)
2169 	    cJSON_AddNumberToObject(j, "dont_fragment", test->settings->dont_fragment);
2170 #endif /* HAVE_DONT_FRAGMENT */
2171 #if defined(HAVE_SSL)
2172 	/* Send authentication parameters */
2173 	if (test->settings->client_username && test->settings->client_password && test->settings->client_rsa_pubkey){
2174 	    int rc = encode_auth_setting(test->settings->client_username, test->settings->client_password, test->settings->client_rsa_pubkey, &test->settings->authtoken);
2175 
2176 	    if (rc) {
2177 		cJSON_Delete(j);
2178 		i_errno = IESENDPARAMS;
2179 		return -1;
2180 	    }
2181 
2182 	    cJSON_AddStringToObject(j, "authtoken", test->settings->authtoken);
2183 	}
2184 #endif // HAVE_SSL
2185 	cJSON_AddStringToObject(j, "client_version", IPERF_VERSION);
2186 
2187 	if (test->debug) {
2188 	    char *str = cJSON_Print(j);
2189 	    printf("send_parameters:\n%s\n", str);
2190 	    cJSON_free(str);
2191 	}
2192 
2193 	if (JSON_write(test->ctrl_sck, j) < 0) {
2194 	    i_errno = IESENDPARAMS;
2195 	    r = -1;
2196 	}
2197 	cJSON_Delete(j);
2198     }
2199     return r;
2200 }
2201 
2202 /*************************************************************/
2203 
2204 static int
get_parameters(struct iperf_test * test)2205 get_parameters(struct iperf_test *test)
2206 {
2207     int r = 0;
2208     cJSON *j;
2209     cJSON *j_p;
2210 
2211     j = JSON_read(test->ctrl_sck);
2212     if (j == NULL) {
2213 	i_errno = IERECVPARAMS;
2214         r = -1;
2215     } else {
2216 	if (test->debug) {
2217             char *str;
2218             str = cJSON_Print(j);
2219             printf("get_parameters:\n%s\n", str );
2220             cJSON_free(str);
2221 	}
2222 
2223 	if ((j_p = cJSON_GetObjectItem(j, "tcp")) != NULL)
2224 	    set_protocol(test, Ptcp);
2225 	if ((j_p = cJSON_GetObjectItem(j, "udp")) != NULL)
2226 	    set_protocol(test, Pudp);
2227         if ((j_p = cJSON_GetObjectItem(j, "sctp")) != NULL)
2228             set_protocol(test, Psctp);
2229 	if ((j_p = cJSON_GetObjectItem(j, "omit")) != NULL)
2230 	    test->omit = j_p->valueint;
2231 	if ((j_p = cJSON_GetObjectItem(j, "server_affinity")) != NULL)
2232 	    test->server_affinity = j_p->valueint;
2233 	if ((j_p = cJSON_GetObjectItem(j, "time")) != NULL)
2234 	    test->duration = j_p->valueint;
2235 	if ((j_p = cJSON_GetObjectItem(j, "num")) != NULL)
2236 	    test->settings->bytes = j_p->valueint;
2237 	if ((j_p = cJSON_GetObjectItem(j, "blockcount")) != NULL)
2238 	    test->settings->blocks = j_p->valueint;
2239 	if ((j_p = cJSON_GetObjectItem(j, "MSS")) != NULL)
2240 	    test->settings->mss = j_p->valueint;
2241 	if ((j_p = cJSON_GetObjectItem(j, "nodelay")) != NULL)
2242 	    test->no_delay = 1;
2243 	if ((j_p = cJSON_GetObjectItem(j, "parallel")) != NULL)
2244 	    test->num_streams = j_p->valueint;
2245 	if ((j_p = cJSON_GetObjectItem(j, "reverse")) != NULL)
2246 	    iperf_set_test_reverse(test, 1);
2247         if ((j_p = cJSON_GetObjectItem(j, "bidirectional")) != NULL)
2248             iperf_set_test_bidirectional(test, 1);
2249 	if ((j_p = cJSON_GetObjectItem(j, "window")) != NULL)
2250 	    test->settings->socket_bufsize = j_p->valueint;
2251 	if ((j_p = cJSON_GetObjectItem(j, "len")) != NULL)
2252 	    test->settings->blksize = j_p->valueint;
2253 	if ((j_p = cJSON_GetObjectItem(j, "bandwidth")) != NULL)
2254 	    test->settings->rate = j_p->valueint;
2255 	if ((j_p = cJSON_GetObjectItem(j, "fqrate")) != NULL)
2256 	    test->settings->fqrate = j_p->valueint;
2257 	if ((j_p = cJSON_GetObjectItem(j, "pacing_timer")) != NULL)
2258 	    test->settings->pacing_timer = j_p->valueint;
2259 	if ((j_p = cJSON_GetObjectItem(j, "burst")) != NULL)
2260 	    test->settings->burst = j_p->valueint;
2261 	if ((j_p = cJSON_GetObjectItem(j, "TOS")) != NULL)
2262 	    test->settings->tos = j_p->valueint;
2263 	if ((j_p = cJSON_GetObjectItem(j, "flowlabel")) != NULL)
2264 	    test->settings->flowlabel = j_p->valueint;
2265 	if ((j_p = cJSON_GetObjectItem(j, "title")) != NULL)
2266 	    test->title = strdup(j_p->valuestring);
2267 	if ((j_p = cJSON_GetObjectItem(j, "extra_data")) != NULL)
2268 	    test->extra_data = strdup(j_p->valuestring);
2269 	if ((j_p = cJSON_GetObjectItem(j, "congestion")) != NULL)
2270 	    test->congestion = strdup(j_p->valuestring);
2271 	if ((j_p = cJSON_GetObjectItem(j, "congestion_used")) != NULL)
2272 	    test->congestion_used = strdup(j_p->valuestring);
2273 	if ((j_p = cJSON_GetObjectItem(j, "get_server_output")) != NULL)
2274 	    iperf_set_test_get_server_output(test, 1);
2275 	if ((j_p = cJSON_GetObjectItem(j, "udp_counters_64bit")) != NULL)
2276 	    iperf_set_test_udp_counters_64bit(test, 1);
2277 	if ((j_p = cJSON_GetObjectItem(j, "repeating_payload")) != NULL)
2278 	    test->repeating_payload = 1;
2279 	if ((j_p = cJSON_GetObjectItem(j, "zerocopy")) != NULL)
2280 	    test->zerocopy = j_p->valueint;
2281 #if defined(HAVE_DONT_FRAGMENT)
2282 	if ((j_p = cJSON_GetObjectItem(j, "dont_fragment")) != NULL)
2283 	    test->settings->dont_fragment = j_p->valueint;
2284 #endif /* HAVE_DONT_FRAGMENT */
2285 #if defined(HAVE_SSL)
2286 	if ((j_p = cJSON_GetObjectItem(j, "authtoken")) != NULL)
2287         test->settings->authtoken = strdup(j_p->valuestring);
2288 #endif //HAVE_SSL
2289 	if (test->mode && test->protocol->id == Ptcp && has_tcpinfo_retransmits())
2290 	    test->sender_has_retransmits = 1;
2291 	if (test->settings->rate)
2292 	    cJSON_AddNumberToObject(test->json_start, "target_bitrate", test->settings->rate);
2293 	cJSON_Delete(j);
2294     }
2295     return r;
2296 }
2297 
2298 /*************************************************************/
2299 
2300 static int
send_results(struct iperf_test * test)2301 send_results(struct iperf_test *test)
2302 {
2303     int r = 0;
2304     cJSON *j;
2305     cJSON *j_streams;
2306     struct iperf_stream *sp;
2307     cJSON *j_stream;
2308     int sender_has_retransmits;
2309     iperf_size_t bytes_transferred;
2310     int retransmits;
2311     struct iperf_time temp_time;
2312     double start_time, end_time;
2313 
2314     j = cJSON_CreateObject();
2315     if (j == NULL) {
2316 	i_errno = IEPACKAGERESULTS;
2317 	r = -1;
2318     } else {
2319 	cJSON_AddNumberToObject(j, "cpu_util_total", test->cpu_util[0]);
2320 	cJSON_AddNumberToObject(j, "cpu_util_user", test->cpu_util[1]);
2321 	cJSON_AddNumberToObject(j, "cpu_util_system", test->cpu_util[2]);
2322 	if ( test->mode == RECEIVER )
2323 	    sender_has_retransmits = -1;
2324 	else
2325 	    sender_has_retransmits = test->sender_has_retransmits;
2326 	cJSON_AddNumberToObject(j, "sender_has_retransmits", sender_has_retransmits);
2327 	if ( test->congestion_used ) {
2328 	    cJSON_AddStringToObject(j, "congestion_used", test->congestion_used);
2329 	}
2330 
2331 	/* If on the server and sending server output, then do this */
2332 	if (test->role == 's' && test->get_server_output) {
2333 	    if (test->json_output) {
2334 		/* Add JSON output */
2335 		cJSON_AddItemReferenceToObject(j, "server_output_json", test->json_top);
2336 	    }
2337 	    else {
2338 		/* Add textual output */
2339 		size_t buflen = 0;
2340 
2341 		/* Figure out how much room we need to hold the complete output string */
2342 		struct iperf_textline *t;
2343 		TAILQ_FOREACH(t, &(test->server_output_list), textlineentries) {
2344 		    buflen += strlen(t->line);
2345 		}
2346 
2347 		/* Allocate and build it up from the component lines */
2348 		char *output = calloc(buflen + 1, 1);
2349 		TAILQ_FOREACH(t, &(test->server_output_list), textlineentries) {
2350 		    strncat(output, t->line, buflen);
2351 		    buflen -= strlen(t->line);
2352 		}
2353 
2354 		cJSON_AddStringToObject(j, "server_output_text", output);
2355         free(output);
2356 	    }
2357 	}
2358 
2359 	j_streams = cJSON_CreateArray();
2360 	if (j_streams == NULL) {
2361 	    i_errno = IEPACKAGERESULTS;
2362 	    r = -1;
2363 	} else {
2364 	    cJSON_AddItemToObject(j, "streams", j_streams);
2365 	    SLIST_FOREACH(sp, &test->streams, streams) {
2366 		j_stream = cJSON_CreateObject();
2367 		if (j_stream == NULL) {
2368 		    i_errno = IEPACKAGERESULTS;
2369 		    r = -1;
2370 		} else {
2371 		    cJSON_AddItemToArray(j_streams, j_stream);
2372 		    bytes_transferred = sp->sender ? (sp->result->bytes_sent - sp->result->bytes_sent_omit) : sp->result->bytes_received;
2373 		    retransmits = (sp->sender && test->sender_has_retransmits) ? sp->result->stream_retrans : -1;
2374 		    cJSON_AddNumberToObject(j_stream, "id", sp->id);
2375 		    cJSON_AddNumberToObject(j_stream, "bytes", bytes_transferred);
2376 		    cJSON_AddNumberToObject(j_stream, "retransmits", retransmits);
2377 		    cJSON_AddNumberToObject(j_stream, "jitter", sp->jitter);
2378 		    cJSON_AddNumberToObject(j_stream, "errors", sp->cnt_error);
2379 		    cJSON_AddNumberToObject(j_stream, "packets", sp->packet_count);
2380 
2381 		    iperf_time_diff(&sp->result->start_time, &sp->result->start_time, &temp_time);
2382 		    start_time = iperf_time_in_secs(&temp_time);
2383 		    iperf_time_diff(&sp->result->start_time, &sp->result->end_time, &temp_time);
2384 		    end_time = iperf_time_in_secs(&temp_time);
2385 		    cJSON_AddNumberToObject(j_stream, "start_time", start_time);
2386 		    cJSON_AddNumberToObject(j_stream, "end_time", end_time);
2387 
2388 		}
2389 	    }
2390 	    if (r == 0 && test->debug) {
2391                 char *str = cJSON_Print(j);
2392 		printf("send_results\n%s\n", str);
2393                 cJSON_free(str);
2394 	    }
2395 	    if (r == 0 && JSON_write(test->ctrl_sck, j) < 0) {
2396 		i_errno = IESENDRESULTS;
2397 		r = -1;
2398 	    }
2399 	}
2400 	cJSON_Delete(j);
2401     }
2402     return r;
2403 }
2404 
2405 /*************************************************************/
2406 
2407 static int
get_results(struct iperf_test * test)2408 get_results(struct iperf_test *test)
2409 {
2410     int r = 0;
2411     cJSON *j;
2412     cJSON *j_cpu_util_total;
2413     cJSON *j_cpu_util_user;
2414     cJSON *j_cpu_util_system;
2415     cJSON *j_remote_congestion_used;
2416     cJSON *j_sender_has_retransmits;
2417     int result_has_retransmits;
2418     cJSON *j_streams;
2419     int n, i;
2420     cJSON *j_stream;
2421     cJSON *j_id;
2422     cJSON *j_bytes;
2423     cJSON *j_retransmits;
2424     cJSON *j_jitter;
2425     cJSON *j_errors;
2426     cJSON *j_packets;
2427     cJSON *j_server_output;
2428     cJSON *j_start_time, *j_end_time;
2429     int sid, cerror, pcount;
2430     double jitter;
2431     iperf_size_t bytes_transferred;
2432     int retransmits;
2433     struct iperf_stream *sp;
2434 
2435     j = JSON_read(test->ctrl_sck);
2436     if (j == NULL) {
2437 	i_errno = IERECVRESULTS;
2438         r = -1;
2439     } else {
2440 	j_cpu_util_total = cJSON_GetObjectItem(j, "cpu_util_total");
2441 	j_cpu_util_user = cJSON_GetObjectItem(j, "cpu_util_user");
2442 	j_cpu_util_system = cJSON_GetObjectItem(j, "cpu_util_system");
2443 	j_sender_has_retransmits = cJSON_GetObjectItem(j, "sender_has_retransmits");
2444 	if (j_cpu_util_total == NULL || j_cpu_util_user == NULL || j_cpu_util_system == NULL || j_sender_has_retransmits == NULL) {
2445 	    i_errno = IERECVRESULTS;
2446 	    r = -1;
2447 	} else {
2448 	    if (test->debug) {
2449                 char *str = cJSON_Print(j);
2450                 printf("get_results\n%s\n", str);
2451                 cJSON_free(str);
2452 	    }
2453 
2454 	    test->remote_cpu_util[0] = j_cpu_util_total->valuedouble;
2455 	    test->remote_cpu_util[1] = j_cpu_util_user->valuedouble;
2456 	    test->remote_cpu_util[2] = j_cpu_util_system->valuedouble;
2457 	    result_has_retransmits = j_sender_has_retransmits->valueint;
2458 	    if ( test->mode == RECEIVER ) {
2459 	        test->sender_has_retransmits = result_has_retransmits;
2460 	        test->other_side_has_retransmits = 0;
2461 	    }
2462 	    else if ( test->mode == BIDIRECTIONAL )
2463 	        test->other_side_has_retransmits = result_has_retransmits;
2464 
2465 	    j_streams = cJSON_GetObjectItem(j, "streams");
2466 	    if (j_streams == NULL) {
2467 		i_errno = IERECVRESULTS;
2468 		r = -1;
2469 	    } else {
2470 	        n = cJSON_GetArraySize(j_streams);
2471 		for (i=0; i<n; ++i) {
2472 		    j_stream = cJSON_GetArrayItem(j_streams, i);
2473 		    if (j_stream == NULL) {
2474 			i_errno = IERECVRESULTS;
2475 			r = -1;
2476 		    } else {
2477 			j_id = cJSON_GetObjectItem(j_stream, "id");
2478 			j_bytes = cJSON_GetObjectItem(j_stream, "bytes");
2479 			j_retransmits = cJSON_GetObjectItem(j_stream, "retransmits");
2480 			j_jitter = cJSON_GetObjectItem(j_stream, "jitter");
2481 			j_errors = cJSON_GetObjectItem(j_stream, "errors");
2482 			j_packets = cJSON_GetObjectItem(j_stream, "packets");
2483 			j_start_time = cJSON_GetObjectItem(j_stream, "start_time");
2484 			j_end_time = cJSON_GetObjectItem(j_stream, "end_time");
2485 			if (j_id == NULL || j_bytes == NULL || j_retransmits == NULL || j_jitter == NULL || j_errors == NULL || j_packets == NULL) {
2486 			    i_errno = IERECVRESULTS;
2487 			    r = -1;
2488 			} else {
2489 			    sid = j_id->valueint;
2490 			    bytes_transferred = j_bytes->valueint;
2491 			    retransmits = j_retransmits->valueint;
2492 			    jitter = j_jitter->valuedouble;
2493 			    cerror = j_errors->valueint;
2494 			    pcount = j_packets->valueint;
2495 			    SLIST_FOREACH(sp, &test->streams, streams)
2496 				if (sp->id == sid) break;
2497 			    if (sp == NULL) {
2498 				i_errno = IESTREAMID;
2499 				r = -1;
2500 			    } else {
2501 				if (sp->sender) {
2502 				    sp->jitter = jitter;
2503 				    sp->cnt_error = cerror;
2504 				    sp->peer_packet_count = pcount;
2505 				    sp->result->bytes_received = bytes_transferred;
2506 				    /*
2507 				     * We have to handle the possibility that
2508 				     * start_time and end_time might not be
2509 				     * available; this is the case for older (pre-3.2)
2510 				     * servers.
2511 				     *
2512 				     * We need to have result structure members to hold
2513 				     * the both sides' start_time and end_time.
2514 				     */
2515 				    if (j_start_time && j_end_time) {
2516 					sp->result->receiver_time = j_end_time->valuedouble - j_start_time->valuedouble;
2517 				    }
2518 				    else {
2519 					sp->result->receiver_time = 0.0;
2520 				    }
2521 				} else {
2522 				    sp->peer_packet_count = pcount;
2523 				    sp->result->bytes_sent = bytes_transferred;
2524 				    sp->result->stream_retrans = retransmits;
2525 				    if (j_start_time && j_end_time) {
2526 					sp->result->sender_time = j_end_time->valuedouble - j_start_time->valuedouble;
2527 				    }
2528 				    else {
2529 					sp->result->sender_time = 0.0;
2530 				    }
2531 				}
2532 			    }
2533 			}
2534 		    }
2535 		}
2536 		/*
2537 		 * If we're the client and we're supposed to get remote results,
2538 		 * look them up and process accordingly.
2539 		 */
2540 		if (test->role == 'c' && iperf_get_test_get_server_output(test)) {
2541 		    /* Look for JSON.  If we find it, grab the object so it doesn't get deleted. */
2542 		    j_server_output = cJSON_DetachItemFromObject(j, "server_output_json");
2543 		    if (j_server_output != NULL) {
2544 			test->json_server_output = j_server_output;
2545 		    }
2546 		    else {
2547 			/* No JSON, look for textual output.  Make a copy of the text for later. */
2548 			j_server_output = cJSON_GetObjectItem(j, "server_output_text");
2549 			if (j_server_output != NULL) {
2550 			    test->server_output_text = strdup(j_server_output->valuestring);
2551 			}
2552 		    }
2553 		}
2554 	    }
2555 	}
2556 
2557 	j_remote_congestion_used = cJSON_GetObjectItem(j, "congestion_used");
2558 	if (j_remote_congestion_used != NULL) {
2559 	    test->remote_congestion_used = strdup(j_remote_congestion_used->valuestring);
2560 	}
2561 
2562 	cJSON_Delete(j);
2563     }
2564     return r;
2565 }
2566 
2567 /*************************************************************/
2568 
2569 static int
JSON_write(int fd,cJSON * json)2570 JSON_write(int fd, cJSON *json)
2571 {
2572     uint32_t hsize, nsize;
2573     char *str;
2574     int r = 0;
2575 
2576     str = cJSON_PrintUnformatted(json);
2577     if (str == NULL)
2578 	r = -1;
2579     else {
2580 	hsize = strlen(str);
2581 	nsize = htonl(hsize);
2582 	if (Nwrite(fd, (char*) &nsize, sizeof(nsize), Ptcp) < 0)
2583 	    r = -1;
2584 	else {
2585 	    if (Nwrite(fd, str, hsize, Ptcp) < 0)
2586 		r = -1;
2587 	}
2588 	cJSON_free(str);
2589     }
2590     return r;
2591 }
2592 
2593 /*************************************************************/
2594 
2595 static cJSON *
JSON_read(int fd)2596 JSON_read(int fd)
2597 {
2598     uint32_t hsize, nsize;
2599     char *str;
2600     cJSON *json = NULL;
2601     int rc;
2602 
2603     /*
2604      * Read a four-byte integer, which is the length of the JSON to follow.
2605      * Then read the JSON into a buffer and parse it.  Return a parsed JSON
2606      * structure, NULL if there was an error.
2607      */
2608     if (Nread(fd, (char*) &nsize, sizeof(nsize), Ptcp) >= 0) {
2609 	hsize = ntohl(nsize);
2610 	/* Allocate a buffer to hold the JSON */
2611 	str = (char *) calloc(sizeof(char), hsize+1);	/* +1 for trailing null */
2612 	if (str != NULL) {
2613 	    rc = Nread(fd, str, hsize, Ptcp);
2614 	    if (rc >= 0) {
2615 		/*
2616 		 * We should be reading in the number of bytes corresponding to the
2617 		 * length in that 4-byte integer.  If we don't the socket might have
2618 		 * prematurely closed.  Only do the JSON parsing if we got the
2619 		 * correct number of bytes.
2620 		 */
2621 		if (rc == hsize) {
2622 		    json = cJSON_Parse(str);
2623 		}
2624 		else {
2625 		    printf("WARNING:  Size of data read does not correspond to offered length\n");
2626 		}
2627 	    }
2628 	}
2629 	free(str);
2630     }
2631     return json;
2632 }
2633 
2634 /*************************************************************/
2635 /**
2636  * add_to_interval_list -- adds new interval to the interval_list
2637  */
2638 
2639 void
add_to_interval_list(struct iperf_stream_result * rp,struct iperf_interval_results * new)2640 add_to_interval_list(struct iperf_stream_result * rp, struct iperf_interval_results * new)
2641 {
2642     struct iperf_interval_results *irp;
2643 
2644     irp = (struct iperf_interval_results *) malloc(sizeof(struct iperf_interval_results));
2645     memcpy(irp, new, sizeof(struct iperf_interval_results));
2646     TAILQ_INSERT_TAIL(&rp->interval_results, irp, irlistentries);
2647 }
2648 
2649 
2650 /************************************************************/
2651 
2652 /**
2653  * connect_msg -- displays connection message
2654  * denoting sender/receiver details
2655  *
2656  */
2657 
2658 void
connect_msg(struct iperf_stream * sp)2659 connect_msg(struct iperf_stream *sp)
2660 {
2661     char ipl[INET6_ADDRSTRLEN], ipr[INET6_ADDRSTRLEN];
2662     int lport, rport;
2663 
2664     if (getsockdomain(sp->socket) == AF_INET) {
2665         inet_ntop(AF_INET, (void *) &((struct sockaddr_in *) &sp->local_addr)->sin_addr, ipl, sizeof(ipl));
2666 	mapped_v4_to_regular_v4(ipl);
2667         inet_ntop(AF_INET, (void *) &((struct sockaddr_in *) &sp->remote_addr)->sin_addr, ipr, sizeof(ipr));
2668 	mapped_v4_to_regular_v4(ipr);
2669         lport = ntohs(((struct sockaddr_in *) &sp->local_addr)->sin_port);
2670         rport = ntohs(((struct sockaddr_in *) &sp->remote_addr)->sin_port);
2671     } else {
2672         inet_ntop(AF_INET6, (void *) &((struct sockaddr_in6 *) &sp->local_addr)->sin6_addr, ipl, sizeof(ipl));
2673 	mapped_v4_to_regular_v4(ipl);
2674         inet_ntop(AF_INET6, (void *) &((struct sockaddr_in6 *) &sp->remote_addr)->sin6_addr, ipr, sizeof(ipr));
2675 	mapped_v4_to_regular_v4(ipr);
2676         lport = ntohs(((struct sockaddr_in6 *) &sp->local_addr)->sin6_port);
2677         rport = ntohs(((struct sockaddr_in6 *) &sp->remote_addr)->sin6_port);
2678     }
2679 
2680     if (sp->test->json_output)
2681         cJSON_AddItemToArray(sp->test->json_connected, iperf_json_printf("socket: %d  local_host: %s  local_port: %d  remote_host: %s  remote_port: %d", (int64_t) sp->socket, ipl, (int64_t) lport, ipr, (int64_t) rport));
2682     else
2683 	iperf_printf(sp->test, report_connected, sp->socket, ipl, lport, ipr, rport);
2684 }
2685 
2686 
2687 /**************************************************************************/
2688 
2689 struct iperf_test *
iperf_new_test()2690 iperf_new_test()
2691 {
2692     struct iperf_test *test;
2693 
2694     test = (struct iperf_test *) malloc(sizeof(struct iperf_test));
2695     if (!test) {
2696         i_errno = IENEWTEST;
2697         return NULL;
2698     }
2699     /* initialize everything to zero */
2700     memset(test, 0, sizeof(struct iperf_test));
2701 
2702     test->settings = (struct iperf_settings *) malloc(sizeof(struct iperf_settings));
2703     if (!test->settings) {
2704         free(test);
2705 	i_errno = IENEWTEST;
2706 	return NULL;
2707     }
2708     memset(test->settings, 0, sizeof(struct iperf_settings));
2709 
2710     test->bitrate_limit_intervals_traffic_bytes = (iperf_size_t *) malloc(sizeof(iperf_size_t) * MAX_INTERVAL);
2711     if (!test->bitrate_limit_intervals_traffic_bytes) {
2712         free(test->settings);
2713         free(test);
2714 	i_errno = IENEWTEST;
2715 	return NULL;
2716     }
2717     memset(test->bitrate_limit_intervals_traffic_bytes, 0, sizeof(sizeof(iperf_size_t) * MAX_INTERVAL));
2718 
2719     /* By default all output goes to stdout */
2720     test->outfile = stdout;
2721 
2722     return test;
2723 }
2724 
2725 /**************************************************************************/
2726 
2727 struct protocol *
protocol_new(void)2728 protocol_new(void)
2729 {
2730     struct protocol *proto;
2731 
2732     proto = malloc(sizeof(struct protocol));
2733     if(!proto) {
2734         return NULL;
2735     }
2736     memset(proto, 0, sizeof(struct protocol));
2737 
2738     return proto;
2739 }
2740 
2741 void
protocol_free(struct protocol * proto)2742 protocol_free(struct protocol *proto)
2743 {
2744     free(proto);
2745 }
2746 
2747 /**************************************************************************/
2748 int
iperf_defaults(struct iperf_test * testp)2749 iperf_defaults(struct iperf_test *testp)
2750 {
2751     struct protocol *tcp, *udp;
2752 #if defined(HAVE_SCTP_H)
2753     struct protocol *sctp;
2754 #endif /* HAVE_SCTP_H */
2755 
2756     testp->omit = OMIT;
2757     testp->duration = DURATION;
2758     testp->diskfile_name = (char*) 0;
2759     testp->affinity = -1;
2760     testp->server_affinity = -1;
2761     TAILQ_INIT(&testp->xbind_addrs);
2762 #if defined(HAVE_CPUSET_SETAFFINITY)
2763     CPU_ZERO(&testp->cpumask);
2764 #endif /* HAVE_CPUSET_SETAFFINITY */
2765     testp->title = NULL;
2766     testp->extra_data = NULL;
2767     testp->congestion = NULL;
2768     testp->congestion_used = NULL;
2769     testp->remote_congestion_used = NULL;
2770     testp->server_port = PORT;
2771     testp->ctrl_sck = -1;
2772     testp->prot_listener = -1;
2773     testp->other_side_has_retransmits = 0;
2774 
2775     testp->stats_callback = iperf_stats_callback;
2776     testp->reporter_callback = iperf_reporter_callback;
2777 
2778     testp->stats_interval = testp->reporter_interval = 1;
2779     testp->num_streams = 1;
2780 
2781     testp->settings->domain = AF_UNSPEC;
2782     testp->settings->unit_format = 'a';
2783     testp->settings->socket_bufsize = 0;    /* use autotuning */
2784     testp->settings->blksize = DEFAULT_TCP_BLKSIZE;
2785     testp->settings->rate = 0;
2786     testp->settings->bitrate_limit = 0;
2787     testp->settings->bitrate_limit_interval = 5;
2788     testp->settings->bitrate_limit_stats_per_interval = 0;
2789     testp->settings->fqrate = 0;
2790     testp->settings->pacing_timer = DEFAULT_PACING_TIMER;
2791     testp->settings->burst = 0;
2792     testp->settings->mss = 0;
2793     testp->settings->bytes = 0;
2794     testp->settings->blocks = 0;
2795     testp->settings->connect_timeout = -1;
2796     testp->settings->rcv_timeout.secs = DEFAULT_NO_MSG_RCVD_TIMEOUT / SEC_TO_mS;
2797     testp->settings->rcv_timeout.usecs = (DEFAULT_NO_MSG_RCVD_TIMEOUT % SEC_TO_mS) * mS_TO_US;
2798     testp->zerocopy = 0;
2799 
2800     memset(testp->cookie, 0, COOKIE_SIZE);
2801 
2802     testp->multisend = 10;	/* arbitrary */
2803 
2804     /* Set up protocol list */
2805     SLIST_INIT(&testp->streams);
2806     SLIST_INIT(&testp->protocols);
2807 
2808     tcp = protocol_new();
2809     if (!tcp)
2810         return -1;
2811 
2812     tcp->id = Ptcp;
2813     tcp->name = "TCP";
2814     tcp->accept = iperf_tcp_accept;
2815     tcp->listen = iperf_tcp_listen;
2816     tcp->connect = iperf_tcp_connect;
2817     tcp->send = iperf_tcp_send;
2818     tcp->recv = iperf_tcp_recv;
2819     tcp->init = NULL;
2820     SLIST_INSERT_HEAD(&testp->protocols, tcp, protocols);
2821 
2822     udp = protocol_new();
2823     if (!udp) {
2824         protocol_free(tcp);
2825         return -1;
2826     }
2827 
2828     udp->id = Pudp;
2829     udp->name = "UDP";
2830     udp->accept = iperf_udp_accept;
2831     udp->listen = iperf_udp_listen;
2832     udp->connect = iperf_udp_connect;
2833     udp->send = iperf_udp_send;
2834     udp->recv = iperf_udp_recv;
2835     udp->init = iperf_udp_init;
2836     SLIST_INSERT_AFTER(tcp, udp, protocols);
2837 
2838     set_protocol(testp, Ptcp);
2839 
2840 #if defined(HAVE_SCTP_H)
2841     sctp = protocol_new();
2842     if (!sctp) {
2843         protocol_free(tcp);
2844         protocol_free(udp);
2845         return -1;
2846     }
2847 
2848     sctp->id = Psctp;
2849     sctp->name = "SCTP";
2850     sctp->accept = iperf_sctp_accept;
2851     sctp->listen = iperf_sctp_listen;
2852     sctp->connect = iperf_sctp_connect;
2853     sctp->send = iperf_sctp_send;
2854     sctp->recv = iperf_sctp_recv;
2855     sctp->init = iperf_sctp_init;
2856 
2857     SLIST_INSERT_AFTER(udp, sctp, protocols);
2858 #endif /* HAVE_SCTP_H */
2859 
2860     testp->on_new_stream = iperf_on_new_stream;
2861     testp->on_test_start = iperf_on_test_start;
2862     testp->on_connect = iperf_on_connect;
2863     testp->on_test_finish = iperf_on_test_finish;
2864 
2865     TAILQ_INIT(&testp->server_output_list);
2866 
2867     return 0;
2868 }
2869 
2870 
2871 /**************************************************************************/
2872 void
iperf_free_test(struct iperf_test * test)2873 iperf_free_test(struct iperf_test *test)
2874 {
2875     struct protocol *prot;
2876     struct iperf_stream *sp;
2877 
2878     /* Free streams */
2879     while (!SLIST_EMPTY(&test->streams)) {
2880         sp = SLIST_FIRST(&test->streams);
2881         SLIST_REMOVE_HEAD(&test->streams, streams);
2882         iperf_free_stream(sp);
2883     }
2884     if (test->server_hostname)
2885 	free(test->server_hostname);
2886     if (test->tmp_template)
2887 	free(test->tmp_template);
2888     if (test->bind_address)
2889 	free(test->bind_address);
2890     if (test->bind_dev)
2891 	free(test->bind_dev);
2892     if (!TAILQ_EMPTY(&test->xbind_addrs)) {
2893         struct xbind_entry *xbe;
2894 
2895         while (!TAILQ_EMPTY(&test->xbind_addrs)) {
2896             xbe = TAILQ_FIRST(&test->xbind_addrs);
2897             TAILQ_REMOVE(&test->xbind_addrs, xbe, link);
2898             if (xbe->ai)
2899                 freeaddrinfo(xbe->ai);
2900             free(xbe->name);
2901             free(xbe);
2902         }
2903     }
2904 #if defined(HAVE_SSL)
2905 
2906     if (test->server_rsa_private_key)
2907       EVP_PKEY_free(test->server_rsa_private_key);
2908     test->server_rsa_private_key = NULL;
2909 
2910     free(test->settings->authtoken);
2911     test->settings->authtoken = NULL;
2912 
2913     free(test->settings->client_username);
2914     test->settings->client_username = NULL;
2915 
2916     free(test->settings->client_password);
2917     test->settings->client_password = NULL;
2918 
2919     if (test->settings->client_rsa_pubkey)
2920       EVP_PKEY_free(test->settings->client_rsa_pubkey);
2921     test->settings->client_rsa_pubkey = NULL;
2922 #endif /* HAVE_SSL */
2923 
2924     if (test->settings)
2925     free(test->settings);
2926     if (test->title)
2927 	free(test->title);
2928     if (test->extra_data)
2929 	free(test->extra_data);
2930     if (test->congestion)
2931 	free(test->congestion);
2932     if (test->congestion_used)
2933 	free(test->congestion_used);
2934     if (test->remote_congestion_used)
2935 	free(test->remote_congestion_used);
2936     if (test->timestamp_format)
2937 	free(test->timestamp_format);
2938     if (test->omit_timer != NULL)
2939 	tmr_cancel(test->omit_timer);
2940     if (test->timer != NULL)
2941 	tmr_cancel(test->timer);
2942     if (test->stats_timer != NULL)
2943 	tmr_cancel(test->stats_timer);
2944     if (test->reporter_timer != NULL)
2945 	tmr_cancel(test->reporter_timer);
2946 
2947     /* Free protocol list */
2948     while (!SLIST_EMPTY(&test->protocols)) {
2949         prot = SLIST_FIRST(&test->protocols);
2950         SLIST_REMOVE_HEAD(&test->protocols, protocols);
2951         free(prot);
2952     }
2953 
2954     if (test->logfile) {
2955 	free(test->logfile);
2956 	test->logfile = NULL;
2957 	if (test->outfile && test->outfile != stdout) {
2958 	    fclose(test->outfile);
2959 	    test->outfile = NULL;
2960 	}
2961     }
2962 
2963     if (test->server_output_text) {
2964 	free(test->server_output_text);
2965 	test->server_output_text = NULL;
2966     }
2967 
2968     if (test->json_output_string) {
2969 	free(test->json_output_string);
2970 	test->json_output_string = NULL;
2971     }
2972 
2973     /* Free output line buffers, if any (on the server only) */
2974     struct iperf_textline *t;
2975     while (!TAILQ_EMPTY(&test->server_output_list)) {
2976 	t = TAILQ_FIRST(&test->server_output_list);
2977 	TAILQ_REMOVE(&test->server_output_list, t, textlineentries);
2978 	free(t->line);
2979 	free(t);
2980     }
2981 
2982     /* sctp_bindx: do not free the arguments, only the resolver results */
2983     if (!TAILQ_EMPTY(&test->xbind_addrs)) {
2984         struct xbind_entry *xbe;
2985 
2986         TAILQ_FOREACH(xbe, &test->xbind_addrs, link) {
2987             if (xbe->ai) {
2988                 freeaddrinfo(xbe->ai);
2989                 xbe->ai = NULL;
2990             }
2991         }
2992     }
2993 
2994     /* Free interval's traffic array for average rate calculations */
2995     if (test->bitrate_limit_intervals_traffic_bytes != NULL)
2996         free(test->bitrate_limit_intervals_traffic_bytes);
2997 
2998     /* XXX: Why are we setting these values to NULL? */
2999     // test->streams = NULL;
3000     test->stats_callback = NULL;
3001     test->reporter_callback = NULL;
3002     free(test);
3003 }
3004 
3005 
3006 void
iperf_reset_test(struct iperf_test * test)3007 iperf_reset_test(struct iperf_test *test)
3008 {
3009     struct iperf_stream *sp;
3010     int i;
3011 
3012     /* Free streams */
3013     while (!SLIST_EMPTY(&test->streams)) {
3014         sp = SLIST_FIRST(&test->streams);
3015         SLIST_REMOVE_HEAD(&test->streams, streams);
3016         iperf_free_stream(sp);
3017     }
3018     if (test->omit_timer != NULL) {
3019 	tmr_cancel(test->omit_timer);
3020 	test->omit_timer = NULL;
3021     }
3022     if (test->timer != NULL) {
3023 	tmr_cancel(test->timer);
3024 	test->timer = NULL;
3025     }
3026     if (test->stats_timer != NULL) {
3027 	tmr_cancel(test->stats_timer);
3028 	test->stats_timer = NULL;
3029     }
3030     if (test->reporter_timer != NULL) {
3031 	tmr_cancel(test->reporter_timer);
3032 	test->reporter_timer = NULL;
3033     }
3034     test->done = 0;
3035 
3036     SLIST_INIT(&test->streams);
3037 
3038     if (test->remote_congestion_used)
3039         free(test->remote_congestion_used);
3040     test->remote_congestion_used = NULL;
3041     test->role = 's';
3042     test->mode = RECEIVER;
3043     test->sender_has_retransmits = 0;
3044     set_protocol(test, Ptcp);
3045     test->omit = OMIT;
3046     test->duration = DURATION;
3047     test->server_affinity = -1;
3048 #if defined(HAVE_CPUSET_SETAFFINITY)
3049     CPU_ZERO(&test->cpumask);
3050 #endif /* HAVE_CPUSET_SETAFFINITY */
3051     test->state = 0;
3052 
3053     test->ctrl_sck = -1;
3054     test->prot_listener = -1;
3055 
3056     test->bytes_sent = 0;
3057     test->blocks_sent = 0;
3058 
3059     test->bytes_received = 0;
3060     test->blocks_received = 0;
3061 
3062     test->other_side_has_retransmits = 0;
3063 
3064     test->bitrate_limit_stats_count = 0;
3065     test->bitrate_limit_last_interval_index = 0;
3066     test->bitrate_limit_exceeded = 0;
3067 
3068     for (i = 0; i < MAX_INTERVAL; i++)
3069         test->bitrate_limit_intervals_traffic_bytes[i] = 0;
3070 
3071     test->reverse = 0;
3072     test->bidirectional = 0;
3073     test->no_delay = 0;
3074 
3075     FD_ZERO(&test->read_set);
3076     FD_ZERO(&test->write_set);
3077 
3078     test->num_streams = 1;
3079     test->settings->socket_bufsize = 0;
3080     test->settings->blksize = DEFAULT_TCP_BLKSIZE;
3081     test->settings->rate = 0;
3082     test->settings->burst = 0;
3083     test->settings->mss = 0;
3084     test->settings->tos = 0;
3085     test->settings->dont_fragment = 0;
3086     test->zerocopy = 0;
3087 
3088 #if defined(HAVE_SSL)
3089     if (test->settings->authtoken) {
3090         free(test->settings->authtoken);
3091         test->settings->authtoken = NULL;
3092     }
3093     if (test->settings->client_username) {
3094         free(test->settings->client_username);
3095         test->settings->client_username = NULL;
3096     }
3097     if (test->settings->client_password) {
3098         free(test->settings->client_password);
3099         test->settings->client_password = NULL;
3100     }
3101     if (test->settings->client_rsa_pubkey) {
3102         EVP_PKEY_free(test->settings->client_rsa_pubkey);
3103         test->settings->client_rsa_pubkey = NULL;
3104     }
3105 #endif /* HAVE_SSL */
3106 
3107     memset(test->cookie, 0, COOKIE_SIZE);
3108     test->multisend = 10;	/* arbitrary */
3109     test->udp_counters_64bit = 0;
3110     if (test->title) {
3111 	free(test->title);
3112 	test->title = NULL;
3113     }
3114     if (test->extra_data) {
3115 	free(test->extra_data);
3116 	test->extra_data = NULL;
3117     }
3118 
3119     /* Free output line buffers, if any (on the server only) */
3120     struct iperf_textline *t;
3121     while (!TAILQ_EMPTY(&test->server_output_list)) {
3122 	t = TAILQ_FIRST(&test->server_output_list);
3123 	TAILQ_REMOVE(&test->server_output_list, t, textlineentries);
3124 	free(t->line);
3125 	free(t);
3126     }
3127 }
3128 
3129 
3130 /* Reset all of a test's stats back to zero.  Called when the omitting
3131 ** period is over.
3132 */
3133 void
iperf_reset_stats(struct iperf_test * test)3134 iperf_reset_stats(struct iperf_test *test)
3135 {
3136     struct iperf_time now;
3137     struct iperf_stream *sp;
3138     struct iperf_stream_result *rp;
3139 
3140     test->bytes_sent = 0;
3141     test->blocks_sent = 0;
3142     iperf_time_now(&now);
3143     SLIST_FOREACH(sp, &test->streams, streams) {
3144 	sp->omitted_packet_count = sp->packet_count;
3145         sp->omitted_cnt_error = sp->cnt_error;
3146         sp->omitted_outoforder_packets = sp->outoforder_packets;
3147 	sp->jitter = 0;
3148 	rp = sp->result;
3149         rp->bytes_sent_omit = rp->bytes_sent;
3150         rp->bytes_received = 0;
3151         rp->bytes_sent_this_interval = rp->bytes_received_this_interval = 0;
3152 	if (test->sender_has_retransmits == 1) {
3153 	    struct iperf_interval_results ir; /* temporary results structure */
3154 	    save_tcpinfo(sp, &ir);
3155 	    rp->stream_prev_total_retrans = get_total_retransmits(&ir);
3156 	}
3157 	rp->stream_retrans = 0;
3158 	rp->start_time = now;
3159     }
3160 }
3161 
3162 
3163 /**************************************************************************/
3164 
3165 /**
3166  * Gather statistics during a test.
3167  * This function works for both the client and server side.
3168  */
3169 void
iperf_stats_callback(struct iperf_test * test)3170 iperf_stats_callback(struct iperf_test *test)
3171 {
3172     struct iperf_stream *sp;
3173     struct iperf_stream_result *rp = NULL;
3174     struct iperf_interval_results *irp, temp;
3175     struct iperf_time temp_time;
3176     iperf_size_t total_interval_bytes_transferred = 0;
3177 
3178     temp.omitted = test->omitting;
3179     SLIST_FOREACH(sp, &test->streams, streams) {
3180         rp = sp->result;
3181 	temp.bytes_transferred = sp->sender ? rp->bytes_sent_this_interval : rp->bytes_received_this_interval;
3182 
3183         // Total bytes transferred this interval
3184 	total_interval_bytes_transferred += rp->bytes_sent_this_interval + rp->bytes_received_this_interval;
3185 
3186 	irp = TAILQ_LAST(&rp->interval_results, irlisthead);
3187         /* result->end_time contains timestamp of previous interval */
3188         if ( irp != NULL ) /* not the 1st interval */
3189             memcpy(&temp.interval_start_time, &rp->end_time, sizeof(struct iperf_time));
3190         else /* or use timestamp from beginning */
3191             memcpy(&temp.interval_start_time, &rp->start_time, sizeof(struct iperf_time));
3192         /* now save time of end of this interval */
3193         iperf_time_now(&rp->end_time);
3194         memcpy(&temp.interval_end_time, &rp->end_time, sizeof(struct iperf_time));
3195         iperf_time_diff(&temp.interval_start_time, &temp.interval_end_time, &temp_time);
3196         temp.interval_duration = iperf_time_in_secs(&temp_time);
3197 	if (test->protocol->id == Ptcp) {
3198 	    if ( has_tcpinfo()) {
3199 		save_tcpinfo(sp, &temp);
3200 		if (test->sender_has_retransmits == 1) {
3201 		    long total_retrans = get_total_retransmits(&temp);
3202 		    temp.interval_retrans = total_retrans - rp->stream_prev_total_retrans;
3203 		    rp->stream_retrans += temp.interval_retrans;
3204 		    rp->stream_prev_total_retrans = total_retrans;
3205 
3206 		    temp.snd_cwnd = get_snd_cwnd(&temp);
3207 		    if (temp.snd_cwnd > rp->stream_max_snd_cwnd) {
3208 			rp->stream_max_snd_cwnd = temp.snd_cwnd;
3209 		    }
3210 
3211 		    temp.snd_wnd = get_snd_wnd(&temp);
3212 		    if (temp.snd_wnd > rp->stream_max_snd_wnd) {
3213 			rp->stream_max_snd_wnd = temp.snd_wnd;
3214 		    }
3215 
3216 		    temp.rtt = get_rtt(&temp);
3217 		    if (temp.rtt > rp->stream_max_rtt) {
3218 			rp->stream_max_rtt = temp.rtt;
3219 		    }
3220 		    if (rp->stream_min_rtt == 0 ||
3221 			temp.rtt < rp->stream_min_rtt) {
3222 			rp->stream_min_rtt = temp.rtt;
3223 		    }
3224 		    rp->stream_sum_rtt += temp.rtt;
3225 		    rp->stream_count_rtt++;
3226 
3227 		    temp.rttvar = get_rttvar(&temp);
3228 		    temp.pmtu = get_pmtu(&temp);
3229 		}
3230 	    }
3231 	} else {
3232 	    if (irp == NULL) {
3233 		temp.interval_packet_count = sp->packet_count;
3234 		temp.interval_outoforder_packets = sp->outoforder_packets;
3235 		temp.interval_cnt_error = sp->cnt_error;
3236 	    } else {
3237 		temp.interval_packet_count = sp->packet_count - irp->packet_count;
3238 		temp.interval_outoforder_packets = sp->outoforder_packets - irp->outoforder_packets;
3239 		temp.interval_cnt_error = sp->cnt_error - irp->cnt_error;
3240 	    }
3241 	    temp.packet_count = sp->packet_count;
3242 	    temp.jitter = sp->jitter;
3243 	    temp.outoforder_packets = sp->outoforder_packets;
3244 	    temp.cnt_error = sp->cnt_error;
3245 	}
3246         add_to_interval_list(rp, &temp);
3247         rp->bytes_sent_this_interval = rp->bytes_received_this_interval = 0;
3248     }
3249 
3250     /* Verify that total server's throughput is not above specified limit */
3251     if (test->role == 's') {
3252 	iperf_check_total_rate(test, total_interval_bytes_transferred);
3253     }
3254 }
3255 
3256 /**
3257  * Print intermediate results during a test (interval report).
3258  * Uses print_interval_results to print the results for each stream,
3259  * then prints an interval summary for all streams in this
3260  * interval.
3261  */
3262 static void
iperf_print_intermediate(struct iperf_test * test)3263 iperf_print_intermediate(struct iperf_test *test)
3264 {
3265     struct iperf_stream *sp = NULL;
3266     struct iperf_interval_results *irp;
3267     struct iperf_time temp_time;
3268     cJSON *json_interval;
3269     cJSON *json_interval_streams;
3270 
3271     int lower_mode, upper_mode;
3272     int current_mode;
3273 
3274     /*
3275      * Due to timing oddities, there can be cases, especially on the
3276      * server side, where at the end of a test there is a fairly short
3277      * interval with no data transferred.  This could caused by
3278      * the control and data flows sharing the same path in the network,
3279      * and having the control messages for stopping the test being
3280      * queued behind the data packets.
3281      *
3282      * We'd like to try to omit that last interval when it happens, to
3283      * avoid cluttering data and output with useless stuff.
3284      * So we're going to try to ignore very short intervals (less than
3285      * 10% of the interval time) that have no data.
3286      */
3287     int interval_ok = 0;
3288     SLIST_FOREACH(sp, &test->streams, streams) {
3289 	irp = TAILQ_LAST(&sp->result->interval_results, irlisthead);
3290 	if (irp) {
3291 	    iperf_time_diff(&irp->interval_start_time, &irp->interval_end_time, &temp_time);
3292 	    double interval_len = iperf_time_in_secs(&temp_time);
3293 	    if (test->debug) {
3294 		printf("interval_len %f bytes_transferred %" PRIu64 "\n", interval_len, irp->bytes_transferred);
3295 	    }
3296 
3297 	    /*
3298 	     * If the interval is at least 10% the normal interval
3299 	     * length, or if there were actual bytes transferred,
3300 	     * then we want to keep this interval.
3301 	     */
3302 	    if (interval_len >= test->stats_interval * 0.10 ||
3303 		irp->bytes_transferred > 0) {
3304 		interval_ok = 1;
3305 		if (test->debug) {
3306 		    printf("interval forces keep\n");
3307 		}
3308 	    }
3309 	}
3310     }
3311     if (!interval_ok) {
3312 	if (test->debug) {
3313 	    printf("ignoring short interval with no data\n");
3314 	}
3315 	return;
3316     }
3317 
3318     if (test->json_output) {
3319         json_interval = cJSON_CreateObject();
3320 	if (json_interval == NULL)
3321 	    return;
3322 	cJSON_AddItemToArray(test->json_intervals, json_interval);
3323         json_interval_streams = cJSON_CreateArray();
3324 	if (json_interval_streams == NULL)
3325 	    return;
3326 	cJSON_AddItemToObject(json_interval, "streams", json_interval_streams);
3327     } else {
3328         json_interval = NULL;
3329         json_interval_streams = NULL;
3330     }
3331 
3332     /*
3333      * We must to sum streams separately.
3334      * For bidirectional mode we must to display
3335      * information about sender and receiver streams.
3336      * For client side we must handle sender streams
3337      * firstly and receiver streams for server side.
3338      * The following design allows us to do this.
3339      */
3340 
3341     if (test->mode == BIDIRECTIONAL) {
3342         if (test->role == 'c') {
3343             lower_mode = -1;
3344             upper_mode = 0;
3345         } else {
3346             lower_mode = 0;
3347             upper_mode = 1;
3348         }
3349     } else {
3350         lower_mode = test->mode;
3351         upper_mode = lower_mode;
3352     }
3353 
3354 
3355     for (current_mode = lower_mode; current_mode <= upper_mode; ++current_mode) {
3356         char ubuf[UNIT_LEN];
3357         char nbuf[UNIT_LEN];
3358         char mbuf[UNIT_LEN];
3359         char zbuf[] = "          ";
3360 
3361         iperf_size_t bytes = 0;
3362         double bandwidth;
3363         int retransmits = 0;
3364         double start_time, end_time;
3365 
3366         int total_packets = 0, lost_packets = 0;
3367         double avg_jitter = 0.0, lost_percent;
3368         int stream_must_be_sender = current_mode * current_mode;
3369 
3370         char *sum_name;
3371 
3372         /*  Print stream role just for bidirectional mode. */
3373 
3374         if (test->mode == BIDIRECTIONAL) {
3375             sprintf(mbuf, "[%s-%s]", stream_must_be_sender?"TX":"RX", test->role == 'c'?"C":"S");
3376         } else {
3377             mbuf[0] = '\0';
3378             zbuf[0] = '\0';
3379         }
3380 
3381         SLIST_FOREACH(sp, &test->streams, streams) {
3382             if (sp->sender == stream_must_be_sender) {
3383                 print_interval_results(test, sp, json_interval_streams);
3384                 /* sum up all streams */
3385                 irp = TAILQ_LAST(&sp->result->interval_results, irlisthead);
3386                 if (irp == NULL) {
3387                     iperf_err(test,
3388                             "iperf_print_intermediate error: interval_results is NULL");
3389                     return;
3390                 }
3391                 bytes += irp->bytes_transferred;
3392                 if (test->protocol->id == Ptcp) {
3393                     if (test->sender_has_retransmits == 1) {
3394                         retransmits += irp->interval_retrans;
3395                     }
3396                 } else {
3397                     total_packets += irp->interval_packet_count;
3398                     lost_packets += irp->interval_cnt_error;
3399                     avg_jitter += irp->jitter;
3400                 }
3401             }
3402         }
3403 
3404         /* next build string with sum of all streams */
3405         if (test->num_streams > 1 || test->json_output) {
3406             /*
3407              * With BIDIR give a different JSON object name to the one sent/receive sums.
3408              * The different name is given to the data sent from the server, which is
3409              * the "reverse" channel.  This makes sure that the name reported on the server
3410              * and client are compatible, and the names are the same as with non-bidir,
3411              * except for when reverse is used.
3412              */
3413             sum_name = "sum";
3414             if (test->mode == BIDIRECTIONAL) {
3415                 if ((test->role == 'c' && !stream_must_be_sender) ||
3416                     (test->role != 'c' && stream_must_be_sender))
3417                 {
3418                     sum_name = "sum_bidir_reverse";
3419                 }
3420             }
3421 
3422             sp = SLIST_FIRST(&test->streams); /* reset back to 1st stream */
3423             /* Only do this of course if there was a first stream */
3424             if (sp) {
3425 	    irp = TAILQ_LAST(&sp->result->interval_results, irlisthead);    /* use 1st stream for timing info */
3426 
3427 	    unit_snprintf(ubuf, UNIT_LEN, (double) bytes, 'A');
3428 	    bandwidth = (double) bytes / (double) irp->interval_duration;
3429 	    unit_snprintf(nbuf, UNIT_LEN, bandwidth, test->settings->unit_format);
3430 
3431 	    iperf_time_diff(&sp->result->start_time,&irp->interval_start_time, &temp_time);
3432 	    start_time = iperf_time_in_secs(&temp_time);
3433 	    iperf_time_diff(&sp->result->start_time,&irp->interval_end_time, &temp_time);
3434 	    end_time = iperf_time_in_secs(&temp_time);
3435                 if (test->protocol->id == Ptcp || test->protocol->id == Psctp) {
3436                     if (test->sender_has_retransmits == 1 && stream_must_be_sender) {
3437                         /* Interval sum, TCP with retransmits. */
3438                         if (test->json_output)
3439                             cJSON_AddItemToObject(json_interval, sum_name, iperf_json_printf("start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  retransmits: %d  omitted: %b sender: %b", (double) start_time, (double) end_time, (double) irp->interval_duration, (int64_t) bytes, bandwidth * 8, (int64_t) retransmits, irp->omitted, stream_must_be_sender)); /* XXX irp->omitted or test->omitting? */
3440                         else
3441                             iperf_printf(test, report_sum_bw_retrans_format, mbuf, start_time, end_time, ubuf, nbuf, retransmits, irp->omitted?report_omitted:""); /* XXX irp->omitted or test->omitting? */
3442                     } else {
3443                         /* Interval sum, TCP without retransmits. */
3444                         if (test->json_output)
3445                             cJSON_AddItemToObject(json_interval, sum_name, iperf_json_printf("start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  omitted: %b sender: %b", (double) start_time, (double) end_time, (double) irp->interval_duration, (int64_t) bytes, bandwidth * 8, test->omitting, stream_must_be_sender));
3446                         else
3447                             iperf_printf(test, report_sum_bw_format, mbuf, start_time, end_time, ubuf, nbuf, test->omitting?report_omitted:"");
3448                     }
3449                 } else {
3450                     /* Interval sum, UDP. */
3451                     if (stream_must_be_sender) {
3452                         if (test->json_output)
3453                             cJSON_AddItemToObject(json_interval, sum_name, iperf_json_printf("start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  packets: %d  omitted: %b sender: %b", (double) start_time, (double) end_time, (double) irp->interval_duration, (int64_t) bytes, bandwidth * 8, (int64_t) total_packets, test->omitting, stream_must_be_sender));
3454                         else
3455                             iperf_printf(test, report_sum_bw_udp_sender_format, mbuf, start_time, end_time, ubuf, nbuf, zbuf, total_packets, test->omitting?report_omitted:"");
3456                     } else {
3457                         avg_jitter /= test->num_streams;
3458                         if (total_packets > 0) {
3459                             lost_percent = 100.0 * lost_packets / total_packets;
3460                         }
3461                         else {
3462                             lost_percent = 0.0;
3463                         }
3464                         if (test->json_output)
3465                             cJSON_AddItemToObject(json_interval, sum_name, iperf_json_printf("start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  jitter_ms: %f  lost_packets: %d  packets: %d  lost_percent: %f  omitted: %b sender: %b", (double) start_time, (double) end_time, (double) irp->interval_duration, (int64_t) bytes, bandwidth * 8, (double) avg_jitter * 1000.0, (int64_t) lost_packets, (int64_t) total_packets, (double) lost_percent, test->omitting, stream_must_be_sender));
3466                         else
3467                             iperf_printf(test, report_sum_bw_udp_format, mbuf, start_time, end_time, ubuf, nbuf, avg_jitter * 1000.0, lost_packets, total_packets, lost_percent, test->omitting?report_omitted:"");
3468                     }
3469                 }
3470             }
3471         }
3472     }
3473 }
3474 
3475 /**
3476  * Print overall summary statistics at the end of a test.
3477  */
3478 static void
iperf_print_results(struct iperf_test * test)3479 iperf_print_results(struct iperf_test *test)
3480 {
3481 
3482     cJSON *json_summary_streams = NULL;
3483 
3484     int lower_mode, upper_mode;
3485     int current_mode;
3486 
3487     char *sum_sent_name, *sum_received_name, *sum_name;
3488 
3489     int tmp_sender_has_retransmits = test->sender_has_retransmits;
3490 
3491     /* print final summary for all intervals */
3492 
3493     if (test->json_output) {
3494         json_summary_streams = cJSON_CreateArray();
3495 	if (json_summary_streams == NULL)
3496 	    return;
3497 	cJSON_AddItemToObject(test->json_end, "streams", json_summary_streams);
3498     } else {
3499 	iperf_printf(test, "%s", report_bw_separator);
3500 	if (test->verbose)
3501 	    iperf_printf(test, "%s", report_summary);
3502 	if (test->protocol->id == Ptcp || test->protocol->id == Psctp) {
3503 	    if (test->sender_has_retransmits || test->other_side_has_retransmits) {
3504 	        if (test->bidirectional)
3505 	            iperf_printf(test, "%s", report_bw_retrans_header_bidir);
3506 	        else
3507 	            iperf_printf(test, "%s", report_bw_retrans_header);
3508 	    }
3509 	    else {
3510 	        if (test->bidirectional)
3511 	            iperf_printf(test, "%s", report_bw_header_bidir);
3512 	        else
3513 	            iperf_printf(test, "%s", report_bw_header);
3514 	    }
3515 	} else {
3516 	    if (test->bidirectional)
3517 	        iperf_printf(test, "%s", report_bw_udp_header_bidir);
3518 	    else
3519 	        iperf_printf(test, "%s", report_bw_udp_header);
3520 	}
3521     }
3522 
3523     /*
3524      * We must to sum streams separately.
3525      * For bidirectional mode we must to display
3526      * information about sender and receiver streams.
3527      * For client side we must handle sender streams
3528      * firstly and receiver streams for server side.
3529      * The following design allows us to do this.
3530      */
3531 
3532     if (test->mode == BIDIRECTIONAL) {
3533         if (test->role == 'c') {
3534             lower_mode = -1;
3535             upper_mode = 0;
3536         } else {
3537             lower_mode = 0;
3538             upper_mode = 1;
3539         }
3540     } else {
3541         lower_mode = test->mode;
3542         upper_mode = lower_mode;
3543     }
3544 
3545 
3546     for (current_mode = lower_mode; current_mode <= upper_mode; ++current_mode) {
3547         cJSON *json_summary_stream = NULL;
3548         int total_retransmits = 0;
3549         int total_packets = 0, lost_packets = 0;
3550         int sender_packet_count = 0, receiver_packet_count = 0; /* for this stream, this interval */
3551         int sender_total_packets = 0, receiver_total_packets = 0; /* running total */
3552         char ubuf[UNIT_LEN];
3553         char nbuf[UNIT_LEN];
3554         struct stat sb;
3555         char sbuf[UNIT_LEN];
3556         struct iperf_stream *sp = NULL;
3557         iperf_size_t bytes_sent, total_sent = 0;
3558         iperf_size_t bytes_received, total_received = 0;
3559         double start_time, end_time = 0.0, avg_jitter = 0.0, lost_percent = 0.0;
3560         double sender_time = 0.0, receiver_time = 0.0;
3561     struct iperf_time temp_time;
3562         double bandwidth;
3563 
3564         char mbuf[UNIT_LEN];
3565         int stream_must_be_sender = current_mode * current_mode;
3566 
3567 
3568         /*  Print stream role just for bidirectional mode. */
3569 
3570         if (test->mode == BIDIRECTIONAL) {
3571             sprintf(mbuf, "[%s-%s]", stream_must_be_sender?"TX":"RX", test->role == 'c'?"C":"S");
3572         } else {
3573             mbuf[0] = '\0';
3574         }
3575 
3576         /* Get sender_has_retransmits for each sender side (client and server) */
3577         if (test->mode == BIDIRECTIONAL && stream_must_be_sender)
3578             test->sender_has_retransmits = tmp_sender_has_retransmits;
3579         else if (test->mode == BIDIRECTIONAL && !stream_must_be_sender)
3580             test->sender_has_retransmits = test->other_side_has_retransmits;
3581 
3582         start_time = 0.;
3583         sp = SLIST_FIRST(&test->streams);
3584 
3585         /*
3586          * If there is at least one stream, then figure out the length of time
3587          * we were running the tests and print out some statistics about
3588          * the streams.  It's possible to not have any streams at all
3589          * if the client got interrupted before it got to do anything.
3590          *
3591          * Also note that we try to keep separate values for the sender
3592          * and receiver ending times.  Earlier iperf (3.1 and earlier)
3593          * servers didn't send that to the clients, so in this case we fall
3594          * back to using the client's ending timestamp.  The fallback is
3595          * basically emulating what iperf 3.1 did.
3596          */
3597 
3598         if (sp) {
3599     iperf_time_diff(&sp->result->start_time, &sp->result->end_time, &temp_time);
3600     end_time = iperf_time_in_secs(&temp_time);
3601         if (sp->sender) {
3602             sp->result->sender_time = end_time;
3603             if (sp->result->receiver_time == 0.0) {
3604                 sp->result->receiver_time = sp->result->sender_time;
3605             }
3606         }
3607         else {
3608             sp->result->receiver_time = end_time;
3609             if (sp->result->sender_time == 0.0) {
3610                 sp->result->sender_time = sp->result->receiver_time;
3611             }
3612         }
3613         sender_time = sp->result->sender_time;
3614         receiver_time = sp->result->receiver_time;
3615         SLIST_FOREACH(sp, &test->streams, streams) {
3616             if (sp->sender == stream_must_be_sender) {
3617                 if (test->json_output) {
3618                     json_summary_stream = cJSON_CreateObject();
3619                     if (json_summary_stream == NULL)
3620                         return;
3621                     cJSON_AddItemToArray(json_summary_streams, json_summary_stream);
3622                 }
3623 
3624                 bytes_sent = sp->result->bytes_sent - sp->result->bytes_sent_omit;
3625                 bytes_received = sp->result->bytes_received;
3626                 total_sent += bytes_sent;
3627                 total_received += bytes_received;
3628 
3629                 if (sp->sender) {
3630                     sender_packet_count = sp->packet_count;
3631                     receiver_packet_count = sp->peer_packet_count;
3632                 }
3633                 else {
3634                     sender_packet_count = sp->peer_packet_count;
3635                     receiver_packet_count = sp->packet_count;
3636                 }
3637 
3638                 if (test->protocol->id == Ptcp || test->protocol->id == Psctp) {
3639                     if (test->sender_has_retransmits) {
3640                         total_retransmits += sp->result->stream_retrans;
3641                     }
3642                 } else {
3643                     /*
3644                      * Running total of the total number of packets.  Use the sender packet count if we
3645                      * have it, otherwise use the receiver packet count.
3646                      */
3647                     int packet_count = sender_packet_count ? sender_packet_count : receiver_packet_count;
3648                     total_packets += (packet_count - sp->omitted_packet_count);
3649                     sender_total_packets += (sender_packet_count - sp->omitted_packet_count);
3650                     receiver_total_packets += (receiver_packet_count - sp->omitted_packet_count);
3651                     lost_packets += (sp->cnt_error - sp->omitted_cnt_error);
3652                     avg_jitter += sp->jitter;
3653                 }
3654 
3655                 unit_snprintf(ubuf, UNIT_LEN, (double) bytes_sent, 'A');
3656                 if (sender_time > 0.0) {
3657                     bandwidth = (double) bytes_sent / (double) sender_time;
3658                 }
3659                 else {
3660                     bandwidth = 0.0;
3661                 }
3662                 unit_snprintf(nbuf, UNIT_LEN, bandwidth, test->settings->unit_format);
3663                 if (test->protocol->id == Ptcp || test->protocol->id == Psctp) {
3664                     if (test->sender_has_retransmits) {
3665                         /* Sender summary, TCP and SCTP with retransmits. */
3666                         if (test->json_output)
3667                             cJSON_AddItemToObject(json_summary_stream, "sender", iperf_json_printf("socket: %d  start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  retransmits: %d  max_snd_cwnd:  %d  max_snd_wnd:  %d  max_rtt:  %d  min_rtt:  %d  mean_rtt:  %d sender: %b", (int64_t) sp->socket, (double) start_time, (double) sender_time, (double) sender_time, (int64_t) bytes_sent, bandwidth * 8, (int64_t) sp->result->stream_retrans, (int64_t) sp->result->stream_max_snd_cwnd, (int64_t) sp->result->stream_max_snd_wnd, (int64_t) sp->result->stream_max_rtt, (int64_t) sp->result->stream_min_rtt, (int64_t) ((sp->result->stream_count_rtt == 0) ? 0 : sp->result->stream_sum_rtt / sp->result->stream_count_rtt), stream_must_be_sender));
3668                         else
3669                             if (test->role == 's' && !sp->sender) {
3670                                 if (test->verbose)
3671                                     iperf_printf(test, report_sender_not_available_format, sp->socket);
3672                             }
3673                             else {
3674                                 iperf_printf(test, report_bw_retrans_format, sp->socket, mbuf, start_time, sender_time, ubuf, nbuf, sp->result->stream_retrans, report_sender);
3675                             }
3676                     } else {
3677                         /* Sender summary, TCP and SCTP without retransmits. */
3678                         if (test->json_output)
3679                             cJSON_AddItemToObject(json_summary_stream, "sender", iperf_json_printf("socket: %d  start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f sender: %b", (int64_t) sp->socket, (double) start_time, (double) sender_time, (double) sender_time, (int64_t) bytes_sent, bandwidth * 8,  stream_must_be_sender));
3680                         else
3681                             if (test->role == 's' && !sp->sender) {
3682                                 if (test->verbose)
3683                                     iperf_printf(test, report_sender_not_available_format, sp->socket);
3684                             }
3685                             else {
3686                                 iperf_printf(test, report_bw_format, sp->socket, mbuf, start_time, sender_time, ubuf, nbuf, report_sender);
3687                             }
3688                     }
3689                 } else {
3690                     /* Sender summary, UDP. */
3691                     if (sender_packet_count - sp->omitted_packet_count > 0) {
3692                         lost_percent = 100.0 * (sp->cnt_error - sp->omitted_cnt_error) / (sender_packet_count - sp->omitted_packet_count);
3693                     }
3694                     else {
3695                         lost_percent = 0.0;
3696                     }
3697                     if (test->json_output) {
3698                         /*
3699                          * For hysterical raisins, we only emit one JSON
3700                          * object for the UDP summary, and it contains
3701                          * information for both the sender and receiver
3702                          * side.
3703                          *
3704                          * The JSON format as currently defined only includes one
3705                          * value for the number of packets.  We usually want that
3706                          * to be the sender's value (how many packets were sent
3707                          * by the sender).  However this value might not be
3708                          * available on the receiver in certain circumstances
3709                          * specifically on the server side for a normal test or
3710                          * the client side for a reverse-mode test.  If this
3711                          * is the case, then use the receiver's count of packets
3712                          * instead.
3713                          */
3714                         int packet_count = sender_packet_count ? sender_packet_count : receiver_packet_count;
3715                         cJSON_AddItemToObject(json_summary_stream, "udp", iperf_json_printf("socket: %d  start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  jitter_ms: %f  lost_packets: %d  packets: %d  lost_percent: %f  out_of_order: %d sender: %b", (int64_t) sp->socket, (double) start_time, (double) sender_time, (double) sender_time, (int64_t) bytes_sent, bandwidth * 8, (double) sp->jitter * 1000.0, (int64_t) (sp->cnt_error - sp->omitted_cnt_error), (int64_t) (packet_count - sp->omitted_packet_count), (double) lost_percent, (int64_t) (sp->outoforder_packets - sp->omitted_outoforder_packets), stream_must_be_sender));
3716                     }
3717                     else {
3718                         /*
3719                          * Due to ordering of messages on the control channel,
3720                          * the server cannot report on client-side summary
3721                          * statistics.  If we're the server, omit one set of
3722                          * summary statistics to avoid giving meaningless
3723                          * results.
3724                          */
3725                         if (test->role == 's' && !sp->sender) {
3726                             if (test->verbose)
3727                                 iperf_printf(test, report_sender_not_available_format, sp->socket);
3728                         }
3729                         else {
3730                             iperf_printf(test, report_bw_udp_format, sp->socket, mbuf, start_time, sender_time, ubuf, nbuf, 0.0, 0, (sender_packet_count - sp->omitted_packet_count), (double) 0, report_sender);
3731                         }
3732                         if ((sp->outoforder_packets - sp->omitted_outoforder_packets) > 0)
3733                           iperf_printf(test, report_sum_outoforder, mbuf, start_time, sender_time, (sp->outoforder_packets - sp->omitted_outoforder_packets));
3734                     }
3735                 }
3736 
3737                 if (sp->diskfile_fd >= 0) {
3738                     if (fstat(sp->diskfile_fd, &sb) == 0) {
3739                         /* In the odd case that it's a zero-sized file, say it was all transferred. */
3740                         int percent_sent = 100, percent_received = 100;
3741                         if (sb.st_size > 0) {
3742                             percent_sent = (int) ( ( (double) bytes_sent / (double) sb.st_size ) * 100.0 );
3743                             percent_received = (int) ( ( (double) bytes_received / (double) sb.st_size ) * 100.0 );
3744                         }
3745                         unit_snprintf(sbuf, UNIT_LEN, (double) sb.st_size, 'A');
3746                         if (test->json_output)
3747                             cJSON_AddItemToObject(json_summary_stream, "diskfile", iperf_json_printf("sent: %d  received: %d  size: %d  percent_sent: %d  percent_received: %d  filename: %s", (int64_t) bytes_sent, (int64_t) bytes_received, (int64_t) sb.st_size, (int64_t) percent_sent, (int64_t) percent_received, test->diskfile_name));
3748                         else
3749                             if (stream_must_be_sender) {
3750                                 iperf_printf(test, report_diskfile, ubuf, sbuf, percent_sent, test->diskfile_name);
3751                             }
3752                             else {
3753                                 unit_snprintf(ubuf, UNIT_LEN, (double) bytes_received, 'A');
3754                                 iperf_printf(test, report_diskfile, ubuf, sbuf, percent_received, test->diskfile_name);
3755                             }
3756                     }
3757                 }
3758 
3759                 unit_snprintf(ubuf, UNIT_LEN, (double) bytes_received, 'A');
3760                 if (receiver_time > 0) {
3761                     bandwidth = (double) bytes_received / (double) receiver_time;
3762                 }
3763                 else {
3764                     bandwidth = 0.0;
3765                 }
3766                 unit_snprintf(nbuf, UNIT_LEN, bandwidth, test->settings->unit_format);
3767                 if (test->protocol->id == Ptcp || test->protocol->id == Psctp) {
3768                     /* Receiver summary, TCP and SCTP */
3769                     if (test->json_output)
3770                         cJSON_AddItemToObject(json_summary_stream, "receiver", iperf_json_printf("socket: %d  start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f sender: %b", (int64_t) sp->socket, (double) start_time, (double) receiver_time, (double) end_time, (int64_t) bytes_received, bandwidth * 8, stream_must_be_sender));
3771                     else
3772                         if (test->role == 's' && sp->sender) {
3773                             if (test->verbose)
3774                                 iperf_printf(test, report_receiver_not_available_format, sp->socket);
3775                         }
3776                         else {
3777                             iperf_printf(test, report_bw_format, sp->socket, mbuf, start_time, receiver_time, ubuf, nbuf, report_receiver);
3778                         }
3779                 }
3780                 else {
3781                     /*
3782                      * Receiver summary, UDP.  Note that JSON was emitted with
3783                      * the sender summary, so we only deal with human-readable
3784                      * data here.
3785                      */
3786                     if (! test->json_output) {
3787                         if (receiver_packet_count - sp->omitted_packet_count > 0) {
3788                             lost_percent = 100.0 * (sp->cnt_error - sp->omitted_cnt_error) / (receiver_packet_count - sp->omitted_packet_count);
3789                         }
3790                         else {
3791                             lost_percent = 0.0;
3792                         }
3793 
3794                         if (test->role == 's' && sp->sender) {
3795                             if (test->verbose)
3796                                 iperf_printf(test, report_receiver_not_available_format, sp->socket);
3797                         }
3798                         else {
3799                             iperf_printf(test, report_bw_udp_format, sp->socket, mbuf, start_time, receiver_time, ubuf, nbuf, sp->jitter * 1000.0, (sp->cnt_error - sp->omitted_cnt_error), (receiver_packet_count - sp->omitted_packet_count), lost_percent, report_receiver);
3800                         }
3801                     }
3802                 }
3803             }
3804         }
3805         }
3806 
3807         if (test->num_streams > 1 || test->json_output) {
3808             /*
3809              * With BIDIR give a different JSON object name to the one sent/receive sums.
3810              * The different name is given to the data sent from the server, which is
3811              * the "reverse" channel.  This makes sure that the name reported on the server
3812              * and client are compatible, and the names are the same as with non-bidir,
3813              * except for when reverse is used.
3814              */
3815             sum_name = "sum";
3816             sum_sent_name = "sum_sent";
3817             sum_received_name = "sum_received";
3818             if (test->mode == BIDIRECTIONAL) {
3819                 if ((test->role == 'c' && !stream_must_be_sender) ||
3820                     (test->role != 'c' && stream_must_be_sender))
3821                 {
3822                     sum_name = "sum_bidir_reverse";
3823                     sum_sent_name = "sum_sent_bidir_reverse";
3824                     sum_received_name = "sum_received_bidir_reverse";
3825                 }
3826 
3827             }
3828 
3829             unit_snprintf(ubuf, UNIT_LEN, (double) total_sent, 'A');
3830             /* If no tests were run, arbitrarily set bandwidth to 0. */
3831             if (sender_time > 0.0) {
3832                 bandwidth = (double) total_sent / (double) sender_time;
3833             }
3834             else {
3835                 bandwidth = 0.0;
3836             }
3837             unit_snprintf(nbuf, UNIT_LEN, bandwidth, test->settings->unit_format);
3838             if (test->protocol->id == Ptcp || test->protocol->id == Psctp) {
3839                 if (test->sender_has_retransmits) {
3840                     /* Summary sum, TCP with retransmits. */
3841                     if (test->json_output)
3842                         cJSON_AddItemToObject(test->json_end, sum_sent_name, iperf_json_printf("start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  retransmits: %d sender: %b", (double) start_time, (double) sender_time, (double) sender_time, (int64_t) total_sent, bandwidth * 8, (int64_t) total_retransmits, stream_must_be_sender));
3843                     else
3844                         if (test->role == 's' && !stream_must_be_sender) {
3845                             if (test->verbose)
3846                                 iperf_printf(test, report_sender_not_available_summary_format, "SUM");
3847                         }
3848                         else {
3849                           iperf_printf(test, report_sum_bw_retrans_format, mbuf, start_time, sender_time, ubuf, nbuf, total_retransmits, report_sender);
3850                         }
3851                 } else {
3852                     /* Summary sum, TCP without retransmits. */
3853                     if (test->json_output)
3854                         cJSON_AddItemToObject(test->json_end, sum_sent_name, iperf_json_printf("start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f sender: %b", (double) start_time, (double) sender_time, (double) sender_time, (int64_t) total_sent, bandwidth * 8, stream_must_be_sender));
3855                     else
3856                         if (test->role == 's' && !stream_must_be_sender) {
3857                             if (test->verbose)
3858                                 iperf_printf(test, report_sender_not_available_summary_format, "SUM");
3859                         }
3860                         else {
3861                             iperf_printf(test, report_sum_bw_format, mbuf, start_time, sender_time, ubuf, nbuf, report_sender);
3862                         }
3863                 }
3864                 unit_snprintf(ubuf, UNIT_LEN, (double) total_received, 'A');
3865                 /* If no tests were run, set received bandwidth to 0 */
3866                 if (receiver_time > 0.0) {
3867                     bandwidth = (double) total_received / (double) receiver_time;
3868                 }
3869                 else {
3870                     bandwidth = 0.0;
3871                 }
3872                 unit_snprintf(nbuf, UNIT_LEN, bandwidth, test->settings->unit_format);
3873                 if (test->json_output)
3874                     cJSON_AddItemToObject(test->json_end, sum_received_name, iperf_json_printf("start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f sender: %b", (double) start_time, (double) receiver_time, (double) receiver_time, (int64_t) total_received, bandwidth * 8, stream_must_be_sender));
3875                 else
3876                     if (test->role == 's' && stream_must_be_sender) {
3877                         if (test->verbose)
3878                             iperf_printf(test, report_receiver_not_available_summary_format, "SUM");
3879                     }
3880                     else {
3881                         iperf_printf(test, report_sum_bw_format, mbuf, start_time, receiver_time, ubuf, nbuf, report_receiver);
3882                     }
3883             } else {
3884                 /* Summary sum, UDP. */
3885                 avg_jitter /= test->num_streams;
3886                 /* If no packets were sent, arbitrarily set loss percentage to 0. */
3887                 if (total_packets > 0) {
3888                     lost_percent = 100.0 * lost_packets / total_packets;
3889                 }
3890                 else {
3891                     lost_percent = 0.0;
3892                 }
3893                 if (test->json_output) {
3894                     /*
3895                      * Original, summary structure. Using this
3896                      * structure is not recommended due to
3897                      * ambiguities between the sender and receiver.
3898                      */
3899                     cJSON_AddItemToObject(test->json_end, sum_name, iperf_json_printf("start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  jitter_ms: %f  lost_packets: %d  packets: %d  lost_percent: %f sender: %b", (double) start_time, (double) receiver_time, (double) receiver_time, (int64_t) total_sent, bandwidth * 8, (double) avg_jitter * 1000.0, (int64_t) lost_packets, (int64_t) total_packets, (double) lost_percent, stream_must_be_sender));
3900                     /*
3901                      * Separate sum_sent and sum_received structures.
3902                      * Using these structures to get the most complete
3903                      * information about UDP transfer.
3904                      */
3905                     cJSON_AddItemToObject(test->json_end, sum_sent_name, iperf_json_printf("start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  jitter_ms: %f  lost_packets: %d  packets: %d  lost_percent: %f  sender: %b", (double) start_time, (double) sender_time, (double) sender_time, (int64_t) total_sent, (double) total_sent * 8 / sender_time, (double) 0.0, (int64_t) 0, (int64_t) sender_total_packets, (double) 0.0, 1));
3906                     cJSON_AddItemToObject(test->json_end, sum_received_name, iperf_json_printf("start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  jitter_ms: %f  lost_packets: %d  packets: %d  lost_percent: %f  sender: %b", (double) start_time, (double) receiver_time, (double) receiver_time, (int64_t) total_received, (double) total_received * 8 / receiver_time, (double) avg_jitter * 1000.0, (int64_t) lost_packets, (int64_t) receiver_total_packets, (double) lost_percent, 0));
3907                 } else {
3908                     /*
3909                      * On the client we have both sender and receiver overall summary
3910                      * stats.  On the server we have only the side that was on the
3911                      * server.  Output whatever we have.
3912                      */
3913                     if (! (test->role == 's' && !stream_must_be_sender) ) {
3914                         unit_snprintf(ubuf, UNIT_LEN, (double) total_sent, 'A');
3915                         iperf_printf(test, report_sum_bw_udp_format, mbuf, start_time, sender_time, ubuf, nbuf, 0.0, 0, sender_total_packets, 0.0, "sender");
3916                     }
3917                     if (! (test->role == 's' && stream_must_be_sender) ) {
3918 
3919                         unit_snprintf(ubuf, UNIT_LEN, (double) total_received, 'A');
3920                         /* Compute received bandwidth. */
3921                         if (end_time > 0.0) {
3922                             bandwidth = (double) total_received / (double) receiver_time;
3923                         }
3924                         else {
3925                             bandwidth = 0.0;
3926                         }
3927                         unit_snprintf(nbuf, UNIT_LEN, bandwidth, test->settings->unit_format);
3928                         iperf_printf(test, report_sum_bw_udp_format, mbuf, start_time, receiver_time, ubuf, nbuf, avg_jitter * 1000.0, lost_packets, receiver_total_packets, lost_percent, "receiver");
3929                     }
3930                 }
3931             }
3932         }
3933 
3934         if (test->json_output && current_mode == upper_mode) {
3935             cJSON_AddItemToObject(test->json_end, "cpu_utilization_percent", iperf_json_printf("host_total: %f  host_user: %f  host_system: %f  remote_total: %f  remote_user: %f  remote_system: %f", (double) test->cpu_util[0], (double) test->cpu_util[1], (double) test->cpu_util[2], (double) test->remote_cpu_util[0], (double) test->remote_cpu_util[1], (double) test->remote_cpu_util[2]));
3936             if (test->protocol->id == Ptcp) {
3937                 char *snd_congestion = NULL, *rcv_congestion = NULL;
3938                 if (stream_must_be_sender) {
3939                     snd_congestion = test->congestion_used;
3940                     rcv_congestion = test->remote_congestion_used;
3941                 }
3942                 else {
3943                     snd_congestion = test->remote_congestion_used;
3944                     rcv_congestion = test->congestion_used;
3945                 }
3946                 if (snd_congestion) {
3947                     cJSON_AddStringToObject(test->json_end, "sender_tcp_congestion", snd_congestion);
3948                 }
3949                 if (rcv_congestion) {
3950                     cJSON_AddStringToObject(test->json_end, "receiver_tcp_congestion", rcv_congestion);
3951                 }
3952             }
3953         }
3954         else {
3955             if (test->verbose) {
3956                 if (stream_must_be_sender) {
3957                     if (test->bidirectional) {
3958                         iperf_printf(test, report_cpu, report_local, stream_must_be_sender?report_sender:report_receiver, test->cpu_util[0], test->cpu_util[1], test->cpu_util[2], report_remote, stream_must_be_sender?report_receiver:report_sender, test->remote_cpu_util[0], test->remote_cpu_util[1], test->remote_cpu_util[2]);
3959                         iperf_printf(test, report_cpu, report_local, !stream_must_be_sender?report_sender:report_receiver, test->cpu_util[0], test->cpu_util[1], test->cpu_util[2], report_remote, !stream_must_be_sender?report_receiver:report_sender, test->remote_cpu_util[0], test->remote_cpu_util[1], test->remote_cpu_util[2]);
3960                     } else
3961                         iperf_printf(test, report_cpu, report_local, stream_must_be_sender?report_sender:report_receiver, test->cpu_util[0], test->cpu_util[1], test->cpu_util[2], report_remote, stream_must_be_sender?report_receiver:report_sender, test->remote_cpu_util[0], test->remote_cpu_util[1], test->remote_cpu_util[2]);
3962                 }
3963                 if (test->protocol->id == Ptcp) {
3964                     char *snd_congestion = NULL, *rcv_congestion = NULL;
3965                     if (stream_must_be_sender) {
3966                         snd_congestion = test->congestion_used;
3967                         rcv_congestion = test->remote_congestion_used;
3968                     }
3969                     else {
3970                         snd_congestion = test->remote_congestion_used;
3971                         rcv_congestion = test->congestion_used;
3972                     }
3973                     if (snd_congestion) {
3974                         iperf_printf(test, "snd_tcp_congestion %s\n", snd_congestion);
3975                     }
3976                     if (rcv_congestion) {
3977                         iperf_printf(test, "rcv_tcp_congestion %s\n", rcv_congestion);
3978                     }
3979                 }
3980             }
3981 
3982             /* Print server output if we're on the client and it was requested/provided */
3983             if (test->role == 'c' && iperf_get_test_get_server_output(test) && !test->json_output) {
3984                 if (test->json_server_output) {
3985 		    char *str = cJSON_Print(test->json_server_output);
3986                     iperf_printf(test, "\nServer JSON output:\n%s\n", str);
3987 		    cJSON_free(str);
3988                     cJSON_Delete(test->json_server_output);
3989                     test->json_server_output = NULL;
3990                 }
3991                 if (test->server_output_text) {
3992                     iperf_printf(test, "\nServer output:\n%s\n", test->server_output_text);
3993                     test->server_output_text = NULL;
3994                 }
3995             }
3996         }
3997     }
3998 
3999     /* Set real sender_has_retransmits for current side */
4000     if (test->mode == BIDIRECTIONAL)
4001         test->sender_has_retransmits = tmp_sender_has_retransmits;
4002 }
4003 
4004 /**************************************************************************/
4005 
4006 /**
4007  * Main report-printing callback.
4008  * Prints results either during a test (interval report only) or
4009  * after the entire test has been run (last interval report plus
4010  * overall summary).
4011  */
4012 void
iperf_reporter_callback(struct iperf_test * test)4013 iperf_reporter_callback(struct iperf_test *test)
4014 {
4015     switch (test->state) {
4016         case TEST_RUNNING:
4017         case STREAM_RUNNING:
4018             /* print interval results for each stream */
4019             iperf_print_intermediate(test);
4020             break;
4021         case TEST_END:
4022         case DISPLAY_RESULTS:
4023             iperf_print_intermediate(test);
4024             iperf_print_results(test);
4025             break;
4026     }
4027 
4028 }
4029 
4030 /**
4031  * Print the interval results for one stream.
4032  * This function needs to know about the overall test so it can determine the
4033  * context for printing headers, separators, etc.
4034  */
4035 static void
print_interval_results(struct iperf_test * test,struct iperf_stream * sp,cJSON * json_interval_streams)4036 print_interval_results(struct iperf_test *test, struct iperf_stream *sp, cJSON *json_interval_streams)
4037 {
4038     char ubuf[UNIT_LEN];
4039     char nbuf[UNIT_LEN];
4040     char cbuf[UNIT_LEN];
4041     char mbuf[UNIT_LEN];
4042     char zbuf[] = "          ";
4043     double st = 0., et = 0.;
4044     struct iperf_time temp_time;
4045     struct iperf_interval_results *irp = NULL;
4046     double bandwidth, lost_percent;
4047 
4048     if (test->mode == BIDIRECTIONAL) {
4049         sprintf(mbuf, "[%s-%s]", sp->sender?"TX":"RX", test->role == 'c'?"C":"S");
4050     } else {
4051         mbuf[0] = '\0';
4052         zbuf[0] = '\0';
4053     }
4054 
4055     irp = TAILQ_LAST(&sp->result->interval_results, irlisthead); /* get last entry in linked list */
4056     if (irp == NULL) {
4057 	iperf_err(test, "print_interval_results error: interval_results is NULL");
4058         return;
4059     }
4060     if (!test->json_output) {
4061 	/* First stream? */
4062 	if (sp == SLIST_FIRST(&test->streams)) {
4063 	    /* It it's the first interval, print the header;
4064 	    ** else if there's more than one stream, print the separator;
4065 	    ** else nothing.
4066 	    */
4067 	    if (iperf_time_compare(&sp->result->start_time, &irp->interval_start_time) == 0) {
4068 		if (test->protocol->id == Ptcp || test->protocol->id == Psctp) {
4069 		    if (test->sender_has_retransmits == 1) {
4070 		        if (test->bidirectional)
4071 		            iperf_printf(test, "%s", report_bw_retrans_cwnd_header_bidir);
4072 		        else
4073 		            iperf_printf(test, "%s", report_bw_retrans_cwnd_header);
4074 		    }
4075 		    else {
4076 	                if (test->bidirectional)
4077 	                    iperf_printf(test, "%s", report_bw_header_bidir);
4078 	                else
4079 	                    iperf_printf(test, "%s", report_bw_header);
4080 	            }
4081 		} else {
4082 		    if (test->mode == SENDER) {
4083 		        iperf_printf(test, "%s", report_bw_udp_sender_header);
4084 		    } else if (test->mode == RECEIVER){
4085 		        iperf_printf(test, "%s", report_bw_udp_header);
4086 		    } else {
4087 		        /* BIDIRECTIONAL */
4088 		        iperf_printf(test, "%s", report_bw_udp_header_bidir);
4089 		    }
4090 		}
4091 	    } else if (test->num_streams > 1)
4092 		iperf_printf(test, "%s", report_bw_separator);
4093 	}
4094     }
4095 
4096     unit_snprintf(ubuf, UNIT_LEN, (double) (irp->bytes_transferred), 'A');
4097     if (irp->interval_duration > 0.0) {
4098 	bandwidth = (double) irp->bytes_transferred / (double) irp->interval_duration;
4099     }
4100     else {
4101 	bandwidth = 0.0;
4102     }
4103     unit_snprintf(nbuf, UNIT_LEN, bandwidth, test->settings->unit_format);
4104 
4105     iperf_time_diff(&sp->result->start_time, &irp->interval_start_time, &temp_time);
4106     st = iperf_time_in_secs(&temp_time);
4107     iperf_time_diff(&sp->result->start_time, &irp->interval_end_time, &temp_time);
4108     et = iperf_time_in_secs(&temp_time);
4109 
4110     if (test->protocol->id == Ptcp || test->protocol->id == Psctp) {
4111 	if (test->sender_has_retransmits == 1 && sp->sender) {
4112 	    /* Interval, TCP with retransmits. */
4113 	    if (test->json_output)
4114 		cJSON_AddItemToArray(json_interval_streams, iperf_json_printf("socket: %d  start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  retransmits: %d  snd_cwnd:  %d  snd_wnd:  %d  rtt:  %d  rttvar: %d  pmtu: %d  omitted: %b sender: %b", (int64_t) sp->socket, (double) st, (double) et, (double) irp->interval_duration, (int64_t) irp->bytes_transferred, bandwidth * 8, (int64_t) irp->interval_retrans, (int64_t) irp->snd_cwnd, (int64_t) irp->snd_wnd, (int64_t) irp->rtt, (int64_t) irp->rttvar, (int64_t) irp->pmtu, irp->omitted, sp->sender));
4115 	    else {
4116 		unit_snprintf(cbuf, UNIT_LEN, irp->snd_cwnd, 'A');
4117 		iperf_printf(test, report_bw_retrans_cwnd_format, sp->socket, mbuf, st, et, ubuf, nbuf, irp->interval_retrans, cbuf, irp->omitted?report_omitted:"");
4118 	    }
4119 	} else {
4120 	    /* Interval, TCP without retransmits. */
4121 	    if (test->json_output)
4122 		cJSON_AddItemToArray(json_interval_streams, iperf_json_printf("socket: %d  start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  omitted: %b sender: %b", (int64_t) sp->socket, (double) st, (double) et, (double) irp->interval_duration, (int64_t) irp->bytes_transferred, bandwidth * 8, irp->omitted, sp->sender));
4123 	    else
4124 		iperf_printf(test, report_bw_format, sp->socket, mbuf, st, et, ubuf, nbuf, irp->omitted?report_omitted:"");
4125 	}
4126     } else {
4127 	/* Interval, UDP. */
4128 	if (sp->sender) {
4129 	    if (test->json_output)
4130 		cJSON_AddItemToArray(json_interval_streams, iperf_json_printf("socket: %d  start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  packets: %d  omitted: %b sender: %b", (int64_t) sp->socket, (double) st, (double) et, (double) irp->interval_duration, (int64_t) irp->bytes_transferred, bandwidth * 8, (int64_t) irp->interval_packet_count, irp->omitted, sp->sender));
4131 	    else
4132 		iperf_printf(test, report_bw_udp_sender_format, sp->socket, mbuf, st, et, ubuf, nbuf, zbuf, irp->interval_packet_count, irp->omitted?report_omitted:"");
4133 	} else {
4134 	    if (irp->interval_packet_count > 0) {
4135 		lost_percent = 100.0 * irp->interval_cnt_error / irp->interval_packet_count;
4136 	    }
4137 	    else {
4138 		lost_percent = 0.0;
4139 	    }
4140 	    if (test->json_output)
4141 		cJSON_AddItemToArray(json_interval_streams, iperf_json_printf("socket: %d  start: %f  end: %f  seconds: %f  bytes: %d  bits_per_second: %f  jitter_ms: %f  lost_packets: %d  packets: %d  lost_percent: %f  omitted: %b sender: %b", (int64_t) sp->socket, (double) st, (double) et, (double) irp->interval_duration, (int64_t) irp->bytes_transferred, bandwidth * 8, (double) irp->jitter * 1000.0, (int64_t) irp->interval_cnt_error, (int64_t) irp->interval_packet_count, (double) lost_percent, irp->omitted, sp->sender));
4142 	    else
4143 		iperf_printf(test, report_bw_udp_format, sp->socket, mbuf, st, et, ubuf, nbuf, irp->jitter * 1000.0, irp->interval_cnt_error, irp->interval_packet_count, lost_percent, irp->omitted?report_omitted:"");
4144 	}
4145     }
4146 
4147     if (test->logfile || test->forceflush)
4148         iflush(test);
4149 }
4150 
4151 /**************************************************************************/
4152 void
iperf_free_stream(struct iperf_stream * sp)4153 iperf_free_stream(struct iperf_stream *sp)
4154 {
4155     struct iperf_interval_results *irp, *nirp;
4156 
4157     /* XXX: need to free interval list too! */
4158     munmap(sp->buffer, sp->test->settings->blksize);
4159     close(sp->buffer_fd);
4160     if (sp->diskfile_fd >= 0)
4161 	close(sp->diskfile_fd);
4162     for (irp = TAILQ_FIRST(&sp->result->interval_results); irp != NULL; irp = nirp) {
4163         nirp = TAILQ_NEXT(irp, irlistentries);
4164         free(irp);
4165     }
4166     free(sp->result);
4167     if (sp->send_timer != NULL)
4168 	tmr_cancel(sp->send_timer);
4169     free(sp);
4170 }
4171 
4172 /**************************************************************************/
4173 struct iperf_stream *
iperf_new_stream(struct iperf_test * test,int s,int sender)4174 iperf_new_stream(struct iperf_test *test, int s, int sender)
4175 {
4176     struct iperf_stream *sp;
4177     int ret = 0;
4178 
4179     char template[1024];
4180     if (test->tmp_template) {
4181         snprintf(template, sizeof(template) / sizeof(char), "%s", test->tmp_template);
4182     } else {
4183         //find the system temporary dir *unix, windows, cygwin support
4184         char* tempdir = getenv("TMPDIR");
4185         if (tempdir == 0){
4186             tempdir = getenv("TEMP");
4187         }
4188         if (tempdir == 0){
4189             tempdir = getenv("TMP");
4190         }
4191         if (tempdir == 0){
4192             tempdir = "/tmp";
4193         }
4194         snprintf(template, sizeof(template) / sizeof(char), "%s/iperf3.XXXXXX", tempdir);
4195     }
4196 
4197     sp = (struct iperf_stream *) malloc(sizeof(struct iperf_stream));
4198     if (!sp) {
4199         i_errno = IECREATESTREAM;
4200         return NULL;
4201     }
4202 
4203     memset(sp, 0, sizeof(struct iperf_stream));
4204 
4205     sp->sender = sender;
4206     sp->test = test;
4207     sp->settings = test->settings;
4208     sp->result = (struct iperf_stream_result *) malloc(sizeof(struct iperf_stream_result));
4209     if (!sp->result) {
4210         free(sp);
4211         i_errno = IECREATESTREAM;
4212         return NULL;
4213     }
4214 
4215     memset(sp->result, 0, sizeof(struct iperf_stream_result));
4216     TAILQ_INIT(&sp->result->interval_results);
4217 
4218     /* Create and randomize the buffer */
4219     sp->buffer_fd = mkstemp(template);
4220     if (sp->buffer_fd == -1) {
4221         i_errno = IECREATESTREAM;
4222         free(sp->result);
4223         free(sp);
4224         return NULL;
4225     }
4226     if (unlink(template) < 0) {
4227         i_errno = IECREATESTREAM;
4228         free(sp->result);
4229         free(sp);
4230         return NULL;
4231     }
4232     if (ftruncate(sp->buffer_fd, test->settings->blksize) < 0) {
4233         i_errno = IECREATESTREAM;
4234         free(sp->result);
4235         free(sp);
4236         return NULL;
4237     }
4238     sp->buffer = (char *) mmap(NULL, test->settings->blksize, PROT_READ|PROT_WRITE, MAP_PRIVATE, sp->buffer_fd, 0);
4239     if (sp->buffer == MAP_FAILED) {
4240         i_errno = IECREATESTREAM;
4241         free(sp->result);
4242         free(sp);
4243         return NULL;
4244     }
4245     sp->pending_size = 0;
4246 
4247     /* Set socket */
4248     sp->socket = s;
4249 
4250     sp->snd = test->protocol->send;
4251     sp->rcv = test->protocol->recv;
4252 
4253     if (test->diskfile_name != (char*) 0) {
4254 	sp->diskfile_fd = open(test->diskfile_name, sender ? O_RDONLY : (O_WRONLY|O_CREAT|O_TRUNC), S_IRUSR|S_IWUSR);
4255 	if (sp->diskfile_fd == -1) {
4256 	    i_errno = IEFILE;
4257             munmap(sp->buffer, sp->test->settings->blksize);
4258             free(sp->result);
4259             free(sp);
4260 	    return NULL;
4261 	}
4262         sp->snd2 = sp->snd;
4263 	sp->snd = diskfile_send;
4264 	sp->rcv2 = sp->rcv;
4265 	sp->rcv = diskfile_recv;
4266     } else
4267         sp->diskfile_fd = -1;
4268 
4269     /* Initialize stream */
4270     if (test->repeating_payload)
4271         fill_with_repeating_pattern(sp->buffer, test->settings->blksize);
4272     else
4273         ret = readentropy(sp->buffer, test->settings->blksize);
4274 
4275     if ((ret < 0) || (iperf_init_stream(sp, test) < 0)) {
4276         close(sp->buffer_fd);
4277         munmap(sp->buffer, sp->test->settings->blksize);
4278         free(sp->result);
4279         free(sp);
4280         return NULL;
4281     }
4282     iperf_add_stream(test, sp);
4283 
4284     return sp;
4285 }
4286 
4287 /**************************************************************************/
4288 int
iperf_init_stream(struct iperf_stream * sp,struct iperf_test * test)4289 iperf_init_stream(struct iperf_stream *sp, struct iperf_test *test)
4290 {
4291     socklen_t len;
4292     int opt;
4293 
4294     len = sizeof(struct sockaddr_storage);
4295     if (getsockname(sp->socket, (struct sockaddr *) &sp->local_addr, &len) < 0) {
4296         i_errno = IEINITSTREAM;
4297         return -1;
4298     }
4299     len = sizeof(struct sockaddr_storage);
4300     if (getpeername(sp->socket, (struct sockaddr *) &sp->remote_addr, &len) < 0) {
4301         i_errno = IEINITSTREAM;
4302         return -1;
4303     }
4304 
4305     /* Set IP TOS */
4306     if ((opt = test->settings->tos)) {
4307         if (getsockdomain(sp->socket) == AF_INET6) {
4308 #ifdef IPV6_TCLASS
4309             if (setsockopt(sp->socket, IPPROTO_IPV6, IPV6_TCLASS, &opt, sizeof(opt)) < 0) {
4310                 i_errno = IESETCOS;
4311                 return -1;
4312             }
4313 #else
4314             i_errno = IESETCOS;
4315             return -1;
4316 #endif
4317         } else {
4318             if (setsockopt(sp->socket, IPPROTO_IP, IP_TOS, &opt, sizeof(opt)) < 0) {
4319                 i_errno = IESETTOS;
4320                 return -1;
4321             }
4322         }
4323     }
4324 
4325 #if defined(HAVE_DONT_FRAGMENT)
4326     /* Set Don't Fragment (DF). Only applicable to IPv4/UDP tests. */
4327     if (iperf_get_test_protocol_id(test) == Pudp &&
4328         getsockdomain(sp->socket) == AF_INET &&
4329         iperf_get_dont_fragment(test)) {
4330 
4331         /*
4332          * There are multiple implementations of this feature depending on the OS.
4333          * We need to handle separately Linux, UNIX, and Windows, as well as
4334          * the case that DF isn't supported at all (such as on macOS).
4335          */
4336 #if defined(IP_MTU_DISCOVER) /* Linux version of IP_DONTFRAG */
4337         opt = IP_PMTUDISC_DO;
4338         if (setsockopt(sp->socket, IPPROTO_IP, IP_MTU_DISCOVER, &opt, sizeof(opt)) < 0) {
4339             i_errno = IESETDONTFRAGMENT;
4340             return -1;
4341         }
4342 #else
4343 #if defined(IP_DONTFRAG) /* UNIX does IP_DONTFRAG */
4344         opt = 1;
4345         if (setsockopt(sp->socket, IPPROTO_IP, IP_DONTFRAG, &opt, sizeof(opt)) < 0) {
4346             i_errno = IESETDONTFRAGMENT;
4347             return -1;
4348         }
4349 #else
4350 #if defined(IP_DONTFRAGMENT) /* Windows does IP_DONTFRAGMENT */
4351         opt = 1;
4352         if (setsockopt(sp->socket, IPPROTO_IP, IP_DONTFRAGMENT, &opt, sizeof(opt)) < 0) {
4353             i_errno = IESETDONTFRAGMENT;
4354             return -1;
4355         }
4356 #else
4357 	i_errno = IESETDONTFRAGMENT;
4358 	return -1;
4359 #endif /* IP_DONTFRAGMENT */
4360 #endif /* IP_DONTFRAG */
4361 #endif /* IP_MTU_DISCOVER */
4362     }
4363 #endif /* HAVE_DONT_FRAGMENT */
4364     return 0;
4365 }
4366 
4367 /**************************************************************************/
4368 void
iperf_add_stream(struct iperf_test * test,struct iperf_stream * sp)4369 iperf_add_stream(struct iperf_test *test, struct iperf_stream *sp)
4370 {
4371     int i;
4372     struct iperf_stream *n, *prev;
4373 
4374     if (SLIST_EMPTY(&test->streams)) {
4375         SLIST_INSERT_HEAD(&test->streams, sp, streams);
4376         sp->id = 1;
4377     } else {
4378         // for (n = test->streams, i = 2; n->next; n = n->next, ++i);
4379         // NOTE: this would ideally be set to 1, however this will not
4380         //       be changed since it is not causing a significant problem
4381         //       and changing it would break multi-stream tests between old
4382         //       and new iperf3 versions.
4383         i = 2;
4384         SLIST_FOREACH(n, &test->streams, streams) {
4385             prev = n;
4386             ++i;
4387         }
4388         SLIST_INSERT_AFTER(prev, sp, streams);
4389         sp->id = i;
4390     }
4391 }
4392 
4393 /* This pair of routines gets inserted into the snd/rcv function pointers
4394 ** when there's a -F flag. They handle the file stuff and call the real
4395 ** snd/rcv functions, which have been saved in snd2/rcv2.
4396 **
4397 ** The advantage of doing it this way is that in the much more common
4398 ** case of no -F flag, there is zero extra overhead.
4399 */
4400 
4401 static int
diskfile_send(struct iperf_stream * sp)4402 diskfile_send(struct iperf_stream *sp)
4403 {
4404     int r;
4405     int buffer_left = sp->diskfile_left; // represents total data in buffer to be sent out
4406     static int rtot;
4407 
4408     /* if needed, read enough data from the disk to fill up the buffer */
4409     if (sp->diskfile_left < sp->test->settings->blksize && !sp->test->done) {
4410     	r = read(sp->diskfile_fd, sp->buffer, sp->test->settings->blksize -
4411     		 sp->diskfile_left);
4412         buffer_left += r;
4413     	rtot += r;
4414     	if (sp->test->debug) {
4415     	    printf("read %d bytes from file, %d total\n", r, rtot);
4416     	}
4417 
4418         // If the buffer doesn't contain a full buffer at this point,
4419         // adjust the size of the data to send.
4420         if (buffer_left != sp->test->settings->blksize) {
4421             if (sp->test->debug)
4422                 printf("possible eof\n");
4423             // setting data size to be sent,
4424             // which is less than full block/buffer size
4425             // (to be used by iperf_tcp_send, etc.)
4426             sp->pending_size = buffer_left;
4427         }
4428 
4429         // If there's no work left, we're done.
4430         if (buffer_left == 0) {
4431     	    sp->test->done = 1;
4432     	    if (sp->test->debug)
4433     		  printf("done\n");
4434     	}
4435     }
4436 
4437     // If there's no data left in the file or in the buffer, we're done.
4438     // No more data available to be sent.
4439     // Return without sending data to the network
4440     if( sp->test->done || buffer_left == 0 ){
4441         if (sp->test->debug)
4442               printf("already done\n");
4443         sp->test->done = 1;
4444         return 0;
4445     }
4446 
4447     r = sp->snd2(sp);
4448     if (r < 0) {
4449 	return r;
4450     }
4451     /*
4452      * Compute how much data is in the buffer but didn't get sent.
4453      * If there are bytes that got left behind, slide them to the
4454      * front of the buffer so they can hopefully go out on the next
4455      * pass.
4456      */
4457     sp->diskfile_left = buffer_left - r;
4458     if (sp->diskfile_left && sp->diskfile_left < sp->test->settings->blksize) {
4459 	memcpy(sp->buffer,
4460 	       sp->buffer + (sp->test->settings->blksize - sp->diskfile_left),
4461 	       sp->diskfile_left);
4462 	if (sp->test->debug)
4463 	    printf("Shifting %d bytes by %d\n", sp->diskfile_left, (sp->test->settings->blksize - sp->diskfile_left));
4464     }
4465     return r;
4466 }
4467 
4468 static int
diskfile_recv(struct iperf_stream * sp)4469 diskfile_recv(struct iperf_stream *sp)
4470 {
4471     int r;
4472 
4473     r = sp->rcv2(sp);
4474     if (r > 0) {
4475 	// NOTE: Currently ignoring the return value of writing to disk
4476 	(void) (write(sp->diskfile_fd, sp->buffer, r) + 1);
4477     }
4478     return r;
4479 }
4480 
4481 
4482 void
iperf_catch_sigend(void (* handler)(int))4483 iperf_catch_sigend(void (*handler)(int))
4484 {
4485 #ifdef SIGINT
4486     signal(SIGINT, handler);
4487 #endif
4488 #ifdef SIGTERM
4489     signal(SIGTERM, handler);
4490 #endif
4491 #ifdef SIGHUP
4492     signal(SIGHUP, handler);
4493 #endif
4494 }
4495 
4496 /**
4497  * Called as a result of getting a signal.
4498  * Depending on the current state of the test (and the role of this
4499  * process) compute and report one more set of ending statistics
4500  * before cleaning up and exiting.
4501  */
4502 void
iperf_got_sigend(struct iperf_test * test)4503 iperf_got_sigend(struct iperf_test *test)
4504 {
4505     /*
4506      * If we're the client, or if we're a server and running a test,
4507      * then dump out the accumulated stats so far.
4508      */
4509     if (test->role == 'c' ||
4510       (test->role == 's' && test->state == TEST_RUNNING)) {
4511 
4512 	test->done = 1;
4513 	cpu_util(test->cpu_util);
4514 	test->stats_callback(test);
4515 	test->state = DISPLAY_RESULTS; /* change local state only */
4516 	if (test->on_test_finish)
4517 	    test->on_test_finish(test);
4518 	test->reporter_callback(test);
4519     }
4520 
4521     if (test->ctrl_sck >= 0) {
4522 	test->state = (test->role == 'c') ? CLIENT_TERMINATE : SERVER_TERMINATE;
4523 	(void) Nwrite(test->ctrl_sck, (char*) &test->state, sizeof(signed char), Ptcp);
4524     }
4525     i_errno = (test->role == 'c') ? IECLIENTTERM : IESERVERTERM;
4526     iperf_errexit(test, "interrupt - %s", iperf_strerror(i_errno));
4527 }
4528 
4529 /* Try to write a PID file if requested, return -1 on an error. */
4530 int
iperf_create_pidfile(struct iperf_test * test)4531 iperf_create_pidfile(struct iperf_test *test)
4532 {
4533     if (test->pidfile) {
4534 	int fd;
4535 	char buf[8];
4536 
4537 	/* See if the file already exists and we can read it. */
4538 	fd = open(test->pidfile, O_RDONLY, 0);
4539 	if (fd >= 0) {
4540 	    if (read(fd, buf, sizeof(buf) - 1) >= 0) {
4541 
4542 		/* We read some bytes, see if they correspond to a valid PID */
4543 		pid_t pid;
4544 		pid = atoi(buf);
4545 		if (pid > 0) {
4546 
4547 		    /* See if the process exists. */
4548 		    if (kill(pid, 0) == 0) {
4549 			/*
4550 			 * Make sure not to try to delete existing PID file by
4551 			 * scribbling over the pathname we'd use to refer to it.
4552 			 * Then exit with an error.
4553 			 */
4554 			free(test->pidfile);
4555 			test->pidfile = NULL;
4556 			iperf_errexit(test, "Another instance of iperf3 appears to be running");
4557 		    }
4558 		}
4559 	    }
4560 	}
4561 
4562 	/*
4563 	 * File didn't exist, we couldn't read it, or it didn't correspond to
4564 	 * a running process.  Try to create it.
4565 	 */
4566 	fd = open(test->pidfile, O_WRONLY | O_CREAT | O_TRUNC, S_IRUSR|S_IWUSR);
4567 	if (fd < 0) {
4568 	    return -1;
4569 	}
4570 	snprintf(buf, sizeof(buf), "%d", getpid()); /* no trailing newline */
4571 	if (write(fd, buf, strlen(buf)) < 0) {
4572 	    return -1;
4573 	}
4574 	if (close(fd) < 0) {
4575 	    return -1;
4576 	};
4577     }
4578     return 0;
4579 }
4580 
4581 /* Get rid of a PID file, return -1 on error. */
4582 int
iperf_delete_pidfile(struct iperf_test * test)4583 iperf_delete_pidfile(struct iperf_test *test)
4584 {
4585     if (test->pidfile) {
4586 	if (unlink(test->pidfile) < 0) {
4587 	    return -1;
4588 	}
4589     }
4590     return 0;
4591 }
4592 
4593 int
iperf_json_start(struct iperf_test * test)4594 iperf_json_start(struct iperf_test *test)
4595 {
4596     test->json_top = cJSON_CreateObject();
4597     if (test->json_top == NULL)
4598         return -1;
4599     test->json_start = cJSON_CreateObject();
4600     if (test->json_start == NULL)
4601         return -1;
4602     cJSON_AddItemToObject(test->json_top, "start", test->json_start);
4603     test->json_connected = cJSON_CreateArray();
4604     if (test->json_connected == NULL)
4605         return -1;
4606     cJSON_AddItemToObject(test->json_start, "connected", test->json_connected);
4607     test->json_intervals = cJSON_CreateArray();
4608     if (test->json_intervals == NULL)
4609         return -1;
4610     cJSON_AddItemToObject(test->json_top, "intervals", test->json_intervals);
4611     test->json_end = cJSON_CreateObject();
4612     if (test->json_end == NULL)
4613         return -1;
4614     cJSON_AddItemToObject(test->json_top, "end", test->json_end);
4615     return 0;
4616 }
4617 
4618 int
iperf_json_finish(struct iperf_test * test)4619 iperf_json_finish(struct iperf_test *test)
4620 {
4621     if (test->title)
4622 	cJSON_AddStringToObject(test->json_top, "title", test->title);
4623     if (test->extra_data)
4624 	cJSON_AddStringToObject(test->json_top, "extra_data", test->extra_data);
4625     /* Include server output */
4626     if (test->json_server_output) {
4627 	cJSON_AddItemToObject(test->json_top, "server_output_json", test->json_server_output);
4628     }
4629     if (test->server_output_text) {
4630 	cJSON_AddStringToObject(test->json_top, "server_output_text", test->server_output_text);
4631     }
4632     // Get ASCII rendering of JSON structure.  Then make our
4633     // own copy of it and return the storage that cJSON allocated
4634     // on our behalf.  We keep our own copy around.
4635     char *str = cJSON_Print(test->json_top);
4636     if (str == NULL)
4637 	return -1;
4638     test->json_output_string = strdup(str);
4639     cJSON_free(str);
4640     if (test->json_output_string == NULL)
4641         return -1;
4642     fprintf(test->outfile, "%s\n", test->json_output_string);
4643     iflush(test);
4644     cJSON_Delete(test->json_top);
4645     test->json_top = test->json_start = test->json_connected = test->json_intervals = test->json_server_output = test->json_end = NULL;
4646     return 0;
4647 }
4648 
4649 
4650 /* CPU affinity stuff - Linux, FreeBSD, and Windows only. */
4651 
4652 int
iperf_setaffinity(struct iperf_test * test,int affinity)4653 iperf_setaffinity(struct iperf_test *test, int affinity)
4654 {
4655 #if defined(HAVE_SCHED_SETAFFINITY)
4656     cpu_set_t cpu_set;
4657 
4658     CPU_ZERO(&cpu_set);
4659     CPU_SET(affinity, &cpu_set);
4660     if (sched_setaffinity(0, sizeof(cpu_set_t), &cpu_set) != 0) {
4661 	i_errno = IEAFFINITY;
4662         return -1;
4663     }
4664     return 0;
4665 #elif defined(HAVE_CPUSET_SETAFFINITY)
4666     cpuset_t cpumask;
4667 
4668     if(cpuset_getaffinity(CPU_LEVEL_WHICH, CPU_WHICH_PID, -1,
4669                           sizeof(cpuset_t), &test->cpumask) != 0) {
4670         i_errno = IEAFFINITY;
4671         return -1;
4672     }
4673 
4674     CPU_ZERO(&cpumask);
4675     CPU_SET(affinity, &cpumask);
4676 
4677     if(cpuset_setaffinity(CPU_LEVEL_WHICH,CPU_WHICH_PID, -1,
4678                           sizeof(cpuset_t), &cpumask) != 0) {
4679         i_errno = IEAFFINITY;
4680         return -1;
4681     }
4682     return 0;
4683 #elif defined(HAVE_SETPROCESSAFFINITYMASK)
4684 	HANDLE process = GetCurrentProcess();
4685 	DWORD_PTR processAffinityMask = 1 << affinity;
4686 
4687 	if (SetProcessAffinityMask(process, processAffinityMask) == 0) {
4688 		i_errno = IEAFFINITY;
4689 		return -1;
4690 	}
4691 	return 0;
4692 #else /* neither HAVE_SCHED_SETAFFINITY nor HAVE_CPUSET_SETAFFINITY nor HAVE_SETPROCESSAFFINITYMASK */
4693     i_errno = IEAFFINITY;
4694     return -1;
4695 #endif /* neither HAVE_SCHED_SETAFFINITY nor HAVE_CPUSET_SETAFFINITY nor HAVE_SETPROCESSAFFINITYMASK */
4696 }
4697 
4698 int
iperf_clearaffinity(struct iperf_test * test)4699 iperf_clearaffinity(struct iperf_test *test)
4700 {
4701 #if defined(HAVE_SCHED_SETAFFINITY)
4702     cpu_set_t cpu_set;
4703     int i;
4704 
4705     CPU_ZERO(&cpu_set);
4706     for (i = 0; i < CPU_SETSIZE; ++i)
4707 	CPU_SET(i, &cpu_set);
4708     if (sched_setaffinity(0, sizeof(cpu_set_t), &cpu_set) != 0) {
4709 	i_errno = IEAFFINITY;
4710         return -1;
4711     }
4712     return 0;
4713 #elif defined(HAVE_CPUSET_SETAFFINITY)
4714     if(cpuset_setaffinity(CPU_LEVEL_WHICH,CPU_WHICH_PID, -1,
4715                           sizeof(cpuset_t), &test->cpumask) != 0) {
4716         i_errno = IEAFFINITY;
4717         return -1;
4718     }
4719     return 0;
4720 #elif defined(HAVE_SETPROCESSAFFINITYMASK)
4721 	HANDLE process = GetCurrentProcess();
4722 	DWORD_PTR processAffinityMask;
4723 	DWORD_PTR lpSystemAffinityMask;
4724 
4725 	if (GetProcessAffinityMask(process, &processAffinityMask, &lpSystemAffinityMask) == 0
4726 			|| SetProcessAffinityMask(process, lpSystemAffinityMask) == 0) {
4727 		i_errno = IEAFFINITY;
4728 		return -1;
4729 	}
4730 	return 0;
4731 #else /* neither HAVE_SCHED_SETAFFINITY nor HAVE_CPUSET_SETAFFINITY nor HAVE_SETPROCESSAFFINITYMASK */
4732     i_errno = IEAFFINITY;
4733     return -1;
4734 #endif /* neither HAVE_SCHED_SETAFFINITY nor HAVE_CPUSET_SETAFFINITY nor HAVE_SETPROCESSAFFINITYMASK */
4735 }
4736 
4737 static char iperf_timestr[100];
4738 static char linebuffer[1024];
4739 
4740 int
iperf_printf(struct iperf_test * test,const char * format,...)4741 iperf_printf(struct iperf_test *test, const char* format, ...)
4742 {
4743     va_list argp;
4744     int r = 0, r0;
4745     time_t now;
4746     struct tm *ltm = NULL;
4747     char *ct = NULL;
4748 
4749     /* Timestamp if requested */
4750     if (iperf_get_test_timestamps(test)) {
4751 	time(&now);
4752 	ltm = localtime(&now);
4753 	strftime(iperf_timestr, sizeof(iperf_timestr), iperf_get_test_timestamp_format(test), ltm);
4754 	ct = iperf_timestr;
4755     }
4756 
4757     /*
4758      * There are roughly two use cases here.  If we're the client,
4759      * want to print stuff directly to the output stream.
4760      * If we're the sender we might need to buffer up output to send
4761      * to the client.
4762      *
4763      * This doesn't make a whole lot of difference except there are
4764      * some chunks of output on the client (on particular the whole
4765      * of the server output with --get-server-output) that could
4766      * easily exceed the size of the line buffer, but which don't need
4767      * to be buffered up anyway.
4768      */
4769     if (test->role == 'c') {
4770 	if (ct) {
4771             r0 = fprintf(test->outfile, "%s", ct);
4772             if (r0 < 0)
4773                 return r0;
4774             r += r0;
4775 	}
4776 	if (test->title) {
4777 	    r0 = fprintf(test->outfile, "%s:  ", test->title);
4778             if (r0 < 0)
4779                 return r0;
4780             r += r0;
4781         }
4782 	va_start(argp, format);
4783 	r0 = vfprintf(test->outfile, format, argp);
4784 	va_end(argp);
4785         if (r0 < 0)
4786             return r0;
4787         r += r0;
4788     }
4789     else if (test->role == 's') {
4790 	if (ct) {
4791 	    r0 = snprintf(linebuffer, sizeof(linebuffer), "%s", ct);
4792             if (r0 < 0)
4793                 return r0;
4794             r += r0;
4795 	}
4796         /* Should always be true as long as sizeof(ct) < sizeof(linebuffer) */
4797         if (r < sizeof(linebuffer)) {
4798             va_start(argp, format);
4799             r0 = vsnprintf(linebuffer + r, sizeof(linebuffer) - r, format, argp);
4800             va_end(argp);
4801             if (r0 < 0)
4802                 return r0;
4803             r += r0;
4804         }
4805 	fprintf(test->outfile, "%s", linebuffer);
4806 
4807 	if (test->role == 's' && iperf_get_test_get_server_output(test)) {
4808 	    struct iperf_textline *l = (struct iperf_textline *) malloc(sizeof(struct iperf_textline));
4809 	    l->line = strdup(linebuffer);
4810 	    TAILQ_INSERT_TAIL(&(test->server_output_list), l, textlineentries);
4811 	}
4812     }
4813     return r;
4814 }
4815 
4816 int
iflush(struct iperf_test * test)4817 iflush(struct iperf_test *test)
4818 {
4819     return fflush(test->outfile);
4820 }
4821