fix: handle invalid ip address

messede-degod · Mar 3, 2024 · 50e1762 · 50e1762
1 parent 98ce245
commit 50e1762
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main/java/com/segfault/sstableMigrator/App.java b/src/main/java/com/segfault/sstableMigrator/App.java
@@ -17,6 +17,7 @@
 import com.maxmind.db.MaxMindDbParameter;
 import com.maxmind.db.Reader;
 import java.net.InetAddress;
+import java.net.UnknownHostException;
 import java.util.Arrays;
 import java.util.HashMap;
 import java.util.Map;
@@ -243,7 +244,13 @@ public void parseAndInsertCSV(String csvString) throws IOException, SkippedEntry
         InetAddress ip16 = null;
         InetAddress ip24 = null;
 
+        try{
         parsedIpAddress = InetAddress.getByName(ipStr);
+        }catch(UnknownHostException e){
+            System.out.println("Invalid Ip Address: "+csvString);
+            return;
+        }
+
         ip8 = App.getIPBlock(parsedIpAddress, (short) 8);
         ip16 = App.getIPBlock(parsedIpAddress, (short) 16);
         ip24 = App.getIPBlock(parsedIpAddress, (short) 24);