Да, это действительно возможно, но это решение специфично для ABI (работа на всех платформах не гарантируется) и широко использует доступную во время выполнения информацию о методах.
Сначала нам нужно получить информацию о методе, который мы оборачиваем блоком. Это делается через NSMethodSignature
, который содержит такую информацию, как:
- Количество аргументов
- Размер (в байтах) каждого аргумента
- Размер возвращаемого типа
Это позволяет нам обернуть (почти) любой метод без специального кода для этого метода, создавая таким образом повторно используемую функцию.
Во-вторых, нам нужен способ безопасной отправки вызовов методов во время выполнения. Мы делаем это с помощью NSInvocation
, что дает нам возможность создать динамический и безопасный вызов метода во время выполнения.
В-третьих, нам нужен блок, который может принимать любое количество передаваемых аргументов, и диспетчеризация этого. Это делается с помощью va_list
API C и должно работать для 99% методов.
Наконец, нам нужно получить возвращаемое значение и иметь возможность вернуть его из нашего блока. Это часть всей операции, которая может не работать из-за странностей с возвращаемыми структурами и тому подобным во время выполнения Objective-C.
Однако, пока вы придерживаетесь примитивных типов и объектов Objective-C, этот код должен отлично работать для вас.
Несколько замечаний по поводу этой реализации:
Это зависит от неопределенного поведения с приведением типов блоков и функций, однако из-за соглашений о вызовах iOS и Mac это не должно вызывать никаких проблем (если только ваш метод не имеет тип возвращаемого значения, отличный от ожидаемого блоком).
Это также зависит от неопределенного поведения с результатом вызова va_arg
с типом, который может не совпадать с переданным, однако, поскольку типы имеют одинаковый размер, это никогда не должно быть проблемой.
Без лишних слов, вот пример кода, за которым следует реализация:
@interface MyObj : NSObject
-(void) doSomething;
@end
@implementation MyObj
-(void) doSomething
{
NSLog(@"This is me, doing something! %p", self);
}
-(id) doSomethingWithArgs:(long) arg :(short) arg2{
return [NSString stringWithFormat:@"%ld %d", arg, arg2];
}
@end
int main() {
// try out our selector wrapping
MyObj *obj = [MyObj new];
id (^asBlock)(long, short) = createBlock(obj, @selector(doSomethingWithArgs::));
NSLog(@"%@", asBlock(123456789, 456));
}
/* WARNING, ABI SPECIFIC, BLAH BLAH BLAH NOT PORTABLE! */
static inline void getArgFromListOfSize(va_list *args, void *first, size_t size, size_t align, void *dst, BOOL isFirst) {
// create a map of sizes to types
switch (size) {
// varargs are weird, and are aligned to 32 bit boundaries. We still only copy the size needed, though.
// these cases should cover all 32 bit pointers (iOS), boolean values, and floats too.
case sizeof(uint8_t): {
uint8_t tmp = isFirst ? (uint32_t) first : va_arg(*args, uint32_t);
memcpy(dst, &tmp, size);
break;
}
case sizeof(uint16_t): {
uint16_t tmp = isFirst ? (uint32_t) first : va_arg(*args, uint32_t);
memcpy(dst, &tmp, size);
break;
}
case sizeof(uint32_t): {
uint32_t tmp = isFirst ? (uint32_t) first : va_arg(*args, uint32_t);
memcpy(dst, &tmp, size);
break;
}
// this should cover 64 bit pointers (Mac), and longs, and doubles
case sizeof(uint64_t): {
uint64_t tmp = isFirst ? (uint64_t) first : va_arg(*args, uint64_t);
memcpy(dst, &tmp, size);
break;
}
/* This has to be commented out to work on iOS (as CGSizes are 64 bits)
// common 'other' types (covers CGSize, CGPoint)
case sizeof(CGPoint): {
CGPoint tmp = isFirst ? *(CGPoint *) &first : va_arg(*args, CGPoint);
memcpy(dst, &tmp, size);
break;
}
*/
// CGRects are fairly common on iOS, so we'll include those as well
case sizeof(CGRect): {
CGRect tmp = isFirst ? *(CGRect *) &first : va_arg(*args, CGRect);
memcpy(dst, &tmp, size);
break;
}
default: {
fprintf(stderr, "WARNING! Could not bind parameter of size %zu, unkown type! Going to have problems down the road!", size);
break;
}
}
}
id createBlock(id self, SEL _cmd) {
NSMethodSignature *methodSig = [self methodSignatureForSelector:_cmd];
if (methodSig == nil)
return nil;
return ^(void *arg, ...) {
NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:methodSig];
[invocation setTarget:self];
[invocation setSelector:_cmd];
NSUInteger argc = [methodSig numberOfArguments];
va_list args;
va_start(args, arg);
for (int argi = 2; argi < argc; argi++) {
const char *type = [methodSig getArgumentTypeAtIndex:argi];
NSUInteger size;
NSUInteger align;
// get the size
NSGetSizeAndAlignment(type, &size, &align);
// find the right type
void *argument = alloca(size);
getArgFromListOfSize(&args, arg, size, align, argument, argi == 2);
[invocation setArgument:argument atIndex:argi];
}
va_end(args);
[invocation invoke];
// get the return value
if (methodSig.methodReturnLength != 0) {
void *retVal = alloca(methodSig.methodReturnLength);
[invocation getReturnValue:retVal];
return *((void **) retVal);
}
return nil;
};
}
Дайте мне знать, если у вас возникнут проблемы с этой реализацией!
person
Richard J. Ross III
schedule
27.04.2013