1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* Copyright (c) 1999 Global Technology Associates, Inc.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
* OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
* OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
* BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
* OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
* EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
* $FreeBSD$
*/
#include <sys/types.h>
#include <sys/reboot.h>
#include <sys/inflate.h>
#include "kgzldr.h"
#define KGZ_HEAD 0xa /* leading bytes to ignore */
#define KGZ_TAIL 0x8 /* trailing bytes to ignore */
#define E_FMT 1 /* Error: Invalid format */
#define E_MEM 2 /* Error: Out of memory */
struct kgz_hdr {
char ident[4]; /* identification */
uint32_t dload; /* decoded image load address */
uint32_t dsize; /* decoded image size */
uint32_t isize; /* image size in memory */
uint32_t entry; /* program entry point */
uint32_t nsize; /* encoded image size */
};
extern struct kgz_hdr kgz; /* header */
extern uint8_t kgz_ndata[]; /* encoded image */
static const char *const msg[] = {
"done",
"invalid format",
"out of memory"
};
static const u_char *ip; /* input pointer */
static u_char *op; /* output pointer */
static struct inflate infl; /* inflate() parameters */
static int decode(void);
static int input(void *);
static int output(void *, u_char *, u_long);
/*
* Uncompress and boot a kernel.
*/
int
boot(int howto)
{
int err;
kgz_con = howto & RB_SERIAL ? KGZ_SIO : KGZ_CRT;
putstr("Uncompressing ... ");
err = decode();
putstr(msg[err]);
putstr("\n");
if (err) {
putstr("System halted");
for (;;)
;
}
return err;
}
/*
* Interface with inflate() to uncompress the data.
*/
static int
decode(void)
{
static u_char slide[GZ_WSIZE];
int err;
ip = kgz_ndata + KGZ_HEAD;
op = (u_char *)kgz.dload;
infl.gz_input = input;
infl.gz_output = output;
infl.gz_slide = slide;
err = inflate(&infl);
return err ? err == 3 ? E_MEM : E_FMT : 0;
}
/*
* Read a byte.
*/
static int
input(void *dummy)
{
if ((size_t)(ip - kgz_ndata) + KGZ_TAIL > kgz.nsize)
return GZ_EOF;
return *ip++;
}
/*
* Write some bytes.
*/
static int
output(void *dummy, u_char * ptr, u_long len)
{
if (op - (u_char *)kgz.dload + len > kgz.dsize)
return -1;
while (len--)
*op++ = *ptr++;
return 0;
}
|
