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