xref: /freebsd/usr.bin/soelim/soelim.c (revision 1d386b48)
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 #include <sys/types.h>
29 
30 #include <ctype.h>
31 #include <err.h>
32 #include <limits.h>
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #include <stringlist.h>
37 #include <unistd.h>
38 
39 #define C_OPTION 0x1
40 
41 static StringList *includes;
42 
43 static void
44 usage(void)
45 {
46 
47 	fprintf(stderr, "usage: soelim [-Crtv] [-I dir] [files]\n");
48 
49 	exit(EXIT_FAILURE);
50 }
51 
52 static FILE *
53 soelim_fopen(const char *name)
54 {
55 	FILE *f;
56 	char path[PATH_MAX];
57 	size_t i;
58 
59 	if (strcmp(name, "-") == 0)
60 		return (stdin);
61 
62 	if ((f = fopen(name, "r")) != NULL)
63 		return (f);
64 
65 	if (*name == '/') {
66 		warn("can't open '%s'", name);
67 		return (NULL);
68 	}
69 
70 	for (i = 0; i < includes->sl_cur; i++) {
71 		snprintf(path, sizeof(path), "%s/%s", includes->sl_str[i],
72 		    name);
73 		if ((f = fopen(path, "r")) != NULL)
74 			return (f);
75 	}
76 
77 	warn("can't open '%s'", name);
78 
79 	return (f);
80 }
81 
82 static int
83 soelim_file(FILE *f, int flag)
84 {
85 	char *line = NULL;
86 	char *walk, *cp;
87 	size_t linecap = 0;
88 	ssize_t linelen;
89 
90 	if (f == NULL)
91 		return (1);
92 
93 	while ((linelen = getline(&line, &linecap, f)) > 0) {
94 		if (strncmp(line, ".so", 3) != 0) {
95 			printf("%s", line);
96 			continue;
97 		}
98 
99 		walk = line + 3;
100 		if (!isspace(*walk) && ((flag & C_OPTION) == 0)) {
101 			printf("%s", line);
102 			continue;
103 		}
104 
105 		while (isspace(*walk))
106 			walk++;
107 
108 		cp = walk;
109 		while (*cp != '\0' && !isspace(*cp))
110 			cp++;
111 		*cp = 0;
112 		if (cp < line + linelen)
113 			cp++;
114 
115 		if (*walk == '\0') {
116 			printf("%s", line);
117 			continue;
118 		}
119 		if (soelim_file(soelim_fopen(walk), flag) == 1) {
120 			free(line);
121 			return (1);
122 		}
123 		if (*cp != '\0')
124 			printf("%s", cp);
125 	}
126 
127 	free(line);
128 	fclose(f);
129 
130 	return (0);
131 }
132 
133 int
134 main(int argc, char **argv)
135 {
136 	int ch, i;
137 	int ret = 0;
138 	int flags = 0;
139 
140 	includes = sl_init();
141 	if (includes == NULL)
142 		err(EXIT_FAILURE, "sl_init()");
143 
144 	while ((ch = getopt(argc, argv, "CrtvI:")) != -1) {
145 		switch (ch) {
146 		case 'C':
147 			flags |= C_OPTION;
148 			break;
149 		case 'r':
150 		case 'v':
151 		case 't':
152 			/* stub compatibility with groff's soelim */
153 			break;
154 		case 'I':
155 			sl_add(includes, optarg);
156 			break;
157 		default:
158 			sl_free(includes, 0);
159 			usage();
160 		}
161 	}
162 
163 	argc -= optind;
164 	argv += optind;
165 
166 	if (argc == 0)
167 		ret = soelim_file(stdin, flags);
168 
169 	for (i = 0; i < argc; i++)
170 		ret = soelim_file(soelim_fopen(argv[i]), flags);
171 
172 	sl_free(includes, 0);
173 
174 	return (ret);
175 }
176