about summary refs log tree commit diff
path: root/lib/util/vasprintf.c
blob: a4229c702fe20548410ba94f8400462a0a4ef6d5 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#define _GNU_SOURCE
   /* Due to conditional compilation, this is GNU source only if the C library
      is GNU.
   */
#include <stdlib.h>
#include <string.h>

#include "nstring.h"

#ifdef __GNUC__
  #define HAVE_VASPRINTF 1
#else
  #define HAVE_VASPRINTF 0
#endif

void
vasprintfN(const char ** const resultP,
           const char *  const format,
           va_list             varargs) {

    char * result;

#if HAVE_VASPRINTF
    vasprintf(&result, format, varargs);

    if (result == NULL)
        *resultP = strsol;
    else
        *resultP = result;
#else
    /* We have a big compromise here.  To do this right, without a
       huge amount of work, we need to go through the variable
       arguments twice: once to determine how much memory to allocate,
       and once to format the string.  On some machines, you can
       simply make two copies of the va_list variable in normal C
       fashion, but on others you need va_copy, which is a relatively
       recent invention.  In particular, the simple va_list copy
       failed on an AMD64 Gcc Linux system in March 2006.  

       So instead, we just allocate 4K and truncate or waste as
       necessary.
    */
    size_t const allocSize = 4096;
    result = malloc(allocSize);
    
    if (result == NULL)
        *resultP = strsol;
    else {
        size_t realLen;

        vsnprintfN(result, allocSize, format, varargs, &realLen);
        
        if (realLen >= allocSize)
            strcpy(result + allocSize - 15, "<<<TRUNCATED");

        *resultP = result;
    }
#endif
}