X-Git-Url: https://git.saurik.com/apple/dyld.git/blobdiff_plain/d3f1e533acc7f70659b8bde9b6c040974f05e03b..bc3b7c8cda49ed8598284a489c0bb9694c67c6a4:/dyld3/MachOAppCache.cpp

diff --git a/dyld3/MachOAppCache.cpp b/dyld3/MachOAppCache.cpp
new file mode 100644
index 0000000..b8130ad
--- /dev/null
+++ b/dyld3/MachOAppCache.cpp
@@ -0,0 +1,163 @@
+/*
+* Copyright (c) 2017 Apple Inc. All rights reserved.
+*
+* @APPLE_LICENSE_HEADER_START@
+*
+* This file contains Original Code and/or Modifications of Original Code
+* as defined in and that are subject to the Apple Public Source License
+* Version 2.0 (the 'License'). You may not use this file except in
+* compliance with the License. Please obtain a copy of the License at
+* http://www.opensource.apple.com/apsl/ and read it before using this
+* file.
+*
+* The Original Code and all software distributed under the License are
+* distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+* EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+* INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+* FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+* Please see the License for the specific language governing rights and
+* limitations under the License.
+*
+* @APPLE_LICENSE_HEADER_END@
+*/
+
+#include "MachOAppCache.h"
+
+#include <list>
+
+#include <CoreFoundation/CFArray.h>
+#include <CoreFoundation/CFPropertyList.h>
+#include <CoreFoundation/CFString.h>
+
+#ifndef LC_FILESET_ENTRY
+#define LC_FILESET_ENTRY      (0x35 | LC_REQ_DYLD) /* used with fileset_entry_command */
+struct fileset_entry_command {
+    uint32_t        cmd;        /* LC_FILESET_ENTRY */
+    uint32_t        cmdsize;    /* includes id string */
+    uint64_t        vmaddr;     /* memory address of the dylib */
+    uint64_t        fileoff;    /* file offset of the dylib */
+    union lc_str    entry_id;   /* contained entry id */
+    uint32_t        reserved;   /* entry_id is 32-bits long, so this is the reserved padding */
+};
+#endif
+
+namespace dyld3 {
+
+void MachOAppCache::forEachDylib(Diagnostics& diag, void (^callback)(const MachOAnalyzer* ma, const char* name, bool& stop)) const {
+    const intptr_t slide = getSlide();
+    forEachLoadCommand(diag, ^(const load_command *cmd, bool &stop) {
+        if (cmd->cmd == LC_FILESET_ENTRY) {
+            const fileset_entry_command* app_cache_cmd = (const fileset_entry_command*)cmd;
+            const char* name = (char*)app_cache_cmd + app_cache_cmd->entry_id.offset;
+            callback((const MachOAnalyzer*)(app_cache_cmd->vmaddr + slide), name, stop);
+            return;
+        }
+    });
+}
+
+void MachOAppCache::forEachPrelinkInfoLibrary(Diagnostics& diags,
+                                              void (^callback)(const char* bundleName, const char* relativePath,
+                                                               const std::vector<const char*>& deps)) const  {
+
+     __block std::list<std::string> nonASCIIStrings;
+     auto getString = ^(Diagnostics& diags, CFStringRef symbolNameRef) {
+         const char* symbolName = CFStringGetCStringPtr(symbolNameRef, kCFStringEncodingUTF8);
+         if ( symbolName != nullptr )
+             return symbolName;
+
+         CFIndex len = CFStringGetMaximumSizeForEncoding(CFStringGetLength(symbolNameRef), kCFStringEncodingUTF8);
+         char buffer[len + 1];
+         if ( !CFStringGetCString(symbolNameRef, buffer, len, kCFStringEncodingUTF8) ) {
+             diags.error("Could not convert string to ASCII");
+             return (const char*)nullptr;
+         }
+         buffer[len] = '\0';
+         nonASCIIStrings.push_back(buffer);
+         return nonASCIIStrings.back().c_str();
+     };
+
+    const uint8_t* prelinkInfoBuffer = nullptr;
+    uint64_t prelinkInfoBufferSize = 0;
+    prelinkInfoBuffer = (const uint8_t*)findSectionContent("__PRELINK_INFO", "__info", prelinkInfoBufferSize);
+    if ( prelinkInfoBuffer == nullptr )
+        return;
+
+    CFReadStreamRef readStreamRef = CFReadStreamCreateWithBytesNoCopy(kCFAllocatorDefault, prelinkInfoBuffer, prelinkInfoBufferSize, kCFAllocatorNull);
+    if ( !CFReadStreamOpen(readStreamRef) ) {
+        fprintf(stderr, "Could not open plist stream\n");
+        exit(1);
+    }
+    CFErrorRef errorRef = nullptr;
+    CFPropertyListRef plistRef = CFPropertyListCreateWithStream(kCFAllocatorDefault, readStreamRef, prelinkInfoBufferSize, kCFPropertyListImmutable, nullptr, &errorRef);
+    if ( errorRef != nullptr ) {
+        CFStringRef stringRef = CFErrorCopyFailureReason(errorRef);
+        fprintf(stderr, "Could not read plist because: %s\n", CFStringGetCStringPtr(stringRef, kCFStringEncodingASCII));
+        CFRelease(stringRef);
+        exit(1);
+    }
+    assert(CFGetTypeID(plistRef) == CFDictionaryGetTypeID());
+
+    // Get the "_PrelinkInfoDictionary" array
+    CFArrayRef prelinkInfoDictionaryArrayRef = (CFArrayRef)CFDictionaryGetValue((CFDictionaryRef)plistRef, CFSTR("_PrelinkInfoDictionary"));
+    assert(CFGetTypeID(prelinkInfoDictionaryArrayRef) == CFArrayGetTypeID());
+
+    for (CFIndex i = 0; i != CFArrayGetCount(prelinkInfoDictionaryArrayRef); ++i) {
+        CFDictionaryRef kextInfoDictionary = (CFDictionaryRef)CFArrayGetValueAtIndex(prelinkInfoDictionaryArrayRef, i);
+        assert(CFGetTypeID(kextInfoDictionary) == CFDictionaryGetTypeID());
+
+        CFStringRef bundleIdentifierStringRef = (CFStringRef)CFDictionaryGetValue((CFDictionaryRef)kextInfoDictionary, CFSTR("CFBundleIdentifier"));
+        assert(CFGetTypeID(bundleIdentifierStringRef) == CFStringGetTypeID());
+
+        const char* bundleID = getString(diags, bundleIdentifierStringRef);
+        if ( bundleID == nullptr )
+            return;
+
+        const char* relativePath = nullptr;
+        CFStringRef relativePathStringRef = (CFStringRef)CFDictionaryGetValue((CFDictionaryRef)kextInfoDictionary, CFSTR("_PrelinkExecutableRelativePath"));
+        if ( relativePathStringRef != nullptr ) {
+            assert(CFGetTypeID(relativePathStringRef) == CFStringGetTypeID());
+            relativePath = getString(diags, relativePathStringRef);
+            if ( relativePath == nullptr )
+                return;
+        }
+
+        std::vector<const char*> dependencies;
+
+        CFDictionaryRef bundleLibrariesDictionaryRef = (CFDictionaryRef)CFDictionaryGetValue((CFDictionaryRef)kextInfoDictionary, CFSTR("OSBundleLibraries"));
+        if (bundleLibrariesDictionaryRef != nullptr) {
+            // Add the libraries to the dependencies
+            // If we didn't have bundle libraries then a placeholder was added
+            assert(CFGetTypeID(bundleLibrariesDictionaryRef) == CFDictionaryGetTypeID());
+
+            struct ApplyContext {
+                Diagnostics*                diagnostics;
+                std::vector<const char*>*   dependencies                                    = nullptr;
+                const char* (^getString)(Diagnostics& diags, CFStringRef symbolNameRef)     = nullptr;
+            };
+
+            CFDictionaryApplierFunction callback = [](const void *key, const void *value, void *context) {
+                CFStringRef keyStringRef = (CFStringRef)key;
+                assert(CFGetTypeID(keyStringRef) == CFStringGetTypeID());
+
+                ApplyContext* applyContext = (ApplyContext*)context;
+                const char* depString = applyContext->getString(*applyContext->diagnostics, keyStringRef);
+                if ( !depString )
+                    return;
+
+                applyContext->dependencies->push_back(depString);
+            };
+
+            ApplyContext applyContext = { &diags, &dependencies, getString };
+            CFDictionaryApplyFunction(bundleLibrariesDictionaryRef, callback, &applyContext);
+
+            if ( diags.hasError() )
+                return;
+        }
+        callback(bundleID, relativePath, dependencies);
+    }
+
+    CFRelease(plistRef);
+    CFRelease(readStreamRef);
+}
+
+} // namespace dyld3