c87a581747
Summary: This fixes the issue mentioned in https://github.com/facebook/flipper/issues/1940 ## Changelog Pull Request resolved: https://github.com/facebook/flipper/pull/1943 Test Plan: User confirmed Reviewed By: mweststrate Differential Revision: D26577560 Pulled By: priteshrnandgaonkar fbshipit-source-id: dec4e34dcec4a3b3c20c11f0f2dc748fe2698693
130 lines
3.8 KiB
Plaintext
Executable File
130 lines
3.8 KiB
Plaintext
Executable File
/*
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#import "FLEXUtility.h"
|
|
|
|
#include <assert.h>
|
|
#include <mach/mach.h>
|
|
#include <mach/mach_time.h>
|
|
#import <objc/runtime.h>
|
|
#import <zlib.h>
|
|
|
|
#import <ImageIO/ImageIO.h>
|
|
|
|
@implementation FLEXUtility
|
|
|
|
+ (SEL)swizzledSelectorForSelector:(SEL)selector {
|
|
return NSSelectorFromString(
|
|
[NSString stringWithFormat:@"_flex_swizzle_%x_%@",
|
|
arc4random(),
|
|
NSStringFromSelector(selector)]);
|
|
}
|
|
|
|
+ (BOOL)instanceRespondsButDoesNotImplementSelector:(SEL)selector
|
|
class:(Class)cls {
|
|
if ([cls instancesRespondToSelector:selector]) {
|
|
unsigned int numMethods = 0;
|
|
Method* methods = class_copyMethodList(cls, &numMethods);
|
|
|
|
BOOL implementsSelector = NO;
|
|
for (int index = 0; index < numMethods; index++) {
|
|
SEL methodSelector = method_getName(methods[index]);
|
|
if (selector == methodSelector) {
|
|
implementsSelector = YES;
|
|
break;
|
|
}
|
|
}
|
|
|
|
free(methods);
|
|
|
|
if (!implementsSelector) {
|
|
return YES;
|
|
}
|
|
}
|
|
|
|
return NO;
|
|
}
|
|
|
|
+ (void)replaceImplementationOfKnownSelector:(SEL)originalSelector
|
|
onClass:(Class)className
|
|
withBlock:(id)block
|
|
swizzledSelector:(SEL)swizzledSelector {
|
|
// This method is only intended for swizzling methods that are know to exist
|
|
// on the class. Bail if that isn't the case.
|
|
Method originalMethod = class_getInstanceMethod(className, originalSelector);
|
|
if (!originalMethod) {
|
|
return;
|
|
}
|
|
|
|
IMP implementation = imp_implementationWithBlock(block);
|
|
class_addMethod(
|
|
className,
|
|
swizzledSelector,
|
|
implementation,
|
|
method_getTypeEncoding(originalMethod));
|
|
Method newMethod = class_getInstanceMethod(className, swizzledSelector);
|
|
method_exchangeImplementations(originalMethod, newMethod);
|
|
}
|
|
|
|
+ (void)replaceImplementationOfSelector:(SEL)selector
|
|
withSelector:(SEL)swizzledSelector
|
|
forClass:(Class)cls
|
|
withMethodDescription:
|
|
(struct objc_method_description)methodDescription
|
|
implementationBlock:(id)implementationBlock
|
|
undefinedBlock:(id)undefinedBlock {
|
|
if ([self instanceRespondsButDoesNotImplementSelector:selector class:cls]) {
|
|
return;
|
|
}
|
|
|
|
IMP implementation = imp_implementationWithBlock((id)(
|
|
[cls instancesRespondToSelector:selector] ? implementationBlock
|
|
: undefinedBlock));
|
|
|
|
Method oldMethod = class_getInstanceMethod(cls, selector);
|
|
if (oldMethod) {
|
|
objc_method_description* description = method_getDescription(oldMethod);
|
|
class_addMethod(cls, swizzledSelector, implementation, description->types);
|
|
Method newMethod = class_getInstanceMethod(cls, swizzledSelector);
|
|
method_exchangeImplementations(oldMethod, newMethod);
|
|
} else {
|
|
class_addMethod(cls, selector, implementation, methodDescription.types);
|
|
}
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation NSNumber (SonarUtility)
|
|
|
|
+ (NSNumber*)random {
|
|
int64_t identifier;
|
|
arc4random_buf(&identifier, sizeof(int64_t));
|
|
return @(identifier);
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation NSDate (SonarUtility)
|
|
|
|
+ (uint64_t)getTimeNanoseconds {
|
|
static struct mach_timebase_info tb_info = {0};
|
|
static dispatch_once_t onceToken;
|
|
dispatch_once(&onceToken, ^{
|
|
__unused int ret = mach_timebase_info(&tb_info);
|
|
assert(0 == ret);
|
|
});
|
|
|
|
return (mach_absolute_time() * tb_info.numer) / tb_info.denom;
|
|
}
|
|
|
|
+ (uint64_t)timestamp {
|
|
const uint64_t nowNanoSeconds = [self getTimeNanoseconds];
|
|
return nowNanoSeconds / 1000000;
|
|
}
|
|
|
|
@end
|