1  // Copyright 2006 Google LLC
  2  //
  3  // Redistribution and use in source and binary forms, with or without
  4  // modification, are permitted provided that the following conditions are
  5  // met:
  6  //
  7  //     * Redistributions of source code must retain the above copyright
  8  // notice, this list of conditions and the following disclaimer.
  9  //     * Redistributions in binary form must reproduce the above
 10  // copyright notice, this list of conditions and the following disclaimer
 11  // in the documentation and/or other materials provided with the
 12  // distribution.
 13  //     * Neither the name of Google LLC nor the names of its
 14  // contributors may be used to endorse or promote products derived from
 15  // this software without specific prior written permission.
 16  //
 17  // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 18  // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 19  // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 20  // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 21  // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 22  // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 23  // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 24  // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 25  // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 26  // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 27  // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 28  
 29  // macho_id.h: Functions to gather identifying information from a macho file
 30  //
 31  // Author: Dan Waylonis
 32  
 33  #ifndef COMMON_MAC_MACHO_ID_H__
 34  #define COMMON_MAC_MACHO_ID_H__
 35  
 36  #include <limits.h>
 37  #include <mach/machine.h>
 38  #include <mach-o/loader.h>
 39  
 40  #include "common/mac/macho_walker.h"
 41  #include "common/md5.h"
 42  
 43  namespace MacFileUtilities {
 44  
 45  class MachoID {
 46   public:
 47    MachoID(const char* path);
 48    MachoID(void* memory, size_t size);
 49    ~MachoID();
 50  
 51    // For the given |cpu_type| and |cpu_subtype|, return a UUID from the LC_UUID
 52    // command.
 53    // Return false if there isn't a LC_UUID command.
 54    bool UUIDCommand(cpu_type_t cpu_type,
 55                     cpu_subtype_t cpu_subtype,
 56                     unsigned char identifier[16]);
 57  
 58    // For the given |cpu_type|, and |cpu_subtype| return the MD5 for the mach-o
 59    // data segment(s).
 60    // Return true on success, false otherwise
 61    bool MD5(cpu_type_t cpu_type,
 62             cpu_subtype_t cpu_subtype,
 63             unsigned char identifier[16]);
 64  
 65   private:
 66    // Signature of class member function to be called with data read from file
 67    typedef void (MachoID::*UpdateFunction)(unsigned char* bytes, size_t size);
 68  
 69    // Update the MD5 value by examining |size| |bytes| and applying the algorithm
 70    // to each byte.
 71    void UpdateMD5(unsigned char* bytes, size_t size);
 72  
 73    // Bottleneck for update routines
 74    void Update(MachoWalker* walker, off_t offset, size_t size);
 75  
 76    // Factory for the MachoWalker
 77    bool WalkHeader(cpu_type_t cpu_type, cpu_subtype_t cpu_subtype,
 78                    MachoWalker::LoadCommandCallback callback, void* context);
 79  
 80    // The callback from the MachoWalker for CRC and MD5
 81    static bool WalkerCB(MachoWalker* walker, load_command* cmd, off_t offset,
 82                         bool swap, void* context);
 83  
 84    // The callback from the MachoWalker for LC_UUID
 85    static bool UUIDWalkerCB(MachoWalker* walker, load_command* cmd, off_t offset,
 86                             bool swap, void* context);
 87  
 88    // File path
 89    char path_[PATH_MAX];
 90  
 91    // Memory region to read from
 92    void* memory_;
 93  
 94    // Size of the memory region
 95    size_t memory_size_;
 96  
 97    // The MD5 context
 98    google_breakpad::MD5Context md5_context_;
 99  
100    // The current update to call from the Update callback
101    UpdateFunction update_function_;
102  };
103  
104  }  // namespace MacFileUtilities
105  
106  #endif  // COMMON_MAC_MACHO_ID_H__