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 <paths.h>
35 #include <stdio.h>
36 #include <stdlib.h>
37 #include <string.h>
38 #include <unistd.h>
39
40 #include "kgzip.h"
41
42 #define FN_SRC 0 /* Filename: source */
43 #define FN_OBJ 1 /* Filename: relocatable */
44 #define FN_KGZ 2 /* Filename: executable */
45 #define FN_CNT 3 /* Number of filenames */
46
47 #define SFX_OBJ ".o" /* Filename suffix: relocatable */
48 #define SFX_KGZ ".kgz" /* Filename suffix: executable */
49 #define SFX_MAX 5 /* Size of larger filename suffix */
50
51 const char *loader = "/usr/lib/kgzldr.o"; /* Default loader */
52 int format; /* Output format */
53
54 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
main(int argc,char * argv[])64 main(int argc, char *argv[])
65 {
66 static char *fn[FN_CNT];
67 struct kgz_hdr kh;
68 const char *output;
69 char *tmpdir;
70 int cflag, vflag, c;
71
72 tmpdir = getenv("TMPDIR");
73 if (asprintf(&tname, "%s/kgzXXXXXXXXXX",
74 tmpdir == NULL ? _PATH_TMP : tmpdir) == -1)
75 errx(1, "Out of memory");
76 output = NULL;
77 cflag = vflag = 0;
78 while ((c = getopt(argc, argv, "cvf:l:o:")) != -1)
79 switch (c) {
80 case 'c':
81 cflag = 1;
82 break;
83 case 'v':
84 vflag = 1;
85 break;
86 case 'f':
87 if (!strcmp(optarg, "aout"))
88 format = F_AOUT;
89 else if (!strcmp(optarg, "elf"))
90 format = F_ELF;
91 else
92 errx(1, "%s: Unknown format", optarg);
93 break;
94 case 'l':
95 loader = optarg;
96 break;
97 case 'o':
98 output = optarg;
99 break;
100 default:
101 usage();
102 }
103 argc -= optind;
104 argv += optind;
105 if (argc != 1)
106 usage();
107 atexit(cleanup);
108 mk_fn(cflag, *argv, output, fn);
109 memset(&kh, 0, sizeof(kh));
110 if (fn[FN_SRC]) {
111 if (!format)
112 format = F_ELF;
113 kgzcmp(&kh, fn[FN_SRC], fn[FN_OBJ]);
114 }
115 if (!cflag)
116 kgzld(&kh, fn[FN_OBJ], fn[FN_KGZ]);
117 if (vflag)
118 printf("dload=%#x dsize=%#x isize=%#x entry=%#x nsize=%#x\n",
119 kh.dload, kh.dsize, kh.isize, kh.entry, kh.nsize);
120 return 0;
121 }
122
123 /*
124 * Clean up after processing.
125 */
126 static void
cleanup(void)127 cleanup(void)
128 {
129 if (tname)
130 unlink(tname);
131 }
132
133 /*
134 * Make the required filenames.
135 */
136 static void
mk_fn(int cflag,const char * f1,const char * f2,char * fn[])137 mk_fn(int cflag, const char *f1, const char *f2, char *fn[])
138 {
139 const char *p, *s;
140 size_t n;
141 int i, fd;
142
143 i = 0;
144 s = strrchr(f1, 0);
145 n = sizeof(SFX_OBJ) - 1;
146 if ((size_t)(s - f1) > n && !memcmp(s - n, SFX_OBJ, n)) {
147 s -= n;
148 i++;
149 }
150 fn[i++] = (char *)f1;
151 if (i == FN_OBJ && !cflag) {
152 if ((fd = mkstemp(tname)) == -1)
153 err(1, NULL);
154 close(fd);
155 fn[i++] = tname;
156 }
157 if (!(fn[i] = (char *)f2)) {
158 p = (p = strrchr(f1, '/')) ? p + 1 : f1;
159 n = (size_t)(s - p);
160 if (!(fn[i] = malloc(n + SFX_MAX)))
161 err(1, NULL);
162 memcpy(fn[i], p, n);
163 strcpy(fn[i] + n, i == FN_OBJ ? SFX_OBJ : SFX_KGZ);
164 }
165 }
166
167 /*
168 * Display usage information.
169 */
170 static void
usage(void)171 usage(void)
172 {
173 fprintf(stderr,
174 "usage: kgzip [-cv] [-f format] [-l loader] [-o output] file\n");
175 exit(1);
176 }
177