Annotation of embedaddon/iperf/src/main.c, revision 1.1
1.1 ! misho 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 TTD@lbl.gov.
! 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: #ifdef HAVE_STDINT_H
! 45: #include <stdint.h>
! 46: #endif
! 47: #include <netinet/tcp.h>
! 48:
! 49: #include "iperf.h"
! 50: #include "iperf_api.h"
! 51: #include "units.h"
! 52: #include "iperf_locale.h"
! 53: #include "net.h"
! 54:
! 55:
! 56: static int run(struct iperf_test *test);
! 57:
! 58:
! 59: /**************************************************************************/
! 60: int
! 61: main(int argc, char **argv)
! 62: {
! 63: struct iperf_test *test;
! 64:
! 65: // XXX: Setting the process affinity requires root on most systems.
! 66: // Is this a feature we really need?
! 67: #ifdef TEST_PROC_AFFINITY
! 68: /* didnt seem to work.... */
! 69: /*
! 70: * increasing the priority of the process to minimise packet generation
! 71: * delay
! 72: */
! 73: int rc = setpriority(PRIO_PROCESS, 0, -15);
! 74:
! 75: if (rc < 0) {
! 76: perror("setpriority:");
! 77: fprintf(stderr, "setting priority to valid level\n");
! 78: rc = setpriority(PRIO_PROCESS, 0, 0);
! 79: }
! 80:
! 81: /* setting the affinity of the process */
! 82: cpu_set_t cpu_set;
! 83: int affinity = -1;
! 84: int ncores = 1;
! 85:
! 86: sched_getaffinity(0, sizeof(cpu_set_t), &cpu_set);
! 87: if (errno)
! 88: perror("couldn't get affinity:");
! 89:
! 90: if ((ncores = sysconf(_SC_NPROCESSORS_CONF)) <= 0)
! 91: err("sysconf: couldn't get _SC_NPROCESSORS_CONF");
! 92:
! 93: CPU_ZERO(&cpu_set);
! 94: CPU_SET(affinity, &cpu_set);
! 95: if (sched_setaffinity(0, sizeof(cpu_set_t), &cpu_set) != 0)
! 96: err("couldn't change CPU affinity");
! 97: #endif
! 98:
! 99: test = iperf_new_test();
! 100: if (!test)
! 101: iperf_errexit(NULL, "create new test error - %s", iperf_strerror(i_errno));
! 102: iperf_defaults(test); /* sets defaults */
! 103:
! 104: if (iperf_parse_arguments(test, argc, argv) < 0) {
! 105: iperf_err(test, "parameter error - %s", iperf_strerror(i_errno));
! 106: fprintf(stderr, "\n");
! 107: usage_long();
! 108: exit(1);
! 109: }
! 110:
! 111: if (run(test) < 0)
! 112: iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
! 113:
! 114: iperf_free_test(test);
! 115:
! 116: return 0;
! 117: }
! 118:
! 119:
! 120: static jmp_buf sigend_jmp_buf;
! 121:
! 122: static void
! 123: sigend_handler(int sig)
! 124: {
! 125: longjmp(sigend_jmp_buf, 1);
! 126: }
! 127:
! 128: /**************************************************************************/
! 129: static int
! 130: run(struct iperf_test *test)
! 131: {
! 132: /* Termination signals. */
! 133: iperf_catch_sigend(sigend_handler);
! 134: if (setjmp(sigend_jmp_buf))
! 135: iperf_got_sigend(test);
! 136:
! 137: switch (test->role) {
! 138: case 's':
! 139: if (test->daemon) {
! 140: int rc = daemon(0, 0);
! 141: if (rc < 0) {
! 142: i_errno = IEDAEMON;
! 143: iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
! 144: }
! 145: }
! 146: if (iperf_create_pidfile(test) < 0) {
! 147: i_errno = IEPIDFILE;
! 148: iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
! 149: }
! 150: for (;;) {
! 151: int rc;
! 152: rc = iperf_run_server(test);
! 153: if (rc < 0) {
! 154: iperf_err(test, "error - %s", iperf_strerror(i_errno));
! 155: if (rc < -1) {
! 156: iperf_errexit(test, "exiting");
! 157: break;
! 158: }
! 159: }
! 160: iperf_reset_test(test);
! 161: if (iperf_get_test_one_off(test))
! 162: break;
! 163: }
! 164: iperf_delete_pidfile(test);
! 165: break;
! 166: case 'c':
! 167: if (iperf_run_client(test) < 0)
! 168: iperf_errexit(test, "error - %s", iperf_strerror(i_errno));
! 169: break;
! 170: default:
! 171: usage();
! 172: break;
! 173: }
! 174:
! 175: iperf_catch_sigend(SIG_DFL);
! 176:
! 177: return 0;
! 178: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>