lib/libssh
1
0
Fork 0
mirror of https://git.libssh.org/projects/libssh.git synced 2025-08-05 20:55:46 +03:00
Code Activity

Reformat tests/benchmarks/benchmarks.c

Browse Source 
tests/benchmarks/benchmarks.c has been reformatted
according to current coding style.

Signed-off-by: Eshan Kelkar <eshankelkar@galorithm.com>
Reviewed-by: Sahana Prasad <sahana@redhat.com>
Reviewed-by: Jakub Jelen <jjelen@redhat.com>
This commit is contained in:
Eshan Kelkar
2023-06-01 20:18:05 +05:30
committed by Sahana Prasad
parent 4768d2970a
commit d8790d06c4
1 changed files with 320 additions and 290 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
tests/benchmarks/benchmarks.c
View File

@@ -73,7 +73,7 @@ struct benchmark benchmarks[]= {
const char *argp_program_version = "libssh benchmarks 2011-08-28";
const char *argp_program_bug_address = "Aris Adamantiadis <aris@0xbadc0de.be>";
static char **cmdline;
static char **cmdline = NULL;
/* Program documentation. */
static char doc[] = "libssh benchmarks";
@@ -128,7 +128,6 @@ static struct argp_option options[] = {
.flags = 0,
.doc = "Upload data using synchronous SFTP",
.group = 0
},
{
.name = "sync-sftp-download",
@@ -137,7 +136,6 @@ static struct argp_option options[] = {
.flags = 0,
.doc = "Download data using synchronous SFTP (slow)",
.group = 0
},
{
.name = "async-sftp-download",
@@ -146,7 +144,6 @@ static struct argp_option options[] = {
.flags = 0,
.doc = "Download data using asynchronous SFTP (fast)",
.group = 0
},
{
.name = "host",
@@ -193,7 +190,8 @@ static struct argp_option options[] = {
};
/* Parse a single option. */
static error_t parse_opt (int key, char *arg, struct argp_state *state) {
static error_t parse_opt (int key, char *arg, struct argp_state *state)
{
/* Get the input argument from argp_parse, which we
* know is a pointer to our arguments structure.
*/
@@ -233,6 +231,7 @@ static error_t parse_opt (int key, char *arg, struct argp_state *state) {
fprintf(stderr, "Too much hosts\n");
return ARGP_ERR_UNKNOWN;
}
arguments->hosts[arguments->nhosts] = arg;
arguments->nhosts++;
break;
@@ -253,7 +252,8 @@ static struct argp argp = {options, parse_opt, NULL, doc, NULL, NULL, NULL};
#endif /* HAVE_ARGP_H */
static void cmdline_parse(int argc, char **argv, struct argument_s *arguments) {
static void cmdline_parse(int argc, char **argv, struct argument_s *arguments)
{
/*
* Parse our arguments; every option seen by parse_opt will
* be reflected in arguments.
@@ -268,19 +268,23 @@ static void cmdline_parse(int argc, char **argv, struct argument_s *arguments) {
#endif /* HAVE_ARGP_H */
}
static void arguments_init(struct argument_s *arguments){
static void arguments_init(struct argument_s *arguments)
{
memset(arguments, 0, sizeof(*arguments));
arguments->chunksize = 32758;
arguments->concurrent_requests = 20;
arguments->datasize = 10;
}
static ssh_session connect_host(const char *host, int verbose, char *cipher){
static ssh_session connect_host(const char *host, int verbose, char *cipher)
{
ssh_session session = ssh_new();
if (session == NULL)
goto error;
if (ssh_options_set(session, SSH_OPTIONS_HOST, host) < 0)
goto error;
ssh_options_set(session, SSH_OPTIONS_LOG_VERBOSITY, &verbose);
if (cipher != NULL) {
if (ssh_options_set(session, SSH_OPTIONS_CIPHERS_C_S, cipher) ||
@@ -288,19 +292,23 @@ static ssh_session connect_host(const char *host, int verbose, char *cipher){
goto error;
}
}
ssh_options_parse_config(session, NULL);
if (ssh_connect(session) == SSH_ERROR)
goto error;
if (ssh_userauth_autopubkey(session, NULL) != SSH_AUTH_SUCCESS)
goto error;
return session;
error:
fprintf(stderr,"Error connecting to \"%s\": %s\n",host,ssh_get_error(session));
fprintf(stderr, "Error connecting to \"%s\": %s\n",
host, ssh_get_error(session));
ssh_free(session);
return NULL;
}
static char *network_speed(float bps){
static char *network_speed(float bps)
{
static char buf[128];
if (bps > 1000 * 1000 * 1000) {
/* Gbps */
@@ -313,44 +321,55 @@ static char *network_speed(float bps){
} else {
snprintf(buf, sizeof(buf), "%f bps", bps);
}
return buf;
}
static void do_benchmarks(ssh_session session, struct argument_s *arguments,
const char *hostname){
const char *hostname)
{
float ping_rtt = 0.0;
float ssh_rtt = 0.0;
float bps = 0.0;
int i;
int err;
struct benchmark *b;
struct benchmark *b = NULL;
if (arguments->verbose > 0)
fprintf(stdout, "Testing ICMP RTT\n");
err = benchmarks_ping_latency(hostname, &ping_rtt);
if (err == 0) {
fprintf(stdout, "ping RTT : %f ms\n", ping_rtt);
}
err = benchmarks_ssh_latency(session, &ssh_rtt);
if (err == 0) {
fprintf(stdout, "SSH RTT : %f ms. Theoretical max BW (win=128K) : %s\n",ssh_rtt,network_speed(128000.0/(ssh_rtt / 1000.0)));
fprintf(stdout,
"SSH RTT : %f ms. Theoretical max BW (win=128K) : %s\n",
ssh_rtt, network_speed(128000.0 / (ssh_rtt / 1000.0)));
}
for (i=0; i < BENCHMARK_NUMBER; ++i){
b = &benchmarks[i];
if (b->enabled) {
err=b->fct(session, arguments, &bps);
if (err == 0) {
fprintf(stdout, "%s : %s : %s\n",hostname, b->name, network_speed(bps));
fprintf(stdout,
"%s : %s : %s\n",
hostname, b->name, network_speed(bps));
}
}
}
}
char *buffer;
char *buffer = NULL;
int main(int argc, char **argv){
int main(int argc, char **argv)
{
struct argument_s arguments;
ssh_session session;
ssh_session session = NULL;
int i;
arguments_init(&arguments);
@@ -359,44 +378,55 @@ int main(int argc, char **argv){
fprintf(stderr, "At least one host (-h) must be specified\n");
return EXIT_FAILURE;
}
if (arguments.ntests == 0) {
for (i=0; i < BENCHMARK_NUMBER; ++i) {
benchmarks[i].enabled = 1;
}
arguments.ntests = BENCHMARK_NUMBER;
}
buffer = malloc(arguments.chunksize > 1024 ? arguments.chunksize : 1024);
if (buffer == NULL) {
fprintf(stderr, "Allocation of chunk buffer failed\n");
return EXIT_FAILURE;
}
if (arguments.verbose > 0) {
fprintf(stdout, "Will try hosts ");
for (i=0; i < arguments.nhosts; ++i) {
fprintf(stdout, "\"%s\" ", arguments.hosts[i]);
}
fprintf(stdout, "with benchmarks ");
for (i = 0; i < BENCHMARK_NUMBER; ++i) {
if (benchmarks[i].enabled)
fprintf(stdout, "\"%s\" ", benchmarks[i].name);
}
fprintf(stdout,"\n");
}
for (i = 0; i < arguments.nhosts; ++i) {
if (arguments.verbose > 0)
fprintf(stdout, "Connecting to \"%s\"...\n", arguments.hosts[i]);
session=connect_host(arguments.hosts[i], arguments.verbose, arguments.cipher);
session = connect_host(arguments.hosts[i],
arguments.verbose,
arguments.cipher);
if (session != NULL && arguments.verbose > 0)
fprintf(stdout, "Success\n");
if (session == NULL) {
fprintf(stderr, "Errors occurred, stopping\n");
return EXIT_FAILURE;
}
do_benchmarks(session, &arguments, arguments.hosts[i]);
ssh_disconnect(session);
ssh_free(session);
}
return EXIT_SUCCESS;
}