程序参数解析函数使用getopt_long()函数，改编自webbench

最新推荐文章于 2022-05-13 11:07:08 发布

禾斗人

最新推荐文章于 2022-05-13 11:07:08 发布

阅读量562

点赞数

分类专栏： C/C++ 文章标签：参数解析 getopt_long

本文链接：https://blog.csdn.net/enwhat/article/details/46124327

版权

C/C++ 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

#include "socket.c"
#include <unistd.h>
#include <sys/param.h>
#include <rpc/types.h>
#include <getopt.h>
#include <strings.h>
#include <time.h>
#include <signal.h>

/* Allow: GET, HEAD, OPTIONS, TRACE */
#define METHOD_GET 0
#define METHOD_HEAD 1
#define METHOD_OPTIONS 2
#define METHOD_TRACE 3
#define PROGRAM_VERSION "1.5"
int method=METHOD_GET;
int clients=1;
int force=0;
int force_reload=0;
int proxyport=80;
int http10=1;
char *proxyhost=NULL;
int benchtime=30;

static const struct option long_options[]=
{
    {"force",no_argument,&force,1},
    {"reload",no_argument,&force_reload,1},
    {"time",required_argument,NULL,'t'},
    {"help",no_argument,NULL,'?'},
    {"http09",no_argument,NULL,'9'},
    {"http10",no_argument,NULL,'1'},
    {"http11",no_argument,NULL,'2'},
    {"get",no_argument,&method,METHOD_GET},
    {"head",no_argument,&method,METHOD_HEAD},
    {"options",no_argument,&method,METHOD_OPTIONS},
    {"trace",no_argument,&method,METHOD_TRACE},
    {"version",no_argument,NULL,'V'},
    {"proxy",required_argument,NULL,'p'},
    {"clients",required_argument,NULL,'c'},
    {NULL,0,NULL,0}
};

/* prototypes */
static void benchcore(const char* host,const int port, const char *request);
static int bench(void);
static void build_request(const char *url);

static void usage(void)
{
    fprintf(stderr,
            "webbench [option]... URL\n"
            " -f|--force               Don't wait for reply from server.\n"
            " -r|--reload              Send reload request - Pragma: no-cache.\n"
            " -t|--time <sec>          Run benchmark for <sec> seconds. Default 30.\n"
            " -p|--proxy <server:port> Use proxy server for request.\n"
            " -c|--clients <n>         Run <n> HTTP clients at once. Default one.\n"
            " -9|--http09              Use HTTP/0.9 style requests.\n"
            " -1|--http10              Use HTTP/1.0 protocol.\n"
            " -2|--http11              Use HTTP/1.1 protocol.\n"
            " --get                    Use GET request method.\n"
            " --head                   Use HEAD request method.\n"
            " --options                Use OPTIONS request method.\n"
            " --trace                  Use TRACE request method.\n"
            " -?|-h|--help             This information.\n"
            " -V|--version             Display program version.\n"
           );
};
int main(int argc, char *argv[])
{
    int opt=0;
    int options_index=0;
    char *tmp=NULL;

    if(argc==1)
    {
        usage();
        return 2;
    }

    while((opt=getopt_long(argc,argv,"912Vfrt:p:c:?h",long_options,&options_index))!=EOF )
    {
        switch(opt)
        {
            case 0 : break;
            case 'f': force=1;break;
            case 'r': force_reload=1;break;
            case '9': http10=0;break;
            case '1': http10=1;break;
            case '2': http10=2;break;
            case 'V': printf(PROGRAM_VERSION"\n");exit(0);
            case 't': benchtime=atoi(optarg);break;
            case 'p':
                      /* proxy server parsing server:port */
                      tmp=strrchr(optarg,':');
                      proxyhost=optarg;
                      if(tmp==NULL)
                      {
                          break;
                      }
                      if(tmp==optarg)
                      {
                          fprintf(stderr,"Error in option --proxy %s: Missing hostname.\n",optarg);
                          return 2;
                      }
                      if(tmp==optarg+strlen(optarg)-1)
                      {
                          fprintf(stderr,"Error in option --proxy %s Port number is missing.\n",optarg);
                          return 2;
                      }
                      *tmp='\0';
                      proxyport=atoi(tmp+1);break;
            case ':':
            case 'h':
            case '?': usage();return 2;break;
            case 'c': clients=atoi(optarg);break;
        }
    }

    if(optind==argc) {
        fprintf(stderr,"webbench: Missing URL!\n");
        usage();
        return 2;
    }
    return 0;
}