Adjust parameters, add logging

log2timeline · Mar 24, 2024 · 1c24253 · 1c24253
1 parent a3edf02
commit 1c24253
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dftimewolf/lib/collectors/grr_hosts.py b/dftimewolf/lib/collectors/grr_hosts.py
@@ -361,6 +361,7 @@ def _DownloadFiles(self, client: Client, flow_id: str) -> Optional[str]:
       os.makedirs(flow_output_dir, exist_ok=True)
 
     if grr_flow.Get().data.name == "TimelineFlow":
+      self.logger.info("Downloading timeline from GRR")
       self._DownloadTimeline(grr_flow, flow_output_dir)
       return flow_output_dir
 
@@ -378,8 +379,9 @@ def _DownloadFiles(self, client: Client, flow_id: str) -> Optional[str]:
       self.logger.warning(
         "Large files detected, downloading blobs instead of archive..."
       )
-      self._DownloadBlobs(grr_flow, flow_output_dir)
+      self._DownloadBlobs(client, files, flow_output_dir)
     else:
+      self.logger.info("Downloading file archive from GRR")
       self._DownloadArchive(grr_flow, flow_output_dir)
 
     return flow_output_dir