Skip to content

Commit 28356ca

Browse files
madsmtmmadsmtm
committed
Add rc::Allocated
1 parent 8529a36 commit 28356ca

10 files changed

+154
-69
lines changed

‎objc2/src/__macro_helpers.rs

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
use crate::rc::{Id, Ownership};
1+
use crate::rc::{Allocated, Id, Ownership};
22
use crate::runtime::{Class, Sel};
33
use crate::{Message, MessageArguments, MessageError, MessageReceiver};
44

@@ -63,31 +63,31 @@ impl<T: ?Sized + Message, O: Ownership> MsgSendId<&'_ Class, Id<T, O>>
6363
}
6464
}
6565

66-
// `alloc`, should mark the return value as "allocated, not initialized" somehow
67-
impl<T: ?Sized + Message, O: Ownership> MsgSendId<&'_ Class, Id<T, O>>
66+
// `alloc`
67+
impl<T: ?Sized + Message, O: Ownership> MsgSendId<&'_ Class, Id<Allocated<T>, O>>
6868
for RetainSemantics<false, true, false, false>
6969
{
7070
#[inline(always)]
7171
unsafe fn send_message_id<A: MessageArguments>(
7272
cls: &Class,
7373
sel: Sel,
7474
args: A,
75-
) -> Result<Option<Id<T, O>>, MessageError> {
76-
unsafe { MessageReceiver::send_message(cls, sel, args).map(|r| Id::new(r)) }
75+
) -> Result<Option<Id<Allocated<T>, O>>, MessageError> {
76+
unsafe { MessageReceiver::send_message(cls, sel, args).map(|r| Id::new_allocated(r)) }
7777
}
7878
}
7979

