Рефакторинг

src/main/java/ru/serega6531/packmate/service/optimization/StreamOptimizer.java

@@ -0,0 +1,96 @@
+package ru.serega6531.packmate.service.optimization;
+
+import lombok.AllArgsConstructor;
+import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
+import ru.serega6531.packmate.model.CtfService;
+import ru.serega6531.packmate.model.Packet;
+
+import java.util.List;
+
+@AllArgsConstructor
+@Slf4j
+public class StreamOptimizer {
+
+    private final CtfService service;
+    private List<Packet> packets;
+
+    /**
+     * Вызвать для выполнения оптимизаций на переданном списке пакетов.
+     */
+    public List<Packet> optimizeStream() {
+        if (service.isProcessChunkedEncoding()) {
+            processChunkedEncoding();
+        }
+
+        if (service.isUngzipHttp()) {
+            unpackGzip();
+        }
+
+        if (service.isParseWebSockets()) {
+            parseWebSockets();
+        }
+
+        if (service.isUrldecodeHttpRequests()) {
+            urldecodeRequests();
+        }
+
+        if (service.isMergeAdjacentPackets()) {
+            mergeAdjacentPackets();
+        }
+
+        return packets;
+    }
+
+    /**
+     * Сжать соседние пакеты в одном направлении в один.
+     * Выполняется после других оптимизаций чтобы правильно определять границы пакетов.
+     */
+    private void mergeAdjacentPackets() {
+        final PacketsMerger merger = new PacketsMerger(packets);
+        merger.mergeAdjacentPackets();
+    }
+
+    /**
+     * Декодирование urlencode с http пакета до смены стороны или окончания стрима
+     */
+    @SneakyThrows
+    private void urldecodeRequests() {
+        final HttpUrldecodeProcessor processor = new HttpUrldecodeProcessor(packets);
+        processor.urldecodeRequests();
+    }
+
+    /**
+     * https://ru.wikipedia.org/wiki/Chunked_transfer_encoding
+     */
+    private void processChunkedEncoding() {
+        HttpChunksProcessor processor = new HttpChunksProcessor(packets);
+        processor.processChunkedEncoding();
+    }
+
+    /**
+     * Попытаться распаковать GZIP из исходящих http пакетов. <br>
+     * GZIP поток начинается на найденном HTTP пакете с заголовком Content-Encoding: gzip
+     * (при этом заголовок HTTP может быть в другом пакете)<br>
+     * Поток заканчивается при обнаружении нового HTTP заголовка,
+     * при смене стороны передачи или при окончании всего стрима
+     */
+    private void unpackGzip() {
+        final HttpGzipProcessor processor = new HttpGzipProcessor(packets);
+        processor.unpackGzip();
+    }
+
+    private void parseWebSockets() {
+        if (!packets.get(0).getContentString().contains("HTTP/")) {
+            return;
+        }
+
+        final WebSocketsParser parser = new WebSocketsParser(packets);
+        if (!parser.isParsed()) {
+            return;
+        }
+
+        packets = parser.getParsedPackets();
+    }
+
+}