重新分配后 vsnprintf 失败

Question

我得到这段代码（如下），它基本上是一个 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, 0, format, argp);
    if(*str)
    {
        *str = realloc(*str, len + 1);
    }
    else
    {
        *str = malloc(len + 1);
        **str = 0;
    }
    if(!*str)
    {
        return false;
    }
    vsnprintf(*str, len, format, argp);
    va_end(argp);
    (*str)[len] = 0;
    return true;
}

这就是您使用函数的方式。

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

编辑：评论来自这里

Answer 1

#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, 0, 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 arguments
    // but in this case it actually doesn't *use* them -- IT DOES consume them!!!!!

    va_end(argp);

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

    if(!*str) return -1;

    va_start(argp, format);
    vsnprintf(*str, len, format, 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() returns a signed 类型（-1 也是一个有效的 return 值：也检查一下）
realloc() 足够聪明，可以理解第一次调用（当第一个参数为 NULL 时），无需单独处理

Answer 2

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

我主要是用这个函数在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);
}

编辑：我现在有一个可用的函数。原来你确实需要 va_end.

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, 0, format, argp);
    va_end(argp);
    int len_str = 0;
    if(*str)
    {
        len_str = strlen(*str);
    }
    *str = realloc(*str, len_str + len_app + 1);
    
    // append
    va_start(argp);
    if(*str)
    {
        if(vsnprintf(&(*str)[len_str], len_app + 1, format, 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, "Hello world!!%d\n", 231);
success &= append_str(&str, "Hello world!!%d\n", 231);
success &= append_str(&str, "Hello world!!%d\n", 231);
success &= append_str(&str, "Hello world!!%d\n", 231);
if(success)
{
    printf("%s\n", str);
}
else
{
    printf("failed to append string.\n");
}
return 0;

重新分配后 vsnprintf 失败

vsnprintf fails after realloc

c

variadic-functions

realloc