80-
// `init`, should mark the input value as "allocated, not initialized" somehow
81-
impl<T: ?Sized + Message, O: Ownership> MsgSendId<Option<Id<T, O>>, Id<T, O>>
80+
// `init`
81+
impl<T: ?Sized + Message, O: Ownership> MsgSendId<Option<Id<Allocated<T>, O>>, Id<T, O>>
8282
for RetainSemantics<false, false, true, false>
8383
{
8484
#[inline(always)]
8585
unsafe fn send_message_id<A: MessageArguments>(
86-
obj: Option<Id<T, O>>,
86+
obj: Option<Id<Allocated<T>, O>>,
8787
sel: Sel,
8888
args: A,
8989
) -> Result<Option<Id<T, O>>, MessageError> {
90-
let ptr = Id::option_into_ptr(obj);
90+
let ptr = Id::option_into_ptr(obj.map(|obj| unsafe { Id::assume_init(obj) }));
9191
// SAFETY: `ptr` may be null here, but that's fine since the return
9292
// is `*mut T`, which is one of the few types where messages to nil is
9393
// allowed.
@@ -177,7 +177,7 @@ mod tests {
177177

178178
use core::ptr;
179179

180-
use crate::rc::{Owned, RcTestObject, Shared, ThreadTestData};
180+
use crate::rc::{Allocated, Owned, RcTestObject, Shared, ThreadTestData};
181181
use crate::runtime::Object;
182182
use crate::{Encoding, RefEncode};
183183

@@ -186,7 +186,7 @@ mod tests {
186186
let mut expected = ThreadTestData::current();
187187
let cls = RcTestObject::class();
188188

189-
let obj: Id<RcTestObject, Shared> = unsafe { msg_send_id![cls, alloc].unwrap() };
189+
let obj: Id<Allocated<RcTestObject>, Shared> = unsafe { msg_send_id![cls, alloc].unwrap() };
190190
expected.alloc += 1;
191191
expected.assert_current();
192192

@@ -216,7 +216,7 @@ mod tests {
216216
let cls = RcTestObject::class();
217217

218218
let zone: *const _NSZone = ptr::null();
219-
let _obj: Id<RcTestObject, Owned> =
219+
let _obj: Id<Allocated<RcTestObject>, Owned> =
220220
unsafe { msg_send_id![cls, allocWithZone: zone].unwrap() };
221221
// `+[NSObject alloc]` delegates to `+[NSObject allocWithZone:]`, but
222222
// `RcTestObject` only catches `alloc`.
@@ -229,14 +229,14 @@ mod tests {
229229
let mut expected = ThreadTestData::current();
230230
let cls = RcTestObject::class();
231231

232-
let obj: Option<Id<RcTestObject, Shared>> = unsafe { msg_send_id![cls, alloc] };
232+
let obj: Option<Id<Allocated<RcTestObject>, Shared>> = unsafe { msg_send_id![cls, alloc] };
233233
expected.alloc += 1;
234234
// Don't check allocation error
235235
let _obj: Id<RcTestObject, Shared> = unsafe { msg_send_id![obj, init].unwrap() };
236236
expected.init += 1;
237237
expected.assert_current();
238238

239-
let obj: Option<Id<RcTestObject, Shared>> = unsafe { msg_send_id![cls, alloc] };
239+
let obj: Option<Id<Allocated<RcTestObject>, Shared>> = unsafe { msg_send_id![cls, alloc] };
240240
expected.alloc += 1;
241241
// Check allocation error before init
242242
let obj = obj.unwrap();

‎objc2/src/macros.rs

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -323,11 +323,12 @@ macro_rules! msg_send_bool {
323323
/// is a generic `Option<Id<T, O>>`.
324324
///
325325
/// - The `alloc` family: The receiver must be `&Class`, and the return type
326-
/// is a generic `Option<Id<T, O>>`. (This will change, see [#172]).
326+
/// is a generic `Option<Id<Allocated<T>, O>>`.
327327
///
328-
/// - The `init` family: The receiver must be `Option<Id<T, O>>` as returned
329-
/// from `alloc`. The receiver is consumed, and a the now-initialized
330-
/// `Option<Id<T, O>>` (with the same `T` and `O`) is returned.
328+
/// - The `init` family: The receiver must be `Option<Id<Allocated<T>, O>>`
329+
/// as returned from `alloc`. The receiver is consumed, and a the
330+
/// now-initialized `Option<Id<T, O>>` (with the same `T` and `O`) is
331+
/// returned.
331332
///
332333
/// - The `copy` family: The receiver may be anything that implements
333334
/// [`MessageReceiver`] and the return type is a generic `Option<Id<T, O>>`.

‎objc2/src/rc/allocated.rs

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
/// TODO
2+
///
3+
/// The reason we use `Option<Id<Allocated<T>, O>>` instead of just `*mut T`
4+
/// is:
5+
/// - To safely know the object is valid (albeit uninitialized)
6+
/// - To allow specifying ownership
7+
/// - To allow releasing allocated (but yet still uninitialized) objects.
8+
#[repr(transparent)]
9+
pub struct Allocated<T: ?Sized> {
10+
_inner: T,
11+
}
12+
13+
// Explicitly don't implement `Message` nor `RefEncode`!

‎objc2/src/rc/id.rs

Lines changed: 31 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ use core::ops::{Deref, DerefMut};
55
use core::panic::{RefUnwindSafe, UnwindSafe};
66
use core::ptr::NonNull;
77

8+
use super::Allocated;
89
use super::AutoreleasePool;
910
use super::{Owned, Ownership, Shared};
1011
use crate::ffi;
@@ -124,6 +125,36 @@ pub struct Id<T: ?Sized, O: Ownership> {
124125
notunwindsafe: PhantomData<&'static mut ()>,
125126
}
126127

128+
impl<T: ?Sized, O: Ownership> Id<T, O> {
129+
#[inline]
130+
unsafe fn new_nonnull(ptr: NonNull<T>) -> Self {
131+
Self {
132+
ptr,
133+
item: PhantomData,
134+
own: PhantomData,
135+
notunwindsafe: PhantomData,
136+
}
137+
}
138+
}
139+
140+
impl<T: Message + ?Sized, O: Ownership> Id<Allocated<T>, O> {
141+
#[inline]
142+
pub(crate) unsafe fn new_allocated(ptr: *mut T) -> Option<Self> {
143+
// SAFETY: Upheld by the caller
144+
NonNull::new(ptr as *mut Allocated<T>).map(|ptr| unsafe { Self::new_nonnull(ptr) })
145+
}
146+
147+
#[inline]
148+
pub(crate) unsafe fn assume_init(this: Self) -> Id<T, O> {
149+
let ptr = ManuallyDrop::new(this).ptr;
150+
let ptr = ptr.as_ptr() as *mut T;
151+
let ptr = unsafe { NonNull::new_unchecked(ptr) };
152+
// SAFETY: The pointer is valid.
153+
// Caller verifies that the object is allocated.
154+
unsafe { Id::new_nonnull(ptr) }
155+
}
156+
}
157+
127158
impl<T: Message + ?Sized, O: Ownership> Id<T, O> {
128159
/// Constructs an [`Id`] to an object that already has +1 retain count.
129160
///
@@ -181,16 +212,6 @@ impl<T: Message + ?Sized, O: Ownership> Id<T, O> {
181212
NonNull::new(ptr).map(|ptr| unsafe { Id::new_nonnull(ptr) })
182213
}
183214

184-
#[inline]
185-
unsafe fn new_nonnull(ptr: NonNull<T>) -> Id<T, O> {
186-
Self {
187-
ptr,
188-
item: PhantomData,
189-
own: PhantomData,
190-
notunwindsafe: PhantomData,
191-
}
192-
}
193-
194215
/// Returns a raw pointer to the object.
195216
///
196217
/// The pointer is valid for at least as long as the `Id` is held.

‎objc2/src/rc/mod.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@
5555
//! assert!(weak.load().is_none());
5656
//! ```
5757
58+
mod allocated;
5859
mod autorelease;
5960
mod id;
6061
mod id_forwarding_impls;
@@ -65,6 +66,7 @@ mod weak_id;
6566
#[cfg(test)]
6667
mod test_object;
6768

69+
pub use self::allocated::Allocated;
6870
pub use self::autorelease::{autoreleasepool, AutoreleasePool, AutoreleaseSafe};
6971
pub use self::id::Id;
7072
pub use self::id_traits::{DefaultId, SliceId, SliceIdMut};

‎tests/assembly/test_msg_send_id/lib.rs

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,18 @@
11
//! Test assembly output of `msg_send_id!` internals.
22
use objc2::__macro_helpers::{MsgSendId, RetainSemantics};
3-
use objc2::rc::{Id, Shared};
3+
use objc2::rc::{Allocated, Id, Shared};
44
use objc2::runtime::{Class, Object, Sel};
55

66
#[no_mangle]
7-
unsafe fn handle_alloc(obj: &Class, sel: Sel) -> Option<Id<Object, Shared>> {
7+
unsafe fn handle_alloc(obj: &Class, sel: Sel) -> Option<Id<Allocated<Object>, Shared>> {
88
<RetainSemantics<false, true, false, false>>::send_message_id(obj, sel, ()).unwrap()
99
}
1010

1111
#[no_mangle]
12-
unsafe fn handle_init(obj: Option<Id<Object, Shared>>, sel: Sel) -> Option<Id<Object, Shared>> {
12+
unsafe fn handle_init(
13+
obj: Option<Id<Allocated<Object>, Shared>>,
14+
sel: Sel,
15+
) -> Option<Id<Object, Shared>> {
1316
<RetainSemantics<false, false, true, false>>::send_message_id(obj, sel, ()).unwrap()
1417
}
1518

@@ -21,7 +24,7 @@ unsafe fn handle_alloc_init(obj: &Class, sel1: Sel, sel2: Sel) -> Option<Id<Obje
2124

2225
#[no_mangle]
2326
unsafe fn handle_alloc_release(cls: &Class, sel: Sel) {
24-
let _obj: Id<Object, Shared> =
27+
let _obj: Id<Allocated<Object>, Shared> =
2528
<RetainSemantics<false, true, false, false>>::send_message_id(cls, sel, ())
2629
.unwrap()
2730
.unwrap_unchecked();

‎tests/ui/msg_send_id_invalid_receiver.rs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,20 @@
11
//! Test compiler output with invalid msg_send_id receivers.
22
use objc2::msg_send_id;
33
use objc2::runtime::{Class, Object};
4-
use objc2::rc::{Id, Shared};
4+
use objc2::rc::{Allocated, Id, Shared};
55

66
fn main() {
77
let obj: &Object;
88
let _: Id<Object, Shared> = unsafe { msg_send_id![obj, new].unwrap() };
9-
let _: Id<Object, Shared> = unsafe { msg_send_id![obj, alloc].unwrap() };
9+
let _: Id<Allocated<Object>, Shared> = unsafe { msg_send_id![obj, alloc].unwrap() };
1010
let _: Id<Object, Shared> = unsafe { msg_send_id![obj, init].unwrap() };
1111

1212
let cls: &Class;
1313
let _: Id<Object, Shared> = unsafe { msg_send_id![cls, init].unwrap() };
14+
let obj: Id<Object, Shared>;
15+
let _: Id<Object, Shared> = unsafe { msg_send_id![obj, init].unwrap() };
16+
let obj: Option<Id<Object, Shared>>;
17+
let _: Id<Object, Shared> = unsafe { msg_send_id![obj, init].unwrap() };
1418

1519
let obj: Id<Object, Shared>;
1620
let _: Id<Object, Shared> = unsafe { msg_send_id![obj, copy].unwrap() };

‎tests/ui/msg_send_id_invalid_receiver.stderr

Lines changed: 44 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -16,13 +16,13 @@ note: associated function defined here
1616
| ^^^^^^^^^^^^^^^
1717

1818
error[E0308]: mismatched types
19-
--> ui/msg_send_id_invalid_receiver.rs:9:55
19+
--> ui/msg_send_id_invalid_receiver.rs:9:66
2020
|
21-
9 | let _: Id<Object, Shared> = unsafe { msg_send_id![obj, alloc].unwrap() };
22-
| -------------^^^--------
23-
| | |
24-
| | expected struct `objc2::runtime::Class`, found struct `objc2::runtime::Object`
25-
| arguments to this function are incorrect
21+
9 | let _: Id<Allocated<Object>, Shared> = unsafe { msg_send_id![obj, alloc].unwrap() };
22+
| -------------^^^--------
23+
| | |
24+
| | expected struct `objc2::runtime::Class`, found struct `objc2::runtime::Object`
25+
| arguments to this function are incorrect
2626
|
2727
= note: expected reference `&objc2::runtime::Class`
2828
found reference `&objc2::runtime::Object`
@@ -41,7 +41,7 @@ error[E0308]: mismatched types
4141
| | expected enum `Option`, found `&objc2::runtime::Object`
4242
| arguments to this function are incorrect
4343
|
44-
= note: expected enum `Option<Id<_, _>>`
44+
= note: expected enum `Option<Id<Allocated<_>, _>>`
4545
found reference `&objc2::runtime::Object`
4646
note: associated function defined here
4747
--> $WORKSPACE/objc2/src/__macro_helpers.rs
@@ -58,18 +58,52 @@ error[E0308]: mismatched types
5858
| | expected enum `Option`, found `&objc2::runtime::Class`
5959
| arguments to this function are incorrect
6060
|
61-
= note: expected enum `Option<Id<_, _>>`
61+
= note: expected enum `Option<Id<Allocated<_>, _>>`
6262
found reference `&objc2::runtime::Class`
6363
note: associated function defined here
6464
--> $WORKSPACE/objc2/src/__macro_helpers.rs
6565
|
6666
| unsafe fn send_message_id<A: MessageArguments>(
6767
| ^^^^^^^^^^^^^^^
6868

69+
error[E0308]: mismatched types
70+
--> ui/msg_send_id_invalid_receiver.rs:15:55
71+
|
72+
15 | let _: Id<Object, Shared> = unsafe { msg_send_id![obj, init].unwrap() };
73+
| -------------^^^-------
74+
| | |
75+
| | expected enum `Option`, found struct `Id`
76+
| arguments to this function are incorrect
77+
|
78+
= note: expected enum `Option<Id<Allocated<_>, _>>`
79+
found struct `Id<objc2::runtime::Object, Shared>`
80+
note: associated function defined here
81+
--> $WORKSPACE/objc2/src/__macro_helpers.rs
82+
|
83+
| unsafe fn send_message_id<A: MessageArguments>(
84+
| ^^^^^^^^^^^^^^^
85+
86+
error[E0308]: mismatched types
87+
--> ui/msg_send_id_invalid_receiver.rs:17:55
88+
|
89+
17 | let _: Id<Object, Shared> = unsafe { msg_send_id![obj, init].unwrap() };
90+
| -------------^^^-------
91+
| | |
92+
| | expected struct `Allocated`, found struct `objc2::runtime::Object`
93+
| arguments to this function are incorrect
94+
|
95+
= note: expected enum `Option<Id<Allocated<_>, _>>`
96+
found enum `Option<Id<objc2::runtime::Object, Shared>>`
97+
note: associated function defined here
98+
--> $WORKSPACE/objc2/src/__macro_helpers.rs
99+
|
100+
| unsafe fn send_message_id<A: MessageArguments>(
101+
| ^^^^^^^^^^^^^^^
102+
69103
error[E0277]: the trait bound `Id<objc2::runtime::Object, Shared>: MessageReceiver` is not satisfied
70-
--> ui/msg_send_id_invalid_receiver.rs:16:42
104+
--> ui/msg_send_id_invalid_receiver.rs:20:42
71105
|
72-
16 | let _: Id<Object, Shared> = unsafe { msg_send_id![obj, copy].unwrap() };
106+
20 | let _: Id<Object, Shared> = unsafe { msg_send_id![obj, copy].unwrap() };
73107
| ^^^^^^^^^^^^^^^^^^^^^^^ the trait `MessageReceiver` is not implemented for `Id<objc2::runtime::Object, Shared>`
74108
|
75109
= help: the following other types implement trait `MessageReceiver`:

‎tests/ui/msg_send_id_invalid_return.rs

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,24 @@
11
//! Test compiler output with invalid msg_send_id receivers.
22
use objc2::msg_send_id;
33
use objc2::runtime::{Class, Object};
4-
use objc2::rc::{Id, Owned, Shared};
4+
use objc2::rc::{Allocated, Id, Owned, Shared};
55
use objc2_foundation::NSObject;
66

77
fn main() {
88
let cls: &Class;
99
let _: &Object = unsafe { msg_send_id![cls, new].unwrap() };
1010
let _: Id<Class, Shared> = unsafe { msg_send_id![cls, new].unwrap() };
1111
let _: &Object = unsafe { msg_send_id![cls, alloc].unwrap() };
12-
let _: Id<Class, Shared> = unsafe { msg_send_id![cls, alloc].unwrap() };
12+
let _: Id<Allocated<Class>, Shared> = unsafe { msg_send_id![cls, alloc].unwrap() };
13+
let _: Id<Object, Shared> = unsafe { msg_send_id![cls, alloc].unwrap() };
1314

14-
let obj: Option<Id<Object, Shared>>;
15+
let obj: Option<Id<Allocated<Object>, Shared>>;
1516
let _: &Object = unsafe { msg_send_id![obj, init].unwrap() };
16-
let obj: Option<Id<Object, Shared>>;
17+
let obj: Option<Id<Allocated<Object>, Shared>>;
1718
let _: Id<Class, Shared> = unsafe { msg_send_id![obj, init].unwrap() };
18-
let obj: Option<Id<Object, Shared>>;
19+
let obj: Option<Id<Allocated<Object>, Shared>>;
1920
let _: Id<NSObject, Shared> = unsafe { msg_send_id![obj, init].unwrap() };
20-
let obj: Option<Id<Object, Shared>>;
21+
let obj: Option<Id<Allocated<Object>, Shared>>;
2122
let _: Id<Object, Owned> = unsafe { msg_send_id![obj, init].unwrap() };
2223

2324
let obj: Id<Object, Shared>;

0 commit comments

Comments
 (0)