Feature: Infoblox manage session with cookies in integration storage

Packs/Infoblox/CONTRIBUTORS.json

@@ -0,0 +1,3 @@
+[
+    "Jesús García Potes <jesusgarciapotes95@gmail.com>"
+]

Packs/Infoblox/Integrations/Infoblox/Infoblox.py

@@ -1,6 +1,8 @@
-from enum import Enum, unique
 import demistomock as demisto  # noqa: F401
 from CommonServerPython import *  # noqa: F401
+from enum import Enum, unique
+
+
 import json
 
 ''' IMPORTS '''
@@ -17,6 +19,7 @@
 INTEGRATION_CONTEXT_NAME = 'Infoblox'
 INTEGRATION_HOST_RECORDS_CONTEXT_NAME = "Host"
 INTEGRATION_NETWORK_INFO_CONTEXT_KEY = "NetworkInfo"
+INTEGRATION_AUTHORIZATION_EXCEPTION_MESSAGE = "Authorization error, check your credentials."
 
 # COMMON RAW RESULT KEYS
 INTEGRATION_COMMON_RAW_RESULT_REFERENCE_KEY = "_ref"
@@ -161,6 +164,69 @@ class IPv4AddressStatus(Enum):
     USED = "USED"
 
 
+def inject_cookies(func: Callable) -> Callable:
+    """
+    Decorator to manage session persistence and handle authentication for API requests.
+
+    This decorator attempts to execute the provided function using existing session cookies
+    stored in the 'integration_context'. If no valid cookies are available, or if the existing
+    session is no longer valid the auth generate new cookies to save, bad credentials force the
+    storage clean.
+
+    The decorator handles saving and loading cookies between different executions, allowing for
+    session persistence across multiple API calls.
+
+    Args:
+        func (Callable): The API request function to be executed.
+
+    Raises:
+        DemistoException: If the API request fails.
+
+    Returns:
+        Callable: The result from executing 'func' with the provided arguments and keyword arguments.
+    """
+
+    @wraps(wrapped=func)
+    def wrapper(client: "InfoBloxNIOSClient", *args, **kwargs):
+
+        def save_cookies_to_context(client: "InfoBloxNIOSClient") -> None:
+            cookies_dict = {}
+            for cookie in client._session.cookies:
+                cookies_dict[cookie.name] = {
+                    'value': cookie.value,
+                    'domain': cookie.domain,
+                    'path': cookie.path
+                }
+            set_integration_context({'cookies': cookies_dict})
+
+        def load_cookies(client: "InfoBloxNIOSClient", cookies_dict: dict) -> None:
+            for name, cookie_data in cookies_dict.items():
+                client._session.cookies.set(
+                    name,
+                    cookie_data['value'],
+                    domain=cookie_data['domain'],
+                    path=cookie_data['path']
+                )
+
+        integration_context = get_integration_context()
+        if (
+            integration_context
+            and (context_cookies := integration_context.get("cookies"))
+        ):
+            load_cookies(client, context_cookies)
+
+        try:
+            response = func(client, *args, **kwargs)
+            save_cookies_to_context(client)
+            return response
+        except DemistoException as error:
+            if error.message and error.message == INTEGRATION_AUTHORIZATION_EXCEPTION_MESSAGE:
+                set_integration_context({})
+            raise error
+
+    return wrapper
+
+
 class InfoBloxNIOSClient(BaseClient):
 
     REQUEST_PARAMS_RETURN_AS_OBJECT_KEY = '_return_as_object'
@@ -182,6 +248,7 @@ def __init__(self, base_url, verify=True, proxy=False, ok_codes=(), headers=None
         super().__init__(base_url, verify, proxy, ok_codes, headers, auth)
         self.params: dict[str, Any] = {self.REQUEST_PARAMS_RETURN_AS_OBJECT_KEY: '1'}
 
+    @inject_cookies
     def _http_request(  # type: ignore[override]
         self, method, url_suffix, full_url=None, headers=None, auth=None,
         json_data=None, params=None, data=None, files=None,
@@ -584,7 +651,7 @@ def get_network_info(
 def parse_demisto_exception(error: DemistoException, field_in_error: str = 'text'):
     err_msg = err_string = error.args[0]
     if '[401]' in err_string:
-        err_msg = 'Authorization error, check your credentials.'
+        err_msg = INTEGRATION_AUTHORIZATION_EXCEPTION_MESSAGE
     elif 'Failed to parse json object' in err_string:
         err_msg = 'Cannot connect to Infoblox server, check your proxy and connection.'
     elif 'Error in API call' in err_string:

Packs/Infoblox/Integrations/Infoblox/Infoblox.yml

@@ -784,7 +784,7 @@ script:
     - contextPath: Infoblox.NetworkInfo.AdditionalFields
       description: The additional fields for network.
       type: Unknown
-  dockerimage: demisto/python3:3.11.10.115186
+  dockerimage: demisto/python3:3.11.10.116949
   runonce: false
   script: ''
   subtype: python3

Packs/Infoblox/ReleaseNotes/1_1_3.md

@@ -2,7 +2,4 @@
 #### Integrations
 
 ##### Infoblox
-- Updated the Docker image to: *demisto/python3:3.11.10.115186*.
-
-
-
+- Updated the Docker image to: *demisto/python3:3.11.10.115186*.

Packs/Infoblox/ReleaseNotes/1_1_4.md

@@ -0,0 +1,7 @@
+
+#### Integrations
+
+##### Infoblox
+
+- Persisting Session with Cookies: Added logic to persist the session using cookies, enabling reusability of authentication tokens across requests. This prevents the need to log in on every API call, improving efficiency and reducing redundant login operations.
+- Updated the Docker image to: *demisto/python3:3.11.10.116949*.

Packs/Infoblox/pack_metadata.json

@@ -2,7 +2,7 @@
     "name": "Infoblox NIOS",
     "description": "Infoblox is a comprehensive solution that consolidates DNS, DHCP, and IP address management into a single platform. It is designed to simplify network management by automating these critical functions and providing a centralized console for managing them.",
     "support": "xsoar",
-    "currentVersion": "1.1.3",
+    "currentVersion": "1.1.4",
     "author": "Cortex XSOAR",
     "url": "https://www.paloaltonetworks.com/cortex",
     "email": "",