fix: #240 (#245)

* fix: #240

.github/workflows/pylint.yml

@@ -2,7 +2,7 @@ name: PyLint
 on: [pull_request]
 
 jobs:
-  build:
+  pylint:
     strategy:
       matrix:
         python-version: ['3.10']

pocsuite3/lib/core/log.py

@@ -9,6 +9,7 @@
 logging.addLevelName(CUSTOM_LOGGING.WARNING, "!")
 
 LOGGER = logging.getLogger("pocsuite")
+sys.stdout = open(sys.stdout.fileno(), mode='w', encoding='utf8', buffering=1)
 LOGGER_HANDLER = logging.StreamHandler(sys.stdout)
 PRIMARY_FMT = (
     "%(cyan)s[%(asctime)s] %(log_color)s[%(levelname)s]%(reset)s %(message)s"

tests/login_demo.py

@@ -38,7 +38,6 @@ def _verify(self):
         result = {}
         payload = "username={0}&password={1}".format(self.get_option("username"), self.get_option("password"))
         r = requests.post(self.url, data=payload)
-        print(r.text)
         if r.status_code == 200:
             result['VerifyInfo'] = {}
             result['VerifyInfo']['URL'] = self.url

tests/test_cmd_diy_options.py

@@ -22,7 +22,10 @@ def test_cmd_run(self):
 
         eval_path = os.path.join(path, "../pocsuite3/cli.py")
         poc_path = os.path.join(path, "login_demo.py")
-        command = '''python3 {0} -u http://www.baidu.com -r {1} --verify -v 2  --password mypass123 --username "asd asd" --testt abctest'''.format(
-            eval_path, poc_path)
+        command = (
+            f'python3 {eval_path} -u http://www.baidu.com -r {poc_path} --verify -v 2  --password mypass123 '
+            '--username "asd asd" --testt abctest'
+        )
         pipeline = os.popen(command)
-        self.assertTrue('1 / 1' in pipeline.read())
+        res = pipeline.buffer.read().decode('utf-8')
+        self.assertTrue('1 / 1' in res)

tests/test_show_poc_options.py

@@ -15,6 +15,9 @@ def verify_result(self):
         pass
 
     def test_cmd_run(self):
-        command = 'pocsuite -r ecshop_rce.py --options'
-        res = os.popen(command).read()
+        pipeline = os.popen('pocsuite -r ecshop_rce.py --options')
+
+        # os.popen default encoding may not be utf-8
+        res = pipeline.buffer.read().decode('utf-8')
+
         self.assertTrue('You can select dict_keys' in res)