vsmprint.c (1445B) download
1/* Copyright (c) 2002-2006 Lucent Technologies; see LICENSE */
2/*
3 * Plan 9 port version must include libc.h in order to
4 * get Plan 9 debugging malloc, which sometimes returns
5 * different pointers than the standard malloc.
6 */
7#ifdef PLAN9PORT
8# include "fmtdef.h"
9
10# include <libc.h>
11# include <u.h>
12#else
13# include "fmt.h"
14# include "fmtdef.h"
15# include "plan9.h"
16
17# include <stdlib.h>
18# include <string.h>
19#endif
20
21static int
22fmtStrFlush(Fmt* f) {
23 char* s;
24 int n;
25
26 if (f->start == nil)
27 return 0;
28 n = (uintptr) f->farg;
29 n *= 2;
30 s = (char*) f->start;
31 f->start = realloc(s, n);
32 if (f->start == nil) {
33 f->farg = nil;
34 f->to = nil;
35 f->stop = nil;
36 free(s);
37 return 0;
38 }
39 f->farg = (void*) (uintptr) n;
40 f->to = (char*) f->start + ((char*) f->to - s);
41 f->stop = (char*) f->start + n - 1;
42 return 1;
43}
44
45int fmtstrinit(Fmt* f) {
46 int n;
47
48 memset(f, 0, sizeof *f);
49 f->runes = 0;
50 n = 32;
51 f->start = malloc(n);
52 if (f->start == nil)
53 return -1;
54 f->to = f->start;
55 f->stop = (char*) f->start + n - 1;
56 f->flush = fmtStrFlush;
57 f->farg = (void*) (uintptr) n;
58 f->nfmt = 0;
59 fmtlocaleinit(f, nil, nil, nil);
60 return 0;
61}
62
63/*
64 * print into an allocated string buffer
65 */
66char* vsmprint(char* fmt, va_list args) {
67 Fmt f;
68 int n;
69
70 if (fmtstrinit(&f) < 0)
71 return nil;
72 VA_COPY(f.args, args);
73 n = dofmt(&f, fmt);
74 VA_END(f.args);
75 if (n < 0) {
76 free(f.start);
77 return nil;
78 }
79 return fmtstrflush(&f);
80}