| Line |
Branch |
Exec |
Source |
| 1 |
|
|
/* |
| 2 |
|
|
* $Id: printbuf.c,v 1.5 2006/01/26 02:16:28 mclark Exp $ |
| 3 |
|
|
* |
| 4 |
|
|
* Copyright (c) 2004, 2005 Metaparadigm Pte. Ltd. |
| 5 |
|
|
* Michael Clark <michael@metaparadigm.com> |
| 6 |
|
|
* |
| 7 |
|
|
* This library is free software; you can redistribute it and/or modify |
| 8 |
|
|
* it under the terms of the MIT license. See COPYING for details. |
| 9 |
|
|
* |
| 10 |
|
|
* |
| 11 |
|
|
* Copyright (c) 2008-2009 Yahoo! Inc. All rights reserved. |
| 12 |
|
|
* The copyrights to the contents of this file are licensed under the MIT License |
| 13 |
|
|
* (https://www.opensource.org/licenses/mit-license.php) |
| 14 |
|
|
*/ |
| 15 |
|
|
|
| 16 |
|
|
#include "config.h" |
| 17 |
|
|
|
| 18 |
|
|
#include <errno.h> |
| 19 |
|
|
#include <limits.h> |
| 20 |
|
|
#include <stdio.h> |
| 21 |
|
|
#include <stdlib.h> |
| 22 |
|
|
#include <string.h> |
| 23 |
|
|
|
| 24 |
|
|
#ifdef HAVE_STDARG_H |
| 25 |
|
|
#include <stdarg.h> |
| 26 |
|
|
#else /* !HAVE_STDARG_H */ |
| 27 |
|
|
#error Not enough var arg support! |
| 28 |
|
|
#endif /* HAVE_STDARG_H */ |
| 29 |
|
|
|
| 30 |
|
|
#include "debug.h" |
| 31 |
|
|
#include "printbuf.h" |
| 32 |
|
|
#include "snprintf_compat.h" |
| 33 |
|
|
#include "vasprintf_compat.h" |
| 34 |
|
|
|
| 35 |
|
|
static int printbuf_extend(struct printbuf *p, int min_size); |
| 36 |
|
|
|
| 37 |
|
✗ |
struct printbuf *printbuf_new(void) |
| 38 |
|
|
{ |
| 39 |
|
|
struct printbuf *p; |
| 40 |
|
|
|
| 41 |
|
✗ |
p = (struct printbuf *)calloc(1, sizeof(struct printbuf)); |
| 42 |
|
✗ |
if (!p) |
| 43 |
|
|
return NULL; |
| 44 |
|
✗ |
p->size = 32; |
| 45 |
|
✗ |
p->bpos = 0; |
| 46 |
|
✗ |
if (!(p->buf = (char *)malloc(p->size))) |
| 47 |
|
|
{ |
| 48 |
|
✗ |
free(p); |
| 49 |
|
✗ |
return NULL; |
| 50 |
|
|
} |
| 51 |
|
✗ |
p->buf[0] = '\0'; |
| 52 |
|
✗ |
return p; |
| 53 |
|
|
} |
| 54 |
|
|
|
| 55 |
|
|
/** |
| 56 |
|
|
* Extend the buffer p so it has a size of at least min_size. |
| 57 |
|
|
* |
| 58 |
|
|
* If the current size is large enough, nothing is changed. |
| 59 |
|
|
* |
| 60 |
|
|
* If extension failed, errno is set to indicate the error. |
| 61 |
|
|
* |
| 62 |
|
|
* Note: this does not check the available space! The caller |
| 63 |
|
|
* is responsible for performing those calculations. |
| 64 |
|
|
*/ |
| 65 |
|
✗ |
static int printbuf_extend(struct printbuf *p, int min_size) |
| 66 |
|
|
{ |
| 67 |
|
|
char *t; |
| 68 |
|
|
int new_size; |
| 69 |
|
|
|
| 70 |
|
✗ |
if (p->size >= min_size) |
| 71 |
|
|
return 0; |
| 72 |
|
|
/* Prevent signed integer overflows with large buffers. */ |
| 73 |
|
✗ |
if (min_size > INT_MAX - 8) |
| 74 |
|
|
{ |
| 75 |
|
✗ |
errno = EFBIG; |
| 76 |
|
✗ |
return -1; |
| 77 |
|
|
} |
| 78 |
|
✗ |
if (p->size > INT_MAX / 2) |
| 79 |
|
✗ |
new_size = min_size + 8; |
| 80 |
|
|
else { |
| 81 |
|
✗ |
new_size = p->size * 2; |
| 82 |
|
✗ |
if (new_size < min_size + 8) |
| 83 |
|
✗ |
new_size = min_size + 8; |
| 84 |
|
|
} |
| 85 |
|
|
#ifdef PRINTBUF_DEBUG |
| 86 |
|
|
MC_DEBUG("printbuf_extend: realloc " |
| 87 |
|
|
"bpos=%d min_size=%d old_size=%d new_size=%d\n", |
| 88 |
|
|
p->bpos, min_size, p->size, new_size); |
| 89 |
|
|
#endif /* PRINTBUF_DEBUG */ |
| 90 |
|
✗ |
if (!(t = (char *)realloc(p->buf, new_size))) |
| 91 |
|
|
return -1; |
| 92 |
|
✗ |
p->size = new_size; |
| 93 |
|
✗ |
p->buf = t; |
| 94 |
|
✗ |
return 0; |
| 95 |
|
|
} |
| 96 |
|
|
|
| 97 |
|
✗ |
int printbuf_memappend(struct printbuf *p, const char *buf, int size) |
| 98 |
|
|
{ |
| 99 |
|
|
/* Prevent signed integer overflows with large buffers. */ |
| 100 |
|
✗ |
if (size < 0 || size > INT_MAX - p->bpos - 1) |
| 101 |
|
|
{ |
| 102 |
|
✗ |
errno = EFBIG; |
| 103 |
|
✗ |
return -1; |
| 104 |
|
|
} |
| 105 |
|
✗ |
if (p->size <= p->bpos + size + 1) |
| 106 |
|
|
{ |
| 107 |
|
✗ |
if (printbuf_extend(p, p->bpos + size + 1) < 0) |
| 108 |
|
|
return -1; |
| 109 |
|
|
} |
| 110 |
|
✗ |
memcpy(p->buf + p->bpos, buf, size); |
| 111 |
|
✗ |
p->bpos += size; |
| 112 |
|
✗ |
p->buf[p->bpos] = '\0'; |
| 113 |
|
✗ |
return size; |
| 114 |
|
|
} |
| 115 |
|
|
|
| 116 |
|
✗ |
int printbuf_memset(struct printbuf *pb, int offset, int charvalue, int len) |
| 117 |
|
|
{ |
| 118 |
|
|
int size_needed; |
| 119 |
|
|
|
| 120 |
|
✗ |
if (offset == -1) |
| 121 |
|
✗ |
offset = pb->bpos; |
| 122 |
|
|
/* Prevent signed integer overflows with large buffers. */ |
| 123 |
|
✗ |
if (len < 0 || offset < -1 || len > INT_MAX - offset) |
| 124 |
|
|
{ |
| 125 |
|
✗ |
errno = EFBIG; |
| 126 |
|
✗ |
return -1; |
| 127 |
|
|
} |
| 128 |
|
✗ |
size_needed = offset + len; |
| 129 |
|
✗ |
if (pb->size < size_needed) |
| 130 |
|
|
{ |
| 131 |
|
✗ |
if (printbuf_extend(pb, size_needed) < 0) |
| 132 |
|
|
return -1; |
| 133 |
|
|
} |
| 134 |
|
|
|
| 135 |
|
✗ |
if (pb->bpos < offset) |
| 136 |
|
✗ |
memset(pb->buf + pb->bpos, '\0', offset - pb->bpos); |
| 137 |
|
✗ |
memset(pb->buf + offset, charvalue, len); |
| 138 |
|
✗ |
if (pb->bpos < size_needed) |
| 139 |
|
✗ |
pb->bpos = size_needed; |
| 140 |
|
|
|
| 141 |
|
|
return 0; |
| 142 |
|
|
} |
| 143 |
|
|
|
| 144 |
|
✗ |
int sprintbuf(struct printbuf *p, const char *msg, ...) |
| 145 |
|
|
{ |
| 146 |
|
|
va_list ap; |
| 147 |
|
|
char *t; |
| 148 |
|
|
int size; |
| 149 |
|
|
char buf[128]; |
| 150 |
|
|
|
| 151 |
|
|
/* use stack buffer first */ |
| 152 |
|
✗ |
va_start(ap, msg); |
| 153 |
|
|
size = vsnprintf(buf, 128, msg, ap); |
| 154 |
|
✗ |
va_end(ap); |
| 155 |
|
|
/* if string is greater than stack buffer, then use dynamic string |
| 156 |
|
|
* with vasprintf. Note: some implementations of vsnprintf return -1 |
| 157 |
|
|
* if output is truncated whereas some return the number of bytes that |
| 158 |
|
|
* would have been written - this code handles both cases. |
| 159 |
|
|
*/ |
| 160 |
|
✗ |
if (size < 0 || size > 127) |
| 161 |
|
|
{ |
| 162 |
|
✗ |
va_start(ap, msg); |
| 163 |
|
✗ |
if ((size = vasprintf(&t, msg, ap)) < 0) |
| 164 |
|
|
{ |
| 165 |
|
✗ |
va_end(ap); |
| 166 |
|
✗ |
return -1; |
| 167 |
|
|
} |
| 168 |
|
✗ |
va_end(ap); |
| 169 |
|
✗ |
size = printbuf_memappend(p, t, size); |
| 170 |
|
✗ |
free(t); |
| 171 |
|
|
} |
| 172 |
|
|
else |
| 173 |
|
|
{ |
| 174 |
|
✗ |
size = printbuf_memappend(p, buf, size); |
| 175 |
|
|
} |
| 176 |
|
|
return size; |
| 177 |
|
|
} |
| 178 |
|
|
|
| 179 |
|
✗ |
void printbuf_reset(struct printbuf *p) |
| 180 |
|
|
{ |
| 181 |
|
✗ |
p->buf[0] = '\0'; |
| 182 |
|
✗ |
p->bpos = 0; |
| 183 |
|
|
} |
| 184 |
|
|
|
| 185 |
|
✗ |
void printbuf_free(struct printbuf *p) |
| 186 |
|
|
{ |
| 187 |
|
✗ |
if (p) |
| 188 |
|
|
{ |
| 189 |
|
✗ |
free(p->buf); |
| 190 |
|
✗ |
free(p); |
| 191 |
|
|
} |
| 192 |
|
|
} |
| 193 |
|
|
|