X-Git-Url: https://git.saurik.com/apple/ld64.git/blobdiff_plain/eaf282aaf65b222563e6b3db98e12d720fb161bf..7f09b9353af9897bf18933788d6a59c152c29edd:/src/ld/passes/huge.cpp

diff --git a/src/ld/passes/huge.cpp b/src/ld/passes/huge.cpp
index 693976c..3797a23 100644
--- a/src/ld/passes/huge.cpp
+++ b/src/ld/passes/huge.cpp
@@ -47,10 +47,53 @@ public:
 	}
 };
 
+class DataPadAtom : public ld::Atom {
+public:
+											DataPadAtom(ld::Internal& state)
+											: ld::Atom(_s_section, ld::Atom::definitionRegular, ld::Atom::combineNever,
+														ld::Atom::scopeTranslationUnit, ld::Atom::typeUnclassified,
+														symbolTableNotIn, false, false, false, ld::Atom::Alignment(3))
+											{ state.addAtom(*this); }
+
+
+	virtual const ld::File*					file() const					{ return NULL; }
+	virtual const char*						name() const					{ return "padding"; }
+	virtual uint64_t						size() const					{ return 8; }
+	virtual uint64_t						objectAddress() const			{ return 0; }
+	virtual void							copyRawContent(uint8_t buffer[]) const { }
+
+protected:
+	virtual									~DataPadAtom() {}
+
+	static ld::Section						_s_section;
+};
+
+ld::Section DataPadAtom::_s_section("__DATA", "__data", ld::Section::typeUnclassified);
+
+
 void doPass(const Options& opts, ld::Internal& state)
 {
 	const bool log = false;
-	
+
+	// <rdar://problem/26015603> add __data section if __DATA segment was gutted by dirty data removal
+	if ( (opts.outputKind() == Options::kDynamicLibrary) && opts.useDataConstSegment() && opts.hasDataSymbolMoves() ) {
+		uint64_t dataAtomsSize = 0;
+		bool foundSegmentDATA_DIRTY = false;
+		for (ld::Internal::FinalSection* sect : state.sections) {
+			if ( strcmp(sect->segmentName(), "__DATA") == 0 ) {
+				for (const ld::Atom* atom : sect->atoms) {
+					dataAtomsSize += atom->size();
+				}
+			}
+			else if ( strcmp(sect->segmentName(), "__DATA_DIRTY") == 0 ) {
+				foundSegmentDATA_DIRTY = true;
+			}
+		}
+		if ( foundSegmentDATA_DIRTY && (dataAtomsSize == 0) ) {
+			new DataPadAtom(state);
+		}
+	}
+
 	// only make make __huge section in final linked images
 	if ( opts.outputKind() == Options::kObjectFile )
 		return;