Fix memory management issues with expanded %include
[nasm/avx512.git] / lib / vsnprintf.c
blob976b0eac95160bfa80d17b3daccdbc65c7b5dbfb
1 /*
2 * vsnprintf()
4 * Poor substitute for a real vsnprintf() function for systems
5 * that don't have them...
6 */
8 #include "compiler.h"
10 #include <stdio.h>
11 #include <stdlib.h>
12 #include <stdarg.h>
13 #include <string.h>
15 #include "nasmlib.h"
17 extern efunc nasm_malloc_error;
19 #define BUFFER_SIZE 65536 /* Bigger than any string we might print... */
21 static char snprintf_buffer[BUFFER_SIZE];
23 int vsnprintf(char *str, size_t size, const char *format, va_list ap)
25 int rv, bytes;
27 if (size > BUFFER_SIZE) {
28 nasm_malloc_error(ERR_PANIC|ERR_NOFILE,
29 "snprintf: size (%d) > BUFFER_SIZE (%d)",
30 size, BUFFER_SIZE);
31 size = BUFFER_SIZE;
34 rv = vsprintf(snprintf_buffer, format, ap);
35 if (rv >= BUFFER_SIZE) {
36 nasm_malloc_error(ERR_PANIC|ERR_NOFILE,
37 "snprintf buffer overflow");
40 if (size > 0) {
41 if ((size_t)rv < size-1)
42 bytes = rv;
43 else
44 bytes = size-1;
46 memcpy(str, snprintf_buffer, bytes);
47 str[bytes] = '\0';
50 return rv;