Index: lib/ReaderWriter/MachO/MachONormalizedFileBinaryReader.cpp =================================================================== --- lib/ReaderWriter/MachO/MachONormalizedFileBinaryReader.cpp +++ lib/ReaderWriter/MachO/MachONormalizedFileBinaryReader.cpp @@ -376,6 +376,15 @@ } } } + if (cmd == LC_ID_DYLIB) { + const dylib_command *dl = reinterpret_cast(lc); + dylib_command tempDL; + if (swap) { + tempDL = *dl; swapStruct(tempDL); dl = &tempDL; + } + + f->installName = lc + dl->dylib.name; + } return false; }); if (ec) Index: lib/ReaderWriter/MachO/MachONormalizedFileBinaryWriter.cpp =================================================================== --- lib/ReaderWriter/MachO/MachONormalizedFileBinaryWriter.cpp +++ lib/ReaderWriter/MachO/MachONormalizedFileBinaryWriter.cpp @@ -346,6 +346,12 @@ ++count; } + // If creating a dylib, add LC_ID_DYLIB. + if (_file.fileType == llvm::MachO::MH_DYLIB) { + size += sizeof(dylib_command) + pointerAlign(_file.installName.size() + 1); + ++count; + } + // Add LC_LOAD_DYLIB for each dependent dylib. for (const DependentDylib &dep : _file.dependentDylibs) { size += sizeof(dylib_command) + pointerAlign(dep.path.size()+1); @@ -704,6 +710,24 @@ lc += sizeof(entry_point_command); } + // Add LC_ID_DYLIB command for dynamic libraries. + if (_file.fileType == llvm::MachO::MH_DYLIB) { + dylib_command *dc = reinterpret_cast(lc); + StringRef path = _file.installName; + uint32_t size = sizeof(dylib_command) + pointerAlign(path.size() + 1); + dc->cmd = LC_ID_DYLIB; + dc->cmdsize = size; + dc->dylib.name = sizeof(dylib_command); // offset + dc->dylib.timestamp = 0; // FIXME + dc->dylib.current_version = 0; // FIXME + dc->dylib.compatibility_version = 0; // FIXME + if (_swap) + swapStruct(*dc); + memcpy(lc + sizeof(dylib_command), path.begin(), path.size()); + lc[sizeof(dylib_command) + path.size()] = '\0'; + lc += size; + } + // Add LC_LOAD_DYLIB commands for (const DependentDylib &dep : _file.dependentDylibs) { dylib_command* dc = reinterpret_cast(lc); @@ -720,7 +744,6 @@ lc[sizeof(dylib_command)+dep.path.size()] = '\0'; lc += size; } - } return ec; } Index: lib/ReaderWriter/MachO/MachONormalizedFileFromAtoms.cpp =================================================================== --- lib/ReaderWriter/MachO/MachONormalizedFileFromAtoms.cpp +++ lib/ReaderWriter/MachO/MachONormalizedFileFromAtoms.cpp @@ -934,6 +934,7 @@ f->arch = context.arch(); f->fileType = context.outputMachOType(); f->flags = util.fileFlags(); + f->installName = context.installName(); util.copySegmentInfo(normFile); util.copySections(normFile); util.addDependentDylibs(atomFile, normFile); Index: lib/ReaderWriter/MachO/MachONormalizedFileToAtoms.cpp =================================================================== --- lib/ReaderWriter/MachO/MachONormalizedFileToAtoms.cpp +++ lib/ReaderWriter/MachO/MachONormalizedFileToAtoms.cpp @@ -606,7 +606,8 @@ ErrorOr> normalizedDylibToAtoms(const NormalizedFile &normalizedFile, StringRef path, bool copyRefs) { - std::unique_ptr file(new MachODylibFile(path)); + std::unique_ptr file( + new MachODylibFile(normalizedFile.installName)); for (auto &sym : normalizedFile.globalSymbols) { assert((sym.scope & N_EXT) && "only expect external symbols here"); Index: lib/ReaderWriter/MachO/MachONormalizedFileYAML.cpp =================================================================== --- lib/ReaderWriter/MachO/MachONormalizedFileYAML.cpp +++ lib/ReaderWriter/MachO/MachONormalizedFileYAML.cpp @@ -508,7 +508,6 @@ } }; - template <> struct ScalarEnumerationTraits { static void enumeration(IO &io, RebaseType &value) { Index: test/mach-o/Inputs/use-dylib-install-names.yaml =================================================================== --- /dev/null +++ test/mach-o/Inputs/use-dylib-install-names.yaml @@ -0,0 +1,28 @@ +--- !mach-o +arch: x86_64 +file-type: MH_OBJECT +flags: [ ] +has-UUID: false +OS: unknown +sections: + - segment: __TEXT + section: __text + type: S_REGULAR + attributes: [ S_ATTR_PURE_INSTRUCTIONS ] + address: 0x0000000000000000 + content: [ 0x55, 0x48, 0x89, 0xE5, 0xE8, 0x00, 0x00, 0x00, + 0x00, 0xE8, 0x00, 0x00, 0x00, 0x00, 0xE8, 0x00, + 0x00, 0x00, 0x00, 0xE8, 0x00, 0x00, 0x00, 0x00, + 0xE8, 0x00, 0x00, 0x00, 0x00, 0x5D, 0xE9, 0x00, + 0x00, 0x00, 0x00 ] +global-symbols: + - name: _foo + type: N_SECT + scope: [ N_EXT ] + sect: 1 + value: 0x0000000000000000 +undefined-symbols: + - name: _myGlobal + type: N_UNDF + scope: [ N_EXT ] + value: 0x0000000000000000 Index: test/mach-o/dylib-install-names.yaml =================================================================== --- /dev/null +++ test/mach-o/dylib-install-names.yaml @@ -0,0 +1,41 @@ +# RUN: lld -flavor darwin -arch x86_64 -install_name libwibble.dylib -dylib %s -o %t.dylib +# RUN: macho-dump %t.dylib | FileCheck %s --check-prefix=CHECK-BINARY-WRITE +# RUN: lld -flavor darwin -arch x86_64 %p/Inputs/use-dylib-install-names.yaml %t.dylib -r -print_atoms | FileCheck %s --check-prefix=CHECK-BINARY-READ + +--- !mach-o +arch: x86_64 +file-type: MH_OBJECT +flags: [ MH_SUBSECTIONS_VIA_SYMBOLS ] +has-UUID: false +OS: unknown +sections: + - segment: __TEXT + section: __text + type: S_REGULAR + attributes: [ S_ATTR_PURE_INSTRUCTIONS, S_ATTR_SOME_INSTRUCTIONS ] + alignment: 4 + address: 0x0000000000000000 + content: [ 0xCC, 0xC3, 0x90, 0xC3, 0x90, 0x90, 0xC3, 0x90, + 0x90, 0x90, 0xC3, 0x90, 0x90, 0x90, 0x90, 0xC3, + 0x31, 0xC0, 0xC3 ] +local-symbols: + - name: _myStatic + type: N_SECT + sect: 1 + value: 0x000000000000000B +global-symbols: + - name: _myGlobal + type: N_SECT + scope: [ N_EXT ] + sect: 1 + value: 0x0000000000000001 +... + + +# CHECK-BINARY-WRITE: (('command', 13) +# CHECK-BINARY-WRITE-NEXT: ('size', 40) +# CHECK-BINARY-WRITE-NEXT: ('install_name', 'libwibble.dylib') + +# CHECK-BINARY-READ: shared-library-atoms: +# CHECK-BINARY-READ: - name: _myGlobal +# CHECK-BINARY-READ: load-name: libwibble.dylib Index: test/mach-o/lit.local.cfg =================================================================== --- test/mach-o/lit.local.cfg +++ test/mach-o/lit.local.cfg @@ -1,3 +1,4 @@ # mach-o test cases encode input files in yaml and use .yaml extension config.suffixes = ['.yaml'] +config.excludes = ['Inputs'] Index: test/mach-o/use-simple-dylib.yaml =================================================================== --- test/mach-o/use-simple-dylib.yaml +++ test/mach-o/use-simple-dylib.yaml @@ -1,5 +1,5 @@ # RUN: lld -flavor darwin -arch x86_64 -print_atoms -r %s | FileCheck %s -# lld -flavor darwin -arch x86_64 -print_atoms -r %s %p/Inputs/simple-dylib.yaml -o - | FileCheck %s + --- !mach-o arch: x86_64 @@ -110,6 +110,8 @@ sect: 1 desc: [ N_SYMBOL_RESOLVER ] value: 0x0000000000000010 + +install-name: libspecial.dylib ... @@ -118,12 +120,12 @@ # CHECK: - name: _myVariablePreviouslyKnownAsPrivateExtern # CHECK: shared-library-atoms: # CHECK: - name: _myHidden -# CHECK: load-name: {{.*}}use-simple-dylib.yaml +# CHECK: load-name: libspecial.dylib # CHECK: - name: _myGlobal -# CHECK: load-name: {{.*}}use-simple-dylib.yaml +# CHECK: load-name: libspecial.dylib # CHECK: - name: _myHiddenWeak -# CHECK: load-name: {{.*}}use-simple-dylib.yaml +# CHECK: load-name: libspecial.dylib # CHECK: - name: _myGlobalWeak -# CHECK: load-name: {{.*}}use-simple-dylib.yaml +# CHECK: load-name: libspecial.dylib # CHECK: - name: _myResolver -# CHECK: load-name: {{.*}}use-simple-dylib.yaml +# CHECK: load-name: libspecial.dylib