Add files via upload

PYAS.py

@@ -1,6 +1,5 @@
-import os, gc, sys, time, json, psutil
-import hashlib, pefile, socket, msvcrt
-import xml.etree.ElementTree as xmlet
+import os, gc, sys, time, json, yara
+import pefile, socket, msvcrt, psutil
 import requests, pyperclip, win32file
 import win32gui, win32api, win32con
 from PYAS_Engine import ListSimHash
@@ -19,11 +18,12 @@ class MainWindow_Controller(QMainWindow):
         self.setWindowFlags(Qt.FramelessWindowHint)
         self.pyas = sys.argv[0].replace("\\", "/")
         self.dir = os.path.dirname(self.pyas)
-        self.pyae_version = "SimHash Engine"
-        self.pyas_version = "3.0.6"
+        self.pyae_version = "Fusion Engine"
+        self.pyas_version = "3.1.0"
         self.ui = Ui_MainWindow()
         self.ui.setupUi(self)
-        self.init_show_pyas()
+        self.init_startup()
+        self.init_rule_dict()
         self.init_data_base()
         self.init_read_json()
         self.init_tray_icon()
@@ -50,12 +50,47 @@ class MainWindow_Controller(QMainWindow):
         self.tray_icon.activated.connect(self.init_show_pyas)
         self.tray_icon.show()
 
+    def init_startup(self):
+        try:
+            if len(sys.argv) > 1:
+                param = sys.argv[1]
+                if "h" in param or "hid" in param:
+                    pass
+                else:
+                    self.init_show_pyas()
+            elif len(sys.argv) <= 1:
+                self.init_show_pyas()
+        except Exception as e:
+            self.bug_event(e)
+
     def init_data_base(self):
         try:
-            file_path = os.path.join(self.dir, "PYAS_Model.json")
-            if os.path.exists(file_path):
-                self.pe = ListSimHash()
-                self.pe.load_model(file_path)
+            file_path = os.path.join(self.dir, "Model")
+            for root, dirs, files in os.walk(file_path):
+                for file in files:
+                    model_path = os.path.join(root, file)
+                    ftype = str(f".{file.split('.')[-1]}").lower()
+                    if ftype in [".json", ".txt"]:
+                        self.pe = ListSimHash()
+                        self.pe.load_model(model_path)
+                    elif ftype in [".h5", ".pkl"]:
+                        pass # Not Sp Yet !
+        except Exception as e:
+            self.bug_event(e)
+
+    def init_rule_dict(self):
+        try:
+            self.compiled_rules = {}
+            file_path = os.path.join(self.dir, "Rules")
+            for root, dirs, files in os.walk(file_path):
+                for file in files:
+                    yara_path = os.path.join(root, file)
+                    ftype = str(f".{file.split('.')[-1]}").lower()
+                    if ftype in [".yara", ".yar"]:
+                        rules = yara.compile(yara_path)
+                    elif ftype in [".yc", ".yrc"]:
+                        rules = yara.load(yara_path)
+                    self.compiled_rules[yara_path] = rules
         except Exception as e:
             self.bug_event(e)
 
@@ -93,19 +128,19 @@ class MainWindow_Controller(QMainWindow):
             if not os.path.exists("C:/ProgramData/PYAS"):
                 os.makedirs("C:/ProgramData/PYAS")
             if not os.path.exists("C:/ProgramData/PYAS/PYAS.json"):
