Enable LZMA compression in bmpbklk_utility.

utility/bmpblk_util.c

 // Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <errno.h>
 #include <fcntl.h>
 #include <limits.h>
+#include <lzma.h>
 #include <stdio.h>
 #include <string.h>
 #include <sys/mman.h>
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <unistd.h>
 
 #include "bmpblk_util.h"
 #include "eficompress.h"
 
@@ skipping 74 matching lines @@
 
 static void *do_efi_decompress(ImageInfo *img) {
   void *ibuf;
   void *sbuf;
   void *obuf;
   uint32_t isize;
   uint32_t ssize;
   uint32_t osize;
   EFI_STATUS r;
 
-  ibuf = ((void *)img) + sizeof(ImageInfo);
+  ibuf = (void*)(img + 1);
   isize = img->compressed_size;
 
   r = EfiGetInfo(ibuf, isize, &osize, &ssize);
   if (EFI_SUCCESS != r) {
     fprintf(stderr, "EfiGetInfo() failed with code %d\n",
             r);
     return 0;
   }
 
   sbuf = malloc(ssize);
@@ skipping 19 matching lines @@
     free(obuf);
     free(sbuf);
     return 0;
   }
 
   free(sbuf);
   return obuf;
 }
 
 
+
+static void *do_lzma_decompress(ImageInfo *img) {
+  void *ibuf;
+  void *obuf;
+  uint32_t isize;
+  uint32_t osize;
+  lzma_stream stream = LZMA_STREAM_INIT;
+  lzma_ret result;
+
+  ibuf = (void*)(img + 1);
+  isize = img->compressed_size;
+  osize = img->original_size;
+  obuf = malloc(osize);
+  if (!obuf) {
+    fprintf(stderr, "Can't allocate %d bytes: %s\n",
+            osize,
+            strerror(errno));
+    return 0;
+  }
+
+  result = lzma_auto_decoder(&stream, -1, 0);
+  if (result != LZMA_OK) {
+    fprintf(stderr, "Unable to initialize auto decoder (error: %d)!\n",
+            result);
+    free(obuf);
+    return 0;
+  }
+
+  stream.next_in = ibuf;
+  stream.avail_in = isize;
+  stream.next_out = obuf;
+  stream.avail_out = osize;
+  result = lzma_code(&stream, LZMA_FINISH);
+  if (result != LZMA_STREAM_END) {
+    fprintf(stderr, "Unalbe to decode data (error: %d)!\n", result);
+    free(obuf);
+    return 0;
+  }
+  lzma_end(&stream);
+  return obuf;
+}
+
+
+
 // Show what's inside. If todir is NULL, just print. Otherwise unpack.
 int dump_bmpblock(const char *infile, int show_as_yaml,
                   const char *todir, int overwrite) {
   void *ptr, *data_ptr;
   size_t length = 0;
   BmpBlockHeader *hdr;
   ImageInfo *img;
   ScreenLayout *scr;
   int loc_num;
   int screen_num;
@@ skipping 107 matching lines @@
       case COMPRESS_EFIv1:
         data_ptr = do_efi_decompress(img);
         if (!data_ptr) {
           fclose(bfp);
           fclose(yfp);
           discard_file(ptr, length);
           return 1;
         }
         free_data = 1;
         break;
+      case COMPRESS_LZMA1:
+        data_ptr = do_lzma_decompress(img);
+        if (!data_ptr) {
+          fclose(bfp);
+          fclose(yfp);
+          discard_file(ptr, length);
+          return 1;
+        }
+        free_data = 1;
+        break;
       default:
         fprintf(stderr, "Unsupported compression method encountered.\n");
         fclose(bfp);
         fclose(yfp);
         discard_file(ptr, length);
         return 1;
       }
       if (1 != fwrite(data_ptr, img->original_size, 1, bfp)) {
         fprintf(stderr, "Unable to write %s: %s\n", full_path_name,
                 strerror(errno));
@@ skipping 48 matching lines @@
   }
 
   if (todir)
     fclose(yfp);
 
   discard_file(ptr, length);
 
   return 0;
 }