alloc/str.rs
1//! Utilities for the `str` primitive type.
2//!
3//! *[See also the `str` primitive type](str).*
4
5#![stable(feature = "rust1", since = "1.0.0")]
6// Many of the usings in this module are only used in the test configuration.
7// It's cleaner to just turn off the unused_imports warning than to fix them.
8#![allow(unused_imports)]
9
10use core::borrow::{Borrow, BorrowMut};
11use core::iter::FusedIterator;
12use core::mem::MaybeUninit;
13#[stable(feature = "encode_utf16", since = "1.8.0")]
14pub use core::str::EncodeUtf16;
15#[stable(feature = "split_ascii_whitespace", since = "1.34.0")]
16pub use core::str::SplitAsciiWhitespace;
17#[stable(feature = "split_inclusive", since = "1.51.0")]
18pub use core::str::SplitInclusive;
19#[stable(feature = "rust1", since = "1.0.0")]
20pub use core::str::SplitWhitespace;
21#[stable(feature = "rust1", since = "1.0.0")]
22pub use core::str::pattern;
23use core::str::pattern::{DoubleEndedSearcher, Pattern, ReverseSearcher, Searcher, Utf8Pattern};
24#[stable(feature = "rust1", since = "1.0.0")]
25pub use core::str::{Bytes, CharIndices, Chars, from_utf8, from_utf8_mut};
26#[stable(feature = "str_escape", since = "1.34.0")]
27pub use core::str::{EscapeDebug, EscapeDefault, EscapeUnicode};
28#[stable(feature = "rust1", since = "1.0.0")]
29pub use core::str::{FromStr, Utf8Error};
30#[allow(deprecated)]
31#[stable(feature = "rust1", since = "1.0.0")]
32pub use core::str::{Lines, LinesAny};
33#[stable(feature = "rust1", since = "1.0.0")]
34pub use core::str::{MatchIndices, RMatchIndices};
35#[stable(feature = "rust1", since = "1.0.0")]
36pub use core::str::{Matches, RMatches};
37#[stable(feature = "rust1", since = "1.0.0")]
38pub use core::str::{ParseBoolError, from_utf8_unchecked, from_utf8_unchecked_mut};
39#[stable(feature = "rust1", since = "1.0.0")]
40pub use core::str::{RSplit, Split};
41#[stable(feature = "rust1", since = "1.0.0")]
42pub use core::str::{RSplitN, SplitN};
43#[stable(feature = "rust1", since = "1.0.0")]
44pub use core::str::{RSplitTerminator, SplitTerminator};
45#[stable(feature = "utf8_chunks", since = "1.79.0")]
46pub use core::str::{Utf8Chunk, Utf8Chunks};
47#[unstable(feature = "str_from_raw_parts", issue = "119206")]
48pub use core::str::{from_raw_parts, from_raw_parts_mut};
49use core::unicode::conversions;
50use core::{mem, ptr};
51
52use crate::borrow::ToOwned;
53use crate::boxed::Box;
54use crate::slice::{Concat, Join, SliceIndex};
55use crate::string::String;
56use crate::vec::Vec;
57
58/// Note: `str` in `Concat<str>` is not meaningful here.
59/// This type parameter of the trait only exists to enable another impl.
60#[cfg(not(no_global_oom_handling))]
61#[unstable(feature = "slice_concat_ext", issue = "27747")]
62impl<S: Borrow<str>> Concat<str> for [S] {
63 type Output = String;
64
65 fn concat(slice: &Self) -> String {
66 Join::join(slice, "")
67 }
68}
69
70#[cfg(not(no_global_oom_handling))]
71#[unstable(feature = "slice_concat_ext", issue = "27747")]
72impl<S: Borrow<str>> Join<&str> for [S] {
73 type Output = String;
74
75 fn join(slice: &Self, sep: &str) -> String {
76 unsafe { String::from_utf8_unchecked(join_generic_copy(slice, sep.as_bytes())) }
77 }
78}
79
80#[cfg(not(no_global_oom_handling))]
81macro_rules! specialize_for_lengths {
82 ($separator:expr, $target:expr, $iter:expr; $($num:expr),*) => {{
83 let mut target = $target;
84 let iter = $iter;
85 let sep_bytes = $separator;
86 match $separator.len() {
87 $(
88 // loops with hardcoded sizes run much faster
89 // specialize the cases with small separator lengths
90 $num => {
91 for s in iter {
92 copy_slice_and_advance!(target, sep_bytes);
93 let content_bytes = s.borrow().as_ref();
94 copy_slice_and_advance!(target, content_bytes);
95 }
96 },
97 )*
98 _ => {
99 // arbitrary non-zero size fallback
100 for s in iter {
101 copy_slice_and_advance!(target, sep_bytes);
102 let content_bytes = s.borrow().as_ref();
103 copy_slice_and_advance!(target, content_bytes);
104 }
105 }
106 }
107 target
108 }}
109}
110
111#[cfg(not(no_global_oom_handling))]
112macro_rules! copy_slice_and_advance {
113 ($target:expr, $bytes:expr) => {
114 let len = $bytes.len();
115 let (head, tail) = { $target }.split_at_mut(len);
116 head.copy_from_slice($bytes);
117 $target = tail;
118 };
119}
120
121// Optimized join implementation that works for both Vec<T> (T: Copy) and String's inner vec
122// Currently (2018-05-13) there is a bug with type inference and specialization (see issue #36262)
123// For this reason SliceConcat<T> is not specialized for T: Copy and SliceConcat<str> is the
124// only user of this function. It is left in place for the time when that is fixed.
125//
126// the bounds for String-join are S: Borrow<str> and for Vec-join Borrow<[T]>
127// [T] and str both impl AsRef<[T]> for some T
128// => s.borrow().as_ref() and we always have slices
129//
130// # Safety notes
131//
132// `Borrow` is a safe trait, and implementations are not required
133// to be deterministic. An inconsistent `Borrow` implementation could return slices
134// of different lengths on consecutive calls (e.g. by using interior mutability).
135//
136// This implementation calls `borrow()` multiple times:
137// 1. To calculate `reserved_len`, all elements are borrowed once.
138// 2. All elements, except the first, are borrowed a second time when building the mapped iterator.
139//
140// Risks and Mitigations:
141// - If elements 2..N GROW on their second borrow, the target slice bounds set by `checked_sub`
142// means that `split_at_mut` inside `copy_slice_and_advance!` will correctly panic.
143// - If elements SHRINK on their second borrow, the spare space is never written, and the final
144// length set via `set_len` masks trailing uninitialized bytes.
145#[cfg(not(no_global_oom_handling))]
146fn join_generic_copy<B, T, S>(slice: &[S], sep: &[T]) -> Vec<T>
147where
148 T: Copy,
149 B: AsRef<[T]> + ?Sized,
150 S: Borrow<B>,
151{
152 let sep_len = sep.len();
153 let mut iter = slice.iter();
154
155 // the first slice is the only one without a separator preceding it
156 // we take care to only borrow this once during the length calculation
157 // to avoid inconsistent Borrow implementations from breaking our assumptions
158 let first = match iter.next() {
159 Some(first) => first.borrow().as_ref(),
160 None => return vec![],
161 };
162
163 // compute the exact total length of the joined Vec
164 // if the `len` calculation overflows, we'll panic
165 // we would have run out of memory anyway and the rest of the function requires
166 // the entire Vec pre-allocated for safety
167 let reserved_len = sep_len
168 .checked_mul(iter.len())
169 .and_then(|n| n.checked_add(first.len()))
170 .and_then(|n| {
171 // iter starts from the second element as we've already taken the first
172 // it's cloned so we can reuse the same iterator below
173 iter.clone().map(|s| s.borrow().as_ref().len()).try_fold(n, usize::checked_add)
174 })
175 .expect("attempt to join into collection with len > usize::MAX");
176
177 // prepare an uninitialized buffer
178 let mut result = Vec::with_capacity(reserved_len);
179 debug_assert!(result.capacity() >= reserved_len);
180
181 result.extend_from_slice(first);
182
183 unsafe {
184 let pos = result.len();
185 debug_assert!(reserved_len >= pos);
186 let target = result.spare_capacity_mut().get_unchecked_mut(..reserved_len - pos);
187
188 // Convert the separator and slices to slices of MaybeUninit
189 // to simplify implementation in specialize_for_lengths.
190 let sep_uninit = core::slice::from_raw_parts(sep.as_ptr().cast(), sep.len());
191 let iter_uninit = iter.map(|it| {
192 let it = it.borrow().as_ref();
193 core::slice::from_raw_parts(it.as_ptr().cast(), it.len())
194 });
195
196 // copy separator and slices over without bounds checks.
197 // `specialize_for_lengths!` internally calls `s.borrow()`, but because it uses
198 // the bounds-checked `split_at_mut` any misbehaving implementation
199 // will not write out of bounds.
200 let remain = specialize_for_lengths!(sep_uninit, target, iter_uninit; 0, 1, 2, 3, 4);
201
202 // A weird borrow implementation may return different
203 // slices for the length calculation and the actual copy.
204 // Make sure we don't expose uninitialized bytes to the caller.
205 let result_len = reserved_len - remain.len();
206 result.set_len(result_len);
207 }
208 result
209}
210
211/// Helper for final sigma lowercase
212#[cfg(not(no_global_oom_handling))]
213fn map_uppercase_sigma(from: &str, i: usize) -> char {
214 fn case_ignorable_then_cased<I: Iterator<Item = char>>(iter: I) -> bool {
215 match iter.skip_while(|&c| c.is_case_ignorable()).next() {
216 Some(c) => c.is_cased(),
217 None => false,
218 }
219 }
220
221 // See https://www.unicode.org/versions/latest/core-spec/chapter-3/#G54277
222 // for the definition of `Final_Sigma`.
223 let is_word_final = case_ignorable_then_cased(from[..i].chars().rev())
224 && !case_ignorable_then_cased(from[i + const { 'Σ'.len_utf8() }..].chars());
225 if is_word_final { 'ς' } else { 'σ' }
226}
227
228#[stable(feature = "rust1", since = "1.0.0")]
229impl Borrow<str> for String {
230 #[inline]
231 fn borrow(&self) -> &str {
232 &self[..]
233 }
234}
235
236#[stable(feature = "string_borrow_mut", since = "1.36.0")]
237impl BorrowMut<str> for String {
238 #[inline]
239 fn borrow_mut(&mut self) -> &mut str {
240 &mut self[..]
241 }
242}
243
244#[cfg(not(no_global_oom_handling))]
245#[stable(feature = "rust1", since = "1.0.0")]
246impl ToOwned for str {
247 type Owned = String;
248
249 #[inline]
250 fn to_owned(&self) -> String {
251 unsafe { String::from_utf8_unchecked(self.as_bytes().to_owned()) }
252 }
253
254 #[inline]
255 fn clone_into(&self, target: &mut String) {
256 target.clear();
257 target.push_str(self);
258 }
259}
260
261/// Methods for string slices.
262impl str {
263 /// Converts a `Box<str>` into a `Box<[u8]>` without copying or allocating.
264 ///
265 /// # Examples
266 ///
267 /// ```
268 /// let s = "this is a string";
269 /// let boxed_str = s.to_owned().into_boxed_str();
270 /// let boxed_bytes = boxed_str.into_boxed_bytes();
271 /// assert_eq!(*boxed_bytes, *s.as_bytes());
272 /// ```
273 #[rustc_allow_incoherent_impl]
274 #[stable(feature = "str_box_extras", since = "1.20.0")]
275 #[must_use = "`self` will be dropped if the result is not used"]
276 #[inline]
277 pub fn into_boxed_bytes(self: Box<Self>) -> Box<[u8]> {
278 self.into()
279 }
280
281 /// Replaces all matches of a pattern with another string.
282 ///
283 /// `replace` creates a new [`String`], and copies the data from this string slice into it.
284 /// While doing so, it attempts to find matches of a pattern. If it finds any, it
285 /// replaces them with the replacement string slice.
286 ///
287 /// # Examples
288 ///
289 /// ```
290 /// let s = "this is old";
291 ///
292 /// assert_eq!("this is new", s.replace("old", "new"));
293 /// assert_eq!("than an old", s.replace("is", "an"));
294 /// ```
295 ///
296 /// When the pattern doesn't match, it returns this string slice as [`String`]:
297 ///
298 /// ```
299 /// let s = "this is old";
300 /// assert_eq!(s, s.replace("cookie monster", "little lamb"));
301 /// ```
302 #[cfg(not(no_global_oom_handling))]
303 #[rustc_allow_incoherent_impl]
304 #[must_use = "this returns the replaced string as a new allocation, \
305 without modifying the original"]
306 #[stable(feature = "rust1", since = "1.0.0")]
307 #[inline]
308 pub fn replace<P: Pattern>(&self, from: P, to: &str) -> String {
309 // Fast path for replacing a single ASCII character with another.
310 if let Some(from_byte) = match from.as_utf8_pattern() {
311 Some(Utf8Pattern::StringPattern(s)) => match s.as_bytes() {
312 [from_byte] => Some(*from_byte),
313 _ => None,
314 },
315 Some(Utf8Pattern::CharPattern(c)) => c.as_ascii().map(|ascii_char| ascii_char.to_u8()),
316 _ => None,
317 } {
318 if let [to_byte] = to.as_bytes() {
319 return unsafe { replace_ascii(self.as_bytes(), from_byte, *to_byte) };
320 }
321 }
322 // Set result capacity to self.len() when from.len() <= to.len()
323 let default_capacity = match from.as_utf8_pattern() {
324 Some(Utf8Pattern::StringPattern(s)) if s.len() <= to.len() => self.len(),
325 Some(Utf8Pattern::CharPattern(c)) if c.len_utf8() <= to.len() => self.len(),
326 _ => 0,
327 };
328 let mut result = String::with_capacity(default_capacity);
329 let mut last_end = 0;
330 for (start, part) in self.match_indices(from) {
331 result.push_str(unsafe { self.get_unchecked(last_end..start) });
332 result.push_str(to);
333 last_end = start + part.len();
334 }
335 result.push_str(unsafe { self.get_unchecked(last_end..self.len()) });
336 result
337 }
338
339 /// Replaces first N matches of a pattern with another string.
340 ///
341 /// `replacen` creates a new [`String`], and copies the data from this string slice into it.
342 /// While doing so, it attempts to find matches of a pattern. If it finds any, it
343 /// replaces them with the replacement string slice at most `count` times.
344 ///
345 /// # Examples
346 ///
347 /// ```
348 /// let s = "foo foo 123 foo";
349 /// assert_eq!("new new 123 foo", s.replacen("foo", "new", 2));
350 /// assert_eq!("faa fao 123 foo", s.replacen('o', "a", 3));
351 /// assert_eq!("foo foo new23 foo", s.replacen(char::is_numeric, "new", 1));
352 /// ```
353 ///
354 /// When the pattern doesn't match, it returns this string slice as [`String`]:
355 ///
356 /// ```
357 /// let s = "this is old";
358 /// assert_eq!(s, s.replacen("cookie monster", "little lamb", 10));
359 /// ```
360 #[cfg(not(no_global_oom_handling))]
361 #[rustc_allow_incoherent_impl]
362 #[doc(alias = "replace_first")]
363 #[must_use = "this returns the replaced string as a new allocation, \
364 without modifying the original"]
365 #[stable(feature = "str_replacen", since = "1.16.0")]
366 pub fn replacen<P: Pattern>(&self, pat: P, to: &str, count: usize) -> String {
367 // Hope to reduce the times of re-allocation
368 let mut result = String::with_capacity(32);
369 let mut last_end = 0;
370 for (start, part) in self.match_indices(pat).take(count) {
371 result.push_str(unsafe { self.get_unchecked(last_end..start) });
372 result.push_str(to);
373 last_end = start + part.len();
374 }
375 result.push_str(unsafe { self.get_unchecked(last_end..self.len()) });
376 result
377 }
378
379 /// Returns the lowercase equivalent of this string slice, as a new [`String`].
380 ///
381 /// 'Lowercase' is defined according to the terms of
382 /// [Chapter 3 (Conformance)](https://www.unicode.org/versions/latest/core-spec/chapter-3/#G34432)
383 /// of the Unicode standard.
384 ///
385 /// Since some characters can expand into multiple characters when changing
386 /// the case, this function returns a [`String`] instead of modifying the
387 /// parameter in-place.
388 ///
389 /// Unlike [`char::to_lowercase()`], this method fully handles the context-dependent
390 /// casing of Greek sigma. However, like that method, it does not handle locale-specific
391 /// casing, like Turkish and Azeri I/ı/İ/i. See its documentation
392 /// for more information.
393 ///
394 /// # Examples
395 ///
396 /// Basic usage:
397 ///
398 /// ```
399 /// let s = "HELLO WORLD";
400 ///
401 /// assert_eq!("hello world", s.to_lowercase());
402 /// ```
403 ///
404 /// Tricky examples, with sigma:
405 ///
406 /// ```
407 /// let sigma = "Σ";
408 ///
409 /// assert_eq!("σ", sigma.to_lowercase());
410 ///
411 /// // but at the end of a word, it's ς, not σ:
412 /// let odysseus = "ὈΔΥΣΣΕΎΣ";
413 ///
414 /// assert_eq!("ὀδυσσεύς", odysseus.to_lowercase());
415 ///
416 /// let odysseus_king_of_ithaca = "Ο ΟΔΥΣΣΈΑΣ ΒΑΣΙΛΙΆΣ ΤΗΣ ΙΘΆΚΗΣ";
417 ///
418 /// assert_eq!("ο οδυσσέας βασιλιάς της ιθάκης", odysseus_king_of_ithaca.to_lowercase());
419 /// ```
420 ///
421 /// Languages without case are not changed:
422 ///
423 /// ```
424 /// let new_year = "农历新年";
425 ///
426 /// assert_eq!(new_year, new_year.to_lowercase());
427 /// ```
428 #[cfg(not(no_global_oom_handling))]
429 #[rustc_allow_incoherent_impl]
430 #[must_use = "this returns the lowercase string as a new String, \
431 without modifying the original"]
432 #[stable(feature = "unicode_case_mapping", since = "1.2.0")]
433 pub fn to_lowercase(&self) -> String {
434 // SAFETY: `to_ascii_lowercase` preserves ASCII bytes, so the converted
435 // prefix remains valid UTF-8.
436 let (mut s, rest) = unsafe { convert_while_ascii(self, u8::to_ascii_lowercase) };
437
438 let prefix_len = s.len();
439
440 for (i, c) in rest.char_indices() {
441 if c == 'Σ' {
442 // Σ maps to σ, except at the end of a word where it maps to ς.
443 // This is the only conditional (contextual) but language-independent mapping
444 // in `SpecialCasing.txt`,
445 // so hard-code it rather than have a generic "condition" mechanism.
446 // See https://github.com/rust-lang/rust/issues/26035
447 let sigma_lowercase = map_uppercase_sigma(self, prefix_len + i);
448 s.push(sigma_lowercase);
449 } else {
450 match conversions::to_lower(c) {
451 [a, '\0', _] => s.push(a),
452 [a, b, '\0'] => {
453 s.push(a);
454 s.push(b);
455 }
456 [a, b, c] => {
457 s.push(a);
458 s.push(b);
459 s.push(c);
460 }
461 }
462 }
463 }
464 return s;
465 }
466
467 /// Returns the titlecase equivalent of this string slice,
468 /// which is assumed to represent a single word,
469 /// as a new [`String`].
470 ///
471 /// Essentially, this consists of uppercasing the first cased letter
472 /// (with [`char::to_titlecase()`]), and lowercasing everything that follows.
473 ///
474 /// 'Titlecase' is defined according to the terms of
475 /// [Chapter 3 (Conformance)](https://www.unicode.org/versions/latest/core-spec/chapter-3/#G34082)
476 /// of the Unicode standard.
477 ///
478 /// Since some characters can expand into multiple characters when changing
479 /// the case, this function returns a [`String`] instead of modifying the
480 /// parameter in-place.
481 ///
482 /// Unlike [`char::to_lowercase()`], this method fully handles the context-dependent
483 /// casing of Greek sigma. However, like that method, it does not handle locale-specific
484 /// casing, like Turkish and Azeri I/ı/İ/i. See its documentation
485 /// for more information.
486 ///
487 /// This method does not perform any kind of word segmentation.
488 ///
489 /// # Examples
490 ///
491 /// Basic usage:
492 ///
493 /// ```
494 /// #![feature(titlecase)]
495 /// let s = "HELLO WORLD";
496 ///
497 /// assert_eq!("Hello world", s.word_to_titlecase());
498 /// ```
499 ///
500 /// The first *cased* letter is uppercased:
501 ///
502 /// ```
503 /// #![feature(titlecase)]
504 /// let the_night_before_christmas = "'twas";
505 ///
506 /// assert_eq!("'Twas", the_night_before_christmas.word_to_titlecase());
507 /// ```
508 ///
509 /// Languages without case are not changed:
510 ///
511 /// ```
512 /// #![feature(titlecase)]
513 /// let new_year = "农历新年";
514 ///
515 /// assert_eq!(new_year, new_year.word_to_titlecase());
516 /// ```
517 ///
518 /// Georgian uppercase ("Mtavruli") letters are not used in titlecase:
519 ///
520 /// ```
521 /// #![feature(titlecase)]
522 /// let georgian = "ერთობაშია";
523 ///
524 /// assert_eq!(georgian, georgian.word_to_titlecase());
525 /// ```
526 ///
527 /// No word segmentation is performed,
528 /// so only the first cased letter in the whole string gets uppercased:
529 ///
530 /// ```
531 /// #![feature(titlecase)]
532 /// let blazingly_fast = "ferris and I";
533 ///
534 /// assert_eq!("Ferris and i", blazingly_fast.word_to_titlecase());
535 /// ```
536 ///
537 /// Tricky examples, with sigma:
538 ///
539 /// ```
540 /// #![feature(titlecase)]
541 /// let odysseus = "ὈΔΥΣΣΕΎΣ";
542 ///
543 /// assert_eq!("Ὀδυσσεύς", odysseus.word_to_titlecase());
544 ///
545 /// let odysseus_king_of_ithaca = "Ο ΟΔΥΣΣΈΑΣ ΒΑΣΙΛΙΆΣ ΤΗΣ ΙΘΆΚΗΣ";
546 ///
547 /// assert_eq!("Ο οδυσσέας βασιλιάς της ιθάκης", odysseus_king_of_ithaca.word_to_titlecase());
548 /// ```
549 #[cfg(not(no_global_oom_handling))]
550 #[rustc_allow_incoherent_impl]
551 #[must_use = "this returns the titlecase word as a new String, \
552 without modifying the original"]
553 #[unstable(feature = "titlecase", issue = "153892")]
554 pub fn word_to_titlecase(&self) -> String {
555 // FIXME: add ASCII fast path
556
557 let mut s = String::with_capacity(self.len());
558 let mut chars = self.char_indices();
559
560 'until_first_cased_char: for (_, c) in chars.by_ref() {
561 if c.is_cased() {
562 s.extend(c.to_titlecase());
563 break 'until_first_cased_char;
564 } else {
565 s.push(c);
566 }
567 }
568
569 for (i, c) in chars {
570 if c == 'Σ' {
571 // Σ maps to σ, except at the end of a word where it maps to ς.
572 // This is the only conditional (contextual) but language-independent mapping
573 // in `SpecialCasing.txt`,
574 // so hard-code it rather than have a generic "condition" mechanism.
575 // See https://github.com/rust-lang/rust/issues/26035
576 let sigma_lowercase = map_uppercase_sigma(self, i);
577 s.push(sigma_lowercase);
578 } else {
579 match conversions::to_lower(c) {
580 [a, '\0', _] => s.push(a),
581 [a, b, '\0'] => {
582 s.push(a);
583 s.push(b);
584 }
585 [a, b, c] => {
586 s.push(a);
587 s.push(b);
588 s.push(c);
589 }
590 }
591 }
592 }
593
594 s
595 }
596
597 /// Returns the uppercase equivalent of this string slice, as a new [`String`].
598 ///
599 /// 'Uppercase' is defined according to the terms of
600 /// [Chapter 3 (Conformance)](https://www.unicode.org/versions/latest/core-spec/chapter-3/#G34431)
601 /// of the Unicode standard.
602 ///
603 /// Since some characters can expand into multiple characters when changing
604 /// the case, this function returns a [`String`] instead of modifying the
605 /// parameter in-place.
606 ///
607 /// Like [`char::to_uppercase()`] this method does not handle language-specific
608 /// casing, like Turkish and Azeri I/ı/İ/i. See that method's documentation
609 /// for more information.
610 ///
611 /// # Examples
612 ///
613 /// Basic usage:
614 ///
615 /// ```
616 /// let s = "hello world";
617 ///
618 /// assert_eq!("HELLO WORLD", s.to_uppercase());
619 /// ```
620 ///
621 /// Scripts without case are not changed:
622 ///
623 /// ```
624 /// let new_year = "农历新年";
625 ///
626 /// assert_eq!(new_year, new_year.to_uppercase());
627 /// ```
628 ///
629 /// One character can become multiple:
630 /// ```
631 /// let s = "tschüß";
632 ///
633 /// assert_eq!("TSCHÜSS", s.to_uppercase());
634 /// ```
635 #[cfg(not(no_global_oom_handling))]
636 #[rustc_allow_incoherent_impl]
637 #[must_use = "this returns the uppercase string as a new String, \
638 without modifying the original"]
639 #[stable(feature = "unicode_case_mapping", since = "1.2.0")]
640 pub fn to_uppercase(&self) -> String {
641 // SAFETY: `to_ascii_uppercase` preserves ASCII bytes, so the converted
642 // prefix remains valid UTF-8.
643 let (mut s, rest) = unsafe { convert_while_ascii(self, u8::to_ascii_uppercase) };
644
645 for c in rest.chars() {
646 match conversions::to_upper(c) {
647 [a, '\0', _] => s.push(a),
648 [a, b, '\0'] => {
649 s.push(a);
650 s.push(b);
651 }
652 [a, b, c] => {
653 s.push(a);
654 s.push(b);
655 s.push(c);
656 }
657 }
658 }
659 s
660 }
661
662 /// Returns the case-folded equivalent of this string slice, as a new [`String`].
663 ///
664 /// Case folding is a transformation, mostly matching lowercase, that is meant to be used
665 /// for case-insensitive string comparisons. Case-folded strings should not usually
666 /// be exposed directly to users.
667 ///
668 /// For the precise specification of case folding, see
669 /// [Chapter 3 (Conformance)](https://www.unicode.org/versions/latest/core-spec/chapter-3/#G63737)
670 /// of the Unicode standard.
671 ///
672 /// Since some characters can expand into multiple characters when case folding,
673 /// this function returns a [`String`] instead of modifying the parameter in-place.
674 ///
675 /// No [normalization] (e.g. NFC) is performed, so visually and semantically identical strings
676 /// might still casefold differently. For example, `"Å"` (U+00C5 LATIN CAPITAL LETTER A WITH RING ABOVE)
677 /// is considered distinct from `"Å"` (A followed by U+030A COMBINING RING ABOVE),
678 /// even though Unicode considers them canonically equivalent.
679 ///
680 /// Like [`char::to_casefold_unnormalized()`] this method does not handle language-specific
681 /// casing, like Turkish and Azeri I/ı/İ/i. See that method's documentation
682 /// for more information.
683 ///
684 /// # Examples
685 ///
686 /// Basic usage:
687 ///
688 /// ```
689 /// #![feature(casefold)]
690 /// let s0 = "HELLO";
691 /// let s1 = "Hello";
692 ///
693 /// assert_eq!(s0.to_casefold_unnormalized(), s1.to_casefold_unnormalized());
694 /// assert_eq!(s0.to_casefold_unnormalized(), "hello")
695 /// ```
696 ///
697 /// Scripts without case are not changed:
698 ///
699 /// ```
700 /// #![feature(casefold)]
701 /// let new_year = "农历新年";
702 ///
703 /// assert_eq!(new_year, new_year.to_casefold_unnormalized());
704 /// ```
705 ///
706 /// One character can become multiple:
707 ///
708 /// ```
709 /// #![feature(casefold)]
710 /// let s0 = "TSCHÜẞ";
711 /// let s1 = "TSCHÜSS";
712 /// let s2 = "tschüß";
713 ///
714 /// assert_eq!(s0.to_casefold_unnormalized(), s1.to_casefold_unnormalized());
715 /// assert_eq!(s0.to_casefold_unnormalized(), s2.to_casefold_unnormalized());
716 /// assert_eq!(s0.to_casefold_unnormalized(), "tschüss");
717 /// ```
718 ///
719 /// No NFC [normalization] is performed:
720 ///
721 /// ```rust
722 /// #![feature(casefold)]
723 /// // These two strings are visually and semantically identical...
724 /// let comp = "Å";
725 /// let decomp = "Å";
726 ///
727 /// // ... but not codepoint-for-codepoint equal.
728 /// assert_eq!(comp, "\u{C5}");
729 /// assert_eq!(decomp, "A\u{030A}");
730 ///
731 /// // Their case-foldings are likewise unequal:
732 /// assert_eq!(comp.to_casefold_unnormalized(), "\u{E5}");
733 /// assert_eq!(decomp.to_casefold_unnormalized(), "a\u{030A}");
734 /// ```
735 ///
736 /// [normalization]: https://www.unicode.org/faq/normalization.html
737 #[cfg(not(no_global_oom_handling))]
738 #[rustc_allow_incoherent_impl]
739 #[must_use = "this returns the case-folded string as a new String, \
740 without modifying the original"]
741 #[unstable(feature = "casefold", issue = "154742")]
742 pub fn to_casefold_unnormalized(&self) -> String {
743 // SAFETY: `to_ascii_lowercase` preserves ASCII bytes, so the converted
744 // prefix remains valid UTF-8.
745 let (mut s, rest) = unsafe { convert_while_ascii(self, u8::to_ascii_lowercase) };
746
747 for c in rest.chars() {
748 match conversions::to_casefold(c) {
749 [a, '\0', _] => s.push(a),
750 [a, b, '\0'] => {
751 s.push(a);
752 s.push(b);
753 }
754 [a, b, c] => {
755 s.push(a);
756 s.push(b);
757 s.push(c);
758 }
759 }
760 }
761 s
762 }
763
764 /// Converts a [`Box<str>`] into a [`String`] without copying or allocating.
765 ///
766 /// # Examples
767 ///
768 /// ```
769 /// let string = String::from("birthday gift");
770 /// let boxed_str = string.clone().into_boxed_str();
771 ///
772 /// assert_eq!(boxed_str.into_string(), string);
773 /// ```
774 #[stable(feature = "box_str", since = "1.4.0")]
775 #[rustc_allow_incoherent_impl]
776 #[must_use = "`self` will be dropped if the result is not used"]
777 #[inline]
778 pub fn into_string(self: Box<Self>) -> String {
779 let slice = Box::<[u8]>::from(self);
780 unsafe { String::from_utf8_unchecked(slice.into_vec()) }
781 }
782
783 /// Creates a new [`String`] by repeating a string `n` times.
784 ///
785 /// # Panics
786 ///
787 /// This function will panic if the capacity would overflow.
788 ///
789 /// # Examples
790 ///
791 /// Basic usage:
792 ///
793 /// ```
794 /// assert_eq!("abc".repeat(4), String::from("abcabcabcabc"));
795 /// ```
796 ///
797 /// A panic upon overflow:
798 ///
799 /// ```should_panic
800 /// // this will panic at runtime
801 /// let huge = "0123456789abcdef".repeat(usize::MAX);
802 /// ```
803 #[cfg(not(no_global_oom_handling))]
804 #[rustc_allow_incoherent_impl]
805 #[must_use]
806 #[stable(feature = "repeat_str", since = "1.16.0")]
807 #[inline]
808 pub fn repeat(&self, n: usize) -> String {
809 unsafe { String::from_utf8_unchecked(self.as_bytes().repeat(n)) }
810 }
811
812 /// Returns a copy of this string where each character is mapped to its
813 /// ASCII upper case equivalent.
814 ///
815 /// ASCII letters 'a' to 'z' are mapped to 'A' to 'Z',
816 /// but non-ASCII letters are unchanged.
817 ///
818 /// To uppercase the value in-place, use [`make_ascii_uppercase`].
819 ///
820 /// To uppercase ASCII characters in addition to non-ASCII characters, use
821 /// [`to_uppercase`].
822 ///
823 /// # Examples
824 ///
825 /// ```
826 /// let s = "Grüße, Jürgen ❤";
827 ///
828 /// assert_eq!("GRüßE, JüRGEN ❤", s.to_ascii_uppercase());
829 /// ```
830 ///
831 /// [`make_ascii_uppercase`]: str::make_ascii_uppercase
832 /// [`to_uppercase`]: #method.to_uppercase
833 #[cfg(not(no_global_oom_handling))]
834 #[rustc_allow_incoherent_impl]
835 #[must_use = "to uppercase the value in-place, use `make_ascii_uppercase()`"]
836 #[stable(feature = "ascii_methods_on_intrinsics", since = "1.23.0")]
837 #[inline]
838 pub fn to_ascii_uppercase(&self) -> String {
839 let mut s = self.to_owned();
840 s.make_ascii_uppercase();
841 s
842 }
843
844 /// Returns a copy of this string where each character is mapped to its
845 /// ASCII lower case equivalent.
846 ///
847 /// ASCII letters 'A' to 'Z' are mapped to 'a' to 'z',
848 /// but non-ASCII letters are unchanged.
849 ///
850 /// To lowercase the value in-place, use [`make_ascii_lowercase`].
851 ///
852 /// To lowercase ASCII characters in addition to non-ASCII characters, use
853 /// [`to_lowercase`].
854 ///
855 /// # Examples
856 ///
857 /// ```
858 /// let s = "Grüße, Jürgen ❤";
859 ///
860 /// assert_eq!("grüße, jürgen ❤", s.to_ascii_lowercase());
861 /// ```
862 ///
863 /// [`make_ascii_lowercase`]: str::make_ascii_lowercase
864 /// [`to_lowercase`]: #method.to_lowercase
865 #[cfg(not(no_global_oom_handling))]
866 #[rustc_allow_incoherent_impl]
867 #[must_use = "to lowercase the value in-place, use `make_ascii_lowercase()`"]
868 #[stable(feature = "ascii_methods_on_intrinsics", since = "1.23.0")]
869 #[inline]
870 pub fn to_ascii_lowercase(&self) -> String {
871 let mut s = self.to_owned();
872 s.make_ascii_lowercase();
873 s
874 }
875}
876
877/// Converts a boxed slice of bytes to a boxed string slice without checking
878/// that the string contains valid UTF-8.
879///
880/// # Safety
881///
882/// * The provided bytes must contain a valid UTF-8 sequence.
883///
884/// # Examples
885///
886/// ```
887/// let smile_utf8 = Box::new([226, 152, 186]);
888/// let smile = unsafe { std::str::from_boxed_utf8_unchecked(smile_utf8) };
889///
890/// assert_eq!("☺", &*smile);
891/// ```
892#[stable(feature = "str_box_extras", since = "1.20.0")]
893#[must_use]
894#[inline]
895pub unsafe fn from_boxed_utf8_unchecked(v: Box<[u8]>) -> Box<str> {
896 unsafe { Box::from_raw(Box::into_raw(v) as *mut str) }
897}
898
899/// Converts leading ascii bytes in `s` by calling the `convert` function.
900///
901/// For better average performance, this happens in chunks of `2*size_of::<usize>()`.
902///
903/// Returns a tuple of the converted prefix and the remainder starting from
904/// the first non-ascii character.
905///
906/// This function is only public so that it can be verified in a codegen test,
907/// see `issue-123712-str-to-lower-autovectorization.rs`.
908///
909/// # Safety
910///
911/// `convert` must return an ASCII byte for every ASCII input byte.
912#[unstable(feature = "str_internals", issue = "none")]
913#[doc(hidden)]
914#[inline]
915#[cfg(not(no_global_oom_handling))]
916pub unsafe fn convert_while_ascii(s: &str, convert: fn(&u8) -> u8) -> (String, &str) {
917 // Process the input in chunks of 16 bytes to enable auto-vectorization.
918 // Previously the chunk size depended on the size of `usize`,
919 // but on 32-bit platforms with sse or neon is also the better choice.
920 // The only downside on other platforms would be a bit more loop-unrolling.
921 const N: usize = 16;
922
923 let mut slice = s.as_bytes();
924 let mut out = Vec::with_capacity(slice.len());
925 let mut out_slice = out.spare_capacity_mut();
926
927 let mut ascii_prefix_len = 0_usize;
928 let mut is_ascii = [false; N];
929
930 while slice.len() >= N {
931 // SAFETY: checked in loop condition
932 let chunk = unsafe { slice.get_unchecked(..N) };
933 // SAFETY: out_slice has at least same length as input slice and gets sliced with the same offsets
934 let out_chunk = unsafe { out_slice.get_unchecked_mut(..N) };
935
936 for j in 0..N {
937 is_ascii[j] = chunk[j] <= 127;
938 }
939
940 // Auto-vectorization for this check is a bit fragile, sum and comparing against the chunk
941 // size gives the best result, specifically a pmovmsk instruction on x86.
942 // See https://github.com/llvm/llvm-project/issues/96395 for why llvm currently does not
943 // currently recognize other similar idioms.
944 if is_ascii.iter().map(|x| *x as u8).sum::<u8>() as usize != N {
945 break;
946 }
947
948 for j in 0..N {
949 out_chunk[j] = MaybeUninit::new(convert(&chunk[j]));
950 }
951
952 ascii_prefix_len += N;
953 slice = unsafe { slice.get_unchecked(N..) };
954 out_slice = unsafe { out_slice.get_unchecked_mut(N..) };
955 }
956
957 // handle the remainder as individual bytes
958 while slice.len() > 0 {
959 let byte = slice[0];
960 if byte > 127 {
961 break;
962 }
963 // SAFETY: out_slice has at least same length as input slice
964 unsafe {
965 *out_slice.get_unchecked_mut(0) = MaybeUninit::new(convert(&byte));
966 }
967 ascii_prefix_len += 1;
968 slice = unsafe { slice.get_unchecked(1..) };
969 out_slice = unsafe { out_slice.get_unchecked_mut(1..) };
970 }
971
972 unsafe {
973 // SAFETY: ascii_prefix_len bytes have been initialized above
974 out.set_len(ascii_prefix_len);
975
976 // SAFETY: We have written only valid ascii to the output vec
977 let ascii_string = String::from_utf8_unchecked(out);
978
979 // SAFETY: we know this is a valid char boundary
980 // since we only skipped over leading ascii bytes
981 let rest = core::str::from_utf8_unchecked(slice);
982
983 (ascii_string, rest)
984 }
985}
986#[inline]
987#[cfg(not(no_global_oom_handling))]
988#[allow(dead_code)]
989/// Faster implementation of string replacement for ASCII to ASCII cases.
990/// Should produce fast vectorized code.
991unsafe fn replace_ascii(utf8_bytes: &[u8], from: u8, to: u8) -> String {
992 let result: Vec<u8> = utf8_bytes.iter().map(|b| if *b == from { to } else { *b }).collect();
993 // SAFETY: We replaced ascii with ascii on valid utf8 strings.
994 unsafe { String::from_utf8_unchecked(result) }
995}