mirror of
https://github.com/cesanta/mongoose.git
synced 2026-10-26 12:14:56 +00:00
Remove dangerous mg_pfn_realloc(). Add %q printf specifier
This commit is contained in:
1 parent
24cb9467c0
commit
2ba3630de9
8 files changed
+65
-96
No files matched your search
@@ -82,20 +82,6 @@ void mg_pfn_iobuf(char ch, void *param) {
|
||||
}
|
||||
}
|
||||
|
||||
// We don't use realloc() in mongoose, so resort to inefficient calloc
|
||||
// Every new character reallocates the whole string
|
||||
void mg_pfn_realloc(char ch, void *param) {
|
||||
char *s, *buf = *(char **) param;
|
||||
size_t len = buf == NULL ? 0 : strlen(buf);
|
||||
if ((s = (char *) calloc(1, len + 2)) != NULL) {
|
||||
if (buf != NULL) memcpy(s, buf, len);
|
||||
s[len] = ch;
|
||||
s[len + 1] = '\0';
|
||||
free(buf);
|
||||
*(char **) param = s;
|
||||
}
|
||||
}
|
||||
|
||||
size_t mg_vsnprintf(char *buf, size_t len, const char *fmt, va_list *ap) {
|
||||
struct mg_iobuf io = {(uint8_t *) buf, len, 0, 0};
|
||||
size_t n = mg_vrprintf(mg_putchar_iobuf_static, &io, fmt, ap);
|
||||
@@ -285,8 +271,7 @@ static char mg_escape(int c) {
|
||||
|
||||
static size_t qcpy(void (*out)(char, void *), void *ptr, char *buf,
|
||||
size_t len) {
|
||||
size_t i = 0, extra = 2;
|
||||
out('"', ptr);
|
||||
size_t i = 0, extra = 0;
|
||||
for (i = 0; i < len && buf[i] != '\0'; i++) {
|
||||
char c = mg_escape(buf[i]);
|
||||
if (c) {
|
||||
@@ -295,10 +280,18 @@ static size_t qcpy(void (*out)(char, void *), void *ptr, char *buf,
|
||||
out(buf[i], ptr);
|
||||
}
|
||||
}
|
||||
out('"', ptr);
|
||||
return i + extra;
|
||||
}
|
||||
|
||||
static size_t Qcpy(void (*out)(char, void *), void *ptr, char *buf,
|
||||
size_t len) {
|
||||
size_t n = 2;
|
||||
out('"', ptr);
|
||||
n += qcpy(out, ptr, buf, len);
|
||||
out('"', ptr);
|
||||
return n;
|
||||
}
|
||||
|
||||
static size_t bcpy(void (*out)(char, void *), void *ptr, uint8_t *buf,
|
||||
size_t len) {
|
||||
size_t i, n = 0;
|
||||
@@ -395,10 +388,11 @@ size_t mg_vrprintf(void (*out)(char, void *), void *param, const char *fmt,
|
||||
size_t len = (size_t) va_arg(*ap, int);
|
||||
uint8_t *buf = va_arg(*ap, uint8_t *);
|
||||
n += bcpy(out, param, buf, len);
|
||||
} else if (c == 's' || c == 'Q') {
|
||||
} else if (c == 's' || c == 'Q' || c == 'q') {
|
||||
char *p = va_arg(*ap, char *);
|
||||
size_t (*f)(void (*)(char, void *), void *, char *, size_t) =
|
||||
c == 's' ? scpy : qcpy;
|
||||
size_t (*f)(void (*)(char, void *), void *, char *, size_t) = scpy;
|
||||
if (c == 'Q') f = Qcpy;
|
||||
if (c == 'q') f = qcpy;
|
||||
if (pr == ~0U) pr = p == NULL ? 0 : strlen(p);
|
||||
for (j = 0; !minus && pr < w && j + pr < w; j++)
|
||||
n += f(out, param, &pad, 1);
|
||||
|
||||
@@ -5,8 +5,7 @@
|
||||
|
||||
typedef void (*mg_pfn_t)(char, void *); // Custom putchar
|
||||
typedef size_t (*mg_pm_t)(mg_pfn_t, void *, va_list *); // %M printer
|
||||
void mg_pfn_realloc(char ch, void *param); // Print to malloced str
|
||||
void mg_pfn_iobuf(char ch, void *param); // Print to iobuf
|
||||
void mg_pfn_iobuf(char ch, void *param); // iobuf printer
|
||||
|
||||
size_t mg_lld(char *buf, int64_t val, bool is_signed, bool is_hex);
|
||||
double mg_atod(const char *buf, int len, int *numlen);
|
||||
|
||||
Reference in new issue
Block a user