timezone support (kevoreilly#2873)

* timezone support

* Update lib/cuckoo/core/database.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

* Update test_analysis_manager.py

---------

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: doomedraven

conf/default/cuckoo.conf.default

@@ -3,6 +3,10 @@
 # Ignore Signals, will quit CAPE inmediatelly instead wait jobs to finish
 ignore_signals = yes
 
+# Specify the timezone for the system (e.g., UTC, Europe/Madrid, America/New_York).
+# Default is utc.
+timezone = utc
+
 # Which category of tasks do you want to analyze?
 categories = static, pcap, url, file
 

lib/cuckoo/core/database.py

@@ -13,11 +13,7 @@
 from contextlib import suppress
 from datetime import datetime, timedelta, timezone
 from typing import Any, List, Optional, Union, Tuple, Dict
-
-
-def _utcnow_naive():
-    """Returns the current time in UTC as a naive datetime object."""
-    return datetime.now(timezone.utc).replace(tzinfo=None)
+import pytz
 
 
 # Sflock does a good filetype recon
@@ -26,7 +22,6 @@ def _utcnow_naive():
 
 from lib.cuckoo.common.cape_utils import static_config_lookup, static_extraction
 from lib.cuckoo.common.colors import red
-from lib.cuckoo.common.config import Config
 from lib.cuckoo.common.constants import CUCKOO_ROOT
 from lib.cuckoo.common.demux import demux_sample
 from lib.cuckoo.common.exceptions import (
@@ -36,6 +31,7 @@ def _utcnow_naive():
     CuckooOperationalError,
     CuckooUnserviceableTaskError,
 )
+from lib.cuckoo.common.config import Config
 from lib.cuckoo.common.integrations.parse_pe import PortableExecutable
 from lib.cuckoo.common.objects import PCAP, URL, File, Static
 from lib.cuckoo.common.path_utils import path_delete, path_exists
@@ -81,6 +77,17 @@ def _utcnow_naive():
 except ImportError:  # pragma: no cover
     raise CuckooDependencyError("Unable to import sqlalchemy (install with `poetry install`)")
 
+cfg = Config("cuckoo")
+tz_name = cfg.cuckoo.get("timezone", "utc")
+
+def _utcnow_naive():
+    """Returns the current time in the configured timezone as a naive datetime object."""
+    try:
+        tz = pytz.timezone(tz_name)
+    except pytz.UnknownTimeZoneError:
+        tz = timezone.utc
+    return datetime.now(tz).replace(tzinfo=None)
+
 
 sandbox_packages = (
     "access",

tests/test_analysis_manager.py

@@ -108,37 +108,7 @@ class TestAnalysisManager:
     def test_init(self, task: Task):
         mgr = AnalysisManager(task=task)
 
-        assert mgr.cfg.cuckoo == {
-            "allow_static": False,
-            "categories": "static, pcap, url, file",
-            "freespace": 50000,
-            "delete_original": False,
-            "tmppath": "/tmp",
-            "terminate_processes": False,
-            "memory_dump": False,
-            "delete_bin_copy": False,
-            "max_machines_count": 10,
-            "reschedule": False,
-            "rooter": "/tmp/cuckoo-rooter",
-            "machinery": "kvm",
-            "machinery_screenshots": False,
-            "delete_archive": True,
-            "max_vmstartup_count": 5,
-            "daydelta": 0,
-            "max_analysis_count": 0,
-            "max_len": 196,
-            "sanitize_len": 32,
-            "sanitize_to_len": 24,
-            "scaling_semaphore": False,
-            "scaling_semaphore_update_timer": 10,
-            "task_pending_timeout": 0,
-            "task_timeout": False,
-            "task_timeout_scan_interval": 30,
-            "freespace_processing": 15000,
-            "ignore_signals": True,
-            "periodic_log": False,
-            "fail_unserviceable": True,
-        }
+        assert mgr.cfg.cuckoo == Config("cuckoo").cuckoo
 
         assert mgr.task.id == task.id