Annotation of mandoc/main.c, Revision 1.4
1.4 ! kristaps 1: /* $Id: main.c,v 1.3 2009/03/19 16:40:49 kristaps Exp $ */
1.1 kristaps 2: /*
3: * Copyright (c) 2008, 2009 Kristaps Dzonsons <kristaps@openbsd.org>
4: *
5: * Permission to use, copy, modify, and distribute this software for any
6: * purpose with or without fee is hereby granted, provided that the
7: * above copyright notice and this permission notice appear in all
8: * copies.
9: *
10: * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL
11: * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
12: * WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE
13: * AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
14: * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
15: * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
16: * TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
17: * PERFORMANCE OF THIS SOFTWARE.
18: */
19: #include <sys/stat.h>
20:
21: #include <assert.h>
22: #include <err.h>
23: #include <fcntl.h>
24: #include <stdio.h>
25: #include <stdlib.h>
26: #include <string.h>
27: #include <unistd.h>
28:
29: #include "mdoc.h"
30:
1.3 kristaps 31: #ifdef __linux__
32: extern int getsubopt(char **, char * const *, char **);
33: # ifndef __dead
34: # define __dead __attribute__((__noreturn__))
35: # endif
36: #endif
37:
1.1 kristaps 38: #define WARN_WALL 0x03 /* All-warnings mask. */
39: #define WARN_WCOMPAT (1 << 0) /* Compatibility warnings. */
40: #define WARN_WSYNTAX (1 << 1) /* Syntax warnings. */
41: #define WARN_WERR (1 << 2) /* Warnings->errors. */
42:
43: enum outt {
44: OUTT_ASCII,
45: OUTT_LATIN1,
46: OUTT_UTF8,
47: OUTT_TREE,
48: OUTT_LINT
49: };
50:
51: typedef int (*out_run)(void *, const struct mdoc *);
52: typedef void (*out_free)(void *);
53:
54: extern char *__progname;
55:
56: extern void *ascii_alloc(void);
57: extern void *latin1_alloc(void);
58: extern void *utf8_alloc(void);
59: extern int terminal_run(void *, const struct mdoc *);
60: extern int tree_run(void *, const struct mdoc *);
61: extern void terminal_free(void *);
62:
63: __dead static void version(void);
64: __dead static void usage(void);
65: static int foptions(int *, char *);
66: static int toptions(enum outt *, char *);
67: static int woptions(int *, char *);
68: static int merr(void *, int, int, const char *);
69: static int mwarn(void *, int, int,
70: enum mdoc_warn, const char *);
71: static int file(char **, size_t *, char **, size_t *,
72: const char *, struct mdoc *);
73: static int fdesc(char **, size_t *, char **, size_t *,
74: const char *, int, struct mdoc *);
75:
76:
77: int
78: main(int argc, char *argv[])
79: {
80: int c, rc, fflags, wflags;
81: struct mdoc_cb cb;
82: struct mdoc *mdoc;
83: char *buf, *line;
84: size_t bufsz, linesz;
85: void *outdata;
86: enum outt outtype;
87: out_run outrun;
88: out_free outfree;
89:
90: fflags = wflags = 0;
91: outtype = OUTT_ASCII;
92:
93: /* LINTED */
94: while (-1 != (c = getopt(argc, argv, "f:VW:T:")))
95: switch (c) {
96: case ('f'):
97: if ( ! foptions(&fflags, optarg))
98: return(0);
99: break;
100: case ('T'):
101: if ( ! toptions(&outtype, optarg))
102: return(0);
103: break;
104: case ('W'):
105: if ( ! woptions(&wflags, optarg))
106: return(0);
107: break;
108: case ('V'):
109: version();
110: /* NOTREACHED */
111: default:
112: usage();
113: /* NOTREACHED */
114: }
115:
116: argc -= optind;
117: argv += optind;
118:
119: /*
120: * Allocate the appropriate front-end. Note that utf8, ascii
121: * and latin1 all resolve to the terminal front-end with
122: * different encodings (see terminal.c). Not all frontends have
123: * cleanup or alloc routines.
124: */
125:
126: switch (outtype) {
127: case (OUTT_LATIN1):
128: outdata = latin1_alloc();
129: outrun = terminal_run;
130: outfree = terminal_free;
131: break;
132: case (OUTT_UTF8):
133: outdata = utf8_alloc();
134: outrun = terminal_run;
135: outfree = terminal_free;
136: break;
137: case (OUTT_TREE):
138: outdata = NULL;
139: outrun = tree_run;
140: outfree = NULL;
141: break;
142: case (OUTT_LINT):
143: outdata = NULL;
144: outrun = NULL;
145: outfree = NULL;
146: break;
147: default:
148: outdata = ascii_alloc();
149: outrun = terminal_run;
150: outfree = terminal_free;
151: break;
152: }
153:
154: /*
155: * All callbacks route into here, where we print them onto the
156: * screen. XXX - for now, no path for debugging messages.
157: */
158:
159: cb.mdoc_msg = NULL;
160: cb.mdoc_err = merr;
161: cb.mdoc_warn = mwarn;
162:
163: buf = line = NULL;
164: bufsz = linesz = 0;
165:
166: mdoc = mdoc_alloc(&wflags, fflags, &cb);
167:
1.4 ! kristaps 168: /*
! 169: * Loop around available files.
! 170: */
1.1 kristaps 171:
1.4 ! kristaps 172: if (NULL == *argv) {
! 173: c = fdesc(&line, &linesz, &buf, &bufsz,
! 174: "stdin", STDIN_FILENO, mdoc);
! 175: rc = 0;
! 176: if (c && NULL == outrun)
! 177: rc = 1;
! 178: else if (c && outrun && (*outrun)(outdata, mdoc))
! 179: rc = 1;
! 180: } else {
! 181: while (*argv) {
! 182: c = file(&line, &linesz, &buf,
! 183: &bufsz, *argv, mdoc);
! 184: if ( ! c)
! 185: break;
! 186: if (outrun && ! (*outrun)(outdata, mdoc))
! 187: break;
! 188: /* Reset the parser for another file. */
! 189: mdoc_reset(mdoc);
! 190: argv++;
! 191: }
! 192: rc = NULL == *argv;
1.1 kristaps 193: }
194:
195: if (buf)
196: free(buf);
197: if (line)
198: free(line);
199: if (outfree)
200: (*outfree)(outdata);
201:
202: mdoc_free(mdoc);
203:
204: return(rc ? EXIT_SUCCESS : EXIT_FAILURE);
205: }
206:
207:
208: __dead static void
209: version(void)
210: {
211:
212: (void)printf("%s %s\n", __progname, VERSION);
213: exit(0);
214: /* NOTREACHED */
215: }
216:
217:
218: __dead static void
219: usage(void)
220: {
221:
222: (void)fprintf(stderr, "usage: %s\n", __progname);
223: exit(1);
224: /* NOTREACHED */
225: }
226:
227:
228: static int
229: file(char **ln, size_t *lnsz, char **buf, size_t *bufsz,
230: const char *file, struct mdoc *mdoc)
231: {
232: int fd, c;
233:
234: if (-1 == (fd = open(file, O_RDONLY, 0))) {
235: warn("%s", file);
236: return(0);
237: }
238:
239: c = fdesc(ln, lnsz, buf, bufsz, file, fd, mdoc);
240:
241: if (-1 == close(fd))
242: warn("%s", file);
243:
244: return(c);
245: }
246:
247:
248: static int
249: fdesc(char **lnp, size_t *lnsz, char **bufp, size_t *bufsz,
250: const char *f, int fd, struct mdoc *mdoc)
251: {
252: size_t sz;
253: ssize_t ssz;
254: struct stat st;
255: int j, i, pos, lnn;
256: char *ln, *buf;
257:
258: buf = *bufp;
259: ln = *lnp;
260:
261: /*
262: * Two buffers: ln and buf. buf is the input buffer, optimised
263: * for each file's block size. ln is a line buffer. Both
264: * growable, hence passed in by ptr-ptr.
265: */
266:
267: if (-1 == fstat(fd, &st)) {
268: warnx("%s", f);
269: sz = BUFSIZ;
270: } else
271: sz = (unsigned)BUFSIZ > st.st_blksize ?
272: (size_t)BUFSIZ : st.st_blksize;
273:
274: if (sz > *bufsz) {
275: if (NULL == (buf = realloc(buf, sz)))
276: err(1, "realloc");
277: *bufp = buf;
278: *bufsz = sz;
279: }
280:
281: /*
282: * Fill buf with file blocksize and parse newlines into ln.
283: */
284:
285: for (lnn = 1, pos = 0; ; ) {
286: if (-1 == (ssz = read(fd, buf, sz))) {
287: warn("%s", f);
288: return(0);
289: } else if (0 == ssz)
290: break;
291:
292: for (i = 0; i < (int)ssz; i++) {
293: if (pos >= (int)*lnsz) {
294: *lnsz += 256; /* Step-size. */
295: ln = realloc(ln, *lnsz);
296: if (NULL == ln)
297: err(1, "realloc");
298: *lnp = ln;
299: }
300:
301: if ('\n' != buf[i]) {
302: ln[pos++] = buf[i];
303: continue;
304: }
305:
306: /* Check for CPP-escaped newline. */
307:
308: if (pos > 0 && '\\' == ln[pos - 1]) {
309: for (j = pos - 1; j >= 0; j--)
310: if ('\\' != ln[j])
311: break;
312:
313: if ( ! ((pos - j) % 2)) {
314: pos--;
315: lnn++;
316: continue;
317: }
318: }
319:
320: ln[pos] = 0;
321: if ( ! mdoc_parseln(mdoc, lnn, ln))
322: return(0);
323: lnn++;
324: pos = 0;
325: }
326: }
327:
328: return(mdoc_endparse(mdoc));
329: }
330:
331:
332: static int
333: toptions(enum outt *tflags, char *arg)
334: {
335:
336: if (0 == strcmp(arg, "ascii"))
337: *tflags = OUTT_ASCII;
338: else if (0 == strcmp(arg, "latin1"))
339: *tflags = OUTT_LATIN1;
340: else if (0 == strcmp(arg, "utf8"))
341: *tflags = OUTT_UTF8;
342: else if (0 == strcmp(arg, "lint"))
343: *tflags = OUTT_LINT;
344: else if (0 == strcmp(arg, "tree"))
345: *tflags = OUTT_TREE;
346: else {
347: warnx("bad argument: -T%s", arg);
348: return(0);
349: }
350:
351: return(1);
352: }
353:
354:
355: /*
356: * Parse out the options for [-fopt...] setting compiler options. These
357: * can be comma-delimited or called again.
358: */
359: static int
360: foptions(int *fflags, char *arg)
361: {
362: char *v;
363: char *toks[4];
364:
365: toks[0] = "ign-scope";
366: toks[1] = "ign-escape";
367: toks[2] = "ign-macro";
368: toks[3] = NULL;
369:
370: while (*arg)
371: switch (getsubopt(&arg, toks, &v)) {
372: case (0):
373: *fflags |= MDOC_IGN_SCOPE;
374: break;
375: case (1):
376: *fflags |= MDOC_IGN_ESCAPE;
377: break;
378: case (2):
379: *fflags |= MDOC_IGN_MACRO;
380: break;
381: default:
382: warnx("bad argument: -f%s", arg);
383: return(0);
384: }
385:
386: return(1);
387: }
388:
389:
390: /*
391: * Parse out the options for [-Werr...], which sets warning modes.
392: * These can be comma-delimited or called again.
393: */
394: static int
395: woptions(int *wflags, char *arg)
396: {
397: char *v;
398: char *toks[5];
399:
400: toks[0] = "all";
401: toks[1] = "compat";
402: toks[2] = "syntax";
403: toks[3] = "error";
404: toks[4] = NULL;
405:
406: while (*arg)
407: switch (getsubopt(&arg, toks, &v)) {
408: case (0):
409: *wflags |= WARN_WALL;
410: break;
411: case (1):
412: *wflags |= WARN_WCOMPAT;
413: break;
414: case (2):
415: *wflags |= WARN_WSYNTAX;
416: break;
417: case (3):
418: *wflags |= WARN_WERR;
419: break;
420: default:
421: warnx("bad argument: -W%s", arg);
422: return(0);
423: }
424:
425: return(1);
426: }
427:
428:
1.2 kristaps 429: /* ARGSUSED */
1.1 kristaps 430: static int
431: merr(void *arg, int line, int col, const char *msg)
432: {
433:
434: warnx("error: %s (line %d, column %d)", msg, line, col);
435: return(0);
436: }
437:
438:
439: static int
440: mwarn(void *arg, int line, int col,
441: enum mdoc_warn type, const char *msg)
442: {
443: int flags;
444: char *wtype;
445:
446: flags = *(int *)arg;
447: wtype = NULL;
448:
449: switch (type) {
450: case (WARN_COMPAT):
451: wtype = "compat";
452: if (flags & WARN_WCOMPAT)
453: break;
454: return(1);
455: case (WARN_SYNTAX):
456: wtype = "syntax";
457: if (flags & WARN_WSYNTAX)
458: break;
459: return(1);
460: }
461:
462: assert(wtype);
463: warnx("%s warning: %s (line %d, column %d)",
464: wtype, msg, line, col);
465:
466: if ( ! (flags & WARN_WERR))
467: return(1);
468:
469: warnx("%s: considering warnings as errors",
470: __progname);
471: return(0);
472: }
473:
474:
CVSweb