micord/ervu-lkrp-av
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add av.check.enabled property

Browse source
This commit is contained in:
gulnaz 2024-09-27 16:35:12 +03:00
parent e7f8aaed29
commit 43e0cafcca
3 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

1
micord.env
View file

@ -16,6 +16,7 @@ ERVU_KAFKA_ERROR_TOPIC_NAME=ervu.lkrp.download.request
ERVU_KAFKA_SUCCESS_TOPIC_NAME=ervu.lkrp.download.request ERVU_KAFKA_SUCCESS_TOPIC_NAME=ervu.lkrp.download.request
ERVU_KAFKA_RESPONSE_TOPIC_NAME=ervu.lkrp.download.response ERVU_KAFKA_RESPONSE_TOPIC_NAME=ervu.lkrp.download.response
AV_CHECK_ENABLED=true
AV_REST_ADDRESS=http://10.10.31.118:8085/scans AV_REST_ADDRESS=http://10.10.31.118:8085/scans
AV_FIRST_TIMEOUT_MILLISECONDS=1000 AV_FIRST_TIMEOUT_MILLISECONDS=1000
AV_RETRY_MAX_ATTEMPTS_COUNT=10 AV_RETRY_MAX_ATTEMPTS_COUNT=10

10
src/main/java/ru/micord/ervu/av/service/FileUploadService.java
View file

@ -51,6 +51,8 @@ import ru.micord.ervu.av.s3.S3Service;
@Service @Service
public class FileUploadService { public class FileUploadService {
private static final Logger logger = LoggerFactory.getLogger(FileUploadService.class); private static final Logger logger = LoggerFactory.getLogger(FileUploadService.class);
@Value("${av.check.enabled}")
private boolean avCheckEnabled;
@Value("${av.rest.address}") @Value("${av.rest.address}")
private String avRestAddress; private String avRestAddress;
@Value("${av.first.timeout.milliseconds}") @Value("${av.first.timeout.milliseconds}")
@ -87,13 +89,17 @@ public class FileUploadService {
Path filePath = Paths.get(fileSavingPath, fileUrl.fileName()); Path filePath = Paths.get(fileSavingPath, fileUrl.fileName());
String downloadUrl = fileUrl.fileUrl(); String downloadUrl = fileUrl.fileUrl();
downloadFile(downloadUrl, filePath); downloadFile(downloadUrl, filePath);
boolean clean = true;
boolean infected = false;
if (avCheckEnabled) {
AvResponse avResponse = checkFile(filePath); AvResponse avResponse = checkFile(filePath);
boolean clean = Arrays.stream(avResponse.verdicts()) clean = Arrays.stream(avResponse.verdicts())
.anyMatch(verdict -> verdict.equalsIgnoreCase(AvResponse.Scan.VERDICT_CLEAN)); .anyMatch(verdict -> verdict.equalsIgnoreCase(AvResponse.Scan.VERDICT_CLEAN));
boolean infected = Arrays.stream(avResponse.verdicts()) infected = Arrays.stream(avResponse.verdicts())
.anyMatch(verdict -> verdict.equalsIgnoreCase(AvResponse.Scan.VERDICT_INFECTED)); .anyMatch(verdict -> verdict.equalsIgnoreCase(AvResponse.Scan.VERDICT_INFECTED));
}
if (infected || !clean) { if (infected || !clean) {
downloadRequest.fileInfo().setFileUrl(null); downloadRequest.fileInfo().setFileUrl(null);

1
src/main/resources/application.properties
View file

@ -42,6 +42,7 @@ kafka.out.error.topic.name=${ERVU_KAFKA_ERROR_TOPIC_NAME}
kafka.out.success.topic.name=${ERVU_KAFKA_SUCCESS_TOPIC_NAME} kafka.out.success.topic.name=${ERVU_KAFKA_SUCCESS_TOPIC_NAME}
kafka.out.response.topic.name=${ERVU_KAFKA_RESPONSE_TOPIC_NAME} kafka.out.response.topic.name=${ERVU_KAFKA_RESPONSE_TOPIC_NAME}
# #
av.check.enabled=${AV_CHECK_ENABLED:false}
av.rest.address=${AV_REST_ADDRESS} av.rest.address=${AV_REST_ADDRESS}
av.first.timeout.milliseconds=${AV_FIRST_TIMEOUT_MILLISECONDS:1000} av.first.timeout.milliseconds=${AV_FIRST_TIMEOUT_MILLISECONDS:1000}
av.retry.max.attempts.count=${AV_RETRY_MAX_ATTEMPTS_COUNT:10} av.retry.max.attempts.count=${AV_RETRY_MAX_ATTEMPTS_COUNT:10}