commit 346472c6be39db0236d606d4215d4f93d12451ef
author Ayush Singh <ayushdevel1325@gmail.com> Wed Oct 25 21:48:41 2023 +0530
committer Ayush Singh <ayushdevel1325@gmail.com> Sat Jan 20 01:46:02 2024 +0530
tree eeb15a7227d2fbb098db4ce4200167ff4eb1346a
parent 07000dc244cbd4fc1f9850618edd2952d65a38bd

Use heap for stdout and stderr

Signed-off-by: Ayush Singh <ayushdevel1325@gmail.com>

library/std/src/sys/pal/uefi/stdio.rs

diff --git a/library/std/src/sys/pal/uefi/stdio.rs b/library/std/src/sys/pal/uefi/stdio.rs
index 7f384bc..9adccee 100644
--- a/library/std/src/sys/pal/uefi/stdio.rs
+++ b/library/std/src/sys/pal/uefi/stdio.rs
@@ -4,8 +4,6 @@
 use crate::os::uefi;
 use crate::ptr::NonNull;
 
-const MAX_BUFFER_SIZE: usize = 8192;
-
 pub struct Stdin {
     pending: Option<char>,
 }
@@ -111,19 +109,14 @@
     protocol: *mut r_efi::protocols::simple_text_output::Protocol,
     buf: &[u8],
 ) -> io::Result<usize> {
-    let mut utf16 = [0; MAX_BUFFER_SIZE / 2];
-
     // Get valid UTF-8 buffer
     let utf8 = match crate::str::from_utf8(buf) {
         Ok(x) => x,
         Err(e) => unsafe { crate::str::from_utf8_unchecked(&buf[..e.valid_up_to()]) },
     };
-    // Clip UTF-8 buffer to max UTF-16 buffer we support
-    let utf8 = &utf8[..utf8.floor_char_boundary(utf16.len() - 1)];
 
-    for (i, ch) in utf8.encode_utf16().enumerate() {
-        utf16[i] = ch;
-    }
+    let mut utf16: Vec<u16> = utf8.encode_utf16().collect();
+    utf16.push(0);
 
     unsafe { simple_text_output(protocol, &mut utf16) }?;