Merge pull request #52 from JarbasHiveMind/release-0.2.0a1

Release 0.2.0a1

Author: JarbasAl

CHANGELOG.md

@@ -1,12 +1,12 @@
 # Changelog
 
-## [0.1.6a1](https://github.com/JarbasHiveMind/hivemind-websocket-client/tree/0.1.6a1) (2025-01-01)
+## [0.2.0a1](https://github.com/JarbasHiveMind/hivemind-websocket-client/tree/0.2.0a1) (2025-01-03)
 
-[Full Changelog](https://github.com/JarbasHiveMind/hivemind-websocket-client/compare/0.1.5...0.1.6a1)
+[Full Changelog](https://github.com/JarbasHiveMind/hivemind-websocket-client/compare/0.1.6...0.2.0a1)
 
 **Merged pull requests:**
 
-- fix: dont leak session data / implement client HELLO [\#48](https://github.com/JarbasHiveMind/hivemind-websocket-client/pull/48) ([JarbasAl](https://github.com/JarbasAl))
+- feat:add chacha20 cipher + z85 encoding [\#50](https://github.com/JarbasHiveMind/hivemind-websocket-client/pull/50) ([JarbasAl](https://github.com/JarbasAl))
 
 
 

hivemind_bus_client/client.py

@@ -17,8 +17,9 @@
 from hivemind_bus_client.message import HiveMessage, HiveMessageType
 from hivemind_bus_client.serialization import HiveMindBinaryPayloadType
 from hivemind_bus_client.serialization import get_bitstring, decode_bitstring
-from hivemind_bus_client.util import serialize_message, \
-    encrypt_as_json, decrypt_from_json, encrypt_bin, decrypt_bin
+from hivemind_bus_client.util import serialize_message
+from hivemind_bus_client.encryption import (encrypt_as_json, decrypt_from_json, encrypt_bin, decrypt_bin,
+                                            SupportedEncodings, SupportedCiphers)
 from poorman_handshake.asymmetric.utils import encrypt_RSA, load_RSA_key, sign_RSA
 
 
@@ -104,6 +105,8 @@ def __init__(self, key: Optional[str] = None,
                  internal_bus: Optional[OVOSBusClient] = None,
                  bin_callbacks: BinaryDataCallbacks = BinaryDataCallbacks()):
         self.bin_callbacks = bin_callbacks
+        self.json_encoding = SupportedEncodings.JSON_HEX  # server defaults before it was made configurable
+        self.cipher = SupportedCiphers.AES_GCM  # server defaults before it was made configurable
 
         self.identity = identity or None
         self._password = password
@@ -271,11 +274,12 @@ def on_message(self, *args):
         if self.crypto_key:
             # handle binary encryption
             if isinstance(message, bytes):
-                message = decrypt_bin(self.crypto_key, message)
+                message = decrypt_bin(self.crypto_key, message, cipher=self.cipher)
             # handle json encryption
             elif "ciphertext" in message:
                 # LOG.debug(f"got encrypted message: {len(message)}")
-                message = decrypt_from_json(self.crypto_key, message)
+                message = decrypt_from_json(self.crypto_key, message,
+                                            cipher=self.cipher, encoding=self.json_encoding)
             else:
                 LOG.debug("Message was unencrypted")
 
@@ -367,14 +371,15 @@ def emit(self, message: Union[MycroftMessage, HiveMessage],
                                        binary_type=binary_type,
                                        hivemeta=message.metadata)
                 if self.crypto_key:
-                    ws_payload = encrypt_bin(self.crypto_key, bitstr.bytes)
+                    ws_payload = encrypt_bin(self.crypto_key, bitstr.bytes, cipher=self.cipher)
                 else:
                     ws_payload = bitstr.bytes
                 self.client.send(ws_payload, ABNF.OPCODE_BINARY)
             else:
                 ws_payload = serialize_message(message)
                 if self.crypto_key:
-                    ws_payload = encrypt_as_json(self.crypto_key, ws_payload)
+                    ws_payload = encrypt_as_json(self.crypto_key, ws_payload,
+                                                 cipher=self.cipher, encoding=self.json_encoding)
                 self.client.send(ws_payload)
 
         except WebSocketConnectionClosedException: