Branch data Line data Source code
1 : : #ifdef STATIC
2 : : /* Pre-boot environment: included */
3 : :
4 : : /* prevent inclusion of _LINUX_KERNEL_H in pre-boot environment: lots
5 : : * errors about console_printk etc... on ARM */
6 : : #define _LINUX_KERNEL_H
7 : :
8 : : #include "zlib_inflate/inftrees.c"
9 : : #include "zlib_inflate/inffast.c"
10 : : #include "zlib_inflate/inflate.c"
11 : :
12 : : #else /* STATIC */
13 : : /* initramfs et al: linked */
14 : :
15 : : #include <linux/zutil.h>
16 : :
17 : : #include "zlib_inflate/inftrees.h"
18 : : #include "zlib_inflate/inffast.h"
19 : : #include "zlib_inflate/inflate.h"
20 : :
21 : : #include "zlib_inflate/infutil.h"
22 : :
23 : : #endif /* STATIC */
24 : :
25 : : #include <linux/decompress/mm.h>
26 : :
27 : : #define GZIP_IOBUF_SIZE (16*1024)
28 : :
29 : 0 : static int INIT nofill(void *buffer, unsigned int len)
30 : : {
31 : 0 : return -1;
32 : : }
33 : :
34 : : /* Included from initramfs et al code */
35 : 0 : STATIC int INIT gunzip(unsigned char *buf, int len,
36 : : int(*fill)(void*, unsigned int),
37 : : int(*flush)(void*, unsigned int),
38 : : unsigned char *out_buf,
39 : : int *pos,
40 : : void(*error)(char *x)) {
41 : : u8 *zbuf;
42 : : struct z_stream_s *strm;
43 : : int rc;
44 : : size_t out_len;
45 : :
46 : : rc = -1;
47 [ # # ]: 0 : if (flush) {
48 : : out_len = 0x8000; /* 32 K */
49 : : out_buf = malloc(out_len);
50 : : } else {
51 : 0 : out_len = ((size_t)~0) - (size_t)out_buf; /* no limit */
52 : : }
53 [ # # ]: 0 : if (!out_buf) {
54 : 0 : error("Out of memory while allocating output buffer");
55 : 0 : goto gunzip_nomem1;
56 : : }
57 : :
58 [ # # ]: 0 : if (buf)
59 : : zbuf = buf;
60 : : else {
61 : : zbuf = malloc(GZIP_IOBUF_SIZE);
62 : : len = 0;
63 : : }
64 [ # # ]: 0 : if (!zbuf) {
65 : 0 : error("Out of memory while allocating input buffer");
66 : 0 : goto gunzip_nomem2;
67 : : }
68 : :
69 : : strm = malloc(sizeof(*strm));
70 [ # # ]: 0 : if (strm == NULL) {
71 : 0 : error("Out of memory while allocating z_stream");
72 : 0 : goto gunzip_nomem3;
73 : : }
74 : :
75 [ # # ]: 0 : strm->workspace = malloc(flush ? zlib_inflate_workspacesize() :
76 : : sizeof(struct inflate_state));
77 [ # # ]: 0 : if (strm->workspace == NULL) {
78 : 0 : error("Out of memory while allocating workspace");
79 : 0 : goto gunzip_nomem4;
80 : : }
81 : :
82 [ # # ]: 0 : if (!fill)
83 : : fill = nofill;
84 : :
85 [ # # ]: 0 : if (len == 0)
86 : 0 : len = fill(zbuf, GZIP_IOBUF_SIZE);
87 : :
88 : : /* verify the gzip header */
89 [ # # ][ # # ]: 0 : if (len < 10 ||
90 [ # # ][ # # ]: 0 : zbuf[0] != 0x1f || zbuf[1] != 0x8b || zbuf[2] != 0x08) {
91 [ # # ]: 0 : if (pos)
92 : 0 : *pos = 0;
93 : 0 : error("Not a gzip file");
94 : 0 : goto gunzip_5;
95 : : }
96 : :
97 : : /* skip over gzip header (1f,8b,08... 10 bytes total +
98 : : * possible asciz filename)
99 : : */
100 : 0 : strm->next_in = zbuf + 10;
101 : 0 : strm->avail_in = len - 10;
102 : : /* skip over asciz filename */
103 [ # # ]: 0 : if (zbuf[3] & 0x8) {
104 : : do {
105 : : /*
106 : : * If the filename doesn't fit into the buffer,
107 : : * the file is very probably corrupt. Don't try
108 : : * to read more data.
109 : : */
110 [ # # ]: 0 : if (strm->avail_in == 0) {
111 : 0 : error("header error");
112 : 0 : goto gunzip_5;
113 : : }
114 : 0 : --strm->avail_in;
115 [ # # ]: 0 : } while (*strm->next_in++);
116 : : }
117 : :
118 : 0 : strm->next_out = out_buf;
119 : 0 : strm->avail_out = out_len;
120 : :
121 : 0 : rc = zlib_inflateInit2(strm, -MAX_WBITS);
122 : :
123 [ # # ]: 0 : if (!flush) {
124 : 0 : WS(strm)->inflate_state.wsize = 0;
125 : 0 : WS(strm)->inflate_state.window = NULL;
126 : : }
127 : :
128 [ # # ]: 0 : while (rc == Z_OK) {
129 [ # # ]: 0 : if (strm->avail_in == 0) {
130 : : /* TODO: handle case where both pos and fill are set */
131 : 0 : len = fill(zbuf, GZIP_IOBUF_SIZE);
132 [ # # ]: 0 : if (len < 0) {
133 : : rc = -1;
134 : 0 : error("read error");
135 : 0 : break;
136 : : }
137 : 0 : strm->next_in = zbuf;
138 : 0 : strm->avail_in = len;
139 : : }
140 : 0 : rc = zlib_inflate(strm, 0);
141 : :
142 : : /* Write any data generated */
143 [ # # ][ # # ]: 0 : if (flush && strm->next_out > out_buf) {
144 : 0 : int l = strm->next_out - out_buf;
145 [ # # ]: 0 : if (l != flush(out_buf, l)) {
146 : : rc = -1;
147 : 0 : error("write error");
148 : 0 : break;
149 : : }
150 : 0 : strm->next_out = out_buf;
151 : 0 : strm->avail_out = out_len;
152 : : }
153 : :
154 : : /* after Z_FINISH, only Z_STREAM_END is "we unpacked it all" */
155 [ # # ]: 0 : if (rc == Z_STREAM_END) {
156 : : rc = 0;
157 : : break;
158 [ # # ]: 0 : } else if (rc != Z_OK) {
159 : 0 : error("uncompression error");
160 : : rc = -1;
161 : : }
162 : : }
163 : :
164 : 0 : zlib_inflateEnd(strm);
165 [ # # ]: 0 : if (pos)
166 : : /* add + 8 to skip over trailer */
167 : 0 : *pos = strm->next_in - zbuf+8;
168 : :
169 : : gunzip_5:
170 : 0 : free(strm->workspace);
171 : : gunzip_nomem4:
172 : 0 : free(strm);
173 : : gunzip_nomem3:
174 [ # # ]: 0 : if (!buf)
175 : 0 : free(zbuf);
176 : : gunzip_nomem2:
177 [ # # ]: 0 : if (flush)
178 : 0 : free(out_buf);
179 : : gunzip_nomem1:
180 : 0 : return rc; /* returns Z_OK (0) if successful */
181 : : }
182 : :
183 : : #define decompress gunzip
|