usr.bin/soelim/soelim.c

   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 }