Use validating stream

src/main/java/BigIntPoint.java

@@ -6,8 +6,7 @@ public class BigIntPoint implements Serializable
     BigInteger x;
     BigInteger y;
 
-    public BigIntPoint(BigInteger x, BigInteger y)
-    {
+    public BigIntPoint(BigInteger x, BigInteger y) {
         this.x = x;
         this.y = y;
     }

src/main/java/EncryptedPathsComparison.java

@@ -127,10 +127,10 @@ public boolean encryptedOnSegment (BigIntPoint p, BigIntPoint q, BigIntPoint r)
 
             z =  a && b && c && d;
 
-        } catch (HomomorphicException | IOException | ClassNotFoundException e){
+        }
+        catch (HomomorphicException | IOException | ClassNotFoundException e){
             System.err.println("Exception in OnSegment" + e.getMessage());
         }
-
         return z;
     }
 

src/main/java/PathsAlice.java

@@ -1,3 +1,4 @@
+import org.apache.commons.io.serialization.ValidatingObjectInputStream;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import security.misc.HomomorphicException;
@@ -6,7 +7,6 @@
 
 import java.io.File;
 import java.io.IOException;
-import java.io.ObjectInputStream;
 import java.math.BigInteger;
 import java.net.Socket;
 import java.util.ArrayList;
@@ -51,7 +51,7 @@ public static void main(String[] args) {
             Socket socket = new Socket (ip_address, pathsalice.port);
             alice.set_socket(socket);
             alice.receivePublicKeys();
-            ObjectInputStream input = new ObjectInputStream(socket.getInputStream());
+            ValidatingObjectInputStream input = get_ois(socket);
             Object object = input.readObject();
 
             if (object instanceof List<?>) {
@@ -80,4 +80,18 @@ public static void main(String[] args) {
             logger.fatal(e.getStackTrace());
         }
     }
+
+    public static ValidatingObjectInputStream get_ois(Socket socket) throws IOException {
+        ValidatingObjectInputStream ois = new ValidatingObjectInputStream(socket.getInputStream());
+        ois.accept(
+                java.util.List.class,
+                BigIntPoint.class,
+                java.lang.Number.class,
+                java.math.BigInteger.class
+        );
+        ois.accept("[B");
+        ois.accept("[L*");
+        return ois;
+    }
+
 }

src/test/java/IntersectTest.java

@@ -1,3 +1,5 @@
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
 import org.junit.Test;
 import org.junit.Before;
 import static org.junit.Assert.assertEquals;
@@ -21,6 +23,7 @@
 import java.util.List;
 
 public class IntersectTest {
+    private static final Logger logger = LogManager.getLogger(IntersectTest.class);
 
     private static KeyPair dgk = null;
     private static KeyPair paillier = null;
@@ -69,11 +72,11 @@ public void encrypted_test_intersections() {
                 System.out.println("Reading the file: " + ownroute);
                 boolean will_collide = Boolean.parseBoolean(expected_result);
 
-                //Parsing routes
+                // Parsing routes
                 List<BigIntPoint> ownroute_list = CleartextPathsComparison.read_all_paths(ownroute);
                 List<BigIntPoint> cryptroute_list = CleartextPathsComparison.read_all_paths(cryptroute);
 
-                //encrypt routes
+                // Encrypt routes
                 List<BigIntPoint> encryptedownroute_list = new ArrayList<>();
                 List<BigIntPoint> encryptedcryptroute_list = new ArrayList<>();
 
@@ -111,7 +114,7 @@ public void encrypted_test_intersections() {
             }
         }
         catch (IOException | ClassNotFoundException | HomomorphicException | InterruptedException e) {
-            System.err.println("Error reading files" + e.getMessage());
+            logger.fatal(e.getStackTrace());
         }
     }
 
@@ -129,12 +132,14 @@ public static Socket connectWithRetry(String host, int port) {
                 // Connection refused, print a message and retry after a delay
                 System.out.println("Connection refused. Retrying in 1 second...");
                 try {
-                    Thread.sleep(1000); // Wait for 5 seconds before retrying
-                } catch (InterruptedException ex) {
-                    ex.printStackTrace();
+                    Thread.sleep(1000); // Wait for 1 second before retrying
+                }
+                catch (InterruptedException ex) {
+                    logger.info(ex.getStackTrace());
                 }
-            } catch (Exception e) {
-                e.printStackTrace();
+            }
+            catch (Exception e) {
+                logger.fatal(e.getStackTrace());
             }
         }
     }