Blame


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 91c13e54 2004-02-29 devnull * WARRANTY. IN PARTICULAR, NEITHER THE AUTHORS NOR LUCENT TECHNOLOGIES MAKE ANY
11 91c13e54 2004-02-29 devnull * 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 91c13e54 2004-02-29 devnull #include <stdarg.h>
15 91c13e54 2004-02-29 devnull #include <string.h>
16 91c13e54 2004-02-29 devnull #include "utf.h"
17 91c13e54 2004-02-29 devnull #include "fmt.h"
18 91c13e54 2004-02-29 devnull #include "fmtdef.h"
19 91c13e54 2004-02-29 devnull
20 91c13e54 2004-02-29 devnull enum
21 91c13e54 2004-02-29 devnull {
22 91c13e54 2004-02-29 devnull Maxfmt = 64
23 91c13e54 2004-02-29 devnull };
24 91c13e54 2004-02-29 devnull
25 91c13e54 2004-02-29 devnull typedef struct Convfmt Convfmt;
26 91c13e54 2004-02-29 devnull struct Convfmt
27 91c13e54 2004-02-29 devnull {
28 91c13e54 2004-02-29 devnull int c;
29 91c13e54 2004-02-29 devnull volatile Fmts fmt; /* for spin lock in fmtfmt; avoids race due to write order */
30 91c13e54 2004-02-29 devnull };
31 91c13e54 2004-02-29 devnull
32 91c13e54 2004-02-29 devnull struct
33 91c13e54 2004-02-29 devnull {
34 91c13e54 2004-02-29 devnull /* lock by calling __fmtlock, __fmtunlock */
35 91c13e54 2004-02-29 devnull int nfmt;
36 91c13e54 2004-02-29 devnull Convfmt fmt[Maxfmt];
37 91c13e54 2004-02-29 devnull } fmtalloc;
38 91c13e54 2004-02-29 devnull
39 91c13e54 2004-02-29 devnull static Convfmt knownfmt[] = {
40 91c13e54 2004-02-29 devnull ' ', __flagfmt,
41 91c13e54 2004-02-29 devnull '#', __flagfmt,
42 91c13e54 2004-02-29 devnull '%', __percentfmt,
43 91c13e54 2004-02-29 devnull '+', __flagfmt,
44 91c13e54 2004-02-29 devnull ',', __flagfmt,
45 91c13e54 2004-02-29 devnull '-', __flagfmt,
46 91c13e54 2004-02-29 devnull 'C', __runefmt, /* Plan 9 addition */
47 91c13e54 2004-02-29 devnull 'E', __efgfmt,
48 91c13e54 2004-02-29 devnull 'F', __efgfmt, /* ANSI only */
49 91c13e54 2004-02-29 devnull 'G', __efgfmt,
50 91c13e54 2004-02-29 devnull 'L', __flagfmt, /* ANSI only */
51 91c13e54 2004-02-29 devnull 'S', __runesfmt, /* Plan 9 addition */
52 91c13e54 2004-02-29 devnull 'X', __ifmt,
53 91c13e54 2004-02-29 devnull 'b', __ifmt, /* Plan 9 addition */
54 91c13e54 2004-02-29 devnull 'c', __charfmt,
55 91c13e54 2004-02-29 devnull 'd', __ifmt,
56 91c13e54 2004-02-29 devnull 'e', __efgfmt,
57 91c13e54 2004-02-29 devnull 'f', __efgfmt,
58 91c13e54 2004-02-29 devnull 'g', __efgfmt,
59 91c13e54 2004-02-29 devnull 'h', __flagfmt,
60 91c13e54 2004-02-29 devnull 'i', __ifmt, /* ANSI only */
61 91c13e54 2004-02-29 devnull 'l', __flagfmt,
62 91c13e54 2004-02-29 devnull 'n', __countfmt,
63 91c13e54 2004-02-29 devnull 'o', __ifmt,
64 91c13e54 2004-02-29 devnull 'p', __ifmt,
65 91c13e54 2004-02-29 devnull 'r', __errfmt,
66 91c13e54 2004-02-29 devnull 's', __strfmt,
67 91c13e54 2004-02-29 devnull 'u', __flagfmt, /* in Unix, __ifmt */
68 91c13e54 2004-02-29 devnull 'x', __ifmt,
69 91c13e54 2004-02-29 devnull 0, nil,
70 91c13e54 2004-02-29 devnull };
71 91c13e54 2004-02-29 devnull
72 91c13e54 2004-02-29 devnull
73 91c13e54 2004-02-29 devnull int (*fmtdoquote)(int);
74 91c13e54 2004-02-29 devnull
75 91c13e54 2004-02-29 devnull /*
76 91c13e54 2004-02-29 devnull * __fmtlock() must be set
77 91c13e54 2004-02-29 devnull */
78 91c13e54 2004-02-29 devnull static int
79 91c13e54 2004-02-29 devnull __fmtinstall(int c, Fmts f)
80 91c13e54 2004-02-29 devnull {
81 91c13e54 2004-02-29 devnull Convfmt *p, *ep;
82 91c13e54 2004-02-29 devnull
83 91c13e54 2004-02-29 devnull if(c<=0 || c>=65536)
84 91c13e54 2004-02-29 devnull return -1;
85 91c13e54 2004-02-29 devnull if(!f)
86 91c13e54 2004-02-29 devnull f = __badfmt;
87 91c13e54 2004-02-29 devnull
88 91c13e54 2004-02-29 devnull ep = &fmtalloc.fmt[fmtalloc.nfmt];
89 91c13e54 2004-02-29 devnull for(p=fmtalloc.fmt; p<ep; p++)
90 91c13e54 2004-02-29 devnull if(p->c == c)
91 91c13e54 2004-02-29 devnull break;
92 91c13e54 2004-02-29 devnull
93 91c13e54 2004-02-29 devnull if(p == &fmtalloc.fmt[Maxfmt])
94 91c13e54 2004-02-29 devnull return -1;
95 91c13e54 2004-02-29 devnull
96 91c13e54 2004-02-29 devnull p->fmt = f;
97 91c13e54 2004-02-29 devnull if(p == ep){ /* installing a new format character */
98 91c13e54 2004-02-29 devnull fmtalloc.nfmt++;
99 91c13e54 2004-02-29 devnull p->c = c;
100 91c13e54 2004-02-29 devnull }
101 91c13e54 2004-02-29 devnull
102 91c13e54 2004-02-29 devnull return 0;
103 91c13e54 2004-02-29 devnull }
104 91c13e54 2004-02-29 devnull
105 91c13e54 2004-02-29 devnull int
106 91c13e54 2004-02-29 devnull fmtinstall(int c, Fmts f)
107 91c13e54 2004-02-29 devnull {
108 91c13e54 2004-02-29 devnull int ret;
109 91c13e54 2004-02-29 devnull
110 91c13e54 2004-02-29 devnull __fmtlock();
111 91c13e54 2004-02-29 devnull ret = __fmtinstall(c, f);
112 91c13e54 2004-02-29 devnull __fmtunlock();
113 91c13e54 2004-02-29 devnull return ret;
114 91c13e54 2004-02-29 devnull }
115 91c13e54 2004-02-29 devnull
116 91c13e54 2004-02-29 devnull static Fmts
117 91c13e54 2004-02-29 devnull fmtfmt(int c)
118 91c13e54 2004-02-29 devnull {
119 91c13e54 2004-02-29 devnull Convfmt *p, *ep;
120 91c13e54 2004-02-29 devnull
121 91c13e54 2004-02-29 devnull ep = &fmtalloc.fmt[fmtalloc.nfmt];
122 91c13e54 2004-02-29 devnull for(p=fmtalloc.fmt; p<ep; p++)
123 91c13e54 2004-02-29 devnull if(p->c == c){
124 91c13e54 2004-02-29 devnull while(p->fmt == nil) /* loop until value is updated */
125 91c13e54 2004-02-29 devnull ;
126 91c13e54 2004-02-29 devnull return p->fmt;
127 91c13e54 2004-02-29 devnull }
128 91c13e54 2004-02-29 devnull
129 91c13e54 2004-02-29 devnull /* is this a predefined format char? */
130 91c13e54 2004-02-29 devnull __fmtlock();
131 91c13e54 2004-02-29 devnull for(p=knownfmt; p->c; p++)
132 91c13e54 2004-02-29 devnull if(p->c == c){
133 91c13e54 2004-02-29 devnull __fmtinstall(p->c, p->fmt);
134 91c13e54 2004-02-29 devnull __fmtunlock();
135 91c13e54 2004-02-29 devnull return p->fmt;
136 91c13e54 2004-02-29 devnull }
137 91c13e54 2004-02-29 devnull __fmtunlock();
138 91c13e54 2004-02-29 devnull
139 91c13e54 2004-02-29 devnull return __badfmt;
140 91c13e54 2004-02-29 devnull }
141 91c13e54 2004-02-29 devnull
142 91c13e54 2004-02-29 devnull void*
143 91c13e54 2004-02-29 devnull __fmtdispatch(Fmt *f, void *fmt, int isrunes)
144 91c13e54 2004-02-29 devnull {
145 91c13e54 2004-02-29 devnull Rune rune, r;
146 91c13e54 2004-02-29 devnull int i, n;
147 91c13e54 2004-02-29 devnull
148 91c13e54 2004-02-29 devnull f->flags = 0;
149 91c13e54 2004-02-29 devnull f->width = f->prec = 0;
150 91c13e54 2004-02-29 devnull
151 91c13e54 2004-02-29 devnull for(;;){
152 91c13e54 2004-02-29 devnull if(isrunes){
153 91c13e54 2004-02-29 devnull r = *(Rune*)fmt;
154 91c13e54 2004-02-29 devnull fmt = (Rune*)fmt + 1;
155 91c13e54 2004-02-29 devnull }else{
156 91c13e54 2004-02-29 devnull fmt = (char*)fmt + chartorune(&rune, (char*)fmt);
157 91c13e54 2004-02-29 devnull r = rune;
158 91c13e54 2004-02-29 devnull }
159 91c13e54 2004-02-29 devnull f->r = r;
160 91c13e54 2004-02-29 devnull switch(r){
161 91c13e54 2004-02-29 devnull case '\0':
162 91c13e54 2004-02-29 devnull return nil;
163 91c13e54 2004-02-29 devnull case '.':
164 91c13e54 2004-02-29 devnull f->flags |= FmtWidth|FmtPrec;
165 91c13e54 2004-02-29 devnull continue;
166 91c13e54 2004-02-29 devnull case '0':
167 91c13e54 2004-02-29 devnull if(!(f->flags & FmtWidth)){
168 91c13e54 2004-02-29 devnull f->flags |= FmtZero;
169 91c13e54 2004-02-29 devnull continue;
170 91c13e54 2004-02-29 devnull }
171 91c13e54 2004-02-29 devnull /* fall through */
172 91c13e54 2004-02-29 devnull case '1': case '2': case '3': case '4':
173 91c13e54 2004-02-29 devnull case '5': case '6': case '7': case '8': case '9':
174 91c13e54 2004-02-29 devnull i = 0;
175 91c13e54 2004-02-29 devnull while(r >= '0' && r <= '9'){
176 91c13e54 2004-02-29 devnull i = i * 10 + r - '0';
177 91c13e54 2004-02-29 devnull if(isrunes){
178 91c13e54 2004-02-29 devnull r = *(Rune*)fmt;
179 91c13e54 2004-02-29 devnull fmt = (Rune*)fmt + 1;
180 91c13e54 2004-02-29 devnull }else{
181 91c13e54 2004-02-29 devnull r = *(char*)fmt;
182 91c13e54 2004-02-29 devnull fmt = (char*)fmt + 1;
183 91c13e54 2004-02-29 devnull }
184 91c13e54 2004-02-29 devnull }
185 91c13e54 2004-02-29 devnull if(isrunes)
186 91c13e54 2004-02-29 devnull fmt = (Rune*)fmt - 1;
187 91c13e54 2004-02-29 devnull else
188 91c13e54 2004-02-29 devnull fmt = (char*)fmt - 1;
189 91c13e54 2004-02-29 devnull numflag:
190 91c13e54 2004-02-29 devnull if(f->flags & FmtWidth){
191 91c13e54 2004-02-29 devnull f->flags |= FmtPrec;
192 91c13e54 2004-02-29 devnull f->prec = i;
193 91c13e54 2004-02-29 devnull }else{
194 91c13e54 2004-02-29 devnull f->flags |= FmtWidth;
195 91c13e54 2004-02-29 devnull f->width = i;
196 91c13e54 2004-02-29 devnull }
197 91c13e54 2004-02-29 devnull continue;
198 91c13e54 2004-02-29 devnull case '*':
199 91c13e54 2004-02-29 devnull i = va_arg(f->args, int);
200 91c13e54 2004-02-29 devnull if(i < 0){
201 91c13e54 2004-02-29 devnull /*
202 91c13e54 2004-02-29 devnull * negative precision =>
203 91c13e54 2004-02-29 devnull * ignore the precision.
204 91c13e54 2004-02-29 devnull */
205 91c13e54 2004-02-29 devnull if(f->flags & FmtPrec){
206 91c13e54 2004-02-29 devnull f->flags &= ~FmtPrec;
207 91c13e54 2004-02-29 devnull f->prec = 0;
208 91c13e54 2004-02-29 devnull continue;
209 91c13e54 2004-02-29 devnull }
210 91c13e54 2004-02-29 devnull i = -i;
211 91c13e54 2004-02-29 devnull f->flags |= FmtLeft;
212 91c13e54 2004-02-29 devnull }
213 91c13e54 2004-02-29 devnull goto numflag;
214 91c13e54 2004-02-29 devnull }
215 91c13e54 2004-02-29 devnull n = (*fmtfmt(r))(f);
216 91c13e54 2004-02-29 devnull if(n < 0)
217 91c13e54 2004-02-29 devnull return nil;
218 91c13e54 2004-02-29 devnull if(n == 0)
219 91c13e54 2004-02-29 devnull return fmt;
220 91c13e54 2004-02-29 devnull }
221 91c13e54 2004-02-29 devnull }