Commit | Line | Data |
---|---|---|
460b9539 | 1 | /* |
2 | * This file is part of DisOrder | |
5aff007d | 3 | * Copyright (C) 2004, 2007, 2008 Richard Kettlewell |
460b9539 | 4 | * |
e7eb3a27 | 5 | * This program is free software: you can redistribute it and/or modify |
460b9539 | 6 | * it under the terms of the GNU General Public License as published by |
e7eb3a27 | 7 | * the Free Software Foundation, either version 3 of the License, or |
460b9539 | 8 | * (at your option) any later version. |
e7eb3a27 RK |
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 | * | |
460b9539 | 15 | * You should have received a copy of the GNU General Public License |
e7eb3a27 | 16 | * along with this program. If not, see <http://www.gnu.org/licenses/>. |
460b9539 | 17 | */ |
132a5a4a RK |
18 | /** @file lib/snprintf.c |
19 | * @brief UTF-8 capable *snprintf workalikes | |
20 | */ | |
460b9539 | 21 | |
22 | #define NO_MEMORY_ALLOCATION | |
23 | /* because used from log.c */ | |
24 | ||
05b75f8d | 25 | #include "common.h" |
460b9539 | 26 | |
460b9539 | 27 | #include <stdarg.h> |
28 | #include <stddef.h> | |
29 | ||
30 | #include "printf.h" | |
31 | #include "sink.h" | |
32 | ||
598b07b7 RK |
33 | /** @brief A @ref sink that stores to a fixed buffer |
34 | * | |
35 | * If there is too much output, it is truncated. | |
36 | */ | |
460b9539 | 37 | struct fixedstr_sink { |
598b07b7 | 38 | /** @brief Base */ |
460b9539 | 39 | struct sink s; |
598b07b7 RK |
40 | |
41 | /** @brief Output buffer */ | |
460b9539 | 42 | char *buffer; |
598b07b7 RK |
43 | |
44 | /** @brief Bytes written so far */ | |
460b9539 | 45 | int nbytes; |
598b07b7 RK |
46 | |
47 | /** @brief Size of buffer */ | |
460b9539 | 48 | size_t size; |
49 | }; | |
50 | ||
51 | static int fixedstr_write(struct sink *f, const void *buffer, int nbytes) { | |
52 | struct fixedstr_sink *s = (struct fixedstr_sink *)f; | |
53 | int count; | |
54 | ||
55 | if((size_t)s->nbytes < s->size) { | |
56 | if((size_t)nbytes > s->size - s->nbytes) | |
57 | count = s->size - s->nbytes; | |
58 | else | |
59 | count = nbytes; | |
60 | memcpy(s->buffer + s->nbytes, buffer, count); | |
61 | } | |
62 | s->nbytes += nbytes; | |
63 | return 0; | |
64 | } | |
65 | ||
66 | int byte_vsnprintf(char buffer[], | |
67 | size_t bufsize, | |
68 | const char *fmt, | |
69 | va_list ap) { | |
70 | struct fixedstr_sink s; | |
71 | int n, m; | |
72 | ||
73 | /* We have to make a sink directly here, since we can't safely do memory | |
74 | * allocation here (we might be formatting the error message from a failed | |
75 | * memory allocation) */ | |
76 | s.s.write = fixedstr_write; | |
77 | s.buffer = buffer; | |
78 | s.nbytes = 0; | |
79 | s.size = bufsize; | |
80 | n = byte_vsinkprintf(&s.s, fmt, ap); | |
81 | if(bufsize) { | |
82 | /* add the null terminator (even if the printf failed) */ | |
83 | m = s.nbytes; | |
84 | if((size_t)m >= bufsize) m = bufsize - 1; | |
85 | buffer[m] = 0; | |
86 | } | |
87 | return n; | |
88 | } | |
89 | ||
90 | int byte_snprintf(char buffer[], size_t bufsize, const char *fmt, ...) { | |
91 | int n; | |
92 | va_list ap; | |
93 | ||
94 | va_start(ap, fmt); | |
95 | n = byte_vsnprintf(buffer, bufsize, fmt, ap); | |
96 | va_end(ap); | |
97 | return n; | |
98 | } | |
99 | ||
100 | /* | |
101 | Local Variables: | |
102 | c-basic-offset:2 | |
103 | comment-column:40 | |
104 | End: | |
105 | */ |