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.
14class 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_
42