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
721pub trait Peered: HandleBased {
723 fn signal_peer(&self, clear_mask: Signals, set_mask: Signals) -> Result<(), Status> {
728 let handle = self.as_handle_ref().raw_handle();
729 let status =
730 unsafe { sys::zx_object_signal_peer(handle, clear_mask.bits(), set_mask.bits()) };
731 ok(status)
732 }
733
734 fn is_closed(&self) -> Result<bool, Status> {
742 match self
743 .as_handle_ref()
744 .wait_one(Signals::OBJECT_PEER_CLOSED, MonotonicInstant::INFINITE_PAST)
745 {
746 WaitResult::Ok(signals) => Ok(signals.contains(Signals::OBJECT_PEER_CLOSED)),
747 WaitResult::TimedOut(_) => Ok(false),
748 WaitResult::Canceled(_) => Err(Status::CANCELED),
749 WaitResult::Err(e) => Err(e),
750 }
751 }
752}
753
754#[derive(Debug, Copy, Clone, Eq, PartialEq)]
758pub struct HandleBasicInfo {
759 pub koid: Koid,
760 pub rights: Rights,
761 pub object_type: ObjectType,
762 pub related_koid: Koid,
763}
764
765impl Default for HandleBasicInfo {
766 fn default() -> Self {
767 Self::from(sys::zx_info_handle_basic_t::default())
768 }
769}
770
771impl From<sys::zx_info_handle_basic_t> for HandleBasicInfo {
772 fn from(info: sys::zx_info_handle_basic_t) -> Self {
773 let sys::zx_info_handle_basic_t { koid, rights, type_, related_koid, .. } = info;
774
775 HandleBasicInfo {
778 koid: Koid::from_raw(koid),
779 rights: Rights::from_bits_truncate(rights),
780 object_type: ObjectType::from_raw(type_),
781 related_koid: Koid::from_raw(related_koid),
782 }
783 }
784}
785
786struct HandleBasicInfoQuery;
789unsafe impl ObjectQuery for HandleBasicInfoQuery {
790 const TOPIC: Topic = Topic::HANDLE_BASIC;
791 type InfoTy = sys::zx_info_handle_basic_t;
792}
793
794sys::zx_info_handle_count_t!(HandleCountInfo);
795
796impl From<sys::zx_info_handle_count_t> for HandleCountInfo {
797 fn from(sys::zx_info_handle_count_t { handle_count }: sys::zx_info_handle_count_t) -> Self {
798 HandleCountInfo { handle_count }
799 }
800}
801
802struct HandleCountInfoQuery;
805unsafe impl ObjectQuery for HandleCountInfoQuery {
806 const TOPIC: Topic = Topic::HANDLE_COUNT;
807 type InfoTy = sys::zx_info_handle_count_t;
808}
809
810#[derive(Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
812pub enum HandleOp<'a> {
813 Move(NullableHandle),
814 Duplicate(HandleRef<'a>),
815}
816
817#[derive(Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
819#[repr(C)]
820pub struct HandleDisposition<'a> {
821 operation: sys::zx_handle_op_t,
823 handle: sys::zx_handle_t,
825 _handle_lifetime: std::marker::PhantomData<&'a ()>,
827
828 pub object_type: ObjectType,
829 pub rights: Rights,
830 pub result: Status,
831}
832
833static_assertions::assert_eq_size!(HandleDisposition<'_>, sys::zx_handle_disposition_t);
834static_assertions::const_assert_eq!(
835 std::mem::offset_of!(HandleDisposition<'_>, operation),
836 std::mem::offset_of!(sys::zx_handle_disposition_t, operation)
837);
838static_assertions::const_assert_eq!(
839 std::mem::offset_of!(HandleDisposition<'_>, handle),
840 std::mem::offset_of!(sys::zx_handle_disposition_t, handle)
841);
842static_assertions::const_assert_eq!(
843 std::mem::offset_of!(HandleDisposition<'_>, object_type),
844 std::mem::offset_of!(sys::zx_handle_disposition_t, type_)
845);
846static_assertions::const_assert_eq!(
847 std::mem::offset_of!(HandleDisposition<'_>, rights),
848 std::mem::offset_of!(sys::zx_handle_disposition_t, rights)
849);
850static_assertions::const_assert_eq!(
851 std::mem::offset_of!(HandleDisposition<'_>, result),
852 std::mem::offset_of!(sys::zx_handle_disposition_t, result)
853);
854
855impl<'a> HandleDisposition<'a> {
856 #[inline]
857 pub fn new(
858 handle_op: HandleOp<'a>,
859 object_type: ObjectType,
860 rights: Rights,
861 status: Status,
862 ) -> Self {
863 let (operation, handle) = match handle_op {
864 HandleOp::Move(h) => (sys::ZX_HANDLE_OP_MOVE, h.into_raw()),
865 HandleOp::Duplicate(h) => (sys::ZX_HANDLE_OP_DUPLICATE, h.raw_handle()),
866 };
867
868 Self {
869 operation,
870 handle,
871 _handle_lifetime: std::marker::PhantomData,
872 object_type,
873 rights: rights,
874 result: status,
875 }
876 }
877
878 pub fn raw_handle(&self) -> sys::zx_handle_t {
879 self.handle
880 }
881
882 pub fn is_move(&self) -> bool {
883 self.operation == sys::ZX_HANDLE_OP_MOVE
884 }
885
886 pub fn is_duplicate(&self) -> bool {
887 self.operation == sys::ZX_HANDLE_OP_DUPLICATE
888 }
889
890 pub fn take_op(&mut self) -> HandleOp<'a> {
891 match self.operation {
892 sys::ZX_HANDLE_OP_MOVE => {
893 HandleOp::Move(unsafe {
896 NullableHandle::from_raw(std::mem::replace(
897 &mut self.handle,
898 sys::ZX_HANDLE_INVALID,
899 ))
900 })
901 }
902 sys::ZX_HANDLE_OP_DUPLICATE => {
903 HandleOp::Duplicate(Unowned {
906 inner: ManuallyDrop::new(unsafe { NullableHandle::from_raw(self.handle) }),
907 marker: PhantomData,
908 })
909 }
910 _ => unreachable!(),
911 }
912 }
913
914 pub fn into_raw(mut self) -> sys::zx_handle_disposition_t {
915 match self.take_op() {
916 HandleOp::Move(mut handle) => sys::zx_handle_disposition_t {
917 operation: sys::ZX_HANDLE_OP_MOVE,
918 handle: std::mem::replace(&mut handle, NullableHandle::invalid()).into_raw(),
919 type_: self.object_type.into_raw(),
920 rights: self.rights.bits(),
921 result: self.result.into_raw(),
922 },
923 HandleOp::Duplicate(handle_ref) => sys::zx_handle_disposition_t {
924 operation: sys::ZX_HANDLE_OP_DUPLICATE,
925 handle: handle_ref.raw_handle(),
926 type_: self.object_type.into_raw(),
927 rights: self.rights.bits(),
928 result: self.result.into_raw(),
929 },
930 }
931 }
932}
933
934impl<'a> Drop for HandleDisposition<'a> {
935 fn drop(&mut self) {
936 if self.operation == sys::ZX_HANDLE_OP_MOVE {
938 unsafe { drop(NullableHandle::from_raw(self.handle)) };
939 }
940 }
941}
942
943#[derive(Debug, Eq, PartialEq, Ord, PartialOrd, Hash)]
947#[repr(C)]
948pub struct HandleInfo {
949 pub handle: NullableHandle,
950 pub object_type: ObjectType,
951 pub rights: Rights,
952
953 pub(crate) _unused: u32,
955}
956
957static_assertions::assert_eq_size!(HandleInfo, sys::zx_handle_info_t);
958static_assertions::const_assert_eq!(
959 std::mem::offset_of!(HandleInfo, handle),
960 std::mem::offset_of!(sys::zx_handle_info_t, handle)
961);
962static_assertions::const_assert_eq!(
963 std::mem::offset_of!(HandleInfo, object_type),
964 std::mem::offset_of!(sys::zx_handle_info_t, ty)
965);
966static_assertions::const_assert_eq!(
967 std::mem::offset_of!(HandleInfo, rights),
968 std::mem::offset_of!(sys::zx_handle_info_t, rights)
969);
970static_assertions::const_assert_eq!(
971 std::mem::offset_of!(HandleInfo, _unused),
972 std::mem::offset_of!(sys::zx_handle_info_t, unused)
973);
974
975impl HandleInfo {
976 pub const fn new(handle: NullableHandle, object_type: ObjectType, rights: Rights) -> Self {
978 Self { handle, object_type, rights, _unused: 0 }
979 }
980
981 pub const unsafe fn from_raw(raw: sys::zx_handle_info_t) -> HandleInfo {
989 HandleInfo::new(
990 unsafe { NullableHandle::from_raw(raw.handle) },
992 ObjectType::from_raw(raw.ty),
993 Rights::from_bits_retain(raw.rights),
994 )
995 }
996}
997
998#[cfg(test)]
999mod tests {
1000 use super::*;
1001 use zx::{
1004 Channel, HandleBased, HandleDisposition, HandleInfo, HandleOp, Name, NullableHandle,
1005 ObjectType, Rights, Vmo,
1006 };
1007 use zx_sys as sys;
1008
1009 #[test]
1010 fn into_raw() {
1011 let vmo = Vmo::create(1).unwrap();
1012 let h = vmo.into_raw();
1013 let vmo2 = Vmo::from(unsafe { NullableHandle::from_raw(h) });
1014 assert!(vmo2.write(b"1", 0).is_ok());
1015 }
1016
1017 #[test]
1018 fn check_raw_valid() {
1019 assert!(Handle::check_raw_valid(sys::ZX_HANDLE_INVALID).is_err());
1020 let vmo = Vmo::create(1).unwrap();
1021 let vmo_raw = vmo.raw_handle();
1022 assert!(Handle::check_raw_valid(vmo_raw).is_ok());
1023 drop(vmo);
1024 assert!(Handle::check_raw_valid(vmo_raw).is_err());
1025 }
1026
1027 #[test]
1029 fn duplicate() {
1030 let hello_length: usize = 5;
1031
1032 let vmo = Vmo::create(hello_length as u64).unwrap();
1034 assert!(vmo.write(b"hello", 0).is_ok());
1035
1036 let readonly_vmo = vmo.duplicate_handle(Rights::READ).unwrap();
1038 let mut read_vec = vec![0; hello_length];
1040 assert!(readonly_vmo.read(&mut read_vec, 0).is_ok());
1041 assert_eq!(read_vec, b"hello");
1042 assert_eq!(readonly_vmo.write(b"", 0), Err(Status::ACCESS_DENIED));
1043
1044 assert!(vmo.write(b"bye", 0).is_ok());
1046 assert!(readonly_vmo.read(&mut read_vec, 0).is_ok());
1047 assert_eq!(read_vec, b"byelo");
1048 }
1049
1050 #[test]
1052 fn replace() {
1053 let hello_length: usize = 5;
1054
1055 let vmo = Vmo::create(hello_length as u64).unwrap();
1057 assert!(vmo.write(b"hello", 0).is_ok());
1058
1059 let readonly_vmo = vmo.replace_handle(Rights::READ).unwrap();
1061 let mut read_vec = vec![0; hello_length];
1063 assert!(readonly_vmo.read(&mut read_vec, 0).is_ok());
1064 assert_eq!(read_vec, b"hello");
1065 assert_eq!(readonly_vmo.write(b"", 0), Err(Status::ACCESS_DENIED));
1066 }
1067
1068 #[test]
1069 fn set_get_name() {
1070 let vmo = Vmo::create(1).unwrap();
1072 let short_name = Name::new("v").unwrap();
1073 assert!(vmo.set_name(&short_name).is_ok());
1074 assert_eq!(vmo.get_name().unwrap(), short_name);
1075 }
1076
1077 #[test]
1078 fn set_get_max_len_name() {
1079 let vmo = Vmo::create(1).unwrap();
1080 let max_len_name = Name::new("a_great_maximum_length_vmo_name").unwrap(); assert!(vmo.set_name(&max_len_name).is_ok());
1082 assert_eq!(vmo.get_name().unwrap(), max_len_name);
1083 }
1084
1085 #[test]
1086 fn basic_info_channel() {
1087 let (side1, side2) = Channel::create();
1088 let info1 = side1.basic_info().expect("side1 basic_info failed");
1089 let info2 = side2.basic_info().expect("side2 basic_info failed");
1090
1091 assert_eq!(info1.koid, info2.related_koid);
1092 assert_eq!(info2.koid, info1.related_koid);
1093
1094 for info in &[info1, info2] {
1095 assert!(info.koid.raw_koid() >= sys::ZX_KOID_FIRST);
1096 assert_eq!(info.object_type, ObjectType::CHANNEL);
1097 assert!(info.rights.contains(Rights::READ | Rights::WRITE | Rights::WAIT));
1098 }
1099
1100 let side1_repl = side1.replace_handle(Rights::READ).expect("side1 replace_handle failed");
1101 let info1_repl = side1_repl.basic_info().expect("side1_repl basic_info failed");
1102 assert_eq!(info1_repl.koid, info1.koid);
1103 assert_eq!(info1_repl.rights, Rights::READ);
1104 }
1105
1106 #[test]
1107 fn basic_info_vmar() {
1108 let root_vmar = fuchsia_runtime::vmar_root_self();
1110 let info = root_vmar.basic_info().expect("vmar basic_info failed");
1111 assert_eq!(info.object_type, ObjectType::VMAR);
1112 assert!(!info.rights.contains(Rights::WAIT));
1113 }
1114
1115 #[test]
1116 fn count_info() {
1117 let vmo0 = Vmo::create(1).unwrap();
1118 let count_info = vmo0.count_info().expect("vmo0 count_info failed");
1119 assert_eq!(count_info.handle_count, 1);
1120
1121 let vmo1 = vmo0.duplicate_handle(Rights::SAME_RIGHTS).expect("vmo duplicate_handle failed");
1122 let count_info = vmo1.count_info().expect("vmo1 count_info failed");
1123 assert_eq!(count_info.handle_count, 2);
1124 }
1125
1126 #[test]
1127 fn raw_handle_disposition() {
1128 const RAW_HANDLE: sys::zx_handle_t = 1;
1129 let hd = HandleDisposition::new(
1130 HandleOp::Move(unsafe { NullableHandle::from_raw(RAW_HANDLE) }),
1131 ObjectType::VMO,
1132 Rights::EXECUTE,
1133 Status::OK,
1134 );
1135 let raw_hd = hd.into_raw();
1136 assert_eq!(raw_hd.operation, sys::ZX_HANDLE_OP_MOVE);
1137 assert_eq!(raw_hd.handle, RAW_HANDLE);
1138 assert_eq!(raw_hd.rights, sys::ZX_RIGHT_EXECUTE);
1139 assert_eq!(raw_hd.type_, sys::ZX_OBJ_TYPE_VMO);
1140 assert_eq!(raw_hd.result, sys::ZX_OK);
1141 }
1142
1143 #[test]
1144 fn regression_nullable_handle_into_raw_recursion() {
1145 let h = NullableHandle::invalid();
1146 assert_eq!(h.into_raw(), sys::ZX_HANDLE_INVALID);
1148
1149 let vmo = Vmo::create(1).unwrap();
1150 let raw = vmo.raw_handle();
1151 let h = vmo.into_handle();
1152 assert_eq!(h.into_raw(), raw);
1154 }
1155
1156 #[test]
1157 fn handle_info_from_raw() {
1158 const RAW_HANDLE: sys::zx_handle_t = 1;
1159 let raw_hi = sys::zx_handle_info_t {
1160 handle: RAW_HANDLE,
1161 ty: sys::ZX_OBJ_TYPE_VMO,
1162 rights: sys::ZX_RIGHT_EXECUTE,
1163 unused: 128,
1164 };
1165 let hi = unsafe { HandleInfo::from_raw(raw_hi) };
1166 assert_eq!(hi.handle.into_raw(), RAW_HANDLE);
1167 assert_eq!(hi.object_type, ObjectType::VMO);
1168 assert_eq!(hi.rights, Rights::EXECUTE);
1169 }
1170
1171 #[test]
1172 fn basic_peer_closed() {
1173 let (lhs, rhs) = crate::EventPair::create();
1174 assert!(!lhs.is_closed().unwrap());
1175 assert!(!rhs.is_closed().unwrap());
1176 drop(rhs);
1177 assert!(lhs.is_closed().unwrap());
1178 }
1179}