首页 > 代码库 > WebBench源码分析与心得

WebBench源码分析与心得

源码:https://github.com/EZLippi/WebBench
 
关键全局变量:
  speed 成功次数
  failed 失败次数
  bytes 接收字节数
  benchtime 执行时长(秒)
 
 
build_request() 函数:生成请求字符串。
一)请求第一行,如:GET /mall HTTP/1.0
  1. 根据命令行参数中的请求方法种类,把"GET"、"HEAD"、"OPTIONS"或"TRACE"相应复制到 request[]。
  2. 在request中,接下来添加一个空格。
  3. 如果命令行参数url不包含"://",则它不是个有效的URL,报错、退出。
  4. 如果url长度超过1500个字符,则太长,退出。
  5. 如果url开头不是"http://",则报错。
  6. 如果没有设置代理,则
    1. 从url中取主机名和端口号,分别填至host和proxyport
    2. 取路径部分,添加至 request
  7. 如果使用HTTP 1.0 或 1.1,则将 " HTTP/1.0" 或 " HTTP/1.1" 添加至 request。(注意前导空格)
    1. 如果使用HTTP 0.9,则无需添加此协议部分。
  8. 换行:添加 "\r\n" 至 request
二)请求后续行:
  1. 若HTTP版本在1.0及以上,则向request添加:"User-Agent: WebBench 1.5\r\n"。注意有换行。
  2. 若不使用代理,且版本在1.0以上,则向request添加主机头("Host: " + host)并换行。
  3. 若不使用代理,且强制刷新,则向request添加缓存控制(不缓存):"Pragma: no-cache\r\n"。
  4. 若版本在1.1及以上,则添加:"Connection: close\r\n"。
  5. 若版本在1.0及以上,则再添加空行:"\r\n"。
  6. 结束。■
 
 
bench()函数:创建各子进程以发送HTTP请求,并收集统计数据。
  1. 检测目标服务器(URL中的host或者代理)是否可以连接:使用connect()。
  2. 创建一对管道:使用pipe()。
  3. 创建指定个数clients的子进程:使用fork()。
  4. 在子进程中:
    1. 调用 benchcore() 执行测试。
    2. 向管道写入三项数据:速度、失败次数、字节数。
    3. 结束。■
[在父进程中]
  1. 循环 clients次,反复执行:
    1. 从管道读入下一个子进程的三项数据:速度、失败次数、字节数。
    2. 若读入项数少于3,说明已无数据可读、或其余子进程已提前结束,跳出循环。
    3. 累计三项数据
  2. 打印统计结果:
    1. 速度:分两项
      1. 总请求数 / 时间 = (speed+failed)/,单位:次/分
      2. 总字节数 / 时间,单位:字节/秒
    2. 成功次数:speed
    3. 失败次数:failed
  3. 结束。■
 
benchcore()函数:(在子进程中)反复发送HTTP请求。
  1. 使用SIGALRM安装定时器:时长来自命令行参数。
  2. 循环 直到定时器超时:
    1. 连接服务器(或代理服务器),得fd;若连接失败,则失败次数 failed 增1,跳至下一轮循环
    2. 向fd写入请求串;若写入失败,则 failed 增1,关闭fd,跳至下一轮循环
    3. 若采用HTTP/0.9协议,则停止发送:使用shutdown();
    4. 若要等服务器响应,则
      1. 反复从fd读出响应数据,直到读入0字节(表示读完),累计字节数至 bytes。
      2. 若出错,则 failed 增1,关闭fd,跳至下一轮循环
    5. 关闭fd。若出错,则 failed 增1。
    6. 累加成功次数 speed。
  3. 结束。■
 
 
学习到的Linux API/C库函数:
getopt、getopt_long() 命令行参数解析,详见《使用GETOPT()处理命令行参数》
strstr() 在字符串中搜索另一个字符串
strchr() 从串头开始搜索单个字符
srrrchr() 从串尾开始搜索单个字符
strncasecmp() 串比较,大小写不敏感
strcspn(s1, s2) 返回n,表示从s1串头开始连续有n个字符都不在字符串s2中。例如 strcspn("My 1984", "0123456789") == 3。
fdopen、fprintf、fscanf、setvbuf 等流式库I/O函数。
fork、pipe 与进程有关的函数。
shutdown() 与 close() 的区别。
 
问题:
1、为什么sleep()能让子进程快点?
    /* fork children */    for (i = 0; i < clients; i++) {        pid = fork();        if (pid <= (pid_t) 0) {            /* child process or error*/            sleep(1); /* make children faster */            break;        }    }

 

杂项:
主机名最大长度:64
命令行最大长度:2048
/* 定义所在文件:/usr/include/asm/param.h */
#define MAXHOSTNAMELEN 64 /* max length of hostname */
#define COMMAND_LINE_SIZE 2048
 
 
 

WebBench源码分析与心得