1 /* $Id: mandocd.c,v 1.6 2017/06/24 14:38:32 schwarze Exp $ */
3 * Copyright (c) 2017 Michael Stapelberg <stapelberg@debian.org>
4 * Copyright (c) 2017 Ingo Schwarze <schwarze@openbsd.org>
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.
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.
24 #include <sys/types.h>
25 #include <sys/socket.h>
50 static void process(struct mparse *, enum outt, void *);
51 static int read_fds(int, int *);
52 static void usage(void) __attribute__((__noreturn__));
57 read_fds(int clientfd, int *fds)
61 unsigned char dummy[1];
66 /* Union used for alignment. */
68 uint8_t controlbuf[CMSG_SPACE(NUM_FDS * sizeof(int))];
72 memset(&msg, '\0', sizeof(msg));
73 msg.msg_control = u.controlbuf;
74 msg.msg_controllen = sizeof(u.controlbuf);
77 * Read a dummy byte - sendmsg cannot send an empty message,
78 * even if we are only interested in the OOB data.
81 iov[0].iov_base = dummy;
82 iov[0].iov_len = sizeof(dummy);
86 switch (recvmsg(clientfd, &msg, 0)) {
96 if ((cmsg = CMSG_FIRSTHDR(&msg)) == NULL) {
97 warnx("CMSG_FIRSTHDR: missing control message");
101 if (cmsg->cmsg_level != SOL_SOCKET ||
102 cmsg->cmsg_type != SCM_RIGHTS ||
103 cmsg->cmsg_len != CMSG_LEN(NUM_FDS * sizeof(int))) {
104 warnx("CMSG_FIRSTHDR: invalid control message");
108 walk = (int *)CMSG_DATA(cmsg);
109 for (cnt = 0; cnt < NUM_FDS; cnt++)
116 main(int argc, char *argv[])
118 struct manoutput options;
119 struct mparse *parser;
132 outtype = OUTT_ASCII;
133 while ((opt = getopt(argc, argv, "I:T:")) != -1) {
136 if (strncmp(optarg, "os=", 3) == 0)
139 warnx("-I %s: Bad argument", optarg);
144 if (strcmp(optarg, "ascii") == 0)
145 outtype = OUTT_ASCII;
146 else if (strcmp(optarg, "utf8") == 0)
148 else if (strcmp(optarg, "html") == 0)
151 warnx("-T %s: Bad argument", optarg);
168 clientfd = strtonum(argv[0], 3, INT_MAX, &errstr);
170 errx(1, "file descriptor %s %s", argv[1], errstr);
173 parser = mparse_alloc(MPARSE_SO | MPARSE_UTF8 | MPARSE_LATIN1,
174 MANDOCERR_MAX, NULL, MANDOC_OS_OTHER, defos);
176 memset(&options, 0, sizeof(options));
179 formatter = ascii_alloc(&options);
182 formatter = utf8_alloc(&options);
185 options.fragment = 1;
186 formatter = html_alloc(&options);
190 state = 1; /* work to do */
193 if ((old_stdin = dup(STDIN_FILENO)) == -1 ||
194 (old_stdout = dup(STDOUT_FILENO)) == -1 ||
195 (old_stderr = dup(STDERR_FILENO)) == -1) {
197 state = -1; /* error */
200 while (state == 1 && (state = read_fds(clientfd, fds)) == 1) {
201 if (dup2(fds[0], STDIN_FILENO) == -1 ||
202 dup2(fds[1], STDOUT_FILENO) == -1 ||
203 dup2(fds[2], STDERR_FILENO) == -1) {
213 process(parser, outtype, formatter);
214 mparse_reset(parser);
218 /* Close file descriptors by restoring the old ones. */
219 if (dup2(old_stderr, STDERR_FILENO) == -1 ||
220 dup2(old_stdout, STDOUT_FILENO) == -1 ||
221 dup2(old_stdin, STDIN_FILENO) == -1) {
232 ascii_free(formatter);
235 html_free(formatter);
240 return state == -1 ? 1 : 0;
244 process(struct mparse *parser, enum outt outtype, void *formatter)
246 struct roff_man *man;
248 mparse_readfd(parser, STDIN_FILENO, "<unixfd>");
249 mparse_result(parser, &man, NULL);
254 if (man->macroset == MACROSET_MDOC) {
259 terminal_mdoc(formatter, man);
262 html_mdoc(formatter, man);
266 if (man->macroset == MACROSET_MAN) {
271 terminal_man(formatter, man);
274 html_man(formatter, man);
283 fprintf(stderr, "usage: mandocd [-I os=name] [-T output] socket_fd\n");