X-Git-Url: http://plrg.eecs.uci.edu/git/?p=pingpong.git;a=blobdiff_plain;f=Code%2FProjects%2FSmartPlugDetector%2Fsrc%2Fmain%2Fjava%2Fedu%2Fuci%2Fiotproject%2FMain.java;h=45fea17814900d7273f94df96eee750b7cf3d774;hp=bc5948dd971135175856e22e414da3f1fe0a2164;hb=e193fa2e8bc7dc4d218e5b6177d13d350712d7fa;hpb=1e72719e0ea1c3ceca5e668bd651bd77c999c5e2 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 bc5948d..45fea17 100644 --- a/Code/Projects/SmartPlugDetector/src/main/java/edu/uci/iotproject/Main.java +++ b/Code/Projects/SmartPlugDetector/src/main/java/edu/uci/iotproject/Main.java @@ -1,6 +1,5 @@ package edu.uci.iotproject; -import edu.uci.iotproject.analysis.PcapPacketPair; import edu.uci.iotproject.analysis.TcpConversationUtils; import edu.uci.iotproject.analysis.TriggerTrafficExtractor; import edu.uci.iotproject.io.TriggerTimesFileReader; @@ -10,8 +9,7 @@ import org.pcap4j.packet.namednumber.DataLinkType; import java.io.EOFException; import java.net.UnknownHostException; import java.time.Instant; -import java.util.ArrayList; -import java.util.List; +import java.util.*; import java.util.concurrent.TimeoutException; /** @@ -29,110 +27,56 @@ 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.dns.pcap"; -// final String fileName = args.length > 0 ? args[0] : "/scratch/June-2018/TPLink/wlan1/tplink.wlan1.local.pcap"; -// //final String fileName = args.length > 0 ? args[0] : "/scratch/June-2018/DLink/wlan1/dlink.wlan1.local.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 list = new ArrayList<>(); -//// //list.add("events.tplinkra.com"); -//// //FlowPattern fp = new FlowPattern("TP_LINK_LOCAL_ON", list, trainingPcap); -//// //List list2 = new ArrayList<>(); -//// //list2.add("devs.tplinkcloud.com"); -//// //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); -// //FlowPattern fp = new FlowPattern("DLINK_LOCAL_ON", "rfe-us-west-1.dch.dlink.com", trainingPcap); -// FlowPatternFinder fpf = new FlowPatternFinder(handle, fp); -// fpf.start(); -//// -//// // ======================== - - /* - PcapReader pcapReader = new PcapReader(args[0]); - PcapProcessingPipeline pipeline = new PcapProcessingPipeline(pcapReader); - TcpReassembler tcpReassembler = new TcpReassembler(); - pipeline.addPcapPacketConsumer(tcpReassembler); - pipeline.executePipeline(); - System.out.println("Pipeline terminated"); - - List> pairs = new ArrayList<>(); - for (Conversation c : tcpReassembler.getTcpConversations()) { - pairs.add(TcpConversationUtils.extractPacketPairs(c)); - } - */ - - /* - // -------- 07-17-2018 -------- - // Only consider packets to/from the TP-Link plug. - PcapReader pcapReader = new PcapReader(args[0], "ip host 192.168.1.159"); - TcpReassembler tcpReassembler = new TcpReassembler(); - PcapPacket packet; - while((packet = pcapReader.readNextPacket()) != null) { - tcpReassembler.consumePacket(packet); - } - // Now we have a set of reassembled TCP conversations. - List conversations = tcpReassembler.getTcpConversations(); - for(Conversation c : conversations) { - List pairs = TcpConversationUtils.extractPacketPairs(c); - for (PcapPacketPair pair : pairs) { - // TODO ... - // 1. discard packets that are not within X seconds after trigger time - // 2. conversations may be (are) with different servers - so need to plot in different plots, one per hostname? - } - } - - // ---------------------------- - */ + // ------------ # Code for extracting traffic generated by a device within x seconds of a trigger # ------------ + // Paths to input and output files (consider supplying these as arguments instead) and IP of the device for + // which traffic is to be extracted: + // D-Link July 26 experiment + final String inputPcapFile = "/Users/varmarken/temp/UCI IoT Project/experiments/2018-07/dlink/dlink.wlan1.local.pcap"; + final String outputPcapFile = "/Users/varmarken/temp/UCI IoT Project/experiments/2018-07/dlink/dlink-processed.pcap"; + final String triggerTimesFile = "/Users/varmarken/temp/UCI IoT Project/experiments/2018-07/dlink/dlink-july-26-2018.timestamps"; + final String deviceIp = "192.168.1.246"; + // TP-Link July 25 experiment +// final String inputPcapFile = "/Users/varmarken/temp/UCI IoT Project/experiments/2018-07/tplink/tplink.wlan1.local.pcap"; +// final String outputPcapFile = "/Users/varmarken/temp/UCI IoT Project/experiments/2018-07/tplink/tplink-processed.pcap"; +// final String triggerTimesFile = "/Users/varmarken/temp/UCI IoT Project/experiments/2018-07/tplink/tplink-july-25-2018.timestamps"; +// final String deviceIp = "192.168.1.159"; - // -------- 07-19-2018 -------- TriggerTimesFileReader ttfr = new TriggerTimesFileReader(); - List triggerTimes = ttfr.readTriggerTimes("/Users/varmarken/Downloads/tplink-feb-13-2018.timestamps", false); -// triggerTimes.stream().forEach(i -> System.out.println(i.atZone(TriggerTimesFileReader.ZONE_ID_LOS_ANGELES).toString())); - String pcapFile = "/Users/varmarken/Development/Repositories/UCI/NetworkingGroup/smart_home_traffic/Code/Projects/SmartPlugDetector/pcap/wlan1.local.dns.pcap"; - String tpLinkPlugIp = "192.168.1.159"; - TriggerTrafficExtractor tte = new TriggerTrafficExtractor(pcapFile, triggerTimes, tpLinkPlugIp); - final PcapDumper outputter = Pcaps.openDead(DataLinkType.EN10MB, 65536).dumpOpen("/Users/varmarken/temp/traces/output/tplink-filtered.pcap"); + List triggerTimes = ttfr.readTriggerTimes(triggerTimesFile, false); + TriggerTrafficExtractor tte = new TriggerTrafficExtractor(inputPcapFile, triggerTimes, deviceIp); + final PcapDumper outputter = Pcaps.openDead(DataLinkType.EN10MB, 65536).dumpOpen(outputPcapFile); + DnsMap dnsMap = new DnsMap(); + TcpReassembler tcpReassembler = new TcpReassembler(); tte.performExtraction(pkt -> { try { outputter.dump(pkt); } catch (NotOpenException e) { e.printStackTrace(); } - }); + }, dnsMap, tcpReassembler); outputter.flush(); outputter.close(); - // ---------------------------- - + // Extract all conversations present in the filtered trace. + List allConversations = tcpReassembler.getTcpConversations(); + // Group conversations by hostname. + Map> convsByHostname = TcpConversationUtils.groupConversationsByHostname(allConversations, dnsMap); + System.out.println("Grouped conversations by hostname."); + // For each hostname, count the frequencies of packet lengths exchanged with that hostname. + final Map> pktLenFreqsByHostname = new HashMap<>(); + convsByHostname.forEach((host, convs) -> pktLenFreqsByHostname.put(host, TcpConversationUtils.countPacketLengthFrequencies(convs))); + System.out.println("Counted frequencies of packet lengths exchanged with each hostname."); + // For each hostname, count the frequencies of packet sequences (i.e., count how many conversations exchange a + // sequence of packets of some specific lengths). + final Map> pktSeqFreqsByHostname = new HashMap<>(); + convsByHostname.forEach((host, convs) -> pktSeqFreqsByHostname.put(host, TcpConversationUtils.countPacketSequenceFrequencies(convs))); + System.out.println("Counted frequencies of packet sequences exchanged with each hostname."); + // For each hostname, count frequencies of packet pairs exchanged with that hostname across all conversations + final Map> pktPairFreqsByHostname = + TcpConversationUtils.countPacketPairFrequenciesByHostname(allConversations, dnsMap); + System.out.println("Counted frequencies of packet pairs per hostname"); + // ------------------------------------------------------------------------------------------------------------- + // ------------------------------------------------------------------------------------------------------------- } }