From 33c8f909bed902b9f7c8d35e06e752e2a9d91388 Mon Sep 17 00:00:00 2001
From: hyugogirubato <65763543+hyugogirubato@users.noreply.github.com>
Date: Sun, 20 Oct 2024 16:07:23 +0200
Subject: [PATCH] Add support for unknown func

---
 keydive/core.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/keydive/core.py b/keydive/core.py
index 6bb7249..792e615 100644
--- a/keydive/core.py
+++ b/keydive/core.py
@@ -187,10 +187,10 @@ class Core:
             logger.log(level=level, msg=data.decode('utf-8'))
             if level in (logging.FATAL, logging.CRITICAL):
                 self.running = False
+        elif isinstance(level, dict) and 'private_key' in level:
+            self.cdm.set_private_key(data=data, name=level['private_key'])
         elif level == 'challenge':
             self.cdm.set_challenge(data=data)
-        elif level == 'private_key':
-            self.cdm.set_private_key(data=data)
         elif level == 'client_id':
             self.cdm.set_client_id(data=data)