feat(protocol): add initial parser for common message header

src/connection.rs

@@ -4,6 +4,7 @@ use tokio::io::{AsyncReadExt, AsyncWriteExt};
 use tokio::net::{TcpStream};
 use std::io;
 
+use crate::protocol::{parce_header, ParseError}; 
 pub struct Connection {
     id: u64,
     client_addr: SocketAddr,
@@ -25,48 +26,72 @@ impl Connection {
         }
     }
 
-    pub async fn run(mut self) -> io::Result<()> {
-        println!(
-            "Обработчик {} создан для {}",
-            self.id,
-            self.client_addr
-        );
+   pub async fn run(mut self) -> io::Result<()> {
+    println!(
+        "Обработчик {} создан для {}",
+        self.id, self.client_addr
+    );
 
-        let mut buffer = [0_u8; 1024];
+    // Буфер для данных тела сообщения (теперь он не хранит заголовок)
+    let mut body_buffer = vec![0_u8; 1024];
 
-        loop {
-            let bytes_read = self.stream.read(&mut buffer).await?;
+    loop {
+        // 1. Создаем буфер СТРОГО под размер заголовка
+        let mut header_buffer = [0_u8; 20];
 
-            if bytes_read == 0 {
+        // 2. Читаем из сети РОВНО 20 байт
+        // Если клиент закроет соединение до того, как пришлет 20 байт,
+        // read_exact вернет ошибку UnexpectedEof.
+        if let Err(e) = self.stream.read_exact(&mut header_buffer).await {
+            if e.kind() == io::ErrorKind::UnexpectedEof {
                 println!(
-                    "Обработчик {}: клиент {} отключился",
-                    self.id,
-                    self.client_addr
+                    "Обработчик {}: клиент {} отключился (или прислал неполный заголовок)",
+                    self.id, self.client_addr
                 );
-
                 return Ok(());
             }
-
-            self.message_count += 1;
-
-            let received =
-                String::from_utf8_lossy(&buffer[..bytes_read]);
-
-            println!(
-                "Обработчик {}: сообщение №{}: {:?}",
-                self.id,
-                self.message_count,
-                received
-            );
-
-            let response = format!(
-                "Обработчик {}: сообщение №{}: {}\n",
-                self.id,
-                self.message_count,
-                received
-            );
-
-            self.stream.write_all(response.as_bytes()).await?;
+            return Err(e); // Другие системные ошибки ввода-вывода (например, обрыв связи)
         }
+
+        // 3. Вызываем ваш парсер заголовка
+        let header = match parce_header(&header_buffer).await {
+            Ok(h) => h,
+            Err(ParseError::InvalidFormat) => {
+                println!("Обработчик {}: Критическая ошибка! Неверный формат заголовка.", self.id);
+                // В сетевых протоколах при неверном заголовке соединение обычно сразу рвут
+                return Err(io::Error::new(io::ErrorKind::InvalidData, "Invalid header format"));
+            }
+            // Если вы добавите новые ошибки, например FilenameTooLong:
+            Err(e) => {
+                println!("Обработчик {}: Ошибка парсинга: {}", self.id, e);
+                return Err(io::Error::new(io::ErrorKind::InvalidData, e.to_string()));
+            }
+        };
+
+        self.message_count += 1;
+        println!("Обработчик {}: Успешно распарсен заголовок №{}: {:?}", self.id, self.message_count, header);
+
+        // 4. Используем данные из распарсенного заголовка!
+        // Теперь мы знаем точный размер тела сообщения благодаря header.body_size
+        let body_len = header.body_size() as usize;
+        
+        // Подгоняем размер буфера под размер тела (чтобы не читать лишнего)
+        if body_len > body_buffer.len() {
+            body_buffer.resize(body_len, 0);
+        }
+
+        // Читаем из сети тело сообщения (ровно столько байт, сколько указано в заголовке)
+        self.stream.read_exact(&mut body_buffer[..body_len]).await?;
+
+        let received_body = String::from_utf8_lossy(&body_buffer[..body_len]);
+        println!("Обработчик {}: Получено тело сообщения: {:?}", self.id, received_body);
+
+        // 5. Отправляем ответ клиенту
+        let response = format!(
+            "Обработчик {}: Заголовок валиден (cmd={}). Тело принято.\n",
+            self.id, header.command()
+        );
+        self.stream.write_all(response.as_bytes()).await?;
     }
+}
 }