每个字符串参数的长度

debugcn 发表于 Dev

phi

最近，我开始对这些stdarg.h函数有所了解，因为我想拥有一些类似的东西，printf但是我不想将其写为控制台，而是以字符串形式返回。

到目前为止，这是我想出的：

char *write(const char *format, ...)
{
    // init
    va_list arg;
    char *string;

    va_start (arg, format);
    vsprintf (string, format, arg);

    // done
    va_end (arg);
    return string;
}

现在的问题是，它string没有保留内存，在那儿我需要修复此功能的方法的帮助，因为我还没有找到任何解决方案。

提前致谢

卡米尔库克

使用snprintf(NULL, 0检查多久缓冲你的需要。然后分配内存。然后打印到字符串。

char *my_write(const char *format, ...) {
    va_list va;
    va_start(va, format);

    // remember to have a separate va_list for each v*print function
    va_list va2;
    va_copy(va2, va);
    const int len = vsnprintf(NULL, 0, format, va2);
    va_end(va2);

    char *string = malloc((len + 1) * sizeof(*string));
    if (string != NULL) {
       vsprintf(string, format, va);
    }
    va_end(va);

    return string;
}

正如@IanAbbott在注释中所建议的，您可以调用va_start两次，这似乎很好地简化了代码：

char *my_write(const char *format, ...) {
    va_list va;

    va_start(va, format);
    const int len = vsnprintf(NULL, 0, format, va);
    va_end(va);

    char *string = malloc((len + 1) * sizeof(*string));
    if (string == NULL) {
       return NULL;
    }

    va_start(va, format);
    vsprintf(string, format, va);
    va_end(va);

    return string;
}

在带有glibc的平台上，您也可以使用vasprintf。请注意，该名称write已被posix write（）函数使用，我建议使用其他名称。有了vasprintfGNU扩展，它就变成了：

#define _GNU_SOURCE
#include <stdio.h>
char *write2(const char *format, ...) {
    va_list va;
    va_start(va, format);
    char *string;
    const int err = vasprintf(&string, format, va);
    va_end(va);
    if (err == -1) {
          return NULL;
    }
    return string;
}

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。