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

Добавлено сохранение tcp пакетов по таймауту

Browse Source
This commit is contained in:
serega6531
2019-05-14 01:44:23 +03:00
parent adba3aa18c
commit 6bbfa49b7a
4 changed files with 48 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -24,7 +24,6 @@ import java.net.Inet4Address;
import java.util.*;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
@Component
@@ -39,7 +38,6 @@ public class PcapWorker {
private final ExecutorService executorService;
private final String localIp;
private final int udpStreamTimeout;
private long packetIdCounter = 0; // оно однопоточное, так что пусть будет без atomic
@@ -53,13 +51,11 @@ public class PcapWorker {
public PcapWorker(ServicesService servicesService,
StreamService streamService,
@Value("${interface-name}") String interfaceName,
@Value("${local-ip}") String localIp,
@Value("${udp-stream-timeout}") int udpStreamTimeout) throws PcapNativeException {
@Value("${local-ip}") String localIp) throws PcapNativeException {
this.servicesService = servicesService;
this.streamService = streamService;
this.localIp = localIp;
this.udpStreamTimeout = udpStreamTimeout;
BasicThreadFactory factory = new BasicThreadFactory.Builder()
.namingPattern("pcap-worker").build();
@@ -205,7 +201,7 @@ public class PcapWorker {
}
}
public int closeUdpStreams() {
public int closeTimeoutStreams(Protocol protocol, long timeoutMillis) {
int streamsClosed = 0;
final Iterator<Map.Entry<UnfinishedStream, List<ru.serega6531.packmate.model.Packet>>> iterator = unfinishedStreams.entrySet().iterator();
@@ -213,9 +209,9 @@ public class PcapWorker {
final Map.Entry<UnfinishedStream, List<ru.serega6531.packmate.model.Packet>> entry = iterator.next();
final UnfinishedStream stream = entry.getKey();
if (stream.getProtocol() == Protocol.UDP) {
if (stream.getProtocol() == protocol) {
final List<ru.serega6531.packmate.model.Packet> packets = entry.getValue();
if (System.currentTimeMillis() - packets.get(packets.size() - 1).getTimestamp() > TimeUnit.SECONDS.toMillis(udpStreamTimeout)) {
if (System.currentTimeMillis() - packets.get(packets.size() - 1).getTimestamp() > timeoutMillis) {
iterator.remove();
streamService.saveNewStream(stream, packets);
streamsClosed++;