15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2010 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// This file contains the interface for an iterator over a portable executable 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// file's resources. 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef CHROME_INSTALLER_TEST_PE_IMAGE_RESOURCES_H_ 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define CHROME_INSTALLER_TEST_PE_IMAGE_RESOURCES_H_ 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <windows.h> 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <vector> 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/basictypes.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace base { namespace win { class PEImage; } } 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace upgrade_test { 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A CopyConstructible and Assignable identifier for resource directory 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// entries. 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class EntryId { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit EntryId(WORD number) : number_(number) { } 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit EntryId(const std::wstring& name) : name_(name), number_() { 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_NE(static_cast<std::wstring::size_type>(0), name.size()); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool IsNamed() const { return !name_.empty(); } 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WORD number() const { return number_; } 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::wstring& name() const { return name_; } 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::wstring name_; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WORD number_; 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; // class EntryId 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A sequence of identifiers comprising the path from the root of an image's 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// resource directory to an individual resource. 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef std::vector<EntryId> EntryPath; 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A callback function invoked once for each data entry in the image's 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// directory of resources. 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |path| - the full path of the data entry, 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |data| - the address of the entry's data. 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |size| - the size, in bytes, of the entry's data. 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |code_page| - the code page to be used to interpret string data in the 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// entry's data. 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |context| - the context given to EnumResources. 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef void (*EnumResource_Fn)(const EntryPath& path, uint8* data, 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD size, DWORD code_page, uintptr_t context); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Enumerates all data entries in |image|'s resource directory. |callback| is 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// invoked (and provided with |context|) once per entry. Returns false if 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// some or all of the resource directory could not be parsed. 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool EnumResources(const base::win::PEImage& image, EnumResource_Fn callback, 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uintptr_t context); 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace upgrade_test 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // CHROME_INSTALLER_TEST_PE_IMAGE_RESOURCES_H_ 63