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$";
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
51const char *loader = "/usr/lib/kgzldr.o";  /* Default loader */
52int format;			/* Output format */
53
54char *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    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 */
126static void
127cleanup(void)
128{
129    if (tname)
130	unlink(tname);
131}
132
133/*
134 * Make the required filenames.
135 */
136static void
137mk_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 */
170static void
171usage(void)
172{
173    fprintf(stderr,
174      "usage: kgzip [-cv] [-f format] [-l loader] [-o output] file\n");
175    exit(1);
176}
177