chrome/installer/util/lzma_util.h

   1 // Copyright (c) 2009 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_INSTALLER_UTIL_LZMA_UTIL_H_
   6 #define CHROME_INSTALLER_UTIL_LZMA_UTIL_H_
   7
   8 #include <windows.h>
   9
  10 #include <set>
  11 #include <string>
  12
  13 #include "base/basictypes.h"
  14
  15 namespace base {
  16 class FilePath;
  17 }
  18
  19 // This is a utility class that acts as a wrapper around LZMA SDK library
  20 class LzmaUtil {
  21  public:
  22   // Utility method that does the job of calling OpenArchive(), UnPack()
  23   // and CloseArchive() in order. Returns error code (NO_ERROR if successful).
  24   static int32 UnPackArchive(const std::wstring& archive,
  25                              const std::wstring& output_dir,
  26                              std::wstring* output_file);
  27
  28   LzmaUtil();
  29   ~LzmaUtil();
  30
  31   DWORD OpenArchive(const std::wstring& archivePath);
  32
  33   // Unpacks the archive to the given location
  34   DWORD UnPack(const std::wstring& location);
  35
  36   // Unpacks the archive to the given location and returns the last file
  37   // extracted from archive. |single_file| is set to true iff only a single
  38   // file is extracted from archive.
  39   DWORD UnPack(const std::wstring& location,
  40                std::wstring* output_file);
  41
  42   void CloseArchive();
  43
  44  protected:
  45   bool CreateDirectory(const base::FilePath& dir);
  46
  47  private:
  48   HANDLE archive_handle_;
  49   std::set<std::wstring> directories_created_;
  50
  51   DISALLOW_COPY_AND_ASSIGN(LzmaUtil);
  52 };
  53
  54 #endif  // CHROME_INSTALLER_UTIL_LZMA_UTIL_H_