Добавлен PcapWorker

2019-04-28 02:21:13 +03:00
parent 8e2342f181
commit 8cb109ac7d
6 changed files with 106 additions and 6 deletions
--- a/src/main/java/ru/serega6531/packmate/PcapWorker.java
+++ b/src/main/java/ru/serega6531/packmate/PcapWorker.java
@@ -0,0 +1,90 @@
+package ru.serega6531.packmate;
+
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.concurrent.BasicThreadFactory;
+import org.pcap4j.core.PcapHandle;
+import org.pcap4j.core.PcapNativeException;
+import org.pcap4j.core.PcapNetworkInterface;
+import org.pcap4j.core.Pcaps;
+import org.pcap4j.packet.Packet;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Component;
+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.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.concurrent.TimeoutException;
+
+@Component
+@Slf4j
+public class PcapWorker {
+
+    private final ServicesService servicesService;
+    private final StreamService streamService;
+    private final PacketService packetService;
+    private final PatternService patternService;
+
+    private PcapNetworkInterface device;
+    private PcapHandle pcap = null;
+    private ExecutorService executorService;
+
+    @Autowired
+    public PcapWorker(ServicesService servicesService,
+                      StreamService streamService,
+                      PacketService packetService,
+                      PatternService patternService,
+                      @Value("${interface-name}") String interfaceName) throws PcapNativeException {
+        this.servicesService = servicesService;
+        this.streamService = streamService;
+        this.packetService = packetService;
+        this.patternService = patternService;
+
+        BasicThreadFactory factory = new BasicThreadFactory.Builder()
+                .namingPattern("pcap-worker").build();
+        executorService = Executors.newSingleThreadExecutor(factory);
+        device = Pcaps.getDevByName(interfaceName);
+    }
+
+    public void start() throws PcapNativeException {
+        System.out.println("Using interface " + device.getName());
+        pcap = device.openLive(65536, PcapNetworkInterface.PromiscuousMode.PROMISCUOUS, 100);
+
+        executorService.execute(() -> {
+            try {
+                log.info("Intercept started");
+                while (true) {
+                    if (pcap.isOpen()) {
+                        try {
+                            final Packet packet = pcap.getNextPacketEx();
+                            processPacket(packet);
+                        } catch (TimeoutException ignored) {
+                        }
+                    } else {
+                        break;
+                    }
+                }
+            } catch (Exception e) {
+                log.error("Error while capturing packet", e);
+                stop();
+            }
+        });
+    }
+
+    @PreDestroy
+    public void stop() {
+        if (pcap != null && pcap.isOpen()) {
+            pcap.close();
+        }
+
+        log.info("Intercept stopped");
+    }
+
+    private void processPacket(Packet packet) {
+        System.out.println(packet);
+    }
+}