Annotation of src/usr.bin/mandoc/term_tag.c, Revision 1.2
1.2 ! schwarze 1: /* $OpenBSD: term_tag.c,v 1.1 2020/03/13 00:31:05 schwarze Exp $ */
1.1 schwarze 2: /*
3: * Copyright (c) 2015,2016,2018,2019,2020 Ingo Schwarze <schwarze@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 above
7: * copyright notice and this permission notice appear in all copies.
8: *
9: * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10: * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11: * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12: * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13: * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14: * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15: * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16: *
17: * Functions to write a ctags(1) file.
18: * For use by the mandoc(1) ASCII and UTF-8 formatters only.
19: */
20: #include <sys/types.h>
21:
22: #include <errno.h>
23: #include <signal.h>
24: #include <stddef.h>
25: #include <stdio.h>
26: #include <stdlib.h>
27: #include <string.h>
28: #include <unistd.h>
29:
30: #include "mandoc.h"
31: #include "roff.h"
32: #include "tag.h"
33: #include "term_tag.h"
34:
35: static void tag_signal(int) __attribute__((__noreturn__));
36:
37: static struct tag_files tag_files;
38:
39:
40: /*
41: * Prepare for using a pager.
42: * Not all pagers are capable of using a tag file,
43: * but for simplicity, create it anyway.
44: */
45: struct tag_files *
1.2 ! schwarze 46: term_tag_init(void)
1.1 schwarze 47: {
48: struct sigaction sa;
49: int ofd; /* In /tmp/, dup(2)ed to stdout. */
50: int tfd;
51:
52: ofd = tfd = -1;
53: tag_files.tfs = NULL;
54: tag_files.tcpgid = -1;
55:
56: /* Clean up when dying from a signal. */
57:
58: memset(&sa, 0, sizeof(sa));
59: sigfillset(&sa.sa_mask);
60: sa.sa_handler = tag_signal;
61: sigaction(SIGHUP, &sa, NULL);
62: sigaction(SIGINT, &sa, NULL);
63: sigaction(SIGTERM, &sa, NULL);
64:
65: /*
66: * POSIX requires that a process calling tcsetpgrp(3)
67: * from the background gets a SIGTTOU signal.
68: * In that case, do not stop.
69: */
70:
71: sa.sa_handler = SIG_IGN;
72: sigaction(SIGTTOU, &sa, NULL);
73:
74: /* Save the original standard output for use by the pager. */
75:
76: if ((tag_files.ofd = dup(STDOUT_FILENO)) == -1) {
77: mandoc_msg(MANDOCERR_DUP, 0, 0, "%s", strerror(errno));
78: goto fail;
79: }
80:
81: /* Create both temporary output files. */
82:
83: (void)strlcpy(tag_files.ofn, "/tmp/man.XXXXXXXXXX",
84: sizeof(tag_files.ofn));
85: (void)strlcpy(tag_files.tfn, "/tmp/man.XXXXXXXXXX",
86: sizeof(tag_files.tfn));
87: if ((ofd = mkstemp(tag_files.ofn)) == -1) {
88: mandoc_msg(MANDOCERR_MKSTEMP, 0, 0,
89: "%s: %s", tag_files.ofn, strerror(errno));
90: goto fail;
91: }
92: if ((tfd = mkstemp(tag_files.tfn)) == -1) {
93: mandoc_msg(MANDOCERR_MKSTEMP, 0, 0,
94: "%s: %s", tag_files.tfn, strerror(errno));
95: goto fail;
96: }
97: if ((tag_files.tfs = fdopen(tfd, "w")) == NULL) {
98: mandoc_msg(MANDOCERR_FDOPEN, 0, 0, "%s", strerror(errno));
99: goto fail;
100: }
101: tfd = -1;
102: if (dup2(ofd, STDOUT_FILENO) == -1) {
103: mandoc_msg(MANDOCERR_DUP, 0, 0, "%s", strerror(errno));
104: goto fail;
105: }
106: close(ofd);
107: return &tag_files;
108:
109: fail:
110: term_tag_unlink();
111: if (ofd != -1)
112: close(ofd);
113: if (tfd != -1)
114: close(tfd);
115: if (tag_files.ofd != -1) {
116: close(tag_files.ofd);
117: tag_files.ofd = -1;
118: }
119: return NULL;
120: }
121:
122: void
123: term_tag_write(struct roff_node *n, size_t line)
124: {
125: const char *cp;
126: int len;
127:
128: if (tag_files.tfs == NULL)
129: return;
130: if (n->string == NULL)
131: n = n->child;
132: cp = n->string;
133: if (cp[0] == '\\' && (cp[1] == '&' || cp[1] == 'e'))
134: cp += 2;
135: len = strcspn(cp, " \t\\");
136: fprintf(tag_files.tfs, "%.*s %s %zu\n",
137: len, cp, tag_files.ofn, line);
138: }
139:
1.2 ! schwarze 140: /*
! 141: * Close both output files and restore the original standard output
! 142: * to the terminal. In the unlikely case that the latter fails,
! 143: * trying to start a pager would be useless, so report the failure
! 144: * to the main program.
! 145: */
! 146: int
! 147: term_tag_close(void)
1.1 schwarze 148: {
1.2 ! schwarze 149: int irc = 0;
! 150:
! 151: if (tag_files.tfs != NULL) {
! 152: fclose(tag_files.tfs);
! 153: tag_files.tfs = NULL;
! 154: }
! 155: if (tag_files.ofd != -1) {
! 156: fflush(stdout);
! 157: if ((irc = dup2(tag_files.ofd, STDOUT_FILENO)) == -1)
! 158: mandoc_msg(MANDOCERR_DUP, 0, 0, "%s", strerror(errno));
! 159: close(tag_files.ofd);
! 160: tag_files.ofd = -1;
1.1 schwarze 161: }
1.2 ! schwarze 162: return irc;
1.1 schwarze 163: }
164:
165: void
166: term_tag_unlink(void)
167: {
168: pid_t tc_pgid;
169:
170: if (tag_files.tcpgid != -1) {
1.2 ! schwarze 171: tc_pgid = tcgetpgrp(STDOUT_FILENO);
1.1 schwarze 172: if (tc_pgid == tag_files.pager_pid ||
173: tc_pgid == getpgid(0) ||
174: getpgid(tc_pgid) == -1)
1.2 ! schwarze 175: (void)tcsetpgrp(STDOUT_FILENO, tag_files.tcpgid);
1.1 schwarze 176: }
177: if (*tag_files.ofn != '\0') {
178: unlink(tag_files.ofn);
179: *tag_files.ofn = '\0';
180: }
181: if (*tag_files.tfn != '\0') {
182: unlink(tag_files.tfn);
183: *tag_files.tfn = '\0';
184: }
185: }
186:
187: static void
188: tag_signal(int signum)
189: {
190: struct sigaction sa;
191:
192: term_tag_unlink();
193: memset(&sa, 0, sizeof(sa));
194: sigemptyset(&sa.sa_mask);
195: sa.sa_handler = SIG_DFL;
196: sigaction(signum, &sa, NULL);
197: kill(getpid(), signum);
198: /* NOTREACHED */
199: _exit(1);
200: }