以下是iperf3的C代码，可以在GitHub上找到完整的代码：
```c
#include <stdio
.h>
#include <stdlib
.h>
#include <string
.h>
#include <unistd
.h>
#include <signal
.h>
#include <errno
.h>
#include <netdb
.h>
#include <sys/types
.h>
#include <sys/socket
.h>
#include <netinet/in
.h>
#include <arpa/inet
.h>
#include "iperf
.h"
#include "iperf_api
.h"
#include "iperf_locale
.h"
#include "units
.h"
#include "net
.h"
#include "timer
.h"
#include "tcp_window_size
.h"
#include "cjson
.h"
#ifdef HAVE_SSL
#include <openssl/ssl
.h>
#include <openssl/err
.h>
#endif
/* Forwards
. */
static int iperf_run(struct iperf_test *test);
/**************************************************************************/
int
main(int argc, char **argv)
{
int i;
struct iperf_test *test;
int result;
struct sigaction sa;
char client_version[10] = IPERF_VERSION;
setlocale(LC_ALL, "");
bindtextdomain("iperf3", LOCALEDIR);
textdomain("iperf3");
/* Initialize settings
. */
test = iperf_new_test();
/* Register a few signal handlers to make clean exit more likely
. */
memset(&sa, 0, sizeof(sa));
sa
.sa_handler = &iperf_got_sigint;
sigaction(SIGINT, &sa, NULL);
sa
.sa_handler = &iperf_got_sigpipe;
sigaction(SIGPIPE, &sa, NULL);
/* Parse command line options, handling some options immediately
. */
i = iperf_parse_arguments(test, argc, argv);
if (i < 0) {
fprintf(stderr, "%s", iperf_strerror(i));
exit(1);
}
if (test
->version) {
printf("%s\n", client_version);
exit(0);
} else if (test
->help) {
iperf_printf(test, "\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n",
"Usage: iperf3 [
-options] [
-s|
-c host] [options]\n",
"Client/Server:",
"
-V,
--version show version information and quit
.",
"
-h,
--help show this message and quit
.",
"
-i,
--interval n seconds between periodic bandwidth reports
.",
"
-f,
--format [kmKM] format to report: K
bits, M
bits, KBytes, MBytes",
"
-P,
--parallel n number of parallel client streams to run
.");
iperf_printf(test, "\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n",
"Client specific:",
"
-c,
--client <host> run in client mode, connecting to <host>
.",
"
-u,
--udp use UDP rather than TCP
.",
"
-b,
--bitrate #[KMG][/#] target bitrate in
bits/sec (0 for unlimited)",
" (default 1 Mbit/sec for UDP, unlimited for TCP)",
"
-t,
--time n time in seconds to transmit for (default 10 secs)",
"
-n,
--bytes n number of bytes to transmit (instead of
-t)",
"
-k,
--blockcount #[KMG] number of blocks (packets) to transmit (instead of
-t or
-n)",
"
-l,
--length #[KMG] length of buffer to read or write",
" (default 128 KB for TCP, dynamic or 8 KB for UDP)",
"
-R,
--reverse reverse the direction of a test (client sends, server receives)
.");
iperf_printf(test, "\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n",
"Server specific:",
"
-s,
--server run in server mode
.",
"
-D,
--daemon run the server as a daemon",
"
-I,
--pidfile file write PID file (default /var/run/iperf3
.pid)",
"
-1,
--one
-off handle one client connection then exit
.",
"
-B,
--bind <host> bind to a specific interface, e
.g
. eth0");
iperf_printf(test, "\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n%s\n\n",
"JSON options:",
"
-J,
--json output in JSON format",
"
--logfile f send output to a log file",
"",
"For more information and tuning options,
see iperf3's man page
.",
"",
"Please report bugs to https://github
.com/esnet/iperf",
"");
exit(0);
}
/* Check for version number consistency
. */
if (test
->protocol
->id == Ptcp && test
->version == 3 && test
->tcp
.omit_version) {
i = 2;
} else {
i = 3;
}
if (strncmp(client_version, test
->version_string, i) != 0) {
iperf_printf(test, "warning: version number mismatch (client %s, server %s)\n",
client_version, test
->version_string);
}
/* Initialize IP protocol */
if (test
->cookiefile) {
test
->cookie = iperf_load_cookie(test
->cookiefile);
if (test
->cookie == NULL) {
iperf_errexit(test, "unable to load cookie from file '%s'", test
->cookiefile);
}
}
if (test
->
reverse)
test
->settings
->
reverse = 1;
if (test
->protocol
->id == Pudp) {
if (test
->settings
->mss || test
->settings
->socket_bufsize) {
iperf_printf(test, "warning: MSS and socket buffer size settings are not used in UDP mode
.\n");
}
if (test
->settings
->no_delay) {
iperf_printf(test, "warning: the TCP_NODELAY option is not used in UDP mode
.\n");
}
if (test
->settings
->pmtu !=
-1) {
iperf_printf(test, "warning: the PMTU option is not used in UDP mode
.\n");
}
}
#ifdef HAVE_SSL
/* Initialize SSL library */
if (test
->protocol
->id == Ptcp && test
->settings
->ssl) {
SSL_load_error_strings();
SSL_library_init();
test
->sslctx = SSL_CTX_new(TLS_client_method());
if (test
->sslctx == NULL) {
iperf_errexit(test, "failed to create SSL context\n");
}
if (test
->settings
->ssl_cafile) {
if (SSL_CTX_load_verify_locations(test
->sslctx, test
->settings
->ssl_cafile, NULL) != 1) {
SSL_CTX_free(test
->sslctx);
iperf_errexit(test, "failed to load CA certificates from %s\n", test
->settings
->ssl_cafile);
}
}
if (test
->settings
->ssl_cert) {
if (SSL_CTX_use_certificate_chain_file(test
->sslctx, test
->settings
->ssl_cert) != 1) {
SSL_CTX_free(test
->sslctx);
iperf_errexit(test, "failed to load SSL certificate from %s\n", test
->settings
->ssl_cert);
}
}
if (test
->settings
->ssl_key) {
if (SSL_CTX_use_PrivateKey_file(test
->sslctx, test
->settings
->ssl_key, SSL_FILETYPE_PEM) != 1) {
SSL_CTX_free(test
->sslctx);
iperf_errexit(test, "failed to load SSL key from %s\n", test
->settings
->ssl_key);
}
}
SSL_CTX_set_verify(test
->sslctx, SSL_VERIFY_PEER, NULL);
}
#endif
/* Daemon mode
. */
if (test
->daemon) {
if (daemon(0, 0) != 0) {
iperf_errexit(test, "error
- failed to become a daemon: %s\n", strerror(errno));
}
if (test
->pidfile) {
FILE *f;
f = fopen(test
->pidfile, "w");
if (f == NULL) {
iperf_errexit(test, "error
- unable to write PID file '%s': %s\n", test
->pidfile, strerror(errno));
}
fprintf(f, "%d\n", getpid());
fclose(f);
}
}
/* Ignore SIGPIPE to simplify error handling */
sa
.sa_handler = SIG_IGN;
sigaction(SIGPIPE, &sa, NULL);
/* If we are doing a single client (
-1) and it is a daemon, don't do
-D again */
if (test
->daemon && test
->num_ostreams == 1)
test
->daemon = 0;
/* Defer daemon mode until after the above check for single client + daemon */
if (test
->daemon) {
if (daemon(0, 0) != 0) {
iperf_errexit(test, "error
- failed to become a daemon: %s\n", strerror(errno));
}
if (test
->pidfile) {
FILE *f;
f = fopen(test
->pidfile, "w");
if (f == NULL) {
iperf_errexit(test, "error
- unable to write PID file '%s': %s\n", test
->pidfile, strerror(errno));
}
fprintf(f, "%d\n", getpid());
fclose(f);
}
}
/* Ignore SIGPIPE to simplify error handling */
sa
.sa_handler = SIG_IGN;
sigaction(SIGPIPE, &sa, NULL);
/* Set up the output stream */
if (test
->json_output) {
test
->outfile = json_get_output_stream(test
->json_output_file);
} else if (test
->logfile) {
test
->outfile = fopen(test
->logfile, "w");
if (test
->outfile == NULL) {
iperf_errexit(test, "error
- unable to write log to file '%s': %s\n", test
->logfile, strerror(errno));
}
} else {
test
->outfile = stdout;
}
/* Start the client or server */
if (test
->server) {
if (test
->daemon) {
iperf_printf(test, "Server listening on port %d\n", test
->server_port);
}
else {
iperf_printf(test, "
-----------------------------------------------------------\n");
iperf_printf(test, "Server listening on %s port %d\n", test
->settings
->domain == AF_INET6 ? "[::]" : "0
.0
.0
.0", test
->server_port);
iperf_printf(test, "
-----------------------------------------------------------\n");
}
result = iperf_run_server(test);
} else {
if (test
->daemon) {
iperf_printf(test, "Client connecting to %s, TCP port %d\n", test
->server_hostname, test
->server_port);
}
else {
if (test
->
reverse)
iperf_printf(test, "
-----------------------------------------------------------\n");
iperf_printf(test, "Client connecting to %s, %s port %d\n", test
->server_hostname, test
->protocol
->name, test
->server_port);
if (test
->
reverse)
iperf_printf(test, "
-----------------------------------------------------------\n");
}
result = iperf_run_client(test);
}
if (test
->cookie)
iperf_delete_cookie(test
->cookie);
#ifdef HAVE_SSL
if (test
->protocol
->id == Ptcp && test
->settings
->ssl) {
SSL_CTX_free(test
->sslctx);
}
ERR_free_strings();
#endif
iperf_free_test(test);
return result;
}
static int
iperf_run(struct iperf_test *test)
{
test
->start_time = milliseconds();
test
->next_time = test
->start_time;
test
->bytes_sent = 0;
test
->blocks_sent = 0;
test
->retransmits = 0;
if (test
->server_hostname) {
test
->server_hostname_len = strlen(test
->server_hostname);
}
if (test
->bind_address) {
test
->bind_address_len = strlen(test
->bind_address);
}
if (test
->json_output) {
cJSON *json_output = cJSON_CreateObject();
if (json_output == NULL) {
iperf_errexit(test, "error
- cJSON_CreateObject failed: %s\n", strerror(errno));
}
cJSON_AddItemToObject(json_output, "start", cJSON_CreateNumber((double) test
->start_time / 1000));
if (test
->verbose) {
cJSON_AddItemToObject(json_output, "verbose", cJSON_CreateNumber(1));
}
cJSON_AddItemToObject(json_output, "system_info", cJSON_CreateObject());
iperf_json_printf(json_output, "version", "%s", test
->version);
iperf_json_printf(json_output, "system_info", "%s", get_system_info());
if (test
->title) {
iperf_json_printf(json_output, "title", "%s", test
->title);
}
if (test
->extra_data) {
cJSON_AddItemToObject(json_output, "extra_data", cJSON_Parse(test
->extra_data));
}
test
->json_start_time = milliseconds();
test
->json_output_string = cJSON_Print(json_output);
cJSON_Delete(json_output);
if (test
->json_output_string == NULL) {
iperf_errexit(test, "error
- cJSON_Print failed: %s\n", strerror(errno));
}
}
if (test
->protocol
->id == Ptcp) {
if (test
->settings
->socket_bufsize) {
if (iperf_set_tcp_windowsize(test) != 0) {
iperf_errexit(test, "error
- %s\n", strerror(errno));
}
}
}
if (test
->
reverse) {
if (test
->protocol
->id == Ptcp) {
if (iperf_create_streams(test, test
->
reverse) < 0) {
iperf_errexit(test, "error
- %s\n", strerror(errno));
}
}
if (iperf_connect(test, test
->
reverse) < 0) {
iperf_errexit(test, "error
- %s\n", strerror(errno));
}
} else {
if (iperf_create_streams(test, test
->
reverse) < 0) {
iperf_errexit(test, "error
- %s\n", strerror(errno));
}
if (iperf_connect(test, test
->
reverse) < 0) {
iperf_errexit(test, "error
- %s\n", strerror(errno));
}
}
if (test
->json_output) {
cJSON *json_output = cJSON_CreateObject();
if (json_output == NULL) {
iperf_errexit(test, "error
- cJSON_CreateObject failed: %s\n", strerror(errno));
}
cJSON_AddItemToObject(json_output, "start", cJSON_CreateNumber((double) test
->start_time / 1000));
if (test
->verbose) {
cJSON_AddItemToObject(json_output, "verbose", cJSON_CreateNumber(1));
}
iperf_json_printf(json_output, "interval", "%d", test
->settings
->stats_interval);
if (test
->title) {
iperf_json_printf(json_output, "title", "%s", test
->title);
}
if (test
->extra_data) {
cJSON_AddItemToObject(json_output, "extra_data", cJSON_Parse(test
->extra_data));
}
iperf_json_printf(json_output, "start_connected", "%d", test
->connected);
cJSON_AddItemToObject(json_output, "intervals", cJSON_CreateArray());
test
->json_output_string = cJSON_Print(json_output);
cJSON_Delete(json_output);
if (test
->json_output_string == NULL) {
iperf_errexit(test, "error
- cJSON_Print failed: %s\n", strerror(errno));
}
}
if (test
->protocol
->id == Pudp) {
if (iperf_udp_connect(test) < 0) {
iperf_errexit(test, "error
- %s\n", strerror(errno));
}
}
if (test
->
reverse) {
if (test
->protocol
->id == Ptcp) {
if (iperf_listen(test) < 0) {
iperf_errexit(test, "error
- %s\n", strerror(errno));
}
}
if (test
->protocol
->id == Pudp || test
->protocol
->id == Psctp) {
if (iperf_run_server_udp(test) < 0) {
iperf_errexit(test, "error
- %s\n", strerror(errno));
}
} else {
if (iperf_run_server_tcp(test) < 0) {
iperf_errexit(test, "error
- %s\n", strerror(errno));
}
}
} else {
if (test
->protocol
->id == Ptcp) {
if (iperf_run_client_tcp(test) < 0) {
iperf_errexit(test, "error
- %s\n", strerror(errno));
}
} else {
if (iperf_run_client_udp(test) < 0) {
iperf_errexit(test, "error
- %s\n", strerror(errno));
}
}
}
if (test
->json_output) {
cJSON *json_output = cJSON_CreateObject();
if (json_output == NULL) {
iperf_errexit(test, "error
- cJSON_CreateObject failed: %s\n", strerror(errno));
}
cJSON_AddItemToObject(json_output, "start", cJSON_CreateNumber((double) test
->start_time / 1000));
if (test
->verbose) {
cJSON_AddItemToObject(json_output, "verbose", cJSON_CreateNumber(1));
}
iperf_json_printf(json_output, "interval", "%d", test
->settings
->stats_interval);
if (test
->title) {
iperf_json_printf(json_output, "title", "%s", test
->title);
}
if (test
扫一扫
2609
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
