- use WARNS?= 6 instead of a hard-wired list. -pedantic doesn't really work
[dragonfly.git] / 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  * $FreeBSD: src/usr.sbin/kgzip/kgzip.c,v 1.3.2.3 2001/07/19 04:42:38 kris Exp $
27  * $DragonFly: src/usr.sbin/kgzip/kgzip.c,v 1.3 2004/08/19 21:36:46 joerg Exp $
28  */
29
30 #include <sys/types.h>
31 #include <err.h>
32 #include <paths.h>
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #include <unistd.h>
37
38 #include "kgzip.h"
39
40 #define FN_SRC          0       /* Filename: source */
41 #define FN_OBJ          1       /* Filename: relocatable */
42 #define FN_KGZ          2       /* Filename: executable */
43 #define FN_CNT          3       /* Number of filenames */
44
45 #define SFX_OBJ         ".o"    /* Filename suffix: relocatable */
46 #define SFX_KGZ         ".kgz"  /* Filename suffix: executable */
47 #define SFX_MAX         5       /* Size of larger filename suffix */
48
49 const char *loader = "/usr/lib/kgzldr.o";  /* Default loader */
50 int format;                     /* Output format */
51
52 char *tname;                    /* Name of temporary file */
53
54 static void cleanup(void);
55 static void mk_fn(int, const char *, const char *, const char *[]);
56 static void usage(void);
57
58 /*
59  * Compress a kernel.
60  */
61 int
62 main(int argc, char *argv[])
63 {
64     static const char *fn[FN_CNT];
65     struct kgz_hdr kh;
66     const char *output;
67     char *tmpdir;
68     int cflag, vflag, c;
69
70     tmpdir = getenv("TMPDIR");
71     if (asprintf(&tname, "%s/kgzXXXXXXXXXX",
72                  tmpdir == NULL ? _PATH_TMP : tmpdir) == -1)
73         errx(1, "Out of memory");
74     output = NULL;
75     cflag = vflag = 0;
76     while ((c = getopt(argc, argv, "cvf:l:o:")) != -1)
77         switch (c) {
78         case 'c':
79             cflag = 1;
80             break;
81         case 'v':
82             vflag = 1;
83             break;
84         case 'f':
85             if (!strcmp(optarg, "aout"))
86                 format = F_AOUT;
87             else if (!strcmp(optarg, "elf"))
88                 format = F_ELF;
89             else
90                 errx(1, "%s: Unknown format", optarg);
91             break;
92         case 'l':
93             loader = optarg;
94             break;
95         case 'o':
96             output = optarg;
97             break;
98         default:
99             usage();
100         }
101     argc -= optind;
102     argv += optind;
103     if (argc != 1)
104         usage();
105     atexit(cleanup);
106     mk_fn(cflag, *argv, output, fn);
107     memset(&kh, 0, sizeof(kh));
108     if (fn[FN_SRC]) {
109         if (!format)
110             format = F_ELF;
111         kgzcmp(&kh, fn[FN_SRC], fn[FN_OBJ]);
112     }
113     if (!cflag)
114         kgzld(&kh, fn[FN_OBJ], fn[FN_KGZ]);
115     if (vflag)
116         printf("dload=%#x dsize=%#x isize=%#x entry=%#x nsize=%#x\n",
117                kh.dload, kh.dsize, kh.isize, kh.entry, kh.nsize);
118     return 0;
119 }
120
121 /*
122  * Clean up after processing.
123  */
124 static void
125 cleanup(void)
126 {
127     if (tname)
128         unlink(tname);
129 }
130
131 /*
132  * Make the required filenames.
133  */
134 static void
135 mk_fn(int cflag, const char *f1, const char *f2, const char *fn[])
136 {
137     const char *p, *s;
138     char *sfx;
139     size_t n;
140     int i, fd;
141
142     i = 0;
143     s = strrchr(f1, 0);
144     n = sizeof(SFX_OBJ) - 1;
145     if ((size_t)(s - f1) > n && !memcmp(s - n, SFX_OBJ, n)) {
146         s -= n;
147         i++;
148     }
149     fn[i++] = f1;
150     if (i == FN_OBJ && !cflag) {
151         if ((fd = mkstemp(tname)) == -1)
152             err(1, NULL);
153         close(fd);
154         fn[i++] = tname;
155     }
156     fn[i] = f2;
157     if (fn[i] == NULL) {
158         p = (p = strrchr(f1, '/')) ? p + 1 : f1;
159         n = (size_t)(s - p);
160         sfx = malloc(n + SFX_MAX);
161         if (sfx == NULL)
162             err(1, NULL);
163         memcpy(sfx, p, n);
164         strcpy(sfx + n, i == FN_OBJ ? SFX_OBJ : SFX_KGZ);
165         fn[i] = sfx;
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 file] [-o filename] file\n");
177     exit(1);
178 }