| 1 | /* |
| 2 | * This file is part of DisOrder |
| 3 | * Copyright (C) 2004, 2007, 2008 Richard Kettlewell |
| 4 | * |
| 5 | * This program is free software: you can redistribute it and/or modify |
| 6 | * it under the terms of the GNU General Public License as published by |
| 7 | * the Free Software Foundation, either version 3 of the License, or |
| 8 | * (at your option) any later version. |
| 9 | * |
| 10 | * This program is distributed in the hope that it will be useful, |
| 11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 13 | * GNU General Public License for more details. |
| 14 | * |
| 15 | * You should have received a copy of the GNU General Public License |
| 16 | * along with this program. If not, see <http://www.gnu.org/licenses/>. |
| 17 | */ |
| 18 | /** @file lib/snprintf.c |
| 19 | * @brief UTF-8 capable *snprintf workalikes |
| 20 | */ |
| 21 | |
| 22 | #define NO_MEMORY_ALLOCATION |
| 23 | /* because used from log.c */ |
| 24 | |
| 25 | #include "common.h" |
| 26 | |
| 27 | #include <stdarg.h> |
| 28 | #include <stddef.h> |
| 29 | |
| 30 | #include "printf.h" |
| 31 | #include "sink.h" |
| 32 | |
| 33 | struct fixedstr_sink { |
| 34 | struct sink s; |
| 35 | char *buffer; |
| 36 | int nbytes; |
| 37 | size_t size; |
| 38 | }; |
| 39 | |
| 40 | static int fixedstr_write(struct sink *f, const void *buffer, int nbytes) { |
| 41 | struct fixedstr_sink *s = (struct fixedstr_sink *)f; |
| 42 | int count; |
| 43 | |
| 44 | if((size_t)s->nbytes < s->size) { |
| 45 | if((size_t)nbytes > s->size - s->nbytes) |
| 46 | count = s->size - s->nbytes; |
| 47 | else |
| 48 | count = nbytes; |
| 49 | memcpy(s->buffer + s->nbytes, buffer, count); |
| 50 | } |
| 51 | s->nbytes += nbytes; |
| 52 | return 0; |
| 53 | } |
| 54 | |
| 55 | int byte_vsnprintf(char buffer[], |
| 56 | size_t bufsize, |
| 57 | const char *fmt, |
| 58 | va_list ap) { |
| 59 | struct fixedstr_sink s; |
| 60 | int n, m; |
| 61 | |
| 62 | /* We have to make a sink directly here, since we can't safely do memory |
| 63 | * allocation here (we might be formatting the error message from a failed |
| 64 | * memory allocation) */ |
| 65 | s.s.write = fixedstr_write; |
| 66 | s.buffer = buffer; |
| 67 | s.nbytes = 0; |
| 68 | s.size = bufsize; |
| 69 | n = byte_vsinkprintf(&s.s, fmt, ap); |
| 70 | if(bufsize) { |
| 71 | /* add the null terminator (even if the printf failed) */ |
| 72 | m = s.nbytes; |
| 73 | if((size_t)m >= bufsize) m = bufsize - 1; |
| 74 | buffer[m] = 0; |
| 75 | } |
| 76 | return n; |
| 77 | } |
| 78 | |
| 79 | int byte_snprintf(char buffer[], size_t bufsize, const char *fmt, ...) { |
| 80 | int n; |
| 81 | va_list ap; |
| 82 | |
| 83 | va_start(ap, fmt); |
| 84 | n = byte_vsnprintf(buffer, bufsize, fmt, ap); |
| 85 | va_end(ap); |
| 86 | return n; |
| 87 | } |
| 88 | |
| 89 | /* |
| 90 | Local Variables: |
| 91 | c-basic-offset:2 |
| 92 | comment-column:40 |
| 93 | End: |
| 94 | */ |