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