最近在写一个压力测试的程序。由于需要向服务器发送协议,拼接了几次字符串后觉得代码太臃肿了,而且容易出错。后来想到了C的printf函数。于是想能不能写一个类似printf的函数,将输入的参数格式化成自己想要的格式?完成后的代码如下:
//格式化字符串
//解析fmt 将fmt中的%s用对应的字符串替代
char *FormatString(const char *fmt,...)
{
char *data = (char *)malloc(strlen(fmt) + 1);
char *p;
va_list va;
int i = 0;
va_start(va, fmt);
memset(data, '\0', strlen(data));
while(*fmt)
{
if(*fmt=='%')
{
fmt++;
switch(*fmt)
{
case 's':
p = va_arg(va, char *);
data = (char *)realloc(data ,strlen(data) + strlen(fmt) + strlen(p) + 1); //add
while(*p)
{
data[i++]=*p;
p++;
}
break;
}
}
else
{
data[i++] = *fmt;
}
fmt++;
}
data[i] = '\0';
va_end(va);
return data;
}
样例输入
FormatString("%s%s","hello world","Test");
FormatString("{%s:%s}","hello world","Test");
此代码在FreeBSD上测试通过
如果各位有更好的方法,希望不吝赐教...