From ce1c57dc79f55157b474b9e73495c56688aec229 Mon Sep 17 00:00:00 2001 From: Greg DiCristofaro <gregd@basistech.com> Date: Thu, 27 Jul 2023 20:22:22 -0400 Subject: [PATCH] some updates --- .../malwarescan/MalwareScanIngestModule.java | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/Core/src/com/basistech/df/cybertriage/autopsy/malwarescan/MalwareScanIngestModule.java b/Core/src/com/basistech/df/cybertriage/autopsy/malwarescan/MalwareScanIngestModule.java index 9b69226c58..bb7df071b2 100644 --- a/Core/src/com/basistech/df/cybertriage/autopsy/malwarescan/MalwareScanIngestModule.java +++ b/Core/src/com/basistech/df/cybertriage/autopsy/malwarescan/MalwareScanIngestModule.java @@ -558,16 +558,16 @@ private List<CTCloudBean> getHashLookupResults(IngestJobState ingestJobState, Li } else if (ingestJobState.getIngestJobContext().fileIngestIsCancelled()) { return Collections.emptyList(); } - + // while we have a valid auth token, also check file uploads. if (ingestJobState.isUploadUnknownFiles()) { - long remainingUploads = remaining(authTokenResponse.getFileUploadLimit(), authTokenResponse.getHashLookupCount()); + long remainingUploads = remaining(authTokenResponse.getFileUploadLimit(), authTokenResponse.getFileUploadCount()); if (remainingUploads <= 0) { ingestJobState.disableUploadUnknownFiles(); notifyWarning( - Bundle.MalwareScanIngestModule_uploadFile_noRemainingFileUploads_title(), - Bundle.MalwareScanIngestModule_uploadFile_noRemainingFileUploads_desc(), - null); + Bundle.MalwareScanIngestModule_uploadFile_noRemainingFileUploads_title(), + Bundle.MalwareScanIngestModule_uploadFile_noRemainingFileUploads_desc(), + null); } } @@ -683,7 +683,10 @@ private boolean uploadFile(IngestJobState ingestJobState, String md5, long objId "MalwareScanIngestModule_longPollForNotFound_timeout_title=File Upload Results Timeout", "MalwareScanIngestModule_longPollForNotFound_timeout_desc=There was a timeout while waiting for file uploads to be processed. Please try again later.",}) private void longPollForNotFound(IngestJobState ingestJobState) throws InterruptedException, CTCloudException, Blackboard.BlackboardException, TskCoreException { - if (!ingestJobState.isDoFileLookups() || !ingestJobState.isQueryForMissing() || MapUtils.isEmpty(ingestJobState.getUnidentifiedHashes())) { + if (!ingestJobState.isDoFileLookups() + || !ingestJobState.isQueryForMissing() + || MapUtils.isEmpty(ingestJobState.getUnidentifiedHashes()) + || ingestJobState.getIngestJobContext().fileIngestIsCancelled()) { return; } -- GitLab