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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
use objc2::rc::Retained;
use objc2::runtime::{NSObject, NSObjectProtocol};
use objc2::{define_class, extern_methods, extern_protocol};
extern_protocol!(
#[allow(clippy::missing_safety_doc)]
#[name = "MainThreadMarkerTestProtocol"]
unsafe trait Proto: NSObjectProtocol {
#[unsafe(method(myMethod:))]
fn protocol_method(mtm: MainThreadMarker, arg: i32) -> i32;
#[unsafe(method(myMethodRetained:))]
fn protocol_method_retained(mtm: MainThreadMarker, arg: &Self) -> Retained<Self>;
}
);
define_class!(
#[unsafe(super(NSObject))]
#[name = "MainThreadMarkerTest"]
#[derive(PartialEq, Eq, Hash, Debug)]
struct Cls;
unsafe impl NSObjectProtocol for Cls {}
unsafe impl Proto for Cls {
#[unsafe(method(myMethod:))]
fn _my_mainthreadonly_method(arg: i32) -> i32 {
arg + 1
}
#[unsafe(method_id(myMethodRetained:))]
fn _my_mainthreadonly_method_retained(arg: &Self) -> Retained<Self> {
unsafe { Retained::retain(arg as *const Self as *mut Self).unwrap() }
}
}
);
// The macro does a textual match; but when users actually use
// `objc2_foundation::MainThreadMarker` to ensure soundness, they will not
// do this!
#[derive(Clone, Copy)]
struct MainThreadMarker {
_some_field: u32,
}
impl Cls {
extern_methods!(
#[unsafe(method(new))]
fn new(mtm: MainThreadMarker) -> Retained<Self>;
#[unsafe(method(myMethod:))]
fn method(mtm: MainThreadMarker, arg: i32, mtm2: MainThreadMarker) -> i32;
#[unsafe(method(myMethodRetained:))]
fn method_retained(
mtm: MainThreadMarker,
arg: &Self,
mtm2: MainThreadMarker,
) -> Retained<Self>;
);
}
#[test]
fn call() {
let mtm = MainThreadMarker { _some_field: 0 };
let obj1 = Cls::new(mtm);
let res = Cls::method(mtm, 2, mtm);
assert_eq!(res, 3);
let res = Cls::protocol_method(mtm, 3);
assert_eq!(res, 4);
let obj2 = Cls::method_retained(mtm, &obj1, mtm);
assert_eq!(obj1, obj2);
let obj2 = Cls::protocol_method_retained(mtm, &obj1);
assert_eq!(obj1, obj2);
}
|
