1 | //! [`CString`] and its related types. |
2 | |
3 | use core::borrow::Borrow; |
4 | use core::ffi::{CStr, c_char}; |
5 | use core::num::NonZero; |
6 | use core::slice::memchr; |
7 | use core::str::{self, FromStr, Utf8Error}; |
8 | use core::{fmt, mem, ops, ptr, slice}; |
9 | |
10 | use crate::borrow::{Cow, ToOwned}; |
11 | use crate::boxed::Box; |
12 | use crate::rc::Rc; |
13 | use crate::string::String; |
14 | #[cfg (target_has_atomic = "ptr" )] |
15 | use crate::sync::Arc; |
16 | use crate::vec::Vec; |
17 | |
18 | /// A type representing an owned, C-compatible, nul-terminated string with no nul bytes in the |
19 | /// middle. |
20 | /// |
21 | /// This type serves the purpose of being able to safely generate a |
22 | /// C-compatible string from a Rust byte slice or vector. An instance of this |
23 | /// type is a static guarantee that the underlying bytes contain no interior 0 |
24 | /// bytes ("nul characters") and that the final byte is 0 ("nul terminator"). |
25 | /// |
26 | /// `CString` is to <code>&[CStr]</code> as [`String`] is to <code>&[str]</code>: the former |
27 | /// in each pair are owned strings; the latter are borrowed |
28 | /// references. |
29 | /// |
30 | /// # Creating a `CString` |
31 | /// |
32 | /// A `CString` is created from either a byte slice or a byte vector, |
33 | /// or anything that implements <code>[Into]<[Vec]<[u8]>></code> (for |
34 | /// example, you can build a `CString` straight out of a [`String`] or |
35 | /// a <code>&[str]</code>, since both implement that trait). |
36 | /// You can create a `CString` from a literal with `CString::from(c"Text")`. |
37 | /// |
38 | /// The [`CString::new`] method will actually check that the provided <code>&[[u8]]</code> |
39 | /// does not have 0 bytes in the middle, and return an error if it |
40 | /// finds one. |
41 | /// |
42 | /// # Extracting a raw pointer to the whole C string |
43 | /// |
44 | /// `CString` implements an [`as_ptr`][`CStr::as_ptr`] method through the [`Deref`] |
45 | /// trait. This method will give you a `*const c_char` which you can |
46 | /// feed directly to extern functions that expect a nul-terminated |
47 | /// string, like C's `strdup()`. Notice that [`as_ptr`][`CStr::as_ptr`] returns a |
48 | /// read-only pointer; if the C code writes to it, that causes |
49 | /// undefined behavior. |
50 | /// |
51 | /// # Extracting a slice of the whole C string |
52 | /// |
53 | /// Alternatively, you can obtain a <code>&[[u8]]</code> slice from a |
54 | /// `CString` with the [`CString::as_bytes`] method. Slices produced in this |
55 | /// way do *not* contain the trailing nul terminator. This is useful |
56 | /// when you will be calling an extern function that takes a `*const |
57 | /// u8` argument which is not necessarily nul-terminated, plus another |
58 | /// argument with the length of the string — like C's `strndup()`. |
59 | /// You can of course get the slice's length with its |
60 | /// [`len`][slice::len] method. |
61 | /// |
62 | /// If you need a <code>&[[u8]]</code> slice *with* the nul terminator, you |
63 | /// can use [`CString::as_bytes_with_nul`] instead. |
64 | /// |
65 | /// Once you have the kind of slice you need (with or without a nul |
66 | /// terminator), you can call the slice's own |
67 | /// [`as_ptr`][slice::as_ptr] method to get a read-only raw pointer to pass to |
68 | /// extern functions. See the documentation for that function for a |
69 | /// discussion on ensuring the lifetime of the raw pointer. |
70 | /// |
71 | /// [str]: prim@str "str" |
72 | /// [`Deref`]: ops::Deref |
73 | /// |
74 | /// # Examples |
75 | /// |
76 | /// ```ignore (extern-declaration) |
77 | /// # fn main() { |
78 | /// use std::ffi::CString; |
79 | /// use std::os::raw::c_char; |
80 | /// |
81 | /// extern "C" { |
82 | /// fn my_printer(s: *const c_char); |
83 | /// } |
84 | /// |
85 | /// // We are certain that our string doesn't have 0 bytes in the middle, |
86 | /// // so we can .expect() |
87 | /// let c_to_print = CString::new("Hello, world!" ).expect("CString::new failed" ); |
88 | /// unsafe { |
89 | /// my_printer(c_to_print.as_ptr()); |
90 | /// } |
91 | /// # } |
92 | /// ``` |
93 | /// |
94 | /// # Safety |
95 | /// |
96 | /// `CString` is intended for working with traditional C-style strings |
97 | /// (a sequence of non-nul bytes terminated by a single nul byte); the |
98 | /// primary use case for these kinds of strings is interoperating with C-like |
99 | /// code. Often you will need to transfer ownership to/from that external |
100 | /// code. It is strongly recommended that you thoroughly read through the |
101 | /// documentation of `CString` before use, as improper ownership management |
102 | /// of `CString` instances can lead to invalid memory accesses, memory leaks, |
103 | /// and other memory errors. |
104 | #[derive (PartialEq, PartialOrd, Eq, Ord, Hash, Clone)] |
105 | #[rustc_diagnostic_item = "cstring_type" ] |
106 | #[stable (feature = "alloc_c_string" , since = "1.64.0" )] |
107 | pub struct CString { |
108 | // Invariant 1: the slice ends with a zero byte and has a length of at least one. |
109 | // Invariant 2: the slice contains only one zero byte. |
110 | // Improper usage of unsafe function can break Invariant 2, but not Invariant 1. |
111 | inner: Box<[u8]>, |
112 | } |
113 | |
114 | /// An error indicating that an interior nul byte was found. |
115 | /// |
116 | /// While Rust strings may contain nul bytes in the middle, C strings |
117 | /// can't, as that byte would effectively truncate the string. |
118 | /// |
119 | /// This error is created by the [`new`][`CString::new`] method on |
120 | /// [`CString`]. See its documentation for more. |
121 | /// |
122 | /// # Examples |
123 | /// |
124 | /// ``` |
125 | /// use std::ffi::{CString, NulError}; |
126 | /// |
127 | /// let _: NulError = CString::new(b"f \0oo" .to_vec()).unwrap_err(); |
128 | /// ``` |
129 | #[derive (Clone, PartialEq, Eq, Debug)] |
130 | #[stable (feature = "alloc_c_string" , since = "1.64.0" )] |
131 | pub struct NulError(usize, Vec<u8>); |
132 | |
133 | #[derive (Clone, PartialEq, Eq, Debug)] |
134 | enum FromBytesWithNulErrorKind { |
135 | InteriorNul(usize), |
136 | NotNulTerminated, |
137 | } |
138 | |
139 | /// An error indicating that a nul byte was not in the expected position. |
140 | /// |
141 | /// The vector used to create a [`CString`] must have one and only one nul byte, |
142 | /// positioned at the end. |
143 | /// |
144 | /// This error is created by the [`CString::from_vec_with_nul`] method. |
145 | /// See its documentation for more. |
146 | /// |
147 | /// # Examples |
148 | /// |
149 | /// ``` |
150 | /// use std::ffi::{CString, FromVecWithNulError}; |
151 | /// |
152 | /// let _: FromVecWithNulError = CString::from_vec_with_nul(b"f \0oo" .to_vec()).unwrap_err(); |
153 | /// ``` |
154 | #[derive (Clone, PartialEq, Eq, Debug)] |
155 | #[stable (feature = "alloc_c_string" , since = "1.64.0" )] |
156 | pub struct FromVecWithNulError { |
157 | error_kind: FromBytesWithNulErrorKind, |
158 | bytes: Vec<u8>, |
159 | } |
160 | |
161 | #[stable (feature = "cstring_from_vec_with_nul" , since = "1.58.0" )] |
162 | impl FromVecWithNulError { |
163 | /// Returns a slice of [`u8`]s bytes that were attempted to convert to a [`CString`]. |
164 | /// |
165 | /// # Examples |
166 | /// |
167 | /// Basic usage: |
168 | /// |
169 | /// ``` |
170 | /// use std::ffi::CString; |
171 | /// |
172 | /// // Some invalid bytes in a vector |
173 | /// let bytes = b"f \0oo" .to_vec(); |
174 | /// |
175 | /// let value = CString::from_vec_with_nul(bytes.clone()); |
176 | /// |
177 | /// assert_eq!(&bytes[..], value.unwrap_err().as_bytes()); |
178 | /// ``` |
179 | #[must_use ] |
180 | #[stable (feature = "cstring_from_vec_with_nul" , since = "1.58.0" )] |
181 | pub fn as_bytes(&self) -> &[u8] { |
182 | &self.bytes[..] |
183 | } |
184 | |
185 | /// Returns the bytes that were attempted to convert to a [`CString`]. |
186 | /// |
187 | /// This method is carefully constructed to avoid allocation. It will |
188 | /// consume the error, moving out the bytes, so that a copy of the bytes |
189 | /// does not need to be made. |
190 | /// |
191 | /// # Examples |
192 | /// |
193 | /// Basic usage: |
194 | /// |
195 | /// ``` |
196 | /// use std::ffi::CString; |
197 | /// |
198 | /// // Some invalid bytes in a vector |
199 | /// let bytes = b"f \0oo" .to_vec(); |
200 | /// |
201 | /// let value = CString::from_vec_with_nul(bytes.clone()); |
202 | /// |
203 | /// assert_eq!(bytes, value.unwrap_err().into_bytes()); |
204 | /// ``` |
205 | #[must_use = "`self` will be dropped if the result is not used" ] |
206 | #[stable (feature = "cstring_from_vec_with_nul" , since = "1.58.0" )] |
207 | pub fn into_bytes(self) -> Vec<u8> { |
208 | self.bytes |
209 | } |
210 | } |
211 | |
212 | /// An error indicating invalid UTF-8 when converting a [`CString`] into a [`String`]. |
213 | /// |
214 | /// `CString` is just a wrapper over a buffer of bytes with a nul terminator; |
215 | /// [`CString::into_string`] performs UTF-8 validation on those bytes and may |
216 | /// return this error. |
217 | /// |
218 | /// This `struct` is created by [`CString::into_string()`]. See |
219 | /// its documentation for more. |
220 | #[derive (Clone, PartialEq, Eq, Debug)] |
221 | #[stable (feature = "alloc_c_string" , since = "1.64.0" )] |
222 | pub struct IntoStringError { |
223 | inner: CString, |
224 | error: Utf8Error, |
225 | } |
226 | |
227 | impl CString { |
228 | /// Creates a new C-compatible string from a container of bytes. |
229 | /// |
230 | /// This function will consume the provided data and use the |
231 | /// underlying bytes to construct a new string, ensuring that |
232 | /// there is a trailing 0 byte. This trailing 0 byte will be |
233 | /// appended by this function; the provided data should *not* |
234 | /// contain any 0 bytes in it. |
235 | /// |
236 | /// # Examples |
237 | /// |
238 | /// ```ignore (extern-declaration) |
239 | /// use std::ffi::CString; |
240 | /// use std::os::raw::c_char; |
241 | /// |
242 | /// extern "C" { fn puts(s: *const c_char); } |
243 | /// |
244 | /// let to_print = CString::new("Hello!" ).expect("CString::new failed" ); |
245 | /// unsafe { |
246 | /// puts(to_print.as_ptr()); |
247 | /// } |
248 | /// ``` |
249 | /// |
250 | /// # Errors |
251 | /// |
252 | /// This function will return an error if the supplied bytes contain an |
253 | /// internal 0 byte. The [`NulError`] returned will contain the bytes as well as |
254 | /// the position of the nul byte. |
255 | #[stable (feature = "rust1" , since = "1.0.0" )] |
256 | pub fn new<T: Into<Vec<u8>>>(t: T) -> Result<CString, NulError> { |
257 | trait SpecNewImpl { |
258 | fn spec_new_impl(self) -> Result<CString, NulError>; |
259 | } |
260 | |
261 | impl<T: Into<Vec<u8>>> SpecNewImpl for T { |
262 | default fn spec_new_impl(self) -> Result<CString, NulError> { |
263 | let bytes: Vec<u8> = self.into(); |
264 | match memchr::memchr(0, &bytes) { |
265 | Some(i) => Err(NulError(i, bytes)), |
266 | None => Ok(unsafe { CString::_from_vec_unchecked(bytes) }), |
267 | } |
268 | } |
269 | } |
270 | |
271 | // Specialization for avoiding reallocation |
272 | #[inline (always)] // Without that it is not inlined into specializations |
273 | fn spec_new_impl_bytes(bytes: &[u8]) -> Result<CString, NulError> { |
274 | // We cannot have such large slice that we would overflow here |
275 | // but using `checked_add` allows LLVM to assume that capacity never overflows |
276 | // and generate twice shorter code. |
277 | // `saturating_add` doesn't help for some reason. |
278 | let capacity = bytes.len().checked_add(1).unwrap(); |
279 | |
280 | // Allocate before validation to avoid duplication of allocation code. |
281 | // We still need to allocate and copy memory even if we get an error. |
282 | let mut buffer = Vec::with_capacity(capacity); |
283 | buffer.extend(bytes); |
284 | |
285 | // Check memory of self instead of new buffer. |
286 | // This allows better optimizations if lto enabled. |
287 | match memchr::memchr(0, bytes) { |
288 | Some(i) => Err(NulError(i, buffer)), |
289 | None => Ok(unsafe { CString::_from_vec_unchecked(buffer) }), |
290 | } |
291 | } |
292 | |
293 | impl SpecNewImpl for &'_ [u8] { |
294 | fn spec_new_impl(self) -> Result<CString, NulError> { |
295 | spec_new_impl_bytes(self) |
296 | } |
297 | } |
298 | |
299 | impl SpecNewImpl for &'_ str { |
300 | fn spec_new_impl(self) -> Result<CString, NulError> { |
301 | spec_new_impl_bytes(self.as_bytes()) |
302 | } |
303 | } |
304 | |
305 | impl SpecNewImpl for &'_ mut [u8] { |
306 | fn spec_new_impl(self) -> Result<CString, NulError> { |
307 | spec_new_impl_bytes(self) |
308 | } |
309 | } |
310 | |
311 | t.spec_new_impl() |
312 | } |
313 | |
314 | /// Creates a C-compatible string by consuming a byte vector, |
315 | /// without checking for interior 0 bytes. |
316 | /// |
317 | /// Trailing 0 byte will be appended by this function. |
318 | /// |
319 | /// This method is equivalent to [`CString::new`] except that no runtime |
320 | /// assertion is made that `v` contains no 0 bytes, and it requires an |
321 | /// actual byte vector, not anything that can be converted to one with Into. |
322 | /// |
323 | /// # Examples |
324 | /// |
325 | /// ``` |
326 | /// use std::ffi::CString; |
327 | /// |
328 | /// let raw = b"foo" .to_vec(); |
329 | /// unsafe { |
330 | /// let c_string = CString::from_vec_unchecked(raw); |
331 | /// } |
332 | /// ``` |
333 | #[must_use ] |
334 | #[stable (feature = "rust1" , since = "1.0.0" )] |
335 | pub unsafe fn from_vec_unchecked(v: Vec<u8>) -> Self { |
336 | debug_assert!(memchr::memchr(0, &v).is_none()); |
337 | unsafe { Self::_from_vec_unchecked(v) } |
338 | } |
339 | |
340 | unsafe fn _from_vec_unchecked(mut v: Vec<u8>) -> Self { |
341 | v.reserve_exact(1); |
342 | v.push(0); |
343 | Self { inner: v.into_boxed_slice() } |
344 | } |
345 | |
346 | /// Retakes ownership of a `CString` that was transferred to C via |
347 | /// [`CString::into_raw`]. |
348 | /// |
349 | /// Additionally, the length of the string will be recalculated from the pointer. |
350 | /// |
351 | /// # Safety |
352 | /// |
353 | /// This should only ever be called with a pointer that was earlier |
354 | /// obtained by calling [`CString::into_raw`], and the memory it points to must not be accessed |
355 | /// through any other pointer during the lifetime of reconstructed `CString`. |
356 | /// Other usage (e.g., trying to take ownership of a string that was allocated by foreign code) |
357 | /// is likely to lead to undefined behavior or allocator corruption. |
358 | /// |
359 | /// This function does not validate ownership of the raw pointer's memory. |
360 | /// A double-free may occur if the function is called twice on the same raw pointer. |
361 | /// Additionally, the caller must ensure the pointer is not dangling. |
362 | /// |
363 | /// It should be noted that the length isn't just "recomputed," but that |
364 | /// the recomputed length must match the original length from the |
365 | /// [`CString::into_raw`] call. This means the [`CString::into_raw`]/`from_raw` |
366 | /// methods should not be used when passing the string to C functions that can |
367 | /// modify the string's length. |
368 | /// |
369 | /// > **Note:** If you need to borrow a string that was allocated by |
370 | /// > foreign code, use [`CStr`]. If you need to take ownership of |
371 | /// > a string that was allocated by foreign code, you will need to |
372 | /// > make your own provisions for freeing it appropriately, likely |
373 | /// > with the foreign code's API to do that. |
374 | /// |
375 | /// # Examples |
376 | /// |
377 | /// Creates a `CString`, pass ownership to an `extern` function (via raw pointer), then retake |
378 | /// ownership with `from_raw`: |
379 | /// |
380 | /// ```ignore (extern-declaration) |
381 | /// use std::ffi::CString; |
382 | /// use std::os::raw::c_char; |
383 | /// |
384 | /// extern "C" { |
385 | /// fn some_extern_function(s: *mut c_char); |
386 | /// } |
387 | /// |
388 | /// let c_string = CString::from(c"Hello!" ); |
389 | /// let raw = c_string.into_raw(); |
390 | /// unsafe { |
391 | /// some_extern_function(raw); |
392 | /// let c_string = CString::from_raw(raw); |
393 | /// } |
394 | /// ``` |
395 | #[must_use = "call `drop(from_raw(ptr))` if you intend to drop the `CString`" ] |
396 | #[stable (feature = "cstr_memory" , since = "1.4.0" )] |
397 | pub unsafe fn from_raw(ptr: *mut c_char) -> CString { |
398 | // SAFETY: This is called with a pointer that was obtained from a call |
399 | // to `CString::into_raw` and the length has not been modified. As such, |
400 | // we know there is a NUL byte (and only one) at the end and that the |
401 | // information about the size of the allocation is correct on Rust's |
402 | // side. |
403 | unsafe { |
404 | unsafe extern "C" { |
405 | /// Provided by libc or compiler_builtins. |
406 | fn strlen(s: *const c_char) -> usize; |
407 | } |
408 | let len = strlen(ptr) + 1; // Including the NUL byte |
409 | let slice = slice::from_raw_parts_mut(ptr, len); |
410 | CString { inner: Box::from_raw(slice as *mut [c_char] as *mut [u8]) } |
411 | } |
412 | } |
413 | |
414 | /// Consumes the `CString` and transfers ownership of the string to a C caller. |
415 | /// |
416 | /// The pointer which this function returns must be returned to Rust and reconstituted using |
417 | /// [`CString::from_raw`] to be properly deallocated. Specifically, one |
418 | /// should *not* use the standard C `free()` function to deallocate |
419 | /// this string. |
420 | /// |
421 | /// Failure to call [`CString::from_raw`] will lead to a memory leak. |
422 | /// |
423 | /// The C side must **not** modify the length of the string (by writing a |
424 | /// nul byte somewhere inside the string or removing the final one) before |
425 | /// it makes it back into Rust using [`CString::from_raw`]. See the safety section |
426 | /// in [`CString::from_raw`]. |
427 | /// |
428 | /// # Examples |
429 | /// |
430 | /// ``` |
431 | /// use std::ffi::CString; |
432 | /// |
433 | /// let c_string = CString::from(c"foo" ); |
434 | /// |
435 | /// let ptr = c_string.into_raw(); |
436 | /// |
437 | /// unsafe { |
438 | /// assert_eq!(b'f' , *ptr as u8); |
439 | /// assert_eq!(b'o' , *ptr.add(1) as u8); |
440 | /// assert_eq!(b'o' , *ptr.add(2) as u8); |
441 | /// assert_eq!(b' \0' , *ptr.add(3) as u8); |
442 | /// |
443 | /// // retake pointer to free memory |
444 | /// let _ = CString::from_raw(ptr); |
445 | /// } |
446 | /// ``` |
447 | #[inline ] |
448 | #[must_use = "`self` will be dropped if the result is not used" ] |
449 | #[stable (feature = "cstr_memory" , since = "1.4.0" )] |
450 | pub fn into_raw(self) -> *mut c_char { |
451 | Box::into_raw(self.into_inner()) as *mut c_char |
452 | } |
453 | |
454 | /// Converts the `CString` into a [`String`] if it contains valid UTF-8 data. |
455 | /// |
456 | /// On failure, ownership of the original `CString` is returned. |
457 | /// |
458 | /// # Examples |
459 | /// |
460 | /// ``` |
461 | /// use std::ffi::CString; |
462 | /// |
463 | /// let valid_utf8 = vec![b'f' , b'o' , b'o' ]; |
464 | /// let cstring = CString::new(valid_utf8).expect("CString::new failed" ); |
465 | /// assert_eq!(cstring.into_string().expect("into_string() call failed" ), "foo" ); |
466 | /// |
467 | /// let invalid_utf8 = vec![b'f' , 0xff, b'o' , b'o' ]; |
468 | /// let cstring = CString::new(invalid_utf8).expect("CString::new failed" ); |
469 | /// let err = cstring.into_string().err().expect("into_string().err() failed" ); |
470 | /// assert_eq!(err.utf8_error().valid_up_to(), 1); |
471 | /// ``` |
472 | #[stable (feature = "cstring_into" , since = "1.7.0" )] |
473 | pub fn into_string(self) -> Result<String, IntoStringError> { |
474 | String::from_utf8(self.into_bytes()).map_err(|e| IntoStringError { |
475 | error: e.utf8_error(), |
476 | inner: unsafe { Self::_from_vec_unchecked(e.into_bytes()) }, |
477 | }) |
478 | } |
479 | |
480 | /// Consumes the `CString` and returns the underlying byte buffer. |
481 | /// |
482 | /// The returned buffer does **not** contain the trailing nul |
483 | /// terminator, and it is guaranteed to not have any interior nul |
484 | /// bytes. |
485 | /// |
486 | /// # Examples |
487 | /// |
488 | /// ``` |
489 | /// use std::ffi::CString; |
490 | /// |
491 | /// let c_string = CString::from(c"foo" ); |
492 | /// let bytes = c_string.into_bytes(); |
493 | /// assert_eq!(bytes, vec![b'f' , b'o' , b'o' ]); |
494 | /// ``` |
495 | #[must_use = "`self` will be dropped if the result is not used" ] |
496 | #[stable (feature = "cstring_into" , since = "1.7.0" )] |
497 | pub fn into_bytes(self) -> Vec<u8> { |
498 | let mut vec = self.into_inner().into_vec(); |
499 | let _nul = vec.pop(); |
500 | debug_assert_eq!(_nul, Some(0u8)); |
501 | vec |
502 | } |
503 | |
504 | /// Equivalent to [`CString::into_bytes()`] except that the |
505 | /// returned vector includes the trailing nul terminator. |
506 | /// |
507 | /// # Examples |
508 | /// |
509 | /// ``` |
510 | /// use std::ffi::CString; |
511 | /// |
512 | /// let c_string = CString::from(c"foo" ); |
513 | /// let bytes = c_string.into_bytes_with_nul(); |
514 | /// assert_eq!(bytes, vec![b'f' , b'o' , b'o' , b' \0' ]); |
515 | /// ``` |
516 | #[must_use = "`self` will be dropped if the result is not used" ] |
517 | #[stable (feature = "cstring_into" , since = "1.7.0" )] |
518 | pub fn into_bytes_with_nul(self) -> Vec<u8> { |
519 | self.into_inner().into_vec() |
520 | } |
521 | |
522 | /// Returns the contents of this `CString` as a slice of bytes. |
523 | /// |
524 | /// The returned slice does **not** contain the trailing nul |
525 | /// terminator, and it is guaranteed to not have any interior nul |
526 | /// bytes. If you need the nul terminator, use |
527 | /// [`CString::as_bytes_with_nul`] instead. |
528 | /// |
529 | /// # Examples |
530 | /// |
531 | /// ``` |
532 | /// use std::ffi::CString; |
533 | /// |
534 | /// let c_string = CString::from(c"foo" ); |
535 | /// let bytes = c_string.as_bytes(); |
536 | /// assert_eq!(bytes, &[b'f' , b'o' , b'o' ]); |
537 | /// ``` |
538 | #[inline ] |
539 | #[must_use ] |
540 | #[stable (feature = "rust1" , since = "1.0.0" )] |
541 | pub fn as_bytes(&self) -> &[u8] { |
542 | // SAFETY: CString has a length at least 1 |
543 | unsafe { self.inner.get_unchecked(..self.inner.len() - 1) } |
544 | } |
545 | |
546 | /// Equivalent to [`CString::as_bytes()`] except that the |
547 | /// returned slice includes the trailing nul terminator. |
548 | /// |
549 | /// # Examples |
550 | /// |
551 | /// ``` |
552 | /// use std::ffi::CString; |
553 | /// |
554 | /// let c_string = CString::from(c"foo" ); |
555 | /// let bytes = c_string.as_bytes_with_nul(); |
556 | /// assert_eq!(bytes, &[b'f' , b'o' , b'o' , b' \0' ]); |
557 | /// ``` |
558 | #[inline ] |
559 | #[must_use ] |
560 | #[stable (feature = "rust1" , since = "1.0.0" )] |
561 | pub fn as_bytes_with_nul(&self) -> &[u8] { |
562 | &self.inner |
563 | } |
564 | |
565 | /// Extracts a [`CStr`] slice containing the entire string. |
566 | /// |
567 | /// # Examples |
568 | /// |
569 | /// ``` |
570 | /// use std::ffi::{CString, CStr}; |
571 | /// |
572 | /// let c_string = CString::from(c"foo" ); |
573 | /// let cstr = c_string.as_c_str(); |
574 | /// assert_eq!(cstr, |
575 | /// CStr::from_bytes_with_nul(b"foo \0" ).expect("CStr::from_bytes_with_nul failed" )); |
576 | /// ``` |
577 | #[inline ] |
578 | #[must_use ] |
579 | #[stable (feature = "as_c_str" , since = "1.20.0" )] |
580 | #[rustc_diagnostic_item = "cstring_as_c_str" ] |
581 | pub fn as_c_str(&self) -> &CStr { |
582 | unsafe { CStr::from_bytes_with_nul_unchecked(self.as_bytes_with_nul()) } |
583 | } |
584 | |
585 | /// Converts this `CString` into a boxed [`CStr`]. |
586 | /// |
587 | /// # Examples |
588 | /// |
589 | /// ``` |
590 | /// let c_string = c"foo" .to_owned(); |
591 | /// let boxed = c_string.into_boxed_c_str(); |
592 | /// assert_eq!(boxed.to_bytes_with_nul(), b"foo \0" ); |
593 | /// ``` |
594 | #[must_use = "`self` will be dropped if the result is not used" ] |
595 | #[stable (feature = "into_boxed_c_str" , since = "1.20.0" )] |
596 | pub fn into_boxed_c_str(self) -> Box<CStr> { |
597 | unsafe { Box::from_raw(Box::into_raw(self.into_inner()) as *mut CStr) } |
598 | } |
599 | |
600 | /// Bypass "move out of struct which implements [`Drop`] trait" restriction. |
601 | #[inline ] |
602 | fn into_inner(self) -> Box<[u8]> { |
603 | // Rationale: `mem::forget(self)` invalidates the previous call to `ptr::read(&self.inner)` |
604 | // so we use `ManuallyDrop` to ensure `self` is not dropped. |
605 | // Then we can return the box directly without invalidating it. |
606 | // See https://github.com/rust-lang/rust/issues/62553. |
607 | let this = mem::ManuallyDrop::new(self); |
608 | unsafe { ptr::read(&this.inner) } |
609 | } |
610 | |
611 | /// Converts a <code>[Vec]<[u8]></code> to a [`CString`] without checking the |
612 | /// invariants on the given [`Vec`]. |
613 | /// |
614 | /// # Safety |
615 | /// |
616 | /// The given [`Vec`] **must** have one nul byte as its last element. |
617 | /// This means it cannot be empty nor have any other nul byte anywhere else. |
618 | /// |
619 | /// # Example |
620 | /// |
621 | /// ``` |
622 | /// use std::ffi::CString; |
623 | /// assert_eq!( |
624 | /// unsafe { CString::from_vec_with_nul_unchecked(b"abc \0" .to_vec()) }, |
625 | /// unsafe { CString::from_vec_unchecked(b"abc" .to_vec()) } |
626 | /// ); |
627 | /// ``` |
628 | #[must_use ] |
629 | #[stable (feature = "cstring_from_vec_with_nul" , since = "1.58.0" )] |
630 | pub unsafe fn from_vec_with_nul_unchecked(v: Vec<u8>) -> Self { |
631 | debug_assert!(memchr::memchr(0, &v).unwrap() + 1 == v.len()); |
632 | unsafe { Self::_from_vec_with_nul_unchecked(v) } |
633 | } |
634 | |
635 | unsafe fn _from_vec_with_nul_unchecked(v: Vec<u8>) -> Self { |
636 | Self { inner: v.into_boxed_slice() } |
637 | } |
638 | |
639 | /// Attempts to converts a <code>[Vec]<[u8]></code> to a [`CString`]. |
640 | /// |
641 | /// Runtime checks are present to ensure there is only one nul byte in the |
642 | /// [`Vec`], its last element. |
643 | /// |
644 | /// # Errors |
645 | /// |
646 | /// If a nul byte is present and not the last element or no nul bytes |
647 | /// is present, an error will be returned. |
648 | /// |
649 | /// # Examples |
650 | /// |
651 | /// A successful conversion will produce the same result as [`CString::new`] |
652 | /// when called without the ending nul byte. |
653 | /// |
654 | /// ``` |
655 | /// use std::ffi::CString; |
656 | /// assert_eq!( |
657 | /// CString::from_vec_with_nul(b"abc \0" .to_vec()) |
658 | /// .expect("CString::from_vec_with_nul failed" ), |
659 | /// c"abc" .to_owned() |
660 | /// ); |
661 | /// ``` |
662 | /// |
663 | /// An incorrectly formatted [`Vec`] will produce an error. |
664 | /// |
665 | /// ``` |
666 | /// use std::ffi::{CString, FromVecWithNulError}; |
667 | /// // Interior nul byte |
668 | /// let _: FromVecWithNulError = CString::from_vec_with_nul(b"a \0bc" .to_vec()).unwrap_err(); |
669 | /// // No nul byte |
670 | /// let _: FromVecWithNulError = CString::from_vec_with_nul(b"abc" .to_vec()).unwrap_err(); |
671 | /// ``` |
672 | #[stable (feature = "cstring_from_vec_with_nul" , since = "1.58.0" )] |
673 | pub fn from_vec_with_nul(v: Vec<u8>) -> Result<Self, FromVecWithNulError> { |
674 | let nul_pos = memchr::memchr(0, &v); |
675 | match nul_pos { |
676 | Some(nul_pos) if nul_pos + 1 == v.len() => { |
677 | // SAFETY: We know there is only one nul byte, at the end |
678 | // of the vec. |
679 | Ok(unsafe { Self::_from_vec_with_nul_unchecked(v) }) |
680 | } |
681 | Some(nul_pos) => Err(FromVecWithNulError { |
682 | error_kind: FromBytesWithNulErrorKind::InteriorNul(nul_pos), |
683 | bytes: v, |
684 | }), |
685 | None => Err(FromVecWithNulError { |
686 | error_kind: FromBytesWithNulErrorKind::NotNulTerminated, |
687 | bytes: v, |
688 | }), |
689 | } |
690 | } |
691 | } |
692 | |
693 | // Turns this `CString` into an empty string to prevent |
694 | // memory-unsafe code from working by accident. Inline |
695 | // to prevent LLVM from optimizing it away in debug builds. |
696 | #[stable (feature = "cstring_drop" , since = "1.13.0" )] |
697 | #[rustc_insignificant_dtor ] |
698 | impl Drop for CString { |
699 | #[inline ] |
700 | fn drop(&mut self) { |
701 | unsafe { |
702 | *self.inner.get_unchecked_mut(index:0) = 0; |
703 | } |
704 | } |
705 | } |
706 | |
707 | #[stable (feature = "rust1" , since = "1.0.0" )] |
708 | impl ops::Deref for CString { |
709 | type Target = CStr; |
710 | |
711 | #[inline ] |
712 | fn deref(&self) -> &CStr { |
713 | self.as_c_str() |
714 | } |
715 | } |
716 | |
717 | #[stable (feature = "rust1" , since = "1.0.0" )] |
718 | impl fmt::Debug for CString { |
719 | fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
720 | fmt::Debug::fmt(self.as_c_str(), f) |
721 | } |
722 | } |
723 | |
724 | #[stable (feature = "cstring_into" , since = "1.7.0" )] |
725 | impl From<CString> for Vec<u8> { |
726 | /// Converts a [`CString`] into a <code>[Vec]<[u8]></code>. |
727 | /// |
728 | /// The conversion consumes the [`CString`], and removes the terminating NUL byte. |
729 | #[inline ] |
730 | fn from(s: CString) -> Vec<u8> { |
731 | s.into_bytes() |
732 | } |
733 | } |
734 | |
735 | #[stable (feature = "cstr_default" , since = "1.10.0" )] |
736 | impl Default for CString { |
737 | /// Creates an empty `CString`. |
738 | fn default() -> CString { |
739 | let a: &CStr = Default::default(); |
740 | a.to_owned() |
741 | } |
742 | } |
743 | |
744 | #[stable (feature = "cstr_borrow" , since = "1.3.0" )] |
745 | impl Borrow<CStr> for CString { |
746 | #[inline ] |
747 | fn borrow(&self) -> &CStr { |
748 | self |
749 | } |
750 | } |
751 | |
752 | #[stable (feature = "cstring_from_cow_cstr" , since = "1.28.0" )] |
753 | impl<'a> From<Cow<'a, CStr>> for CString { |
754 | /// Converts a `Cow<'a, CStr>` into a `CString`, by copying the contents if they are |
755 | /// borrowed. |
756 | #[inline ] |
757 | fn from(s: Cow<'a, CStr>) -> Self { |
758 | s.into_owned() |
759 | } |
760 | } |
761 | |
762 | #[stable (feature = "box_from_c_str" , since = "1.17.0" )] |
763 | impl From<&CStr> for Box<CStr> { |
764 | /// Converts a `&CStr` into a `Box<CStr>`, |
765 | /// by copying the contents into a newly allocated [`Box`]. |
766 | fn from(s: &CStr) -> Box<CStr> { |
767 | let boxed: Box<[u8]> = Box::from(s.to_bytes_with_nul()); |
768 | unsafe { Box::from_raw(Box::into_raw(boxed) as *mut CStr) } |
769 | } |
770 | } |
771 | |
772 | #[stable (feature = "box_from_mut_slice" , since = "1.84.0" )] |
773 | impl From<&mut CStr> for Box<CStr> { |
774 | /// Converts a `&mut CStr` into a `Box<CStr>`, |
775 | /// by copying the contents into a newly allocated [`Box`]. |
776 | fn from(s: &mut CStr) -> Box<CStr> { |
777 | Self::from(&*s) |
778 | } |
779 | } |
780 | |
781 | #[stable (feature = "box_from_cow" , since = "1.45.0" )] |
782 | impl From<Cow<'_, CStr>> for Box<CStr> { |
783 | /// Converts a `Cow<'a, CStr>` into a `Box<CStr>`, |
784 | /// by copying the contents if they are borrowed. |
785 | #[inline ] |
786 | fn from(cow: Cow<'_, CStr>) -> Box<CStr> { |
787 | match cow { |
788 | Cow::Borrowed(s: &CStr) => Box::from(s), |
789 | Cow::Owned(s: CString) => Box::from(s), |
790 | } |
791 | } |
792 | } |
793 | |
794 | #[stable (feature = "c_string_from_box" , since = "1.18.0" )] |
795 | impl From<Box<CStr>> for CString { |
796 | /// Converts a <code>[Box]<[CStr]></code> into a [`CString`] without copying or allocating. |
797 | #[inline ] |
798 | fn from(s: Box<CStr>) -> CString { |
799 | let raw: *mut [u8] = Box::into_raw(s) as *mut [u8]; |
800 | CString { inner: unsafe { Box::from_raw(raw) } } |
801 | } |
802 | } |
803 | |
804 | #[stable (feature = "cstring_from_vec_of_nonzerou8" , since = "1.43.0" )] |
805 | impl From<Vec<NonZero<u8>>> for CString { |
806 | /// Converts a <code>[Vec]<[NonZero]<[u8]>></code> into a [`CString`] without |
807 | /// copying nor checking for inner nul bytes. |
808 | #[inline ] |
809 | fn from(v: Vec<NonZero<u8>>) -> CString { |
810 | unsafe { |
811 | // Transmute `Vec<NonZero<u8>>` to `Vec<u8>`. |
812 | let v: Vec<u8> = { |
813 | // SAFETY: |
814 | // - transmuting between `NonZero<u8>` and `u8` is sound; |
815 | // - `alloc::Layout<NonZero<u8>> == alloc::Layout<u8>`. |
816 | let (ptr: *mut NonZero, len: usize, cap: usize): (*mut NonZero<u8>, _, _) = Vec::into_raw_parts(self:v); |
817 | Vec::from_raw_parts(ptr.cast::<u8>(), length:len, capacity:cap) |
818 | }; |
819 | // SAFETY: `v` cannot contain nul bytes, given the type-level |
820 | // invariant of `NonZero<u8>`. |
821 | Self::_from_vec_unchecked(v) |
822 | } |
823 | } |
824 | } |
825 | |
826 | #[stable (feature = "c_string_from_str" , since = "1.85.0" )] |
827 | impl FromStr for CString { |
828 | type Err = NulError; |
829 | |
830 | /// Converts a string `s` into a [`CString`]. |
831 | /// |
832 | /// This method is equivalent to [`CString::new`]. |
833 | #[inline ] |
834 | fn from_str(s: &str) -> Result<Self, Self::Err> { |
835 | Self::new(s) |
836 | } |
837 | } |
838 | |
839 | #[stable (feature = "c_string_from_str" , since = "1.85.0" )] |
840 | impl TryFrom<CString> for String { |
841 | type Error = IntoStringError; |
842 | |
843 | /// Converts a [`CString`] into a [`String`] if it contains valid UTF-8 data. |
844 | /// |
845 | /// This method is equivalent to [`CString::into_string`]. |
846 | #[inline ] |
847 | fn try_from(value: CString) -> Result<Self, Self::Error> { |
848 | value.into_string() |
849 | } |
850 | } |
851 | |
852 | #[stable (feature = "more_box_slice_clone" , since = "1.29.0" )] |
853 | impl Clone for Box<CStr> { |
854 | #[inline ] |
855 | fn clone(&self) -> Self { |
856 | (**self).into() |
857 | } |
858 | } |
859 | |
860 | #[stable (feature = "box_from_c_string" , since = "1.20.0" )] |
861 | impl From<CString> for Box<CStr> { |
862 | /// Converts a [`CString`] into a <code>[Box]<[CStr]></code> without copying or allocating. |
863 | #[inline ] |
864 | fn from(s: CString) -> Box<CStr> { |
865 | s.into_boxed_c_str() |
866 | } |
867 | } |
868 | |
869 | #[stable (feature = "cow_from_cstr" , since = "1.28.0" )] |
870 | impl<'a> From<CString> for Cow<'a, CStr> { |
871 | /// Converts a [`CString`] into an owned [`Cow`] without copying or allocating. |
872 | #[inline ] |
873 | fn from(s: CString) -> Cow<'a, CStr> { |
874 | Cow::Owned(s) |
875 | } |
876 | } |
877 | |
878 | #[stable (feature = "cow_from_cstr" , since = "1.28.0" )] |
879 | impl<'a> From<&'a CStr> for Cow<'a, CStr> { |
880 | /// Converts a [`CStr`] into a borrowed [`Cow`] without copying or allocating. |
881 | #[inline ] |
882 | fn from(s: &'a CStr) -> Cow<'a, CStr> { |
883 | Cow::Borrowed(s) |
884 | } |
885 | } |
886 | |
887 | #[stable (feature = "cow_from_cstr" , since = "1.28.0" )] |
888 | impl<'a> From<&'a CString> for Cow<'a, CStr> { |
889 | /// Converts a `&`[`CString`] into a borrowed [`Cow`] without copying or allocating. |
890 | #[inline ] |
891 | fn from(s: &'a CString) -> Cow<'a, CStr> { |
892 | Cow::Borrowed(s.as_c_str()) |
893 | } |
894 | } |
895 | |
896 | #[cfg (target_has_atomic = "ptr" )] |
897 | #[stable (feature = "shared_from_slice2" , since = "1.24.0" )] |
898 | impl From<CString> for Arc<CStr> { |
899 | /// Converts a [`CString`] into an <code>[Arc]<[CStr]></code> by moving the [`CString`] |
900 | /// data into a new [`Arc`] buffer. |
901 | #[inline ] |
902 | fn from(s: CString) -> Arc<CStr> { |
903 | let arc: Arc<[u8]> = Arc::from(s.into_inner()); |
904 | unsafe { Arc::from_raw(ptr:Arc::into_raw(this:arc) as *const CStr) } |
905 | } |
906 | } |
907 | |
908 | #[cfg (target_has_atomic = "ptr" )] |
909 | #[stable (feature = "shared_from_slice2" , since = "1.24.0" )] |
910 | impl From<&CStr> for Arc<CStr> { |
911 | /// Converts a `&CStr` into a `Arc<CStr>`, |
912 | /// by copying the contents into a newly allocated [`Arc`]. |
913 | #[inline ] |
914 | fn from(s: &CStr) -> Arc<CStr> { |
915 | let arc: Arc<[u8]> = Arc::from(s.to_bytes_with_nul()); |
916 | unsafe { Arc::from_raw(ptr:Arc::into_raw(this:arc) as *const CStr) } |
917 | } |
918 | } |
919 | |
920 | #[cfg (target_has_atomic = "ptr" )] |
921 | #[stable (feature = "shared_from_mut_slice" , since = "1.84.0" )] |
922 | impl From<&mut CStr> for Arc<CStr> { |
923 | /// Converts a `&mut CStr` into a `Arc<CStr>`, |
924 | /// by copying the contents into a newly allocated [`Arc`]. |
925 | #[inline ] |
926 | fn from(s: &mut CStr) -> Arc<CStr> { |
927 | Arc::from(&*s) |
928 | } |
929 | } |
930 | |
931 | #[stable (feature = "shared_from_slice2" , since = "1.24.0" )] |
932 | impl From<CString> for Rc<CStr> { |
933 | /// Converts a [`CString`] into an <code>[Rc]<[CStr]></code> by moving the [`CString`] |
934 | /// data into a new [`Rc`] buffer. |
935 | #[inline ] |
936 | fn from(s: CString) -> Rc<CStr> { |
937 | let rc: Rc<[u8]> = Rc::from(s.into_inner()); |
938 | unsafe { Rc::from_raw(ptr:Rc::into_raw(this:rc) as *const CStr) } |
939 | } |
940 | } |
941 | |
942 | #[stable (feature = "shared_from_slice2" , since = "1.24.0" )] |
943 | impl From<&CStr> for Rc<CStr> { |
944 | /// Converts a `&CStr` into a `Rc<CStr>`, |
945 | /// by copying the contents into a newly allocated [`Rc`]. |
946 | #[inline ] |
947 | fn from(s: &CStr) -> Rc<CStr> { |
948 | let rc: Rc<[u8]> = Rc::from(s.to_bytes_with_nul()); |
949 | unsafe { Rc::from_raw(ptr:Rc::into_raw(this:rc) as *const CStr) } |
950 | } |
951 | } |
952 | |
953 | #[stable (feature = "shared_from_mut_slice" , since = "1.84.0" )] |
954 | impl From<&mut CStr> for Rc<CStr> { |
955 | /// Converts a `&mut CStr` into a `Rc<CStr>`, |
956 | /// by copying the contents into a newly allocated [`Rc`]. |
957 | #[inline ] |
958 | fn from(s: &mut CStr) -> Rc<CStr> { |
959 | Rc::from(&*s) |
960 | } |
961 | } |
962 | |
963 | #[cfg (not(no_global_oom_handling))] |
964 | #[stable (feature = "more_rc_default_impls" , since = "1.80.0" )] |
965 | impl Default for Rc<CStr> { |
966 | /// Creates an empty CStr inside an Rc |
967 | /// |
968 | /// This may or may not share an allocation with other Rcs on the same thread. |
969 | #[inline ] |
970 | fn default() -> Self { |
971 | let rc: Rc<[u8]> = Rc::<[u8]>::from(*b" \0" ); |
972 | // `[u8]` has the same layout as `CStr`, and it is `NUL` terminated. |
973 | unsafe { Rc::from_raw(ptr:Rc::into_raw(this:rc) as *const CStr) } |
974 | } |
975 | } |
976 | |
977 | #[stable (feature = "default_box_extra" , since = "1.17.0" )] |
978 | impl Default for Box<CStr> { |
979 | fn default() -> Box<CStr> { |
980 | let boxed: Box<[u8]> = Box::from([0]); |
981 | unsafe { Box::from_raw(Box::into_raw(boxed) as *mut CStr) } |
982 | } |
983 | } |
984 | |
985 | impl NulError { |
986 | /// Returns the position of the nul byte in the slice that caused |
987 | /// [`CString::new`] to fail. |
988 | /// |
989 | /// # Examples |
990 | /// |
991 | /// ``` |
992 | /// use std::ffi::CString; |
993 | /// |
994 | /// let nul_error = CString::new("foo \0bar" ).unwrap_err(); |
995 | /// assert_eq!(nul_error.nul_position(), 3); |
996 | /// |
997 | /// let nul_error = CString::new("foo bar \0" ).unwrap_err(); |
998 | /// assert_eq!(nul_error.nul_position(), 7); |
999 | /// ``` |
1000 | #[must_use ] |
1001 | #[stable (feature = "rust1" , since = "1.0.0" )] |
1002 | pub fn nul_position(&self) -> usize { |
1003 | self.0 |
1004 | } |
1005 | |
1006 | /// Consumes this error, returning the underlying vector of bytes which |
1007 | /// generated the error in the first place. |
1008 | /// |
1009 | /// # Examples |
1010 | /// |
1011 | /// ``` |
1012 | /// use std::ffi::CString; |
1013 | /// |
1014 | /// let nul_error = CString::new("foo \0bar" ).unwrap_err(); |
1015 | /// assert_eq!(nul_error.into_vec(), b"foo \0bar" ); |
1016 | /// ``` |
1017 | #[must_use = "`self` will be dropped if the result is not used" ] |
1018 | #[stable (feature = "rust1" , since = "1.0.0" )] |
1019 | pub fn into_vec(self) -> Vec<u8> { |
1020 | self.1 |
1021 | } |
1022 | } |
1023 | |
1024 | #[stable (feature = "rust1" , since = "1.0.0" )] |
1025 | impl fmt::Display for NulError { |
1026 | fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
1027 | write!(f, "nul byte found in provided data at position: {}" , self.0) |
1028 | } |
1029 | } |
1030 | |
1031 | #[stable (feature = "cstring_from_vec_with_nul" , since = "1.58.0" )] |
1032 | impl fmt::Display for FromVecWithNulError { |
1033 | fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
1034 | match self.error_kind { |
1035 | FromBytesWithNulErrorKind::InteriorNul(pos: usize) => { |
1036 | write!(f, "data provided contains an interior nul byte at pos {pos}" ) |
1037 | } |
1038 | FromBytesWithNulErrorKind::NotNulTerminated => { |
1039 | write!(f, "data provided is not nul terminated" ) |
1040 | } |
1041 | } |
1042 | } |
1043 | } |
1044 | |
1045 | impl IntoStringError { |
1046 | /// Consumes this error, returning original [`CString`] which generated the |
1047 | /// error. |
1048 | #[must_use = "`self` will be dropped if the result is not used" ] |
1049 | #[stable (feature = "cstring_into" , since = "1.7.0" )] |
1050 | pub fn into_cstring(self) -> CString { |
1051 | self.inner |
1052 | } |
1053 | |
1054 | /// Access the underlying UTF-8 error that was the cause of this error. |
1055 | #[must_use ] |
1056 | #[stable (feature = "cstring_into" , since = "1.7.0" )] |
1057 | pub fn utf8_error(&self) -> Utf8Error { |
1058 | self.error |
1059 | } |
1060 | } |
1061 | |
1062 | impl IntoStringError { |
1063 | fn description(&self) -> &str { |
1064 | "C string contained non-utf8 bytes" |
1065 | } |
1066 | } |
1067 | |
1068 | #[stable (feature = "cstring_into" , since = "1.7.0" )] |
1069 | impl fmt::Display for IntoStringError { |
1070 | #[allow (deprecated, deprecated_in_future)] |
1071 | fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
1072 | self.description().fmt(f) |
1073 | } |
1074 | } |
1075 | |
1076 | #[stable (feature = "cstr_borrow" , since = "1.3.0" )] |
1077 | impl ToOwned for CStr { |
1078 | type Owned = CString; |
1079 | |
1080 | fn to_owned(&self) -> CString { |
1081 | CString { inner: self.to_bytes_with_nul().into() } |
1082 | } |
1083 | |
1084 | fn clone_into(&self, target: &mut CString) { |
1085 | let mut b: Vec = mem::take(&mut target.inner).into_vec(); |
1086 | self.to_bytes_with_nul().clone_into(&mut b); |
1087 | target.inner = b.into_boxed_slice(); |
1088 | } |
1089 | } |
1090 | |
1091 | #[stable (feature = "cstring_asref" , since = "1.7.0" )] |
1092 | impl From<&CStr> for CString { |
1093 | /// Converts a <code>&[CStr]</code> into a [`CString`] |
1094 | /// by copying the contents into a new allocation. |
1095 | fn from(s: &CStr) -> CString { |
1096 | s.to_owned() |
1097 | } |
1098 | } |
1099 | |
1100 | #[stable (feature = "cstring_asref" , since = "1.7.0" )] |
1101 | impl ops::Index<ops::RangeFull> for CString { |
1102 | type Output = CStr; |
1103 | |
1104 | #[inline ] |
1105 | fn index(&self, _index: ops::RangeFull) -> &CStr { |
1106 | self |
1107 | } |
1108 | } |
1109 | |
1110 | #[stable (feature = "cstring_asref" , since = "1.7.0" )] |
1111 | impl AsRef<CStr> for CString { |
1112 | #[inline ] |
1113 | fn as_ref(&self) -> &CStr { |
1114 | self |
1115 | } |
1116 | } |
1117 | |
1118 | impl CStr { |
1119 | /// Converts a `CStr` into a <code>[Cow]<[str]></code>. |
1120 | /// |
1121 | /// If the contents of the `CStr` are valid UTF-8 data, this |
1122 | /// function will return a <code>[Cow]::[Borrowed]\(&[str])</code> |
1123 | /// with the corresponding <code>&[str]</code> slice. Otherwise, it will |
1124 | /// replace any invalid UTF-8 sequences with |
1125 | /// [`U+FFFD REPLACEMENT CHARACTER`][U+FFFD] and return a |
1126 | /// <code>[Cow]::[Owned]\([String])</code> with the result. |
1127 | /// |
1128 | /// [str]: prim@str "str" |
1129 | /// [Borrowed]: Cow::Borrowed |
1130 | /// [Owned]: Cow::Owned |
1131 | /// [U+FFFD]: core::char::REPLACEMENT_CHARACTER "std::char::REPLACEMENT_CHARACTER" |
1132 | /// |
1133 | /// # Examples |
1134 | /// |
1135 | /// Calling `to_string_lossy` on a `CStr` containing valid UTF-8. The leading |
1136 | /// `c` on the string literal denotes a `CStr`. |
1137 | /// |
1138 | /// ``` |
1139 | /// use std::borrow::Cow; |
1140 | /// |
1141 | /// assert_eq!(c"Hello World" .to_string_lossy(), Cow::Borrowed("Hello World" )); |
1142 | /// ``` |
1143 | /// |
1144 | /// Calling `to_string_lossy` on a `CStr` containing invalid UTF-8: |
1145 | /// |
1146 | /// ``` |
1147 | /// use std::borrow::Cow; |
1148 | /// |
1149 | /// assert_eq!( |
1150 | /// c"Hello \xF0\x90\x80World" .to_string_lossy(), |
1151 | /// Cow::Owned(String::from("Hello �World" )) as Cow<'_, str> |
1152 | /// ); |
1153 | /// ``` |
1154 | #[rustc_allow_incoherent_impl ] |
1155 | #[must_use = "this returns the result of the operation, \ |
1156 | without modifying the original" ] |
1157 | #[stable (feature = "cstr_to_str" , since = "1.4.0" )] |
1158 | pub fn to_string_lossy(&self) -> Cow<'_, str> { |
1159 | String::from_utf8_lossy(self.to_bytes()) |
1160 | } |
1161 | |
1162 | /// Converts a <code>[Box]<[CStr]></code> into a [`CString`] without copying or allocating. |
1163 | /// |
1164 | /// # Examples |
1165 | /// |
1166 | /// ``` |
1167 | /// use std::ffi::{CStr, CString}; |
1168 | /// |
1169 | /// let boxed: Box<CStr> = Box::from(c"foo" ); |
1170 | /// let c_string: CString = c"foo" .to_owned(); |
1171 | /// |
1172 | /// assert_eq!(boxed.into_c_string(), c_string); |
1173 | /// ``` |
1174 | #[rustc_allow_incoherent_impl ] |
1175 | #[must_use = "`self` will be dropped if the result is not used" ] |
1176 | #[stable (feature = "into_boxed_c_str" , since = "1.20.0" )] |
1177 | pub fn into_c_string(self: Box<Self>) -> CString { |
1178 | CString::from(self) |
1179 | } |
1180 | } |
1181 | |
1182 | #[stable (feature = "rust1" , since = "1.0.0" )] |
1183 | impl core::error::Error for NulError { |
1184 | #[allow (deprecated)] |
1185 | fn description(&self) -> &str { |
1186 | "nul byte found in data" |
1187 | } |
1188 | } |
1189 | |
1190 | #[stable (feature = "cstring_from_vec_with_nul" , since = "1.58.0" )] |
1191 | impl core::error::Error for FromVecWithNulError {} |
1192 | |
1193 | #[stable (feature = "cstring_into" , since = "1.7.0" )] |
1194 | impl core::error::Error for IntoStringError { |
1195 | #[allow (deprecated)] |
1196 | fn description(&self) -> &str { |
1197 | "C string contained non-utf8 bytes" |
1198 | } |
1199 | |
1200 | fn source(&self) -> Option<&(dyn core::error::Error + 'static)> { |
1201 | Some(&self.error) |
1202 | } |
1203 | } |
1204 | |