diff options
| author | Benoît du Garreau <bdgdlm@outlook.com> | 2024-03-13 15:09:39 +0100 |
|---|---|---|
| committer | Benoît du Garreau <bdgdlm@outlook.com> | 2024-04-12 09:43:39 +0200 |
| commit | b07c1f7f4d46f083725a03f85a3c6cd3447267ae (patch) | |
| tree | f17e2ec4615f2852c01f09a4468943b692260f12 /library/std/src/io/impls.rs | |
| parent | 6475796a813f2153d3bfc92c6089ed0f0c1c5b46 (diff) | |
| download | rust-b07c1f7f4d46f083725a03f85a3c6cd3447267ae.tar.gz rust-b07c1f7f4d46f083725a03f85a3c6cd3447267ae.zip | |
Improve several `Read` implementations
Diffstat (limited to 'library/std/src/io/impls.rs')
| -rw-r--r-- | library/std/src/io/impls.rs | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/library/std/src/io/impls.rs b/library/std/src/io/impls.rs index dd7e0725176..9f9ee4af5c8 100644 --- a/library/std/src/io/impls.rs +++ b/library/std/src/io/impls.rs @@ -329,8 +329,9 @@ impl Read for &[u8] { #[inline] fn read_to_string(&mut self, buf: &mut String) -> io::Result<usize> { let content = str::from_utf8(self).map_err(|_| io::Error::INVALID_UTF8)?; - buf.push_str(content); let len = self.len(); + buf.try_reserve(len)?; + buf.push_str(content); *self = &self[len..]; Ok(len) } @@ -478,6 +479,7 @@ impl<A: Allocator> Read for VecDeque<u8, A> { let len = self.len(); let content = self.make_contiguous(); let string = str::from_utf8(content).map_err(|_| io::Error::INVALID_UTF8)?; + buf.try_reserve(len)?; buf.push_str(string); self.clear(); Ok(len) |