Enable LZMA compression in bmpbklk_utility.

utility/bmpblk_utility.cc

 // 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.
 //
 // Utility for manipulating firmware screen block (BMPBLOCK) in GBB.
 //
 
 #include "bmpblk_utility.h"
 
 #include <assert.h>
 #include <errno.h>
 #include <getopt.h>
+#include <lzma.h>
 #include <stdarg.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <yaml.h>
 
 extern "C" {
 #include "eficompress.h"
 }
 
@@ skipping 296 matching lines @@
         if (EFI_SUCCESS != EfiCompress((uint8_t *)content.c_str(), tmpsize,
                                        tmpbuf, &tmpsize)) {
           error("Unable to compress!\n");
         }
         it->second.data.compression = compression_;
         it->second.compressed_content.assign((const char *)tmpbuf, tmpsize);
         it->second.data.compressed_size = tmpsize;
         free(tmpbuf);
       }
       break;
+    case COMPRESS_LZMA1:
+      {
+        // Calculate the worst case of buffer size.
+        uint32_t tmpsize = lzma_stream_buffer_bound(content.size());
+        uint8_t *tmpbuf = (uint8_t *)malloc(tmpsize);
+        lzma_stream stream = LZMA_STREAM_INIT;
+        lzma_options_lzma options;
+        lzma_ret result;
+
+        lzma_lzma_preset(&options, 9);
+        result = lzma_alone_encoder(&stream, &options);
+        if (result != LZMA_OK) {
+          error("Unable to initialize easy encoder (error: %d)!\n", result);
+        }
+
+        stream.next_in = (uint8_t *)content.data();
+        stream.avail_in = content.size();
+        stream.next_out = tmpbuf;
+        stream.avail_out = tmpsize;
+        result = lzma_code(&stream, LZMA_FINISH);
+        if (result != LZMA_STREAM_END) {
+          error("Unable to encode data (error: %d)!\n", result);
+        }
+
+        it->second.data.compression = compression_;
+        it->second.compressed_content.assign((const char *)tmpbuf,
+                                             tmpsize - stream.avail_out);
+        it->second.data.compressed_size = tmpsize - stream.avail_out;
+        lzma_end(&stream);
+        free(tmpbuf);
+      }
+      break;
     default:
       error("Unsupported compression method attempted.\n");
     }
   }
 }
 
 const string BmpBlockUtil::read_image_file(const char *filename) {
   string content;
   vector<char> buffer;
 
@@ skipping 189 matching lines @@
 static void usagehelp_exit(const char *prog_name) {
   printf(
     "\n"
     "To create a new BMPBLOCK file using config from YAML file:\n"
     "\n"
     "  %s [-z NUM] -c YAML BMPBLOCK\n"
     "\n"
     "    -z NUM  = compression algorithm to use\n"
     "              0 = none\n"
     "              1 = EFIv1\n"
-    "              2 = TBD\n"
+    "              2 = LZMA1\n"
     "\n", prog_name);
   printf(
     "To display the contents of a BMPBLOCK:\n"
     "\n"
     "  %s [-y] BMPBLOCK\n"
     "\n"
     "    -y  = display as yaml\n"
     "\n", prog_name);
   printf(
     "To unpack a BMPBLOCK file:\n"
@@ skipping 96 matching lines @@
   else if (extract_mode) {
     return dump_bmpblock(bmpblock_fn, 1, extract_dir, overwrite);
   } else {
     return dump_bmpblock(bmpblock_fn, show_as_yaml, 0, 0);
   }
 
   return 0;
 }
 
 #endif  // WITH_UTIL_MAIN