bprint-test: avoid z modifier, mingw fails with it
[ffmpeg.git] / libavutil / bprint.c
1 /*
2  * Copyright (c) 2012 Nicolas George
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #include <stdarg.h>
22 #include <stdio.h>
23 #include <string.h>
24 #include "bprint.h"
25 #include "common.h"
26 #include "error.h"
27 #include "mem.h"
28
29 #define av_bprint_room(buf) ((buf)->size - FFMIN((buf)->len, (buf)->size))
30 #define av_bprint_is_allocated(buf) ((buf)->str != (buf)->reserved_internal_buffer)
31
32 static int av_bprint_alloc(AVBPrint *buf, unsigned room)
33 {
34     char *old_str, *new_str;
35     unsigned min_size, new_size;
36
37     if (buf->size == buf->size_max)
38         return AVERROR(EIO);
39     if (!av_bprint_is_complete(buf))
40         return AVERROR_INVALIDDATA; /* it is already truncated anyway */
41     min_size = buf->len + 1 + FFMIN(UINT_MAX - buf->len - 1, room);
42     new_size = buf->size > buf->size_max / 2 ? buf->size_max : buf->size * 2;
43     if (new_size < min_size)
44         new_size = FFMIN(buf->size_max, min_size);
45     old_str = av_bprint_is_allocated(buf) ? buf->str : NULL;
46     new_str = av_realloc(old_str, new_size);
47     if (!new_str)
48         return AVERROR(ENOMEM);
49     if (!old_str)
50         memcpy(new_str, buf->str, buf->len + 1);
51     buf->str  = new_str;
52     buf->size = new_size;
53     return 0;
54 }
55
56 static void av_bprint_grow(AVBPrint *buf, unsigned extra_len)
57 {
58     /* arbitrary margin to avoid small overflows */
59     extra_len = FFMIN(extra_len, UINT_MAX - 5 - buf->len);
60     buf->len += extra_len;
61     if (buf->size)
62         buf->str[FFMIN(buf->len, buf->size - 1)] = 0;
63 }
64
65 void av_bprint_init(AVBPrint *buf, unsigned size_init, unsigned size_max)
66 {
67     unsigned size_auto = (char *)buf + sizeof(*buf) -
68                          buf->reserved_internal_buffer;
69
70     if (size_max == 1)
71         size_max = size_auto;
72     buf->str      = buf->reserved_internal_buffer;
73     buf->len      = 0;
74     buf->size     = FFMIN(size_auto, size_max);
75     buf->size_max = size_max;
76     *buf->str = 0;
77     if (size_init > buf->size)
78         av_bprint_alloc(buf, size_init - 1);
79 }
80
81 void av_bprint_init_for_buffer(AVBPrint *buf, char *buffer, unsigned size)
82 {
83     buf->str      = buffer;
84     buf->len      = 0;
85     buf->size     = size;
86     buf->size_max = size;
87     *buf->str = 0;
88 }
89
90 void av_bprintf(AVBPrint *buf, const char *fmt, ...)
91 {
92     unsigned room;
93     char *dst;
94     va_list vl;
95     int extra_len;
96
97     while (1) {
98         room = av_bprint_room(buf);
99         dst = room ? buf->str + buf->len : NULL;
100         va_start(vl, fmt);
101         extra_len = vsnprintf(dst, room, fmt, vl);
102         va_end(vl);
103         if (extra_len <= 0)
104             return;
105         if (extra_len < room)
106             break;
107         if (av_bprint_alloc(buf, extra_len))
108             break;
109     }
110     av_bprint_grow(buf, extra_len);
111 }
112
113 void av_bprint_chars(AVBPrint *buf, char c, unsigned n)
114 {
115     unsigned room, real_n;
116
117     while (1) {
118         room = av_bprint_room(buf);
119         if (n < room)
120             break;
121         if (av_bprint_alloc(buf, n))
122             break;
123     }
124     if (room) {
125         real_n = FFMIN(n, room - 1);
126         memset(buf->str + buf->len, c, real_n);
127     }
128     av_bprint_grow(buf, n);
129 }
130
131 void av_bprint_clear(AVBPrint *buf)
132 {
133     if (buf->len) {
134         *buf->str = 0;
135         buf->len  = 0;
136     }
137 }
138
139 int av_bprint_finalize(AVBPrint *buf, char **ret_str)
140 {
141     unsigned real_size = FFMIN(buf->len + 1, buf->size);
142     char *str;
143     int ret = 0;
144
145     if (ret_str) {
146         if (av_bprint_is_allocated(buf)) {
147             str = av_realloc(buf->str, real_size);
148             if (!str)
149                 str = buf->str;
150             buf->str = NULL;
151         } else {
152             str = av_malloc(real_size);
153             if (str)
154                 memcpy(str, buf->str, real_size);
155             else
156                 ret = AVERROR(ENOMEM);
157         }
158         *ret_str = str;
159     } else {
160         if (av_bprint_is_allocated(buf))
161             av_freep(&buf->str);
162     }
163     buf->size = real_size;
164     return ret;
165 }
166
167 #ifdef TEST
168
169 #undef printf
170
171 static void bprint_pascal(AVBPrint *b, unsigned size)
172 {
173     unsigned p[size + 1], i, j;
174
175     p[0] = 1;
176     av_bprintf(b, "%8d\n", 1);
177     for (i = 1; i <= size; i++) {
178         p[i] = 1;
179         for (j = i - 1; j > 0; j--)
180             p[j] = p[j] + p[j - 1];
181         for (j = 0; j <= i; j++)
182             av_bprintf(b, "%8d", p[j]);
183         av_bprintf(b, "\n");
184     }
185 }
186
187 int main(void)
188 {
189     AVBPrint b;
190     char buf[256];
191
192     av_bprint_init(&b, 0, -1);
193     bprint_pascal(&b, 5);
194     printf("Short text in unlimited buffer: %u/%u\n", (unsigned)strlen(b.str), b.len);
195     printf("%s\n", b.str);
196     av_bprint_finalize(&b, NULL);
197
198     av_bprint_init(&b, 0, -1);
199     bprint_pascal(&b, 25);
200     printf("Long text in unlimited buffer: %u/%u\n", (unsigned)strlen(b.str), b.len);
201     av_bprint_finalize(&b, NULL);
202
203     av_bprint_init(&b, 0, 2048);
204     bprint_pascal(&b, 25);
205     printf("Long text in limited buffer: %u/%u\n", (unsigned)strlen(b.str), b.len);
206     av_bprint_finalize(&b, NULL);
207
208     av_bprint_init(&b, 0, 1);
209     bprint_pascal(&b, 5);
210     printf("Short text in automatic buffer: %u/%u\n", (unsigned)strlen(b.str), b.len);
211
212     av_bprint_init(&b, 0, 1);
213     bprint_pascal(&b, 25);
214     printf("Long text in automatic buffer: %u/%u\n", (unsigned)strlen(b.str)/8*8, b.len);
215     /* Note that the size of the automatic buffer is arch-dependant. */
216
217     av_bprint_init(&b, 0, 0);
218     bprint_pascal(&b, 25);
219     printf("Long text count only buffer: %u/%u\n", (unsigned)strlen(b.str), b.len);
220
221     av_bprint_init_for_buffer(&b, buf, sizeof(buf));
222     bprint_pascal(&b, 25);
223     printf("Long text count only buffer: %u/%u\n", (unsigned)strlen(buf), b.len);
224
225     return 0;
226 }
227
228 #endif