1 /*
2 * The authors of this software are Rob Pike and Ken Thompson.
3 * Copyright (c) 2002 by Lucent Technologies.
4 * Permission to use, copy, modify, and distribute this software for any
5 * purpose without fee is hereby granted, provided that this entire notice
6 * is included in all copies of any software which is or includes a copy
7 * or modification of this software and in all copies of the supporting
8 * documentation for such software.
9 * THIS SOFTWARE IS BEING PROVIDED "AS IS", WITHOUT ANY EXPRESS OR IMPLIED
10 * WARRANTY. IN PARTICULAR, NEITHER THE AUTHORS NOR LUCENT TECHNOLOGIES MAKE ANY
11 * REPRESENTATION OR WARRANTY OF ANY KIND CONCERNING THE MERCHANTABILITY
12 * OF THIS SOFTWARE OR ITS FITNESS FOR ANY PARTICULAR PURPOSE.
13 */
14 #include "lib9.h"
15 #include "fmtdef.h"
16
17 static int
fmtStrFlush(Fmt * f)18 fmtStrFlush(Fmt *f)
19 {
20 char *s;
21 int n;
22
23 if(f->start == nil)
24 return 0;
25 n = (int)f->farg;
26 n += 256;
27 f->farg = (void*)n;
28 s = f->start;
29 f->start = realloc(s, n);
30 if(f->start == nil){
31 free(s);
32 f->to = nil;
33 f->stop = nil;
34 return 0;
35 }
36 f->to = (char*)f->start + ((char*)f->to - s);
37 f->stop = (char*)f->start + n - 1;
38 return 1;
39 }
40
41 int
fmtstrinit(Fmt * f)42 fmtstrinit(Fmt *f)
43 {
44 int n;
45
46 memset(f, 0, sizeof(*f));
47 n = 32;
48 f->start = malloc(n);
49 if(f->start == nil)
50 return -1;
51 f->to = f->start;
52 f->stop = (char*)f->start + n - 1;
53 f->flush = fmtStrFlush;
54 f->farg = (void*)n;
55 f->nfmt = 0;
56 return 0;
57 }
58
59 /*
60 * print into an allocated string buffer
61 */
62 char*
vsmprint(char * fmt,va_list args)63 vsmprint(char *fmt, va_list args)
64 {
65 Fmt f;
66 int n;
67
68 if(fmtstrinit(&f) < 0)
69 return nil;
70 va_copy(f.args, args);
71 n = dofmt(&f, fmt);
72 va_end(f.args);
73 if(n < 0){
74 free(f.start);
75 f.start = nil;
76 return nil;
77 }
78 return fmtstrflush(&f);
79 }
80