webrtc_m130/third_party/ocmock/OCMock/NSObject+OCMAdditions.m
Artem Titov 739351d476 Roll chromium_revision 95336cb92b..191d55580e (557816:557824)
Change log: 95336cb92b..191d55580e
Full diff: 95336cb92b..191d55580e

Roll chromium third_party 4e16929f46..3a8f2a9e1e
Change log: 4e16929f46..3a8f2a9e1e

Changed dependencies:
* src/tools: c44a3f5eca..f524a53b81
DEPS diff: 95336cb92b..191d55580e/DEPS

No update to Clang.

TBR=titovartem@google.com,
BUG=None
CQ_INCLUDE_TRYBOTS=master.internal.tryserver.corp.webrtc:linux_internal

Change-Id: Ic9c4a62b050383646e9fcf5cc07a5653c14ac06e
Reviewed-on: https://webrtc-review.googlesource.com/76120
Reviewed-by: Patrik Höglund <phoglund@webrtc.org>
Reviewed-by: Karl Wiberg <kwiberg@webrtc.org>
Reviewed-by: Artem Titov <titovartem@webrtc.org>
Commit-Queue: Artem Titov <titovartem@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#23205}
2018-05-11 11:17:05 +00:00

76 lines
2.4 KiB
Objective-C

/*
* Copyright (c) 2009-2015 Erik Doernenburg and contributors
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may
* not use these files except in compliance with the License. You may obtain
* a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
#import "NSObject+OCMAdditions.h"
#import "NSMethodSignature+OCMAdditions.h"
#import <objc/runtime.h>
@implementation NSObject(OCMAdditions)
+ (IMP)instanceMethodForwarderForSelector:(SEL)aSelector
{
// use sel_registerName() and not @selector to avoid warning
SEL selectorWithNoImplementation = sel_registerName("methodWhichMustNotExist::::");
#ifndef __arm64__
static NSMutableDictionary *_OCMReturnTypeCache;
if(_OCMReturnTypeCache == nil)
_OCMReturnTypeCache = [[NSMutableDictionary alloc] init];
BOOL needsStructureReturn;
void *rawCacheKey[2] = { (void *)self, aSelector };
NSData *cacheKey = [NSData dataWithBytes:rawCacheKey length:sizeof(rawCacheKey)];
NSNumber *cachedValue = [_OCMReturnTypeCache objectForKey:cacheKey];
if(cachedValue == nil)
{
NSMethodSignature *sig = [self instanceMethodSignatureForSelector:aSelector];
needsStructureReturn = [sig usesSpecialStructureReturn];
[_OCMReturnTypeCache setObject:@(needsStructureReturn) forKey:cacheKey];
}
else
{
needsStructureReturn = [cachedValue boolValue];
}
if(needsStructureReturn)
return class_getMethodImplementation_stret([NSObject class], selectorWithNoImplementation);
#endif
return class_getMethodImplementation([NSObject class], selectorWithNoImplementation);
}
+ (void)enumerateMethodsInClass:(Class)aClass usingBlock:(void (^)(Class cls, SEL sel))aBlock
{
for(Class cls = aClass; cls != nil; cls = class_getSuperclass(cls))
{
Method *methodList = class_copyMethodList(cls, NULL);
if(methodList == NULL)
continue;
for(Method *mPtr = methodList; *mPtr != NULL; mPtr++)
{
SEL sel = method_getName(*mPtr);
aBlock(cls, sel);
}
free(methodList);
}
}
@end