1 91c13e54 2004-02-29 devnull /*
 2 91c13e54 2004-02-29 devnull  * The authors of this software are Rob Pike and Ken Thompson.
 3 91c13e54 2004-02-29 devnull  *              Copyright (c) 2002 by Lucent Technologies.
 4 91c13e54 2004-02-29 devnull  * Permission to use, copy, modify, and distribute this software for any
 5 91c13e54 2004-02-29 devnull  * purpose without fee is hereby granted, provided that this entire notice
 6 91c13e54 2004-02-29 devnull  * is included in all copies of any software which is or includes a copy
 7 91c13e54 2004-02-29 devnull  * or modification of this software and in all copies of the supporting
 8 91c13e54 2004-02-29 devnull  * documentation for such software.
 9 91c13e54 2004-02-29 devnull  * THIS SOFTWARE IS BEING PROVIDED "AS IS", WITHOUT ANY EXPRESS OR IMPLIED
10 c90dd38f 2004-12-29 devnull  * WARRANTY.  IN PARTICULAR, NEITHER THE AUTHORS NOR LUCENT TECHNOLOGIES MAKE
11 c90dd38f 2004-12-29 devnull  * ANY REPRESENTATION OR WARRANTY OF ANY KIND CONCERNING THE MERCHANTABILITY
12 91c13e54 2004-02-29 devnull  * OF THIS SOFTWARE OR ITS FITNESS FOR ANY PARTICULAR PURPOSE.
13 91c13e54 2004-02-29 devnull  */
14 df121a00 2004-12-26 devnull /*
15 df121a00 2004-12-26 devnull  * Plan 9 port version must include libc.h in order to 
16 df121a00 2004-12-26 devnull  * get Plan 9 debugging malloc, which sometimes returns
17 df121a00 2004-12-26 devnull  * different pointers than the standard malloc. 
18 df121a00 2004-12-26 devnull  */
19 df121a00 2004-12-26 devnull #ifdef PLAN9PORT
20 df121a00 2004-12-26 devnull #include <u.h>
21 df121a00 2004-12-26 devnull #include <libc.h>
22 eb7f99c5 2004-12-27 devnull #include "fmtdef.h"
23 df121a00 2004-12-26 devnull #else
24 91c13e54 2004-02-29 devnull #include <stdlib.h>
25 e5aa96ac 2004-12-26 devnull #include "plan9.h"
26 91c13e54 2004-02-29 devnull #include "fmt.h"
27 91c13e54 2004-02-29 devnull #include "fmtdef.h"
28 df121a00 2004-12-26 devnull #endif
29 91c13e54 2004-02-29 devnull 
30 df121a00 2004-12-26 devnull static int
31 df121a00 2004-12-26 devnull runeFmtStrFlush(Fmt *f)
32 df121a00 2004-12-26 devnull {
33 df121a00 2004-12-26 devnull 	Rune *s;
34 df121a00 2004-12-26 devnull 	int n;
35 df121a00 2004-12-26 devnull 
36 df121a00 2004-12-26 devnull 	if(f->start == nil)
37 df121a00 2004-12-26 devnull 		return 0;
38 df121a00 2004-12-26 devnull 	n = (int)f->farg;
39 df121a00 2004-12-26 devnull 	n *= 2;
40 df121a00 2004-12-26 devnull 	s = (Rune*)f->start;
41 df121a00 2004-12-26 devnull 	f->start = realloc(s, sizeof(Rune)*n);
42 df121a00 2004-12-26 devnull 	if(f->start == nil){
43 df121a00 2004-12-26 devnull 		f->farg = nil;
44 df121a00 2004-12-26 devnull 		f->to = nil;
45 df121a00 2004-12-26 devnull 		f->stop = nil;
46 df121a00 2004-12-26 devnull 		free(s);
47 df121a00 2004-12-26 devnull 		return 0;
48 df121a00 2004-12-26 devnull 	}
49 df121a00 2004-12-26 devnull 	f->farg = (void*)n;
50 df121a00 2004-12-26 devnull 	f->to = (Rune*)f->start + ((Rune*)f->to - s);
51 df121a00 2004-12-26 devnull 	f->stop = (Rune*)f->start + n - 1;
52 df121a00 2004-12-26 devnull 	return 1;
53 df121a00 2004-12-26 devnull }
54 df121a00 2004-12-26 devnull 
55 df121a00 2004-12-26 devnull int
56 df121a00 2004-12-26 devnull runefmtstrinit(Fmt *f)
57 df121a00 2004-12-26 devnull {
58 df121a00 2004-12-26 devnull 	int n;
59 df121a00 2004-12-26 devnull 
60 df121a00 2004-12-26 devnull 	memset(f, 0, sizeof *f);
61 df121a00 2004-12-26 devnull 	f->runes = 1;
62 df121a00 2004-12-26 devnull 	n = 32;
63 df121a00 2004-12-26 devnull 	f->start = malloc(sizeof(Rune)*n);
64 df121a00 2004-12-26 devnull 	if(f->start == nil)
65 df121a00 2004-12-26 devnull 		return -1;
66 df121a00 2004-12-26 devnull 	f->to = f->start;
67 df121a00 2004-12-26 devnull 	f->stop = (Rune*)f->start + n - 1;
68 df121a00 2004-12-26 devnull 	f->flush = runeFmtStrFlush;
69 df121a00 2004-12-26 devnull 	f->farg = (void*)n;
70 df121a00 2004-12-26 devnull 	f->nfmt = 0;
71 df121a00 2004-12-26 devnull 	return 0;
72 df121a00 2004-12-26 devnull }
73 df121a00 2004-12-26 devnull 
74 91c13e54 2004-02-29 devnull /*
75 91c13e54 2004-02-29 devnull  * print into an allocated string buffer
76 91c13e54 2004-02-29 devnull  */
77 91c13e54 2004-02-29 devnull Rune*
78 91c13e54 2004-02-29 devnull runevsmprint(char *fmt, va_list args)
79 91c13e54 2004-02-29 devnull {
80 91c13e54 2004-02-29 devnull 	Fmt f;
81 91c13e54 2004-02-29 devnull 	int n;
82 91c13e54 2004-02-29 devnull 
83 91c13e54 2004-02-29 devnull 	if(runefmtstrinit(&f) < 0)
84 91c13e54 2004-02-29 devnull 		return nil;
85 df121a00 2004-12-26 devnull 	VA_COPY(f.args,args);
86 91c13e54 2004-02-29 devnull 	n = dofmt(&f, fmt);
87 df121a00 2004-12-26 devnull 	VA_END(f.args);
88 df121a00 2004-12-26 devnull 	if(f.start == nil)
89 91c13e54 2004-02-29 devnull 		return nil;
90 df121a00 2004-12-26 devnull 	if(n < 0){
91 df121a00 2004-12-26 devnull 		free(f.start);
92 df121a00 2004-12-26 devnull 		return nil;
93 df121a00 2004-12-26 devnull 	}
94 91c13e54 2004-02-29 devnull 	*(Rune*)f.to = '\0';
95 91c13e54 2004-02-29 devnull 	return (Rune*)f.start;
96 91c13e54 2004-02-29 devnull }