v0.2.6: preserve buffered lines and add request read timeouts

1 files changed, 13 insertions, 13 deletions

diff --git a/src/bin/server/tls.rs b/src/bin/server/tls.rs
index acf3315..62a25db 100644
--- a/src/bin/server/tls.rs
+++ b/src/bin/server/tls.rs
@@ -1,6 +1,8 @@
 use std::io::{Read, Write};
 use std::net::TcpStream;
 
+use zagan_bet::protocol::LineBuffer;
+
 #[derive(Clone)]
 pub struct TlsServer {
     acceptor: native_tls::TlsAcceptor,
@@ -42,27 +44,25 @@ impl TlsServer {
 
 pub struct TlsStream {
     stream: native_tls::TlsStream<TcpStream>,
+    lines: LineBuffer,
 }
 
 impl TlsStream {
     pub fn new(stream: native_tls::TlsStream<TcpStream>) -> Self {
-        Self { stream }
+        Self {
+            stream,
+            lines: LineBuffer::new(),
+        }
     }
 
     pub fn read_line(&mut self, s: &mut String) -> std::io::Result<usize> {
-        let mut buf = [0u8; 1024];
-        loop {
-            match self.stream.read(&mut buf) {
-                Ok(0) => return Ok(0),
-                Ok(n) => {
-                    let chunk = std::str::from_utf8(&buf[..n]).unwrap_or("");
-                    s.push_str(chunk);
-                    if s.contains('\n') {
-                        return Ok(n);
-                    }
-                }
-                Err(e) => return Err(e),
+        match self.lines.read_line(&mut self.stream)? {
+            Some(line) => {
+                s.push_str(&line);
+                s.push('\n');
+                Ok(line.len() + 1)
             }
+            None => Ok(0),
         }
     }