🗓 Nov 4, 2025 8:57:45 PM

🐛 fix with with write_bytes
✨ salsa20 encrypt/decrypt

Author: securisec

chepy/core.py

@@ -1096,10 +1096,11 @@ def write_binary(self, path: str) -> None:  # pragma: no cover
         Examples:
             >>> c = Chepy("some data").write_binary('/some/path/file')
         """
+        data = self._convert_to_bytes()
         if isinstance(path, bytes):  # pragma: no cover
             path = path.decode()
         with open(str(self._abs_path(path)), "wb+") as f:
-            f.write(self.state)
+            f.write(data)
         self._info_logger("File written to {}".format(self._abs_path(path)))
         return None
 

chepy/modules/encryptionencoding.py

@@ -22,6 +22,7 @@
 
 
 AES = lazy_import.lazy_module("Crypto.Cipher.AES")
+Salsa20 = lazy_import.lazy_module("Crypto.Cipher.Salsa20")
 ARC4 = lazy_import.lazy_module("Crypto.Cipher.ARC4")
 DES = lazy_import.lazy_module("Crypto.Cipher.DES")
 ChaCha20 = lazy_import.lazy_module("Crypto.Cipher.ChaCha20")
@@ -705,6 +706,58 @@ def des_decrypt(
             self.state = cipher.decrypt(self._convert_to_bytes())
             return self
 
+    @ChepyDecorators.call_stack
+    def salsa20_encrypt(
+        self,
+        key: str,
+        nonce: str = "0000000000000000",
+        key_format: str = "hex",
+        nonce_format: str = "hex",
+    ) -> EncryptionEncodingT:
+        """Encrypt raw state with Salsa 20 rounds
+
+        Args:
+            key (str): Required. The secret key
+            nonce (str, optional): Nonce. Defaults to '0000000000000000'.
+            key_format (str, optional): Format of key. Defaults to 'hex'.
+            nonce_format (str, optional): Format of nonce. Defaults to 'hex'.
+
+        Returns:
+            Chepy: The Chepy object.
+        """
+
+        key, nonce = self._convert_key(key, nonce, key_format, nonce_format)
+
+        cipher = Salsa20.new(key=key, nonce=nonce)
+        self.state = cipher.encrypt(self._convert_to_bytes())
+        return self
+
+    @ChepyDecorators.call_stack
+    def salsa20_decrypt(
+        self,
+        key: str,
+        nonce: str = "0000000000000000",
+        key_format: str = "hex",
+        nonce_format: str = "hex",
+    ) -> EncryptionEncodingT:
+        """Decrypt raw state encrypted with ChaCha 20 rounds.
+
+        Args:
+            key (str): Required. The secret key
+            nonce (str, optional): nonce for certain modes only. Defaults to '0000000000000000'.
+            key_format (str, optional): Format of key. Defaults to 'hex'.
+            nonce_format (str, optional): Format of nonce. Defaults to 'hex'.
+
+        Returns:
+            Chepy: The Chepy object.
+        """
+
+        key, nonce = self._convert_key(key, nonce, key_format, nonce_format)
+
+        cipher = Salsa20.new(key=key, nonce=nonce)
+        self.state = cipher.decrypt(self._convert_to_bytes())
+        return self
+
     @ChepyDecorators.call_stack
     def chacha_encrypt(
         self,

chepy/modules/encryptionencoding.pyi

@@ -31,6 +31,8 @@ class EncryptionEncoding(ChepyCore):
     def rc4_decrypt(self: EncryptionEncodingT, key: str, key_format: RC4_FORMAT=...) -> EncryptionEncodingT: ...
     def des_encrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: Literal["CBC", "OFB", "CTR", "ECB"]="CBC", key_format: FORMAT="hex", iv_format: FORMAT="hex") -> EncryptionEncodingT: ...
     def des_decrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: Literal["CBC", "OFB", "CTR", "ECB"]="CBC", key_format: FORMAT="hex", iv_format: FORMAT="hex") -> EncryptionEncodingT: ...
+    def salsa20_encrypt(self: EncryptionEncodingT, key: str, nonce: str=..., key_format: FORMAT="hex", nonce_format: FORMAT="hex") -> EncryptionEncodingT: ...
+    def salsa20_decrypt(self: EncryptionEncodingT, key: str, nonce: str=..., key_format: FORMAT="hex", nonce_format: FORMAT="hex") -> EncryptionEncodingT: ...
     def chacha_encrypt(self: EncryptionEncodingT, key: str, nonce: str=..., key_format: FORMAT=..., nonce_format: FORMAT=...) -> EncryptionEncodingT: ...
     def chacha_decrypt(self: EncryptionEncodingT, key: str, nonce: str=..., key_format: FORMAT=..., nonce_format: FORMAT=...) -> EncryptionEncodingT: ...
     def triple_des_encrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: Literal["CBC", "OFB", "CTR", "ECB"]="CBC", key_format: FORMAT="hex", iv_format: FORMAT="hex") -> EncryptionEncodingT: ...

tests/test_encryptionencoding.py

@@ -766,6 +766,29 @@ def test_monoalphabetic_substitution():
     assert Chepy("lol").monoalphabetic_substitution({"l": "t", "o": "s"}).o == b"tst"
 
 
+def test_salsa20():
+    data = "securisec"
+    enc_data = "07 89 19 8b d9 85 59 c7 45".replace(" ", "")
+    assert (
+        Chepy(data)
+        .salsa20_encrypt(
+            key="ff000000000000000000000000000000", nonce="0000000000000000"
+        )
+        .to_hex()
+        .o
+        == enc_data.encode()
+    )
+    assert (
+        Chepy(enc_data)
+        .from_hex()
+        .salsa20_decrypt(
+            key="ff000000000000000000000000000000", nonce="0000000000000000"
+        )
+        .o
+        == data.encode()
+    )
+
+
 def test_chacha_decrypt():
     assert (
         Chepy("0d118d5f5807747b085473553146a3c76cf1ef61b976519240")