kgzip.c revision 50479
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 28static const char rcsid[] = 29 "$FreeBSD: head/usr.sbin/kgzip/kgzip.c 50479 1999-08-28 01:35:59Z peter $"; 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 52const char *loader = "/usr/lib/kgzldr.o"; /* Default loader */ 53 54static const char *tname; /* Name of temporary file */ 55 56static void cleanup(void); 57static void mk_fn(int, const char *, const char *, char *[]); 58static void usage(void); 59 60/* 61 * Compress a kernel. 62 */ 63int 64main(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 */ 110static void 111cleanup(void) 112{ 113 if (tname) 114 unlink(tname); 115} 116 117/* 118 * Make the required filenames. 119 */ 120static void 121mk_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 */ 153static void 154usage(void) 155{ 156 fprintf(stderr, 157 "usage: kgzip [-cv] [-l file] [-o filename] file\n"); 158 exit(1); 159} 160