1use crate::{
8 Koid, MonotonicInstant, Name, ObjectQuery, ObjectType, Port, Property, PropertyQuery, Rights,
9 Signals, Status, Topic, WaitAsyncOpts, ok, sys,
10};
11use std::marker::PhantomData;
12use std::mem::{ManuallyDrop, MaybeUninit};
13use std::num::NonZeroU32;
14use zerocopy::{FromBytes, Immutable, IntoBytes};
15
16pub(crate) const INFO_VEC_SIZE_INITIAL: usize = 16;
18
19#[derive(Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
37#[repr(transparent)]
38pub struct Handle(NonZeroU32);
39
40static_assertions::const_assert_eq!(sys::ZX_HANDLE_INVALID, 0);
45static_assertions::assert_eq_size!(Handle, sys::zx_handle_t);
46static_assertions::assert_eq_align!(Handle, sys::zx_handle_t);
47static_assertions::assert_eq_size!(Option<Handle>, sys::zx_handle_t);
48static_assertions::assert_eq_align!(Option<Handle>, sys::zx_handle_t);
49
50impl Handle {
51 pub unsafe fn from_raw(raw: sys::zx_handle_t) -> Self {
62 debug_assert!(Self::check_raw_valid(raw).is_ok());
63 Self(NonZeroU32::new(raw).unwrap())
64 }
65
66 pub fn check_raw_valid(raw: sys::zx_handle_t) -> Result<(), Status> {
73 ok(unsafe { sys::zx_handle_check_valid(raw) })
75 }
76
77 pub const fn raw_handle(&self) -> sys::zx_handle_t {
79 self.0.get()
80 }
81
82 pub fn into_raw(self) -> sys::zx_handle_t {
84 let ret = self.0.get();
85 std::mem::forget(self);
86 ret
87 }
88
89 pub fn duplicate(&self, rights: Rights) -> Result<Self, Status> {
93 let mut out = 0;
94 let status =
96 unsafe { sys::zx_handle_duplicate(self.raw_handle(), rights.bits(), &mut out) };
97 ok(status)?;
98
99 Ok(unsafe { Self::from_raw(out) })
101 }
102
103 pub fn replace(self, rights: Rights) -> Result<Self, Status> {
107 let mut out = 0;
108
109 let status = unsafe { sys::zx_handle_replace(self.into_raw(), rights.bits(), &mut out) };
111 ok(status)?;
112
113 unsafe { Ok(Self::from_raw(out)) }
115 }
116
117 pub fn signal(&self, clear_mask: Signals, set_mask: Signals) -> Result<(), Status> {
119 ok(unsafe { sys::zx_object_signal(self.raw_handle(), clear_mask.bits(), set_mask.bits()) })
121 }
122
123 pub fn wait_one(&self, signals: Signals, deadline: MonotonicInstant) -> WaitResult {
126 let mut pending = Signals::empty().bits();
127 let status = unsafe {
129 sys::zx_object_wait_one(
130 self.raw_handle(),
131 signals.bits(),
132 deadline.into_nanos(),
133 &mut pending,
134 )
135 };
136 let signals = Signals::from_bits_truncate(pending);
137 match ok(status) {
138 Ok(()) => WaitResult::Ok(signals),
139 Err(Status::TIMED_OUT) => WaitResult::TimedOut(signals),
140 Err(Status::CANCELED) => WaitResult::Canceled(signals),
141 Err(e) => WaitResult::Err(e),
142 }
143 }
144
145 pub fn wait_async(
148 &self,
149 port: &Port,
150 key: u64,
151 signals: Signals,
152 options: WaitAsyncOpts,
153 ) -> Result<(), Status> {
154 ok(unsafe {
156 sys::zx_object_wait_async(
157 self.raw_handle(),
158 port.raw_handle(),
159 key,
160 signals.bits(),
161 options.bits(),
162 )
163 })
164 }
165
166 pub fn get_name(&self) -> Result<Name, Status> {
172 self.get_property::<NameProperty>()
173 }
174
175 pub fn set_name(&self, name: &Name) -> Result<(), Status> {
185 self.set_property::<NameProperty>(&name)
186 }
187
188 pub(crate) fn get_property<P: PropertyQuery>(&self) -> Result<P::PropTy, Status>
190 where
191 P::PropTy: FromBytes + Immutable,
192 {
193 let mut out = ::std::mem::MaybeUninit::<P::PropTy>::uninit();
194
195 let status = unsafe {
197 sys::zx_object_get_property(
198 self.raw_handle(),
199 *P::PROPERTY,
200 out.as_mut_ptr().cast::<u8>(),
201 std::mem::size_of::<P::PropTy>(),
202 )
203 };
204 Status::ok(status).map(|_| unsafe { out.assume_init() })
205 }
206
207 pub(crate) fn set_property<P: PropertyQuery>(&self, val: &P::PropTy) -> Result<(), Status>
209 where
210 P::PropTy: IntoBytes + Immutable,
211 {
212 let status = unsafe {
213 sys::zx_object_set_property(
214 self.raw_handle(),
215 *P::PROPERTY,
216 std::ptr::from_ref(val).cast::<u8>(),
217 std::mem::size_of::<P::PropTy>(),
218 )
219 };
220 Status::ok(status)
221 }
222
223 pub fn basic_info(&self) -> Result<HandleBasicInfo, Status> {
227 Ok(HandleBasicInfo::from(self.get_info_single::<HandleBasicInfoQuery>()?))
228 }
229
230 pub fn count_info(&self) -> Result<HandleCountInfo, Status> {
234 Ok(HandleCountInfo::from(self.get_info_single::<HandleCountInfoQuery>()?))
235 }
236
237 pub fn koid(&self) -> Result<Koid, Status> {
239 self.basic_info().map(|info| info.koid)
240 }
241
242 pub(crate) fn get_info<'a, Q: ObjectQuery>(
245 &self,
246 out: &'a mut [MaybeUninit<Q::InfoTy>],
247 ) -> Result<(&'a mut [Q::InfoTy], &'a mut [MaybeUninit<Q::InfoTy>], usize), Status>
248 where
249 Q::InfoTy: FromBytes + Immutable,
250 {
251 let mut actual = 0;
252 let mut avail = 0;
253
254 let status = unsafe {
257 sys::zx_object_get_info(
258 self.raw_handle(),
259 *Q::TOPIC,
260 out.as_mut_ptr().cast::<u8>(),
261 std::mem::size_of_val(out),
262 &mut actual,
263 &mut avail,
264 )
265 };
266 ok(status)?;
267
268 let (initialized, uninit) = out.split_at_mut(actual);
269
270 let initialized: &mut [Q::InfoTy] = unsafe {
274 std::slice::from_raw_parts_mut(
275 initialized.as_mut_ptr().cast::<Q::InfoTy>(),
276 initialized.len(),
277 )
278 };
279
280 Ok((initialized, uninit, avail))
281 }
282
283 pub(crate) fn get_info_single<Q: ObjectQuery>(&self) -> Result<Q::InfoTy, Status>
285 where
286 Q::InfoTy: Copy + FromBytes + Immutable,
287 {
288 let mut info = MaybeUninit::<Q::InfoTy>::uninit();
289 let (info, _uninit, _avail) = self.get_info::<Q>(std::slice::from_mut(&mut info))?;
290 Ok(info[0])
291 }
292
293 pub(crate) fn get_info_vec<Q: ObjectQuery>(&self) -> Result<Vec<Q::InfoTy>, Status> {
297 let mut out = Vec::<Q::InfoTy>::with_capacity(INFO_VEC_SIZE_INITIAL);
299 loop {
300 let (init, _uninit, avail) = self.get_info::<Q>(out.spare_capacity_mut())?;
301 let num_initialized = init.len();
302 if num_initialized == avail {
303 unsafe { out.set_len(num_initialized) };
305 return Ok(out);
306 } else {
307 if avail > out.capacity() {
308 out.reserve_exact(avail - out.len());
309 }
310 }
311 }
312 }
313}
314
315impl Drop for Handle {
316 fn drop(&mut self) {
317 unsafe { sys::zx_handle_close(self.0.get()) };
319 }
320}
321
322#[derive(Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
335#[repr(transparent)]
336pub struct NullableHandle(Option<Handle>);
337
338impl AsHandleRef for NullableHandle {
339 fn as_handle_ref(&self) -> HandleRef<'_> {
340 if let Some(inner) = &self.0 {
341 unsafe { Unowned::from_raw_handle(inner.raw_handle()) }
343 } else {
344 unsafe { Unowned::from_raw_handle(sys::ZX_HANDLE_INVALID) }
346 }
347 }
348}
349
350impl HandleBased for NullableHandle {
351 fn into_raw(self) -> sys::zx_handle_t {
352 if let Some(inner) = self.0 { inner.into_raw() } else { sys::ZX_HANDLE_INVALID }
353 }
354}
355
356impl NullableHandle {
357 #[inline(always)]
359 pub const fn invalid() -> Self {
360 Self(None)
361 }
362
363 pub const unsafe fn from_raw(raw: sys::zx_handle_t) -> Self {
375 if let Some(inner) = NonZeroU32::new(raw) { Self(Some(Handle(inner))) } else { Self(None) }
379 }
380
381 pub const fn raw_handle(&self) -> sys::zx_handle_t {
382 if let Some(inner) = &self.0 { inner.raw_handle() } else { sys::ZX_HANDLE_INVALID }
383 }
384
385 pub fn into_raw(self) -> sys::zx_handle_t {
386 self.0.map(Handle::into_raw).unwrap_or(sys::ZX_HANDLE_INVALID)
387 }
388
389 pub fn as_handle_ref(&self) -> HandleRef<'_> {
390 AsHandleRef::as_handle_ref(self)
391 }
392
393 pub const fn is_invalid(&self) -> bool {
394 self.0.is_none()
395 }
396
397 pub fn duplicate(&self, rights: Rights) -> Result<Self, Status> {
398 self.0
399 .as_ref()
400 .map(|h| h.duplicate(rights).map(|new| Self(Some(new))))
401 .unwrap_or(Err(Status::BAD_HANDLE))
402 }
403
404 pub fn replace(mut self, rights: Rights) -> Result<Self, Status> {
405 if let Some(inner) = self.0.take() {
406 let inner = inner.replace(rights)?;
407 Ok(Self(Some(inner)))
408 } else {
409 Ok(Self(None))
410 }
411 }
412
413 pub fn signal(&self, clear_mask: Signals, set_mask: Signals) -> Result<(), Status> {
414 self.0.as_ref().map(|h| h.signal(clear_mask, set_mask)).unwrap_or(Err(Status::BAD_HANDLE))
415 }
416
417 pub fn wait_one(&self, signals: Signals, deadline: MonotonicInstant) -> WaitResult {
418 self.0
419 .as_ref()
420 .map(|h| h.wait_one(signals, deadline))
421 .unwrap_or(WaitResult::Err(Status::BAD_HANDLE))
422 }
423
424 pub fn wait_async(
425 &self,
426 port: &Port,
427 key: u64,
428 signals: Signals,
429 options: WaitAsyncOpts,
430 ) -> Result<(), Status> {
431 self.0
432 .as_ref()
433 .map(|h| h.wait_async(port, key, signals, options))
434 .unwrap_or(Err(Status::BAD_HANDLE))
435 }
436
437 pub fn get_name(&self) -> Result<Name, Status> {
438 self.0.as_ref().map(Handle::get_name).unwrap_or(Err(Status::BAD_HANDLE))
439 }
440
441 pub fn set_name(&self, name: &Name) -> Result<(), Status> {
442 self.0.as_ref().map(|h| h.set_name(name)).unwrap_or(Err(Status::BAD_HANDLE))
443 }
444
445 pub fn basic_info(&self) -> Result<HandleBasicInfo, Status> {
446 self.0.as_ref().map(Handle::basic_info).unwrap_or(Err(Status::BAD_HANDLE))
447 }
448
449 pub fn count_info(&self) -> Result<HandleCountInfo, Status> {
450 self.0.as_ref().map(Handle::count_info).unwrap_or(Err(Status::BAD_HANDLE))
451 }
452
453 pub fn koid(&self) -> Result<Koid, Status> {
454 self.0.as_ref().map(Handle::koid).unwrap_or(Err(Status::BAD_HANDLE))
455 }
456
457 pub(crate) fn get_info<'a, Q: ObjectQuery>(
458 &self,
459 out: &'a mut [MaybeUninit<Q::InfoTy>],
460 ) -> Result<(&'a mut [Q::InfoTy], &'a mut [MaybeUninit<Q::InfoTy>], usize), Status>
461 where
462 Q::InfoTy: FromBytes + Immutable,
463 {
464 self.0.as_ref().map(|h| h.get_info::<Q>(out)).unwrap_or(Err(Status::BAD_HANDLE))
465 }
466
467 pub(crate) fn get_info_single<Q: ObjectQuery>(&self) -> Result<Q::InfoTy, Status>
468 where
469 Q::InfoTy: Copy + FromBytes + Immutable,
470 {
471 self.0.as_ref().map(|h| h.get_info_single::<Q>()).unwrap_or(Err(Status::BAD_HANDLE))
472 }
473
474 pub(crate) fn get_info_vec<Q: ObjectQuery>(&self) -> Result<Vec<Q::InfoTy>, Status> {
475 self.0.as_ref().map(|h| h.get_info_vec::<Q>()).unwrap_or(Err(Status::BAD_HANDLE))
476 }
477
478 pub(crate) fn get_property<P: PropertyQuery>(&self) -> Result<P::PropTy, Status>
479 where
480 P::PropTy: FromBytes + Immutable,
481 {
482 self.0.as_ref().map(|h| h.get_property::<P>()).unwrap_or(Err(Status::BAD_HANDLE))
483 }
484
485 pub(crate) fn set_property<P: PropertyQuery>(&self, val: &P::PropTy) -> Result<(), Status>
486 where
487 P::PropTy: IntoBytes + Immutable,
488 {
489 self.0.as_ref().map(|h| h.set_property::<P>(val)).unwrap_or(Err(Status::BAD_HANDLE))
490 }
491}
492
493struct NameProperty();
494unsafe impl PropertyQuery for NameProperty {
495 const PROPERTY: Property = Property::NAME;
496 type PropTy = Name;
499}
500
501#[derive(Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
505#[repr(transparent)]
506pub struct Unowned<'a, T> {
507 inner: ManuallyDrop<T>,
508 marker: PhantomData<&'a T>,
509}
510
511impl<'a, T: Into<NullableHandle>> ::std::ops::Deref for Unowned<'a, T> {
512 type Target = T;
513
514 fn deref(&self) -> &Self::Target {
515 &*self.inner
516 }
517}
518
519impl<'a, T: AsHandleRef> AsHandleRef for Unowned<'a, T> {
520 fn as_handle_ref(&self) -> HandleRef<'_> {
521 self.inner.as_handle_ref()
522 }
523}
524
525impl<T: HandleBased> Clone for Unowned<'_, T> {
526 fn clone(&self) -> Self {
527 unsafe { Self::from_raw_handle(self.inner.as_handle_ref().raw_handle()) }
528 }
529}
530
531pub type HandleRef<'a> = Unowned<'a, NullableHandle>;
532
533impl<'a, T: Into<NullableHandle>> Unowned<'a, T> {
534 pub fn new<U: AsHandleRef + From<NullableHandle>>(inner: &'a U) -> Self
538 where
539 T: From<U>,
540 {
541 Unowned {
544 inner: ManuallyDrop::new(T::from(U::from(unsafe {
545 NullableHandle::from_raw(inner.as_handle_ref().raw_handle())
546 }))),
547 marker: PhantomData,
548 }
549 }
550}
551
552impl<'a, T: HandleBased> Unowned<'a, T> {
553 pub unsafe fn from_raw_handle(handle: sys::zx_handle_t) -> Self {
565 Unowned {
566 inner: ManuallyDrop::new(T::from(unsafe { NullableHandle::from_raw(handle) })),
567 marker: PhantomData,
568 }
569 }
570
571 pub fn raw_handle(&self) -> sys::zx_handle_t {
573 NullableHandle::raw_handle(&*self.inner.as_handle_ref())
574 }
575}
576
577#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
583pub enum WaitResult {
584 Ok(Signals),
586
587 TimedOut(Signals),
591
592 Canceled(Signals),
599
600 Err(Status),
603}
604
605impl WaitResult {
606 pub const fn to_result(self) -> Result<Signals, Status> {
609 match self {
610 WaitResult::Ok(signals) => Ok(signals),
611 WaitResult::TimedOut(_signals) => Err(Status::TIMED_OUT),
612 WaitResult::Canceled(_signals) => Err(Status::CANCELED),
613 WaitResult::Err(status) => Err(status),
614 }
615 }
616
617 #[must_use = "if you intended to assert that this is ok, consider `.unwrap()` instead"]
622 #[inline]
623 pub const fn is_ok(&self) -> bool {
624 self.to_result().is_ok()
625 }
626
627 #[must_use = "if you intended to assert that this is err, consider `.unwrap_err()` instead"]
628 #[inline]
629 pub const fn is_err(&self) -> bool {
630 self.to_result().is_err()
631 }
632
633 #[inline]
634 pub fn map<U, F: FnOnce(Signals) -> U>(self, op: F) -> Result<U, Status> {
635 self.to_result().map(op)
636 }
637
638 #[inline]
639 pub fn map_err<F, O: FnOnce(Status) -> F>(self, op: O) -> Result<Signals, F> {
640 self.to_result().map_err(op)
641 }
642
643 #[inline]
644 #[track_caller]
645 pub fn expect(self, msg: &str) -> Signals {
646 self.to_result().expect(msg)
647 }
648
649 #[inline]
650 #[track_caller]
651 pub fn expect_err(self, msg: &str) -> Status {
652 self.to_result().expect_err(msg)
653 }
654
655 #[inline(always)]
656 #[track_caller]
657 pub fn unwrap(self) -> Signals {
658 self.to_result().unwrap()
659 }
660}
661
662impl<'a> Unowned<'a, NullableHandle> {
663 pub fn cast<T: HandleBased>(self) -> Unowned<'a, T> {
665 unsafe { Unowned::from_raw_handle(self.raw_handle()) }
667 }
668}
669
670pub trait AsHandleRef {
672 fn as_handle_ref(&self) -> HandleRef<'_>;
675}
676
677impl<T: AsHandleRef> AsHandleRef for &T {
678 fn as_handle_ref(&self) -> HandleRef<'_> {
679 (*self).as_handle_ref()
680 }
681}
682
683pub trait HandleBased: AsHandleRef + From<NullableHandle> + Into<NullableHandle> {
691 fn duplicate_handle(&self, rights: Rights) -> Result<Self, Status> {
695 self.as_handle_ref().duplicate(rights).map(|handle| Self::from(handle))
696 }
697
698 fn replace_handle(self, rights: Rights) -> Result<Self, Status> {
703 <Self as Into<NullableHandle>>::into(self).replace(rights).map(|handle| Self::from(handle))
704 }
705
706 fn into_handle(self) -> NullableHandle {
710 self.into()
711 }
712
713 fn into_raw(self) -> sys::zx_handle_t {
717 self.into_handle().into_raw()
718 }
719
720 fn from_handle(handle: NullableHandle) -> Self {
724 Self::from(handle)
725 }
726}
727
728pub trait Peered: HandleBased {
730 fn signal_peer(&self, clear_mask: Signals, set_mask: Signals) -> Result<(), Status> {
735 let handle = self.as_handle_ref().raw_handle();
736 let status =
737 unsafe { sys::zx_object_signal_peer(handle, clear_mask.bits(), set_mask.bits()) };
738 ok(status)
739 }
740
741 fn is_closed(&self) -> Result<bool, Status> {
749 match self
750 .as_handle_ref()
751 .wait_one(Signals::OBJECT_PEER_CLOSED, MonotonicInstant::INFINITE_PAST)
752 {
753 WaitResult::Ok(signals) => Ok(signals.contains(Signals::OBJECT_PEER_CLOSED)),
754 WaitResult::TimedOut(_) => Ok(false),
755 WaitResult::Canceled(_) => Err(Status::CANCELED),
756 WaitResult::Err(e) => Err(e),
757 }
758 }
759}
760
761#[derive(Debug, Copy, Clone, Eq, PartialEq)]
765pub struct HandleBasicInfo {
766 pub koid: Koid,
767 pub rights: Rights,
768 pub object_type: ObjectType,
769 pub related_koid: Koid,
770}
771
772impl Default for HandleBasicInfo {
773 fn default() -> Self {
774 Self::from(sys::zx_info_handle_basic_t::default())
775 }
776}
777
778impl From<sys::zx_info_handle_basic_t> for HandleBasicInfo {
779 fn from(info: sys::zx_info_handle_basic_t) -> Self {
780 let sys::zx_info_handle_basic_t { koid, rights, type_, related_koid, .. } = info;
781
782 HandleBasicInfo {
785 koid: Koid::from_raw(koid),
786 rights: Rights::from_bits_truncate(rights),
787 object_type: ObjectType::from_raw(type_),
788 related_koid: Koid::from_raw(related_koid),
789 }
790 }
791}
792
793struct HandleBasicInfoQuery;
796unsafe impl ObjectQuery for HandleBasicInfoQuery {
797 const TOPIC: Topic = Topic::HANDLE_BASIC;
798 type InfoTy = sys::zx_info_handle_basic_t;
799}
800
801sys::zx_info_handle_count_t!(HandleCountInfo);
802
803impl From<sys::zx_info_handle_count_t> for HandleCountInfo {
804 fn from(sys::zx_info_handle_count_t { handle_count }: sys::zx_info_handle_count_t) -> Self {
805 HandleCountInfo { handle_count }
806 }
807}
808
809struct HandleCountInfoQuery;
812unsafe impl ObjectQuery for HandleCountInfoQuery {
813 const TOPIC: Topic = Topic::HANDLE_COUNT;
814 type InfoTy = sys::zx_info_handle_count_t;
815}
816
817#[derive(Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
819pub enum HandleOp<'a> {
820 Move(NullableHandle),
821 Duplicate(HandleRef<'a>),
822}
823
824#[derive(Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
826#[repr(C)]
827pub struct HandleDisposition<'a> {
828 operation: sys::zx_handle_op_t,
830 handle: sys::zx_handle_t,
832 _handle_lifetime: std::marker::PhantomData<&'a ()>,
834
835 pub object_type: ObjectType,
836 pub rights: Rights,
837 pub result: Status,
838}
839
840static_assertions::assert_eq_size!(HandleDisposition<'_>, sys::zx_handle_disposition_t);
841static_assertions::const_assert_eq!(
842 std::mem::offset_of!(HandleDisposition<'_>, operation),
843 std::mem::offset_of!(sys::zx_handle_disposition_t, operation)
844);
845static_assertions::const_assert_eq!(
846 std::mem::offset_of!(HandleDisposition<'_>, handle),
847 std::mem::offset_of!(sys::zx_handle_disposition_t, handle)
848);
849static_assertions::const_assert_eq!(
850 std::mem::offset_of!(HandleDisposition<'_>, object_type),
851 std::mem::offset_of!(sys::zx_handle_disposition_t, type_)
852);
853static_assertions::const_assert_eq!(
854 std::mem::offset_of!(HandleDisposition<'_>, rights),
855 std::mem::offset_of!(sys::zx_handle_disposition_t, rights)
856);
857static_assertions::const_assert_eq!(
858 std::mem::offset_of!(HandleDisposition<'_>, result),
859 std::mem::offset_of!(sys::zx_handle_disposition_t, result)
860);
861
862impl<'a> HandleDisposition<'a> {
863 #[inline]
864 pub fn new(
865 handle_op: HandleOp<'a>,
866 object_type: ObjectType,
867 rights: Rights,
868 status: Status,
869 ) -> Self {
870 let (operation, handle) = match handle_op {
871 HandleOp::Move(h) => (sys::ZX_HANDLE_OP_MOVE, h.into_raw()),
872 HandleOp::Duplicate(h) => (sys::ZX_HANDLE_OP_DUPLICATE, h.raw_handle()),
873 };
874
875 Self {
876 operation,
877 handle,
878 _handle_lifetime: std::marker::PhantomData,
879 object_type,
880 rights: rights,
881 result: status,
882 }
883 }
884
885 pub fn raw_handle(&self) -> sys::zx_handle_t {
886 self.handle
887 }
888
889 pub fn is_move(&self) -> bool {
890 self.operation == sys::ZX_HANDLE_OP_MOVE
891 }
892
893 pub fn is_duplicate(&self) -> bool {
894 self.operation == sys::ZX_HANDLE_OP_DUPLICATE
895 }
896
897 pub fn take_op(&mut self) -> HandleOp<'a> {
898 match self.operation {
899 sys::ZX_HANDLE_OP_MOVE => {
900 HandleOp::Move(unsafe {
903 NullableHandle::from_raw(std::mem::replace(
904 &mut self.handle,
905 sys::ZX_HANDLE_INVALID,
906 ))
907 })
908 }
909 sys::ZX_HANDLE_OP_DUPLICATE => {
910 HandleOp::Duplicate(Unowned {
913 inner: ManuallyDrop::new(unsafe { NullableHandle::from_raw(self.handle) }),
914 marker: PhantomData,
915 })
916 }
917 _ => unreachable!(),
918 }
919 }
920
921 pub fn into_raw(mut self) -> sys::zx_handle_disposition_t {
922 match self.take_op() {
923 HandleOp::Move(mut handle) => sys::zx_handle_disposition_t {
924 operation: sys::ZX_HANDLE_OP_MOVE,
925 handle: std::mem::replace(&mut handle, NullableHandle::invalid()).into_raw(),
926 type_: self.object_type.into_raw(),
927 rights: self.rights.bits(),
928 result: self.result.into_raw(),
929 },
930 HandleOp::Duplicate(handle_ref) => sys::zx_handle_disposition_t {
931 operation: sys::ZX_HANDLE_OP_DUPLICATE,
932 handle: handle_ref.raw_handle(),
933 type_: self.object_type.into_raw(),
934 rights: self.rights.bits(),
935 result: self.result.into_raw(),
936 },
937 }
938 }
939}
940
941impl<'a> Drop for HandleDisposition<'a> {
942 fn drop(&mut self) {
943 if self.operation == sys::ZX_HANDLE_OP_MOVE {
945 unsafe { drop(NullableHandle::from_raw(self.handle)) };
946 }
947 }
948}
949
950#[derive(Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
954#[repr(C)]
955pub struct HandleInfo {
956 pub handle: NullableHandle,
957 pub object_type: ObjectType,
958 pub rights: Rights,
959
960 pub(crate) _unused: u32,
962}
963
964static_assertions::assert_eq_size!(HandleInfo, sys::zx_handle_info_t);
965static_assertions::const_assert_eq!(
966 std::mem::offset_of!(HandleInfo, handle),
967 std::mem::offset_of!(sys::zx_handle_info_t, handle)
968);
969static_assertions::const_assert_eq!(
970 std::mem::offset_of!(HandleInfo, object_type),
971 std::mem::offset_of!(sys::zx_handle_info_t, ty)
972);
973static_assertions::const_assert_eq!(
974 std::mem::offset_of!(HandleInfo, rights),
975 std::mem::offset_of!(sys::zx_handle_info_t, rights)
976);
977static_assertions::const_assert_eq!(
978 std::mem::offset_of!(HandleInfo, _unused),
979 std::mem::offset_of!(sys::zx_handle_info_t, unused)
980);
981
982impl HandleInfo {
983 pub const fn new(handle: NullableHandle, object_type: ObjectType, rights: Rights) -> Self {
985 Self { handle, object_type, rights, _unused: 0 }
986 }
987
988 pub const unsafe fn from_raw(raw: sys::zx_handle_info_t) -> HandleInfo {
996 HandleInfo::new(
997 unsafe { NullableHandle::from_raw(raw.handle) },
999 ObjectType::from_raw(raw.ty),
1000 Rights::from_bits_retain(raw.rights),
1001 )
1002 }
1003}
1004
1005#[cfg(test)]
1006mod tests {
1007 use super::*;
1008 use zx::{
1011 Channel, HandleBased, HandleDisposition, HandleInfo, HandleOp, Name, NullableHandle,
1012 ObjectType, Rights, Vmo,
1013 };
1014 use zx_sys as sys;
1015
1016 #[test]
1017 fn into_raw() {
1018 let vmo = Vmo::create(1).unwrap();
1019 let h = vmo.into_raw();
1020 let vmo2 = Vmo::from(unsafe { NullableHandle::from_raw(h) });
1021 assert!(vmo2.write(b"1", 0).is_ok());
1022 }
1023
1024 #[test]
1025 fn check_raw_valid() {
1026 assert!(Handle::check_raw_valid(sys::ZX_HANDLE_INVALID).is_err());
1027 let vmo = Vmo::create(1).unwrap();
1028 let vmo_raw = vmo.raw_handle();
1029 assert!(Handle::check_raw_valid(vmo_raw).is_ok());
1030 drop(vmo);
1031 assert!(Handle::check_raw_valid(vmo_raw).is_err());
1032 }
1033
1034 #[test]
1036 fn duplicate() {
1037 let hello_length: usize = 5;
1038
1039 let vmo = Vmo::create(hello_length as u64).unwrap();
1041 assert!(vmo.write(b"hello", 0).is_ok());
1042
1043 let readonly_vmo = vmo.duplicate_handle(Rights::READ).unwrap();
1045 let mut read_vec = vec![0; hello_length];
1047 assert!(readonly_vmo.read(&mut read_vec, 0).is_ok());
1048 assert_eq!(read_vec, b"hello");
1049 assert_eq!(readonly_vmo.write(b"", 0), Err(Status::ACCESS_DENIED));
1050
1051 assert!(vmo.write(b"bye", 0).is_ok());
1053 assert!(readonly_vmo.read(&mut read_vec, 0).is_ok());
1054 assert_eq!(read_vec, b"byelo");
1055 }
1056
1057 #[test]
1059 fn replace() {
1060 let hello_length: usize = 5;
1061
1062 let vmo = Vmo::create(hello_length as u64).unwrap();
1064 assert!(vmo.write(b"hello", 0).is_ok());
1065
1066 let readonly_vmo = vmo.replace_handle(Rights::READ).unwrap();
1068 let mut read_vec = vec![0; hello_length];
1070 assert!(readonly_vmo.read(&mut read_vec, 0).is_ok());
1071 assert_eq!(read_vec, b"hello");
1072 assert_eq!(readonly_vmo.write(b"", 0), Err(Status::ACCESS_DENIED));
1073 }
1074
1075 #[test]
1076 fn set_get_name() {
1077 let vmo = Vmo::create(1).unwrap();
1079 let short_name = Name::new("v").unwrap();
1080 assert!(vmo.set_name(&short_name).is_ok());
1081 assert_eq!(vmo.get_name().unwrap(), short_name);
1082 }
1083
1084 #[test]
1085 fn set_get_max_len_name() {
1086 let vmo = Vmo::create(1).unwrap();
1087 let max_len_name = Name::new("a_great_maximum_length_vmo_name").unwrap(); assert!(vmo.set_name(&max_len_name).is_ok());
1089 assert_eq!(vmo.get_name().unwrap(), max_len_name);
1090 }
1091
1092 #[test]
1093 fn basic_info_channel() {
1094 let (side1, side2) = Channel::create();
1095 let info1 = side1.basic_info().expect("side1 basic_info failed");
1096 let info2 = side2.basic_info().expect("side2 basic_info failed");
1097
1098 assert_eq!(info1.koid, info2.related_koid);
1099 assert_eq!(info2.koid, info1.related_koid);
1100
1101 for info in &[info1, info2] {
1102 assert!(info.koid.raw_koid() >= sys::ZX_KOID_FIRST);
1103 assert_eq!(info.object_type, ObjectType::CHANNEL);
1104 assert!(info.rights.contains(Rights::READ | Rights::WRITE | Rights::WAIT));
1105 }
1106
1107 let side1_repl = side1.replace_handle(Rights::READ).expect("side1 replace_handle failed");
1108 let info1_repl = side1_repl.basic_info().expect("side1_repl basic_info failed");
1109 assert_eq!(info1_repl.koid, info1.koid);
1110 assert_eq!(info1_repl.rights, Rights::READ);
1111 }
1112
1113 #[test]
1114 fn basic_info_vmar() {
1115 let root_vmar = fuchsia_runtime::vmar_root_self();
1117 let info = root_vmar.basic_info().expect("vmar basic_info failed");
1118 assert_eq!(info.object_type, ObjectType::VMAR);
1119 assert!(!info.rights.contains(Rights::WAIT));
1120 }
1121
1122 #[test]
1123 fn count_info() {
1124 let vmo0 = Vmo::create(1).unwrap();
1125 let count_info = vmo0.count_info().expect("vmo0 count_info failed");
1126 assert_eq!(count_info.handle_count, 1);
1127
1128 let vmo1 = vmo0.duplicate_handle(Rights::SAME_RIGHTS).expect("vmo duplicate_handle failed");
1129 let count_info = vmo1.count_info().expect("vmo1 count_info failed");
1130 assert_eq!(count_info.handle_count, 2);
1131 }
1132
1133 #[test]
1134 fn raw_handle_disposition() {
1135 const RAW_HANDLE: sys::zx_handle_t = 1;
1136 let hd = HandleDisposition::new(
1137 HandleOp::Move(unsafe { NullableHandle::from_raw(RAW_HANDLE) }),
1138 ObjectType::VMO,
1139 Rights::EXECUTE,
1140 Status::OK,
1141 );
1142 let raw_hd = hd.into_raw();
1143 assert_eq!(raw_hd.operation, sys::ZX_HANDLE_OP_MOVE);
1144 assert_eq!(raw_hd.handle, RAW_HANDLE);
1145 assert_eq!(raw_hd.rights, sys::ZX_RIGHT_EXECUTE);
1146 assert_eq!(raw_hd.type_, sys::ZX_OBJ_TYPE_VMO);
1147 assert_eq!(raw_hd.result, sys::ZX_OK);
1148 }
1149
1150 #[test]
1151 fn regression_nullable_handle_into_raw_recursion() {
1152 let h = NullableHandle::invalid();
1153 assert_eq!(h.into_raw(), sys::ZX_HANDLE_INVALID);
1155
1156 let vmo = Vmo::create(1).unwrap();
1157 let raw = vmo.raw_handle();
1158 let h = vmo.into_handle();
1159 assert_eq!(h.into_raw(), raw);
1161 }
1162
1163 #[test]
1164 fn handle_info_from_raw() {
1165 const RAW_HANDLE: sys::zx_handle_t = 1;
1166 let raw_hi = sys::zx_handle_info_t {
1167 handle: RAW_HANDLE,
1168 ty: sys::ZX_OBJ_TYPE_VMO,
1169 rights: sys::ZX_RIGHT_EXECUTE,
1170 unused: 128,
1171 };
1172 let hi = unsafe { HandleInfo::from_raw(raw_hi) };
1173 assert_eq!(hi.handle.into_raw(), RAW_HANDLE);
1174 assert_eq!(hi.object_type, ObjectType::VMO);
1175 assert_eq!(hi.rights, Rights::EXECUTE);
1176 }
1177
1178 #[test]
1179 fn basic_peer_closed() {
1180 let (lhs, rhs) = crate::EventPair::create();
1181 assert!(!lhs.is_closed().unwrap());
1182 assert!(!rhs.is_closed().unwrap());
1183 drop(rhs);
1184 assert!(lhs.is_closed().unwrap());
1185 }
1186}