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