Annotation of mandoc/mandocd.c, Revision 1.7
1.7 ! schwarze 1: /* $Id: mandocd.c,v 1.6 2017/06/24 14:38:32 schwarze Exp $ */
1.1 schwarze 2: /*
3: * Copyright (c) 2017 Michael Stapelberg <stapelberg@debian.org>
4: * Copyright (c) 2017 Ingo Schwarze <schwarze@openbsd.org>
5: *
6: * Permission to use, copy, modify, and distribute this software for any
7: * purpose with or without fee is hereby granted, provided that the above
8: * copyright notice and this permission notice appear in all copies.
9: *
10: * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHORS DISCLAIM ALL WARRANTIES
11: * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12: * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR
13: * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14: * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15: * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16: * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17: */
18: #include "config.h"
1.4 schwarze 19:
20: #if HAVE_CMSG_XPG42
21: #define _XPG4_2
22: #endif
1.1 schwarze 23:
24: #include <sys/types.h>
25: #include <sys/socket.h>
26:
27: #if HAVE_ERR
28: #include <err.h>
29: #endif
30: #include <limits.h>
1.2 schwarze 31: #include <stdint.h>
1.1 schwarze 32: #include <stdio.h>
33: #include <stdlib.h>
34: #include <string.h>
35: #include <unistd.h>
36:
37: #include "mandoc.h"
38: #include "roff.h"
39: #include "mdoc.h"
40: #include "man.h"
1.7 ! schwarze 41: #include "mandoc_parse.h"
1.1 schwarze 42: #include "main.h"
43: #include "manconf.h"
44:
45: enum outt {
46: OUTT_ASCII = 0,
47: OUTT_UTF8,
48: OUTT_HTML
49: };
50:
51: static void process(struct mparse *, enum outt, void *);
52: static int read_fds(int, int *);
1.5 schwarze 53: static void usage(void) __attribute__((__noreturn__));
1.1 schwarze 54:
55:
56: #define NUM_FDS 3
57: static int
58: read_fds(int clientfd, int *fds)
59: {
60: struct msghdr msg;
61: struct iovec iov[1];
62: unsigned char dummy[1];
63: struct cmsghdr *cmsg;
64: int *walk;
65: int cnt;
66:
67: /* Union used for alignment. */
68: union {
69: uint8_t controlbuf[CMSG_SPACE(NUM_FDS * sizeof(int))];
70: struct cmsghdr align;
71: } u;
72:
73: memset(&msg, '\0', sizeof(msg));
74: msg.msg_control = u.controlbuf;
75: msg.msg_controllen = sizeof(u.controlbuf);
76:
77: /*
78: * Read a dummy byte - sendmsg cannot send an empty message,
79: * even if we are only interested in the OOB data.
80: */
81:
82: iov[0].iov_base = dummy;
83: iov[0].iov_len = sizeof(dummy);
84: msg.msg_iov = iov;
85: msg.msg_iovlen = 1;
86:
87: switch (recvmsg(clientfd, &msg, 0)) {
88: case -1:
89: warn("recvmsg");
90: return -1;
91: case 0:
92: return 0;
93: default:
94: break;
95: }
96:
97: if ((cmsg = CMSG_FIRSTHDR(&msg)) == NULL) {
98: warnx("CMSG_FIRSTHDR: missing control message");
99: return -1;
100: }
101:
102: if (cmsg->cmsg_level != SOL_SOCKET ||
103: cmsg->cmsg_type != SCM_RIGHTS ||
104: cmsg->cmsg_len != CMSG_LEN(NUM_FDS * sizeof(int))) {
105: warnx("CMSG_FIRSTHDR: invalid control message");
106: return -1;
107: }
108:
109: walk = (int *)CMSG_DATA(cmsg);
110: for (cnt = 0; cnt < NUM_FDS; cnt++)
111: fds[cnt] = *walk++;
112:
113: return 1;
114: }
115:
116: int
117: main(int argc, char *argv[])
118: {
119: struct manoutput options;
120: struct mparse *parser;
121: void *formatter;
1.3 schwarze 122: const char *defos;
1.1 schwarze 123: const char *errstr;
124: int clientfd;
125: int old_stdin;
126: int old_stdout;
127: int old_stderr;
128: int fds[3];
129: int state, opt;
130: enum outt outtype;
131:
1.3 schwarze 132: defos = NULL;
1.1 schwarze 133: outtype = OUTT_ASCII;
1.3 schwarze 134: while ((opt = getopt(argc, argv, "I:T:")) != -1) {
1.1 schwarze 135: switch (opt) {
1.3 schwarze 136: case 'I':
137: if (strncmp(optarg, "os=", 3) == 0)
138: defos = optarg + 3;
139: else {
140: warnx("-I %s: Bad argument", optarg);
141: usage();
142: }
143: break;
1.1 schwarze 144: case 'T':
145: if (strcmp(optarg, "ascii") == 0)
146: outtype = OUTT_ASCII;
147: else if (strcmp(optarg, "utf8") == 0)
148: outtype = OUTT_UTF8;
149: else if (strcmp(optarg, "html") == 0)
150: outtype = OUTT_HTML;
151: else {
152: warnx("-T %s: Bad argument", optarg);
153: usage();
154: }
155: break;
156: default:
157: usage();
158: }
159: }
160:
161: if (argc > 0) {
162: argc -= optind;
163: argv += optind;
164: }
165: if (argc != 1)
166: usage();
167:
168: errstr = NULL;
169: clientfd = strtonum(argv[0], 3, INT_MAX, &errstr);
170: if (errstr)
171: errx(1, "file descriptor %s %s", argv[1], errstr);
172:
173: mchars_alloc();
174: parser = mparse_alloc(MPARSE_SO | MPARSE_UTF8 | MPARSE_LATIN1,
1.6 schwarze 175: MANDOCERR_MAX, NULL, MANDOC_OS_OTHER, defos);
1.1 schwarze 176:
177: memset(&options, 0, sizeof(options));
178: switch (outtype) {
179: case OUTT_ASCII:
180: formatter = ascii_alloc(&options);
181: break;
182: case OUTT_UTF8:
183: formatter = utf8_alloc(&options);
184: break;
185: case OUTT_HTML:
186: options.fragment = 1;
187: formatter = html_alloc(&options);
188: break;
189: }
190:
191: state = 1; /* work to do */
192: fflush(stdout);
193: fflush(stderr);
194: if ((old_stdin = dup(STDIN_FILENO)) == -1 ||
195: (old_stdout = dup(STDOUT_FILENO)) == -1 ||
196: (old_stderr = dup(STDERR_FILENO)) == -1) {
197: warn("dup");
198: state = -1; /* error */
199: }
200:
201: while (state == 1 && (state = read_fds(clientfd, fds)) == 1) {
202: if (dup2(fds[0], STDIN_FILENO) == -1 ||
203: dup2(fds[1], STDOUT_FILENO) == -1 ||
204: dup2(fds[2], STDERR_FILENO) == -1) {
205: warn("dup2");
206: state = -1;
207: break;
208: }
209:
210: close(fds[0]);
211: close(fds[1]);
212: close(fds[2]);
213:
214: process(parser, outtype, formatter);
215: mparse_reset(parser);
216:
217: fflush(stdout);
218: fflush(stderr);
219: /* Close file descriptors by restoring the old ones. */
220: if (dup2(old_stderr, STDERR_FILENO) == -1 ||
221: dup2(old_stdout, STDOUT_FILENO) == -1 ||
222: dup2(old_stdin, STDIN_FILENO) == -1) {
223: warn("dup2");
224: state = -1;
225: break;
226: }
227: }
228:
229: close(clientfd);
230: switch (outtype) {
231: case OUTT_ASCII:
232: case OUTT_UTF8:
233: ascii_free(formatter);
234: break;
235: case OUTT_HTML:
236: html_free(formatter);
237: break;
238: }
239: mparse_free(parser);
240: mchars_free();
241: return state == -1 ? 1 : 0;
242: }
243:
244: static void
245: process(struct mparse *parser, enum outt outtype, void *formatter)
246: {
247: struct roff_man *man;
248:
249: mparse_readfd(parser, STDIN_FILENO, "<unixfd>");
250: mparse_result(parser, &man, NULL);
251:
252: if (man == NULL)
253: return;
254:
255: if (man->macroset == MACROSET_MDOC) {
256: mdoc_validate(man);
257: switch (outtype) {
258: case OUTT_ASCII:
259: case OUTT_UTF8:
260: terminal_mdoc(formatter, man);
261: break;
262: case OUTT_HTML:
263: html_mdoc(formatter, man);
264: break;
265: }
266: }
267: if (man->macroset == MACROSET_MAN) {
268: man_validate(man);
269: switch (outtype) {
270: case OUTT_ASCII:
271: case OUTT_UTF8:
272: terminal_man(formatter, man);
273: break;
274: case OUTT_HTML:
275: html_man(formatter, man);
276: break;
277: }
278: }
279: }
280:
281: void
282: usage(void)
283: {
1.3 schwarze 284: fprintf(stderr, "usage: mandocd [-I os=name] [-T output] socket_fd\n");
1.1 schwarze 285: exit(1);
286: }
CVSweb