Skip to content

Commit 96e3eda

Browse files
committed
2.3.3 - Support new protobuf library version (#21)
1 parent 3ff3888 commit 96e3eda

File tree

2 files changed

+12
-6
lines changed

2 files changed

+12
-6
lines changed

FlipperNested/bridge.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import inspect
12
import serial
23
import serial.tools.list_ports
34
from google.protobuf.internal.encoder import _VarintBytes
@@ -122,16 +123,21 @@ def get_files(self, path="/ext") -> list:
122123
cmd_data.path = path
123124
rep_data = self._rpc_send_and_read_answer(cmd_data, "storage_list_request")
124125

125-
storage_response.extend(
126-
MessageToDict(message=rep_data.storage_list_response, including_default_value_fields=True, )["file"])
126+
storage_response.extend(self._message_to_dict(rep_data.storage_list_response)["file"])
127127

128128
while rep_data.has_next:
129129
rep_data = self._rpc_read_answer()
130-
storage_response.extend(
131-
MessageToDict(message=rep_data.storage_list_response, including_default_value_fields=True, )["file"])
130+
storage_response.extend(self._message_to_dict(rep_data.storage_list_response)["file"])
132131

133132
return storage_response
134133

134+
@staticmethod
135+
def _message_to_dict(message):
136+
if 'including_default_value_fields' in inspect.signature(MessageToDict).parameters:
137+
return MessageToDict(message=message, including_default_value_fields=True)
138+
else:
139+
return MessageToDict(message=message, always_print_fields_with_no_presence=True)
140+
135141
def file_read(self, path=None):
136142
storage_response = []
137143
cmd_data = storage_pb2.ReadRequest()
@@ -182,7 +188,7 @@ def mkdir(self, path="/ext"):
182188
cmd_data = storage_pb2.MkdirRequest()
183189
cmd_data.path = path
184190

185-
rep_data = self._rpc_send_and_read_answer(cmd_data, "storage_mkdir_request")
191+
self._rpc_send_and_read_answer(cmd_data, "storage_mkdir_request")
186192

187193
def file_rename(self, old, new):
188194
cmd_data = storage_pb2.RenameRequest()

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
with open("README.md", "r") as fh:
3434
long_description = fh.read()
3535

36-
setuptools.setup(name="FlipperNested", version="2.3.2", author="AloneLiberty",
36+
setuptools.setup(name="FlipperNested", version="2.3.3", author="AloneLiberty",
3737
description="Recover keys from collected nonces", long_description=long_description,
3838
long_description_content_type="text/markdown",
3939
url="https://github.com/AloneLiberty/FlipperNestedRecovery",

0 commit comments

Comments
 (0)