Fix: IP Address Validation

config.yml

@@ -16,6 +16,6 @@ message: #Optional during receive
 file: #Optional during receive
 - "./dev_build.sh"
 debug: 1 #Compulsory. 0 is for off and 1 and above for on
-blacklists:
-- 127.0.0.1
-- 34.138.139.178
+# blacklists:
+# - 127.0.0.1
+# - 34.138.139.178

npm/package.json

@@ -1,13 +1,13 @@
 {
   "name": "@onboardbase/secure-share",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "Share anything with teammates across machines via CLI",
   "scripts": {
     "test": "jest",
     "postinstall": "node ./install.js"
   },
   "bin": {
-    "share": "run.js"
+    "scs": "run.js"
   },
   "repository": {
     "type": "git",

share/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "scs"
-version = "0.1.1"
+version = "0.1.2"
 edition = "2021"
 default-run = "scs"
 description = "Open source p2p share for devs to share anything with teammates across machines securely."

share/src/main.rs

@@ -14,7 +14,7 @@ mod network;
 #[derive(Parser, Debug)]
 #[command(name = "scs")]
 #[command(author = "Onboardbase. <onboardbase.com>")]
-#[command(version = "0.1.1")]
+#[command(version = "0.1.2")]
 #[command(about = "Share anything with teammates across machines via CLI.", long_about = None)]
 pub struct Cli {
     /// Separated list of secrets to share. Key-Value pair is seperated by a comma. "my_key,my_value"

share/src/network/hole_puncher.rs

@@ -141,6 +141,7 @@
         }
     }
     let mut connection_deets = ConnectionDetails::new();
+
     block_on(async {
         loop {
             match swarm.next().await.unwrap() {
@@ -167,19 +168,27 @@
                         // println!("deetttss {:#?}", connection_id);
                         error!("This IP address is present in your blacklist.");
                         swarm.close_connection(connection_id);
+                        exit(1);
                     }
 
                     if !is_ip_whitelisted(&event, &config) {
+                        error!("This IP address is not present in your whitleist.");
                         swarm.close_connection(connection_id);
+                        exit(1);
                     }
                 }
                 SwarmEvent::Behaviour(Event::Ping(_)) => {}
                 SwarmEvent::IncomingConnection { connection_id, .. } => {
-                    connection_deets.save_id(connection_id);
+                    debug!("INCOMING CONNECTION: {:?}", connection_id);
                 }
                 SwarmEvent::ConnectionEstablished {
-                    peer_id, endpoint, ..
+                    peer_id,
+                    endpoint,
+                    connection_id,
+                    ..
                 } => {
+                    connection_deets.save_id(connection_id);
+
                     let addr = endpoint.get_remote_address();
                     info!("Established connection to {peer_id} via {addr}");