]> diplodocus.org Git - nmh/blob - uip/fmtdump.c
lock_file.c: close(2) file descriptor on failure, avoiding leak.
[nmh] / uip / fmtdump.c
1 /* fmtdump.c -- compile format file and dump out instructions
2 *
3 * This code is Copyright (c) 2002, by the authors of nmh. See the
4 * COPYRIGHT file in the root directory of the nmh distribution for
5 * complete copyright information.
6 */
7
8 #include <h/mh.h>
9 #include <h/fmt_scan.h>
10 #include <h/fmt_compile.h>
11 #include <h/scansbr.h>
12 #include <h/utils.h>
13
14 #define FMTDUMP_SWITCHES \
15 X("form formatfile", 0, FORMSW) \
16 X("format string", 5, FMTSW) \
17 X("version", 0, VERSIONSW) \
18 X("help", 0, HELPSW) \
19
20 #define X(sw, minchars, id) id,
21 DEFINE_SWITCH_ENUM(FMTDUMP);
22 #undef X
23
24 #define X(sw, minchars, id) { sw, minchars, id },
25 DEFINE_SWITCH_ARRAY(FMTDUMP, switches);
26 #undef X
27
28 /* for assignlabel */
29 static struct format *lvec[128];
30 static int lused = 0;
31
32 /*
33 * static prototypes
34 */
35 static void fmt_dump (struct format *);
36 static void dumpone(struct format *);
37 static int findlabel(struct format *);
38 static void assignlabel(struct format *);
39 static char *f_typestr(int);
40 static char *c_typestr(int);
41 static char *c_flagsstr(int);
42 static void litputs(char *);
43 static void litputc(char);
44
45
46 int
47 main (int argc, char **argv)
48 {
49 char *cp, *form = NULL, *format = NULL;
50 char buf[BUFSIZ], *nfs, **argp, **arguments;
51 struct format *fmt;
52
53 if (nmh_init(argv[0], 2)) { return 1; }
54
55 arguments = getarguments (invo_name, argc, argv, 1);
56 argp = arguments;
57
58 while ((cp = *argp++)) {
59 if (*cp == '-') {
60 switch (smatch (++cp, switches)) {
61 case AMBIGSW:
62 ambigsw (cp, switches);
63 done (1);
64 case UNKWNSW:
65 adios (NULL, "-%s unknown", cp);
66
67 case HELPSW:
68 snprintf (buf, sizeof(buf), "%s [switches]", invo_name);
69 print_help (buf, switches, 1);
70 done (0);
71 case VERSIONSW:
72 print_version(invo_name);
73 done (0);
74
75 case FORMSW:
76 if (!(form = *argp++) || *form == '-')
77 adios (NULL, "missing argument to %s", argp[-2]);
78 format = NULL;
79 continue;
80 case FMTSW:
81 if (!(format = *argp++) || *format == '-')
82 adios (NULL, "missing argument to %s", argp[-2]);
83 form = NULL;
84 continue;
85
86 }
87 }
88 if (form)
89 adios (NULL, "only one form at a time!");
90 form = cp;
91 }
92
93 /*
94 * Get new format string. Must be before chdir().
95 */
96 nfs = new_fs (form, format, FORMAT);
97 (void) fmt_compile(nfs, &fmt, 1);
98
99 fmt_dump(fmt);
100
101 fmt_free(fmt, 1);
102
103 done(0);
104 return 1;
105 }
106
107 static void
108 fmt_dump (struct format *fmth)
109 {
110 int i;
111 struct format *fmt, *addr;
112
113 /* Assign labels */
114 for (fmt = fmth; fmt; ++fmt) {
115 i = fmt->f_type;
116 if (i == FT_IF_S ||
117 i == FT_IF_S_NULL ||
118 i == FT_IF_V_EQ ||
119 i == FT_IF_V_NE ||
120 i == FT_IF_V_GT ||
121 i == FT_IF_MATCH ||
122 i == FT_IF_AMATCH ||
123 i == FT_GOTO) {
124 addr = fmt + fmt->f_skip;
125 if (findlabel(addr) < 0)
126 assignlabel(addr);
127 }
128 if (fmt->f_type == FT_DONE && fmt->f_value == 0)
129 break;
130 }
131
132 /* Dump them out! */
133 for (fmt = fmth; fmt; ++fmt) {
134 dumpone(fmt);
135 if (fmt->f_type == FT_DONE && fmt->f_value == 0)
136 break;
137 }
138 }
139
140 static void
141 dumpone(struct format *fmt)
142 {
143 int i;
144
145 if ((i = findlabel(fmt)) >= 0)
146 printf("L%d:", i);
147 putchar('\t');
148
149 fputs(f_typestr((int)fmt->f_type), stdout);
150
151 switch (fmt->f_type) {
152
153 case FT_COMP:
154 case FT_LS_COMP:
155 case FT_LV_COMPFLAG:
156 case FT_LV_COMP:
157 printf(", comp ");
158 litputs(fmt->f_comp->c_name);
159 if (fmt->f_comp->c_type)
160 printf(", c_type %s", c_typestr(fmt->f_comp->c_type));
161 if (fmt->f_comp->c_flags)
162 printf(", c_flags %s", c_flagsstr(fmt->f_comp->c_flags));
163 break;
164
165 case FT_LV_SEC:
166 case FT_LV_MIN:
167 case FT_LV_HOUR:
168 case FT_LV_MDAY:
169 case FT_LV_MON:
170 case FT_LS_MONTH:
171 case FT_LS_LMONTH:
172 case FT_LS_ZONE:
173 case FT_LV_YEAR:
174 case FT_LV_WDAY:
175 case FT_LS_DAY:
176 case FT_LS_WEEKDAY:
177 case FT_LV_YDAY:
178 case FT_LV_ZONE:
179 case FT_LV_CLOCK:
180 case FT_LV_RCLOCK:
181 case FT_LV_DAYF:
182 case FT_LV_ZONEF:
183 case FT_LV_DST:
184 case FT_LS_822DATE:
185 case FT_LS_PRETTY:
186 case FT_LOCALDATE:
187 case FT_GMTDATE:
188 case FT_PARSEDATE:
189 printf(", c_name ");
190 litputs(fmt->f_comp->c_name);
191 if (fmt->f_comp->c_type)
192 printf(", c_type %s", c_typestr(fmt->f_comp->c_type));
193 if (fmt->f_comp->c_flags)
194 printf(", c_flags %s", c_flagsstr(fmt->f_comp->c_flags));
195 break;
196
197 case FT_LS_ADDR:
198 case FT_LS_PERS:
199 case FT_LS_MBOX:
200 case FT_LS_HOST:
201 case FT_LS_PATH:
202 case FT_LS_GNAME:
203 case FT_LS_NOTE:
204 case FT_LS_822ADDR:
205 case FT_LV_HOSTTYPE:
206 case FT_LV_INGRPF:
207 case FT_LV_NOHOSTF:
208 case FT_LS_FRIENDLY:
209 case FT_PARSEADDR:
210 case FT_MYMBOX:
211 case FT_GETMYMBOX:
212 case FT_GETMYADDR:
213 printf(", c_name ");
214 litputs(fmt->f_comp->c_name);
215 if (fmt->f_comp->c_type)
216 printf(", c_type %s", c_typestr(fmt->f_comp->c_type));
217 if (fmt->f_comp->c_flags)
218 printf(", c_flags %s", c_flagsstr(fmt->f_comp->c_flags));
219 break;
220
221 case FT_COMPF:
222 printf(", width %d, fill '", fmt->f_width);
223 litputc(fmt->f_fill);
224 printf("' name ");
225 litputs(fmt->f_comp->c_name);
226 if (fmt->f_comp->c_type)
227 printf(", c_type %s", c_typestr(fmt->f_comp->c_type));
228 if (fmt->f_comp->c_flags)
229 printf(", c_flags %s", c_flagsstr(fmt->f_comp->c_flags));
230 break;
231
232 case FT_STRF:
233 case FT_NUMF:
234 printf(", width %d, fill '", fmt->f_width);
235 litputc(fmt->f_fill);
236 putchar('\'');
237 break;
238
239 case FT_LIT:
240 putchar(' ');
241 litputs(fmt->f_text);
242 break;
243
244 case FT_LITF:
245 printf(", width %d, fill '", fmt->f_width);
246 litputc(fmt->f_fill);
247 printf("' ");
248 litputs(fmt->f_text);
249 break;
250
251 case FT_CHAR:
252 putchar(' ');
253 putchar('\'');
254 litputc(fmt->f_char);
255 putchar('\'');
256 break;
257
258
259 case FT_IF_S:
260 case FT_IF_S_NULL:
261 case FT_IF_MATCH:
262 case FT_IF_AMATCH:
263 printf(" continue else goto");
264 /* FALLTHRU */
265 case FT_GOTO:
266 i = findlabel(fmt + fmt->f_skip);
267 printf(" L%d", i);
268 break;
269
270 case FT_IF_V_EQ:
271 case FT_IF_V_NE:
272 case FT_IF_V_GT:
273 i = findlabel(fmt + fmt->f_skip);
274 printf(" %d continue else goto L%d", fmt->f_value, i);
275 break;
276
277 case FT_V_EQ:
278 case FT_V_NE:
279 case FT_V_GT:
280 case FT_LV_LIT:
281 case FT_LV_PLUS_L:
282 case FT_LV_MINUS_L:
283 case FT_LV_MULTIPLY_L:
284 case FT_LV_DIVIDE_L:
285 case FT_LV_MODULO_L:
286 printf(" value %d", fmt->f_value);
287 break;
288
289 case FT_LS_LIT:
290 printf(" str ");
291 litputs(fmt->f_text);
292 break;
293
294 case FT_LS_GETENV:
295 printf(" getenv ");
296 litputs(fmt->f_text);
297 break;
298
299 case FT_LS_DECODECOMP:
300 printf(", comp ");
301 litputs(fmt->f_comp->c_name);
302 break;
303
304 case FT_LS_DECODE:
305 break;
306
307 case FT_LS_TRIM:
308 printf(", width %d", fmt->f_width);
309 break;
310
311 case FT_LV_DAT:
312 printf(", value dat[%d]", fmt->f_value);
313 break;
314 }
315 putchar('\n');
316 }
317
318 static int
319 findlabel(struct format *addr)
320 {
321 int i;
322
323 for (i = 0; i < lused; ++i)
324 if (addr == lvec[i])
325 return(i);
326 return(-1);
327 }
328
329 static void
330 assignlabel(struct format *addr)
331 {
332 lvec[lused++] = addr;
333 }
334
335 static char *
336 f_typestr(int t)
337 {
338 static char buf[32];
339
340 switch (t) {
341 case FT_COMP: return "COMP";
342 case FT_COMPF: return "COMPF";
343 case FT_LIT: return "LIT";
344 case FT_LITF: return "LITF";
345 case FT_CHAR: return "CHAR";
346 case FT_NUM: return "NUM";
347 case FT_NUMF: return "NUMF";
348 case FT_STR: return "STR";
349 case FT_STRF: return "STRF";
350 case FT_STRFW: return "STRFW";
351 case FT_STRLIT: return "STRLIT";
352 case FT_STRLITZ: return "STRLITZ";
353 case FT_PUTADDR: return "PUTADDR";
354 case FT_LS_COMP: return "LS_COMP";
355 case FT_LS_LIT: return "LS_LIT";
356 case FT_LS_GETENV: return "LS_GETENV";
357 case FT_LS_CFIND: return "LS_CFIND";
358 case FT_LS_DECODECOMP: return "LS_DECODECOMP";
359 case FT_LS_DECODE: return "LS_DECODE";
360 case FT_LS_TRIM: return "LS_TRIM";
361 case FT_LV_COMP: return "LV_COMP";
362 case FT_LV_COMPFLAG: return "LV_COMPFLAG";
363 case FT_LV_LIT: return "LV_LIT";
364 case FT_LV_DAT: return "LV_DAT";
365 case FT_LV_STRLEN: return "LV_STRLEN";
366 case FT_LV_PLUS_L: return "LV_PLUS_L";
367 case FT_LV_MINUS_L: return "LV_MINUS_L";
368 case FT_LV_MULTIPLY_L: return "LV_MULTIPLY_L";
369 case FT_LV_DIVIDE_L: return "LV_DIVIDE_L";
370 case FT_LV_MODULO_L: return "LV_MODULO_L";
371 case FT_LV_CHAR_LEFT: return "LV_CHAR_LEFT";
372 case FT_LS_MONTH: return "LS_MONTH";
373 case FT_LS_LMONTH: return "LS_LMONTH";
374 case FT_LS_ZONE: return "LS_ZONE";
375 case FT_LS_DAY: return "LS_DAY";
376 case FT_LS_WEEKDAY: return "LS_WEEKDAY";
377 case FT_LS_822DATE: return "LS_822DATE";
378 case FT_LS_PRETTY: return "LS_PRETTY";
379 case FT_LS_KILO: return "LS_KILO";
380 case FT_LS_KIBI: return "LS_KIBI";
381 case FT_LV_SEC: return "LV_SEC";
382 case FT_LV_MIN: return "LV_MIN";
383 case FT_LV_HOUR: return "LV_HOUR";
384 case FT_LV_MDAY: return "LV_MDAY";
385 case FT_LV_MON: return "LV_MON";
386 case FT_LV_YEAR: return "LV_YEAR";
387 case FT_LV_YDAY: return "LV_YDAY";
388 case FT_LV_WDAY: return "LV_WDAY";
389 case FT_LV_ZONE: return "LV_ZONE";
390 case FT_LV_CLOCK: return "LV_CLOCK";
391 case FT_LV_RCLOCK: return "LV_RCLOCK";
392 case FT_LV_DAYF: return "LV_DAYF";
393 case FT_LV_DST: return "LV_DST";
394 case FT_LV_ZONEF: return "LV_ZONEF";
395 case FT_LS_PERS: return "LS_PERS";
396 case FT_LS_MBOX: return "LS_MBOX";
397 case FT_LS_HOST: return "LS_HOST";
398 case FT_LS_PATH: return "LS_PATH";
399 case FT_LS_GNAME: return "LS_GNAME";
400 case FT_LS_NOTE: return "LS_NOTE";
401 case FT_LS_ADDR: return "LS_ADDR";
402 case FT_LS_822ADDR: return "LS_822ADDR";
403 case FT_LS_FRIENDLY: return "LS_FRIENDLY";
404 case FT_LV_HOSTTYPE: return "LV_HOSTTYPE";
405 case FT_LV_INGRPF: return "LV_INGRPF";
406 case FT_LS_UNQUOTE: return "LS_UNQUOTE";
407 case FT_LV_NOHOSTF: return "LV_NOHOSTF";
408 case FT_LOCALDATE: return "LOCALDATE";
409 case FT_GMTDATE: return "GMTDATE";
410 case FT_PARSEDATE: return "PARSEDATE";
411 case FT_PARSEADDR: return "PARSEADDR";
412 case FT_FORMATADDR: return "FORMATADDR";
413 case FT_CONCATADDR: return "CONCATADDR";
414 case FT_MYMBOX: return "MYMBOX";
415 case FT_GETMYMBOX: return "GETMYMBOX";
416 case FT_GETMYADDR: return "GETMYADDR";
417 case FT_SAVESTR: return "SAVESTR";
418 case FT_DONE: return "DONE";
419 case FT_PAUSE: return "PAUSE";
420 case FT_NOP: return "NOP";
421 case FT_GOTO: return "GOTO";
422 case FT_IF_S_NULL: return "IF_S_NULL";
423 case FT_IF_S: return "IF_S";
424 case FT_IF_V_EQ: return "IF_V_EQ";
425 case FT_IF_V_NE: return "IF_V_NE";
426 case FT_IF_V_GT: return "IF_V_GT";
427 case FT_IF_MATCH: return "IF_MATCH";
428 case FT_IF_AMATCH: return "IF_AMATCH";
429 case FT_S_NULL: return "S_NULL";
430 case FT_S_NONNULL: return "S_NONNULL";
431 case FT_V_EQ: return "V_EQ";
432 case FT_V_NE: return "V_NE";
433 case FT_V_GT: return "V_GT";
434 case FT_V_MATCH: return "V_MATCH";
435 case FT_V_AMATCH: return "V_AMATCH";
436 default:
437 snprintf(buf, sizeof buf, "/* ??? #%d */", t);
438 return(buf);
439 }
440 }
441
442 #define FNORD(v, s) if (t & (v)) { \
443 if (i++ > 0) \
444 strcat(buf, "|"); \
445 strcat(buf, s); }
446
447 static char *
448 c_typestr(int t)
449 {
450 int i;
451 static char buf[64];
452
453 buf[0] = '\0';
454 if (t & ~(CT_ADDR|CT_DATE))
455 snprintf(buf, sizeof buf, "0x%x ", t);
456 strcat(buf, "<");
457 i = 0;
458 FNORD(CT_ADDR, "ADDR");
459 FNORD(CT_DATE, "DATE");
460 strcat(buf, ">");
461 return(buf);
462 }
463
464 static char *
465 c_flagsstr(int t)
466 {
467 int i;
468 static char buf[64];
469
470 buf[0] = '\0';
471 if (t & ~(CF_TRUE|CF_PARSED|CF_DATEFAB|CF_TRIMMED))
472 snprintf(buf, sizeof buf, "0x%x ", t);
473 strcat(buf, "<");
474 i = 0;
475 FNORD(CF_TRUE, "TRUE");
476 FNORD(CF_PARSED, "PARSED");
477 FNORD(CF_DATEFAB, "DATEFAB");
478 FNORD(CF_TRIMMED, "TRIMMED");
479 strcat(buf, ">");
480 return(buf);
481 }
482 #undef FNORD
483
484 static void
485 litputs(char *s)
486 {
487 if (s) {
488 putchar('"');
489 while (*s)
490 litputc(*s++);
491 putchar('"');
492 } else
493 fputs("<nil>", stdout);
494 }
495
496 static void
497 litputc(char c)
498 {
499 if (c & ~ 0177) {
500 putchar('M');
501 putchar('-');
502 c &= 0177;
503 }
504 if (c < 0x20 || c == 0177) {
505 if (c == '\b') {
506 putchar('\\');
507 putchar('b');
508 } else if (c == '\f') {
509 putchar('\\');
510 putchar('f');
511 } else if (c == '\n') {
512 putchar('\\');
513 putchar('n');
514 } else if (c == '\r') {
515 putchar('\\');
516 putchar('r');
517 } else if (c == '\t') {
518 putchar('\\');
519 putchar('t');
520 } else {
521 putchar('^');
522 putchar(c ^ 0x40); /* DEL to ?, others to alpha */
523 }
524 } else
525 putchar(c);
526 }