src/grpc/server.rs

@@ -93,10 +93,7 @@ impl NodeServer {
 
                 let conn = if let Err(e) = conn {
                     println!("Error accepting TLS connection: {e}");
-                    let attack_error_messages = ["handshake", "certificate", "quote"];
-
-                    let err_str = e.to_string().to_lowercase();
-                    if attack_error_messages.iter().any(|att_er_str| err_str.contains(att_er_str)) {
+                    if e.to_string().contains("HandshakeFailure") {
                         state.increase_net_attacks().await;
                     }
                     return;
@@ -200,7 +197,6 @@ impl Update for NodeServer {
                             yield Ok(update.update);
                         }
                         // TODO: check if disconnect client if too many connections are active
-                        // Its tested and working
                         if tx.receiver_count() > 9 {
                             error_status = Status::internal("Already have too many clients. Connect to another server.");
                             break;

src/main.rs

@@ -55,9 +55,7 @@ pub async fn heartbeat(
         state.remove_inactive_nodes().await;
         let connected_ips = state.get_connected_ips().await;
         println!("Connected nodes ({}): {:?}", connected_ips.len(), connected_ips);
-        let my_node_info = state.get_my_info().await;
-        my_node_info.save();
-        let _ = tx.send((state.get_my_ip().await, my_node_info).into());
+        let _ = tx.send((state.get_my_ip().await, state.get_my_info().await).into());
         if connected_ips.len() < NUM_CONNECTIONS {
             if let Some(node_ip) = state.get_random_disconnected_ip().await {
                 println!("Dialing random node {}", node_ip);