pyntcli 0.1.81__py3-none-any.whl → 0.1.82__py3-none-any.whl

pyntcli/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.81"
+__version__ = "0.1.82"

pyntcli/commands/burp.py

@@ -10,6 +10,7 @@ from subprocess import Popen, PIPE
 from functools import partial
 import xmltodict
 import base64
+import pyntcli.log.log as log
 from xml.parsers.expat import ExpatError
 
 from pyntcli.pynt_docker import pynt_container
@@ -30,6 +31,9 @@ methods = [
     "connect",
 ]
 
+logger = log.get_logger()
+supported_encoding = ["utf-8", "latin1", "utf-16", "cp1252"]
+
 
 def is_valid_method(method: str):
     if method.lower() in methods:
@@ -37,9 +41,31 @@ def is_valid_method(method: str):
     return False
 
 
+def decode_request(item) -> str:
+    error = None
+    for encoding in supported_encoding:
+        try:
+            return base64.b64decode(item["request"]["#text"]).decode(encoding)
+        except UnicodeDecodeError as e:
+            error = e
+            continue
+        except Exception as e:
+            error = e
+            break
+
+    ui_thread.print(
+        ui_thread.PrinterText(
+            f"Error decoding request: {error}"),
+        ui_thread.PrinterText.WARNING)
+    logger.error(f"Error decoding request: {error}")
+
+    raise e
+
+
 def replay_req(item, proxy_port):
     url = item["url"]
-    decoded_req = base64.b64decode(item["request"]["#text"]).decode("utf-8")
+    decoded_req = decode_request(item)
+
     method = decoded_req.split("\r\n")[0].split(" ")[0]
 
     if not is_valid_method(method):
@@ -289,12 +315,12 @@ class BurpCommand(sub_command.PyntSubCommand):
                 json_report_path = util.get_user_report_path(full_path, "json")
 
             if html_report:
-                with open(html_report_path, "w",encoding="utf-8") as html_file:
+                with open(html_report_path, "w", encoding="utf-8") as html_file:
                     html_file.write(html_report)
                 webbrowser.open("file://{}".format(html_report_path))
 
             if json_report:
-                with open(json_report_path, "w",encoding="utf-8") as json_file:
+                with open(json_report_path, "w", encoding="utf-8") as json_file:
                     json_file.write(json_report)
                 reporter = cli_reporter.PyntReporter(json_report_path)
                 reporter.print_summary()

{pyntcli-0.1.81.dist-info → pyntcli-0.1.82.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyntcli
-Version: 0.1.81
+Version: 0.1.82
 Summary: Command line utility to handle all of Pynt's different integrations
 Author-email: Pynt-io <support@pynt.io>
 Project-URL: Homepage, https://pynt.io

{pyntcli-0.1.81.dist-info → pyntcli-0.1.82.dist-info}/RECORD

@@ -1,11 +1,11 @@
-pyntcli/__init__.py,sha256=4reVRut6pFQ0MxDHMjUh2C4vPGa3csSjDj46nUuBiXs,23
+pyntcli/__init__.py,sha256=YThvBsPSPw2rCgZQb5SxE-vfr4Y0SBQRhFo1VmMzEQg,23
 pyntcli/main.py,sha256=qTGMG2JSpieg-1RRgBz9dPL4vHo1WJJIxTPQYAuzxuY,5138
 pyntcli/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/analytics/send.py,sha256=pJOyOWl3g_Vm9apKK3LzNVqsnC6zsWA1bCK3ZegbLpc,3637
 pyntcli/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/auth/login.py,sha256=TljsRXbEkNI1YUrKm5mlTw4YiecYScYUsit8Z8vstss,5228
 pyntcli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pyntcli/commands/burp.py,sha256=KuxeX9I0ZXSVAMHdt9OYFok79X9IK6z71-ZN-_fnrvU,10565
+pyntcli/commands/burp.py,sha256=2FqLc65SjuXdenUS1ggC6dGRU_iSA52Ft_8OnfGsm_8,11193
 pyntcli/commands/command.py,sha256=0O7Za_cjT6vDkDfM0OTMPB6DLI3U1r1R7lXQydz7458,9495
 pyntcli/commands/har.py,sha256=mSCbTUnxQrKzJd-dAWoc6Tkw6tU1LDH7Ha1w2ylrrrg,3654
 pyntcli/commands/id_command.py,sha256=UBEgMIpm4vauTCsKyixltiGUolNg_OfHEJvJ_i5BpJY,943
@@ -35,8 +35,8 @@ pyntcli/ui/ui_thread.py,sha256=OVTbiIFMg2KgxAvHf7yy86xGm4RVS2vj_VYZkMi-SRY,4956
 tests/conftest.py,sha256=gToq5K74GtgeGQXjFvXSzMaE6axBYxAzcFG5XJPOXjI,427
 tests/auth/test_login.py,sha256=KFlzWhXBAuwdi7GXf16gCB3ya94LQG2wjcSChE149rQ,3798
 tests/store/test_cred_store.py,sha256=_7-917EtNC9eKEumO2_lt-7KuDmCwOZFaowCm7DbA_A,254
-pyntcli-0.1.81.dist-info/METADATA,sha256=D5hWFtXYjvdJp7uIjMBlj8Z_yYP0pobzSV1Ce6T1onM,463
-pyntcli-0.1.81.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-pyntcli-0.1.81.dist-info/entry_points.txt,sha256=kcGmqAxXDttNk2EPRcqunc_LTVp61gzakz0v-GEE2SY,43
-pyntcli-0.1.81.dist-info/top_level.txt,sha256=u9MDStwVHB7UG8PUcODeWCul_NvzL2EzoLvSlgwLHFs,30
-pyntcli-0.1.81.dist-info/RECORD,,
+pyntcli-0.1.82.dist-info/METADATA,sha256=WpnF74sRKWiYhqK37FpYb6ggOl_FscYOIjVjaAedZuU,463
+pyntcli-0.1.82.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+pyntcli-0.1.82.dist-info/entry_points.txt,sha256=kcGmqAxXDttNk2EPRcqunc_LTVp61gzakz0v-GEE2SY,43
+pyntcli-0.1.82.dist-info/top_level.txt,sha256=u9MDStwVHB7UG8PUcODeWCul_NvzL2EzoLvSlgwLHFs,30
+pyntcli-0.1.82.dist-info/RECORD,,