Merge branch 'feature/SUPPORT-8818_log' into develop

This commit is contained in:
adel.kalimullin 2024-12-24 10:02:53 +03:00
commit a3d0971f6f
4 changed files with 72 additions and 53 deletions

35
README.md Normal file
View file

@ -0,0 +1,35 @@
# **av-service**
`av-service` — это сервис, взаимодействующий с Kaspersky Endpoint Security для Linux (KESL) на одной машине.
Основная задача сервиса — обеспечение защиты и анализа файлов с помощью интеграции с KESL.
Сервис использует консольные команды для выполнения операций и взаимодействия с KESL, таких как запуск задач проверки и получение результатов анализа.
---
## **Основные функции**
- Проверка файлов на наличие угроз с использованием KESL.
- Обработка файлов через POST-запросы.
- Логирование результатов проверки.
---
## **Требования**
- **Java 17**
- **KESL** (установлен и настроен на одной машине с сервисом)
---
## **Конфигурация и запуск**
### **1. Конфигурация параметров для сервиса**
Для корректного запуска сервиса добавьте следующие параметры в файл `application.properties`:
```properties
file.upload.directory=/tmp/uploaded_files # Путь для хранения временных файлов
spring.servlet.multipart.max-file-size=10MB # Максимальный размер загружаемого файла
spring.servlet.multipart.max-request-size=10MB # Максимальный размер запроса
server.port=8080 # Порт, на котором работает сервис
```
### **2. Запуск JAR-файла с конфигурационным файлом**
```bash
java -jar app.jar --spring.config.location=file:/path/to/your/application.properties
```

View file

@ -28,10 +28,15 @@ public class FileScanController {
@PostMapping("/scan-file")
public ResponseEntity<?> scanFile(@RequestPart("file") MultipartFile file) {
if (file.isEmpty()) {
LOGGER.warn("The uploaded file is empty or missing.");
return ResponseEntity.badRequest().build();
}
try {
ScanResult result = fileScanService.scanFile(file);
LOGGER.info("Scan result for file {}: status - {}, verdicts - {}",
file.getOriginalFilename(),
result.status(),
String.join(", ", result.verdicts()));
return ResponseEntity.ok(result);
}
catch (Exception e) {

View file

@ -1,42 +0,0 @@
package ru.micord.av.service.service;
import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import ru.micord.av.service.exception.AvException;
/**
* @author Adel Kalimullin
*/
@Service
public class FileManager {
@Value("${file.upload.directory}")
public String uploadDirectory;
public File saveFile(MultipartFile file) throws IOException {
Path uploadPath = Paths.get(uploadDirectory);
if (!Files.exists(uploadPath)) {
Files.createDirectories(uploadPath);
}
Path tempFile = Files.createTempFile(uploadPath, "kesl-upload-", ".tmp");
file.transferTo(tempFile.toFile());
return tempFile.toFile();
}
public void deleteFile(File file) {
if (file != null && file.exists()) {
try {
Files.delete(file.toPath());
}
catch (IOException e) {
throw new AvException("Ошибка удаления временного файла: " + file.getAbsolutePath(), e);
}
}
}
}

View file

@ -3,12 +3,18 @@ package ru.micord.av.service.service;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import ru.micord.av.service.exception.AvException;
@ -19,20 +25,18 @@ import ru.micord.av.service.model.ScanResult;
*/
@Service
public class FileScanService {
public static String KESL_CONTROL = "kesl-control";
public static String KESL_SCAN = "--scan-file";
private final FileManager fileManager;
public FileScanService(FileManager fileManager) {
this.fileManager = fileManager;
}
private static final String KESL_CONTROL = "kesl-control";
private static final String KESL_SCAN = "--scan-file";
private static final Logger LOGGER = LoggerFactory.getLogger(FileScanService.class);
@Value("${file.upload.directory}")
private String uploadDirectory;
public ScanResult scanFile(MultipartFile file) {
File tempFile = null;
LocalDateTime startTime;
LocalDateTime stopTime;
try {
tempFile = fileManager.saveFile(file);
tempFile = saveFile(file);
startTime = LocalDateTime.now();
String rawResult = runKeslScan(tempFile);
stopTime = LocalDateTime.now();
@ -41,12 +45,29 @@ public class FileScanService {
);
}
catch (Exception e) {
throw new AvException("Ошибка при сканировании файла: " + file.getOriginalFilename(), e);
throw new AvException("Error scanning file: " + file.getOriginalFilename(), e);
}
finally {
fileManager.deleteFile(tempFile);
try {
if (tempFile != null && tempFile.exists()) {
Files.delete(tempFile.toPath());
}
}
catch (IOException e) {
LOGGER.warn("Error deleting file: " + tempFile.getAbsolutePath(), e);
}
}
}
private File saveFile(MultipartFile file) throws IOException {
Path uploadPath = Paths.get(uploadDirectory);
if (!Files.exists(uploadPath)) {
Files.createDirectories(uploadPath);
}
Path tempFile = Files.createTempFile(uploadPath, "kesl-upload-", ".tmp");
file.transferTo(tempFile.toFile());
return tempFile.toFile();
}
private String runKeslScan(File file) throws IOException, InterruptedException {
ProcessBuilder processBuilder = new ProcessBuilder(KESL_CONTROL, KESL_SCAN,
@ -58,7 +79,7 @@ public class FileScanService {
String result = new String(inputStream.readAllBytes());
int exitCode = process.waitFor();
if (exitCode != 0 && exitCode != 72) {
throw new AvException("Ошибка KESL, код завершения: " + exitCode);
throw new AvException("KESL error, exit code: " + exitCode);
}
return result;
}