FLEXRuntimeUtility.m 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728
  1. //
  2. // FLEXRuntimeUtility.m
  3. // Flipboard
  4. //
  5. // Created by Ryan Olson on 6/8/14.
  6. // Copyright (c) 2014 Flipboard. All rights reserved.
  7. //
  8. #import <UIKit/UIKit.h>
  9. #import "FLEXRuntimeUtility.h"
  10. // See https://developer.apple.com/library/ios/documentation/Cocoa/Conceptual/ObjCRuntimeGuide/Articles/ocrtPropertyIntrospection.html#//apple_ref/doc/uid/TP40008048-CH101-SW6
  11. NSString *const kFLEXUtilityAttributeTypeEncoding = @"T";
  12. NSString *const kFLEXUtilityAttributeBackingIvar = @"V";
  13. NSString *const kFLEXUtilityAttributeReadOnly = @"R";
  14. NSString *const kFLEXUtilityAttributeCopy = @"C";
  15. NSString *const kFLEXUtilityAttributeRetain = @"&";
  16. NSString *const kFLEXUtilityAttributeNonAtomic = @"N";
  17. NSString *const kFLEXUtilityAttributeCustomGetter = @"G";
  18. NSString *const kFLEXUtilityAttributeCustomSetter = @"S";
  19. NSString *const kFLEXUtilityAttributeDynamic = @"D";
  20. NSString *const kFLEXUtilityAttributeWeak = @"W";
  21. NSString *const kFLEXUtilityAttributeGarbageCollectable = @"P";
  22. NSString *const kFLEXUtilityAttributeOldStyleTypeEncoding = @"t";
  23. static NSString *const FLEXRuntimeUtilityErrorDomain = @"FLEXRuntimeUtilityErrorDomain";
  24. typedef NS_ENUM(NSInteger, FLEXRuntimeUtilityErrorCode) {
  25. FLEXRuntimeUtilityErrorCodeDoesNotRecognizeSelector = 0,
  26. FLEXRuntimeUtilityErrorCodeInvocationFailed = 1,
  27. FLEXRuntimeUtilityErrorCodeArgumentTypeMismatch = 2
  28. };
  29. // Arguments 0 and 1 are self and _cmd always
  30. const unsigned int kFLEXNumberOfImplicitArgs = 2;
  31. @implementation FLEXRuntimeUtility
  32. #pragma mark - Property Helpers (Public)
  33. + (NSString *)prettyNameForProperty:(objc_property_t)property
  34. {
  35. NSString *name = @(property_getName(property));
  36. NSString *encoding = [self typeEncodingForProperty:property];
  37. NSString *readableType = [self readableTypeForEncoding:encoding];
  38. return [self appendName:name toType:readableType];
  39. }
  40. + (NSString *)typeEncodingForProperty:(objc_property_t)property
  41. {
  42. NSDictionary<NSString *, NSString *> *attributesDictionary = [self attributesDictionaryForProperty:property];
  43. return attributesDictionary[kFLEXUtilityAttributeTypeEncoding];
  44. }
  45. + (BOOL)isReadonlyProperty:(objc_property_t)property
  46. {
  47. return [[self attributesDictionaryForProperty:property] objectForKey:kFLEXUtilityAttributeReadOnly] != nil;
  48. }
  49. + (SEL)setterSelectorForProperty:(objc_property_t)property
  50. {
  51. SEL setterSelector = NULL;
  52. NSString *setterSelectorString = [[self attributesDictionaryForProperty:property] objectForKey:kFLEXUtilityAttributeCustomSetter];
  53. if (!setterSelectorString) {
  54. NSString *propertyName = @(property_getName(property));
  55. setterSelectorString = [NSString stringWithFormat:@"set%@%@:", [[propertyName substringToIndex:1] uppercaseString], [propertyName substringFromIndex:1]];
  56. }
  57. if (setterSelectorString) {
  58. setterSelector = NSSelectorFromString(setterSelectorString);
  59. }
  60. return setterSelector;
  61. }
  62. + (NSString *)fullDescriptionForProperty:(objc_property_t)property
  63. {
  64. NSDictionary<NSString *, NSString *> *attributesDictionary = [self attributesDictionaryForProperty:property];
  65. NSMutableArray<NSString *> *attributesStrings = [NSMutableArray array];
  66. // Atomicity
  67. if (attributesDictionary[kFLEXUtilityAttributeNonAtomic]) {
  68. [attributesStrings addObject:@"nonatomic"];
  69. } else {
  70. [attributesStrings addObject:@"atomic"];
  71. }
  72. // Storage
  73. if (attributesDictionary[kFLEXUtilityAttributeRetain]) {
  74. [attributesStrings addObject:@"strong"];
  75. } else if (attributesDictionary[kFLEXUtilityAttributeCopy]) {
  76. [attributesStrings addObject:@"copy"];
  77. } else if (attributesDictionary[kFLEXUtilityAttributeWeak]) {
  78. [attributesStrings addObject:@"weak"];
  79. } else {
  80. [attributesStrings addObject:@"assign"];
  81. }
  82. // Mutability
  83. if (attributesDictionary[kFLEXUtilityAttributeReadOnly]) {
  84. [attributesStrings addObject:@"readonly"];
  85. } else {
  86. [attributesStrings addObject:@"readwrite"];
  87. }
  88. // Custom getter/setter
  89. NSString *customGetter = attributesDictionary[kFLEXUtilityAttributeCustomGetter];
  90. NSString *customSetter = attributesDictionary[kFLEXUtilityAttributeCustomSetter];
  91. if (customGetter) {
  92. [attributesStrings addObject:[NSString stringWithFormat:@"getter=%@", customGetter]];
  93. }
  94. if (customSetter) {
  95. [attributesStrings addObject:[NSString stringWithFormat:@"setter=%@", customSetter]];
  96. }
  97. NSString *attributesString = [attributesStrings componentsJoinedByString:@", "];
  98. NSString *shortName = [self prettyNameForProperty:property];
  99. return [NSString stringWithFormat:@"@property (%@) %@", attributesString, shortName];
  100. }
  101. + (id)valueForProperty:(objc_property_t)property onObject:(id)object
  102. {
  103. NSString *customGetterString = nil;
  104. char *customGetterName = property_copyAttributeValue(property, "G");
  105. if (customGetterName) {
  106. customGetterString = @(customGetterName);
  107. free(customGetterName);
  108. }
  109. SEL getterSelector;
  110. if ([customGetterString length] > 0) {
  111. getterSelector = NSSelectorFromString(customGetterString);
  112. } else {
  113. NSString *propertyName = @(property_getName(property));
  114. getterSelector = NSSelectorFromString(propertyName);
  115. }
  116. return [self performSelector:getterSelector onObject:object withArguments:nil error:NULL];
  117. }
  118. + (NSString *)descriptionForIvarOrPropertyValue:(id)value
  119. {
  120. NSString *description = nil;
  121. // Special case BOOL for better readability.
  122. if ([value isKindOfClass:[NSValue class]]) {
  123. const char *type = [value objCType];
  124. if (strcmp(type, @encode(BOOL)) == 0) {
  125. BOOL boolValue = NO;
  126. [value getValue:&boolValue];
  127. description = boolValue ? @"YES" : @"NO";
  128. } else if (strcmp(type, @encode(SEL)) == 0) {
  129. SEL selector = NULL;
  130. [value getValue:&selector];
  131. description = NSStringFromSelector(selector);
  132. }
  133. }
  134. if (!description) {
  135. // Single line display - replace newlines and tabs with spaces.
  136. description = [[value description] stringByReplacingOccurrencesOfString:@"\n" withString:@" "];
  137. description = [description stringByReplacingOccurrencesOfString:@"\t" withString:@" "];
  138. }
  139. if (!description) {
  140. description = @"nil";
  141. }
  142. return description;
  143. }
  144. + (void)tryAddPropertyWithName:(const char *)name attributes:(NSDictionary<NSString *, NSString *> *)attributePairs toClass:(__unsafe_unretained Class)theClass
  145. {
  146. objc_property_t property = class_getProperty(theClass, name);
  147. if (!property) {
  148. unsigned int totalAttributesCount = (unsigned int)[attributePairs count];
  149. objc_property_attribute_t *attributes = malloc(sizeof(objc_property_attribute_t) * totalAttributesCount);
  150. if (attributes) {
  151. unsigned int attributeIndex = 0;
  152. for (NSString *attributeName in [attributePairs allKeys]) {
  153. objc_property_attribute_t attribute;
  154. attribute.name = [attributeName UTF8String];
  155. attribute.value = [attributePairs[attributeName] UTF8String];
  156. attributes[attributeIndex++] = attribute;
  157. }
  158. class_addProperty(theClass, name, attributes, totalAttributesCount);
  159. free(attributes);
  160. }
  161. }
  162. }
  163. #pragma mark - Ivar Helpers (Public)
  164. + (NSString *)prettyNameForIvar:(Ivar)ivar
  165. {
  166. const char *nameCString = ivar_getName(ivar);
  167. NSString *name = nameCString ? @(nameCString) : nil;
  168. const char *encodingCString = ivar_getTypeEncoding(ivar);
  169. NSString *encoding = encodingCString ? @(encodingCString) : nil;
  170. NSString *readableType = [self readableTypeForEncoding:encoding];
  171. return [self appendName:name toType:readableType];
  172. }
  173. + (id)valueForIvar:(Ivar)ivar onObject:(id)object
  174. {
  175. id value = nil;
  176. const char *type = ivar_getTypeEncoding(ivar);
  177. #ifdef __arm64__
  178. // See http://www.sealiesoftware.com/blog/archive/2013/09/24/objc_explain_Non-pointer_isa.html
  179. const char *name = ivar_getName(ivar);
  180. if (type[0] == @encode(Class)[0] && strcmp(name, "isa") == 0) {
  181. value = object_getClass(object);
  182. } else
  183. #endif
  184. if (type[0] == @encode(id)[0] || type[0] == @encode(Class)[0]) {
  185. value = object_getIvar(object, ivar);
  186. } else {
  187. ptrdiff_t offset = ivar_getOffset(ivar);
  188. void *pointer = (__bridge void *)object + offset;
  189. value = [self valueForPrimitivePointer:pointer objCType:type];
  190. }
  191. return value;
  192. }
  193. + (void)setValue:(id)value forIvar:(Ivar)ivar onObject:(id)object
  194. {
  195. const char *typeEncodingCString = ivar_getTypeEncoding(ivar);
  196. if (typeEncodingCString[0] == '@') {
  197. object_setIvar(object, ivar, value);
  198. } else if ([value isKindOfClass:[NSValue class]]) {
  199. // Primitive - unbox the NSValue.
  200. NSValue *valueValue = (NSValue *)value;
  201. // Make sure that the box contained the correct type.
  202. NSAssert(strcmp([valueValue objCType], typeEncodingCString) == 0, @"Type encoding mismatch (value: %s; ivar: %s) in setting ivar named: %s on object: %@", [valueValue objCType], typeEncodingCString, ivar_getName(ivar), object);
  203. NSUInteger bufferSize = 0;
  204. @try {
  205. // NSGetSizeAndAlignment barfs on type encoding for bitfields.
  206. NSGetSizeAndAlignment(typeEncodingCString, &bufferSize, NULL);
  207. } @catch (NSException *exception) { }
  208. if (bufferSize > 0) {
  209. void *buffer = calloc(bufferSize, 1);
  210. [valueValue getValue:buffer];
  211. ptrdiff_t offset = ivar_getOffset(ivar);
  212. void *pointer = (__bridge void *)object + offset;
  213. memcpy(pointer, buffer, bufferSize);
  214. free(buffer);
  215. }
  216. }
  217. }
  218. #pragma mark - Method Helpers (Public)
  219. + (NSString *)prettyNameForMethod:(Method)method isClassMethod:(BOOL)isClassMethod
  220. {
  221. NSString *selectorName = NSStringFromSelector(method_getName(method));
  222. NSString *methodTypeString = isClassMethod ? @"+" : @"-";
  223. char *returnType = method_copyReturnType(method);
  224. NSString *readableReturnType = [self readableTypeForEncoding:@(returnType)];
  225. free(returnType);
  226. NSString *prettyName = [NSString stringWithFormat:@"%@ (%@)", methodTypeString, readableReturnType];
  227. NSArray<NSString *> *components = [self prettyArgumentComponentsForMethod:method];
  228. if ([components count] > 0) {
  229. prettyName = [prettyName stringByAppendingString:[components componentsJoinedByString:@" "]];
  230. } else {
  231. prettyName = [prettyName stringByAppendingString:selectorName];
  232. }
  233. return prettyName;
  234. }
  235. + (NSArray<NSString *> *)prettyArgumentComponentsForMethod:(Method)method
  236. {
  237. NSMutableArray<NSString *> *components = [NSMutableArray array];
  238. NSString *selectorName = NSStringFromSelector(method_getName(method));
  239. NSMutableArray<NSString *> *selectorComponents = [[selectorName componentsSeparatedByString:@":"] mutableCopy];
  240. // this is a workaround cause method_getNumberOfArguments() returns wrong number for some methods
  241. if (selectorComponents.count == 1) {
  242. return [selectorComponents copy];
  243. }
  244. if ([selectorComponents.lastObject isEqualToString:@""]) {
  245. [selectorComponents removeLastObject];
  246. }
  247. for (unsigned int argIndex = 0; argIndex < selectorComponents.count; argIndex++) {
  248. char *argType = method_copyArgumentType(method, argIndex + kFLEXNumberOfImplicitArgs);
  249. NSString *readableArgType = (argType != NULL) ? [self readableTypeForEncoding:@(argType)] : nil;
  250. free(argType);
  251. NSString *prettyComponent = [NSString stringWithFormat:@"%@:(%@) ", [selectorComponents objectAtIndex:argIndex], readableArgType];
  252. [components addObject:prettyComponent];
  253. }
  254. return components;
  255. }
  256. #pragma mark - Method Calling/Field Editing (Public)
  257. + (id)performSelector:(SEL)selector onObject:(id)object withArguments:(NSArray *)arguments error:(NSError * __autoreleasing *)error
  258. {
  259. // Bail if the object won't respond to this selector.
  260. if (![object respondsToSelector:selector]) {
  261. if (error) {
  262. NSDictionary<NSString *, id> *userInfo = @{ NSLocalizedDescriptionKey : [NSString stringWithFormat:@"%@ does not respond to the selector %@", object, NSStringFromSelector(selector)]};
  263. *error = [NSError errorWithDomain:FLEXRuntimeUtilityErrorDomain code:FLEXRuntimeUtilityErrorCodeDoesNotRecognizeSelector userInfo:userInfo];
  264. }
  265. return nil;
  266. }
  267. // Build the invocation
  268. NSMethodSignature *methodSignature = [object methodSignatureForSelector:selector];
  269. NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:methodSignature];
  270. [invocation setSelector:selector];
  271. [invocation setTarget:object];
  272. [invocation retainArguments];
  273. // Always self and _cmd
  274. NSUInteger numberOfArguments = [methodSignature numberOfArguments];
  275. for (NSUInteger argumentIndex = kFLEXNumberOfImplicitArgs; argumentIndex < numberOfArguments; argumentIndex++) {
  276. NSUInteger argumentsArrayIndex = argumentIndex - kFLEXNumberOfImplicitArgs;
  277. id argumentObject = [arguments count] > argumentsArrayIndex ? arguments[argumentsArrayIndex] : nil;
  278. // NSNull in the arguments array can be passed as a placeholder to indicate nil. We only need to set the argument if it will be non-nil.
  279. if (argumentObject && ![argumentObject isKindOfClass:[NSNull class]]) {
  280. const char *typeEncodingCString = [methodSignature getArgumentTypeAtIndex:argumentIndex];
  281. if (typeEncodingCString[0] == @encode(id)[0] || typeEncodingCString[0] == @encode(Class)[0] || [self isTollFreeBridgedValue:argumentObject forCFType:typeEncodingCString]) {
  282. // Object
  283. [invocation setArgument:&argumentObject atIndex:argumentIndex];
  284. } else if (strcmp(typeEncodingCString, @encode(CGColorRef)) == 0 && [argumentObject isKindOfClass:[UIColor class]]) {
  285. // Bridging UIColor to CGColorRef
  286. CGColorRef colorRef = [argumentObject CGColor];
  287. [invocation setArgument:&colorRef atIndex:argumentIndex];
  288. } else if ([argumentObject isKindOfClass:[NSValue class]]) {
  289. // Primitive boxed in NSValue
  290. NSValue *argumentValue = (NSValue *)argumentObject;
  291. // Ensure that the type encoding on the NSValue matches the type encoding of the argument in the method signature
  292. if (strcmp([argumentValue objCType], typeEncodingCString) != 0) {
  293. if (error) {
  294. NSDictionary<NSString *, id> *userInfo = @{ NSLocalizedDescriptionKey : [NSString stringWithFormat:@"Type encoding mismatch for agrument at index %lu. Value type: %s; Method argument type: %s.", (unsigned long)argumentsArrayIndex, [argumentValue objCType], typeEncodingCString]};
  295. *error = [NSError errorWithDomain:FLEXRuntimeUtilityErrorDomain code:FLEXRuntimeUtilityErrorCodeArgumentTypeMismatch userInfo:userInfo];
  296. }
  297. return nil;
  298. }
  299. NSUInteger bufferSize = 0;
  300. @try {
  301. // NSGetSizeAndAlignment barfs on type encoding for bitfields.
  302. NSGetSizeAndAlignment(typeEncodingCString, &bufferSize, NULL);
  303. } @catch (NSException *exception) { }
  304. if (bufferSize > 0) {
  305. void *buffer = calloc(bufferSize, 1);
  306. [argumentValue getValue:buffer];
  307. [invocation setArgument:buffer atIndex:argumentIndex];
  308. free(buffer);
  309. }
  310. }
  311. }
  312. }
  313. // Try to invoke the invocation but guard against an exception being thrown.
  314. BOOL successfullyInvoked = NO;
  315. @try {
  316. // Some methods are not fit to be called...
  317. // Looking at you -[UIResponder(UITextInputAdditions) _caretRect]
  318. [invocation invoke];
  319. successfullyInvoked = YES;
  320. } @catch (NSException *exception) {
  321. // Bummer...
  322. if (error) {
  323. // "… on <class>" / "… on instance of <class>"
  324. NSString *class = NSStringFromClass([object class]);
  325. NSString *calledOn = object == [object class] ? class : [@"an instance of " stringByAppendingString:class];
  326. NSString *message = [NSString stringWithFormat:@"Exception '%@' thrown while performing selector '%@' on %@.\nReason:\n\n%@",
  327. exception.name,
  328. NSStringFromSelector(selector),
  329. calledOn,
  330. exception.reason];
  331. *error = [NSError errorWithDomain:FLEXRuntimeUtilityErrorDomain
  332. code:FLEXRuntimeUtilityErrorCodeInvocationFailed
  333. userInfo:@{ NSLocalizedDescriptionKey : message }];
  334. }
  335. }
  336. // Retreive the return value and box if necessary.
  337. id returnObject = nil;
  338. if (successfullyInvoked) {
  339. const char *returnType = [methodSignature methodReturnType];
  340. if (returnType[0] == @encode(id)[0] || returnType[0] == @encode(Class)[0]) {
  341. __unsafe_unretained id objectReturnedFromMethod = nil;
  342. [invocation getReturnValue:&objectReturnedFromMethod];
  343. returnObject = objectReturnedFromMethod;
  344. } else if (returnType[0] != @encode(void)[0]) {
  345. void *returnValue = malloc([methodSignature methodReturnLength]);
  346. if (returnValue) {
  347. [invocation getReturnValue:returnValue];
  348. returnObject = [self valueForPrimitivePointer:returnValue objCType:returnType];
  349. free(returnValue);
  350. }
  351. }
  352. }
  353. return returnObject;
  354. }
  355. + (BOOL)isTollFreeBridgedValue:(id)value forCFType:(const char *)typeEncoding
  356. {
  357. // See https://developer.apple.com/library/ios/documentation/general/conceptual/CocoaEncyclopedia/Toll-FreeBridgin/Toll-FreeBridgin.html
  358. #define CASE(cftype, foundationClass) \
  359. if(strcmp(typeEncoding, @encode(cftype)) == 0) { \
  360. return [value isKindOfClass:[foundationClass class]]; \
  361. }
  362. CASE(CFArrayRef, NSArray);
  363. CASE(CFAttributedStringRef, NSAttributedString);
  364. CASE(CFCalendarRef, NSCalendar);
  365. CASE(CFCharacterSetRef, NSCharacterSet);
  366. CASE(CFDataRef, NSData);
  367. CASE(CFDateRef, NSDate);
  368. CASE(CFDictionaryRef, NSDictionary);
  369. CASE(CFErrorRef, NSError);
  370. CASE(CFLocaleRef, NSLocale);
  371. CASE(CFMutableArrayRef, NSMutableArray);
  372. CASE(CFMutableAttributedStringRef, NSMutableAttributedString);
  373. CASE(CFMutableCharacterSetRef, NSMutableCharacterSet);
  374. CASE(CFMutableDataRef, NSMutableData);
  375. CASE(CFMutableDictionaryRef, NSMutableDictionary);
  376. CASE(CFMutableSetRef, NSMutableSet);
  377. CASE(CFMutableStringRef, NSMutableString);
  378. CASE(CFNumberRef, NSNumber);
  379. CASE(CFReadStreamRef, NSInputStream);
  380. CASE(CFRunLoopTimerRef, NSTimer);
  381. CASE(CFSetRef, NSSet);
  382. CASE(CFStringRef, NSString);
  383. CASE(CFTimeZoneRef, NSTimeZone);
  384. CASE(CFURLRef, NSURL);
  385. CASE(CFWriteStreamRef, NSOutputStream);
  386. #undef CASE
  387. return NO;
  388. }
  389. + (NSString *)editableJSONStringForObject:(id)object
  390. {
  391. NSString *editableDescription = nil;
  392. if (object) {
  393. // This is a hack to use JSON serialization for our editable objects.
  394. // NSJSONSerialization doesn't allow writing fragments - the top level object must be an array or dictionary.
  395. // We always wrap the object inside an array and then strip the outer square braces off the final string.
  396. NSArray *wrappedObject = @[object];
  397. if ([NSJSONSerialization isValidJSONObject:wrappedObject]) {
  398. NSString *wrappedDescription = [[NSString alloc] initWithData:[NSJSONSerialization dataWithJSONObject:wrappedObject options:0 error:NULL] encoding:NSUTF8StringEncoding];
  399. editableDescription = [wrappedDescription substringWithRange:NSMakeRange(1, [wrappedDescription length] - 2)];
  400. }
  401. }
  402. return editableDescription;
  403. }
  404. + (id)objectValueFromEditableJSONString:(NSString *)string
  405. {
  406. id value = nil;
  407. // nil for empty string/whitespace
  408. if ([[string stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]] length] > 0) {
  409. value = [NSJSONSerialization JSONObjectWithData:[string dataUsingEncoding:NSUTF8StringEncoding] options:NSJSONReadingAllowFragments error:NULL];
  410. }
  411. return value;
  412. }
  413. + (NSValue *)valueForNumberWithObjCType:(const char *)typeEncoding fromInputString:(NSString *)inputString
  414. {
  415. NSNumberFormatter *formatter = [[NSNumberFormatter alloc] init];
  416. [formatter setNumberStyle:NSNumberFormatterDecimalStyle];
  417. NSNumber *number = [formatter numberFromString:inputString];
  418. // Make sure we box the number with the correct type encoding so it can be propperly unboxed later via getValue:
  419. NSValue *value = nil;
  420. if (strcmp(typeEncoding, @encode(char)) == 0) {
  421. char primitiveValue = [number charValue];
  422. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  423. } else if (strcmp(typeEncoding, @encode(int)) == 0) {
  424. int primitiveValue = [number intValue];
  425. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  426. } else if (strcmp(typeEncoding, @encode(short)) == 0) {
  427. short primitiveValue = [number shortValue];
  428. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  429. } else if (strcmp(typeEncoding, @encode(long)) == 0) {
  430. long primitiveValue = [number longValue];
  431. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  432. } else if (strcmp(typeEncoding, @encode(long long)) == 0) {
  433. long long primitiveValue = [number longLongValue];
  434. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  435. } else if (strcmp(typeEncoding, @encode(unsigned char)) == 0) {
  436. unsigned char primitiveValue = [number unsignedCharValue];
  437. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  438. } else if (strcmp(typeEncoding, @encode(unsigned int)) == 0) {
  439. unsigned int primitiveValue = [number unsignedIntValue];
  440. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  441. } else if (strcmp(typeEncoding, @encode(unsigned short)) == 0) {
  442. unsigned short primitiveValue = [number unsignedShortValue];
  443. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  444. } else if (strcmp(typeEncoding, @encode(unsigned long)) == 0) {
  445. unsigned long primitiveValue = [number unsignedLongValue];
  446. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  447. } else if (strcmp(typeEncoding, @encode(unsigned long long)) == 0) {
  448. unsigned long long primitiveValue = [number unsignedLongValue];
  449. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  450. } else if (strcmp(typeEncoding, @encode(float)) == 0) {
  451. float primitiveValue = [number floatValue];
  452. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  453. } else if (strcmp(typeEncoding, @encode(double)) == 0) {
  454. double primitiveValue = [number doubleValue];
  455. value = [NSValue value:&primitiveValue withObjCType:typeEncoding];
  456. }
  457. return value;
  458. }
  459. + (void)enumerateTypesInStructEncoding:(const char *)structEncoding usingBlock:(void (^)(NSString *structName, const char *fieldTypeEncoding, NSString *prettyTypeEncoding, NSUInteger fieldIndex, NSUInteger fieldOffset))typeBlock
  460. {
  461. if (structEncoding && structEncoding[0] == '{') {
  462. const char *equals = strchr(structEncoding, '=');
  463. if (equals) {
  464. const char *nameStart = structEncoding + 1;
  465. NSString *structName = [@(structEncoding) substringWithRange:NSMakeRange(nameStart - structEncoding, equals - nameStart)];
  466. NSUInteger fieldAlignment = 0;
  467. NSUInteger structSize = 0;
  468. @try {
  469. // NSGetSizeAndAlignment barfs on type encoding for bitfields.
  470. NSGetSizeAndAlignment(structEncoding, &structSize, &fieldAlignment);
  471. } @catch (NSException *exception) { }
  472. if (structSize > 0) {
  473. NSUInteger runningFieldIndex = 0;
  474. NSUInteger runningFieldOffset = 0;
  475. const char *typeStart = equals + 1;
  476. while (*typeStart != '}') {
  477. NSUInteger fieldSize = 0;
  478. // If the struct type encoding was successfully handled by NSGetSizeAndAlignment above, we *should* be ok with the field here.
  479. const char *nextTypeStart = NSGetSizeAndAlignment(typeStart, &fieldSize, NULL);
  480. NSString *typeEncoding = [@(structEncoding) substringWithRange:NSMakeRange(typeStart - structEncoding, nextTypeStart - typeStart)];
  481. typeBlock(structName, [typeEncoding UTF8String], [self readableTypeForEncoding:typeEncoding], runningFieldIndex, runningFieldOffset);
  482. runningFieldOffset += fieldSize;
  483. // Padding to keep propper alignment. __attribute((packed)) structs will break here.
  484. // The type encoding is no different for packed structs, so it's not clear there's anything we can do for those.
  485. if (runningFieldOffset % fieldAlignment != 0) {
  486. runningFieldOffset += fieldAlignment - runningFieldOffset % fieldAlignment;
  487. }
  488. runningFieldIndex++;
  489. typeStart = nextTypeStart;
  490. }
  491. }
  492. }
  493. }
  494. }
  495. #pragma mark - Internal Helpers
  496. + (NSDictionary<NSString *, NSString *> *)attributesDictionaryForProperty:(objc_property_t)property
  497. {
  498. NSString *attributes = @(property_getAttributes(property));
  499. // Thanks to MAObjcRuntime for inspiration here.
  500. NSArray<NSString *> *attributePairs = [attributes componentsSeparatedByString:@","];
  501. NSMutableDictionary<NSString *, NSString *> *attributesDictionary = [NSMutableDictionary dictionaryWithCapacity:[attributePairs count]];
  502. for (NSString *attributePair in attributePairs) {
  503. [attributesDictionary setObject:[attributePair substringFromIndex:1] forKey:[attributePair substringToIndex:1]];
  504. }
  505. return attributesDictionary;
  506. }
  507. + (NSString *)appendName:(NSString *)name toType:(NSString *)type
  508. {
  509. NSString *combined = nil;
  510. if ([type characterAtIndex:[type length] - 1] == '*') {
  511. combined = [type stringByAppendingString:name];
  512. } else {
  513. combined = [type stringByAppendingFormat:@" %@", name];
  514. }
  515. return combined;
  516. }
  517. + (NSString *)readableTypeForEncoding:(NSString *)encodingString
  518. {
  519. if (!encodingString) {
  520. return nil;
  521. }
  522. // See https://developer.apple.com/library/mac/#documentation/Cocoa/Conceptual/ObjCRuntimeGuide/Articles/ocrtTypeEncodings.html
  523. // class-dump has a much nicer and much more complete implementation for this task, but it is distributed under GPLv2 :/
  524. // See https://github.com/nygard/class-dump/blob/master/Source/CDType.m
  525. // Warning: this method uses multiple middle returns and macros to cut down on boilerplate.
  526. // The use of macros here was inspired by https://www.mikeash.com/pyblog/friday-qa-2013-02-08-lets-build-key-value-coding.html
  527. const char *encodingCString = [encodingString UTF8String];
  528. // Objects
  529. if (encodingCString[0] == '@') {
  530. NSString *class = [encodingString substringFromIndex:1];
  531. class = [class stringByReplacingOccurrencesOfString:@"\"" withString:@""];
  532. if ([class length] == 0 || [class isEqual:@"?"]) {
  533. class = @"id";
  534. } else {
  535. class = [class stringByAppendingString:@" *"];
  536. }
  537. return class;
  538. }
  539. // C Types
  540. #define TRANSLATE(ctype) \
  541. if (strcmp(encodingCString, @encode(ctype)) == 0) { \
  542. return (NSString *)CFSTR(#ctype); \
  543. }
  544. // Order matters here since some of the cocoa types are typedefed to c types.
  545. // We can't recover the exact mapping, but we choose to prefer the cocoa types.
  546. // This is not an exhaustive list, but it covers the most common types
  547. TRANSLATE(CGRect);
  548. TRANSLATE(CGPoint);
  549. TRANSLATE(CGSize);
  550. TRANSLATE(UIEdgeInsets);
  551. TRANSLATE(UIOffset);
  552. TRANSLATE(NSRange);
  553. TRANSLATE(CGAffineTransform);
  554. TRANSLATE(CATransform3D);
  555. TRANSLATE(CGColorRef);
  556. TRANSLATE(CGPathRef);
  557. TRANSLATE(CGContextRef);
  558. TRANSLATE(NSInteger);
  559. TRANSLATE(NSUInteger);
  560. TRANSLATE(CGFloat);
  561. TRANSLATE(BOOL);
  562. TRANSLATE(int);
  563. TRANSLATE(short);
  564. TRANSLATE(long);
  565. TRANSLATE(long long);
  566. TRANSLATE(unsigned char);
  567. TRANSLATE(unsigned int);
  568. TRANSLATE(unsigned short);
  569. TRANSLATE(unsigned long);
  570. TRANSLATE(unsigned long long);
  571. TRANSLATE(float);
  572. TRANSLATE(double);
  573. TRANSLATE(long double);
  574. TRANSLATE(char *);
  575. TRANSLATE(Class);
  576. TRANSLATE(objc_property_t);
  577. TRANSLATE(Ivar);
  578. TRANSLATE(Method);
  579. TRANSLATE(Category);
  580. TRANSLATE(NSZone *);
  581. TRANSLATE(SEL);
  582. TRANSLATE(void);
  583. #undef TRANSLATE
  584. // Qualifier Prefixes
  585. // Do this after the checks above since some of the direct translations (i.e. Method) contain a prefix.
  586. #define RECURSIVE_TRANSLATE(prefix, formatString) \
  587. if (encodingCString[0] == prefix) { \
  588. NSString *recursiveType = [self readableTypeForEncoding:[encodingString substringFromIndex:1]]; \
  589. return [NSString stringWithFormat:formatString, recursiveType]; \
  590. }
  591. // If there's a qualifier prefix on the encoding, translate it and then
  592. // recursively call this method with the rest of the encoding string.
  593. RECURSIVE_TRANSLATE('^', @"%@ *");
  594. RECURSIVE_TRANSLATE('r', @"const %@");
  595. RECURSIVE_TRANSLATE('n', @"in %@");
  596. RECURSIVE_TRANSLATE('N', @"inout %@");
  597. RECURSIVE_TRANSLATE('o', @"out %@");
  598. RECURSIVE_TRANSLATE('O', @"bycopy %@");
  599. RECURSIVE_TRANSLATE('R', @"byref %@");
  600. RECURSIVE_TRANSLATE('V', @"oneway %@");
  601. RECURSIVE_TRANSLATE('b', @"bitfield(%@)");
  602. #undef RECURSIVE_TRANSLATE
  603. // If we couldn't translate, just return the original encoding string
  604. return encodingString;
  605. }
  606. + (NSValue *)valueForPrimitivePointer:(void *)pointer objCType:(const char *)type
  607. {
  608. // CASE macro inspired by https://www.mikeash.com/pyblog/friday-qa-2013-02-08-lets-build-key-value-coding.html
  609. #define CASE(ctype, selectorpart) \
  610. if(strcmp(type, @encode(ctype)) == 0) { \
  611. return [NSNumber numberWith ## selectorpart: *(ctype *)pointer]; \
  612. }
  613. CASE(BOOL, Bool);
  614. CASE(unsigned char, UnsignedChar);
  615. CASE(short, Short);
  616. CASE(unsigned short, UnsignedShort);
  617. CASE(int, Int);
  618. CASE(unsigned int, UnsignedInt);
  619. CASE(long, Long);
  620. CASE(unsigned long, UnsignedLong);
  621. CASE(long long, LongLong);
  622. CASE(unsigned long long, UnsignedLongLong);
  623. CASE(float, Float);
  624. CASE(double, Double);
  625. #undef CASE
  626. NSValue *value = nil;
  627. @try {
  628. value = [NSValue valueWithBytes:pointer objCType:type];
  629. } @catch (NSException *exception) {
  630. // Certain type encodings are not supported by valueWithBytes:objCType:. Just fail silently if an exception is thrown.
  631. }
  632. return value;
  633. }
  634. @end