diff options
Diffstat (limited to 'tools')
| -rwxr-xr-x | tools/wasptool | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/tools/wasptool b/tools/wasptool index c80a5b7..befcd98 100755 --- a/tools/wasptool +++ b/tools/wasptool @@ -254,7 +254,15 @@ def handle_binary_download(c, tname, fname): while True: draw_pbar(100 * bytes_read / sz, verbose) reply = c.run_command('repr(f.read(24))') - data = eval(reply.replace('\\\\', '\\').strip('"')) + reply = reply.replace('\\\\', '\\') + if reply.startswith('"'): + # "b'..CONTENT..'" + reply = reply[1:-1] + elif reply.startswith("'"): + # 'b\'..CONTENT..\'' + reply = reply[1:-1].replace("\\'", "'") + data = print(reply) + data = eval(reply) if len(data) == 0: break bytes_read += len(data) |