michael@0: // Copyright (c) 2006, Google Inc. michael@0: // All rights reserved. michael@0: // michael@0: // Redistribution and use in source and binary forms, with or without michael@0: // modification, are permitted provided that the following conditions are michael@0: // met: michael@0: // michael@0: // * Redistributions of source code must retain the above copyright michael@0: // notice, this list of conditions and the following disclaimer. michael@0: // * Redistributions in binary form must reproduce the above michael@0: // copyright notice, this list of conditions and the following disclaimer michael@0: // in the documentation and/or other materials provided with the michael@0: // distribution. michael@0: // * Neither the name of Google Inc. nor the names of its michael@0: // contributors may be used to endorse or promote products derived from michael@0: // this software without specific prior written permission. michael@0: // michael@0: // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS michael@0: // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT michael@0: // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR michael@0: // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT michael@0: // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, michael@0: // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT michael@0: // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, michael@0: // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY michael@0: // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT michael@0: // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE michael@0: // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. michael@0: michael@0: // macho_id.h: Functions to gather identifying information from a macho file michael@0: // michael@0: // Author: Dan Waylonis michael@0: michael@0: #ifndef COMMON_MAC_MACHO_ID_H__ michael@0: #define COMMON_MAC_MACHO_ID_H__ michael@0: michael@0: #include michael@0: #include michael@0: #include michael@0: michael@0: #include "common/mac/macho_walker.h" michael@0: #include "common/md5.h" michael@0: michael@0: namespace MacFileUtilities { michael@0: michael@0: class MachoID { michael@0: public: michael@0: MachoID(const char *path); michael@0: MachoID(const char *path, void *memory, size_t size); michael@0: ~MachoID(); michael@0: michael@0: // For the given |cpu_type| and |cpu_subtype|, return a UUID from the LC_UUID michael@0: // command. michael@0: // Return false if there isn't a LC_UUID command. michael@0: bool UUIDCommand(cpu_type_t cpu_type, michael@0: cpu_subtype_t cpu_subtype, michael@0: unsigned char identifier[16]); michael@0: michael@0: // For the given |cpu_type| and |cpu_subtype|, return a UUID from the michael@0: // LC_ID_DYLIB command. michael@0: // Return false if there isn't a LC_ID_DYLIB command. michael@0: bool IDCommand(cpu_type_t cpu_type, michael@0: cpu_subtype_t cpu_subtype, michael@0: unsigned char identifier[16]); michael@0: michael@0: // For the given |cpu_type| and |cpu_subtype|, return the Adler32 CRC for the michael@0: // mach-o data segment(s). michael@0: // Return 0 on error (e.g., if the file is not a mach-o file) michael@0: uint32_t Adler32(cpu_type_t cpu_type, michael@0: cpu_subtype_t cpu_subtype); michael@0: michael@0: // For the given |cpu_type|, and |cpu_subtype| return the MD5 for the mach-o michael@0: // data segment(s). michael@0: // Return true on success, false otherwise michael@0: bool MD5(cpu_type_t cpu_type, michael@0: cpu_subtype_t cpu_subtype, michael@0: unsigned char identifier[16]); michael@0: michael@0: private: michael@0: // Signature of class member function to be called with data read from file michael@0: typedef void (MachoID::*UpdateFunction)(unsigned char *bytes, size_t size); michael@0: michael@0: // Update the CRC value by examining |size| |bytes| and applying the algorithm michael@0: // to each byte. michael@0: void UpdateCRC(unsigned char *bytes, size_t size); michael@0: michael@0: // Update the MD5 value by examining |size| |bytes| and applying the algorithm michael@0: // to each byte. michael@0: void UpdateMD5(unsigned char *bytes, size_t size); michael@0: michael@0: // Bottleneck for update routines michael@0: void Update(MachoWalker *walker, off_t offset, size_t size); michael@0: michael@0: // Factory for the MachoWalker michael@0: bool WalkHeader(cpu_type_t cpu_type, cpu_subtype_t cpu_subtype, michael@0: MachoWalker::LoadCommandCallback callback, void *context); michael@0: michael@0: // The callback from the MachoWalker for CRC and MD5 michael@0: static bool WalkerCB(MachoWalker *walker, load_command *cmd, off_t offset, michael@0: bool swap, void *context); michael@0: michael@0: // The callback from the MachoWalker for LC_UUID michael@0: static bool UUIDWalkerCB(MachoWalker *walker, load_command *cmd, off_t offset, michael@0: bool swap, void *context); michael@0: michael@0: // The callback from the MachoWalker for LC_ID_DYLIB michael@0: static bool IDWalkerCB(MachoWalker *walker, load_command *cmd, off_t offset, michael@0: bool swap, void *context); michael@0: michael@0: // File path michael@0: char path_[PATH_MAX]; michael@0: michael@0: // Memory region to read from michael@0: void *memory_; michael@0: michael@0: // Size of the memory region michael@0: size_t memory_size_; michael@0: michael@0: // The current crc value michael@0: uint32_t crc_; michael@0: michael@0: // The MD5 context michael@0: google_breakpad::MD5Context md5_context_; michael@0: michael@0: // The current update to call from the Update callback michael@0: UpdateFunction update_function_; michael@0: }; michael@0: michael@0: } // namespace MacFileUtilities michael@0: michael@0: #endif // COMMON_MAC_MACHO_ID_H__