Add support for unknown func

2024-10-20 16:07:23 +02:00 · 2024-10-20 16:07:23 +02:00 · 33c8f909be
parent e84686fa97
commit 33c8f909be
1 changed files with 2 additions and 2 deletions
--- 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)