add benchmark mode

This commit is contained in:
ubuntu201711081 2020-12-05 17:33:41 +00:00
parent 62c116200c
commit b7cbf73ce3
4 changed files with 63 additions and 22 deletions

View File

@ -7,13 +7,16 @@ To build source, you should install `build-essential`.
Usage: Usage:
```bash ```bash
./server [OPTION]... ./server [OPTION]...
./client SERVERNAME PORT FILENAME ./client SERVERNAME PORT [option] [FILENAME]...
``` ```
Server OPTION and arguments: Server OPTION and arguments:
- `-p port` :set to port binding. couldn't set to 0 - `-p port` :set to port binding. couldn't set to 0
- `-h` :print help message. - `-h` :print help message.
Client option and arguments:
- `-b` :benchmark mode
Available macro: Available macro:
For server For server

View File

@ -12,6 +12,9 @@
#include <stdbool.h> #include <stdbool.h>
#include <assert.h> #include <assert.h>
#include <fcntl.h> #include <fcntl.h>
#include <time.h>
#include <sys/time.h>
#include <sys/times.h>
#include "socket_wrapper.h" #include "socket_wrapper.h"
#ifdef USE_SENDFILE #ifdef USE_SENDFILE
@ -169,24 +172,33 @@ int main(int argc, const char *argv[]){
const char * filename; const char * filename;
const char * server_name; const char * server_name;
in_port_t server_port = 0; in_port_t server_port = 0;
int arg_filename_start = 3;
int sock, err; int sock, err;
if (argc != 4){ int retval = 0;
fprintf(stderr,"USAUE: %s SERVERNAME PORT FILENAME\n",argv[0]); bool benchmode = false;
clock_t clock_sum = 0;
clock_t begin_sclock;
clock_t begin_uclock;
int op_count = 0;
if (argc < 4){
fprintf(stderr,"USAUE: %s SERVERNAME PORT [Option]... [FILENAME]...\n",argv[0]);
return 1; return 1;
} }
server_name = argv[1]; server_name = argv[1];
server_port = atoi(argv[2]); server_port = atoi(argv[2]);
filename = argv[3]; if (strcmp("-b",argv[arg_filename_start])==0
if (server_port == 0) ||strcmp("--benchmark",argv[arg_filename_start])==0){
{ arg_filename_start++;
benchmode = true;
}
if (server_port == 0){
fprintf(stderr,"port invalid"); fprintf(stderr,"port invalid");
return 1; return 1;
} }
sock = socket(AF_INET,SOCK_STREAM,0);
if(sock < 0){
perror("sock create fail");
return 1;
}
err = getsockaddrbyname(AF_INET,SOCK_STREAM,0,server_name,(struct sockaddr *)&addr); err = getsockaddrbyname(AF_INET,SOCK_STREAM,0,server_name,(struct sockaddr *)&addr);
if (err != 0){ if (err != 0){
int check; int check;
@ -196,21 +208,49 @@ int main(int argc, const char *argv[]){
assert(check != -1); assert(check != -1);
if (check == 0){ if (check == 0){
fprintf(stderr,"parsing fail : invaild format\n"); fprintf(stderr,"parsing fail : invaild format\n");
close(sock);
return 1; return 1;
} }
} }
addr.sin_family = AF_INET; addr.sin_family = AF_INET;
addr.sin_port = htons(server_port); addr.sin_port = htons(server_port);
if (benchmode){
struct tms t;
times(&t);
begin_sclock = t.tms_stime;
begin_uclock = t.tms_utime;
}
while (arg_filename_start < argc){
filename = argv[arg_filename_start++];
sock = socket(AF_INET,SOCK_STREAM,0);
if(sock < 0){
perror("sock create fail");
return 1;
}
if(connect(sock,(struct sockaddr *)&addr,sizeof(addr)) < 0){ if(connect(sock,(struct sockaddr *)&addr,sizeof(addr)) < 0){
perror("connect failed"); perror("connect failed");
return 1; return 1;
} }
if(sendReadOp(sock,filename) == 0){ if(sendReadOp(sock,filename) == 0){
int ret = recvData(sock,filename); int ret = recvData(sock,filename);
close(sock); if (ret < 0){
return ret; fprintf(stderr,"%d",ret);
}
retval += ret;
} }
close(sock); close(sock);
return 0; op_count++;
}
if (benchmode){
struct tms t;
times(&t);
clock_sum += t.tms_stime - begin_sclock;
clock_sum += t.tms_utime - begin_uclock;
}
if (benchmode){
fprintf(stdout,"operation: %lf ticks/op\n",((double)clock_sum)/((double)op_count));
}
return retval;
} }

View File

@ -12,7 +12,6 @@
#include <stdbool.h> #include <stdbool.h>
#include <assert.h> #include <assert.h>
#include <fcntl.h> #include <fcntl.h>
#include "socket_wrapper.h" #include "socket_wrapper.h"
#ifdef USE_SENDFILE #ifdef USE_SENDFILE

View File

@ -288,7 +288,6 @@ void * worker_proc(void * data){
} }
} }
static pthread_t worker_threads[MAX_THREAD_NUMBER]; static pthread_t worker_threads[MAX_THREAD_NUMBER];
static int sock; static int sock;