-                self.write_config({"high_sensitive":0,"cloud_services":1,
+                self.write_config({"high_sensitive":0,"cloud_services":"http://127.0.0.1/",
                 "language":"en_US","theme_color":"White","theme_custom":""})
             with open("C:/ProgramData/PYAS/PYAS.json", "r") as f:
                 self.json = json.load(f)
         except:
-            self.write_config({"high_sensitive":0,"cloud_services":1,
+            self.write_config({"high_sensitive":0,"cloud_services":"http://127.0.0.1/",
             "language":"en_US","theme_color":"White","theme_custom":""})
             with open("C:/ProgramData/PYAS/PYAS.json", "r") as f:
                 self.json = json.load(f)
 
     def init_config_json(self):
         self.json["high_sensitive"] = self.json.get("high_sensitive", 0)
-        self.json["cloud_services"] = self.json.get("cloud_services", 1)
+        self.json["cloud_services"] = self.json.get("cloud_services", "http://127.0.0.1/")
         self.json["language"] = self.json.get("language", "en_US")
         self.json["theme_color"] = self.json.get("theme_color", "White")
         self.json["theme_custom"] = self.json.get("theme_custom", "")
@@ -114,11 +149,6 @@ class MainWindow_Controller(QMainWindow):
             self.ui.high_sensitivity_switch_Button.setStyleSheet("""
             QPushButton{border:none;background-color:rgb(200, 250, 200);border-radius: 10px;}
             QPushButton:hover{background-color:rgb(210, 250, 210);}""")
-        if self.json["cloud_services"] == 1:
-            self.ui.cloud_services_switch_Button.setText(self.trans("已開啟"))
-            self.ui.cloud_services_switch_Button.setStyleSheet("""
-            QPushButton{border:none;background-color:rgb(200, 250, 200);border-radius: 10px;}
-            QPushButton:hover{background-color:rgb(210, 250, 210);}""")
         if self.json["language"] == "zh_TW":
             self.ui.Language_Traditional_Chinese.setChecked(True)
         elif self.json["language"] == "zh_CN":
@@ -170,7 +200,7 @@ class MainWindow_Controller(QMainWindow):
         self.ui.Protection_switch_Button_4.clicked.connect(self.protect_reg_init)
         self.ui.Protection_switch_Button_5.clicked.connect(self.protect_net_init)
         self.ui.high_sensitivity_switch_Button.clicked.connect(self.change_sensitive)
-        self.ui.cloud_services_switch_Button.clicked.connect(self.change_cloud_service)
+        self.ui.cloud_services_switch_Button.clicked.connect(self.cloud_service)
         self.ui.Add_White_list_Button.clicked.connect(self.add_white_list)
         self.ui.Language_Traditional_Chinese.clicked.connect(self.init_change_lang)
         self.ui.Language_Simplified_Chinese.clicked.connect(self.init_change_lang)
@@ -292,9 +322,9 @@ class MainWindow_Controller(QMainWindow):
         self.ui.high_sensitivity_title.setText(self.trans("高靈敏度模式"))
         self.ui.high_sensitivity_illustrate.setText(self.trans("啟用此選項可以提高引擎的靈敏度"))
         self.ui.high_sensitivity_switch_Button.setText(self.trans(self.ui.high_sensitivity_switch_Button.text()))
-        self.ui.cloud_services_title.setText(self.trans("雲端掃描服務"))
-        self.ui.cloud_services_illustrate.setText(self.trans("啟用此選項可以連接雲端掃描服務"))
-        self.ui.cloud_services_switch_Button.setText(self.trans(self.ui.cloud_services_switch_Button.text()))
+        self.ui.cloud_services_title.setText(self.trans("上報雲端分析"))
+        self.ui.cloud_services_illustrate.setText(self.trans("此選項可以選擇檔案並上報雲端分析"))
+        self.ui.cloud_services_switch_Button.setText(self.trans("選擇"))
         self.ui.Add_White_list_title.setText(self.trans("增加到白名單"))
         self.ui.Add_White_list_illustrate.setText(self.trans("此選項可以選擇檔案並增加到白名單"))
         self.ui.Add_White_list_Button.setText(self.trans(self.ui.Add_White_list_Button.text()))
@@ -564,8 +594,8 @@ class MainWindow_Controller(QMainWindow):
                 self.setWindowOpacity(self.pyas_opacity/100)
             else:
                 self.timer.stop()
-        self.show()
         self.pyas_opacity = 0
+        self.show()
         self.timer = QTimer()
         self.timer.timeout.connect(update_opacity)
         self.timer.start(2)
@@ -589,6 +619,12 @@ class MainWindow_Controller(QMainWindow):
 
     def closeEvent(self, event):
         if self.question_event("您確定要退出 PYAS 和所有防護嗎?"):
+            self.block_window = False
+            self.proc_protect = False
+            self.file_protect = False
+            self.mbr_protect = False
+            self.reg_protect = False
+            self.net_protect = False
             self.virus_scan_break()
             event.accept()
         else:
@@ -627,8 +663,10 @@ class MainWindow_Controller(QMainWindow):
     def show_menu(self):
         self.WindowMenu = QMenu()
         Main_Settings = QAction(self.trans("設定"),self)
+        #Main_Update = QAction(self.trans("更新"),self)
         Main_About = QAction(self.trans("關於"),self)
         self.WindowMenu.addAction(Main_Settings)
+        #self.WindowMenu.addAction(Main_Update)
         self.WindowMenu.addAction(Main_About)
         Qusetion = self.WindowMenu.exec_(self.ui.Menu_Button.mapToGlobal(QPoint(0, 30)))
         if Qusetion == Main_About and self.ui.About_widget.isHidden():
@@ -646,6 +684,15 @@ class MainWindow_Controller(QMainWindow):
             self.ui.Window_widget.raise_()
             self.change_animation_3(self.ui.Setting_widget,0.5)
             self.change_animation_5(self.ui.Setting_widget,10,50,831,481)
+        #if Qusetion == Main_Update:
+            #self.update_database()
+
+    def update_database(self):
+        try:
+            if self.question_event("您確定要更新數據庫嗎?"):
+                pass
+        except Exception as e:
+            self.bug_event(e)
 
     def change_sensitive(self):
         sw_state = self.ui.high_sensitivity_switch_Button.text()
@@ -663,21 +710,17 @@ class MainWindow_Controller(QMainWindow):
             QPushButton:hover{background-color:rgb(210,250,210);}""")
         self.write_config(self.json)
 
-    def change_cloud_service(self):
-        sw_state = self.ui.cloud_services_switch_Button.text()
-        if sw_state == self.trans("已關閉"):
-            self.json["cloud_services"] = 1
-            self.ui.cloud_services_switch_Button.setText(self.trans("已開啟"))
-            self.ui.cloud_services_switch_Button.setStyleSheet("""
-            QPushButton{border:none;background-color:rgb(200,250,200);border-radius: 10px;}
-            QPushButton:hover{background-color:rgb(210,250,210);}""")
-        elif sw_state == self.trans("已開啟"):
-            self.json["cloud_services"] = 0
-            self.ui.cloud_services_switch_Button.setText(self.trans("已關閉"))
-            self.ui.cloud_services_switch_Button.setStyleSheet("""
-            QPushButton{border:none;background-color:rgb(230,230,230);border-radius: 10px;}
-            QPushButton:hover{background-color:rgb(220,220,220);}""")
-        self.write_config(self.json)
+    def cloud_service(self):
+        try:
+            file = str(QFileDialog.getOpenFileName(self,self.trans("上報雲端分析"),"C:/")[0]).replace("\\", "/")
+            if file and self.question_event("您確定要上報雲端分析嗎?"):
+                response = requests.post(self.json["cloud_services"], files={'file': open(file, 'rb')})
+                if response.status_code == 200:
+                    self.info_event(f"成功上報雲端分析: "+file)
+                else:
+                    self.bug_event(response.status_code)
+        except Exception as e:
+            self.bug_event(e)
 
     def init_scan(self):
         try:
@@ -747,14 +790,15 @@ class MainWindow_Controller(QMainWindow):
 
     def write_scan(self, state, file):
         try:
-            self.lock_file(file, True)
-            self.virus_list.append(file)
-            self.virus_list_ui.append(f"[{state}] {file}")
-            item = QListWidgetItem()
-            item.setText(f"[{state}] {file}")
-            item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
-            item.setCheckState(Qt.Checked)
-            self.ui.Virus_Scan_output.addItem(item)
+            if state and file:
+                self.lock_file(file, True)
+                self.virus_list.append(file)
+                self.virus_list_ui.append(f"[{state}] {file}")
+                item = QListWidgetItem()
+                item.setText(f"[{state}] {file}")
+                item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
+                item.setCheckState(Qt.Checked)
+                self.ui.Virus_Scan_output.addItem(item)
         except:
             pass
 
@@ -790,9 +834,9 @@ class MainWindow_Controller(QMainWindow):
     def file_scan(self):
         try:
             file = str(QFileDialog.getOpenFileName(self,self.trans("病毒掃描"),"C:/")[0])
-            if file and file not in self.whitelist:
+            if file and file not in self.whitelist and file != self.pyas:
                 self.init_scan()
-                self.start_scan(file)
+                self.write_scan(self.start_scan(file),file)
                 self.answer_scan()
         except Exception as e:
             self.bug_event(e)
@@ -820,22 +864,6 @@ class MainWindow_Controller(QMainWindow):
             self.bug_event(e)
             self.virus_scan_break()
 
-    def start_scan(self, file):
-        try:
-            file_type = str(f".{file.split('.')[-1]}").lower()
-            if self.json["high_sensitive"] and file != self.pyas:
-                if self.api_scan(file):
-                    self.write_scan(self.trans("惡意"),file)
-                elif self.pe_scan(file):
-                    self.write_scan(self.trans("可疑"),file)
-            elif file_type in slist and file != self.pyas:
-                if self.api_scan(file):
-                    self.write_scan(self.trans("惡意"),file)
-                elif self.pe_scan(file):
-                    self.write_scan(self.trans("可疑"),file)
-        except:
-            pass
-
     def traverse_path(self,path):
         for fd in os.listdir(path):
             try:
@@ -844,33 +872,33 @@ class MainWindow_Controller(QMainWindow):
                     break
                 elif os.path.isdir(file):
                     self.traverse_path(file)
-                elif file not in self.whitelist:
+                elif file not in self.whitelist and file != self.pyas:
                     self.ui.Virus_Scan_text.setText(self.trans(f"正在掃描: ")+file)
                     QApplication.processEvents()
-                    self.start_scan(file)
+                    self.write_scan(self.start_scan(file),file)
                 gc.collect()
             except:
                 pass
 
-    def api_scan(self, file):
+    def start_scan(self, file):
         try:
-            if self.json["cloud_services"]:
-                with open(file, "rb") as f:
-                    text = str(hashlib.md5(f.read()).hexdigest())
-                strBody = f'-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="md5s"\r\n\r\n{text}\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="format"\r\n\r\nXML\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="product"\r\n\r\n360zip\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="combo"\r\n\r\n360zip_main\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="v"\r\n\r\n2\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="osver"\r\n\r\n5.1\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="vk"\r\n\r\na03bc211\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="mid"\r\n\r\n8a40d9eff408a78fe9ec10a0e7e60f62\r\n-------------------------------7d83e2d7a141e--'
-                response = requests.post('http://qup.f.360.cn/file_health_info.php', data=strBody, timeout=5)
-                return response.status_code == 200 and float(xmlet.fromstring(response.text).find('.//e_level').text) > 50
+            ftype = str(f".{file.split('.')[-1]}").lower()
+            label, level = self.pe_scan(file)
+            if label and "Unknown" in label:
+                if self.rule_scan(file):
+                    return "Rules"
+            elif label and "White" not in label:
+                if self.json["high_sensitive"]:
+                    return label
+                elif level and level >= 0.9:
+                    return label
+            elif self.json["high_sensitive"]:
+                if self.rule_scan(file):
+                    return "Rules"
             return False
         except:
             return False
 
-    def sign_scan(self, file):
-        try:
-            with pefile.PE(file, fast_load=True) as pe:
-                return pe.OPTIONAL_HEADER.DATA_DIRECTORY[pefile.DIRECTORY_ENTRY["IMAGE_DIRECTORY_ENTRY_SECURITY"]].VirtualAddress == 0
-        except:
-            return True
-
     def pe_scan(self, file):
         try:
             fn = []
@@ -881,21 +909,36 @@ class MainWindow_Controller(QMainWindow):
                             fn.append(str(func.name, "utf-8"))
                         except:
                             pass
-            out = self.pe.predict_all(fn)
-            v, w = out["Virus"], out["White"]
-            print(v, w, file)
-            if self.json["high_sensitive"]:
-                return v >= w
-            return v > w and v == 1.0
+            QApplication.processEvents()
+            return self.pe.predict(fn)
+        except:
+            return False, False
+
+    def sign_scan(self, file):
+        try:
+            with pefile.PE(file, fast_load=True) as pe:
+                return pe.OPTIONAL_HEADER.DATA_DIRECTORY[pefile.DIRECTORY_ENTRY["IMAGE_DIRECTORY_ENTRY_SECURITY"]].VirtualAddress == 0
+        except:
+            return True
+
+    def rule_scan(self, file):
+        try:
+            with open(file, "rb") as f:
+                data = f.read()
+            for name, rules in self.compiled_rules.items():
+                QApplication.processEvents()
+                if rules.match(data=data):
+                    return True
+            return False
         except:
             return False
 
     def proc_scan(self, p):
         try:
-            for entry in psutil.Process(p.pid).memory_maps():
+            for entry in p.memory_maps():
                 file = entry.path.replace("\\", "/")
                 if ":/Windows" not in file and ":/Program" not in file:
-                    if self.api_scan(file) or self.pe_scan(file):
+                    if self.start_scan(file):
                         return True
             return False
         except:
@@ -1002,8 +1045,7 @@ class MainWindow_Controller(QMainWindow):
                 for i in Permission:
                     try:
                         win32api.RegDeleteValue(key,i)
-                        file = self.proc.exe().replace("\\", "/")
-                        self.kill_process(self.proc, "惡意行為攔截", file)
+                        self.kill_process(self.proc, "惡意行為攔截")
                     except:
                         pass
                 win32api.RegCloseKey(key)
@@ -1227,68 +1269,88 @@ class MainWindow_Controller(QMainWindow):
             Thread(target=self.protect_net_thread, daemon=True).start()
 
     def protect_proc_thread(self):
-        existing_processes = set()
+        self.proc = None
+        existing_process = set()
         for p in psutil.process_iter():
-            existing_processes.add(p.pid)
-        psutil.Process(os.getpid()).nice(256)
+            existing_process.add(p.pid)
         while self.proc_protect:
             time.sleep(0.01)
+            new_process = set()
             for p in psutil.process_iter():
+                new_process.add(p.pid)
+            for pid in new_process - existing_process:
                 try:
-                    if psutil.pid_exists(p.pid) and p.pid not in existing_processes:
-                        name, file, cmd = p.name(), p.exe().replace("\\", "/"), p.cmdline()
-                        if file != self.pyas and file not in self.whitelist:
-                            self.lock_process(p, True)
-                            if "powershell" in name and self.api_scan(cmd[-1].split("'")[-2]):
-                                file = cmd[-1].split("'")[-2].replace("\\", "/")
-                                self.kill_process(p, "惡意腳本攔截", file)
-                            elif "cmd.exe" in name and self.api_scan(" ".join(cmd[2:])):
-                                file = " ".join(cmd[2:]).replace("\\", "/")
-                                self.kill_process(p, "惡意腳本攔截", file)
-                            elif ":/Windows" in file and self.api_scan(cmd[-1]):
-                                file = cmd[-1].replace("\\", "/")
-                                self.kill_process(p, "惡意軟體攔截", file)
-                            elif ":/Windows" not in file and self.proc_scan(p):
-                                self.kill_process(p, "惡意軟體攔截", file)
-                            elif ":/Windows" not in file and self.sign_scan(file):
-                                self.proc = p
-                            self.lock_process(p, False)
-                            existing_processes.add(p.pid)
+                    p = psutil.Process(pid)
+                    self.handle_new_process(p)
                 except:
-                    self.lock_process(p, False)
+                    pass
+            existing_process = new_process
+
+    def handle_new_process(self, p):
+        try:
+            name, file, cmd = p.name(), p.exe().replace("\\", "/"), p.cmdline()
+            if file != self.pyas and file not in self.whitelist:
+                self.lock_process(p, True)
+                if "powershell" in name and self.start_scan(cmd[-1].split("'")[-2]):
+                    self.kill_process(p, "惡意腳本攔截")
+                elif "cmd.exe" in name and self.start_scan(" ".join(cmd[2:])):
+                    self.kill_process(p, "惡意腳本攔截")
+                elif ":/Windows" in file and self.start_scan(cmd[-1]):
+                    self.kill_process(p, "惡意軟體攔截")
+                elif ":/Windows" not in file and self.proc_scan(p):
+                    self.kill_process(p, "惡意軟體攔截")
+                elif ":/Windows" not in file and self.sign_scan(file):
+                    self.proc = p
+                self.lock_process(p, False)
+                gc.collect()
+        except:
+            self.lock_process(p, False)
 
     def lock_process(self, p, lock):
         try:
             if lock:
-                psutil.Process(p.pid).suspend()
+                p.suspend()
             else:
-                psutil.Process(p.pid).resume()
+                p.resume()
         except:
-            return False
+            pass
 
-    def kill_process(self, parent, info, file):
+    def kill_process(self, p, info):
         try:
-            parent.kill()
-            self.send_notify(self.trans(f"{info}: ")+file)
-            for child in parent.children(recursive=True):
-                child.kill()
+            if p.is_running():
+                file = p.exe().replace("\\", "/")
+                self.send_notify(self.trans(f"{info}: ")+file)
+                for child in p.children(recursive=True):
+                    child.kill()
+                p.kill()
+            self.proc = None
         except:
             pass
 
     def protect_file_thread(self):
         hDir = win32file.CreateFile("C:/",win32con.GENERIC_READ,win32con.FILE_SHARE_READ|win32con.FILE_SHARE_WRITE|win32con.FILE_SHARE_DELETE,None,win32con.OPEN_EXISTING,win32con.FILE_FLAG_BACKUP_SEMANTICS,None)
+        self.ransom_counts = 0
         while self.file_protect:
             for action, file in win32file.ReadDirectoryChangesW(hDir,10485760,True,win32con.FILE_NOTIFY_CHANGE_FILE_NAME|win32con.FILE_NOTIFY_CHANGE_DIR_NAME|win32con.FILE_NOTIFY_CHANGE_ATTRIBUTES|win32con.FILE_NOTIFY_CHANGE_SIZE|win32con.FILE_NOTIFY_CHANGE_LAST_WRITE|win32con.FILE_NOTIFY_CHANGE_SECURITY,None,None):
                 try:
                     fpath = str(f"C:/{file}").replace("\\", "/")
                     ftype = str(f".{fpath.split('.')[-1]}").lower()
-                    if action == 2 and ":/Users" in fpath and ftype in alist:
-                        file = self.proc.exe().replace("\\", "/")
-                        if "/AppData/" not in file and ":/Program" not in file:
-                            self.kill_process(self.proc, "勒索行為攔截", file)
-                    elif action == 3 and ":/Users" in fpath and "/AppData/" not in fpath:
-                        if os.path.getsize(fpath) <= 52428800 and ftype in slist:
-                            if self.api_scan(fpath) or self.pe_scan(fpath):
+                    if self.ransom_counts >= 3 and self.proc:
+                        self.ransom_counts = 0
+                        self.kill_process(self.proc, "勒索行為攔截")
+                    elif ":/Windows" in fpath and "/Temp/" not in fpath:
+                        if action == 2 and ftype in alist:
+                            self.ransom_counts += 1
+                        elif action == 4 and ftype in alist:
+                            self.ransom_counts += 1
+                    elif ":/Users" in fpath and "/AppData/" not in fpath:
+                        if action == 2 and ftype in alist:
+                            self.ransom_counts += 1
+                        elif action == 4 and ftype in alist:
+                            self.ransom_counts += 1
+                        elif action == 3 and ftype in slist and os.path.getsize(fpath) <= 52428800:
+                            self.send_notify(self.trans(f"正在掃描: ")+fpath)
+                            if self.sign_scan(file) and self.start_scan(fpath):
                                 os.remove(fpath)
                                 self.send_notify(self.trans("惡意軟體刪除: ")+fpath)
                 except:
@@ -1300,15 +1362,13 @@ class MainWindow_Controller(QMainWindow):
                 time.sleep(0.2)
                 with open(r"\\.\PhysicalDrive0", "r+b") as f:
                     if self.mbr_value[510:512] != b'\x55\xAA':
-                        file = self.proc.exe().replace("\\", "/")
-                        self.kill_process(self.proc, "惡意行為攔截", file)
+                        self.kill_process(self.proc, "惡意行為攔截")
                     elif f.read(512) != self.mbr_value:
                         f.seek(0)
                         f.write(self.mbr_value)
-                        file = self.proc.exe().replace("\\", "/")
-                        self.kill_process(self.proc, "惡意行為攔截", file)
+                        self.kill_process(self.proc, "惡意行為攔截")
             except:
-                pass
+                self.kill_process(self.proc, "惡意行為攔截")
 
     def protect_reg_thread(self):
         while self.reg_protect:
@@ -1327,10 +1387,9 @@ class MainWindow_Controller(QMainWindow):
                 time.sleep(0.2)
                 local = socket.gethostbyname(socket.gethostname())
                 for conn in self.proc.connections():
-                    file = self.proc.exe().replace("\\", "/")
-                    if "/AppData/" not in file and ":/Program" not in file:
+                    if ":/Windows" not in file and ":/Program" not in file:
                         if conn.laddr.ip == local:
-                            self.kill_process(self.proc, "網路通訊攔截", file)
+                            self.kill_process(self.proc, "網路通訊攔截")
             except:
                 pass
 

PYAS_Engine.py

@@ -35,7 +35,7 @@ class ListSimHash:
         features = dict(Counter(sorted(content)))
         for f, w in features.items():
             count += w
-            batch.append(hashlib.md5(json.dumps(f).encode('utf-8')).digest() * w)
+            batch.append(hashlib.sha256(json.dumps(f).encode('utf-8')).digest() * w)
             if len(batch) >= 10000:
                 sums.append(self.sum_hashes(batch))
                 batch = []
@@ -47,7 +47,7 @@ class ListSimHash:
 
     def sum_hashes(self, digests):
         bitarray = numpy.unpackbits(numpy.frombuffer(b''.join(digests), dtype='>B'))
-        return numpy.sum(numpy.reshape(bitarray, (-1, 128)), 0)
+        return numpy.sum(numpy.reshape(bitarray, (-1, 256)), 0)
 
     def progress_bar(self, iteration, total, prefix='', suffix='', length=50, fill='█'):
         percent = min(100.0, max(0.0, 100 * (iteration / float(total))))
@@ -81,4 +81,4 @@ class ListSimHash:
 
     def similar(self, x, y):
         hamming_distance = bin(x ^ y).count('1')
-        return 1 - hamming_distance / 128
+        return 1 - hamming_distance / 256

PYAS_Language.py

@@ -4,7 +4,7 @@ translate_dict = {
               "軟體":"软件","檔案":"文件","確定":"确认","設定":"设置","關於":"关于",
               "錯誤":"错误","實時":"实时","發現":"发现","關閉":"关闭","裝置":"设备",
               "總數":"总数","註冊":"注册","圖標":"图标","路徑":"路径","網路":"网络",
-              "位元":"字节","彈窗":"弹窗","靈敏":"敏感","誤報":"误报","複製":"复制",
+              "位元":"字节","彈窗":"弹窗","靈敏":"敏感","複製":"复制","誤":"误","報":"报",
               "訊":"讯","項":"项","掃":"扫","試":"试","黃":"黄","嗎":"吗","防":"保",
               "鑰":"钥","當":"当","後":"后","護":"护","導":"导","區":"区","復":"复",
               "攔":"拦","請":"请","統":"统","刪":"删","進":"进","個":"个","選":"选",
@@ -47,8 +47,9 @@ translate_dict = {
               "增加到":"Add File To ","白名單":"White List ","軟體":"Software ",
               "增加":"Add","移除":"Delete","項目":"Items ","生成":"Create ","稱":"",
               "複製":"Copy ","開啟":"Open ","此":"This ","並":"And ","全盤":"Full ",
-              "已經是":"Already ","數據庫":"Database ","誤報":"False Positive ",
+              "已經在":"Already ","數據庫":"Database ","誤報":"False Positive ",
               "高":"High ","新":"New ","和":"And ","可能會":"Possible ","會":"Can ",
               "日誌":"Log","已":"Has ","彈窗":"Window ","要":"To ","驅動":"Drive ",
               "的":"","顯示":"","個":"","嗎":"","中":"","了":"","行":"","立即":"",
-              "密鑰":"Key ","檢測":"Detect ","製作":"Make","  ":" "}}
+              "密鑰":"Key ","檢測":"Detect ","製作":"Make","運行":"Running",
+              "自動":"Auto ","上報":"Report ","分析":"Analysis ","  ":" "}}

PYAS_Model.json

@@ -1 +1 @@
-{"Virus": [261318414186602332274638176436767278092, 138027491891768042595443233665640385515, 110357087572968576943094470150101987119, 9625936089192481713091234633207548145, 89628141614819275279501592421270658611, 146573019333239184433892013297795962994, 264541877314875483358485439014616010241, 270110569967616684464025230399867156597, 317767290038007567692141082439138927345, 89378687258170412207570345845754816506, 295491687946453794463476073658069199320, 95651985866123533435079044203969354868, 127221069722257753511418504633757572348, 95651985871075295954016225919772204788, 95067756796967225976055101492886444145, 105638824952820428052005376209508641704, 18885788501897915917239730749935859793, 5340774248576581325353819949912031489, 10415246439899713411029597770193411184, 115920025419090664908814181482465136245, 230418814927598623996991522865620895293, 264555963058906459534362468816145940272, 116667039230079911205687280113439840357, 188173655331549916271106901837471791624, 10580035239361434345353152238353587444, 94568728460636955139799806421985758457, 799868415884320332536138129537778698, 307822865305226887780395456446154510073, 894545638663197114170691953086909252, 307408049865722049855100904733744729280, 281395069059347277039468499503754758797, 292376319578571219959135358926367053520, 22744055344958721368630687581251847176, 315759149185061187179391478424305256898, 255349869106614732679810651086278546442, 95562399119788972561845288789790891122, 145716650028545025053236635988099140033, 95315785306000038407304639293114375376, 133237106118017744142061791759548170883, 10159525814797641639514752130410024305, 273929518968914349238520945266572101465, 6803412338476848613694005563166320666, 94821380653995263056920242811808354032, 110684881900941951709859261701564766961, 105787286451833022929870383496648011249, 201783522381387157888199381457569830339, 105453663146661214242864956330772860016, 171522714730243767599747450471850906780, 106285815540718769887458629487913147505, 95232708608642156662699453866682554993, 106285815540717561551934825217444224113, 95559052534575041823542189821460810608, 173324173206867137081876750353781475864, 106280623246394143467097978838728292593, 305010456281757544171469123445447117681, 222253752239577871278657531622986138289, 95563048166632561463187231032275512432, 95559057605177442847140260259121040241, 108010814655804390841927389574188560953, 95318319972117721345303221798918291186, 94820584548412181705953186602301163257, 106198844586809641198104478887777078384, 94818541689698630943530703193593974512, 106119662154515582210221674170892663408, 143088668467296939342778818474878950647, 20881613063668797611160339818896853168, 179710050855676845982141937253360453053, 31018193447326218505924275935240096368, 105455042350511842344155948214062747248, 89670460543521034706908097860639104624, 105949568716338471638768471668461534962, 137521268273704145345759177480636558327, 107244620726926889373575057027717450762, 106201415362574605111426884001172884597, 110684892675972091428741019788518218489, 10332169700047182535948410013892054257, 105576376886969565715901422869178687219, 95145598865408927095247444731707367024, 105039010693725170460743583068113023705, 94987377577609631169063907051139341428, 100629312664727978741143555078594927795, 110684882592911957008875737340312369917, 148748393343889971317559945312215468786, 93989806751692355346109462915386284256, 100878647073583869926186814218125905113, 94662072863013290725024950240040430327, 105949568706125465724368274041836011250, 114380135952416886874949645426965728392, 94987377598500779420569757831453449840, 20215706740235798874506125693183790832, 110684912958381676190945511157408029425, 300558662558418689647121992472167150144, 132011016412170201801452237305311191441, 109023346904934388839803760995464461049, 336983008266790509743060629130917677774, 143175020845977086618208175741834533027, 172003113737697601641723035082316482624, 105787367661937540200783035652543688177, 95319034907171535697107975830043465952, 127216527016950104546694906326424083058, 124017012583193473160915936873388365506, 95065159985079431125978131216282192112, 10580035240599374384629243014119103732, 10496943991486934504031035936730090609, 31018193447326218505924275935240096368, 10248965448213412269216028810651597936, 100963036754215599502702416838888975610, 79036768232931406417351794113275712068, 105532141803327881065557146349585635569, 162357332045345094604141502928937491106, 116582703050825520068375122378543716978, 95150935027941497330695649661434404081, 176177498613631404614794194516575496896, 95484514342568128463910419849540603504, 105867687984072457600475937752904732400, 66833476002653370931812317778542555093, 9915421242395013439142248483403569396, 136271405179532594000996159423307548914, 47343025305025541868952332865095320248, 169502105399457292840800686344751518387, 274028058373054263663577471026209984675, 95650044561417669529155573147468110960, 100968228555433537808235270915390512872, 26Line truncated
+{"Unknown": [38413146381396798812836662586219087594480760119241756145287379993031670453973, 96759775057307603829199050115151442660431386536877578909394799687553144825391], "White": [92860071170291586757870565976957954052649178883790647665503983139114277816413, 50625869624179221836524636484722091669820057030729734765256925026245308185143, 79303266820014134748061006437377483029320717142203104125806969476209010566146, 50391760446499089493384482654016917630509060287507523223191455177998253781572, 34733894651460458125941968926483828168477919119649839915356573816494375383659, 16447578251589668808184123721615810667991570306415913809506896532563163557975, 18668853151306203943053249777678854424409647574074706653144503993068509738028, 16447129799599854768187525514242197965987640167415805965954293039200573272151, 42965648781535883376480003940402366467478625051031283341289116138472520399484, 45637511530246203537935382208327538642024163565740899790237852717178844340385, 100662025713257198141308144602569453810226448934776598411982222249289646045820, 2813878798545665948450767843887867208740292941021073849284371582085785669685, 54695748401254225125045716155327877471149349614359435230254611073117412307518, 27428665293760567560050957450577436769403462770366418052177943137905489504373, 32018095881556959863902669246154132691484124511748151111295836576906027842532, 82735669648112373508385530969898006750362429118827185652448450648454052106395, 27853809242729492000984914092533936796891917926358854729439436487683423285261, 20030429995379197455160031864140436552884002299333955332088108013764717533796, 82338140557728967648601024370995428985866052631101882076231147335348455671038, 71339449324799277130427957011381342555145697382104653874846579353871764366968, 2432232479351500614611699169184443988440809841079862157291791071419485606436, 9463527399566410848087352636323112952591923803840644609432039785310381355253, 17287926889345790925003722717807755244540044472850272987265624141065166747251, 31507509644511868166890345905396462199402122513677304280568315890707028374136, 92802443753971611323223428952970105040389824130209263287070763891097867755624, 45133408084728538869955951560598604064861024543940290379768365214821771604584, 43161770374046173333870752071317153674965637163816017117315179871715332666383, 55200886519442939000007636174371797768786544863806926514781108850256791225983, 102124766481586899841647737641698083212305154684644839980199278690335383712492, 52992286898317605666180786036367620316269750824654214415314225077986227890804, 96307454545313440124283308270263825589093792315386382296096934270155352236733, 36073459579994433898643313577597321150634749356791285363958792075031385371364, 2671602537216404752611755689384448521170725107261295121211275163689399347529, 31174448189678483951801405655647041557287598300853213718404097030076935043252, 6997953573828232905847825866256444005094471443450029313008738013465446666864, 41288968033780608769032189481784193916730262129846893330891418484950645535282, 56724589038911745751056508177185337741291265472470897077742981563488398484126, 3353938097938067320481103667469354721822387389147415219233498938990215839034, 21195721951654457114265478558602466514207347021822959737607658828495062116920, 2916561284514958799877078174216081002558870786437593202464334407456915288119, 93137488384582768487662730765406914258099784103320160609442537879778117334700, 21195721951654457114265478558602377311225888285332501136798116331518898022972, 85036278951571446667616241901438186945472743486751871120301019617002115253240, 14623491695050192413097067629676318500123044497134202517578785087180871258165, 2022308710750553104057453378681448926020203033444470910346149752954363011822, 16435447118324352510213480738895714506136417806014749563314727656196027207884, 20110908101408861236178371339075650317185505287106087252378028881915405550816, 97262306467112548995108413358594977802790545321935863056096364390382492643957, 96614497885586867524036482497010189662872359368622249290846422070379022959337, 1895288773648887609359329471072769468085207279473521707976899919066265183340, 2651591106060448314208209841614702280337184080195391257416359882876000875095, 114619913163036058498985801603082782363327281628924544162718017899237779173997, 2481883953883130814217575190747389944608730963897863224598923997640828351014, 45000300637061336566799421885334051647634671537987073940268834014403441552901, 79047998258892679051555952202178856054614923426747250310189496005785954516966, 92887093185298651693450115442544102614083847612212241647641096093293220418098, 7140769773119319253009453104401896171570019200295580355390762162793695491644, 17012496707164921800767104952565848716212381967476981614870260538046339441223, 55577404054451500506896476492698809966352434385747804245366700591039970820656, 42705951800174486650758821275676750976088492908259259071295096144374682903086, 49945662818205539045803005800855628211260206264132745568273429553565001939621, 1971375Line truncated

PYAS_Version.py

@@ -2,7 +2,7 @@ import pyinstaller_versionfile
 
 pyinstaller_versionfile.create_versionfile(
 output_file="versionfile.txt",
-version='3.0.6',
+version='3.0.9',
 company_name="PYAS Security",
 file_description="Python Antivirus Software",
 internal_name="PYAS",

README.md

@@ -37,35 +37,6 @@ pywin32==306
 PyQt5==5.15.9
 ```
 
-## Hash Scanning
-
-Use Qihoo 360 cloud service to scan known malicious files
-
-```
-import hashlib, requests
-import xml.etree.ElementTree as xmlet
-
-def hash_scan(file):
-    try:
-        with open(file, "rb") as f:
-            text = str(hashlib.md5(f.read()).hexdigest())
-        strBody = f'-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="md5s"\r\n\r\n{text}\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="format"\r\n\r\nXML\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="product"\r\n\r\n360zip\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="combo"\r\n\r\n360zip_main\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="v"\r\n\r\n2\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="osver"\r\n\r\n5.1\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="vk"\r\n\r\na03bc211\r\n-------------------------------7d83e2d7a141e\r\nContent-Disposition: form-data; name="mid"\r\n\r\n8a40d9eff408a78fe9ec10a0e7e60f62\r\n-------------------------------7d83e2d7a141e--'
-        response = requests.post('http://qup.f.360.cn/file_health_info.php', data=strBody, timeout=3)
-        if response.status_code == 200:
-            level = float(xmlet.fromstring(response.text).find('.//e_level').text)
-            if level > 50:
-                return "Virus"
-            elif level <= 10:
-                return "Safe"
-            return "Unknown"
-        return f"Error: {response.status_code}"
-    except Exception as e:
-        return f"Error: {e}"
-
-info = hash_scan("path to file")
-print(info)
-```
-
 ## Pefile Scanning
 
 Get the pefile file function import table for scanning