Добавлено общее хранилище ключей

This commit is contained in:
serega6531
2020-04-24 00:08:13 +03:00
parent 200dde7940
commit 441e210ea7
4 changed files with 24 additions and 2 deletions

View File

@@ -1,5 +1,7 @@
package ru.serega6531.packmate.service.optimization;
import org.springframework.boot.context.event.ApplicationReadyEvent;
import org.springframework.context.event.EventListener;
import org.springframework.stereotype.Service;
import ru.serega6531.packmate.utils.TlsUtils;
@@ -23,7 +25,6 @@ public class RsaKeysHolder {
X509KeyManager keyManager = TlsUtils.createKeyManager(pemFile, keyFile);
// X509Certificate[] certificateChain = keyManager.getCertificateChain("1");
RSAPrivateKey privateKey = ((RSAPrivateKey) keyManager.getPrivateKey("1"));
keys.put(privateKey.getModulus(), privateKey);
}
@@ -32,4 +33,9 @@ public class RsaKeysHolder {
return keys.get(modulus);
}
@EventListener(ApplicationReadyEvent.class)
public void afterStartup(ApplicationReadyEvent event) {
//TODO load keys
}
}

View File

@@ -12,6 +12,7 @@ import java.util.List;
@Slf4j
public class StreamOptimizer {
private final RsaKeysHolder keysHolder;
private final CtfService service;
private List<Packet> packets;
@@ -19,6 +20,10 @@ public class StreamOptimizer {
* Вызвать для выполнения оптимизаций на переданном списке пакетов.
*/
public List<Packet> optimizeStream() {
if (service.isDecryptTls()) {
decryptTls();
}
if (service.isProcessChunkedEncoding()) {
processChunkedEncoding();
}
@@ -42,6 +47,11 @@ public class StreamOptimizer {
return packets;
}
private void decryptTls() {
final TlsDecryptor tlsDecryptor = new TlsDecryptor(packets, keysHolder);
tlsDecryptor.decryptTls(); // TODO
}
/**
* Сжать соседние пакеты в одном направлении в один.
* Выполняется после других оптимизаций чтобы правильно определять границы пакетов.