aboutsummaryrefslogtreecommitdiff
path: root/test/test-imp-forwarding.m
blob: da2ece27451a126bf767332db0ffa13b1fa3a006 (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
#include "../lib/objc.c"
#include <objc/runtime.h>
#include <stdio.h>

static void imp1(id self, SEL sel, int a, int b) {
    printf("imp1: self=%p 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) {
    printf("imp2: self=%p sel=%s a=%d b=%d\n", self, sel_getName(sel), a, b);
    return (struct big) {{0}};
}

int main() {
    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 (*)(id, SEL, int, int)) new)(@"bar", sel, 1, 2);
}