int
main(int argc, char **argv)
{
struct iperf_test *test;
// XXX: Setting the process affinity requires root on most systems.
// Is this a feature we really need?
#ifdef TEST_PROC_AFFINITY
/* didnt seem to work.... */
/*
* increasing the priority of the process to minimise packet generation
* delay
*/
int rc = setpriority(PRIO_PROCESS, 0, -15);
if (rc < 0) {
perror("setpriority:");
fprintf(stderr, "setting priority to valid level\n");
rc = setpriority(PRIO_PROCESS, 0, 0);
}
/* setting the affinity of the process */
cpu_set_t cpu_set;
int affinity = -1;
int ncores = 1;
sched_getaffinity(0, sizeof(cpu_set_t), &cpu_set);
if (errno)
perror("couldn't get affinity:");
if ((ncores = sysconf(_SC_NPROCESSORS_CONF)) <= 0)
err("sysconf: couldn't get _SC_NPROCESSORS_CONF");
CPU_ZERO(&cpu_set);
CPU_SET(affinity, &cpu_set);
if (sched_setaffinity(0, sizeof(cpu_set_t), &cpu_set) != 0)
err("couldn't change CPU affinity");
#endif
test = iperf_new_test();
if (!test)
iperf_errexit(NULL, "create new test error - %s", iperf_strerror(i_errno));
iperf_defaults(test); /* sets defaults */
if (iperf_parse_arguments(test, argc, argv) < 0) {
iperf_err(test, "parameter error - %s", iperf_strerror(i_errno));
fprintf(stderr, "\n");
usage_long();
exit(1);
}
if (run(test) < 0)
iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
iperf_free_test(test);
return 0;
}
static int
run(struct iperf_test *test)
{
int consecutive_errors;
/* Termination signals. */
iperf_catch_sigend(sigend_handler);
if (setjmp(sigend_jmp_buf))
iperf_got_sigend(test);
switch (test->role) {
case 's':
if (test->daemon) {
int rc = daemon(0, 0);
if (rc < 0) {
i_errno = IEDAEMON;
iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
}
}
consecutive_errors = 0;
if (iperf_create_pidfile(test) < 0) {
i_errno = IEPIDFILE;
iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
}
for (;;) {
if (iperf_run_server(test) < 0) {
iperf_err(test, "error - %s", iperf_strerror(i_errno));
++consecutive_errors;
if (consecutive_errors >= 5) {
iperf_errexit(test, "too many errors, exiting");
break;
}
} else
consecutive_errors = 0;
iperf_reset_test(test);
if (iperf_get_test_one_off(test))
break;
}
iperf_delete_pidfile(test);
break;
case 'c':
if (iperf_run_client(test) < 0)
iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
break;
default:
usage();
break;
}
iperf_catch_sigend(SIG_DFL);
return 0;
}
static int
run(struct iperf_test *test)
{
int consecutive_errors;
switch (test->role) {
case 's':
if (test->daemon) {
int rc = daemon(0, 0);
if (rc < 0) {
i_errno = IEDAEMON;
iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
}
}
consecutive_errors = 0;
if (iperf_create_pidfile(test) < 0) {
i_errno = IEPIDFILE;
iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
}
for (;;) {
if (iperf_run_server(test) < 0) {
iperf_err(test, "error - %s", iperf_strerror(i_errno));
fprintf(stderr, "\n");
++consecutive_errors;
if (consecutive_errors >= 5) {
fprintf(stderr, "too many errors, exiting\n");
break;
}
} else
consecutive_errors = 0;
iperf_reset_test(test);
if (iperf_get_test_one_off(test))
break;
}
iperf_delete_pidfile(test);
break;
case 'c':
if (iperf_run_client(test) < 0)
iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
break;
default:
usage();
break;
}
return 0;
}
int
main( int argc, char** argv )
{
char* argv0;
int port;
struct iperf_test *test;
int consecutive_errors;
argv0 = strrchr( argv[0], '/' );
if ( argv0 != (char*) 0 )
++argv0;
else
argv0 = argv[0];
if ( argc != 2 ) {
fprintf( stderr, "usage: %s [port]\n", argv0 );
exit( EXIT_FAILURE );
}
port = atoi( argv[1] );
test = iperf_new_test();
if ( test == NULL ) {
fprintf( stderr, "%s: failed to create test\n", argv0 );
exit( EXIT_FAILURE );
}
iperf_defaults( test );
iperf_set_verbose( test, 1 );
iperf_set_test_role( test, 's' );
iperf_set_test_server_port( test, port );
consecutive_errors = 0;
for (;;) {
if ( iperf_run_server( test ) < 0 ) {
fprintf( stderr, "%s: error - %s\n\n", argv0, iperf_strerror( i_errno ) );
++consecutive_errors;
if (consecutive_errors >= 5) {
fprintf(stderr, "%s: too many errors, exitting\n", argv0);
break;
}
} else
consecutive_errors = 0;
iperf_reset_test( test );
}
iperf_free_test( test );
exit( EXIT_SUCCESS );
}
void
iperf_error(char *estr)
{
LOGI("%s: %s\n", estr, iperf_strerror(i_errno));
}
