Hochschulprojekte/OrientDBApplication
0
0
Fork 0
Code Issues Pull Requests Activity

Small refactoring, renamed Ipv4PacketModel to IpPacketModel

Browse Source
This commit is contained in:
Jan Philipp Timme 2016-12-05 19:39:20 +01:00
parent a895d8741e
commit 4d0585b68a
Signed by untrusted user: JPT
GPG Key ID: 5F2C85EC6F3754B7
3 changed files with 14 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/main/java/de/hsh/inform/orientdb_project/model/Ipv4PacketModel.java → src/main/java/de/hsh/inform/orientdb_project/model/IpPacketModel.java
View File

@ -1,14 +1,12 @@
package de.hsh.inform.orientdb_project.model; package de.hsh.inform.orientdb_project.model;
import java.net.Inet4Address;
import org.pcap4j.packet.IpV4Packet; import org.pcap4j.packet.IpV4Packet;
import com.orientechnologies.orient.core.metadata.schema.OType; import com.orientechnologies.orient.core.metadata.schema.OType;
import com.tinkerpop.blueprints.impls.orient.OrientGraphNoTx; import com.tinkerpop.blueprints.impls.orient.OrientGraphNoTx;
import com.tinkerpop.blueprints.impls.orient.OrientVertexType; import com.tinkerpop.blueprints.impls.orient.OrientVertexType;
public class Ipv4PacketModel { public class IpPacketModel {
public long ts; public long ts;
public int ms; public int ms;
@ -30,14 +28,13 @@ public class Ipv4PacketModel {
ipPacketType.createProperty("payloadSize", OType.INTEGER); ipPacketType.createProperty("payloadSize", OType.INTEGER);
} }
public Ipv4PacketModel(IpV4Packet ipv4, Inet4Address sourceIp, Inet4Address targetIp, long ts, int ms) { public IpPacketModel(IpV4Packet ipv4, long ts, int ms) {
this.ts = ts; this.ts = ts;
this.ms = ms; this.ms = ms;
this.sourceIp = sourceIp.toString().split("/")[1]; this.sourceIp = ipv4.getHeader().getSrcAddr().toString().split("/")[1];
this.targetIp = targetIp.toString().split("/")[1]; this.targetIp = ipv4.getHeader().getDstAddr().toString().split("/")[1];
this.size = ipv4.getRawData().length; this.size = ipv4.getRawData().length;
this.payloadSize = ipv4.getRawData().length - ipv4.getHeader().length(); this.payloadSize = ipv4.getRawData().length - ipv4.getHeader().length();
} }
public Object[] getArguments() { public Object[] getArguments() {

18
src/main/java/de/hsh/inform/orientdb_project/orientdb/NodeBasedImportService.java
View File

@ -19,7 +19,7 @@ import de.hsh.inform.orientdb_project.model.ArpPacketModel;
import de.hsh.inform.orientdb_project.model.EthernetFrameModel; import de.hsh.inform.orientdb_project.model.EthernetFrameModel;
import de.hsh.inform.orientdb_project.model.HostModel; import de.hsh.inform.orientdb_project.model.HostModel;
import de.hsh.inform.orientdb_project.model.IcmpPacketModel; import de.hsh.inform.orientdb_project.model.IcmpPacketModel;
import de.hsh.inform.orientdb_project.model.Ipv4PacketModel; import de.hsh.inform.orientdb_project.model.IpPacketModel;
import de.hsh.inform.orientdb_project.model.TcpConnectionModel; import de.hsh.inform.orientdb_project.model.TcpConnectionModel;
import de.hsh.inform.orientdb_project.model.TcpPacketModel; import de.hsh.inform.orientdb_project.model.TcpPacketModel;
import de.hsh.inform.orientdb_project.model.UdpPacketModel; import de.hsh.inform.orientdb_project.model.UdpPacketModel;
@ -38,7 +38,7 @@ public class NodeBasedImportService extends AbstractNetdataImportService {
// References to already created model instances (these are reseted before processing a new ethernetFrame) // References to already created model instances (these are reseted before processing a new ethernetFrame)
private EthernetFrameModel ethernetFrameModel; private EthernetFrameModel ethernetFrameModel;
private ArpPacketModel arpPacketModel; private ArpPacketModel arpPacketModel;
private Ipv4PacketModel ipv4PacketModel; private IpPacketModel ipPacketModel;
private TcpPacketModel tcpPacketModel; private TcpPacketModel tcpPacketModel;
private UdpPacketModel udpPacketModel; private UdpPacketModel udpPacketModel;
private IcmpPacketModel icmpPacketModel; private IcmpPacketModel icmpPacketModel;
@ -69,7 +69,7 @@ public class NodeBasedImportService extends AbstractNetdataImportService {
// Also clean model instances // Also clean model instances
this.ethernetFrameModel = null; this.ethernetFrameModel = null;
this.arpPacketModel = null; this.arpPacketModel = null;
this.ipv4PacketModel = null; this.ipPacketModel = null;
this.tcpPacketModel = null; this.tcpPacketModel = null;
this.udpPacketModel = null; this.udpPacketModel = null;
this.icmpPacketModel = null; this.icmpPacketModel = null;
@ -88,13 +88,11 @@ public class NodeBasedImportService extends AbstractNetdataImportService {
} }
public void handleIpV4Packet(IpV4Packet ipv4, long ts, int ms) { public void handleIpV4Packet(IpV4Packet ipv4, long ts, int ms) {
Inet4Address sourceIp = ipv4.getHeader().getSrcAddr(); this.ipPacketModel = new IpPacketModel(ipv4, ts, ms);
Inet4Address targetIp = ipv4.getHeader().getDstAddr(); this.ipPacketVertex = this.og.addVertex("class:IpPacket", this.ipPacketModel.getArguments());
// Add hosts to database if new // Add hosts to database if new
this.addHostIfNew(sourceIp); this.addHostIfNew(ipv4.getHeader().getSrcAddr());
this.addHostIfNew(targetIp); this.addHostIfNew(ipv4.getHeader().getDstAddr());
this.ipv4PacketModel = new Ipv4PacketModel(ipv4, sourceIp, targetIp, ts, ms);
this.ipPacketVertex = this.og.addVertex("class:IpPacket", this.ipv4PacketModel.getArguments());
// Wire up to its ethernet frame // Wire up to its ethernet frame
Edge containsEdge = this.og.addEdge("class:contains", this.ethernetFrameVertex, this.ipPacketVertex, "contains"); Edge containsEdge = this.og.addEdge("class:contains", this.ethernetFrameVertex, this.ipPacketVertex, "contains");
Edge isContainedInEdge = this.og.addEdge("class:isContainedIn", this.ipPacketVertex, this.ethernetFrameVertex, "isContainedIn"); Edge isContainedInEdge = this.og.addEdge("class:isContainedIn", this.ipPacketVertex, this.ethernetFrameVertex, "isContainedIn");
@ -119,7 +117,7 @@ public class NodeBasedImportService extends AbstractNetdataImportService {
TcpConnectionModel tcpConnection = this.getTcpConnectionFor(tcp); TcpConnectionModel tcpConnection = this.getTcpConnectionFor(tcp);
if(tcpConnection != null) { // If connection exists ... if(tcpConnection != null) { // If connection exists ...
if(ts - tcpConnection.endTs < 2) { // ... and still "up to date" aka time difference < 2s if(ts - tcpConnection.endTs < 2) { // ... and still "up to date" aka time difference < 2s
if(tcpConnection.sourceIp.equals(this.ipv4PacketModel.sourceIp)) { if(tcpConnection.sourceIp.equals(this.ipPacketModel.sourceIp)) {
// Update connection data in direction SourceIp -> TargetIp // Update connection data in direction SourceIp -> TargetIp
tcpConnection.addVolumeSourceToTarget(this.tcpPacketModel.payloadSize); tcpConnection.addVolumeSourceToTarget(this.tcpPacketModel.payloadSize);
} else { } else {

6
src/main/java/de/hsh/inform/orientdb_project/orientdb/OrientDbHelperService.java
View File

@ -4,18 +4,16 @@ import java.io.IOException;
import com.orientechnologies.orient.client.remote.OServerAdmin; import com.orientechnologies.orient.client.remote.OServerAdmin;
import com.orientechnologies.orient.core.intent.OIntentMassiveInsert; import com.orientechnologies.orient.core.intent.OIntentMassiveInsert;
import com.orientechnologies.orient.core.metadata.schema.OType;
import com.tinkerpop.blueprints.impls.orient.OrientConfigurableGraph.THREAD_MODE; import com.tinkerpop.blueprints.impls.orient.OrientConfigurableGraph.THREAD_MODE;
import com.tinkerpop.blueprints.impls.orient.OrientEdgeType; import com.tinkerpop.blueprints.impls.orient.OrientEdgeType;
import com.tinkerpop.blueprints.impls.orient.OrientGraphFactory; import com.tinkerpop.blueprints.impls.orient.OrientGraphFactory;
import com.tinkerpop.blueprints.impls.orient.OrientGraphNoTx; import com.tinkerpop.blueprints.impls.orient.OrientGraphNoTx;
import com.tinkerpop.blueprints.impls.orient.OrientVertexType;
import de.hsh.inform.orientdb_project.model.ArpPacketModel; import de.hsh.inform.orientdb_project.model.ArpPacketModel;
import de.hsh.inform.orientdb_project.model.EthernetFrameModel; import de.hsh.inform.orientdb_project.model.EthernetFrameModel;
import de.hsh.inform.orientdb_project.model.HostModel; import de.hsh.inform.orientdb_project.model.HostModel;
import de.hsh.inform.orientdb_project.model.IcmpPacketModel; import de.hsh.inform.orientdb_project.model.IcmpPacketModel;
import de.hsh.inform.orientdb_project.model.Ipv4PacketModel; import de.hsh.inform.orientdb_project.model.IpPacketModel;
import de.hsh.inform.orientdb_project.model.TcpConnectionModel; import de.hsh.inform.orientdb_project.model.TcpConnectionModel;
import de.hsh.inform.orientdb_project.model.TcpPacketModel; import de.hsh.inform.orientdb_project.model.TcpPacketModel;
import de.hsh.inform.orientdb_project.model.UdpPacketModel; import de.hsh.inform.orientdb_project.model.UdpPacketModel;
@ -127,7 +125,7 @@ public class OrientDbHelperService {
// These are the vertex types used in our model // These are the vertex types used in our model
EthernetFrameModel.createType(og); EthernetFrameModel.createType(og);
ArpPacketModel.createType(og); ArpPacketModel.createType(og);
Ipv4PacketModel.createType(og); IpPacketModel.createType(og);
UdpPacketModel.createType(og); UdpPacketModel.createType(og);
TcpPacketModel.createType(og); TcpPacketModel.createType(og);
IcmpPacketModel.createType(og); IcmpPacketModel.createType(og);