chrome/installer/mini_installer/pe_resource.h

   1 // Copyright (c) 2006-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_MINI_INSTALLER_PE_RESOURCE_H_
   6 #define CHROME_INSTALLER_MINI_INSTALLER_PE_RESOURCE_H_
   7
   8 #include <windows.h>
   9
  10 // This class models a windows PE resource. It does not pretend to be a full
  11 // API wrapper and it is just concerned with loading it to memory and writing
  12 // it to disk. Each resource is unique only in the context of a loaded module,
  13 // that is why you need to specify one on each constructor.
  14 class PEResource {
  15  public:
  16   // This ctor takes the handle to the resource and the module where it was
  17   // found. Ownership of the resource is transfered to this object.
  18   PEResource(HRSRC resource, HMODULE module);
  19
  20   // This ctor takes the resource name, the resource type and the module where
  21   // to look for the resource. If the resource is found IsValid() returns true.
  22   PEResource(const wchar_t* name, const wchar_t* type, HMODULE module);
  23
  24   // Returns true if the resource is valid.
  25   bool IsValid();
  26
  27   // Returns the size in bytes of the resource. Returns zero if the resource is
  28   // not valid.
  29   size_t Size();
  30
  31   // Creates a file in 'path' with a copy of the resource. If the resource can
  32   // not be loaded into memory or if it cannot be written to disk it returns
  33   // false.
  34   bool WriteToDisk(const wchar_t* path);
  35
  36  private:
  37   HRSRC resource_;
  38   HMODULE module_;
  39 };
  40
  41 #endif  // CHROME_INSTALLER_MINI_INSTALLER_PE_RESOURCE_H_