template/template.c

   1 /*
   2  * Copyright (c) 2022 Omar Polo <op@openbsd.org>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #include <stdio.h>
  18 #include <stdlib.h>
  19 #include <unistd.h>
  20
  21 #include "got_compat.h"
  22
  23 int      parse(FILE *, const char *);
  24
  25 int      nodebug;
  26
  27 static void __dead
  28 usage(void)
  29 {
  30         fprintf(stderr, "usage: %s [-G] [-o out] [file ...]\n", getprogname());
  31         exit(1);
  32 }
  33
  34 int
  35 main(int argc, char **argv)
  36 {
  37         FILE            *fp = stdout;
  38         const char      *out = NULL;
  39         int              ch, i;
  40
  41         while ((ch = getopt(argc, argv, "Go:")) != -1) {
  42                 switch (ch) {
  43                 case 'G':
  44                         nodebug = 1;
  45                         break;
  46                 case 'o':
  47                         out = optarg;
  48                         break;
  49                 default:
  50                         usage();
  51                 }
  52         }
  53         argc -= optind;
  54         argv += optind;
  55
  56         if (out && (fp = fopen(out, "w")) == NULL)
  57                 err(1, "can't open %s", out);
  58
  59         if (out && unveil(out, "wc") == -1)
  60                 err(1, "unveil %s", out);
  61         if (unveil("/", "r") == -1)
  62                 err(1, "unveil /");
  63         if (pledge(out ? "stdio rpath cpath" : "stdio rpath", NULL) == -1)
  64                 err(1, "pledge");
  65
  66         if (argc == 0) {
  67                 nodebug = 1;
  68                 if (parse(fp, "/dev/stdin") == -1)
  69                         goto err;
  70         } else {
  71                 for (i = 0; i < argc; ++i)
  72                         if (parse(fp, argv[i]) == -1)
  73                                 goto err;
  74         }
  75
  76         if (ferror(fp))
  77                 goto err;
  78
  79         if (fclose(fp) == -1) {
  80                 fp = NULL;
  81                 goto err;
  82         }
  83
  84         return (0);
  85
  86 err:
  87         if (fp)
  88                 fclose(fp);
  89         if (out && unlink(out) == -1)
  90                 err(1, "unlink %s", out);
  91         return (1);
  92 }