core/ffi/c_str.rs
1//! [`CStr`] and its related types.
2
3use crate::cmp::Ordering;
4use crate::error::Error;
5use crate::ffi::c_char;
6use crate::intrinsics::const_eval_select;
7use crate::iter::FusedIterator;
8use crate::marker::PhantomData;
9use crate::ptr::NonNull;
10use crate::slice::memchr;
11use crate::{fmt, ops, range, slice, str};
12
13// FIXME: because this is doc(inline)d, we *have* to use intra-doc links because the actual link
14// depends on where the item is being documented. however, since this is libcore, we can't
15// actually reference libstd or liballoc in intra-doc links. so, the best we can do is remove the
16// links to `CString` and `String` for now until a solution is developed
17
18/// A dynamically-sized view of a C string.
19///
20/// The type `&CStr` represents a reference to a borrowed nul-terminated
21/// array of bytes. It can be constructed safely from a <code>&[[u8]]</code>
22/// slice, or unsafely from a raw `*const c_char`. It can be expressed as a
23/// literal in the form `c"Hello world"`.
24///
25/// The `&CStr` can then be converted to a Rust <code>&[str]</code> by performing
26/// UTF-8 validation, or into an owned `CString`.
27///
28/// `&CStr` is to `CString` as <code>&[str]</code> is to `String`: the former
29/// in each pair are borrowing references; the latter are owned
30/// strings.
31///
32/// Note that this structure does **not** have a guaranteed layout (the `repr(transparent)`
33/// notwithstanding) and should not be placed in the signatures of FFI functions.
34/// Instead, safe wrappers of FFI functions may leverage [`CStr::as_ptr`] and the unsafe
35/// [`CStr::from_ptr`] constructor to provide a safe interface to other consumers.
36///
37/// # Examples
38///
39/// Inspecting a foreign C string:
40///
41/// ```
42/// use std::ffi::CStr;
43/// use std::os::raw::c_char;
44///
45/// # /* Extern functions are awkward in doc comments - fake it instead
46/// extern "C" { fn my_string() -> *const c_char; }
47/// # */ unsafe extern "C" fn my_string() -> *const c_char { c"hello".as_ptr() }
48///
49/// unsafe {
50/// let slice = CStr::from_ptr(my_string());
51/// println!("string buffer size without nul terminator: {}", slice.to_bytes().len());
52/// }
53/// ```
54///
55/// Passing a Rust-originating C string:
56///
57/// ```
58/// use std::ffi::CStr;
59/// use std::os::raw::c_char;
60///
61/// fn work(data: &CStr) {
62/// unsafe extern "C" fn work_with(s: *const c_char) {}
63/// unsafe { work_with(data.as_ptr()) }
64/// }
65///
66/// let s = c"Hello world!";
67/// work(&s);
68/// ```
69///
70/// Converting a foreign C string into a Rust `String`:
71///
72/// ```
73/// use std::ffi::CStr;
74/// use std::os::raw::c_char;
75///
76/// # /* Extern functions are awkward in doc comments - fake it instead
77/// extern "C" { fn my_string() -> *const c_char; }
78/// # */ unsafe extern "C" fn my_string() -> *const c_char { c"hello".as_ptr() }
79///
80/// fn my_string_safe() -> String {
81/// let cstr = unsafe { CStr::from_ptr(my_string()) };
82/// // Get a copy-on-write Cow<'_, str>, then extract the
83/// // allocated String (or allocate a fresh one if needed).
84/// cstr.to_string_lossy().into_owned()
85/// }
86///
87/// println!("string: {}", my_string_safe());
88/// ```
89///
90/// [str]: prim@str "str"
91#[derive(PartialEq, Eq, Hash)]
92#[stable(feature = "core_c_str", since = "1.64.0")]
93#[rustc_diagnostic_item = "cstr_type"]
94#[rustc_has_incoherent_inherent_impls]
95#[lang = "CStr"]
96// `fn from` in `impl From<&CStr> for Box<CStr>` current implementation relies
97// on `CStr` being layout-compatible with `[u8]`.
98// However, `CStr` layout is considered an implementation detail and must not be relied upon. We
99// want `repr(transparent)` but we don't want it to show up in rustdoc, so we hide it under
100// `cfg(doc)`. This is an ad-hoc implementation of attribute privacy.
101#[repr(transparent)]
102pub struct CStr {
103 // FIXME: this should not be represented with a DST slice but rather with
104 // just a raw `c_char` along with some form of marker to make
105 // this an unsized type. Essentially `sizeof(&CStr)` should be the
106 // same as `sizeof(&c_char)` but `CStr` should be an unsized type.
107 inner: [c_char],
108}
109
110/// An error indicating that a nul byte was not in the expected position.
111///
112/// The slice used to create a [`CStr`] must have one and only one nul byte,
113/// positioned at the end.
114///
115/// This error is created by the [`CStr::from_bytes_with_nul`] method.
116/// See its documentation for more.
117///
118/// # Examples
119///
120/// ```
121/// use std::ffi::{CStr, FromBytesWithNulError};
122///
123/// let _: FromBytesWithNulError = CStr::from_bytes_with_nul(b"f\0oo").unwrap_err();
124/// ```
125#[derive(Clone, Copy, PartialEq, Eq, Debug)]
126#[stable(feature = "core_c_str", since = "1.64.0")]
127pub enum FromBytesWithNulError {
128 /// Data provided contains an interior nul byte at byte `position`.
129 InteriorNul {
130 /// The position of the interior nul byte.
131 position: usize,
132 },
133 /// Data provided is not nul terminated.
134 NotNulTerminated,
135}
136
137#[stable(feature = "frombyteswithnulerror_impls", since = "1.17.0")]
138impl fmt::Display for FromBytesWithNulError {
139 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
140 match self {
141 Self::InteriorNul { position } => {
142 write!(f, "data provided contains an interior nul byte at byte position {position}")
143 }
144 Self::NotNulTerminated => write!(f, "data provided is not nul terminated"),
145 }
146 }
147}
148
149#[stable(feature = "frombyteswithnulerror_impls", since = "1.17.0")]
150impl Error for FromBytesWithNulError {}
151
152/// An error indicating that no nul byte was present.
153///
154/// A slice used to create a [`CStr`] must contain a nul byte somewhere
155/// within the slice.
156///
157/// This error is created by the [`CStr::from_bytes_until_nul`] method.
158#[derive(Clone, Copy, PartialEq, Eq, Debug)]
159#[stable(feature = "cstr_from_bytes_until_nul", since = "1.69.0")]
160pub struct FromBytesUntilNulError(());
161
162#[stable(feature = "cstr_from_bytes_until_nul", since = "1.69.0")]
163impl fmt::Display for FromBytesUntilNulError {
164 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
165 write!(f, "data provided does not contain a nul")
166 }
167}
168
169/// Shows the underlying bytes as a normal string, with invalid UTF-8
170/// presented as hex escape sequences.
171#[stable(feature = "cstr_debug", since = "1.3.0")]
172impl fmt::Debug for CStr {
173 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
174 fmt::Debug::fmt(crate::bstr::ByteStr::from_bytes(self.to_bytes()), f)
175 }
176}
177
178#[stable(feature = "cstr_default", since = "1.10.0")]
179#[rustc_const_unstable(feature = "const_default", issue = "143894")]
180const impl Default for &CStr {
181 #[inline]
182 fn default() -> Self {
183 c""
184 }
185}
186
187impl CStr {
188 /// Wraps a raw C string with a safe C string wrapper.
189 ///
190 /// This function will wrap the provided `ptr` with a `CStr` wrapper, which
191 /// allows inspection and interoperation of non-owned C strings. The total
192 /// size of the terminated buffer must be smaller than [`isize::MAX`] **bytes**
193 /// in memory (a restriction from [`slice::from_raw_parts`]).
194 ///
195 /// # Safety
196 ///
197 /// * The memory pointed to by `ptr` must contain a valid nul terminator at the
198 /// end of the string.
199 ///
200 /// * `ptr` must be [valid] for reads of bytes up to and including the nul terminator.
201 /// This means in particular:
202 ///
203 /// * The entire memory range of this `CStr` must be contained within a single allocation!
204 /// * `ptr` must be non-null even for a zero-length cstr.
205 ///
206 /// * The memory referenced by the returned `CStr` must not be mutated for
207 /// the duration of lifetime `'a`.
208 ///
209 /// * The nul terminator must be within `isize::MAX` from `ptr`
210 ///
211 /// > **Note**: This operation is intended to be a 0-cost cast but it is
212 /// > currently implemented with an up-front calculation of the length of
213 /// > the string. This is not guaranteed to always be the case.
214 ///
215 /// # Caveat
216 ///
217 /// The lifetime for the returned slice is inferred from its usage. To prevent accidental misuse,
218 /// it's suggested to tie the lifetime to whichever source lifetime is safe in the context,
219 /// such as by providing a helper function taking the lifetime of a host value for the slice,
220 /// or by explicit annotation.
221 ///
222 /// # Examples
223 ///
224 /// ```
225 /// use std::ffi::{c_char, CStr};
226 ///
227 /// fn my_string() -> *const c_char {
228 /// c"hello".as_ptr()
229 /// }
230 ///
231 /// unsafe {
232 /// let slice = CStr::from_ptr(my_string());
233 /// assert_eq!(slice.to_str().unwrap(), "hello");
234 /// }
235 /// ```
236 ///
237 /// ```
238 /// use std::ffi::{c_char, CStr};
239 ///
240 /// const HELLO_PTR: *const c_char = {
241 /// const BYTES: &[u8] = b"Hello, world!\0";
242 /// BYTES.as_ptr().cast()
243 /// };
244 /// const HELLO: &CStr = unsafe { CStr::from_ptr(HELLO_PTR) };
245 ///
246 /// assert_eq!(c"Hello, world!", HELLO);
247 /// ```
248 ///
249 /// [valid]: core::ptr#safety
250 #[inline] // inline is necessary for codegen to see strlen.
251 #[must_use]
252 #[stable(feature = "rust1", since = "1.0.0")]
253 #[rustc_const_stable(feature = "const_cstr_from_ptr", since = "1.81.0")]
254 pub const unsafe fn from_ptr<'a>(ptr: *const c_char) -> &'a CStr {
255 // SAFETY: The caller has provided a pointer that points to a valid C
256 // string with a NUL terminator less than `isize::MAX` from `ptr`.
257 let len = unsafe { strlen(ptr) };
258
259 // SAFETY: The caller has provided a valid pointer with length less than
260 // `isize::MAX`, so `from_raw_parts` is safe. The content remains valid
261 // and doesn't change for the lifetime of the returned `CStr`. This
262 // means the call to `from_bytes_with_nul_unchecked` is correct.
263 //
264 // The cast from c_char to u8 is ok because a c_char is always one byte.
265 unsafe { Self::from_bytes_with_nul_unchecked(slice::from_raw_parts(ptr.cast(), len + 1)) }
266 }
267
268 /// Creates a C string wrapper from a byte slice with any number of nuls.
269 ///
270 /// This method will create a `CStr` from any byte slice that contains at
271 /// least one nul byte. Unlike with [`CStr::from_bytes_with_nul`], the caller
272 /// does not need to know where the nul byte is located.
273 ///
274 /// If the first byte is a nul character, this method will return an
275 /// empty `CStr`. If multiple nul characters are present, the `CStr` will
276 /// end at the first one.
277 ///
278 /// If the slice only has a single nul byte at the end, this method is
279 /// equivalent to [`CStr::from_bytes_with_nul`].
280 ///
281 /// # Examples
282 /// ```
283 /// use std::ffi::CStr;
284 ///
285 /// let mut buffer = [0u8; 16];
286 /// unsafe {
287 /// // Here we might call an unsafe C function that writes a string
288 /// // into the buffer.
289 /// let buf_ptr = buffer.as_mut_ptr();
290 /// buf_ptr.write_bytes(b'A', 8);
291 /// }
292 /// // Attempt to extract a C nul-terminated string from the buffer.
293 /// let c_str = CStr::from_bytes_until_nul(&buffer[..]).unwrap();
294 /// assert_eq!(c_str.to_str().unwrap(), "AAAAAAAA");
295 /// ```
296 ///
297 #[stable(feature = "cstr_from_bytes_until_nul", since = "1.69.0")]
298 #[rustc_const_stable(feature = "cstr_from_bytes_until_nul", since = "1.69.0")]
299 pub const fn from_bytes_until_nul(bytes: &[u8]) -> Result<&CStr, FromBytesUntilNulError> {
300 let nul_pos = memchr::memchr(0, bytes);
301 match nul_pos {
302 Some(nul_pos) => {
303 // FIXME(const-hack) replace with range index
304 // SAFETY: nul_pos + 1 <= bytes.len()
305 let subslice = unsafe { crate::slice::from_raw_parts(bytes.as_ptr(), nul_pos + 1) };
306 // SAFETY: We know there is a nul byte at nul_pos, so this slice
307 // (ending at the nul byte) is a well-formed C string.
308 Ok(unsafe { CStr::from_bytes_with_nul_unchecked(subslice) })
309 }
310 None => Err(FromBytesUntilNulError(())),
311 }
312 }
313
314 /// Creates a C string wrapper from a byte slice with exactly one nul
315 /// terminator.
316 ///
317 /// This function will cast the provided `bytes` to a `CStr`
318 /// wrapper after ensuring that the byte slice is nul-terminated
319 /// and does not contain any interior nul bytes.
320 ///
321 /// If the nul byte may not be at the end,
322 /// [`CStr::from_bytes_until_nul`] can be used instead.
323 ///
324 /// # Examples
325 ///
326 /// ```
327 /// use std::ffi::CStr;
328 ///
329 /// let cstr = CStr::from_bytes_with_nul(b"hello\0");
330 /// assert_eq!(cstr, Ok(c"hello"));
331 /// ```
332 ///
333 /// Creating a `CStr` without a trailing nul terminator is an error:
334 ///
335 /// ```
336 /// use std::ffi::{CStr, FromBytesWithNulError};
337 ///
338 /// let cstr = CStr::from_bytes_with_nul(b"hello");
339 /// assert_eq!(cstr, Err(FromBytesWithNulError::NotNulTerminated));
340 /// ```
341 ///
342 /// Creating a `CStr` with an interior nul byte is an error:
343 ///
344 /// ```
345 /// use std::ffi::{CStr, FromBytesWithNulError};
346 ///
347 /// let cstr = CStr::from_bytes_with_nul(b"he\0llo\0");
348 /// assert_eq!(cstr, Err(FromBytesWithNulError::InteriorNul { position: 2 }));
349 /// ```
350 #[stable(feature = "cstr_from_bytes", since = "1.10.0")]
351 #[rustc_const_stable(feature = "const_cstr_methods", since = "1.72.0")]
352 pub const fn from_bytes_with_nul(bytes: &[u8]) -> Result<&Self, FromBytesWithNulError> {
353 let nul_pos = memchr::memchr(0, bytes);
354 match nul_pos {
355 Some(nul_pos) if nul_pos + 1 == bytes.len() => {
356 // SAFETY: We know there is only one nul byte, at the end
357 // of the byte slice.
358 Ok(unsafe { Self::from_bytes_with_nul_unchecked(bytes) })
359 }
360 Some(position) => Err(FromBytesWithNulError::InteriorNul { position }),
361 None => Err(FromBytesWithNulError::NotNulTerminated),
362 }
363 }
364
365 /// Unsafely creates a C string wrapper from a byte slice.
366 ///
367 /// This function will cast the provided `bytes` to a `CStr` wrapper without
368 /// performing any sanity checks.
369 ///
370 /// # Safety
371 /// The provided slice **must** be nul-terminated and not contain any interior
372 /// nul bytes.
373 ///
374 /// # Examples
375 ///
376 /// ```
377 /// use std::ffi::CStr;
378 ///
379 /// let bytes = b"Hello world!\0";
380 ///
381 /// let cstr = unsafe { CStr::from_bytes_with_nul_unchecked(bytes) };
382 /// assert_eq!(cstr.to_bytes_with_nul(), bytes);
383 /// ```
384 #[inline]
385 #[must_use]
386 #[stable(feature = "cstr_from_bytes", since = "1.10.0")]
387 #[rustc_const_stable(feature = "const_cstr_unchecked", since = "1.59.0")]
388 #[rustc_allow_const_fn_unstable(const_eval_select)]
389 pub const unsafe fn from_bytes_with_nul_unchecked(bytes: &[u8]) -> &CStr {
390 const_eval_select!(
391 @capture { bytes: &[u8] } -> &CStr:
392 if const {
393 // Saturating so that an empty slice panics in the assert with a good
394 // message, not here due to underflow.
395 let mut i = bytes.len().saturating_sub(1);
396 assert!(!bytes.is_empty() && bytes[i] == 0, "input was not nul-terminated");
397
398 // Ending nul byte exists, skip to the rest.
399 while i != 0 {
400 i -= 1;
401 let byte = bytes[i];
402 assert!(byte != 0, "input contained interior nul");
403 }
404
405 // SAFETY: See runtime cast comment below.
406 unsafe { &*(bytes as *const [u8] as *const CStr) }
407 } else {
408 // Chance at catching some UB at runtime with debug builds.
409 debug_assert!(!bytes.is_empty() && bytes[bytes.len() - 1] == 0);
410
411 // SAFETY: Casting to CStr is safe because its internal representation
412 // is a [u8] too (safe only inside std).
413 // Dereferencing the obtained pointer is safe because it comes from a
414 // reference. Making a reference is then safe because its lifetime
415 // is bound by the lifetime of the given `bytes`.
416 unsafe { &*(bytes as *const [u8] as *const CStr) }
417 }
418 )
419 }
420
421 /// Returns the inner pointer to this C string.
422 ///
423 /// The returned pointer will be valid for as long as `self` is, and points
424 /// to a contiguous region of memory terminated with a 0 byte to represent
425 /// the end of the string.
426 ///
427 /// The type of the returned pointer is
428 /// [`*const c_char`][crate::ffi::c_char], and whether it's
429 /// an alias for `*const i8` or `*const u8` is platform-specific.
430 ///
431 /// **WARNING**
432 ///
433 /// The returned pointer is read-only; writing to it (including passing it
434 /// to C code that writes to it) causes undefined behavior.
435 ///
436 /// It is your responsibility to make sure that the underlying memory is not
437 /// freed too early. For example, the following code will cause undefined
438 /// behavior when `ptr` is used inside the `unsafe` block:
439 ///
440 /// ```no_run
441 /// # #![expect(dangling_pointers_from_temporaries)]
442 /// use std::ffi::{CStr, CString};
443 ///
444 /// // 💀 The meaning of this entire program is undefined,
445 /// // 💀 and nothing about its behavior is guaranteed,
446 /// // 💀 not even that its behavior resembles the code as written,
447 /// // 💀 just because it contains a single instance of undefined behavior!
448 ///
449 /// // 🚨 creates a dangling pointer to a temporary `CString`
450 /// // 🚨 that is deallocated at the end of the statement
451 /// let ptr = CString::new("Hi!".to_uppercase()).unwrap().as_ptr();
452 ///
453 /// // without undefined behavior, you would expect that `ptr` equals:
454 /// dbg!(CStr::from_bytes_with_nul(b"HI!\0").unwrap());
455 ///
456 /// // 🙏 Possibly the program behaved as expected so far,
457 /// // 🙏 and this just shows `ptr` is now garbage..., but
458 /// // 💀 this violates `CStr::from_ptr`'s safety contract
459 /// // 💀 leading to a dereference of a dangling pointer,
460 /// // 💀 which is immediate undefined behavior.
461 /// // 💀 *BOOM*, you're dead, your entire program has no meaning.
462 /// dbg!(unsafe { CStr::from_ptr(ptr) });
463 /// ```
464 ///
465 /// This happens because, the pointer returned by `as_ptr` does not carry any
466 /// lifetime information, and the `CString` is deallocated immediately after
467 /// the expression that it is part of has been evaluated.
468 /// To fix the problem, bind the `CString` to a local variable:
469 ///
470 /// ```
471 /// use std::ffi::{CStr, CString};
472 ///
473 /// let c_str = CString::new("Hi!".to_uppercase()).unwrap();
474 /// let ptr = c_str.as_ptr();
475 ///
476 /// assert_eq!(unsafe { CStr::from_ptr(ptr) }, c"HI!");
477 /// ```
478 #[inline]
479 #[must_use]
480 #[stable(feature = "rust1", since = "1.0.0")]
481 #[rustc_const_stable(feature = "const_str_as_ptr", since = "1.32.0")]
482 #[rustc_as_ptr]
483 #[rustc_never_returns_null_ptr]
484 pub const fn as_ptr(&self) -> *const c_char {
485 self.inner.as_ptr()
486 }
487
488 /// We could eventually expose this publicly, if we wanted.
489 #[inline]
490 #[must_use]
491 const fn as_non_null_ptr(&self) -> NonNull<c_char> {
492 // FIXME(const_trait_impl) replace with `NonNull::from`
493 // SAFETY: a reference is never null
494 unsafe { NonNull::new_unchecked(&self.inner as *const [c_char] as *mut [c_char]) }
495 .as_non_null_ptr()
496 }
497
498 /// Returns the length of `self`. Like C's `strlen`, this does not include the nul terminator.
499 ///
500 /// > **Note**: This method is currently implemented as a constant-time
501 /// > cast, but it is planned to alter its definition in the future to
502 /// > perform the length calculation whenever this method is called.
503 ///
504 /// # Examples
505 ///
506 /// ```
507 /// assert_eq!(c"foo".count_bytes(), 3);
508 /// assert_eq!(c"".count_bytes(), 0);
509 /// ```
510 #[inline]
511 #[must_use]
512 #[doc(alias("len", "strlen"))]
513 #[stable(feature = "cstr_count_bytes", since = "1.79.0")]
514 #[rustc_const_stable(feature = "const_cstr_from_ptr", since = "1.81.0")]
515 pub const fn count_bytes(&self) -> usize {
516 self.inner.len() - 1
517 }
518
519 /// Returns `true` if `self.to_bytes()` has a length of 0.
520 ///
521 /// # Examples
522 ///
523 /// ```
524 /// assert!(!c"foo".is_empty());
525 /// assert!(c"".is_empty());
526 /// ```
527 #[inline]
528 #[stable(feature = "cstr_is_empty", since = "1.71.0")]
529 #[rustc_const_stable(feature = "cstr_is_empty", since = "1.71.0")]
530 pub const fn is_empty(&self) -> bool {
531 // SAFETY: We know there is at least one byte; for empty strings it
532 // is the NUL terminator.
533 // FIXME(const-hack): use get_unchecked
534 unsafe { *self.inner.as_ptr() == 0 }
535 }
536
537 /// Converts this C string to a byte slice.
538 ///
539 /// The returned slice will **not** contain the trailing nul terminator that this C
540 /// string has.
541 ///
542 /// > **Note**: This method is currently implemented as a constant-time
543 /// > cast, but it is planned to alter its definition in the future to
544 /// > perform the length calculation whenever this method is called.
545 ///
546 /// # Examples
547 ///
548 /// ```
549 /// assert_eq!(c"foo".to_bytes(), b"foo");
550 /// ```
551 #[inline]
552 #[must_use = "this returns the result of the operation, \
553 without modifying the original"]
554 #[stable(feature = "rust1", since = "1.0.0")]
555 #[rustc_const_stable(feature = "const_cstr_methods", since = "1.72.0")]
556 pub const fn to_bytes(&self) -> &[u8] {
557 let bytes = self.to_bytes_with_nul();
558 // FIXME(const-hack) replace with range index
559 // SAFETY: to_bytes_with_nul returns slice with length at least 1
560 unsafe { slice::from_raw_parts(bytes.as_ptr(), bytes.len() - 1) }
561 }
562
563 /// Converts this C string to a byte slice containing the trailing 0 byte.
564 ///
565 /// This function is the equivalent of [`CStr::to_bytes`] except that it
566 /// will retain the trailing nul terminator instead of chopping it off.
567 ///
568 /// > **Note**: This method is currently implemented as a 0-cost cast, but
569 /// > it is planned to alter its definition in the future to perform the
570 /// > length calculation whenever this method is called.
571 ///
572 /// # Examples
573 ///
574 /// ```
575 /// assert_eq!(c"foo".to_bytes_with_nul(), b"foo\0");
576 /// ```
577 #[inline]
578 #[must_use = "this returns the result of the operation, \
579 without modifying the original"]
580 #[stable(feature = "rust1", since = "1.0.0")]
581 #[rustc_const_stable(feature = "const_cstr_methods", since = "1.72.0")]
582 pub const fn to_bytes_with_nul(&self) -> &[u8] {
583 // SAFETY: Transmuting a slice of `c_char`s to a slice of `u8`s
584 // is safe on all supported targets.
585 let bytes = unsafe { &*((&raw const self.inner) as *const [u8]) };
586
587 // SAFETY: A valid `CStr` always contains at least its trailing nul byte.
588 unsafe { crate::hint::assert_unchecked(!bytes.is_empty()) };
589
590 bytes
591 }
592
593 /// Iterates over the bytes in this C string.
594 ///
595 /// The returned iterator will **not** contain the trailing nul terminator
596 /// that this C string has.
597 ///
598 /// # Examples
599 ///
600 /// ```
601 /// #![feature(cstr_bytes)]
602 ///
603 /// assert!(c"foo".bytes().eq(*b"foo"));
604 /// ```
605 #[inline]
606 #[unstable(feature = "cstr_bytes", issue = "112115")]
607 pub fn bytes(&self) -> Bytes<'_> {
608 Bytes::new(self)
609 }
610
611 /// Yields a <code>&[str]</code> slice if the `CStr` contains valid UTF-8.
612 ///
613 /// If the contents of the `CStr` are valid UTF-8 data, this
614 /// function will return the corresponding <code>&[str]</code> slice. Otherwise,
615 /// it will return an error with details of where UTF-8 validation failed.
616 ///
617 /// [str]: prim@str "str"
618 ///
619 /// # Examples
620 ///
621 /// ```
622 /// assert_eq!(c"foo".to_str(), Ok("foo"));
623 /// ```
624 #[stable(feature = "cstr_to_str", since = "1.4.0")]
625 #[rustc_const_stable(feature = "const_cstr_methods", since = "1.72.0")]
626 pub const fn to_str(&self) -> Result<&str, str::Utf8Error> {
627 // N.B., when `CStr` is changed to perform the length check in `.to_bytes()`
628 // instead of in `from_ptr()`, it may be worth considering if this should
629 // be rewritten to do the UTF-8 check inline with the length calculation
630 // instead of doing it afterwards.
631 str::from_utf8(self.to_bytes())
632 }
633
634 /// Returns an object that implements [`Display`] for safely printing a [`CStr`] that may
635 /// contain non-Unicode data.
636 ///
637 /// Behaves as if `self` were first lossily converted to a `str`, with invalid UTF-8 presented
638 /// as the Unicode replacement character: �.
639 ///
640 /// [`Display`]: fmt::Display
641 ///
642 /// # Examples
643 ///
644 /// ```
645 /// #![feature(cstr_display)]
646 ///
647 /// let cstr = c"Hello, world!";
648 /// println!("{}", cstr.display());
649 /// ```
650 #[unstable(feature = "cstr_display", issue = "139984")]
651 #[must_use = "this does not display the `CStr`; \
652 it returns an object that can be displayed"]
653 #[inline]
654 pub fn display(&self) -> impl fmt::Display {
655 crate::bstr::ByteStr::from_bytes(self.to_bytes())
656 }
657
658 /// Returns the same string as a string slice `&CStr`.
659 ///
660 /// This method is redundant when used directly on `&CStr`, but
661 /// it helps dereferencing other string-like types to string slices,
662 /// for example references to `Box<CStr>` or `Arc<CStr>`.
663 #[inline]
664 #[unstable(feature = "str_as_str", issue = "130366")]
665 pub const fn as_c_str(&self) -> &CStr {
666 self
667 }
668}
669
670#[stable(feature = "c_string_eq_c_str", since = "1.90.0")]
671impl PartialEq<&Self> for CStr {
672 #[inline]
673 fn eq(&self, other: &&Self) -> bool {
674 *self == **other
675 }
676
677 #[inline]
678 fn ne(&self, other: &&Self) -> bool {
679 *self != **other
680 }
681}
682
683// `.to_bytes()` representations are compared instead of the inner `[c_char]`s,
684// because `c_char` is `i8` (not `u8`) on some platforms.
685// That is why this is implemented manually and not derived.
686#[stable(feature = "rust1", since = "1.0.0")]
687impl PartialOrd for CStr {
688 #[inline]
689 fn partial_cmp(&self, other: &CStr) -> Option<Ordering> {
690 self.to_bytes().partial_cmp(other.to_bytes())
691 }
692}
693
694#[stable(feature = "rust1", since = "1.0.0")]
695impl Ord for CStr {
696 #[inline]
697 fn cmp(&self, other: &CStr) -> Ordering {
698 self.to_bytes().cmp(other.to_bytes())
699 }
700}
701
702#[stable(feature = "cstr_range_from", since = "1.47.0")]
703impl ops::Index<ops::RangeFrom<usize>> for CStr {
704 type Output = CStr;
705
706 #[inline]
707 fn index(&self, index: ops::RangeFrom<usize>) -> &CStr {
708 let bytes = self.to_bytes_with_nul();
709 // we need to manually check the starting index to account for the null
710 // byte, since otherwise we could get an empty string that doesn't end
711 // in a null.
712 if index.start < bytes.len() {
713 // SAFETY: Non-empty tail of a valid `CStr` is still a valid `CStr`.
714 unsafe { CStr::from_bytes_with_nul_unchecked(&bytes[index.start..]) }
715 } else {
716 panic!(
717 "index out of bounds: the len is {} but the index is {}",
718 bytes.len(),
719 index.start
720 );
721 }
722 }
723}
724
725#[stable(feature = "new_range_from_api", since = "1.96.0")]
726impl ops::Index<range::RangeFrom<usize>> for CStr {
727 type Output = CStr;
728
729 #[inline]
730 fn index(&self, index: range::RangeFrom<usize>) -> &CStr {
731 ops::Index::index(self, ops::RangeFrom::from(index))
732 }
733}
734
735#[stable(feature = "cstring_asref", since = "1.7.0")]
736#[rustc_const_unstable(feature = "const_convert", issue = "143773")]
737const impl AsRef<CStr> for CStr {
738 #[inline]
739 fn as_ref(&self) -> &CStr {
740 self
741 }
742}
743
744/// Calculate the length of a nul-terminated string. Defers to C's `strlen` when possible.
745///
746/// # Safety
747///
748/// The pointer must point to a valid buffer that contains a NUL terminator. The NUL must be
749/// located within `isize::MAX` from `ptr`.
750#[inline]
751#[unstable(feature = "cstr_internals", issue = "none")]
752#[rustc_allow_const_fn_unstable(const_eval_select)]
753const unsafe fn strlen(ptr: *const c_char) -> usize {
754 const_eval_select!(
755 @capture { s: *const c_char = ptr } -> usize:
756 if const {
757 let mut len = 0;
758
759 // SAFETY: Outer caller has provided a pointer to a valid C string.
760 while unsafe { *s.add(len) } != 0 {
761 len += 1;
762 }
763
764 len
765 } else {
766 unsafe extern "C" {
767 /// Provided by libc or compiler_builtins.
768 fn strlen(s: *const c_char) -> usize;
769 }
770
771 // SAFETY: Outer caller has provided a pointer to a valid C string.
772 unsafe { strlen(s) }
773 }
774 )
775}
776
777/// An iterator over the bytes of a [`CStr`], without the nul terminator.
778///
779/// This struct is created by the [`bytes`] method on [`CStr`].
780/// See its documentation for more.
781///
782/// [`bytes`]: CStr::bytes
783#[must_use = "iterators are lazy and do nothing unless consumed"]
784#[unstable(feature = "cstr_bytes", issue = "112115")]
785#[derive(Clone, Debug)]
786pub struct Bytes<'a> {
787 // since we know the string is nul-terminated, we only need one pointer
788 ptr: NonNull<u8>,
789 phantom: PhantomData<&'a [c_char]>,
790}
791
792#[unstable(feature = "cstr_bytes", issue = "112115")]
793unsafe impl Send for Bytes<'_> {}
794
795#[unstable(feature = "cstr_bytes", issue = "112115")]
796unsafe impl Sync for Bytes<'_> {}
797
798impl<'a> Bytes<'a> {
799 #[inline]
800 fn new(s: &'a CStr) -> Self {
801 Self { ptr: s.as_non_null_ptr().cast(), phantom: PhantomData }
802 }
803
804 #[inline]
805 fn is_empty(&self) -> bool {
806 // SAFETY: We uphold that the pointer is always valid to dereference
807 // by starting with a valid C string and then never incrementing beyond
808 // the nul terminator.
809 unsafe { self.ptr.read() == 0 }
810 }
811}
812
813#[unstable(feature = "cstr_bytes", issue = "112115")]
814impl Iterator for Bytes<'_> {
815 type Item = u8;
816
817 #[inline]
818 fn next(&mut self) -> Option<u8> {
819 // SAFETY: We only choose a pointer from a valid C string, which must
820 // be non-null and contain at least one value. Since we always stop at
821 // the nul terminator, which is guaranteed to exist, we can assume that
822 // the pointer is non-null and valid. This lets us safely dereference
823 // it and assume that adding 1 will create a new, non-null, valid
824 // pointer.
825 unsafe {
826 let ret = self.ptr.read();
827 if ret == 0 {
828 None
829 } else {
830 self.ptr = self.ptr.add(1);
831 Some(ret)
832 }
833 }
834 }
835
836 #[inline]
837 fn size_hint(&self) -> (usize, Option<usize>) {
838 if self.is_empty() { (0, Some(0)) } else { (1, None) }
839 }
840
841 #[inline]
842 fn count(self) -> usize {
843 // SAFETY: We always hold a valid pointer to a C string
844 unsafe { strlen(self.ptr.as_ptr().cast()) }
845 }
846}
847
848#[unstable(feature = "cstr_bytes", issue = "112115")]
849impl FusedIterator for Bytes<'_> {}