X-Git-Url: https://git.saurik.com/apple/ld64.git/blobdiff_plain/b1f7435d66a93f03b77932b3a9ad8a83ce5e1ebc..e667b16e39593d19d0e25a8440a563eac3572653:/src/ld/passes/objc.cpp

diff --git a/src/ld/passes/objc.cpp b/src/ld/passes/objc.cpp
index d471d98..2bbf54a 100644
--- a/src/ld/passes/objc.cpp
+++ b/src/ld/passes/objc.cpp
@@ -54,6 +54,7 @@ struct objc_image_info  {
 #define OBJC_IMAGE_REQUIRES_GC			(1<<2)
 #define OBJC_IMAGE_OPTIMIZED_BY_DYLD	(1<<3)
 #define OBJC_IMAGE_SUPPORTS_COMPACTION	(1<<4)
+#define OBJC_IMAGE_IS_SIMULATED			(1<<5)
 
 
 
@@ -64,7 +65,7 @@ template <typename A>
 class ObjCImageInfoAtom : public ld::Atom {
 public:
 											ObjCImageInfoAtom(ld::File::ObjcConstraint objcConstraint, 
-															bool compaction, bool abi2);
+															bool compaction, bool abi2, uint8_t swiftVersion);
 
 	virtual const ld::File*					file() const					{ return NULL; }
 	virtual const char*						name() const					{ return "objc image info"; }
@@ -88,7 +89,7 @@ template <typename A> ld::Section ObjCImageInfoAtom<A>::_s_sectionABI2("__DATA",
 
 template <typename A>
 ObjCImageInfoAtom<A>::ObjCImageInfoAtom(ld::File::ObjcConstraint objcConstraint, bool compaction, 
-										bool abi2)
+										bool abi2, uint8_t swiftVersion)
 	: ld::Atom(abi2 ? _s_sectionABI2 : _s_sectionABI1, ld::Atom::definitionRegular, ld::Atom::combineNever,
 							ld::Atom::scopeLinkageUnit, ld::Atom::typeUnclassified, 
 							symbolTableNotIn, false, false, false, ld::Atom::Alignment(2))
@@ -111,8 +112,14 @@ ObjCImageInfoAtom<A>::ObjCImageInfoAtom(ld::File::ObjcConstraint objcConstraint,
 			if ( compaction ) 
 				value |= OBJC_IMAGE_SUPPORTS_COMPACTION;
 			break;
+		case ld::File::objcConstraintRetainReleaseForSimulator:
+				value |= OBJC_IMAGE_IS_SIMULATED;
+			break;
 	}
 
+	// provide swift language version in final binary for runtime to inspect
+	value |= (swiftVersion << 8);
+
 	_content.version = 0;
 	A::P::E::set32(_content.flags, value);
 }
@@ -358,7 +365,7 @@ void ObjCData<A>::setPointerInContent(ld::Internal& state, const ld::Atom* conte
 template <typename A>
 class Category : public ObjCData<A> {
 public:
-	static const ld::Atom*	getClass(ld::Internal& state, const ld::Atom* contentAtom);
+	static const ld::Atom*	getClass(ld::Internal& state, const ld::Atom* contentAtom, bool& hasAddend);
 	static const ld::Atom*	getInstanceMethods(ld::Internal& state, const ld::Atom* contentAtom);
 	static const ld::Atom*	getClassMethods(ld::Internal& state, const ld::Atom* contentAtom);
 	static const ld::Atom*	getProtocols(ld::Internal& state, const ld::Atom* contentAtom);
@@ -370,9 +377,9 @@ private:
 
 
 template <typename A>
-const ld::Atom*	Category<A>::getClass(ld::Internal& state, const ld::Atom* contentAtom)
+const ld::Atom*	Category<A>::getClass(ld::Internal& state, const ld::Atom* contentAtom, bool& hasAddend)
 {
-	return ObjCData<A>::getPointerInContent(state, contentAtom, sizeof(pint_t)); // category_t.cls
+	return ObjCData<A>::getPointerInContent(state, contentAtom, sizeof(pint_t), &hasAddend); // category_t.cls
 }
 
 template <typename A>
@@ -834,10 +841,17 @@ void OptimizeCategories<A>::doit(const Options& opts, ld::Internal& state)
 				// ignore categories also in __objc_nlcatlist
 				if ( nlcatListAtoms.count(categoryAtom) != 0 )
 					continue;
-				const ld::Atom* categoryOnClassAtom = Category<A>::getClass(state, categoryAtom); 
+				const ld::Atom* categoryOnClassAtom = Category<A>::getClass(state, categoryAtom, hasAddend); 
 				assert(categoryOnClassAtom != NULL);
+				// only look at classes defined in this image
 				if ( categoryOnClassAtom->definition() != ld::Atom::definitionProxy ) {
-					// only look at classes defined in this image
+					// <rdar://problem/16107696> for now, back off optimization on new style classes
+					if ( hasAddend != 0 )
+						continue;
+					// <rdar://problem/17249777> don't apply categories to swift classes
+					if ( categoryOnClassAtom->hasFixupsOfKind(ld::Fixup::kindNoneGroupSubordinate) )
+						continue;
+
 					CatMap::iterator pos = classToCategories.find(categoryOnClassAtom);
 					if ( pos == classToCategories.end() ) {
 						classToCategories[categoryOnClassAtom] = new std::vector<const ld::Atom*>();
@@ -1167,19 +1181,27 @@ void doPass(const Options& opts, ld::Internal& state)
 #if SUPPORT_ARCH_x86_64
 			case CPU_TYPE_X86_64:
 				state.addAtom(*new ObjCImageInfoAtom<x86_64>(state.objcObjectConstraint, compaction, 
-								true));
+							  true, state.swiftVersion));
 				break;
 #endif
 #if SUPPORT_ARCH_i386
 			case CPU_TYPE_I386:
 				state.addAtom(*new ObjCImageInfoAtom<x86>(state.objcObjectConstraint, compaction, 
-							opts.objCABIVersion2POverride() ? true : false));
+							  opts.objCABIVersion2POverride() ? true : false, state.swiftVersion));
 				break;
 #endif
+#if SUPPORT_ARCH_arm_any
 			case CPU_TYPE_ARM:
 				state.addAtom(*new ObjCImageInfoAtom<arm>(state.objcObjectConstraint, compaction, 
-							true));
+							  true, state.swiftVersion));
 				break;
+#endif
+#if SUPPORT_ARCH_arm64
+			case CPU_TYPE_ARM64:
+				state.addAtom(*new ObjCImageInfoAtom<arm64>(state.objcObjectConstraint, compaction, 
+							  true, state.swiftVersion));
+				break;
+#endif
 			default:
 				assert(0 && "unknown objc arch");
 		}	
@@ -1195,16 +1217,19 @@ void doPass(const Options& opts, ld::Internal& state)
 #endif
 #if SUPPORT_ARCH_i386
 			case CPU_TYPE_I386:
-				// disable optimization until fully tested
 				if ( opts.objCABIVersion2POverride() )
                     OptimizeCategories<x86>::doit(opts, state);
 				break;
 #endif
 #if SUPPORT_ARCH_arm_any
 			case CPU_TYPE_ARM:
-				// disable optimization until fully tested
 				OptimizeCategories<arm>::doit(opts, state);
 				break;
+#endif
+#if SUPPORT_ARCH_arm64
+			case CPU_TYPE_ARM64:
+				// disabled until tested
+				break;
 #endif
 			default:
 				assert(0 && "unknown objc arch");