xref: /freebsd/usr.bin/soelim/soelim.c (revision 193d9e768ba63fcfb187cfd17f461f7d41345048)
1 /*-
2  * Copyright (c) 2014 Baptiste Daroussin <bapt@FreeBSD.org>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer
10  *    in this position and unchanged.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18  * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26 
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
29 
30 #include <sys/types.h>
31 
32 #include <ctype.h>
33 #include <err.h>
34 #include <limits.h>
35 #include <stdio.h>
36 #include <stdlib.h>
37 #include <string.h>
38 #include <stringlist.h>
39 #include <unistd.h>
40 
41 #define C_OPTION 0x1
42 
43 static StringList *includes;
44 
45 static void
46 usage(void)
47 {
48 
49 	fprintf(stderr, "usage: soelim [-Crtv] [-I dir] [files]\n");
50 
51 	exit(EXIT_FAILURE);
52 }
53 
54 static FILE *
55 soelim_fopen(const char *name)
56 {
57 	FILE *f;
58 	char path[PATH_MAX];
59 	size_t i;
60 
61 	if (strcmp(name, "-") == 0)
62 		return (stdin);
63 
64 	if ((f = fopen(name, "r")) != NULL)
65 		return (f);
66 
67 	if (*name == '/') {
68 		warn("can't open '%s'", name);
69 		return (NULL);
70 	}
71 
72 	for (i = 0; i < includes->sl_cur; i++) {
73 		snprintf(path, sizeof(path), "%s/%s", includes->sl_str[i],
74 		    name);
75 		if ((f = fopen(path, "r")) != NULL)
76 			return (f);
77 	}
78 
79 	warn("can't open '%s'", name);
80 
81 	return (f);
82 }
83 
84 static int
85 soelim_file(FILE *f, int flag)
86 {
87 	char *line = NULL;
88 	char *walk, *cp;
89 	size_t linecap = 0;
90 	ssize_t linelen;
91 
92 	if (f == NULL)
93 		return (1);
94 
95 	while ((linelen = getline(&line, &linecap, f)) > 0) {
96 		if (strncmp(line, ".so", 3) != 0) {
97 			printf("%s", line);
98 			continue;
99 		}
100 
101 		walk = line + 3;
102 		if (!isspace(*walk) && ((flag & C_OPTION) == 0)) {
103 			printf("%s", line);
104 			continue;
105 		}
106 
107 		while (isspace(*walk))
108 			walk++;
109 
110 		cp = walk;
111 		while (*cp != '\0' && !isspace(*cp))
112 			cp++;
113 		*cp = 0;
114 		if (cp < line + linelen)
115 			cp++;
116 
117 		if (*walk == '\0') {
118 			printf("%s", line);
119 			continue;
120 		}
121 		if (soelim_file(soelim_fopen(walk), flag) == 1) {
122 			free(line);
123 			return (1);
124 		}
125 		if (*cp != '\0')
126 			printf("%s", cp);
127 	}
128 
129 	free(line);
130 	fclose(f);
131 
132 	return (0);
133 }
134 
135 int
136 main(int argc, char **argv)
137 {
138 	int ch, i;
139 	int ret = 0;
140 	int flags = 0;
141 
142 	includes = sl_init();
143 	if (includes == NULL)
144 		err(EXIT_FAILURE, "sl_init()");
145 
146 	while ((ch = getopt(argc, argv, "CrtvI:")) != -1) {
147 		switch (ch) {
148 		case 'C':
149 			flags |= C_OPTION;
150 			break;
151 		case 'r':
152 		case 'v':
153 		case 't':
154 			/* stub compatibility with groff's soelim */
155 			break;
156 		case 'I':
157 			sl_add(includes, optarg);
158 			break;
159 		default:
160 			sl_free(includes, 0);
161 			usage();
162 		}
163 	}
164 
165 	argc -= optind;
166 	argv += optind;
167 
168 	if (argc == 0)
169 		ret = soelim_file(stdin, flags);
170 
171 	for (i = 0; i < argc; i++)
172 		ret = soelim_file(soelim_fopen(argv[i]), flags);
173 
174 	sl_free(includes, 0);
175 
176 	return (ret);
177 }
178