Getting pairs of packets from the client-server communications.

[pingpong.git] / Code / Projects / SmartPlugDetector / src / main / java / edu / uci / iotproject / Main.java

diff --git a/Code/Projects/SmartPlugDetector/src/main/java/edu/uci/iotproject/Main.java b/Code/Projects/SmartPlugDetector/src/main/java/edu/uci/iotproject/Main.java
index 9696bd6d56053e2d34a576e042a08b77a00fe2a6..c94f335e6395d7cf326bbfa251b3d72c28b07364 100644 (file)
--- a/Code/Projects/SmartPlugDetector/src/main/java/edu/uci/iotproject/Main.java
+++ b/Code/Projects/SmartPlugDetector/src/main/java/edu/uci/iotproject/Main.java
@@ -25,33 +25,33 @@ public class Main {
     public static void main(String[] args) throws PcapNativeException, NotOpenException, EOFException, TimeoutException, UnknownHostException {
         // -------------------------------------------------------------------------------------------------------------
         // Example/debug code for searching for a pattern at the MAC layer.
-        String fileName = "./pcap/mac-tplink.local.pcapng";
-        PcapHandle handle;
-        try {
-            handle = Pcaps.openOffline(fileName, PcapHandle.TimestampPrecision.NANO);
-        } catch (PcapNativeException pne) {
-            handle = Pcaps.openOffline(fileName);
-        }
-//        Arrays.asList(1590, 1590, 1590, 1001, 337, 197, 636, 1311, 177) // Full pattern (all non-zero payload packets).
-        MacLayerFlowPattern pattern = new MacLayerFlowPattern("TP_LINK_LOCAL_OFF_MAC", "50:c7:bf:33:1f:09", Arrays.asList(637, 1312));
-        MacLayerFlowPatternFinder finder = new MacLayerFlowPatternFinder(handle, pattern);
-        finder.findFlowPattern();
-        // -------------------------------------------------------------------------------------------------------------
-
-//        final String fileName = args.length > 0 ? args[0] : "/home/rtrimana/pcap_processing/smart_home_traffic/Code/Projects/SmartPlugDetector/pcap/wlan1.local.remote.dns.pcap";
-//        final String trainingFileName = "./pcap/TP_LINK_LOCAL_ON_SUBSET.pcap";
-//        //final String trainingFileName = "./pcap/TP_LINK_REMOTE_ON.pcap";
-//
-//        // ====== Debug code ======
+//        String fileName = "./pcap/mac-tplink.local.pcapng";
 //        PcapHandle handle;
-//        PcapHandle trainingPcap;
 //        try {
 //            handle = Pcaps.openOffline(fileName, PcapHandle.TimestampPrecision.NANO);
-//            trainingPcap = Pcaps.openOffline(trainingFileName, PcapHandle.TimestampPrecision.NANO);
 //        } catch (PcapNativeException pne) {
 //            handle = Pcaps.openOffline(fileName);
-//            trainingPcap = Pcaps.openOffline(trainingFileName);
 //        }
+//        Arrays.asList(1590, 1590, 1590, 1001, 337, 197, 636, 1311, 177) // Full pattern (all non-zero payload packets).
+//        MacLayerFlowPattern pattern = new MacLayerFlowPattern("TP_LINK_LOCAL_OFF_MAC", "50:c7:bf:33:1f:09", Arrays.asList(637, 1312));
+//        MacLayerFlowPatternFinder finder = new MacLayerFlowPatternFinder(handle, pattern);
+//        finder.findFlowPattern();
+        // -------------------------------------------------------------------------------------------------------------
+
+        final String fileName = args.length > 0 ? args[0] : "/home/rtrimana/pcap_processing/smart_home_traffic/Code/Projects/SmartPlugDetector/pcap/wlan1.local.dns.pcap";
+        final String trainingFileName = "./pcap/TP_LINK_LOCAL_ON_SUBSET.pcap";
+//        final String trainingFileName = "./pcap/TP_LINK_LOCAL_ON.pcap";
+//
+//        // ====== Debug code ======
+        PcapHandle handle;
+        PcapHandle trainingPcap;
+        try {
+            handle = Pcaps.openOffline(fileName, PcapHandle.TimestampPrecision.NANO);
+            trainingPcap = Pcaps.openOffline(trainingFileName, PcapHandle.TimestampPrecision.NANO);
+        } catch (PcapNativeException pne) {
+            handle = Pcaps.openOffline(fileName);
+            trainingPcap = Pcaps.openOffline(trainingFileName);
+        }
 //
 //        // TODO: The followings are the way to extract multiple hostnames and their associated packet lengths lists
 //        //List<String> list = new ArrayList<>();
@@ -62,9 +62,9 @@ public class Main {
 //        //list2.add("events.tplinkra.com");
 //        //FlowPattern fp3 = new FlowPattern("TP_LINK_REMOTE_ON", list2, trainingPcap);
 //
-//        FlowPattern fp = new FlowPattern("TP_LINK_LOCAL_ON", "events.tplinkra.com", trainingPcap);
-//        FlowPatternFinder fpf = new FlowPatternFinder(handle, fp);
-//        fpf.start();
+        FlowPattern fp = new FlowPattern("TP_LINK_LOCAL_ON", "events.tplinkra.com", trainingPcap);
+        FlowPatternFinder fpf = new FlowPatternFinder(handle, fp);
+        fpf.start();
 //
 //        // ========================
     }