src/libstd/ffi/c_str.rs

   1 // Copyright 2012 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 use ascii;
  12 use borrow::{Cow, Borrow};
  13 use cmp::Ordering;
  14 use error::Error;
  15 use fmt::{self, Write};
  16 use io;
  17 use libc;
  18 use mem;
  19 use memchr;
  20 use ops;
  21 use os::raw::c_char;
  22 use ptr;
  23 use slice;
  24 use str::{self, Utf8Error};
  25
  26 /// A type representing an owned C-compatible string
  27 ///
  28 /// This type serves the primary purpose of being able to safely generate a
  29 /// C-compatible string from a Rust byte slice or vector. An instance of this
  30 /// type is a static guarantee that the underlying bytes contain no interior 0
  31 /// bytes and the final byte is 0.
  32 ///
  33 /// A `CString` is created from either a byte slice or a byte vector. After
  34 /// being created, a `CString` predominately inherits all of its methods from
  35 /// the `Deref` implementation to `[c_char]`. Note that the underlying array
  36 /// is represented as an array of `c_char` as opposed to `u8`. A `u8` slice
  37 /// can be obtained with the `as_bytes` method.  Slices produced from a `CString`
  38 /// do *not* contain the trailing nul terminator unless otherwise specified.
  39 ///
  40 /// # Examples
  41 ///
  42 /// ```no_run
  43 /// # fn main() {
  44 /// use std::ffi::CString;
  45 /// use std::os::raw::c_char;
  46 ///
  47 /// extern {
  48 ///     fn my_printer(s: *const c_char);
  49 /// }
  50 ///
  51 /// let c_to_print = CString::new("Hello, world!").unwrap();
  52 /// unsafe {
  53 ///     my_printer(c_to_print.as_ptr());
  54 /// }
  55 /// # }
  56 /// ```
  57 ///
  58 /// # Safety
  59 ///
  60 /// `CString` is intended for working with traditional C-style strings
  61 /// (a sequence of non-null bytes terminated by a single null byte); the
  62 /// primary use case for these kinds of strings is interoperating with C-like
  63 /// code. Often you will need to transfer ownership to/from that external
  64 /// code. It is strongly recommended that you thoroughly read through the
  65 /// documentation of `CString` before use, as improper ownership management
  66 /// of `CString` instances can lead to invalid memory accesses, memory leaks,
  67 /// and other memory errors.
  68
  69 #[derive(PartialEq, PartialOrd, Eq, Ord, Hash, Clone)]
  70 #[stable(feature = "rust1", since = "1.0.0")]
  71 pub struct CString {
  72     // Invariant 1: the slice ends with a zero byte and has a length of at least one.
  73     // Invariant 2: the slice contains only one zero byte.
  74     // Improper usage of unsafe function can break Invariant 2, but not Invariant 1.
  75     inner: Box<[u8]>,
  76 }
  77
  78 /// Representation of a borrowed C string.
  79 ///
  80 /// This dynamically sized type is only safely constructed via a borrowed
  81 /// version of an instance of `CString`. This type can be constructed from a raw
  82 /// C string as well and represents a C string borrowed from another location.
  83 ///
  84 /// Note that this structure is **not** `repr(C)` and is not recommended to be
  85 /// placed in the signatures of FFI functions. Instead safe wrappers of FFI
  86 /// functions may leverage the unsafe `from_ptr` constructor to provide a safe
  87 /// interface to other consumers.
  88 ///
  89 /// # Examples
  90 ///
  91 /// Inspecting a foreign C string
  92 ///
  93 /// ```no_run
  94 /// use std::ffi::CStr;
  95 /// use std::os::raw::c_char;
  96 ///
  97 /// extern { fn my_string() -> *const c_char; }
  98 ///
  99 /// unsafe {
 100 ///     let slice = CStr::from_ptr(my_string());
 101 ///     println!("string length: {}", slice.to_bytes().len());
 102 /// }
 103 /// ```
 104 ///
 105 /// Passing a Rust-originating C string
 106 ///
 107 /// ```no_run
 108 /// use std::ffi::{CString, CStr};
 109 /// use std::os::raw::c_char;
 110 ///
 111 /// fn work(data: &CStr) {
 112 ///     extern { fn work_with(data: *const c_char); }
 113 ///
 114 ///     unsafe { work_with(data.as_ptr()) }
 115 /// }
 116 ///
 117 /// let s = CString::new("data data data data").unwrap();
 118 /// work(&s);
 119 /// ```
 120 ///
 121 /// Converting a foreign C string into a Rust `String`
 122 ///
 123 /// ```no_run
 124 /// use std::ffi::CStr;
 125 /// use std::os::raw::c_char;
 126 ///
 127 /// extern { fn my_string() -> *const c_char; }
 128 ///
 129 /// fn my_string_safe() -> String {
 130 ///     unsafe {
 131 ///         CStr::from_ptr(my_string()).to_string_lossy().into_owned()
 132 ///     }
 133 /// }
 134 ///
 135 /// println!("string: {}", my_string_safe());
 136 /// ```
 137 #[derive(Hash)]
 138 #[stable(feature = "rust1", since = "1.0.0")]
 139 pub struct CStr {
 140     // FIXME: this should not be represented with a DST slice but rather with
 141     //        just a raw `c_char` along with some form of marker to make
 142     //        this an unsized type. Essentially `sizeof(&CStr)` should be the
 143     //        same as `sizeof(&c_char)` but `CStr` should be an unsized type.
 144     inner: [c_char]
 145 }
 146
 147 /// An error returned from `CString::new` to indicate that a nul byte was found
 148 /// in the vector provided.
 149 #[derive(Clone, PartialEq, Eq, Debug)]
 150 #[stable(feature = "rust1", since = "1.0.0")]
 151 pub struct NulError(usize, Vec<u8>);
 152
 153 /// An error returned from `CStr::from_bytes_with_nul` to indicate that a nul
 154 /// byte was found too early in the slice provided or one wasn't found at all.
 155 #[derive(Clone, PartialEq, Eq, Debug)]
 156 #[stable(feature = "cstr_from_bytes", since = "1.10.0")]
 157 pub struct FromBytesWithNulError {
 158     kind: FromBytesWithNulErrorKind,
 159 }
 160
 161 #[derive(Clone, PartialEq, Eq, Debug)]
 162 enum FromBytesWithNulErrorKind {
 163     InteriorNul(usize),
 164     NotNulTerminated,
 165 }
 166
 167 impl FromBytesWithNulError {
 168     fn interior_nul(pos: usize) -> FromBytesWithNulError {
 169         FromBytesWithNulError {
 170             kind: FromBytesWithNulErrorKind::InteriorNul(pos),
 171         }
 172     }
 173     fn not_nul_terminated() -> FromBytesWithNulError {
 174         FromBytesWithNulError {
 175             kind: FromBytesWithNulErrorKind::NotNulTerminated,
 176         }
 177     }
 178 }
 179
 180 /// An error returned from `CString::into_string` to indicate that a UTF-8 error
 181 /// was encountered during the conversion.
 182 #[derive(Clone, PartialEq, Eq, Debug)]
 183 #[stable(feature = "cstring_into", since = "1.7.0")]
 184 pub struct IntoStringError {
 185     inner: CString,
 186     error: Utf8Error,
 187 }
 188
 189 impl CString {
 190     /// Creates a new C-compatible string from a container of bytes.
 191     ///
 192     /// This method will consume the provided data and use the underlying bytes
 193     /// to construct a new string, ensuring that there is a trailing 0 byte.
 194     ///
 195     /// # Examples
 196     ///
 197     /// ```no_run
 198     /// use std::ffi::CString;
 199     /// use std::os::raw::c_char;
 200     ///
 201     /// extern { fn puts(s: *const c_char); }
 202     ///
 203     /// let to_print = CString::new("Hello!").unwrap();
 204     /// unsafe {
 205     ///     puts(to_print.as_ptr());
 206     /// }
 207     /// ```
 208     ///
 209     /// # Errors
 210     ///
 211     /// This function will return an error if the bytes yielded contain an
 212     /// internal 0 byte. The error returned will contain the bytes as well as
 213     /// the position of the nul byte.
 214     #[stable(feature = "rust1", since = "1.0.0")]
 215     pub fn new<T: Into<Vec<u8>>>(t: T) -> Result<CString, NulError> {
 216         Self::_new(t.into())
 217     }
 218
 219     fn _new(bytes: Vec<u8>) -> Result<CString, NulError> {
 220         match memchr::memchr(0, &bytes) {
 221             Some(i) => Err(NulError(i, bytes)),
 222             None => Ok(unsafe { CString::from_vec_unchecked(bytes) }),
 223         }
 224     }
 225
 226     /// Creates a C-compatible string from a byte vector without checking for
 227     /// interior 0 bytes.
 228     ///
 229     /// This method is equivalent to `new` except that no runtime assertion
 230     /// is made that `v` contains no 0 bytes, and it requires an actual
 231     /// byte vector, not anything that can be converted to one with Into.
 232     ///
 233     /// # Examples
 234     ///
 235     /// ```
 236     /// use std::ffi::CString;
 237     ///
 238     /// let raw = b"foo".to_vec();
 239     /// unsafe {
 240     ///     let c_string = CString::from_vec_unchecked(raw);
 241     /// }
 242     /// ```
 243     #[stable(feature = "rust1", since = "1.0.0")]
 244     pub unsafe fn from_vec_unchecked(mut v: Vec<u8>) -> CString {
 245         v.reserve_exact(1);
 246         v.push(0);
 247         CString { inner: v.into_boxed_slice() }
 248     }
 249
 250     /// Retakes ownership of a `CString` that was transferred to C.
 251     ///
 252     /// Additionally, the length of the string will be recalculated from the pointer.
 253     ///
 254     /// # Safety
 255     ///
 256     /// This should only ever be called with a pointer that was earlier
 257     /// obtained by calling `into_raw` on a `CString`. Other usage (e.g. trying to take
 258     /// ownership of a string that was allocated by foreign code) is likely to lead
 259     /// to undefined behavior or allocator corruption.
 260     #[stable(feature = "cstr_memory", since = "1.4.0")]
 261     pub unsafe fn from_raw(ptr: *mut c_char) -> CString {
 262         let len = libc::strlen(ptr) + 1; // Including the NUL byte
 263         let slice = slice::from_raw_parts(ptr, len as usize);
 264         CString { inner: mem::transmute(slice) }
 265     }
 266
 267     /// Transfers ownership of the string to a C caller.
 268     ///
 269     /// The pointer must be returned to Rust and reconstituted using
 270     /// `from_raw` to be properly deallocated. Specifically, one
 271     /// should *not* use the standard C `free` function to deallocate
 272     /// this string.
 273     ///
 274     /// Failure to call `from_raw` will lead to a memory leak.
 275     #[stable(feature = "cstr_memory", since = "1.4.0")]
 276     pub fn into_raw(self) -> *mut c_char {
 277         Box::into_raw(self.into_inner()) as *mut c_char
 278     }
 279
 280     /// Converts the `CString` into a `String` if it contains valid Unicode data.
 281     ///
 282     /// On failure, ownership of the original `CString` is returned.
 283     #[stable(feature = "cstring_into", since = "1.7.0")]
 284     pub fn into_string(self) -> Result<String, IntoStringError> {
 285         String::from_utf8(self.into_bytes())
 286             .map_err(|e| IntoStringError {
 287                 error: e.utf8_error(),
 288                 inner: unsafe { CString::from_vec_unchecked(e.into_bytes()) },
 289             })
 290     }
 291
 292     /// Returns the underlying byte buffer.
 293     ///
 294     /// The returned buffer does **not** contain the trailing nul separator and
 295     /// it is guaranteed to not have any interior nul bytes.
 296     #[stable(feature = "cstring_into", since = "1.7.0")]
 297     pub fn into_bytes(self) -> Vec<u8> {
 298         let mut vec = self.into_inner().into_vec();
 299         let _nul = vec.pop();
 300         debug_assert_eq!(_nul, Some(0u8));
 301         vec
 302     }
 303
 304     /// Equivalent to the `into_bytes` function except that the returned vector
 305     /// includes the trailing nul byte.
 306     #[stable(feature = "cstring_into", since = "1.7.0")]
 307     pub fn into_bytes_with_nul(self) -> Vec<u8> {
 308         self.into_inner().into_vec()
 309     }
 310
 311     /// Returns the contents of this `CString` as a slice of bytes.
 312     ///
 313     /// The returned slice does **not** contain the trailing nul separator and
 314     /// it is guaranteed to not have any interior nul bytes.
 315     #[stable(feature = "rust1", since = "1.0.0")]
 316     pub fn as_bytes(&self) -> &[u8] {
 317         &self.inner[..self.inner.len() - 1]
 318     }
 319
 320     /// Equivalent to the `as_bytes` function except that the returned slice
 321     /// includes the trailing nul byte.
 322     #[stable(feature = "rust1", since = "1.0.0")]
 323     pub fn as_bytes_with_nul(&self) -> &[u8] {
 324         &self.inner
 325     }
 326
 327     /// Converts this `CString` into a boxed `CStr`.
 328     #[unstable(feature = "into_boxed_c_str", issue = "40380")]
 329     pub fn into_boxed_c_str(self) -> Box<CStr> {
 330         unsafe { mem::transmute(self.into_inner()) }
 331     }
 332
 333     // Bypass "move out of struct which implements `Drop` trait" restriction.
 334     fn into_inner(self) -> Box<[u8]> {
 335         unsafe {
 336             let result = ptr::read(&self.inner);
 337             mem::forget(self);
 338             result
 339         }
 340     }
 341 }
 342
 343 // Turns this `CString` into an empty string to prevent
 344 // memory unsafe code from working by accident. Inline
 345 // to prevent LLVM from optimizing it away in debug builds.
 346 #[stable(feature = "cstring_drop", since = "1.13.0")]
 347 impl Drop for CString {
 348     #[inline]
 349     fn drop(&mut self) {
 350         unsafe { *self.inner.get_unchecked_mut(0) = 0; }
 351     }
 352 }
 353
 354 #[stable(feature = "rust1", since = "1.0.0")]
 355 impl ops::Deref for CString {
 356     type Target = CStr;
 357
 358     fn deref(&self) -> &CStr {
 359         unsafe { CStr::from_bytes_with_nul_unchecked(self.as_bytes_with_nul()) }
 360     }
 361 }
 362
 363 #[stable(feature = "rust1", since = "1.0.0")]
 364 impl fmt::Debug for CString {
 365     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
 366         fmt::Debug::fmt(&**self, f)
 367     }
 368 }
 369
 370 #[stable(feature = "cstring_into", since = "1.7.0")]
 371 impl From<CString> for Vec<u8> {
 372     fn from(s: CString) -> Vec<u8> {
 373         s.into_bytes()
 374     }
 375 }
 376
 377 #[stable(feature = "cstr_debug", since = "1.3.0")]
 378 impl fmt::Debug for CStr {
 379     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
 380         write!(f, "\"")?;
 381         for byte in self.to_bytes().iter().flat_map(|&b| ascii::escape_default(b)) {
 382             f.write_char(byte as char)?;
 383         }
 384         write!(f, "\"")
 385     }
 386 }
 387
 388 #[stable(feature = "cstr_default", since = "1.10.0")]
 389 impl<'a> Default for &'a CStr {
 390     fn default() -> &'a CStr {
 391         static SLICE: &'static [c_char] = &[0];
 392         unsafe { CStr::from_ptr(SLICE.as_ptr()) }
 393     }
 394 }
 395
 396 #[stable(feature = "cstr_default", since = "1.10.0")]
 397 impl Default for CString {
 398     /// Creates an empty `CString`.
 399     fn default() -> CString {
 400         let a: &CStr = Default::default();
 401         a.to_owned()
 402     }
 403 }
 404
 405 #[stable(feature = "cstr_borrow", since = "1.3.0")]
 406 impl Borrow<CStr> for CString {
 407     fn borrow(&self) -> &CStr { self }
 408 }
 409
 410 #[stable(feature = "box_from_c_str", since = "1.17.0")]
 411 impl<'a> From<&'a CStr> for Box<CStr> {
 412     fn from(s: &'a CStr) -> Box<CStr> {
 413         let boxed: Box<[u8]> = Box::from(s.to_bytes_with_nul());
 414         unsafe { mem::transmute(boxed) }
 415     }
 416 }
 417
 418 #[stable(feature = "c_string_from_box", since = "1.17.0")]
 419 impl From<Box<CStr>> for CString {
 420     fn from(s: Box<CStr>) -> CString {
 421         s.into_c_string()
 422     }
 423 }
 424
 425 #[stable(feature = "box_from_c_string", since = "1.17.0")]
 426 impl Into<Box<CStr>> for CString {
 427     fn into(self) -> Box<CStr> {
 428         self.into_boxed_c_str()
 429     }
 430 }
 431
 432 #[stable(feature = "default_box_extra", since = "1.17.0")]
 433 impl Default for Box<CStr> {
 434     fn default() -> Box<CStr> {
 435         let boxed: Box<[u8]> = Box::from([0]);
 436         unsafe { mem::transmute(boxed) }
 437     }
 438 }
 439
 440 impl NulError {
 441     /// Returns the position of the nul byte in the slice that was provided to
 442     /// `CString::new`.
 443     ///
 444     /// # Examples
 445     ///
 446     /// ```
 447     /// use std::ffi::CString;
 448     ///
 449     /// let nul_error = CString::new("foo\0bar").unwrap_err();
 450     /// assert_eq!(nul_error.nul_position(), 3);
 451     ///
 452     /// let nul_error = CString::new("foo bar\0").unwrap_err();
 453     /// assert_eq!(nul_error.nul_position(), 7);
 454     /// ```
 455     #[stable(feature = "rust1", since = "1.0.0")]
 456     pub fn nul_position(&self) -> usize { self.0 }
 457
 458     /// Consumes this error, returning the underlying vector of bytes which
 459     /// generated the error in the first place.
 460     ///
 461     /// # Examples
 462     ///
 463     /// ```
 464     /// use std::ffi::CString;
 465     ///
 466     /// let nul_error = CString::new("foo\0bar").unwrap_err();
 467     /// assert_eq!(nul_error.into_vec(), b"foo\0bar");
 468     /// ```
 469     #[stable(feature = "rust1", since = "1.0.0")]
 470     pub fn into_vec(self) -> Vec<u8> { self.1 }
 471 }
 472
 473 #[stable(feature = "rust1", since = "1.0.0")]
 474 impl Error for NulError {
 475     fn description(&self) -> &str { "nul byte found in data" }
 476 }
 477
 478 #[stable(feature = "rust1", since = "1.0.0")]
 479 impl fmt::Display for NulError {
 480     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
 481         write!(f, "nul byte found in provided data at position: {}", self.0)
 482     }
 483 }
 484
 485 #[stable(feature = "rust1", since = "1.0.0")]
 486 impl From<NulError> for io::Error {
 487     fn from(_: NulError) -> io::Error {
 488         io::Error::new(io::ErrorKind::InvalidInput,
 489                        "data provided contains a nul byte")
 490     }
 491 }
 492
 493 #[stable(feature = "frombyteswithnulerror_impls", since = "1.17.0")]
 494 impl Error for FromBytesWithNulError {
 495     fn description(&self) -> &str {
 496         match self.kind {
 497             FromBytesWithNulErrorKind::InteriorNul(..) =>
 498                 "data provided contains an interior nul byte",
 499             FromBytesWithNulErrorKind::NotNulTerminated =>
 500                 "data provided is not nul terminated",
 501         }
 502     }
 503 }
 504
 505 #[stable(feature = "frombyteswithnulerror_impls", since = "1.17.0")]
 506 impl fmt::Display for FromBytesWithNulError {
 507     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
 508         f.write_str(self.description())?;
 509         if let FromBytesWithNulErrorKind::InteriorNul(pos) = self.kind {
 510             write!(f, " at byte pos {}", pos)?;
 511         }
 512         Ok(())
 513     }
 514 }
 515
 516 impl IntoStringError {
 517     /// Consumes this error, returning original `CString` which generated the
 518     /// error.
 519     #[stable(feature = "cstring_into", since = "1.7.0")]
 520     pub fn into_cstring(self) -> CString {
 521         self.inner
 522     }
 523
 524     /// Access the underlying UTF-8 error that was the cause of this error.
 525     #[stable(feature = "cstring_into", since = "1.7.0")]
 526     pub fn utf8_error(&self) -> Utf8Error {
 527         self.error
 528     }
 529 }
 530
 531 #[stable(feature = "cstring_into", since = "1.7.0")]
 532 impl Error for IntoStringError {
 533     fn description(&self) -> &str {
 534         "C string contained non-utf8 bytes"
 535     }
 536
 537     fn cause(&self) -> Option<&Error> {
 538         Some(&self.error)
 539     }
 540 }
 541
 542 #[stable(feature = "cstring_into", since = "1.7.0")]
 543 impl fmt::Display for IntoStringError {
 544     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
 545         self.description().fmt(f)
 546     }
 547 }
 548
 549 impl CStr {
 550     /// Casts a raw C string to a safe C string wrapper.
 551     ///
 552     /// This function will cast the provided `ptr` to the `CStr` wrapper which
 553     /// allows inspection and interoperation of non-owned C strings. This method
 554     /// is unsafe for a number of reasons:
 555     ///
 556     /// * There is no guarantee to the validity of `ptr`
 557     /// * The returned lifetime is not guaranteed to be the actual lifetime of
 558     ///   `ptr`
 559     /// * There is no guarantee that the memory pointed to by `ptr` contains a
 560     ///   valid nul terminator byte at the end of the string.
 561     ///
 562     /// > **Note**: This operation is intended to be a 0-cost cast but it is
 563     /// > currently implemented with an up-front calculation of the length of
 564     /// > the string. This is not guaranteed to always be the case.
 565     ///
 566     /// # Examples
 567     ///
 568     /// ```no_run
 569     /// # fn main() {
 570     /// use std::ffi::CStr;
 571     /// use std::os::raw::c_char;
 572     ///
 573     /// extern {
 574     ///     fn my_string() -> *const c_char;
 575     /// }
 576     ///
 577     /// unsafe {
 578     ///     let slice = CStr::from_ptr(my_string());
 579     ///     println!("string returned: {}", slice.to_str().unwrap());
 580     /// }
 581     /// # }
 582     /// ```
 583     #[stable(feature = "rust1", since = "1.0.0")]
 584     pub unsafe fn from_ptr<'a>(ptr: *const c_char) -> &'a CStr {
 585         let len = libc::strlen(ptr);
 586         let ptr = ptr as *const u8;
 587         CStr::from_bytes_with_nul_unchecked(slice::from_raw_parts(ptr, len as usize + 1))
 588     }
 589
 590     /// Creates a C string wrapper from a byte slice.
 591     ///
 592     /// This function will cast the provided `bytes` to a `CStr` wrapper after
 593     /// ensuring that it is null terminated and does not contain any interior
 594     /// nul bytes.
 595     ///
 596     /// # Examples
 597     ///
 598     /// ```
 599     /// use std::ffi::CStr;
 600     ///
 601     /// let cstr = CStr::from_bytes_with_nul(b"hello\0");
 602     /// assert!(cstr.is_ok());
 603     /// ```
 604     #[stable(feature = "cstr_from_bytes", since = "1.10.0")]
 605     pub fn from_bytes_with_nul(bytes: &[u8])
 606                                -> Result<&CStr, FromBytesWithNulError> {
 607         let nul_pos = memchr::memchr(0, bytes);
 608         if let Some(nul_pos) = nul_pos {
 609             if nul_pos + 1 != bytes.len() {
 610                 return Err(FromBytesWithNulError::interior_nul(nul_pos));
 611             }
 612             Ok(unsafe { CStr::from_bytes_with_nul_unchecked(bytes) })
 613         } else {
 614             Err(FromBytesWithNulError::not_nul_terminated())
 615         }
 616     }
 617
 618     /// Unsafely creates a C string wrapper from a byte slice.
 619     ///
 620     /// This function will cast the provided `bytes` to a `CStr` wrapper without
 621     /// performing any sanity checks. The provided slice must be null terminated
 622     /// and not contain any interior nul bytes.
 623     ///
 624     /// # Examples
 625     ///
 626     /// ```
 627     /// use std::ffi::{CStr, CString};
 628     ///
 629     /// unsafe {
 630     ///     let cstring = CString::new("hello").unwrap();
 631     ///     let cstr = CStr::from_bytes_with_nul_unchecked(cstring.to_bytes_with_nul());
 632     ///     assert_eq!(cstr, &*cstring);
 633     /// }
 634     /// ```
 635     #[stable(feature = "cstr_from_bytes", since = "1.10.0")]
 636     pub unsafe fn from_bytes_with_nul_unchecked(bytes: &[u8]) -> &CStr {
 637         mem::transmute(bytes)
 638     }
 639
 640     /// Returns the inner pointer to this C string.
 641     ///
 642     /// The returned pointer will be valid for as long as `self` is and points
 643     /// to a contiguous region of memory terminated with a 0 byte to represent
 644     /// the end of the string.
 645     ///
 646     /// **WARNING**
 647     ///
 648     /// It is your responsibility to make sure that the underlying memory is not
 649     /// freed too early. For example, the following code will cause undefined
 650     /// behaviour when `ptr` is used inside the `unsafe` block:
 651     ///
 652     /// ```no_run
 653     /// use std::ffi::{CString};
 654     ///
 655     /// let ptr = CString::new("Hello").unwrap().as_ptr();
 656     /// unsafe {
 657     ///     // `ptr` is dangling
 658     ///     *ptr;
 659     /// }
 660     /// ```
 661     ///
 662     /// This happens because the pointer returned by `as_ptr` does not carry any
 663     /// lifetime information and the string is deallocated immediately after
 664     /// the `CString::new("Hello").unwrap().as_ptr()` expression is evaluated.
 665     /// To fix the problem, bind the string to a local variable:
 666     ///
 667     /// ```no_run
 668     /// use std::ffi::{CString};
 669     ///
 670     /// let hello = CString::new("Hello").unwrap();
 671     /// let ptr = hello.as_ptr();
 672     /// unsafe {
 673     ///     // `ptr` is valid because `hello` is in scope
 674     ///     *ptr;
 675     /// }
 676     /// ```
 677     #[stable(feature = "rust1", since = "1.0.0")]
 678     pub fn as_ptr(&self) -> *const c_char {
 679         self.inner.as_ptr()
 680     }
 681
 682     /// Converts this C string to a byte slice.
 683     ///
 684     /// This function will calculate the length of this string (which normally
 685     /// requires a linear amount of work to be done) and then return the
 686     /// resulting slice of `u8` elements.
 687     ///
 688     /// The returned slice will **not** contain the trailing nul that this C
 689     /// string has.
 690     ///
 691     /// > **Note**: This method is currently implemented as a 0-cost cast, but
 692     /// > it is planned to alter its definition in the future to perform the
 693     /// > length calculation whenever this method is called.
 694     #[stable(feature = "rust1", since = "1.0.0")]
 695     pub fn to_bytes(&self) -> &[u8] {
 696         let bytes = self.to_bytes_with_nul();
 697         &bytes[..bytes.len() - 1]
 698     }
 699
 700     /// Converts this C string to a byte slice containing the trailing 0 byte.
 701     ///
 702     /// This function is the equivalent of `to_bytes` except that it will retain
 703     /// the trailing nul instead of chopping it off.
 704     ///
 705     /// > **Note**: This method is currently implemented as a 0-cost cast, but
 706     /// > it is planned to alter its definition in the future to perform the
 707     /// > length calculation whenever this method is called.
 708     #[stable(feature = "rust1", since = "1.0.0")]
 709     pub fn to_bytes_with_nul(&self) -> &[u8] {
 710         unsafe { mem::transmute(&self.inner) }
 711     }
 712
 713     /// Yields a `&str` slice if the `CStr` contains valid UTF-8.
 714     ///
 715     /// This function will calculate the length of this string and check for
 716     /// UTF-8 validity, and then return the `&str` if it's valid.
 717     ///
 718     /// > **Note**: This method is currently implemented to check for validity
 719     /// > after a 0-cost cast, but it is planned to alter its definition in the
 720     /// > future to perform the length calculation in addition to the UTF-8
 721     /// > check whenever this method is called.
 722     #[stable(feature = "cstr_to_str", since = "1.4.0")]
 723     pub fn to_str(&self) -> Result<&str, str::Utf8Error> {
 724         // NB: When CStr is changed to perform the length check in .to_bytes()
 725         // instead of in from_ptr(), it may be worth considering if this should
 726         // be rewritten to do the UTF-8 check inline with the length calculation
 727         // instead of doing it afterwards.
 728         str::from_utf8(self.to_bytes())
 729     }
 730
 731     /// Converts a `CStr` into a `Cow<str>`.
 732     ///
 733     /// This function will calculate the length of this string (which normally
 734     /// requires a linear amount of work to be done) and then return the
 735     /// resulting slice as a `Cow<str>`, replacing any invalid UTF-8 sequences
 736     /// with `U+FFFD REPLACEMENT CHARACTER`.
 737     ///
 738     /// > **Note**: This method is currently implemented to check for validity
 739     /// > after a 0-cost cast, but it is planned to alter its definition in the
 740     /// > future to perform the length calculation in addition to the UTF-8
 741     /// > check whenever this method is called.
 742     #[stable(feature = "cstr_to_str", since = "1.4.0")]
 743     pub fn to_string_lossy(&self) -> Cow<str> {
 744         String::from_utf8_lossy(self.to_bytes())
 745     }
 746
 747     /// Converts a `Box<CStr>` into a `CString` without copying or allocating.
 748     #[unstable(feature = "into_boxed_c_str", issue = "40380")]
 749     pub fn into_c_string(self: Box<CStr>) -> CString {
 750         unsafe { mem::transmute(self) }
 751     }
 752 }
 753
 754 #[stable(feature = "rust1", since = "1.0.0")]
 755 impl PartialEq for CStr {
 756     fn eq(&self, other: &CStr) -> bool {
 757         self.to_bytes().eq(other.to_bytes())
 758     }
 759 }
 760 #[stable(feature = "rust1", since = "1.0.0")]
 761 impl Eq for CStr {}
 762 #[stable(feature = "rust1", since = "1.0.0")]
 763 impl PartialOrd for CStr {
 764     fn partial_cmp(&self, other: &CStr) -> Option<Ordering> {
 765         self.to_bytes().partial_cmp(&other.to_bytes())
 766     }
 767 }
 768 #[stable(feature = "rust1", since = "1.0.0")]
 769 impl Ord for CStr {
 770     fn cmp(&self, other: &CStr) -> Ordering {
 771         self.to_bytes().cmp(&other.to_bytes())
 772     }
 773 }
 774
 775 #[stable(feature = "cstr_borrow", since = "1.3.0")]
 776 impl ToOwned for CStr {
 777     type Owned = CString;
 778
 779     fn to_owned(&self) -> CString {
 780         CString { inner: self.to_bytes_with_nul().into() }
 781     }
 782 }
 783
 784 #[stable(feature = "cstring_asref", since = "1.7.0")]
 785 impl<'a> From<&'a CStr> for CString {
 786     fn from(s: &'a CStr) -> CString {
 787         s.to_owned()
 788     }
 789 }
 790
 791 #[stable(feature = "cstring_asref", since = "1.7.0")]
 792 impl ops::Index<ops::RangeFull> for CString {
 793     type Output = CStr;
 794
 795     #[inline]
 796     fn index(&self, _index: ops::RangeFull) -> &CStr {
 797         self
 798     }
 799 }
 800
 801 #[stable(feature = "cstring_asref", since = "1.7.0")]
 802 impl AsRef<CStr> for CStr {
 803     fn as_ref(&self) -> &CStr {
 804         self
 805     }
 806 }
 807
 808 #[stable(feature = "cstring_asref", since = "1.7.0")]
 809 impl AsRef<CStr> for CString {
 810     fn as_ref(&self) -> &CStr {
 811         self
 812     }
 813 }
 814
 815 #[cfg(test)]
 816 mod tests {
 817     use super::*;
 818     use os::raw::c_char;
 819     use borrow::Cow::{Borrowed, Owned};
 820     use hash::{Hash, Hasher};
 821     use collections::hash_map::DefaultHasher;
 822
 823     #[test]
 824     fn c_to_rust() {
 825         let data = b"123\0";
 826         let ptr = data.as_ptr() as *const c_char;
 827         unsafe {
 828             assert_eq!(CStr::from_ptr(ptr).to_bytes(), b"123");
 829             assert_eq!(CStr::from_ptr(ptr).to_bytes_with_nul(), b"123\0");
 830         }
 831     }
 832
 833     #[test]
 834     fn simple() {
 835         let s = CString::new("1234").unwrap();
 836         assert_eq!(s.as_bytes(), b"1234");
 837         assert_eq!(s.as_bytes_with_nul(), b"1234\0");
 838     }
 839
 840     #[test]
 841     fn build_with_zero1() {
 842         assert!(CString::new(&b"\0"[..]).is_err());
 843     }
 844     #[test]
 845     fn build_with_zero2() {
 846         assert!(CString::new(vec![0]).is_err());
 847     }
 848
 849     #[test]
 850     fn build_with_zero3() {
 851         unsafe {
 852             let s = CString::from_vec_unchecked(vec![0]);
 853             assert_eq!(s.as_bytes(), b"\0");
 854         }
 855     }
 856
 857     #[test]
 858     fn formatted() {
 859         let s = CString::new(&b"abc\x01\x02\n\xE2\x80\xA6\xFF"[..]).unwrap();
 860         assert_eq!(format!("{:?}", s), r#""abc\x01\x02\n\xe2\x80\xa6\xff""#);
 861     }
 862
 863     #[test]
 864     fn borrowed() {
 865         unsafe {
 866             let s = CStr::from_ptr(b"12\0".as_ptr() as *const _);
 867             assert_eq!(s.to_bytes(), b"12");
 868             assert_eq!(s.to_bytes_with_nul(), b"12\0");
 869         }
 870     }
 871
 872     #[test]
 873     fn to_str() {
 874         let data = b"123\xE2\x80\xA6\0";
 875         let ptr = data.as_ptr() as *const c_char;
 876         unsafe {
 877             assert_eq!(CStr::from_ptr(ptr).to_str(), Ok("123…"));
 878             assert_eq!(CStr::from_ptr(ptr).to_string_lossy(), Borrowed("123…"));
 879         }
 880         let data = b"123\xE2\0";
 881         let ptr = data.as_ptr() as *const c_char;
 882         unsafe {
 883             assert!(CStr::from_ptr(ptr).to_str().is_err());
 884             assert_eq!(CStr::from_ptr(ptr).to_string_lossy(), Owned::<str>(format!("123\u{FFFD}")));
 885         }
 886     }
 887
 888     #[test]
 889     fn to_owned() {
 890         let data = b"123\0";
 891         let ptr = data.as_ptr() as *const c_char;
 892
 893         let owned = unsafe { CStr::from_ptr(ptr).to_owned() };
 894         assert_eq!(owned.as_bytes_with_nul(), data);
 895     }
 896
 897     #[test]
 898     fn equal_hash() {
 899         let data = b"123\xE2\xFA\xA6\0";
 900         let ptr = data.as_ptr() as *const c_char;
 901         let cstr: &'static CStr = unsafe { CStr::from_ptr(ptr) };
 902
 903         let mut s = DefaultHasher::new();
 904         cstr.hash(&mut s);
 905         let cstr_hash = s.finish();
 906         let mut s = DefaultHasher::new();
 907         CString::new(&data[..data.len() - 1]).unwrap().hash(&mut s);
 908         let cstring_hash = s.finish();
 909
 910         assert_eq!(cstr_hash, cstring_hash);
 911     }
 912
 913     #[test]
 914     fn from_bytes_with_nul() {
 915         let data = b"123\0";
 916         let cstr = CStr::from_bytes_with_nul(data);
 917         assert_eq!(cstr.map(CStr::to_bytes), Ok(&b"123"[..]));
 918         let cstr = CStr::from_bytes_with_nul(data);
 919         assert_eq!(cstr.map(CStr::to_bytes_with_nul), Ok(&b"123\0"[..]));
 920
 921         unsafe {
 922             let cstr = CStr::from_bytes_with_nul(data);
 923             let cstr_unchecked = CStr::from_bytes_with_nul_unchecked(data);
 924             assert_eq!(cstr, Ok(cstr_unchecked));
 925         }
 926     }
 927
 928     #[test]
 929     fn from_bytes_with_nul_unterminated() {
 930         let data = b"123";
 931         let cstr = CStr::from_bytes_with_nul(data);
 932         assert!(cstr.is_err());
 933     }
 934
 935     #[test]
 936     fn from_bytes_with_nul_interior() {
 937         let data = b"1\023\0";
 938         let cstr = CStr::from_bytes_with_nul(data);
 939         assert!(cstr.is_err());
 940     }
 941
 942     #[test]
 943     fn into_boxed() {
 944         let orig: &[u8] = b"Hello, world!\0";
 945         let cstr = CStr::from_bytes_with_nul(orig).unwrap();
 946         let boxed: Box<CStr> = Box::from(cstr);
 947         let cstring = cstr.to_owned().into_boxed_c_str().into_c_string();
 948         assert_eq!(cstr, &*boxed);
 949         assert_eq!(&*boxed, &*cstring);
 950         assert_eq!(&*cstring, cstr);
 951     }
 952
 953     #[test]
 954     fn boxed_default() {
 955         let boxed = <Box<CStr>>::default();
 956         assert_eq!(boxed.to_bytes_with_nul(), &[0]);
 957     }
 958 }