usr.sbin/kgzip/kgzip.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
   3  *
   4  * Copyright (c) 1999 Global Technology Associates, Inc.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS``AS IS'' AND
  17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS
  20  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
  21  * OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
  22  * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
  23  * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  24  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
  25  * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
  26  * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 #ifndef lint
  30 static const char rcsid[] =
  31   "$FreeBSD$";
  32 #endif /* not lint */
  33
  34 #include <sys/types.h>
  35 #include <err.h>
  36 #include <paths.h>
  37 #include <stdio.h>
  38 #include <stdlib.h>
  39 #include <string.h>
  40 #include <unistd.h>
  41
  42 #include "kgzip.h"
  43
  44 #define FN_SRC          0       /* Filename: source */
  45 #define FN_OBJ          1       /* Filename: relocatable */
  46 #define FN_KGZ          2       /* Filename: executable */
  47 #define FN_CNT          3       /* Number of filenames */
  48
  49 #define SFX_OBJ         ".o"    /* Filename suffix: relocatable */
  50 #define SFX_KGZ         ".kgz"  /* Filename suffix: executable */
  51 #define SFX_MAX         5       /* Size of larger filename suffix */
  52
  53 const char *loader = "/usr/lib/kgzldr.o";  /* Default loader */
  54 int format;                     /* Output format */
  55
  56 char *tname;                    /* Name of temporary file */
  57
  58 static void cleanup(void);
  59 static void mk_fn(int, const char *, const char *, char *[]);
  60 static void usage(void);
  61
  62 /*
  63  * Compress a kernel.
  64  */
  65 int
  66 main(int argc, char *argv[])
  67 {
  68     static char *fn[FN_CNT];
  69     struct kgz_hdr kh;
  70     const char *output;
  71     char *tmpdir;
  72     int cflag, vflag, c;
  73
  74     tmpdir = getenv("TMPDIR");
  75     if (asprintf(&tname, "%s/kgzXXXXXXXXXX",
  76                  tmpdir == NULL ? _PATH_TMP : tmpdir) == -1)
  77         errx(1, "Out of memory");
  78     output = NULL;
  79     cflag = vflag = 0;
  80     while ((c = getopt(argc, argv, "cvf:l:o:")) != -1)
  81         switch (c) {
  82         case 'c':
  83             cflag = 1;
  84             break;
  85         case 'v':
  86             vflag = 1;
  87             break;
  88         case 'f':
  89             if (!strcmp(optarg, "aout"))
  90                 format = F_AOUT;
  91             else if (!strcmp(optarg, "elf"))
  92                 format = F_ELF;
  93             else
  94                 errx(1, "%s: Unknown format", optarg);
  95             break;
  96         case 'l':
  97             loader = optarg;
  98             break;
  99         case 'o':
 100             output = optarg;
 101             break;
 102         default:
 103             usage();
 104         }
 105     argc -= optind;
 106     argv += optind;
 107     if (argc != 1)
 108         usage();
 109     atexit(cleanup);
 110     mk_fn(cflag, *argv, output, fn);
 111     memset(&kh, 0, sizeof(kh));
 112     if (fn[FN_SRC]) {
 113         if (!format)
 114             format = F_ELF;
 115         kgzcmp(&kh, fn[FN_SRC], fn[FN_OBJ]);
 116     }
 117     if (!cflag)
 118         kgzld(&kh, fn[FN_OBJ], fn[FN_KGZ]);
 119     if (vflag)
 120         printf("dload=%#x dsize=%#x isize=%#x entry=%#x nsize=%#x\n",
 121                kh.dload, kh.dsize, kh.isize, kh.entry, kh.nsize);
 122     return 0;
 123 }
 124
 125 /*
 126  * Clean up after processing.
 127  */
 128 static void
 129 cleanup(void)
 130 {
 131     if (tname)
 132         unlink(tname);
 133 }
 134
 135 /*
 136  * Make the required filenames.
 137  */
 138 static void
 139 mk_fn(int cflag, const char *f1, const char *f2, char *fn[])
 140 {
 141     const char *p, *s;
 142     size_t n;
 143     int i, fd;
 144
 145     i = 0;
 146     s = strrchr(f1, 0);
 147     n = sizeof(SFX_OBJ) - 1;
 148     if ((size_t)(s - f1) > n && !memcmp(s - n, SFX_OBJ, n)) {
 149         s -= n;
 150         i++;
 151     }
 152     fn[i++] = (char *)f1;
 153     if (i == FN_OBJ && !cflag) {
 154         if ((fd = mkstemp(tname)) == -1)
 155             err(1, NULL);
 156         close(fd);
 157         fn[i++] = tname;
 158     }
 159     if (!(fn[i] = (char *)f2)) {
 160         p = (p = strrchr(f1, '/')) ? p + 1 : f1;
 161         n = (size_t)(s - p);
 162         if (!(fn[i] = malloc(n + SFX_MAX)))
 163             err(1, NULL);
 164         memcpy(fn[i], p, n);
 165         strcpy(fn[i] + n, i == FN_OBJ ? SFX_OBJ : SFX_KGZ);
 166     }
 167 }
 168
 169 /*
 170  * Display usage information.
 171  */
 172 static void
 173 usage(void)
 174 {
 175     fprintf(stderr,
 176       "usage: kgzip [-cv] [-f format] [-l loader] [-o output] file\n");
 177     exit(1);
 178 }