Blame


1 7a4ee46d 2003-11-23 devnull #include "stdinc.h"
2 7a4ee46d 2003-11-23 devnull #include "dat.h"
3 7a4ee46d 2003-11-23 devnull #include "fns.h"
4 7a4ee46d 2003-11-23 devnull
5 7a4ee46d 2003-11-23 devnull int
6 7a4ee46d 2003-11-23 devnull namecmp(char *s, char *t)
7 7a4ee46d 2003-11-23 devnull {
8 7a4ee46d 2003-11-23 devnull return strncmp(s, t, ANameSize);
9 7a4ee46d 2003-11-23 devnull }
10 7a4ee46d 2003-11-23 devnull
11 7a4ee46d 2003-11-23 devnull void
12 7a4ee46d 2003-11-23 devnull namecp(char *dst, char *src)
13 7a4ee46d 2003-11-23 devnull {
14 7a4ee46d 2003-11-23 devnull strncpy(dst, src, ANameSize - 1);
15 7a4ee46d 2003-11-23 devnull dst[ANameSize - 1] = '\0';
16 7a4ee46d 2003-11-23 devnull }
17 7a4ee46d 2003-11-23 devnull
18 7a4ee46d 2003-11-23 devnull int
19 7a4ee46d 2003-11-23 devnull nameok(char *name)
20 7a4ee46d 2003-11-23 devnull {
21 7a4ee46d 2003-11-23 devnull char *t;
22 7a4ee46d 2003-11-23 devnull int c;
23 7a4ee46d 2003-11-23 devnull
24 7a4ee46d 2003-11-23 devnull if(name == nil)
25 7a4ee46d 2003-11-23 devnull return -1;
26 7a4ee46d 2003-11-23 devnull for(t = name; c = *t; t++)
27 7a4ee46d 2003-11-23 devnull if(t - name >= ANameSize
28 7a4ee46d 2003-11-23 devnull || c < ' ' || c >= 0x7f)
29 7a4ee46d 2003-11-23 devnull return -1;
30 7a4ee46d 2003-11-23 devnull return 0;
31 7a4ee46d 2003-11-23 devnull }
32 7a4ee46d 2003-11-23 devnull
33 7a4ee46d 2003-11-23 devnull int
34 7a4ee46d 2003-11-23 devnull stru32int(char *s, u32int *r)
35 7a4ee46d 2003-11-23 devnull {
36 7a4ee46d 2003-11-23 devnull char *t;
37 7a4ee46d 2003-11-23 devnull u32int n, nn, m;
38 7a4ee46d 2003-11-23 devnull int c;
39 7a4ee46d 2003-11-23 devnull
40 7a4ee46d 2003-11-23 devnull m = TWID32 / 10;
41 7a4ee46d 2003-11-23 devnull n = 0;
42 7a4ee46d 2003-11-23 devnull for(t = s; ; t++){
43 7a4ee46d 2003-11-23 devnull c = *t;
44 7a4ee46d 2003-11-23 devnull if(c < '0' || c > '9')
45 7a4ee46d 2003-11-23 devnull break;
46 7a4ee46d 2003-11-23 devnull if(n > m)
47 7a4ee46d 2003-11-23 devnull return -1;
48 7a4ee46d 2003-11-23 devnull nn = n * 10 + c - '0';
49 7a4ee46d 2003-11-23 devnull if(nn < n)
50 7a4ee46d 2003-11-23 devnull return -1;
51 7a4ee46d 2003-11-23 devnull n = nn;
52 7a4ee46d 2003-11-23 devnull }
53 7a4ee46d 2003-11-23 devnull *r = n;
54 7a4ee46d 2003-11-23 devnull return s != t && *t == '\0';
55 7a4ee46d 2003-11-23 devnull }
56 7a4ee46d 2003-11-23 devnull
57 7a4ee46d 2003-11-23 devnull int
58 7a4ee46d 2003-11-23 devnull stru64int(char *s, u64int *r)
59 7a4ee46d 2003-11-23 devnull {
60 7a4ee46d 2003-11-23 devnull char *t;
61 7a4ee46d 2003-11-23 devnull u64int n, nn, m;
62 7a4ee46d 2003-11-23 devnull int c;
63 7a4ee46d 2003-11-23 devnull
64 7a4ee46d 2003-11-23 devnull m = TWID64 / 10;
65 7a4ee46d 2003-11-23 devnull n = 0;
66 7a4ee46d 2003-11-23 devnull for(t = s; ; t++){
67 7a4ee46d 2003-11-23 devnull c = *t;
68 7a4ee46d 2003-11-23 devnull if(c < '0' || c > '9')
69 7a4ee46d 2003-11-23 devnull break;
70 7a4ee46d 2003-11-23 devnull if(n > m)
71 7a4ee46d 2003-11-23 devnull return -1;
72 7a4ee46d 2003-11-23 devnull nn = n * 10 + c - '0';
73 7a4ee46d 2003-11-23 devnull if(nn < n)
74 7a4ee46d 2003-11-23 devnull return -1;
75 7a4ee46d 2003-11-23 devnull n = nn;
76 7a4ee46d 2003-11-23 devnull }
77 7a4ee46d 2003-11-23 devnull *r = n;
78 7a4ee46d 2003-11-23 devnull return s != t && *t == '\0';
79 7a4ee46d 2003-11-23 devnull }
80 7a4ee46d 2003-11-23 devnull
81 7a4ee46d 2003-11-23 devnull int
82 7a4ee46d 2003-11-23 devnull vttypevalid(int type)
83 7a4ee46d 2003-11-23 devnull {
84 7a4ee46d 2003-11-23 devnull return type < VtMaxType;
85 7a4ee46d 2003-11-23 devnull }
86 7a4ee46d 2003-11-23 devnull
87 7a4ee46d 2003-11-23 devnull void
88 7a4ee46d 2003-11-23 devnull fmtzbinit(Fmt *f, ZBlock *b)
89 7a4ee46d 2003-11-23 devnull {
90 7a4ee46d 2003-11-23 devnull f->runes = 0;
91 7a4ee46d 2003-11-23 devnull f->start = b->data;
92 7a4ee46d 2003-11-23 devnull f->to = f->start;
93 7a4ee46d 2003-11-23 devnull f->stop = (char*)f->start + b->len;
94 7a4ee46d 2003-11-23 devnull f->flush = nil;
95 7a4ee46d 2003-11-23 devnull f->farg = nil;
96 7a4ee46d 2003-11-23 devnull f->nfmt = 0;
97 7a4ee46d 2003-11-23 devnull f->args = nil;
98 7a4ee46d 2003-11-23 devnull }
99 7a4ee46d 2003-11-23 devnull
100 7a4ee46d 2003-11-23 devnull static int
101 7a4ee46d 2003-11-23 devnull sflush(Fmt *f)
102 7a4ee46d 2003-11-23 devnull {
103 7a4ee46d 2003-11-23 devnull char *s;
104 7a4ee46d 2003-11-23 devnull int n;
105 7a4ee46d 2003-11-23 devnull
106 7a4ee46d 2003-11-23 devnull n = (int)f->farg;
107 7a4ee46d 2003-11-23 devnull n += 256;
108 7a4ee46d 2003-11-23 devnull f->farg = (void*)n;
109 7a4ee46d 2003-11-23 devnull s = f->start;
110 7a4ee46d 2003-11-23 devnull f->start = realloc(s, n);
111 7a4ee46d 2003-11-23 devnull if(f->start == nil){
112 7a4ee46d 2003-11-23 devnull f->start = s;
113 7a4ee46d 2003-11-23 devnull return 0;
114 7a4ee46d 2003-11-23 devnull }
115 7a4ee46d 2003-11-23 devnull f->to = (char*)f->start + ((char*)f->to - s);
116 7a4ee46d 2003-11-23 devnull f->stop = (char*)f->start + n - 1;
117 7a4ee46d 2003-11-23 devnull return 1;
118 7a4ee46d 2003-11-23 devnull }
119 7a4ee46d 2003-11-23 devnull
120 7a4ee46d 2003-11-23 devnull static char*
121 7a4ee46d 2003-11-23 devnull logit(int severity, char *fmt, va_list args)
122 7a4ee46d 2003-11-23 devnull {
123 7a4ee46d 2003-11-23 devnull Fmt f;
124 7a4ee46d 2003-11-23 devnull int n;
125 7a4ee46d 2003-11-23 devnull
126 7a4ee46d 2003-11-23 devnull f.runes = 0;
127 7a4ee46d 2003-11-23 devnull n = 32;
128 7a4ee46d 2003-11-23 devnull f.start = malloc(n);
129 7a4ee46d 2003-11-23 devnull if(f.start == nil)
130 7a4ee46d 2003-11-23 devnull return nil;
131 7a4ee46d 2003-11-23 devnull f.to = f.start;
132 7a4ee46d 2003-11-23 devnull f.stop = (char*)f.start + n - 1;
133 7a4ee46d 2003-11-23 devnull f.flush = sflush;
134 7a4ee46d 2003-11-23 devnull f.farg = (void*)n;
135 7a4ee46d 2003-11-23 devnull f.nfmt = 0;
136 7a4ee46d 2003-11-23 devnull f.args = args;
137 7a4ee46d 2003-11-23 devnull n = dofmt(&f, fmt);
138 7a4ee46d 2003-11-23 devnull if(n < 0)
139 7a4ee46d 2003-11-23 devnull {
140 7a4ee46d 2003-11-23 devnull fprint(2, "dofmt %s failed\n", fmt);
141 7a4ee46d 2003-11-23 devnull return nil;
142 7a4ee46d 2003-11-23 devnull }
143 7a4ee46d 2003-11-23 devnull *(char*)f.to = '\0';
144 7a4ee46d 2003-11-23 devnull
145 7a4ee46d 2003-11-23 devnull if(argv0 == nil)
146 7a4ee46d 2003-11-23 devnull fprint(2, "%s: err %d: %s\n", argv0, severity, f.start);
147 7a4ee46d 2003-11-23 devnull else
148 7a4ee46d 2003-11-23 devnull fprint(2, "err %d: %s\n", severity, f.start);
149 7a4ee46d 2003-11-23 devnull return f.start;
150 7a4ee46d 2003-11-23 devnull }
151 7a4ee46d 2003-11-23 devnull
152 7a4ee46d 2003-11-23 devnull void
153 7a4ee46d 2003-11-23 devnull seterr(int severity, char *fmt, ...)
154 7a4ee46d 2003-11-23 devnull {
155 7a4ee46d 2003-11-23 devnull char *s;
156 7a4ee46d 2003-11-23 devnull va_list args;
157 7a4ee46d 2003-11-23 devnull
158 7a4ee46d 2003-11-23 devnull va_start(args, fmt);
159 7a4ee46d 2003-11-23 devnull s = logit(severity, fmt, args);
160 7a4ee46d 2003-11-23 devnull va_end(args);
161 7a4ee46d 2003-11-23 devnull if(s == nil)
162 7a4ee46d 2003-11-23 devnull werrstr("error setting error");
163 7a4ee46d 2003-11-23 devnull else{
164 7a4ee46d 2003-11-23 devnull werrstr("%s", s);
165 7a4ee46d 2003-11-23 devnull free(s);
166 7a4ee46d 2003-11-23 devnull }
167 7a4ee46d 2003-11-23 devnull }
168 7a4ee46d 2003-11-23 devnull
169 7a4ee46d 2003-11-23 devnull void
170 7a4ee46d 2003-11-23 devnull logerr(int severity, char *fmt, ...)
171 7a4ee46d 2003-11-23 devnull {
172 7a4ee46d 2003-11-23 devnull char *s;
173 7a4ee46d 2003-11-23 devnull va_list args;
174 7a4ee46d 2003-11-23 devnull
175 7a4ee46d 2003-11-23 devnull va_start(args, fmt);
176 7a4ee46d 2003-11-23 devnull s = logit(severity, fmt, args);
177 7a4ee46d 2003-11-23 devnull va_end(args);
178 7a4ee46d 2003-11-23 devnull free(s);
179 7a4ee46d 2003-11-23 devnull }
180 7a4ee46d 2003-11-23 devnull
181 7a4ee46d 2003-11-23 devnull u32int
182 7a4ee46d 2003-11-23 devnull now(void)
183 7a4ee46d 2003-11-23 devnull {
184 7a4ee46d 2003-11-23 devnull return time(nil);
185 7a4ee46d 2003-11-23 devnull }
186 7a4ee46d 2003-11-23 devnull
187 7a4ee46d 2003-11-23 devnull void
188 7a4ee46d 2003-11-23 devnull fatal(char *fmt, ...)
189 7a4ee46d 2003-11-23 devnull {
190 7a4ee46d 2003-11-23 devnull Fmt f;
191 7a4ee46d 2003-11-23 devnull char buf[256];
192 7a4ee46d 2003-11-23 devnull
193 7a4ee46d 2003-11-23 devnull f.runes = 0;
194 7a4ee46d 2003-11-23 devnull f.start = buf;
195 7a4ee46d 2003-11-23 devnull f.to = buf;
196 7a4ee46d 2003-11-23 devnull f.stop = buf + sizeof(buf);
197 7a4ee46d 2003-11-23 devnull f.flush = fmtfdflush;
198 7a4ee46d 2003-11-23 devnull f.farg = (void*)2;
199 7a4ee46d 2003-11-23 devnull f.nfmt = 0;
200 7a4ee46d 2003-11-23 devnull fmtprint(&f, "fatal %s error:", argv0);
201 7a4ee46d 2003-11-23 devnull va_start(f.args, fmt);
202 7a4ee46d 2003-11-23 devnull dofmt(&f, fmt);
203 7a4ee46d 2003-11-23 devnull va_end(f.args);
204 7a4ee46d 2003-11-23 devnull fmtprint(&f, "\n");
205 7a4ee46d 2003-11-23 devnull fmtfdflush(&f);
206 7a4ee46d 2003-11-23 devnull if(0)
207 7a4ee46d 2003-11-23 devnull abort();
208 7a4ee46d 2003-11-23 devnull threadexitsall(buf);
209 7a4ee46d 2003-11-23 devnull }
210 7a4ee46d 2003-11-23 devnull
211 7a4ee46d 2003-11-23 devnull ZBlock *
212 7a4ee46d 2003-11-23 devnull alloczblock(u32int size, int zeroed)
213 7a4ee46d 2003-11-23 devnull {
214 7a4ee46d 2003-11-23 devnull ZBlock *b;
215 7a4ee46d 2003-11-23 devnull static ZBlock z;
216 7a4ee46d 2003-11-23 devnull
217 7a4ee46d 2003-11-23 devnull b = malloc(sizeof(ZBlock) + size);
218 7a4ee46d 2003-11-23 devnull if(b == nil){
219 7a4ee46d 2003-11-23 devnull seterr(EOk, "out of memory");
220 7a4ee46d 2003-11-23 devnull return nil;
221 7a4ee46d 2003-11-23 devnull }
222 7a4ee46d 2003-11-23 devnull
223 7a4ee46d 2003-11-23 devnull *b = z;
224 7a4ee46d 2003-11-23 devnull b->data = (u8int*)&b[1];
225 7a4ee46d 2003-11-23 devnull b->len = size;
226 7a4ee46d 2003-11-23 devnull if(zeroed)
227 7a4ee46d 2003-11-23 devnull memset(b->data, 0, size);
228 7a4ee46d 2003-11-23 devnull return b;
229 7a4ee46d 2003-11-23 devnull }
230 7a4ee46d 2003-11-23 devnull
231 7a4ee46d 2003-11-23 devnull void
232 7a4ee46d 2003-11-23 devnull freezblock(ZBlock *b)
233 7a4ee46d 2003-11-23 devnull {
234 7a4ee46d 2003-11-23 devnull free(b);
235 7a4ee46d 2003-11-23 devnull }
236 7a4ee46d 2003-11-23 devnull
237 7a4ee46d 2003-11-23 devnull ZBlock*
238 7a4ee46d 2003-11-23 devnull packet2zblock(Packet *p, u32int size)
239 7a4ee46d 2003-11-23 devnull {
240 7a4ee46d 2003-11-23 devnull ZBlock *b;
241 7a4ee46d 2003-11-23 devnull
242 7a4ee46d 2003-11-23 devnull if(p == nil)
243 7a4ee46d 2003-11-23 devnull return nil;
244 7a4ee46d 2003-11-23 devnull b = alloczblock(size, 0);
245 7a4ee46d 2003-11-23 devnull if(b == nil)
246 7a4ee46d 2003-11-23 devnull return nil;
247 7a4ee46d 2003-11-23 devnull b->len = size;
248 7a4ee46d 2003-11-23 devnull if(packetcopy(p, b->data, 0, size) < 0){
249 7a4ee46d 2003-11-23 devnull freezblock(b);
250 7a4ee46d 2003-11-23 devnull return nil;
251 7a4ee46d 2003-11-23 devnull }
252 7a4ee46d 2003-11-23 devnull return b;
253 7a4ee46d 2003-11-23 devnull }
254 7a4ee46d 2003-11-23 devnull
255 7a4ee46d 2003-11-23 devnull Packet*
256 7a4ee46d 2003-11-23 devnull zblock2packet(ZBlock *zb, u32int size)
257 7a4ee46d 2003-11-23 devnull {
258 7a4ee46d 2003-11-23 devnull Packet *p;
259 7a4ee46d 2003-11-23 devnull
260 7a4ee46d 2003-11-23 devnull if(zb == nil)
261 7a4ee46d 2003-11-23 devnull return nil;
262 7a4ee46d 2003-11-23 devnull p = packetalloc();
263 7a4ee46d 2003-11-23 devnull packetappend(p, zb->data, size);
264 7a4ee46d 2003-11-23 devnull return p;
265 7a4ee46d 2003-11-23 devnull }
266 7a4ee46d 2003-11-23 devnull
267 7a4ee46d 2003-11-23 devnull void *
268 7a4ee46d 2003-11-23 devnull emalloc(ulong n)
269 7a4ee46d 2003-11-23 devnull {
270 7a4ee46d 2003-11-23 devnull void *p;
271 7a4ee46d 2003-11-23 devnull
272 7a4ee46d 2003-11-23 devnull p = malloc(n);
273 7a4ee46d 2003-11-23 devnull if(p == nil)
274 7a4ee46d 2003-11-23 devnull sysfatal("out of memory");
275 7a4ee46d 2003-11-23 devnull memset(p, 0xa5, n);
276 7a4ee46d 2003-11-23 devnull if(0)print("emalloc %p-%p by %lux\n", p, (char*)p+n, getcallerpc(&n));
277 7a4ee46d 2003-11-23 devnull return p;
278 7a4ee46d 2003-11-23 devnull }
279 7a4ee46d 2003-11-23 devnull
280 7a4ee46d 2003-11-23 devnull void *
281 7a4ee46d 2003-11-23 devnull ezmalloc(ulong n)
282 7a4ee46d 2003-11-23 devnull {
283 7a4ee46d 2003-11-23 devnull void *p;
284 7a4ee46d 2003-11-23 devnull
285 7a4ee46d 2003-11-23 devnull p = malloc(n);
286 7a4ee46d 2003-11-23 devnull if(p == nil)
287 7a4ee46d 2003-11-23 devnull sysfatal("out of memory");
288 7a4ee46d 2003-11-23 devnull memset(p, 0, n);
289 7a4ee46d 2003-11-23 devnull if(0)print("ezmalloc %p-%p by %lux\n", p, (char*)p+n, getcallerpc(&n));
290 7a4ee46d 2003-11-23 devnull return p;
291 7a4ee46d 2003-11-23 devnull }
292 7a4ee46d 2003-11-23 devnull
293 7a4ee46d 2003-11-23 devnull void *
294 7a4ee46d 2003-11-23 devnull erealloc(void *p, ulong n)
295 7a4ee46d 2003-11-23 devnull {
296 7a4ee46d 2003-11-23 devnull p = realloc(p, n);
297 7a4ee46d 2003-11-23 devnull if(p == nil)
298 7a4ee46d 2003-11-23 devnull sysfatal("out of memory");
299 7a4ee46d 2003-11-23 devnull if(0)print("erealloc %p-%p by %lux\n", p, (char*)p+n, getcallerpc(&p));
300 7a4ee46d 2003-11-23 devnull return p;
301 7a4ee46d 2003-11-23 devnull }
302 7a4ee46d 2003-11-23 devnull
303 7a4ee46d 2003-11-23 devnull char *
304 7a4ee46d 2003-11-23 devnull estrdup(char *s)
305 7a4ee46d 2003-11-23 devnull {
306 7a4ee46d 2003-11-23 devnull char *t;
307 7a4ee46d 2003-11-23 devnull int n;
308 7a4ee46d 2003-11-23 devnull
309 7a4ee46d 2003-11-23 devnull n = strlen(s) + 1;
310 7a4ee46d 2003-11-23 devnull t = emalloc(n);
311 7a4ee46d 2003-11-23 devnull memmove(t, s, n);
312 7a4ee46d 2003-11-23 devnull if(0)print("estrdup %p-%p by %lux\n", t, (char*)t+n, getcallerpc(&s));
313 7a4ee46d 2003-11-23 devnull return t;
314 7a4ee46d 2003-11-23 devnull }
315 7a4ee46d 2003-11-23 devnull
316 7a4ee46d 2003-11-23 devnull ZBlock*
317 7a4ee46d 2003-11-23 devnull readfile(char *name)
318 7a4ee46d 2003-11-23 devnull {
319 7a4ee46d 2003-11-23 devnull Part *p;
320 7a4ee46d 2003-11-23 devnull ZBlock *b;
321 7a4ee46d 2003-11-23 devnull
322 7a4ee46d 2003-11-23 devnull p = initpart(name, 1);
323 7a4ee46d 2003-11-23 devnull if(p == nil)
324 7a4ee46d 2003-11-23 devnull return nil;
325 7a4ee46d 2003-11-23 devnull b = alloczblock(p->size, 0);
326 7a4ee46d 2003-11-23 devnull if(b == nil){
327 7a4ee46d 2003-11-23 devnull seterr(EOk, "can't alloc %s: %r", name);
328 7a4ee46d 2003-11-23 devnull freepart(p);
329 7a4ee46d 2003-11-23 devnull return nil;
330 7a4ee46d 2003-11-23 devnull }
331 7a4ee46d 2003-11-23 devnull if(readpart(p, 0, b->data, p->size) < 0){
332 7a4ee46d 2003-11-23 devnull seterr(EOk, "can't read %s: %r", name);
333 7a4ee46d 2003-11-23 devnull freepart(p);
334 7a4ee46d 2003-11-23 devnull freezblock(b);
335 7a4ee46d 2003-11-23 devnull return nil;
336 7a4ee46d 2003-11-23 devnull }
337 7a4ee46d 2003-11-23 devnull freepart(p);
338 7a4ee46d 2003-11-23 devnull return b;
339 7a4ee46d 2003-11-23 devnull }
340 7a4ee46d 2003-11-23 devnull
341 7a4ee46d 2003-11-23 devnull /*
342 7a4ee46d 2003-11-23 devnull * return floor(log2(v))
343 7a4ee46d 2003-11-23 devnull */
344 7a4ee46d 2003-11-23 devnull int
345 7a4ee46d 2003-11-23 devnull u64log2(u64int v)
346 7a4ee46d 2003-11-23 devnull {
347 7a4ee46d 2003-11-23 devnull int i;
348 7a4ee46d 2003-11-23 devnull
349 7a4ee46d 2003-11-23 devnull for(i = 0; i < 64; i++)
350 7a4ee46d 2003-11-23 devnull if((v >> i) <= 1)
351 7a4ee46d 2003-11-23 devnull break;
352 7a4ee46d 2003-11-23 devnull return i;
353 7a4ee46d 2003-11-23 devnull }
354 7a4ee46d 2003-11-23 devnull
355 7a4ee46d 2003-11-23 devnull int
356 7a4ee46d 2003-11-23 devnull vtproc(void (*fn)(void*), void *arg)
357 7a4ee46d 2003-11-23 devnull {
358 7a4ee46d 2003-11-23 devnull proccreate(fn, arg, 256*1024);
359 7a4ee46d 2003-11-23 devnull return 0;
360 7a4ee46d 2003-11-23 devnull }
361 7a4ee46d 2003-11-23 devnull