diff --git a/scripts/bdecode.py b/scripts/bdecode.py index 2e01bca..d29405e 100755 --- a/scripts/bdecode.py +++ b/scripts/bdecode.py @@ -1,10 +1,10 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- -import bencoder import click import json from pprint import pprint +from fastbencode import bdecode def debyte_list(lst): @@ -45,10 +45,10 @@ def main(**kwargs): with open(input_file, 'rb') as fd: data = fd.read() if no_decode: - d = bencoder.bdecode(data[8:]) + d = bdecode(data[8:]) pprint(d) else: - d = debyte_dict(bencoder.bdecode(data[8:])) + d = debyte_dict(bdecode(data[8:])) print(json.dumps(d, indent=4)) diff --git a/scripts/bencode.py b/scripts/bencode.py index 7856902..10fb5a8 100755 --- a/scripts/bencode.py +++ b/scripts/bencode.py @@ -1,10 +1,10 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- -import bencoder import click import json from pprint import pprint +from fastbencode import bencode_utf8 def enbyte_list(lst): @@ -62,7 +62,7 @@ def main(**kwargs): data = enbyte_dict(data) with open(output_file, 'wb') as wfd: - wfd.write(MAGIC_UNENC + bencoder.bencode(data)) + wfd.write(MAGIC_UNENC + bencode_utf8(data)) if __name__ == '__main__': diff --git a/src/jmclient/storage.py b/src/jmclient/storage.py index 72730cc..786361c 100644 --- a/src/jmclient/storage.py +++ b/src/jmclient/storage.py @@ -414,11 +414,11 @@ class DKGRecoveryStorage(object): @staticmethod def _serialize(data): - return bencoder.bencode(data) + return bencode_utf8(data) @staticmethod def _deserialize(data): - return bencoder.bdecode(data) + return bdecode(data) @staticmethod def _get_lock_filename(path: str) -> str: diff --git a/test/jmclient/test_frost_wallet.py b/test/jmclient/test_frost_wallet.py index 36f7f36..58145c8 100644 --- a/test/jmclient/test_frost_wallet.py +++ b/test/jmclient/test_frost_wallet.py @@ -5,7 +5,7 @@ from pprint import pprint from unittest import IsolatedAsyncioTestCase -import bencoder +from fastbencode import bdecode import jmclient # install asyncioreactor from twisted.internet import reactor @@ -240,7 +240,7 @@ class AsyncioTestCase(IsolatedAsyncioTestCase): ext_recovery_bytes = decrypt_ext_recovery(wlt._hostseckey, ext_recovery) - saved_dkg = bencoder.bdecode(wlt._dkg_storage.file_data[8:]) + saved_dkg = bdecode(wlt._dkg_storage.file_data[8:]) STORAGE_KEY = DKGManager.STORAGE_KEY HOSTPUBKEYS_SUBKEY = DKGManager.HOSTPUBKEYS_SUBKEY PUBKEY_SUBKEY = DKGManager.PUBKEY_SUBKEY @@ -256,12 +256,12 @@ class AsyncioTestCase(IsolatedAsyncioTestCase): assert saved_dkg[STORAGE_KEY][T_SUBKEY] == dict() assert saved_dkg[STORAGE_KEY][SESSIONS_SUBKEY] == dict() - saved_rec = bencoder.bdecode(wlt._recovery_storage.file_data[8:]) + saved_rec = bdecode(wlt._recovery_storage.file_data[8:]) assert saved_rec[b'dkg'] == dict() wlt.dkg.save() - saved_dkg = bencoder.bdecode(wlt._dkg_storage.file_data[8:]) + saved_dkg = bdecode(wlt._dkg_storage.file_data[8:]) assert set(saved_dkg[STORAGE_KEY][SECSHARE_SUBKEY].keys()) == set([ b'\xaa'*32, b'\xbb'*32, @@ -286,7 +286,7 @@ class AsyncioTestCase(IsolatedAsyncioTestCase): ext_recovery_bytes ]) - saved_rec = bencoder.bdecode(wlt._recovery_storage.file_data[8:]) + saved_rec = bdecode(wlt._recovery_storage.file_data[8:]) RECOVERY_STORAGE_KEY = DKGManager.RECOVERY_STORAGE_KEY assert set(saved_rec[RECOVERY_STORAGE_KEY].keys()) == set([ b'\xaa'*32,