Code/Projects/SmartPlugDetector/src/main/java/edu/uci/iotproject/Main.java

   1 package edu.uci.iotproject;
   2
   3 import edu.uci.iotproject.maclayer.MacLayerFlowPattern;
   4 import edu.uci.iotproject.maclayer.MacLayerFlowPatternFinder;
   5 import org.pcap4j.core.*;
   6
   7 import java.io.EOFException;
   8 import java.net.UnknownHostException;
   9 import java.util.*;
  10 import java.util.concurrent.TimeoutException;
  11
  12 /**
  13  * This is a system that reads PCAP files to compare
  14  * patterns of DNS hostnames, packet sequences, and packet
  15  * lengths with training data to determine certain events
  16  * or actions for smart home devices.
  17  *
  18  * @author Janus Varmarken
  19  * @author Rahmadi Trimananda (rtrimana@uci.edu)
  20  * @version 0.1
  21  */
  22 public class Main {
  23
  24
  25     public static void main(String[] args) throws PcapNativeException, NotOpenException, EOFException, TimeoutException, UnknownHostException {
  26         // -------------------------------------------------------------------------------------------------------------
  27         // Example/debug code for searching for a pattern at the MAC layer.
  28 //        String fileName = "./pcap/mac-tplink.local.pcapng";
  29 //        PcapHandle handle;
  30 //        try {
  31 //            handle = Pcaps.openOffline(fileName, PcapHandle.TimestampPrecision.NANO);
  32 //        } catch (PcapNativeException pne) {
  33 //            handle = Pcaps.openOffline(fileName);
  34 //        }
  35 //        Arrays.asList(1590, 1590, 1590, 1001, 337, 197, 636, 1311, 177) // Full pattern (all non-zero payload packets).
  36 //        MacLayerFlowPattern pattern = new MacLayerFlowPattern("TP_LINK_LOCAL_OFF_MAC", "50:c7:bf:33:1f:09", Arrays.asList(637, 1312));
  37 //        MacLayerFlowPatternFinder finder = new MacLayerFlowPatternFinder(handle, pattern);
  38 //        finder.findFlowPattern();
  39         // -------------------------------------------------------------------------------------------------------------
  40
  41         //final String fileName = args.length > 0 ? args[0] : "/home/rtrimana/pcap_processing/smart_home_traffic/Code/Projects/SmartPlugDetector/pcap/wlan1.local.dns.pcap";
  42         final String fileName = args.length > 0 ? args[0] : "/scratch/June-2018/TPLink/wlan1/tplink.wlan1.local.pcap";
  43         //final String fileName = args.length > 0 ? args[0] : "/scratch/June-2018/DLink/wlan1/dlink.wlan1.local.pcap";
  44         final String trainingFileName = "./pcap/TP_LINK_LOCAL_ON_SUBSET.pcap";
  45 //        final String trainingFileName = "./pcap/TP_LINK_LOCAL_ON.pcap";
  46 //
  47 //        // ====== Debug code ======
  48         PcapHandle handle;
  49         PcapHandle trainingPcap;
  50         try {
  51             handle = Pcaps.openOffline(fileName, PcapHandle.TimestampPrecision.NANO);
  52             trainingPcap = Pcaps.openOffline(trainingFileName, PcapHandle.TimestampPrecision.NANO);
  53         } catch (PcapNativeException pne) {
  54             handle = Pcaps.openOffline(fileName);
  55             trainingPcap = Pcaps.openOffline(trainingFileName);
  56         }
  57 //
  58 //        // TODO: The followings are the way to extract multiple hostnames and their associated packet lengths lists
  59 //        //List<String> list = new ArrayList<>();
  60 //        //list.add("events.tplinkra.com");
  61 //        //FlowPattern fp = new FlowPattern("TP_LINK_LOCAL_ON", list, trainingPcap);
  62 //        //List<String> list2 = new ArrayList<>();
  63 //        //list2.add("devs.tplinkcloud.com");
  64 //        //list2.add("events.tplinkra.com");
  65 //        //FlowPattern fp3 = new FlowPattern("TP_LINK_REMOTE_ON", list2, trainingPcap);
  66 //
  67         FlowPattern fp = new FlowPattern("TP_LINK_LOCAL_ON", "events.tplinkra.com", trainingPcap);
  68         //FlowPattern fp = new FlowPattern("DLINK_LOCAL_ON", "rfe-us-west-1.dch.dlink.com", trainingPcap);
  69         FlowPatternFinder fpf = new FlowPatternFinder(handle, fp);
  70         fpf.start();
  71 //
  72 //        // ========================
  73     }
  74 }