From 4d93229f190bf0651ff2202eb59ca83120e8746a Mon Sep 17 00:00:00 2001
From: Valentyn Faychuk <valy@detee.ltd>
Date: Sat, 7 Dec 2024 22:55:50 +0200
Subject: [PATCH] precise network attacks metric

---
 src/grpc/client.rs | 8 ++++++--
 src/grpc/server.rs | 7 ++++---
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/grpc/client.rs b/src/grpc/client.rs
index fc700b1..418ad5b 100644
--- a/src/grpc/client.rs
+++ b/src/grpc/client.rs
@@ -89,7 +89,9 @@ impl ConnManager {
         let rx_stream = BroadcastStream::new(rx).filter_map(|n| n.ok());
         let response = client.get_updates(rx_stream).await.map_err(|e| {
             println!("Error connecting to {node_ip}: {e}");
-            self.state.increase_net_attacks();
+            if e.to_string().contains("QuoteVerifyError") {
+                self.state.increase_net_attacks();
+            }
             e
         })?;
         let mut resp_stream = response.into_inner();
@@ -159,7 +161,9 @@ pub async fn key_grabber(
 
     let response = client.get_keys(tonic::Request::new(Empty {})).await.map_err(|e| {
         println!("Error getting keys from {node_ip}: {e}");
-        state.increase_net_attacks();
+        if e.to_string().contains("QuoteVerifyError") {
+            state.increase_net_attacks();
+        }
         e
     })?;
     Ok(response.into_inner())
diff --git a/src/grpc/server.rs b/src/grpc/server.rs
index 338721f..a233d54 100644
--- a/src/grpc/server.rs
+++ b/src/grpc/server.rs
@@ -55,7 +55,6 @@ impl MyServer {
                 Ok(incoming) => incoming,
                 Err(e) => {
                     println!("Error accepting connection: {}", e);
-                    state.increase_net_attacks();
                     continue;
                 }
             };
@@ -79,8 +78,10 @@ impl MyServer {
                     .await;
 
                 let conn = if let Err(e) = conn {
-                    println!("Error accepting TLS connection: {}", e);
-                    state.increase_net_attacks();
+                    println!("Error accepting TLS connection: {e}");
+                    if e.to_string().contains("HandshakeFailure") {
+                        state.increase_net_attacks();
+                    }
                     return;
                 } else {
                     conn.unwrap()