Skip to content

Commit adaae45

Browse files
committedMay 21, 2013
auto merge of #6604 : bjz/rust/ptr-to-option, r=brson
2 parents 6c0a469 + eef03c3 commit adaae45

File tree

1 file changed

+51
-0
lines changed

1 file changed

+51
-0
lines changed
 

Diff for: ‎src/libcore/ptr.rs

+51
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
use cast;
1414
use libc;
1515
use libc::{c_void, size_t};
16+
use option::{Option, Some, None};
1617
use sys;
1718

1819
#[cfg(not(test))] use cmp::{Eq, Ord};
@@ -209,6 +210,7 @@ pub unsafe fn array_each<T>(arr: **T, cb: &fn(*T)) {
209210
pub trait Ptr<T> {
210211
fn is_null(&const self) -> bool;
211212
fn is_not_null(&const self) -> bool;
213+
unsafe fn to_option(&const self) -> Option<&T>;
212214
fn offset(&self, count: uint) -> Self;
213215
}
214216

@@ -222,6 +224,23 @@ impl<T> Ptr<T> for *T {
222224
#[inline(always)]
223225
fn is_not_null(&const self) -> bool { is_not_null(*self) }
224226

227+
///
228+
/// Returns `None` if the pointer is null, or else returns the value wrapped
229+
/// in `Some`.
230+
///
231+
/// # Safety Notes
232+
///
233+
/// While this method is useful for null-safety, it is important to note
234+
/// that this is still an unsafe operation because the returned value could
235+
/// be pointing to invalid memory.
236+
///
237+
#[inline(always)]
238+
unsafe fn to_option(&const self) -> Option<&T> {
239+
if self.is_null() { None } else {
240+
Some(cast::transmute(*self))
241+
}
242+
}
243+
225244
/// Calculates the offset from a pointer.
226245
#[inline(always)]
227246
fn offset(&self, count: uint) -> *T { offset(*self, count) }
@@ -237,6 +256,23 @@ impl<T> Ptr<T> for *mut T {
237256
#[inline(always)]
238257
fn is_not_null(&const self) -> bool { is_not_null(*self) }
239258

259+
///
260+
/// Returns `None` if the pointer is null, or else returns the value wrapped
261+
/// in `Some`.
262+
///
263+
/// # Safety Notes
264+
///
265+
/// While this method is useful for null-safety, it is important to note
266+
/// that this is still an unsafe operation because the returned value could
267+
/// be pointing to invalid memory.
268+
///
269+
#[inline(always)]
270+
unsafe fn to_option(&const self) -> Option<&T> {
271+
if self.is_null() { None } else {
272+
Some(cast::transmute(*self))
273+
}
274+
}
275+
240276
/// Calculates the offset from a mutable pointer.
241277
#[inline(always)]
242278
fn offset(&self, count: uint) -> *mut T { mut_offset(*self, count) }
@@ -423,6 +459,21 @@ pub mod ptr_tests {
423459
assert!(mq.is_not_null());
424460
}
425461

462+
#[test]
463+
fn test_to_option() {
464+
let p: *int = null();
465+
// FIXME (#6641): Usage of unsafe methods in safe code doesn't cause an error.
466+
assert_eq!(p.to_option(), None);
467+
468+
let q: *int = &2;
469+
assert_eq!(q.to_option().unwrap(), &2); // FIXME (#6641)
470+
471+
let p: *mut int = mut_null();
472+
assert_eq!(p.to_option(), None); // FIXME (#6641)
473+
474+
let q: *mut int = &mut 2;
475+
assert_eq!(q.to_option().unwrap(), &2); // FIXME (#6641)
476+
}
426477

427478
#[test]
428479
fn test_ptr_array_each_with_len() {

0 commit comments

Comments
 (0)
Please sign in to comment.