aboutsummaryrefslogtreecommitdiff
path: root/tests/benchmarks/bench_raw.c
diff options
context:
space:
mode:
authorAris Adamantiadis <aris@0xbadc0de.be>2011-08-28 13:29:25 +0300
committerAris Adamantiadis <aris@0xbadc0de.be>2011-09-02 11:43:05 +0300
commit5d9cd8bf39cac07fb674a4bb8c7667dba1773ba2 (patch)
tree08faf1862ca40a5c64e2e5ac86b945fd6586d959 /tests/benchmarks/bench_raw.c
parenta2c94abb92ead1503a0d6284609af41e53bef402 (diff)
downloadlibssh-5d9cd8bf39cac07fb674a4bb8c7667dba1773ba2.tar.gz
libssh-5d9cd8bf39cac07fb674a4bb8c7667dba1773ba2.tar.xz
libssh-5d9cd8bf39cac07fb674a4bb8c7667dba1773ba2.zip
benchmarks: added raw_download test
Diffstat (limited to 'tests/benchmarks/bench_raw.c')
-rw-r--r--tests/benchmarks/bench_raw.c104
1 files changed, 100 insertions, 4 deletions
diff --git a/tests/benchmarks/bench_raw.c b/tests/benchmarks/bench_raw.c
index c870942a..431659c5 100644
--- a/tests/benchmarks/bench_raw.c
+++ b/tests/benchmarks/bench_raw.c
@@ -109,13 +109,12 @@ int benchmarks_raw_up (ssh_session session, struct argument_s *args,
unsigned long bytes=0x1000000;
char *script=get_python_eater(bytes);
char cmd[128];
- char buffer[1024];
+ static char buffer[0x10000];
int err;
ssh_channel channel;
struct timestamp_struct ts;
float ms=0.0;
unsigned long total=0;
- (void)bps;
err=upload_script(session,"/tmp/eater.py",script);
free(script);
@@ -144,8 +143,8 @@ int benchmarks_raw_up (ssh_session session, struct argument_s *args,
while(total < bytes){
unsigned long towrite = bytes - total;
int w;
- if(towrite > 0x1000)
- towrite = 0x1000;
+ if(towrite > 0x10000)
+ towrite = 32758;
w=ssh_channel_write(channel,buffer,towrite);
if(w == SSH_ERROR)
goto error;
@@ -180,3 +179,100 @@ error:
}
return -1;
}
+
+const char python_giver[] =
+"#!/usr/bin/python\n"
+"import sys\n"
+"r=sys.stdin.read(2)\n"
+"towrite=XXXXXXXXXX\n"
+"wrote=0\n"
+"while(wrote < towrite):\n"
+" buffersize=towrite-wrote\n"
+" if(buffersize > 4096):\n"
+" buffersize=4096\n"
+" sys.stdout.write('A'*buffersize)\n"
+" wrote+=buffersize\n"
+"sys.stdout.flush()\n";
+
+static char *get_python_giver(unsigned long bytes){
+ char *giver=malloc(sizeof(python_giver));
+ char *ptr;
+ char buffer[12];
+
+ memcpy(giver,python_giver,sizeof(python_giver));
+ ptr=strstr(giver,"XXXXXXXXXX");
+ if(!ptr){
+ free(giver);
+ return NULL;
+ }
+ sprintf(buffer,"0x%.8lx",bytes);
+ memcpy(ptr,buffer,10);
+ return giver;
+}
+
+/** @internal
+ * @brief benchmarks a raw download (simple upload in a SSH channel) using an
+ * existing SSH session.
+ * @param[in] session Open SSH session
+ * @param[in] args Parsed command line arguments
+ * @param[out] bps The calculated bytes per second obtained via benchmark.
+ * @return 0 on success, -1 on error.
+ */
+int benchmarks_raw_down (ssh_session session, struct argument_s *args,
+ float *bps){
+ unsigned long bytes=0x1000000;
+ char *script=get_python_giver(bytes);
+ char cmd[128];
+ static char buffer[0x10000];
+ int err;
+ ssh_channel channel;
+ struct timestamp_struct ts;
+ float ms=0.0;
+ unsigned long total=0;
+
+ err=upload_script(session,"/tmp/giver.py",script);
+ free(script);
+ if(err<0)
+ return err;
+ channel=ssh_channel_new(session);
+ if(channel == NULL)
+ goto error;
+ if(ssh_channel_open_session(channel)==SSH_ERROR)
+ goto error;
+ snprintf(cmd,sizeof(cmd),"%s /tmp/giver.py", PYTHON_PATH);
+ if(ssh_channel_request_exec(channel,cmd)==SSH_ERROR)
+ goto error;
+ if((err=ssh_channel_write(channel,"go",2))==SSH_ERROR)
+ goto error;
+ if(args->verbose>0)
+ fprintf(stdout,"Starting download of %lu bytes now\n",bytes);
+ timestamp_init(&ts);
+ while(total < bytes){
+ unsigned long toread = bytes - total;
+ int r;
+ if(toread > sizeof(buffer))
+ toread = sizeof(buffer);
+ r=ssh_channel_read(channel,buffer,toread,0);
+ if(r == SSH_ERROR)
+ goto error;
+ total += r;
+ }
+
+ if(args->verbose>0)
+ fprintf(stdout,"Finished download\n");
+ ms=elapsed_time(&ts);
+ *bps=8000 * (float)bytes / ms;
+ if(args->verbose > 0)
+ fprintf(stdout,"Download took %f ms for %lu bytes, at %f bps\n",ms,
+ bytes,*bps);
+ ssh_channel_close(channel);
+ ssh_channel_free(channel);
+ return 0;
+error:
+ fprintf(stderr,"Error during raw upload : %s\n",ssh_get_error(session));
+ if(channel){
+ ssh_channel_close(channel);
+ ssh_channel_free(channel);
+ }
+ return -1;
+}