1 /* 2 * iperf, Copyright (c) 2014, 2015, 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 #ifndef __IPERF_API_H 28 #define __IPERF_API_H 29 30 #include <sys/time.h> 31 #include <setjmp.h> 32 33 struct iperf_test; 34 struct iperf_stream_result; 35 struct iperf_interval_results; 36 struct iperf_stream; 37 38 /* default settings */ 39 #define Ptcp SOCK_STREAM 40 #define Pudp SOCK_DGRAM 41 #define Psctp 12 42 #define DEFAULT_UDP_BLKSIZE 8192 43 #define DEFAULT_TCP_BLKSIZE (128 * 1024) /* default read/write block size */ 44 #define DEFAULT_SCTP_BLKSIZE (64 * 1024) 45 46 /* short option equivalents, used to support options that only have long form */ 47 #define OPT_SCTP 1 48 #define OPT_LOGFILE 2 49 #define OPT_GET_SERVER_OUTPUT 3 50 #define OPT_UDP_COUNTERS_64BIT 4 51 #define OPT_CLIENT_PORT 5 52 #define OPT_NUMSTREAMS 6 53 54 /* states */ 55 #define TEST_START 1 56 #define TEST_RUNNING 2 57 #define RESULT_REQUEST 3 58 #define TEST_END 4 59 #define STREAM_BEGIN 5 60 #define STREAM_RUNNING 6 61 #define STREAM_END 7 62 #define ALL_STREAMS_END 8 63 #define PARAM_EXCHANGE 9 64 #define CREATE_STREAMS 10 65 #define SERVER_TERMINATE 11 66 #define CLIENT_TERMINATE 12 67 #define EXCHANGE_RESULTS 13 68 #define DISPLAY_RESULTS 14 69 #define IPERF_START 15 70 #define IPERF_DONE 16 71 #define ACCESS_DENIED (-1) 72 #define SERVER_ERROR (-2) 73 74 /* Getter routines for some fields inside iperf_test. */ 75 int iperf_get_verbose( struct iperf_test* ipt ); 76 int iperf_get_control_socket( struct iperf_test* ipt ); 77 int iperf_get_test_omit( struct iperf_test* ipt ); 78 int iperf_get_test_duration( struct iperf_test* ipt ); 79 char iperf_get_test_role( struct iperf_test* ipt ); 80 int iperf_get_test_reverse( struct iperf_test* ipt ); 81 int iperf_get_test_blksize( struct iperf_test* ipt ); 82 FILE* iperf_get_test_outfile( struct iperf_test* ipt ); 83 uint64_t iperf_get_test_rate( struct iperf_test* ipt ); 84 int iperf_get_test_burst( struct iperf_test* ipt ); 85 int iperf_get_test_socket_bufsize( struct iperf_test* ipt ); 86 double iperf_get_test_reporter_interval( struct iperf_test* ipt ); 87 double iperf_get_test_stats_interval( struct iperf_test* ipt ); 88 int iperf_get_test_num_streams( struct iperf_test* ipt ); 89 int iperf_get_test_server_port( struct iperf_test* ipt ); 90 char* iperf_get_test_server_hostname( struct iperf_test* ipt ); 91 char* iperf_get_test_template( struct iperf_test* ipt ); 92 int iperf_get_test_protocol_id( struct iperf_test* ipt ); 93 int iperf_get_test_json_output( struct iperf_test* ipt ); 94 char* iperf_get_test_json_output_string ( struct iperf_test* ipt ); 95 int iperf_get_test_zerocopy( struct iperf_test* ipt ); 96 int iperf_get_test_get_server_output( struct iperf_test* ipt ); 97 char* iperf_get_test_bind_address ( struct iperf_test* ipt ); 98 int iperf_get_test_udp_counters_64bit( struct iperf_test* ipt ); 99 int iperf_get_test_one_off( struct iperf_test* ipt ); 100 101 /* Setter routines for some fields inside iperf_test. */ 102 void iperf_set_verbose( struct iperf_test* ipt, int verbose ); 103 void iperf_set_control_socket( struct iperf_test* ipt, int ctrl_sck ); 104 void iperf_set_test_omit( struct iperf_test* ipt, int omit ); 105 void iperf_set_test_duration( struct iperf_test* ipt, int duration ); 106 void iperf_set_test_reporter_interval( struct iperf_test* ipt, double reporter_interval ); 107 void iperf_set_test_stats_interval( struct iperf_test* ipt, double stats_interval ); 108 void iperf_set_test_state( struct iperf_test* ipt, signed char state ); 109 void iperf_set_test_blksize( struct iperf_test* ipt, int blksize ); 110 void iperf_set_test_rate( struct iperf_test* ipt, uint64_t rate ); 111 void iperf_set_test_burst( struct iperf_test* ipt, int burst ); 112 void iperf_set_test_server_port( struct iperf_test* ipt, int server_port ); 113 void iperf_set_test_socket_bufsize( struct iperf_test* ipt, int socket_bufsize ); 114 void iperf_set_test_num_streams( struct iperf_test* ipt, int num_streams ); 115 void iperf_set_test_role( struct iperf_test* ipt, char role ); 116 void iperf_set_test_server_hostname( struct iperf_test* ipt, char* server_hostname ); 117 void iperf_set_test_template( struct iperf_test *ipt, char *tmp_template ); 118 void iperf_set_test_reverse( struct iperf_test* ipt, int reverse ); 119 void iperf_set_test_json_output( struct iperf_test* ipt, int json_output ); 120 int iperf_has_zerocopy( void ); 121 void iperf_set_test_zerocopy( struct iperf_test* ipt, int zerocopy ); 122 void iperf_set_test_get_server_output( struct iperf_test* ipt, int get_server_output ); 123 void iperf_set_test_bind_address( struct iperf_test* ipt, char *bind_address ); 124 void iperf_set_test_udp_counters_64bit( struct iperf_test* ipt, int udp_counters_64bit ); 125 void iperf_set_test_one_off( struct iperf_test* ipt, int one_off ); 126 127 /** 128 * exchange_parameters - handles the param_Exchange part for client 129 * 130 */ 131 int iperf_exchange_parameters(struct iperf_test * test); 132 133 /** 134 * add_to_interval_list -- adds new interval to the interval_list 135 * 136 */ 137 void add_to_interval_list(struct iperf_stream_result * rp, struct iperf_interval_results *temp); 138 139 /** 140 * connect_msg -- displays connection message 141 * denoting senfer/receiver details 142 * 143 */ 144 void connect_msg(struct iperf_stream * sp); 145 146 /** 147 * iperf_stats_callback -- handles the statistic gathering 148 * 149 */ 150 void iperf_stats_callback(struct iperf_test * test); 151 152 /** 153 * iperf_reporter_callback -- handles the report printing 154 * 155 */ 156 void iperf_reporter_callback(struct iperf_test * test); 157 158 /** 159 * iperf_new_test -- return a new iperf_test with default values 160 * 161 * returns NULL on failure 162 * 163 */ 164 struct iperf_test *iperf_new_test(); 165 166 int iperf_defaults(struct iperf_test * testp); 167 168 /** 169 * iperf_free_test -- free resources used by test, calls iperf_free_stream to 170 * free streams 171 * 172 */ 173 void iperf_free_test(struct iperf_test * testp); 174 175 /** 176 * iperf_new_stream -- return a net iperf_stream with default values 177 * 178 * returns NULL on failure 179 * 180 */ 181 struct iperf_stream *iperf_new_stream(struct iperf_test *, int); 182 183 /** 184 * iperf_add_stream -- add a stream to a test 185 * 186 */ 187 void iperf_add_stream(struct iperf_test * test, struct iperf_stream * stream); 188 189 /** 190 * iperf_init_stream -- init resources associated with test 191 * 192 */ 193 int iperf_init_stream(struct iperf_stream *, struct iperf_test *); 194 195 /** 196 * iperf_free_stream -- free resources associated with test 197 * 198 */ 199 void iperf_free_stream(struct iperf_stream * sp); 200 201 int has_tcpinfo(void); 202 int has_tcpinfo_retransmits(void); 203 void save_tcpinfo(struct iperf_stream *sp, struct iperf_interval_results *irp); 204 long get_total_retransmits(struct iperf_interval_results *irp); 205 long get_snd_cwnd(struct iperf_interval_results *irp); 206 long get_rtt(struct iperf_interval_results *irp); 207 void print_tcpinfo(struct iperf_test *test); 208 void build_tcpinfo_message(struct iperf_interval_results *r, char *message); 209 210 int iperf_set_send_state(struct iperf_test *test, signed char state); 211 void iperf_check_throttle(struct iperf_stream *sp, struct timeval *nowP); 212 int iperf_send(struct iperf_test *, fd_set *) /* __attribute__((hot)) */; 213 int iperf_recv(struct iperf_test *, fd_set *); 214 void iperf_catch_sigend(void (*handler)(int)); 215 void iperf_got_sigend(struct iperf_test *test) __attribute__ ((noreturn)); 216 void usage(); 217 void usage_long(); 218 void warning(char *); 219 int iperf_exchange_results(struct iperf_test *); 220 int iperf_init_test(struct iperf_test *); 221 int iperf_create_send_timers(struct iperf_test *); 222 int iperf_parse_arguments(struct iperf_test *, int, char **); 223 void iperf_reset_test(struct iperf_test *); 224 void iperf_reset_stats(struct iperf_test * test); 225 226 struct protocol *get_protocol(struct iperf_test *, int); 227 int set_protocol(struct iperf_test *, int); 228 229 void iperf_on_new_stream(struct iperf_stream *); 230 void iperf_on_test_start(struct iperf_test *); 231 void iperf_on_connect(struct iperf_test *); 232 void iperf_on_test_finish(struct iperf_test *); 233 234 extern jmp_buf env; 235 236 /* Client routines. */ 237 int iperf_run_client(struct iperf_test *); 238 int iperf_connect(struct iperf_test *); 239 int iperf_create_streams(struct iperf_test *); 240 int iperf_handle_message_client(struct iperf_test *); 241 int iperf_client_end(struct iperf_test *); 242 243 /* Server routines. */ 244 int iperf_run_server(struct iperf_test *); 245 int iperf_server_listen(struct iperf_test *); 246 int iperf_accept(struct iperf_test *); 247 int iperf_handle_message_server(struct iperf_test *); 248 void iperf_test_reset(struct iperf_test *); 249 int iperf_create_pidfile(struct iperf_test *); 250 int iperf_delete_pidfile(struct iperf_test *); 251 252 /* JSON output routines. */ 253 int iperf_json_start(struct iperf_test *); 254 int iperf_json_finish(struct iperf_test *); 255 256 /* CPU affinity routines */ 257 int iperf_setaffinity(struct iperf_test *, int affinity); 258 int iperf_clearaffinity(struct iperf_test *); 259 260 /* Custom printf routine. */ 261 int iprintf(struct iperf_test *test, const char *format, ...) __attribute__ ((format(printf,2,3))); 262 int iflush(struct iperf_test *test); 263 264 /* Error routines. */ 265 void iperf_err(struct iperf_test *test, const char *format, ...) __attribute__ ((format(printf,2,3))); 266 void iperf_errexit(struct iperf_test *test, const char *format, ...) __attribute__ ((format(printf,2,3),noreturn)); 267 char *iperf_strerror(int); 268 extern int i_errno; 269 enum { 270 IENONE = 0, // No error 271 /* Parameter errors */ 272 IESERVCLIENT = 1, // Iperf cannot be both server and client 273 IENOROLE = 2, // Iperf must either be a client (-c) or server (-s) 274 IESERVERONLY = 3, // This option is server only 275 IECLIENTONLY = 4, // This option is client only 276 IEDURATION = 5, // test duration too long. Maximum value = %dMAX_TIME 277 IENUMSTREAMS = 6, // Number of parallel streams too large. Maximum value = %dMAX_STREAMS 278 IEBLOCKSIZE = 7, // Block size too large. Maximum value = %dMAX_BLOCKSIZE 279 IEBUFSIZE = 8, // Socket buffer size too large. Maximum value = %dMAX_TCP_BUFFER 280 IEINTERVAL = 9, // Invalid report interval (min = %gMIN_INTERVAL, max = %gMAX_INTERVAL seconds) 281 IEMSS = 10, // MSS too large. Maximum value = %dMAX_MSS 282 IENOSENDFILE = 11, // This OS does not support sendfile 283 IEOMIT = 12, // Bogus value for --omit 284 IEUNIMP = 13, // Not implemented yet 285 IEFILE = 14, // -F file couldn't be opened 286 IEBURST = 15, // Invalid burst count. Maximum value = %dMAX_BURST 287 IEENDCONDITIONS = 16, // Only one test end condition (-t, -n, -k) may be specified 288 IELOGFILE = 17, // Can't open log file 289 IENOSCTP = 18, // No SCTP support available 290 IEBIND = 19, // Local port specified with no local bind option 291 IEUDPBLOCKSIZE = 20, // Block size too large. Maximum value = %dMAX_UDP_BLOCKSIZE 292 IEBADTOS = 21, // Bad TOS value 293 /* Test errors */ 294 IENEWTEST = 100, // Unable to create a new test (check perror) 295 IEINITTEST = 101, // Test initialization failed (check perror) 296 IELISTEN = 102, // Unable to listen for connections (check perror) 297 IECONNECT = 103, // Unable to connect to server (check herror/perror) [from netdial] 298 IEACCEPT = 104, // Unable to accept connection from client (check herror/perror) 299 IESENDCOOKIE = 105, // Unable to send cookie to server (check perror) 300 IERECVCOOKIE = 106, // Unable to receive cookie from client (check perror) 301 IECTRLWRITE = 107, // Unable to write to the control socket (check perror) 302 IECTRLREAD = 108, // Unable to read from the control socket (check perror) 303 IECTRLCLOSE = 109, // Control socket has closed unexpectedly 304 IEMESSAGE = 110, // Received an unknown message 305 IESENDMESSAGE = 111, // Unable to send control message to client/server (check perror) 306 IERECVMESSAGE = 112, // Unable to receive control message from client/server (check perror) 307 IESENDPARAMS = 113, // Unable to send parameters to server (check perror) 308 IERECVPARAMS = 114, // Unable to receive parameters from client (check perror) 309 IEPACKAGERESULTS = 115, // Unable to package results (check perror) 310 IESENDRESULTS = 116, // Unable to send results to client/server (check perror) 311 IERECVRESULTS = 117, // Unable to receive results from client/server (check perror) 312 IESELECT = 118, // Select failed (check perror) 313 IECLIENTTERM = 119, // The client has terminated 314 IESERVERTERM = 120, // The server has terminated 315 IEACCESSDENIED = 121, // The server is busy running a test. Try again later. 316 IESETNODELAY = 122, // Unable to set TCP/SCTP NODELAY (check perror) 317 IESETMSS = 123, // Unable to set TCP/SCTP MSS (check perror) 318 IESETBUF = 124, // Unable to set socket buffer size (check perror) 319 IESETTOS = 125, // Unable to set IP TOS (check perror) 320 IESETCOS = 126, // Unable to set IPv6 traffic class (check perror) 321 IESETFLOW = 127, // Unable to set IPv6 flow label 322 IEREUSEADDR = 128, // Unable to set reuse address on socket (check perror) 323 IENONBLOCKING = 129, // Unable to set socket to non-blocking (check perror) 324 IESETWINDOWSIZE = 130, // Unable to set socket window size (check perror) 325 IEPROTOCOL = 131, // Protocol does not exist 326 IEAFFINITY = 132, // Unable to set CPU affinity (check perror) 327 IEDAEMON = 133, // Unable to become a daemon process 328 IESETCONGESTION = 134, // Unable to set TCP_CONGESTION 329 IEPIDFILE = 135, // Unable to write PID file 330 IEV6ONLY = 136, // Unable to set/unset IPV6_V6ONLY (check perror) 331 IESETSCTPDISABLEFRAG = 137, // Unable to set SCTP Fragmentation (check perror) 332 IESETSCTPNSTREAM= 138, // Unable to set SCTP number of streams (check perror) 333 IESETSCTPBINDX= 139, // Unable to process sctp_bindx() parameters 334 /* Stream errors */ 335 IECREATESTREAM = 200, // Unable to create a new stream (check herror/perror) 336 IEINITSTREAM = 201, // Unable to initialize stream (check herror/perror) 337 IESTREAMLISTEN = 202, // Unable to start stream listener (check perror) 338 IESTREAMCONNECT = 203, // Unable to connect stream (check herror/perror) 339 IESTREAMACCEPT = 204, // Unable to accepte stream connection (check perror) 340 IESTREAMWRITE = 205, // Unable to write to stream socket (check perror) 341 IESTREAMREAD = 206, // Unable to read from stream (check perror) 342 IESTREAMCLOSE = 207, // Stream has closed unexpectedly 343 IESTREAMID = 208, // Stream has invalid ID 344 /* Timer errors */ 345 IENEWTIMER = 300, // Unable to create new timer (check perror) 346 IEUPDATETIMER = 301, // Unable to update timer (check perror) 347 }; 348 349 #endif /* !__IPERF_API_H */ 350