vsmprint.c (1445B)
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 21 static int 22 fmtStrFlush(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 45 int 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 */ 66 char* 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 }