Mercurial > hg-old > index.cgi

comparison lib/argp-fmtstream.c @ 434:b8bf63962a99 3.0

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Added various generated files for release
author lost@l-w.ca
date Fri, 29 Oct 2010 19:20:39 -0600
parents
children
comparison
equal deleted inserted replaced
433:b142b473f0ee 434:b8bf63962a99
1 /* Word-wrapping and line-truncating streams
2 Copyright (C) 1997-1999, 2001-2003, 2005, 2009-2010 Free Software
3 Foundation, Inc.
4 This file is part of the GNU C Library.
5 Written by Miles Bader <miles@gnu.ai.mit.edu>.
6
7 This program is free software: you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
11
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>. */
19
20 /* This package emulates glibc `line_wrap_stream' semantics for systems that
21 don't have that. */
22
23 #ifdef HAVE_CONFIG_H
24 # include <config.h>
25 #endif
26
27 #include <stdlib.h>
28 #include <string.h>
29 #include <errno.h>
30 #include <stdarg.h>
31 #include <ctype.h>
32
33 #include "argp-fmtstream.h"
34 #include "argp-namefrob.h"
35
36 #ifndef ARGP_FMTSTREAM_USE_LINEWRAP
37
38 #ifndef isblank
39 #define isblank(ch) ((ch)==' ' || (ch)=='\t')
40 #endif
41
42 #if defined _LIBC && defined USE_IN_LIBIO
43 # include <wchar.h>
44 # include <libio/libioP.h>
45 # define __vsnprintf(s, l, f, a) _IO_vsnprintf (s, l, f, a)
46 #endif
47
48 #define INIT_BUF_SIZE 200
49 #define PRINTF_SIZE_GUESS 150
50
51 /* Return an argp_fmtstream that outputs to STREAM, and which prefixes lines
52 written on it with LMARGIN spaces and limits them to RMARGIN columns
53 total. If WMARGIN >= 0, words that extend past RMARGIN are wrapped by
54 replacing the whitespace before them with a newline and WMARGIN spaces.
55 Otherwise, chars beyond RMARGIN are simply dropped until a newline.
56 Returns NULL if there was an error. */
57 argp_fmtstream_t
58 __argp_make_fmtstream (FILE *stream,
59 size_t lmargin, size_t rmargin, ssize_t wmargin)
60 {
61 argp_fmtstream_t fs;
62
63 fs = (struct argp_fmtstream *) malloc (sizeof (struct argp_fmtstream));
64 if (fs != NULL)
65 {
66 fs->stream = stream;
67
68 fs->lmargin = lmargin;
69 fs->rmargin = rmargin;
70 fs->wmargin = wmargin;
71 fs->point_col = 0;
72 fs->point_offs = 0;
73
74 fs->buf = (char *) malloc (INIT_BUF_SIZE);
75 if (! fs->buf)
76 {
77 free (fs);
78 fs = 0;
79 }
80 else
81 {
82 fs->p = fs->buf;
83 fs->end = fs->buf + INIT_BUF_SIZE;
84 }
85 }
86
87 return fs;
88 }
89 #if 0
90 /* Not exported. */
91 #ifdef weak_alias
92 weak_alias (__argp_make_fmtstream, argp_make_fmtstream)
93 #endif
94 #endif
95
96 /* Flush FS to its stream, and free it (but don't close the stream). */
97 void
98 __argp_fmtstream_free (argp_fmtstream_t fs)
99 {
100 __argp_fmtstream_update (fs);
101 if (fs->p > fs->buf)
102 {
103 #ifdef USE_IN_LIBIO
104 __fxprintf (fs->stream, "%.*s", (int) (fs->p - fs->buf), fs->buf);
105 #else
106 fwrite_unlocked (fs->buf, 1, fs->p - fs->buf, fs->stream);
107 #endif
108 }
109 free (fs->buf);
110 free (fs);
111 }
112 #if 0
113 /* Not exported. */
114 #ifdef weak_alias
115 weak_alias (__argp_fmtstream_free, argp_fmtstream_free)
116 #endif
117 #endif
118
119 /* Process FS's buffer so that line wrapping is done from POINT_OFFS to the
120 end of its buffer. This code is mostly from glibc stdio/linewrap.c. */
121 void
122 __argp_fmtstream_update (argp_fmtstream_t fs)
123 {
124 char *buf, *nl;
125 size_t len;
126
127 /* Scan the buffer for newlines. */
128 buf = fs->buf + fs->point_offs;
129 while (buf < fs->p)
130 {
131 size_t r;
132
133 if (fs->point_col == 0 && fs->lmargin != 0)
134 {
135 /* We are starting a new line. Print spaces to the left margin. */
136 const size_t pad = fs->lmargin;
137 if (fs->p + pad < fs->end)
138 {
139 /* We can fit in them in the buffer by moving the
140 buffer text up and filling in the beginning. */
141 memmove (buf + pad, buf, fs->p - buf);
142 fs->p += pad; /* Compensate for bigger buffer. */
143 memset (buf, ' ', pad); /* Fill in the spaces. */
144 buf += pad; /* Don't bother searching them. */
145 }
146 else
147 {
148 /* No buffer space for spaces. Must flush. */
149 size_t i;
150 for (i = 0; i < pad; i++)
151 {
152 #ifdef USE_IN_LIBIO
153 if (_IO_fwide (fs->stream, 0) > 0)
154 putwc_unlocked (L' ', fs->stream);
155 else
156 #endif
157 putc_unlocked (' ', fs->stream);
158 }
159 }
160 fs->point_col = pad;
161 }
162
163 len = fs->p - buf;
164 nl = memchr (buf, '\n', len);
165
166 if (fs->point_col < 0)
167 fs->point_col = 0;
168
169 if (!nl)
170 {
171 /* The buffer ends in a partial line. */
172
173 if (fs->point_col + len < fs->rmargin)
174 {
175 /* The remaining buffer text is a partial line and fits
176 within the maximum line width. Advance point for the
177 characters to be written and stop scanning. */
178 fs->point_col += len;
179 break;
180 }
181 else
182 /* Set the end-of-line pointer for the code below to
183 the end of the buffer. */
184 nl = fs->p;
185 }
186 else if (fs->point_col + (nl - buf) < (ssize_t) fs->rmargin)
187 {
188 /* The buffer contains a full line that fits within the maximum
189 line width. Reset point and scan the next line. */
190 fs->point_col = 0;
191 buf = nl + 1;
192 continue;
193 }
194
195 /* This line is too long. */
196 r = fs->rmargin - 1;
197
198 if (fs->wmargin < 0)
199 {
200 /* Truncate the line by overwriting the excess with the
201 newline and anything after it in the buffer. */
202 if (nl < fs->p)
203 {
204 memmove (buf + (r - fs->point_col), nl, fs->p - nl);
205 fs->p -= buf + (r - fs->point_col) - nl;
206 /* Reset point for the next line and start scanning it. */
207 fs->point_col = 0;
208 buf += r + 1; /* Skip full line plus \n. */
209 }
210 else
211 {
212 /* The buffer ends with a partial line that is beyond the
213 maximum line width. Advance point for the characters
214 written, and discard those past the max from the buffer. */
215 fs->point_col += len;
216 fs->p -= fs->point_col - r;
217 break;
218 }
219 }
220 else
221 {
222 /* Do word wrap. Go to the column just past the maximum line
223 width and scan back for the beginning of the word there.
224 Then insert a line break. */
225
226 char *p, *nextline;
227 int i;
228
229 p = buf + (r + 1 - fs->point_col);
230 while (p >= buf && !isblank ((unsigned char) *p))
231 --p;
232 nextline = p + 1; /* This will begin the next line. */
233
234 if (nextline > buf)
235 {
236 /* Swallow separating blanks. */
237 if (p >= buf)
238 do
239 --p;
240 while (p >= buf && isblank ((unsigned char) *p));
241 nl = p + 1; /* The newline will replace the first blank. */
242 }
243 else
244 {
245 /* A single word that is greater than the maximum line width.
246 Oh well. Put it on an overlong line by itself. */
247 p = buf + (r + 1 - fs->point_col);
248 /* Find the end of the long word. */
249 if (p < nl)
250 do
251 ++p;
252 while (p < nl && !isblank ((unsigned char) *p));
253 if (p == nl)
254 {
255 /* It already ends a line. No fussing required. */
256 fs->point_col = 0;
257 buf = nl + 1;
258 continue;
259 }
260 /* We will move the newline to replace the first blank. */
261 nl = p;
262 /* Swallow separating blanks. */
263 do
264 ++p;
265 while (isblank ((unsigned char) *p));
266 /* The next line will start here. */
267 nextline = p;
268 }
269
270 /* Note: There are a bunch of tests below for
271 NEXTLINE == BUF + LEN + 1; this case is where NL happens to fall
272 at the end of the buffer, and NEXTLINE is in fact empty (and so
273 we need not be careful to maintain its contents). */
274
275 if ((nextline == buf + len + 1
276 ? fs->end - nl < fs->wmargin + 1
277 : nextline - (nl + 1) < fs->wmargin)
278 && fs->p > nextline)
279 {
280 /* The margin needs more blanks than we removed. */
281 if (fs->end - fs->p > fs->wmargin + 1)
282 /* Make some space for them. */
283 {
284 size_t mv = fs->p - nextline;
285 memmove (nl + 1 + fs->wmargin, nextline, mv);
286 nextline = nl + 1 + fs->wmargin;
287 len = nextline + mv - buf;
288 *nl++ = '\n';
289 }
290 else
291 /* Output the first line so we can use the space. */
292 {
293 #ifdef _LIBC
294 __fxprintf (fs->stream, "%.*s\n",
295 (int) (nl - fs->buf), fs->buf);
296 #else
297 if (nl > fs->buf)
298 fwrite_unlocked (fs->buf, 1, nl - fs->buf, fs->stream);
299 putc_unlocked ('\n', fs->stream);
300 #endif
301
302 len += buf - fs->buf;
303 nl = buf = fs->buf;
304 }
305 }
306 else
307 /* We can fit the newline and blanks in before
308 the next word. */
309 *nl++ = '\n';
310
311 if (nextline - nl >= fs->wmargin
312 || (nextline == buf + len + 1 && fs->end - nextline >= fs->wmargin))
313 /* Add blanks up to the wrap margin column. */
314 for (i = 0; i < fs->wmargin; ++i)
315 *nl++ = ' ';
316 else
317 for (i = 0; i < fs->wmargin; ++i)
318 #ifdef USE_IN_LIBIO
319 if (_IO_fwide (fs->stream, 0) > 0)
320 putwc_unlocked (L' ', fs->stream);
321 else
322 #endif
323 putc_unlocked (' ', fs->stream);
324
325 /* Copy the tail of the original buffer into the current buffer
326 position. */
327 if (nl < nextline)
328 memmove (nl, nextline, buf + len - nextline);
329 len -= nextline - buf;
330
331 /* Continue the scan on the remaining lines in the buffer. */
332 buf = nl;
333
334 /* Restore bufp to include all the remaining text. */
335 fs->p = nl + len;
336
337 /* Reset the counter of what has been output this line. If wmargin
338 is 0, we want to avoid the lmargin getting added, so we set
339 point_col to a magic value of -1 in that case. */
340 fs->point_col = fs->wmargin ? fs->wmargin : -1;
341 }
342 }
343
344 /* Remember that we've scanned as far as the end of the buffer. */
345 fs->point_offs = fs->p - fs->buf;
346 }
347
348 /* Ensure that FS has space for AMOUNT more bytes in its buffer, either by
349 growing the buffer, or by flushing it. True is returned iff we succeed. */
350 int
351 __argp_fmtstream_ensure (struct argp_fmtstream *fs, size_t amount)
352 {
353 if ((size_t) (fs->end - fs->p) < amount)
354 {
355 ssize_t wrote;
356
357 /* Flush FS's buffer. */
358 __argp_fmtstream_update (fs);
359
360 #ifdef _LIBC
361 __fxprintf (fs->stream, "%.*s", (int) (fs->p - fs->buf), fs->buf);
362 wrote = fs->p - fs->buf;
363 #else
364 wrote = fwrite_unlocked (fs->buf, 1, fs->p - fs->buf, fs->stream);
365 #endif
366 if (wrote == fs->p - fs->buf)
367 {
368 fs->p = fs->buf;
369 fs->point_offs = 0;
370 }
371 else
372 {
373 fs->p -= wrote;
374 fs->point_offs -= wrote;
375 memmove (fs->buf, fs->buf + wrote, fs->p - fs->buf);
376 return 0;
377 }
378
379 if ((size_t) (fs->end - fs->buf) < amount)
380 /* Gotta grow the buffer. */
381 {
382 size_t old_size = fs->end - fs->buf;
383 size_t new_size = old_size + amount;
384 char *new_buf;
385
386 if (new_size < old_size || ! (new_buf = realloc (fs->buf, new_size)))
387 {
388 __set_errno (ENOMEM);
389 return 0;
390 }
391
392 fs->buf = new_buf;
393 fs->end = new_buf + new_size;
394 fs->p = fs->buf;
395 }
396 }
397
398 return 1;
399 }
400
401 ssize_t
402 __argp_fmtstream_printf (struct argp_fmtstream *fs, const char *fmt, ...)
403 {
404 int out;
405 size_t avail;
406 size_t size_guess = PRINTF_SIZE_GUESS; /* How much space to reserve. */
407
408 do
409 {
410 va_list args;
411
412 if (! __argp_fmtstream_ensure (fs, size_guess))
413 return -1;
414
415 va_start (args, fmt);
416 avail = fs->end - fs->p;
417 out = __vsnprintf (fs->p, avail, fmt, args);
418 va_end (args);
419 if ((size_t) out >= avail)
420 size_guess = out + 1;
421 }
422 while ((size_t) out >= avail);
423
424 fs->p += out;
425
426 return out;
427 }
428 #if 0
429 /* Not exported. */
430 #ifdef weak_alias
431 weak_alias (__argp_fmtstream_printf, argp_fmtstream_printf)
432 #endif
433 #endif
434
435 #endif /* !ARGP_FMTSTREAM_USE_LINEWRAP */