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: // file_id.cc: Return a unique identifier for a file
michael@0: //
michael@0: // See file_id.h for documentation
michael@0: //
michael@0: // Author: Dan Waylonis
michael@0: 
michael@0: #include <fcntl.h>
michael@0: #include <string.h>
michael@0: #include <unistd.h>
michael@0: 
michael@0: #include "common/mac/file_id.h"
michael@0: #include "common/mac/macho_id.h"
michael@0: 
michael@0: using MacFileUtilities::MachoID;
michael@0: 
michael@0: namespace google_breakpad {
michael@0: 
michael@0: FileID::FileID(const char *path) {
michael@0:   strlcpy(path_, path, sizeof(path_));
michael@0: }
michael@0: 
michael@0: bool FileID::FileIdentifier(unsigned char identifier[16]) {
michael@0:   int fd = open(path_, O_RDONLY);
michael@0:   if (fd == -1)
michael@0:     return false;
michael@0: 
michael@0:   MD5Context md5;
michael@0:   MD5Init(&md5);
michael@0: 
michael@0:   // Read 4k x 2 bytes at a time.  This is faster than just 4k bytes, but
michael@0:   // doesn't seem to be an unreasonable size for the stack.
michael@0:   unsigned char buffer[4096 * 2];
michael@0:   size_t buffer_size = sizeof(buffer);
michael@0:   while ((buffer_size = read(fd, buffer, buffer_size) > 0)) {
michael@0:     MD5Update(&md5, buffer, buffer_size);
michael@0:   }
michael@0: 
michael@0:   close(fd);
michael@0:   MD5Final(identifier, &md5);
michael@0: 
michael@0:   return true;
michael@0: }
michael@0: 
michael@0: bool FileID::MachoIdentifier(cpu_type_t cpu_type,
michael@0:                              cpu_subtype_t cpu_subtype,
michael@0:                              unsigned char identifier[16]) {
michael@0:   MachoID macho(path_);
michael@0: 
michael@0:   if (macho.UUIDCommand(cpu_type, cpu_subtype, identifier))
michael@0:     return true;
michael@0: 
michael@0:   return macho.MD5(cpu_type, cpu_subtype, identifier);
michael@0: }
michael@0: 
michael@0: // static
michael@0: void FileID::ConvertIdentifierToString(const unsigned char identifier[16],
michael@0:                                        char *buffer, int buffer_length) {
michael@0:   int buffer_idx = 0;
michael@0:   for (int idx = 0; (buffer_idx < buffer_length) && (idx < 16); ++idx) {
michael@0:     int hi = (identifier[idx] >> 4) & 0x0F;
michael@0:     int lo = (identifier[idx]) & 0x0F;
michael@0: 
michael@0:     if (idx == 4 || idx == 6 || idx == 8 || idx == 10)
michael@0:       buffer[buffer_idx++] = '-';
michael@0: 
michael@0:     buffer[buffer_idx++] =
michael@0:         static_cast<char>((hi >= 10) ? ('A' + hi - 10) : ('0' + hi));
michael@0:     buffer[buffer_idx++] =
michael@0:         static_cast<char>((lo >= 10) ? ('A' + lo - 10) : ('0' + lo));
michael@0:   }
michael@0: 
michael@0:   // NULL terminate
michael@0:   buffer[(buffer_idx < buffer_length) ? buffer_idx : buffer_idx - 1] = 0;
michael@0: }
michael@0: 
michael@0: }  // namespace google_breakpad