WebBench源码分析与心得

Posted 2020-08-22 xxfcz

源码：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

二）请求后续行：

9. 若HTTP版本在1.0及以上，则向request添加："User-Agent: WebBench 1.5\\r\\n"。注意有换行。
10. 若不使用代理，且版本在1.0以上，则向request添加主机头（"Host: " + host）并换行。
11. 若不使用代理，且强制刷新，则向request添加缓存控制（不缓存）："Pragma: no-cache\\r\\n"。
12. 若版本在1.1及以上，则添加："Connection: close\\r\\n"。
13. 若版本在1.0及以上，则再添加空行："\\r\\n"。
14. 结束。■

 

 

bench()函数：创建各子进程以发送HTTP请求，并收集统计数据。

1. 检测目标服务器（URL中的host或者代理）是否可以连接：使用connect()。
2. 创建一对管道：使用pipe()。
3. 创建指定个数clients的子进程：使用fork()。
4. 在子进程中：
1. 调用 benchcore() 执行测试。
2. 向管道写入三项数据：速度、失败次数、字节数。
3. 结束。■

[在父进程中]

5. 循环 clients次，反复执行：
1. 从管道读入下一个子进程的三项数据：速度、失败次数、字节数。
2. 若读入项数少于3，说明已无数据可读、或其余子进程已提前结束，跳出循环。
3. 累计三项数据
6. 打印统计结果：
1. 速度：分两项
1. 总请求数 / 时间 = (speed+failed)/，单位：次/分
2. 总字节数 / 时间，单位：字节/秒
2. 成功次数：speed
3. 失败次数：failed
7. 结束。■

 

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