lib/decompress_unlzo.c

   1 /*
   2  * LZO decompressor for the Linux kernel. Code borrowed from the lzo
   3  * implementation by Markus Franz Xaver Johannes Oberhumer.
   4  *
   5  * Linux kernel adaptation:
   6  * Copyright (C) 2009
   7  * Albin Tonnerre, Free Electrons <albin.tonnerre@free-electrons.com>
   8  *
   9  * Original code:
  10  * Copyright (C) 1996-2005 Markus Franz Xaver Johannes Oberhumer
  11  * All Rights Reserved.
  12  *
  13  * lzop and the LZO library are free software; you can redistribute them
  14  * and/or modify them under the terms of the GNU General Public License as
  15  * published by the Free Software Foundation; either version 2 of
  16  * the License, or (at your option) any later version.
  17  *
  18  * This program is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU General Public License for more details.
  22  *
  23  * You should have received a copy of the GNU General Public License
  24  * along with this program; see the file COPYING.
  25  * If not, write to the Free Software Foundation, Inc.,
  26  * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  27  *
  28  * Markus F.X.J. Oberhumer
  29  * <markus@oberhumer.com>
  30  * http://www.oberhumer.com/opensource/lzop/
  31  */
  32
  33 #ifdef STATIC
  34 #define PREBOOT
  35 #include "lzo/lzo1x_decompress_safe.c"
  36 #else
  37 #include <linux/decompress/unlzo.h>
  38 #endif
  39
  40 #include <linux/types.h>
  41 #include <linux/lzo.h>
  42 #include <linux/decompress/mm.h>
  43
  44 #include <linux/compiler.h>
  45 #include <asm/unaligned.h>
  46
  47 static const unsigned char lzop_magic[] = {
  48         0x89, 0x4c, 0x5a, 0x4f, 0x00, 0x0d, 0x0a, 0x1a, 0x0a };
  49
  50 #define LZO_BLOCK_SIZE        (256*1024l)
  51 #define HEADER_HAS_FILTER      0x00000800L
  52 #define HEADER_SIZE_MIN       (9 + 7     + 4 + 8     + 1       + 4)
  53 #define HEADER_SIZE_MAX       (9 + 7 + 1 + 8 + 8 + 4 + 1 + 255 + 4)
  54
  55 STATIC inline long INIT parse_header(u8 *input, long *skip, long in_len)
  56 {
  57         int l;
  58         u8 *parse = input;
  59         u8 *end = input + in_len;
  60         u8 level = 0;
  61         u16 version;
  62
  63         /*
  64          * Check that there's enough input to possibly have a valid header.
  65          * Then it is possible to parse several fields until the minimum
  66          * size may have been used.
  67          */
  68         if (in_len < HEADER_SIZE_MIN)
  69                 return 0;
  70
  71         /* read magic: 9 first bits */
  72         for (l = 0; l < 9; l++) {
  73                 if (*parse++ != lzop_magic[l])
  74                         return 0;
  75         }
  76         /* get version (2bytes), skip library version (2),
  77          * 'need to be extracted' version (2) and
  78          * method (1) */
  79         version = get_unaligned_be16(parse);
  80         parse += 7;
  81         if (version >= 0x0940)
  82                 level = *parse++;
  83         if (get_unaligned_be32(parse) & HEADER_HAS_FILTER)
  84                 parse += 8; /* flags + filter info */
  85         else
  86                 parse += 4; /* flags */
  87
  88         /*
  89          * At least mode, mtime_low, filename length, and checksum must
  90          * be left to be parsed. If also mtime_high is present, it's OK
  91          * because the next input buffer check is after reading the
  92          * filename length.
  93          */
  94         if (end - parse < 8 + 1 + 4)
  95                 return 0;
  96
  97         /* skip mode and mtime_low */
  98         parse += 8;
  99         if (version >= 0x0940)
 100                 parse += 4;     /* skip mtime_high */
 101
 102         l = *parse++;
 103         /* don't care about the file name, and skip checksum */
 104         if (end - parse < l + 4)
 105                 return 0;
 106         parse += l + 4;
 107
 108         *skip = parse - input;
 109         return 1;
 110 }
 111
 112 STATIC int INIT unlzo(u8 *input, long in_len,
 113                                 long (*fill)(void *, unsigned long),
 114                                 long (*flush)(void *, unsigned long),
 115                                 u8 *output, long *posp,
 116                                 void (*error) (char *x))
 117 {
 118         u8 r = 0;
 119         long skip = 0;
 120         u32 src_len, dst_len;
 121         size_t tmp;
 122         u8 *in_buf, *in_buf_save, *out_buf;
 123         int ret = -1;
 124
 125         if (output) {
 126                 out_buf = output;
 127         } else if (!flush) {
 128                 error("NULL output pointer and no flush function provided");
 129                 goto exit;
 130         } else {
 131                 out_buf = malloc(LZO_BLOCK_SIZE);
 132                 if (!out_buf) {
 133                         error("Could not allocate output buffer");
 134                         goto exit;
 135                 }
 136         }
 137
 138         if (input && fill) {
 139                 error("Both input pointer and fill function provided, don't know what to do");
 140                 goto exit_1;
 141         } else if (input) {
 142                 in_buf = input;
 143         } else if (!fill) {
 144                 error("NULL input pointer and missing fill function");
 145                 goto exit_1;
 146         } else {
 147                 in_buf = malloc(lzo1x_worst_compress(LZO_BLOCK_SIZE));
 148                 if (!in_buf) {
 149                         error("Could not allocate input buffer");
 150                         goto exit_1;
 151                 }
 152         }
 153         in_buf_save = in_buf;
 154
 155         if (posp)
 156                 *posp = 0;
 157
 158         if (fill) {
 159                 /*
 160                  * Start from in_buf + HEADER_SIZE_MAX to make it possible
 161                  * to use memcpy() to copy the unused data to the beginning
 162                  * of the buffer. This way memmove() isn't needed which
 163                  * is missing from pre-boot environments of most archs.
 164                  */
 165                 in_buf += HEADER_SIZE_MAX;
 166                 in_len = fill(in_buf, HEADER_SIZE_MAX);
 167         }
 168
 169         if (!parse_header(in_buf, &skip, in_len)) {
 170                 error("invalid header");
 171                 goto exit_2;
 172         }
 173         in_buf += skip;
 174         in_len -= skip;
 175
 176         if (fill) {
 177                 /* Move the unused data to the beginning of the buffer. */
 178                 memcpy(in_buf_save, in_buf, in_len);
 179                 in_buf = in_buf_save;
 180         }
 181
 182         if (posp)
 183                 *posp = skip;
 184
 185         for (;;) {
 186                 /* read uncompressed block size */
 187                 if (fill && in_len < 4) {
 188                         skip = fill(in_buf + in_len, 4 - in_len);
 189                         if (skip > 0)
 190                                 in_len += skip;
 191                 }
 192                 if (in_len < 4) {
 193                         error("file corrupted");
 194                         goto exit_2;
 195                 }
 196                 dst_len = get_unaligned_be32(in_buf);
 197                 in_buf += 4;
 198                 in_len -= 4;
 199
 200                 /* exit if last block */
 201                 if (dst_len == 0) {
 202                         if (posp)
 203                                 *posp += 4;
 204                         break;
 205                 }
 206
 207                 if (dst_len > LZO_BLOCK_SIZE) {
 208                         error("dest len longer than block size");
 209                         goto exit_2;
 210                 }
 211
 212                 /* read compressed block size, and skip block checksum info */
 213                 if (fill && in_len < 8) {
 214                         skip = fill(in_buf + in_len, 8 - in_len);
 215                         if (skip > 0)
 216                                 in_len += skip;
 217                 }
 218                 if (in_len < 8) {
 219                         error("file corrupted");
 220                         goto exit_2;
 221                 }
 222                 src_len = get_unaligned_be32(in_buf);
 223                 in_buf += 8;
 224                 in_len -= 8;
 225
 226                 if (src_len <= 0 || src_len > dst_len) {
 227                         error("file corrupted");
 228                         goto exit_2;
 229                 }
 230
 231                 /* decompress */
 232                 if (fill && in_len < src_len) {
 233                         skip = fill(in_buf + in_len, src_len - in_len);
 234                         if (skip > 0)
 235                                 in_len += skip;
 236                 }
 237                 if (in_len < src_len) {
 238                         error("file corrupted");
 239                         goto exit_2;
 240                 }
 241                 tmp = dst_len;
 242
 243                 /* When the input data is not compressed at all,
 244                  * lzo1x_decompress_safe will fail, so call memcpy()
 245                  * instead */
 246                 if (unlikely(dst_len == src_len))
 247                         memcpy(out_buf, in_buf, src_len);
 248                 else {
 249                         r = lzo1x_decompress_safe((u8 *) in_buf, src_len,
 250                                                 out_buf, &tmp);
 251
 252                         if (r != LZO_E_OK || dst_len != tmp) {
 253                                 error("Compressed data violation");
 254                                 goto exit_2;
 255                         }
 256                 }
 257
 258                 if (flush && flush(out_buf, dst_len) != dst_len)
 259                         goto exit_2;
 260                 if (output)
 261                         out_buf += dst_len;
 262                 if (posp)
 263                         *posp += src_len + 12;
 264
 265                 in_buf += src_len;
 266                 in_len -= src_len;
 267                 if (fill) {
 268                         /*
 269                          * If there happens to still be unused data left in
 270                          * in_buf, move it to the beginning of the buffer.
 271                          * Use a loop to avoid memmove() dependency.
 272                          */
 273                         if (in_len > 0)
 274                                 for (skip = 0; skip < in_len; ++skip)
 275                                         in_buf_save[skip] = in_buf[skip];
 276                         in_buf = in_buf_save;
 277                 }
 278         }
 279
 280         ret = 0;
 281 exit_2:
 282         if (!input)
 283                 free(in_buf_save);
 284 exit_1:
 285         if (!output)
 286                 free(out_buf);
 287 exit:
 288         return ret;
 289 }
 290
 291 #ifdef PREBOOT
 292 STATIC int INIT __decompress(unsigned char *buf, long len,
 293                            long (*fill)(void*, unsigned long),
 294                            long (*flush)(void*, unsigned long),
 295                            unsigned char *out_buf, long olen,
 296                            long *pos,
 297                            void (*error)(char *x))
 298 {
 299         return unlzo(buf, len, fill, flush, out_buf, pos, error);
 300 }
 301 #endif