重新分配后 vsnprintf 失败

问题描述

我得到的这段代码（如下）基本上是一个 snprintf 函数，但速度有点慢，这意味着它也使用 malloc/realloc 处理缓冲区大小。

代码如何工作，直到 realloc 更改地址（当字符串随着时间的推移变得越来越大时）......如果发生这种情况，输出将完全错误。

bool write_str(char **str,const char *format,...)
{
    // va arg stuff
    va_list argp;
    va_start(argp,format);
    // remember to have a separate va_list for each v*print function,because v*print consumes the function
    // but in this case it actually doesn't need it
    size_t len = vsnprintf(0,format,argp);
    if(*str)
    {
        *str = realloc(*str,len + 1);
    }
    else
    {
        *str = malloc(len + 1);
        **str = 0;
    }
    if(!*str)
    {
        return false;
    }
    vsnprintf(*str,len,argp);
    va_end(argp);
    (*str)[len] = 0;
    return true;
}

这就是您使用该功能的方式。

char *str = 0;
write_str(&str,"That is how %d use it\n",2);

编辑：评论来自这里Length of each string argument

解决方法

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <stdarg.h>

int write_str(char **str,const char *format,...)
{
    // va arg stuff
    int len;
    va_list argp;

    va_start(argp,format);

    len = vsnprintf(NULL,format,argp);
    // Remember to read the comments (or remove them)
    // remember to have a separate va_list for each v*print function
    //,because v*print consumes the function
    // but in this case it actually doesn't need it

    va_end(argp);

    *str = realloc(*str,len + 1);

    if(!*str) return -1;

    va_start(argp,format);
    vsnprintf(*str,len,argp);
    va_end(argp);

    return 0;
}

int main(void)
{

char *str = NULL;
write_str(&str,"That is how %d use it\n",2);
puts(str);

write_str(&str,"That is how again %d use it\n",666);
puts(str);

return 0;
}

一些注意事项：

argp 被 vsnprintf() 消耗；如果使用两次，则需要复制或重新初始化它
vsnprintf() 返回一个 signed 类型（-1 也是一个有效的返回值：也请检查一下）
realloc() 足够聪明，可以理解第一次调用（当第一个参数为 NULL 时），无需单独处理

终于找到了导致我的问题的原因。有点傻——应该把我真正用过的代码贴出来。

我主要是用这个函数将一个字符串重复追加到itslef。但问题是，realloc 可以改变字符串的地址，因此传递的地址不再相同，导致垃圾输出。

char *str = 0;
write_str(&str,"string");
for(int i = 0; i < 100; i++)
{
    write_str(&str,"%s%s",str,"string"); // the right str argument may change address,don't use it like that!
    printf("%s\n",str);
}

编辑：我现在得到了一个有效的函数。（还没有彻底测试过，但看起来很有希望。等我测试得更彻底了，我会调整这个说法）

bool append_str(char **str,char *format,...)
{
    if(!str || !format) return false;
    
    bool result = false;
    va_list argp;
    va_start(argp,format);
    
    // get memory
    int len_app = vsnprintf(0,argp);
    int len_str = 0;
    if(*str)
    {
        len_str = strlen(*str);
    }
    *str = realloc(*str,len_str + len_app + 1);
    
    // append
    if(*str)
    {
        if(vsnprintf(&(*str)[len_str],len_app + 1,argp) >= 0)
        {
            // successful
            result = true;
        }
        else
        {
            // encoding error
            free(*str);
            *str = 0;
        }
    }
    va_end(argp);
    
    return result;
}

以及如何使用它的示例：

char *str = 0;
bool success = true;
success &= append_str(&str,"Hello world!!%d\n",231);
success &= append_str(&str,231);
if(success)
{
    printf("%s\n",str);
}
else
{
    printf("failed to append string.\n");
}
return 0;

c realloc variadic-functions