123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135 |
- // Copyright (c) 2011 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #include "base/native_library.h"
- #include <dlfcn.h>
- #include <mach-o/getsect.h>
- #include "base/files/file_path.h"
- #include "base/files/file_util.h"
- #include "base/logging.h"
- #include "base/mac/scoped_cftyperef.h"
- #include "base/strings/strcat.h"
- #include "base/strings/string_piece.h"
- #include "base/strings/string_util.h"
- #include "base/strings/utf_string_conversions.h"
- #include "base/threading/thread_restrictions.h"
- namespace base {
- static NativeLibraryObjCStatus GetObjCStatusForImage(
- const void* function_pointer) {
- Dl_info info;
- if (!dladdr(function_pointer, &info))
- return OBJC_UNKNOWN;
- // See if the image contains an "ObjC image info" segment. This method
- // of testing is used in _CFBundleGrokObjcImageInfoFromFile in
- // CF-744/CFBundle.c, around lines 2447-2474.
- //
- // In 64-bit images, ObjC can be recognized in __DATA,__objc_imageinfo.
- const section_64* section = getsectbynamefromheader_64(
- reinterpret_cast<const struct mach_header_64*>(info.dli_fbase), SEG_DATA,
- "__objc_imageinfo");
- if (section)
- return OBJC_PRESENT;
- // ....except when "SharedRegionEncodingV2" is on, it's in
- // __DATA_CONST,__objc_image_info (see https://crbug.com/1220459#c16)
- section = getsectbynamefromheader_64(
- reinterpret_cast<const struct mach_header_64*>(info.dli_fbase),
- "__DATA_CONST", "__objc_imageinfo");
- return section ? OBJC_PRESENT : OBJC_NOT_PRESENT;
- }
- std::string NativeLibraryLoadError::ToString() const {
- return message;
- }
- NativeLibrary LoadNativeLibraryWithOptions(const FilePath& library_path,
- const NativeLibraryOptions& options,
- NativeLibraryLoadError* error) {
- // dlopen() etc. open the file off disk.
- if (library_path.Extension() == "dylib" || !DirectoryExists(library_path)) {
- void* dylib = dlopen(library_path.value().c_str(), RTLD_LAZY);
- if (!dylib) {
- if (error)
- error->message = dlerror();
- return nullptr;
- }
- NativeLibrary native_lib = new NativeLibraryStruct();
- native_lib->type = DYNAMIC_LIB;
- native_lib->dylib = dylib;
- native_lib->objc_status = OBJC_UNKNOWN;
- return native_lib;
- }
- ScopedCFTypeRef<CFURLRef> url(CFURLCreateFromFileSystemRepresentation(
- kCFAllocatorDefault, (const UInt8*)library_path.value().c_str(),
- checked_cast<CFIndex>(library_path.value().length()), true));
- if (!url)
- return nullptr;
- CFBundleRef bundle = CFBundleCreate(kCFAllocatorDefault, url.get());
- if (!bundle)
- return nullptr;
- NativeLibrary native_lib = new NativeLibraryStruct();
- native_lib->type = BUNDLE;
- native_lib->bundle = bundle;
- native_lib->bundle_resource_ref = CFBundleOpenBundleResourceMap(bundle);
- native_lib->objc_status = OBJC_UNKNOWN;
- return native_lib;
- }
- void UnloadNativeLibrary(NativeLibrary library) {
- if (library->objc_status == OBJC_NOT_PRESENT) {
- if (library->type == BUNDLE) {
- CFBundleCloseBundleResourceMap(library->bundle,
- library->bundle_resource_ref);
- CFRelease(library->bundle);
- } else {
- dlclose(library->dylib);
- }
- } else {
- VLOG(2) << "Not unloading NativeLibrary because it may contain an ObjC "
- "segment. library->objc_status = " << library->objc_status;
- // Deliberately do not CFRelease the bundle or dlclose the dylib because
- // doing so can corrupt the ObjC runtime method caches. See
- // http://crbug.com/172319 for details.
- }
- delete library;
- }
- void* GetFunctionPointerFromNativeLibrary(NativeLibrary library,
- StringPiece name) {
- void* function_pointer = nullptr;
- // Get the function pointer using the right API for the type.
- if (library->type == BUNDLE) {
- ScopedCFTypeRef<CFStringRef> symbol_name(CFStringCreateWithCString(
- kCFAllocatorDefault, name.data(), kCFStringEncodingUTF8));
- function_pointer = CFBundleGetFunctionPointerForName(library->bundle,
- symbol_name);
- } else {
- function_pointer = dlsym(library->dylib, name.data());
- }
- // If this library hasn't been tested for having ObjC, use the function
- // pointer to look up the section information for the library.
- if (function_pointer && library->objc_status == OBJC_UNKNOWN)
- library->objc_status = GetObjCStatusForImage(function_pointer);
- return function_pointer;
- }
- std::string GetNativeLibraryName(StringPiece name) {
- DCHECK(IsStringASCII(name));
- return StrCat({"lib", name, ".dylib"});
- }
- std::string GetLoadableModuleName(StringPiece name) {
- DCHECK(IsStringASCII(name));
- return StrCat({name, ".so"});
- }
- } // namespace base
|