From d7a0f3d526e7f213c3de54816feefb1a60b5a474 Mon Sep 17 00:00:00 2001 From: Tom Almeida <tommoa256@gmail.com> Date: Tue, 3 Jul 2018 12:36:41 +0800 Subject: [PATCH] Ensure gets stops on newline or bufchar --- src/stdio/src/lib.rs | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/stdio/src/lib.rs b/src/stdio/src/lib.rs index 4c00df7c..d0b38f94 100644 --- a/src/stdio/src/lib.rs +++ b/src/stdio/src/lib.rs @@ -351,6 +351,9 @@ pub extern "C" fn fgets(s: *mut c_char, n: c_int, stream: &mut FILE) -> *mut c_c if let Some((rpos, rend)) = stream.read { for _ in (0..(n-1) as usize).take_while(|x| rpos + x < rend) { st[diff] = stream.buf[rpos + diff] as i8; + if st[diff] == b'\n' as i8 || st[diff] == stream.buf_char { + break; + } diff += 1; } stream.read = Some((rpos+diff, rend)); -- GitLab