aboutsummaryrefslogtreecommitdiff
path: root/test/test-imp-forwarding.m
blob: e7bff27caa27c659e61f423fd879a21ea8a7a11b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#include "../lib/objc.c"
#include <objc/runtime.h>
#include <stdio.h>
#import <Foundation/Foundation.h>

static void imp1(id self, SEL sel, int a, int b) {
    NSLog(@"imp1: self=%@ sel=%s a=%d b=%d\n", self, sel_getName(sel), a, b);
}

struct big {
    int x[10];
};

static struct big imp2(id self, SEL sel, int a, int b) {
    NSLog(@"imp2: self=%@ sel=%s a=%d b=%d\n", self, sel_getName(sel), a, b);
    return (struct big) {{0}};
}

struct big (^test)(id, int) = ^(id self, int a) {
    NSLog(@"self=%@  a=%d", self, a);
    return (struct big) {{0}};
};

int main() {
    IMP testi = imp_implementationWithBlock(test);
    ((struct big (*)(id, SEL, int)) testi)(@"test", @selector(dumb), 5);
    IMP old = (IMP) imp1;
    SEL sel = @selector(some);
    struct temp_block_literal temp_block = get_temp_block(&old, sel);
    IMP new = imp_implementationWithBlock((id) &temp_block);
    ((void (*)(id, SEL, int, int)) new)(@"foo", sel, 1, 2);
    old = (IMP) imp2;
    struct big big = ((struct big (*)(id, SEL, int, int)) new)(@"bar", sel, 1, 2);
    printf("out? %d\n", big.x[0]);
}