nasmlib/asprintf.c

   1 /* ----------------------------------------------------------------------- *
   2  *
   3  *   Copyright 1996-2018 The NASM Authors - All Rights Reserved
   4  *   See the file AUTHORS included with the NASM distribution for
   5  *   the specific copyright holders.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following
   9  *   conditions are met:
  10  *
  11  *   * Redistributions of source code must retain the above copyright
  12  *     notice, this list of conditions and the following disclaimer.
  13  *   * Redistributions in binary form must reproduce the above
  14  *     copyright notice, this list of conditions and the following
  15  *     disclaimer in the documentation and/or other materials provided
  16  *     with the distribution.
  17  *
  18  *     THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
  19  *     CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
  20  *     INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  21  *     MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  22  *     DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  23  *     CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24  *     SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  25  *     NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  26  *     LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  *     HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  28  *     CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  29  *     OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
  30  *     EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31  *
  32  * ----------------------------------------------------------------------- */
  33
  34 #include "compiler.h"
  35 #include "nasmlib.h"
  36 #include "alloc.h"
  37
  38 /*
  39  * nasm_[v]asprintf() are variants of the semi-standard [v]asprintf()
  40  * functions, except that we return the pointer instead of a count.
  41  * The length of the string (with or without the final NUL) is available
  42  * by calling nasm_last_string_{len,size}() afterwards.
  43  *
  44  * nasm_[v]axprintf() are similar, but allocates a user-defined amount
  45  * of storage before the string, and returns a pointer to the
  46  * allocated buffer. The size of that area is not included in the value
  47  * returned by nasm_last_string_size().
  48  */
  49
  50 void *nasm_vaxprintf(size_t extra, const char *fmt, va_list ap)
  51 {
  52     char *strp;
  53     va_list xap;
  54     size_t bytes;
  55
  56     va_copy(xap, ap);
  57     bytes = vsnprintf(NULL, 0, fmt, xap) + 1;
  58     _nasm_last_string_size = bytes;
  59     va_end(xap);
  60
  61     strp = nasm_malloc(extra+bytes);
  62     memset(strp, 0, extra);
  63     vsnprintf(strp+extra, bytes, fmt, ap);
  64     return strp;
  65 }
  66
  67 char *nasm_vasprintf(const char *fmt, va_list ap)
  68 {
  69     return nasm_vaxprintf(0, fmt, ap);
  70 }
  71
  72 void *nasm_axprintf(size_t extra, const char *fmt, ...)
  73 {
  74     va_list ap;
  75     void *strp;
  76
  77     va_start(ap, fmt);
  78     strp = nasm_vaxprintf(extra, fmt, ap);
  79     va_end(ap);
  80
  81     return strp;
  82 }
  83
  84 char *nasm_asprintf(const char *fmt, ...)
  85 {
  86     va_list ap;
  87     char *strp;
  88
  89     va_start(ap, fmt);
  90     strp = nasm_vaxprintf(0, fmt, ap);
  91     va_end(ap);
  92
  93     return strp;
  94 }