aquatic http load test: register response types

aquatic_http_load_test/src/network.rs

@@ -81,31 +81,11 @@ impl Connection {
         state.statistics.bytes_received
             .fetch_add(self.bytes_read, Ordering::SeqCst);
 
-        let res_response = Response::from_bytes(
+        let interesting_bytes = &self.read_buffer[..self.bytes_read];
-            &self.read_buffer[..self.bytes_read]
-        );
 
         self.bytes_read = 0;
 
-        match res_response {
+        Self::register_response_type(state, interesting_bytes);
-            Ok(Response::Announce(_)) => {
-                state.statistics.responses_announce
-                    .fetch_add(1, Ordering::SeqCst);
-            },
-            Ok(Response::Scrape(_)) => {
-                state.statistics.responses_scrape
-                    .fetch_add(1, Ordering::SeqCst);
-            },
-            Ok(Response::Failure(_)) => {
-                state.statistics.responses_failure
-                    .fetch_add(1, Ordering::SeqCst);
-            },
-            Err(err) => {
-                eprintln!("response from bytes error: {}", err);
-
-                return;
-            }
-        }
 
         self.send_request(
             config,
@@ -114,6 +94,35 @@ impl Connection {
         );
     }
 
+    /// Ultra-crappy byte searches to determine response type with some degree
+    /// of certainty.
+    fn register_response_type(
+        state: &LoadTestState,
+        response_bytes: &[u8],
+    ){
+        for chunk in response_bytes.windows(12){
+            if chunk == b"e8:intervali" {
+                state.statistics.responses_announce.fetch_add(1, Ordering::SeqCst);
+
+                return;
+            }
+        }
+        for chunk in response_bytes.windows(9){
+            if chunk == b"d5:filesd" {
+                state.statistics.responses_scrape.fetch_add(1, Ordering::SeqCst);
+
+                return;
+            }
+        }
+        for chunk in response_bytes.windows(18){
+            if chunk == b"d14:failure_reason" {
+                state.statistics.responses_failure.fetch_add(1, Ordering::SeqCst);
+
+                return;
+            }
+        }
+    }
+
     pub fn send_request(
         &mut self,
         config: &Config,