usr.sbin/kgzip/kgzip.c

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