0xb00b5/0xb00b5-packmate
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Добавлен анализ пакета

Browse Source
This commit is contained in:
serega6531
2019-04-28 03:04:30 +03:00
parent 6eddd6f64f
commit 5f668baf05
5 changed files with 65 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
src/main/java/ru/serega6531/packmate/PcapWorker.java
View File

@@ -6,16 +6,21 @@ import org.pcap4j.core.PcapHandle;
import org.pcap4j.core.PcapNativeException;
import org.pcap4j.core.PcapNetworkInterface;
import org.pcap4j.core.Pcaps;
import org.pcap4j.packet.IpV4Packet;
import org.pcap4j.packet.Packet;
import org.pcap4j.packet.TcpPacket;
import org.pcap4j.packet.UdpPacket;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import ru.serega6531.packmate.model.CtfService;
import ru.serega6531.packmate.service.PacketService;
import ru.serega6531.packmate.service.PatternService;
import ru.serega6531.packmate.service.ServicesService;
import ru.serega6531.packmate.service.StreamService;
import javax.annotation.PreDestroy;
import java.util.Optional;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeoutException;
@@ -29,21 +34,26 @@ public class PcapWorker {
private final PacketService packetService;
private final PatternService patternService;
private PcapNetworkInterface device;
private final PcapNetworkInterface device;
private PcapHandle pcap = null;
private ExecutorService executorService;
private final ExecutorService executorService;
private final String localIp;
@Autowired
public PcapWorker(ServicesService servicesService,
StreamService streamService,
PacketService packetService,
PatternService patternService,
@Value("${interface-name}") String interfaceName) throws PcapNativeException {
@Value("${interface-name}") String interfaceName,
@Value("${local-ip}") String localIp) throws PcapNativeException {
this.servicesService = servicesService;
this.streamService = streamService;
this.packetService = packetService;
this.patternService = patternService;
this.localIp = localIp;
BasicThreadFactory factory = new BasicThreadFactory.Builder()
.namingPattern("pcap-worker").build();
executorService = Executors.newSingleThreadExecutor(factory);
@@ -84,7 +94,45 @@ public class PcapWorker {
log.info("Intercept stopped");
}
private void processPacket(Packet packet) {
System.out.println(packet);
private void processPacket(Packet rawPacket) {
String sourceIp = null;
String destIp = null;
int sourcePort = -1;
int destPort = -1;
byte[] content = null;
if(rawPacket.contains(IpV4Packet.class)){
final IpV4Packet.IpV4Header header = rawPacket.get(IpV4Packet.class).getHeader();
sourceIp = header.getSrcAddr().getHostAddress();
destIp = header.getDstAddr().getHostAddress();
}
if(rawPacket.contains(TcpPacket.class)) {
final TcpPacket packet = rawPacket.get(TcpPacket.class);
final TcpPacket.TcpHeader header = packet.getHeader();
sourcePort = header.getSrcPort().valueAsInt();
destPort = header.getDstPort().valueAsInt();
content = packet.getRawData();
} else if(rawPacket.contains(UdpPacket.class)) {
final UdpPacket packet = rawPacket.get(UdpPacket.class);
final UdpPacket.UdpHeader header = packet.getHeader();
sourcePort = header.getSrcPort().valueAsInt();
destPort = header.getDstPort().valueAsInt();
content = packet.getRawData();
}
if(sourceIp != null && sourcePort != -1) {
Optional<CtfService> serviceOptional = Optional.empty();
if(sourceIp.equals(localIp)) {
serviceOptional = servicesService.findByPort(sourcePort);
} else if(destIp.equals(localIp)) {
serviceOptional = servicesService.findByPort(destPort);
}
if(serviceOptional.isPresent()) {
log.info("{} {}:{} -> {}:{}", serviceOptional, sourceIp, sourcePort, destIp, destPort);
}
}
}
}