1 /*
2 * iperf, Copyright (c) 2014, 2015, 2017, 2019, 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
25 * file for complete information.
26 */
27 #include "iperf_config.h"
28
29 #include <stdio.h>
30 #include <stdlib.h>
31 #include <string.h>
32 #include <getopt.h>
33 #include <errno.h>
34 #include <signal.h>
35 #include <unistd.h>
36 #ifdef HAVE_STDINT_H
37 #include <stdint.h>
38 #endif
39 #include <sys/socket.h>
40 #include <sys/types.h>
41 #include <netinet/in.h>
42 #include <arpa/inet.h>
43 #include <netdb.h>
44
45 #include "iperf.h"
46 #include "iperf_api.h"
47 #include "iperf_util.h"
48 #include "iperf_locale.h"
49 #include "net.h"
50 #include "units.h"
51
52
53 static int run(struct iperf_test *test);
54
55
56 /**************************************************************************/
57 int
main(int argc,char ** argv)58 main(int argc, char **argv)
59 {
60 struct iperf_test *test;
61
62 // XXX: Setting the process affinity requires root on most systems.
63 // Is this a feature we really need?
64 #ifdef TEST_PROC_AFFINITY
65 /* didnt seem to work.... */
66 /*
67 * increasing the priority of the process to minimise packet generation
68 * delay
69 */
70 int rc = setpriority(PRIO_PROCESS, 0, -15);
71
72 if (rc < 0) {
73 perror("setpriority:");
74 fprintf(stderr, "setting priority to valid level\n");
75 rc = setpriority(PRIO_PROCESS, 0, 0);
76 }
77
78 /* setting the affinity of the process */
79 cpu_set_t cpu_set;
80 int affinity = -1;
81 int ncores = 1;
82
83 sched_getaffinity(0, sizeof(cpu_set_t), &cpu_set);
84 if (errno)
85 perror("couldn't get affinity:");
86
87 if ((ncores = sysconf(_SC_NPROCESSORS_CONF)) <= 0)
88 err("sysconf: couldn't get _SC_NPROCESSORS_CONF");
89
90 CPU_ZERO(&cpu_set);
91 CPU_SET(affinity, &cpu_set);
92 if (sched_setaffinity(0, sizeof(cpu_set_t), &cpu_set) != 0)
93 err("couldn't change CPU affinity");
94 #endif
95
96 test = iperf_new_test();
97 if (!test)
98 iperf_errexit(NULL, "create new test error - %s", iperf_strerror(i_errno));
99 iperf_defaults(test); /* sets defaults */
100
101 if (iperf_parse_arguments(test, argc, argv) < 0) {
102 iperf_err(test, "parameter error - %s", iperf_strerror(i_errno));
103 fprintf(stderr, "\n");
104 usage_long(stdout);
105 exit(1);
106 }
107
108 if (run(test) < 0)
109 iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
110
111 iperf_free_test(test);
112
113 return 0;
114 }
115
116
117 static jmp_buf sigend_jmp_buf;
118
119 static void __attribute__ ((noreturn))
sigend_handler(int sig)120 sigend_handler(int sig)
121 {
122 longjmp(sigend_jmp_buf, 1);
123 }
124
125 /**************************************************************************/
126 static int
run(struct iperf_test * test)127 run(struct iperf_test *test)
128 {
129 /* Termination signals. */
130 iperf_catch_sigend(sigend_handler);
131 if (setjmp(sigend_jmp_buf))
132 iperf_got_sigend(test);
133
134 /* Ignore SIGPIPE to simplify error handling */
135 signal(SIGPIPE, SIG_IGN);
136
137 switch (test->role) {
138 case 's':
139 if (test->daemon) {
140 int rc;
141 rc = daemon(0, 0);
142 if (rc < 0) {
143 i_errno = IEDAEMON;
144 iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
145 }
146 }
147 if (iperf_create_pidfile(test) < 0) {
148 i_errno = IEPIDFILE;
149 iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
150 }
151 for (;;) {
152 int rc;
153 rc = iperf_run_server(test);
154 if (rc < 0) {
155 iperf_err(test, "error - %s", iperf_strerror(i_errno));
156 if (rc < -1) {
157 iperf_errexit(test, "exiting");
158 }
159 }
160 iperf_reset_test(test);
161 if (iperf_get_test_one_off(test)) {
162 /* Authentication failure doesn't count for 1-off test */
163 if (rc < 0 && i_errno == IEAUTHTEST) {
164 continue;
165 }
166 break;
167 }
168 }
169 iperf_delete_pidfile(test);
170 break;
171 case 'c':
172 if (iperf_run_client(test) < 0)
173 iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
174 break;
175 default:
176 usage();
177 break;
178 }
179
180 iperf_catch_sigend(SIG_DFL);
181 signal(SIGPIPE, SIG_DFL);
182
183 return 0;
184 }
185