From d87134340a67318678ef6a7b3adef931bbb136c3 Mon Sep 17 00:00:00 2001 From: Paul Makles Date: Mon, 5 Jun 2023 13:23:10 +0100 Subject: [PATCH] refactor: use &[u8] not Vec --- src/lib.rs | 6 +++--- tests/clamav_client.rs | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/lib.rs b/src/lib.rs index 55261af..de48a0d 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -50,7 +50,7 @@ where Ok(response) } -fn scan_buffer(buffer: Vec, mut stream: RW, chunk_size: Option) -> IoResult +fn scan_buffer(buffer: &[u8], mut stream: RW, chunk_size: Option) -> IoResult where RW: Read + Write, { @@ -92,7 +92,7 @@ where } #[cfg(target_family = "unix")] -pub fn scan_buffer_socket

(buffer: Vec, socket_path: P, chunk_size: Option) -> IoResult +pub fn scan_buffer_socket

(buffer: &[u8], socket_path: P, chunk_size: Option) -> IoResult where P: AsRef, { @@ -119,7 +119,7 @@ where scan(file_path, chunk_size, stream) } -pub fn scan_buffer_tcp(buffer: Vec, host_address: A, chunk_size: Option) -> IoResult +pub fn scan_buffer_tcp(buffer: &[u8], host_address: A, chunk_size: Option) -> IoResult where A: ToSocketAddrs, { diff --git a/tests/clamav_client.rs b/tests/clamav_client.rs index e2e0870..6624016 100644 --- a/tests/clamav_client.rs +++ b/tests/clamav_client.rs @@ -66,7 +66,7 @@ fn test_scan_buffer_tcp() { .bytes() .collect::>(); - let response = scan_buffer_tcp(buffer, TEST_HOST_ADDRESS, None).expect(&err_msg); + let response = scan_buffer_tcp(&buffer, TEST_HOST_ADDRESS, None).expect(&err_msg); assert_eq!(&response, EICAR_BUFFER_FOUND_RESPONSE); let is_clean = clean(&response);