1 85231fd8 2006-05-21 devnull /* Copyright (c) 2002-2006 Lucent Technologies; see LICENSE */
2 91c13e54 2004-02-29 devnull #include <stdarg.h>
3 91c13e54 2004-02-29 devnull #include <string.h>
4 e5aa96ac 2004-12-26 devnull #include "plan9.h"
5 91c13e54 2004-02-29 devnull #include "fmt.h"
6 91c13e54 2004-02-29 devnull #include "fmtdef.h"
10 91c13e54 2004-02-29 devnull Maxfmt = 64
13 91c13e54 2004-02-29 devnull typedef struct Convfmt Convfmt;
14 91c13e54 2004-02-29 devnull struct Convfmt
17 91c13e54 2004-02-29 devnull volatile Fmts fmt; /* for spin lock in fmtfmt; avoids race due to write order */
20 85231fd8 2006-05-21 devnull static struct
22 91c13e54 2004-02-29 devnull /* lock by calling __fmtlock, __fmtunlock */
23 91c13e54 2004-02-29 devnull int nfmt;
24 91c13e54 2004-02-29 devnull Convfmt fmt[Maxfmt];
25 91c13e54 2004-02-29 devnull } fmtalloc;
27 91c13e54 2004-02-29 devnull static Convfmt knownfmt[] = {
28 91c13e54 2004-02-29 devnull ' ', __flagfmt,
29 91c13e54 2004-02-29 devnull '#', __flagfmt,
30 91c13e54 2004-02-29 devnull '%', __percentfmt,
31 85231fd8 2006-05-21 devnull '\'', __flagfmt,
32 91c13e54 2004-02-29 devnull '+', __flagfmt,
33 91c13e54 2004-02-29 devnull ',', __flagfmt,
34 91c13e54 2004-02-29 devnull '-', __flagfmt,
35 91c13e54 2004-02-29 devnull 'C', __runefmt, /* Plan 9 addition */
36 91c13e54 2004-02-29 devnull 'E', __efgfmt,
37 df121a00 2004-12-26 devnull #ifndef PLAN9PORT
38 91c13e54 2004-02-29 devnull 'F', __efgfmt, /* ANSI only */
40 91c13e54 2004-02-29 devnull 'G', __efgfmt,
41 df121a00 2004-12-26 devnull #ifndef PLAN9PORT
42 91c13e54 2004-02-29 devnull 'L', __flagfmt, /* ANSI only */
44 91c13e54 2004-02-29 devnull 'S', __runesfmt, /* Plan 9 addition */
45 91c13e54 2004-02-29 devnull 'X', __ifmt,
46 91c13e54 2004-02-29 devnull 'b', __ifmt, /* Plan 9 addition */
47 91c13e54 2004-02-29 devnull 'c', __charfmt,
48 91c13e54 2004-02-29 devnull 'd', __ifmt,
49 91c13e54 2004-02-29 devnull 'e', __efgfmt,
50 91c13e54 2004-02-29 devnull 'f', __efgfmt,
51 91c13e54 2004-02-29 devnull 'g', __efgfmt,
52 91c13e54 2004-02-29 devnull 'h', __flagfmt,
53 df121a00 2004-12-26 devnull #ifndef PLAN9PORT
54 91c13e54 2004-02-29 devnull 'i', __ifmt, /* ANSI only */
56 91c13e54 2004-02-29 devnull 'l', __flagfmt,
57 91c13e54 2004-02-29 devnull 'n', __countfmt,
58 91c13e54 2004-02-29 devnull 'o', __ifmt,
59 91c13e54 2004-02-29 devnull 'p', __ifmt,
60 91c13e54 2004-02-29 devnull 'r', __errfmt,
61 91c13e54 2004-02-29 devnull 's', __strfmt,
62 e5aa96ac 2004-12-26 devnull #ifdef PLAN9PORT
63 e5aa96ac 2004-12-26 devnull 'u', __flagfmt,
65 e5aa96ac 2004-12-26 devnull 'u', __ifmt,
67 91c13e54 2004-02-29 devnull 'x', __ifmt,
72 91c13e54 2004-02-29 devnull int (*fmtdoquote)(int);
75 91c13e54 2004-02-29 devnull * __fmtlock() must be set
77 91c13e54 2004-02-29 devnull static int
78 91c13e54 2004-02-29 devnull __fmtinstall(int c, Fmts f)
80 91c13e54 2004-02-29 devnull Convfmt *p, *ep;
82 91c13e54 2004-02-29 devnull if(c<=0 || c>=65536)
83 91c13e54 2004-02-29 devnull return -1;
85 91c13e54 2004-02-29 devnull f = __badfmt;
87 91c13e54 2004-02-29 devnull ep = &fmtalloc.fmt[fmtalloc.nfmt];
88 91c13e54 2004-02-29 devnull for(p=fmtalloc.fmt; p<ep; p++)
89 91c13e54 2004-02-29 devnull if(p->c == c)
92 91c13e54 2004-02-29 devnull if(p == &fmtalloc.fmt[Maxfmt])
93 91c13e54 2004-02-29 devnull return -1;
95 91c13e54 2004-02-29 devnull p->fmt = f;
96 91c13e54 2004-02-29 devnull if(p == ep){ /* installing a new format character */
97 91c13e54 2004-02-29 devnull fmtalloc.nfmt++;
98 91c13e54 2004-02-29 devnull p->c = c;
101 91c13e54 2004-02-29 devnull return 0;
105 e5aa96ac 2004-12-26 devnull fmtinstall(int c, int (*f)(Fmt*))
107 91c13e54 2004-02-29 devnull int ret;
109 91c13e54 2004-02-29 devnull __fmtlock();
110 91c13e54 2004-02-29 devnull ret = __fmtinstall(c, f);
111 91c13e54 2004-02-29 devnull __fmtunlock();
112 91c13e54 2004-02-29 devnull return ret;
115 91c13e54 2004-02-29 devnull static Fmts
116 91c13e54 2004-02-29 devnull fmtfmt(int c)
118 91c13e54 2004-02-29 devnull Convfmt *p, *ep;
120 91c13e54 2004-02-29 devnull ep = &fmtalloc.fmt[fmtalloc.nfmt];
121 91c13e54 2004-02-29 devnull for(p=fmtalloc.fmt; p<ep; p++)
122 91c13e54 2004-02-29 devnull if(p->c == c){
123 91c13e54 2004-02-29 devnull while(p->fmt == nil) /* loop until value is updated */
125 91c13e54 2004-02-29 devnull return p->fmt;
128 91c13e54 2004-02-29 devnull /* is this a predefined format char? */
129 91c13e54 2004-02-29 devnull __fmtlock();
130 91c13e54 2004-02-29 devnull for(p=knownfmt; p->c; p++)
131 91c13e54 2004-02-29 devnull if(p->c == c){
132 91c13e54 2004-02-29 devnull __fmtinstall(p->c, p->fmt);
133 91c13e54 2004-02-29 devnull __fmtunlock();
134 91c13e54 2004-02-29 devnull return p->fmt;
136 91c13e54 2004-02-29 devnull __fmtunlock();
138 91c13e54 2004-02-29 devnull return __badfmt;
142 91c13e54 2004-02-29 devnull __fmtdispatch(Fmt *f, void *fmt, int isrunes)
144 91c13e54 2004-02-29 devnull Rune rune, r;
145 91c13e54 2004-02-29 devnull int i, n;
147 91c13e54 2004-02-29 devnull f->flags = 0;
148 91c13e54 2004-02-29 devnull f->width = f->prec = 0;
150 91c13e54 2004-02-29 devnull for(;;){
151 91c13e54 2004-02-29 devnull if(isrunes){
152 91c13e54 2004-02-29 devnull r = *(Rune*)fmt;
153 91c13e54 2004-02-29 devnull fmt = (Rune*)fmt + 1;
155 91c13e54 2004-02-29 devnull fmt = (char*)fmt + chartorune(&rune, (char*)fmt);
156 91c13e54 2004-02-29 devnull r = rune;
158 91c13e54 2004-02-29 devnull f->r = r;
159 91c13e54 2004-02-29 devnull switch(r){
160 91c13e54 2004-02-29 devnull case '\0':
161 91c13e54 2004-02-29 devnull return nil;
162 91c13e54 2004-02-29 devnull case '.':
163 91c13e54 2004-02-29 devnull f->flags |= FmtWidth|FmtPrec;
164 91c13e54 2004-02-29 devnull continue;
165 91c13e54 2004-02-29 devnull case '0':
166 91c13e54 2004-02-29 devnull if(!(f->flags & FmtWidth)){
167 91c13e54 2004-02-29 devnull f->flags |= FmtZero;
168 91c13e54 2004-02-29 devnull continue;
170 91c13e54 2004-02-29 devnull /* fall through */
171 91c13e54 2004-02-29 devnull case '1': case '2': case '3': case '4':
172 91c13e54 2004-02-29 devnull case '5': case '6': case '7': case '8': case '9':
174 91c13e54 2004-02-29 devnull while(r >= '0' && r <= '9'){
175 91c13e54 2004-02-29 devnull i = i * 10 + r - '0';
176 91c13e54 2004-02-29 devnull if(isrunes){
177 91c13e54 2004-02-29 devnull r = *(Rune*)fmt;
178 91c13e54 2004-02-29 devnull fmt = (Rune*)fmt + 1;
180 91c13e54 2004-02-29 devnull r = *(char*)fmt;
181 91c13e54 2004-02-29 devnull fmt = (char*)fmt + 1;
184 91c13e54 2004-02-29 devnull if(isrunes)
185 91c13e54 2004-02-29 devnull fmt = (Rune*)fmt - 1;
187 91c13e54 2004-02-29 devnull fmt = (char*)fmt - 1;
188 91c13e54 2004-02-29 devnull numflag:
189 91c13e54 2004-02-29 devnull if(f->flags & FmtWidth){
190 91c13e54 2004-02-29 devnull f->flags |= FmtPrec;
191 91c13e54 2004-02-29 devnull f->prec = i;
193 91c13e54 2004-02-29 devnull f->flags |= FmtWidth;
194 91c13e54 2004-02-29 devnull f->width = i;
196 91c13e54 2004-02-29 devnull continue;
197 91c13e54 2004-02-29 devnull case '*':
198 91c13e54 2004-02-29 devnull i = va_arg(f->args, int);
199 91c13e54 2004-02-29 devnull if(i < 0){
201 91c13e54 2004-02-29 devnull * negative precision =>
202 91c13e54 2004-02-29 devnull * ignore the precision.
204 91c13e54 2004-02-29 devnull if(f->flags & FmtPrec){
205 91c13e54 2004-02-29 devnull f->flags &= ~FmtPrec;
206 91c13e54 2004-02-29 devnull f->prec = 0;
207 91c13e54 2004-02-29 devnull continue;
210 91c13e54 2004-02-29 devnull f->flags |= FmtLeft;
212 91c13e54 2004-02-29 devnull goto numflag;
214 91c13e54 2004-02-29 devnull n = (*fmtfmt(r))(f);
215 91c13e54 2004-02-29 devnull if(n < 0)
216 91c13e54 2004-02-29 devnull return nil;
217 91c13e54 2004-02-29 devnull if(n == 0)
218 91c13e54 2004-02-29 devnull return fmt;