Delete PYAS_Function.7z

PYAS.py

@@ -3,11 +3,14 @@ import hashlib, pefile, socket, msvcrt
 import requests, pyperclip, win32file
 import win32gui, win32api, win32con
 import xml.etree.ElementTree as xmlet
-from PYAS_Rules import pyasrule_dict
-from PYAS_Function import function_list
+from PYAS_Database import model_dict
+from PYAS_Function_Virus import func_virus
+from PYAS_Function_Safe import func_safe
 from PYAS_Extension import slist, alist
 from PYAS_Language import translate_dict
 from PYAS_Interface import Ui_MainWindow
+from PYML import ListClassifier
+from binascii import hexlify
 from threading import Thread
 from PyQt5.QtWidgets import *
 from PyQt5.QtCore import *
@@ -19,10 +22,11 @@ class MainWindow_Controller(QMainWindow):
         self.setAttribute(Qt.WA_TranslucentBackground)
         self.setWindowFlags(Qt.FramelessWindowHint)
         self.pyas = str(sys.argv[0]).replace("\\", "/")
-        self.pyas_version = "2.9.2"
+        self.pyas_version = "2.9.7"
         self.ui = Ui_MainWindow()
         self.ui.setupUi(self)
         self.init_tray_icon()
+        self.init_data_base()
         self.init_virus_list()
         self.init_state_safe()
         self.init_config_path()
@@ -36,6 +40,10 @@ class MainWindow_Controller(QMainWindow):
         self.show_pyas_ui()
         self.init_threads()
 
+    def init_data_base(self):
+        self.ml = ListClassifier()
+        self.ml.load_model(model_dict)
+
     def init_threads(self):
         self.protect_proc_init()
         self.protect_file_init()
@@ -269,7 +277,7 @@ class MainWindow_Controller(QMainWindow):
         self.ui.Window_Block_Button.setText(self.trans("軟體彈窗攔截"))
         self.ui.Repair_System_Network_Button.setText(self.trans("系統網路修復"))
         self.ui.About_Back.setText(self.trans("返回"))
-        self.ui.PYAS_Version.setText(self.trans(f"PYAS V{self.pyas_version} (ML Engine)"))
+        self.ui.PYAS_Version.setText(self.trans(f"PYAS V{self.pyas_version} (ML Engine Beta)"))
         self.ui.GUI_Made_title.setText(self.trans("介面製作:"))
         self.ui.GUI_Made_Name.setText(self.trans("mtkiao"))
         self.ui.Core_Made_title.setText(self.trans("核心製作:"))
@@ -305,21 +313,21 @@ class MainWindow_Controller(QMainWindow):
         if self.ui.Theme_White.isChecked():
             self.ui.Window_widget.setStyleSheet("""QWidget#Window_widget {background-color:rgb(240,240,240);}""")
             self.ui.Navigation_Bar.setStyleSheet("""QWidget#Navigation_Bar {background-color:rgb(230,230,230);}""")
-        elif self.ui.Theme_Red.isChecked():
-            self.ui.Window_widget.setStyleSheet("""QWidget#Window_widget {background-color:rgb(255,150,150);}""")
-            self.ui.Navigation_Bar.setStyleSheet("""QWidget#Navigation_Bar {background-color:rgb(255,140,140);}""")
         elif self.ui.Theme_Black.isChecked():
-            self.ui.Window_widget.setStyleSheet("""QWidget#Window_widget {background-color:rgb(150,150,150);}""")
-            self.ui.Navigation_Bar.setStyleSheet("""QWidget#Navigation_Bar {background-color:rgb(140,140,140);}""")
+            self.ui.Window_widget.setStyleSheet("""QWidget#Window_widget {background-color:rgb(230,230,230);}""")
+            self.ui.Navigation_Bar.setStyleSheet("""QWidget#Navigation_Bar {background-color:rgb(220,220,220);}""")
+        elif self.ui.Theme_Red.isChecked():
+            self.ui.Window_widget.setStyleSheet("""QWidget#Window_widget {background-color:rgb(250,230,230);}""")
+            self.ui.Navigation_Bar.setStyleSheet("""QWidget#Navigation_Bar {background-color:rgb(250,220,220);}""")
         elif self.ui.Theme_Green.isChecked():
-            self.ui.Window_widget.setStyleSheet("""QWidget#Window_widget {background-color:rgb(150,255,150);}""")
-            self.ui.Navigation_Bar.setStyleSheet("""QWidget#Navigation_Bar {background-color:rgb(130,255,130);}""")
+            self.ui.Window_widget.setStyleSheet("""QWidget#Window_widget {background-color:rgb(230,250,230);}""")
+            self.ui.Navigation_Bar.setStyleSheet("""QWidget#Navigation_Bar {background-color:rgb(220,250,220);}""")
         elif self.ui.Theme_Yellow.isChecked():
-            self.ui.Window_widget.setStyleSheet("""QWidget#Window_widget {background-color:rgb(255,255,150);}""")
-            self.ui.Navigation_Bar.setStyleSheet("""QWidget#Navigation_Bar {background-color:rgb(240,240,140);}""")
+            self.ui.Window_widget.setStyleSheet("""QWidget#Window_widget {background-color:rgb(250,250,230);}""")
+            self.ui.Navigation_Bar.setStyleSheet("""QWidget#Navigation_Bar {background-color:rgb(250,250,220);}""")
         elif self.ui.Theme_Blue.isChecked():
-            self.ui.Window_widget.setStyleSheet("""QWidget#Window_widget {background-color:rgb(0,200,255);}""")
-            self.ui.Navigation_Bar.setStyleSheet("""QWidget#Navigation_Bar {background-color:rgb(0,190,255);}""")
+            self.ui.Window_widget.setStyleSheet("""QWidget#Window_widget {background-color:rgb(230,250,250);}""")
+            self.ui.Navigation_Bar.setStyleSheet("""QWidget#Navigation_Bar {background-color:rgb(220,250,250);}""")
 
     def change_animation(self,widget):
         x, y = 170, widget.pos().y()
@@ -532,8 +540,9 @@ class MainWindow_Controller(QMainWindow):
         self.timer.start(2)
 
     def showMinimized(self):
-        self.hide_pyas_ui()
-        self.send_notify(self.trans("PYAS 已最小化到系統托盤圖標"))
+        if self.block_window:
+            self.hide_pyas_ui()
+            self.send_notify(self.trans("PYAS 已最小化到系統托盤圖標"))
 
     def closeEvent(self, event):
         if QMessageBox.warning(self,self.trans("警告"),self.trans("您確定要退出 PYAS 和所有防護嗎?"),QMessageBox.Yes|QMessageBox.No) == 16384:
@@ -777,14 +786,18 @@ class MainWindow_Controller(QMainWindow):
             file_type = str(f".{file.split('.')[-1]}").lower()
             if self.high_sensitivity and file != self.pyas:
                 if self.api_scan(file):
-                    self.write_scan(self.trans("惡意"),file)
+                    self.write_scan(self.trans("Vir.cloud.md5"),file)
+                elif self.ml_scan(file):
+                    self.write_scan(self.trans("Sus.hdfile.ml"),file)
                 elif self.pe_scan(file):
-                    self.write_scan(self.trans("可疑"),file)
-            elif file_type in slist and self.sign_scan(file):
+                    self.write_scan(self.trans("Sus.pefile.he"),file)
+            elif file_type in slist and file != self.pyas:
                 if self.api_scan(file):
-                    self.write_scan(self.trans("惡意"),file)
+                    self.write_scan(self.trans("Vir.cloud.md5"),file)
+                elif self.ml_scan(file):
+                    self.write_scan(self.trans("Sus.hdfile.ml"),file)
                 elif self.pe_scan(file):
-                    self.write_scan(self.trans("可疑"),file)
+                    self.write_scan(self.trans("Sus.pefile.he"),file)
         except:
             pass
 
@@ -836,12 +849,46 @@ class MainWindow_Controller(QMainWindow):
                         except:
                             pass
             max_vfl = []
-            for vfl in function_list:
+            for vfl in func_virus:
                 QApplication.processEvents()
                 max_vfl.append(len(set(fn)&set(vfl))/len(set(fn)|set(vfl)))
-            if self.high_sensitivity and max(max_vfl) == 1.0:
-                return True
-            elif "_CorExeMain" not in fn and max(max_vfl) == 1.0:
+            max_sfl = []
+            for sfl in func_safe:
+                QApplication.processEvents()
+                max_sfl.append(len(set(fn)&set(sfl))/len(set(fn)|set(sfl)))
+            if self.high_sensitivity:
+                return max(max_vfl) >= max(max_sfl)
+            elif "_CorExeMain" not in fn:
+                return max(max_vfl) >= max(max_sfl)
+            return False
+        except:
+            return False
+
+    def ml_scan(self, file):
+        try:
+            data = []
+            if os.path.getsize(file) <= 209715200:
+                with open(file, 'rb') as f:
+                    for block in iter(lambda: f.read(3),b''):
+                        QApplication.processEvents()
+                        data.append(str(hexlify(block),'utf-8'))
+            if self.high_sensitivity:
+                return self.ml.predict(data) == 1
+            elif self.sign_scan(file):
+                return self.ml.predict(data) == 1
+            return False
+        except:
+            return False
+
+    def scan_proc(self, value):
+        try:
+            if isinstance(value, int):
+                for entry in psutil.Process(value).memory_maps():
+                    file = entry.path.replace("\\", "/")
+                    if ":/Windows" not in file and ":/Program" not in file and "/AppData/" not in file:
+                        if self.api_scan(file) or self.pe_scan(file) or self.ml_scan(file):
+                            return True
+            elif self.api_scan(value) or self.pe_scan(value) or self.ml_scan(value):
                 return True
             return False
         except:
@@ -948,6 +995,8 @@ class MainWindow_Controller(QMainWindow):
                 for i in Permission:
                     try:
                         win32api.RegDeleteValue(key,i)
+                        self.proc.kill()
+                        self.send_notify(self.trans("惡意行為攔截: ")+self.proc.name().replace("\\", "/"))
                     except:
                         pass
                 win32api.RegCloseKey(key)
@@ -1175,48 +1224,48 @@ class MainWindow_Controller(QMainWindow):
             time.sleep(0.05)
             for p in psutil.process_iter():
                 try:
-                    if p.pid not in existing_processes:
+                    name, file, cmd = p.name(), p.exe().replace("\\", "/"), p.cmdline()
+                    if p.pid not in existing_processes and file not in self.whitelist:
                         existing_processes.add(p.pid)
                         psutil.Process(p.pid).suspend()
-                        name, file, cmd = p.name(), p.exe().replace("\\", "/"), p.cmdline()
-                        if self.protect_proc_scan(name, file, cmd):
+                        if "powershell" in name and self.scan_proc(cmd[-1].split("'")[-2]):
+                            p.kill()
+                            self.send_notify(self.trans("惡意腳本攔截: ")+str(cmd[-1].split("'")[-2]))
+                        elif "cmd.exe" in name and self.scan_proc(" ".join(cmd[2:])):
+                            p.kill()
+                            self.send_notify(self.trans("惡意腳本攔截: ")+str(" ".join(cmd[2:])))
+                        elif ":/Program" not in file and self.scan_proc(cmd[-1]):
+                            p.kill()
+                            self.send_notify(self.trans("惡意軟體攔截: ")+str(cmd[-1]))
+                        elif ":/Windows" not in file and self.scan_proc(p.pid):
                             p.kill()
                             self.send_notify(self.trans("惡意軟體攔截: ")+file)
-                        elif ":/Windows" not in file and self.sign_scan(file):
+                        elif file != self.pyas and self.sign_scan(file):
                             self.proc = p
                         psutil.Process(p.pid).resume()
                         gc.collect()
                 except:
                     pass
 
-    def protect_proc_scan(self, name, file, cmd):
-        try:
-            if file != self.pyas and file not in self.whitelist:
-                if "powershell" in name:
-                    file = str(cmd[-1].split("'")[-2])
-                elif "cmd.exe" in name:
-                    file = str(" ".join(cmd[2:]))
-                elif ":/Windows" in file:
-                    file = str(cmd[-1])
-                elif ":/Program" in file:
-                    file = str(cmd[-1])
-                return self.api_scan(file) or self.pe_scan(file)
-            return False
-        except:
-            pass
-
     def protect_file_thread(self):
         hDir = win32file.CreateFile("C:/Users/",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_block = False
         while self.file_protect:
-            for action, file in win32file.ReadDirectoryChangesW(hDir,1024,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):
+            for action, file in win32file.ReadDirectoryChangesW(hDir,1048576,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:
-                    if action == 2 and str(f".{file.split('.')[-1]}").lower() in alist:
-                        if self.ransom_block and "AppData" not in file:
+                    full_path = str(f"C:/Users/{file}").replace("\\", "/")
+                    file_type = str(f".{full_path.split('.')[-1]}").lower()
+                    if action == 2 and "/AppData/" not in full_path and file_type in alist:
+                        if self.ransom_block:
                             self.proc.kill()
                             self.ransom_block = False
-                            self.send_notify(self.trans("勒索軟體攔截: ")+self.proc.name())
+                            self.send_notify(self.trans("勒索軟體攔截: ")+self.proc.exe().replace("\\", "/"))
                         self.ransom_block = True
+                    #elif action == 3 and "/AppData/" not in full_path and file_type in slist:
+                        #if os.path.getsize(full_path) <= 20971520:
+                            #if self.sign_scan(full_path) and self.api_scan(full_path):
+                                #self.lock_file(full_path)
+                                #self.send_notify(self.trans("惡意軟體攔截: ")+full_path)
                 except:
                     pass
 
@@ -1227,12 +1276,12 @@ class MainWindow_Controller(QMainWindow):
                 with open(r"\\.\PhysicalDrive0", "r+b") as f:
                     if self.mbr_value[510:512] != b'\x55\xAA':
                         self.proc.kill()
-                        self.send_notify(self.trans("惡意行為攔截: ")+self.proc.name())
+                        self.send_notify(self.trans("惡意行為攔截: ")+self.proc.exe().replace("\\", "/"))
                     elif f.read(512) != self.mbr_value:
                         f.seek(0)
                         f.write(self.mbr_value)
                         self.proc.kill()
-                        self.send_notify(self.trans("惡意行為攔截: ")+self.proc.name())
+                        self.send_notify(self.trans("惡意行為攔截: ")+self.proc.exe().replace("\\", "/"))
             except:
                 pass
 
@@ -1251,10 +1300,11 @@ class MainWindow_Controller(QMainWindow):
         while self.net_protect:
             try:
                 time.sleep(0.5)
+                local_host = socket.gethostbyname(socket.gethostname())
                 for conn in self.proc.connections():
-                    if socket.gethostbyname(socket.gethostname()) == conn.laddr.ip:
+                    if conn.laddr.ip == local_host:
                         self.proc.kill()
-                        self.send_notify(self.trans("網路通訊攔截: ")+self.proc.name())
+                        self.send_notify(self.trans("網路通訊攔截: ")+self.proc.exe().replace("\\", "/"))
             except:
                 pass
 

PYAS_Extension.py

@@ -1,6 +1,5 @@
 slist = [".exe",".dll",".com",".msi",".scr",
-         ".bat",".cmd",".ps1",".vbs",".jar",
-         ".js",".elf",".doc",".xls",".xlsx"]
+         ".js",".bat",".cmd",".ps1",".vbs"]
 
 alist = [".ppt",".pptx",".wps",".txt",".rtf",
          ".pdf",".xls",".xlsx",".doc",".docx",

PYAS_Language.py

@@ -1,53 +1,53 @@
-translate_dict = {"zh_TW": {"已开启":"已開啟","已关闭":"已關閉","On":"已開啟","Off":"已關閉"},
-                  "zh_CN": {"已開啟":"已开启","已關閉":"已关闭","On":"已开启","Off":"已关闭",
-                            "軟體":"软件","檔案":"文件","確定":"确认","設定":"设置","關於":"关于",
-                            "錯誤":"错误","實時":"实时","發現":"发现","關閉":"关闭","裝置":"设备",
-                            "總數":"总数","註冊":"注册","圖標":"图标","路徑":"路径","網路":"网络",
-                            "位元":"字节","彈窗":"弹窗","靈敏":"敏感","誤報":"误报","複製":"复制",
-                            "訊":"讯","項":"项","掃":"扫","試":"试","黃":"黄","嗎":"吗","防":"保",
-                            "鑰":"钥","當":"当","後":"后","護":"护","導":"导","區":"区","復":"复",
-                            "攔":"拦","請":"请","統":"统","刪":"删","進":"进","個":"个","選":"选",
-                            "啟":"启","監":"监","並":"并","徑":"径","盤":"盘","隱":"隐","動":"动",
-                            "狀":"状","誌":"志","這":"这","會":"会","題":"题","擇":"择","語":"语",
-                            "無":"无","顯":"显","變":"变","戶":"户","碼":"码","雲":"云","訂":"订",
-                            "稱":"称","類":"类","輸":"输","執":"执","數":"数","尋":"寻","網":"网",
-                            "詢":"询","製":"制","員":"员","許":"许","條":"条","藍":"蓝","紅":"红",
-                            "綠":"绿","態":"态","務":"务","連":"连","檢":"检","測":"测","關":"关",
-                            "強":"强","驅":"驱","為":"为","腳":"脚","增":"增","單":"单","開":"开",
-                            "惡":"恶",},
-                  "en_US": {"已開啟":"On","已關閉":"Off","已开启":"On","已关闭":"Off","，":", ",
-                            "立即刪除":"Delete Now","安全模式":"Safe Mode","官方網站":"Website",
-                            "惡意檔案":"Virus ","惡意軟體":"Malware ","勒索軟體":"Ransom ",
-                            "惡意腳本":"Script ","惡意行為":"Behavior ","可疑軟體":"Suspicious ",
-                            "許可條款":"License Terms","惡意":"Malicious","可疑":"Suspicious",
-                            "啟用":"Enable ","禁用":"Disable ","正在初始化中":"Initializing ",
-                            "正在掃描":"Scanning ","正在刪除":"Deleting ","最小化到":"Minimize To ",
-                            "確定要":"Sure Want To ","靈敏度":"Sensitivity ","更新":"Update",
-                            "白色主題":"White","黑色主題":"Black","黃色主題":"Yellow","紅色主題":"Red",
-                            "藍色主題":"Blue","綠色主題":"Green","語言":"Language ","製作":"Make",
-                            "測試人員":"Testers","版本資訊":"Information","通訊":"Communication ",
-                            "註冊表":"Registry ","退出":"Quit ","檔案":"File ","當前":"Currently ",
-                            "狀態":"State ","掃描":"Scan ","工具":"Tools ","防護":"Protect ",
-                            "失敗":"Failed ","請":"Please ","路徑":"Path ","方式":"Method ",
-                            "需要":"Need ","病毒":"Virus ","選擇":"Select ","返回":"Back",
-                            "受到":"Been ","錯誤":"Error","完成":"Complete ","引擎":"Engine ",
-                            "托盤":"Tray ","圖標":"Icon ","不安全":"Unsafe","增強":"Enhanced ",
-                            "系統":"System ","雲端":"Cloud ","重啟":"Reboot ","修復":"Repair ",
-                            "裝置":"Device ","發現":"Found ","成功":"Success ","清理":"Cleanup ",
-                            "確定":"OK","執行":"Execute ","警告":"Warning ","使用":"Use ",
-                            "網路":"Network ","變更":"Change ","總數":"Total","結束":"End ",
-                            "選項":"Options ","監控":"Monitor ","連接":"Connect ","可以":"To ",
-                            "進程":"Process ","攔截":"Block ","引導":"Boot ","刪除":"Delete ",
-                            "分區":"Partition ","設定":"Settings","關於":"About","模式":"Mode ",
-                            "提高":"Improve ","安全":"Security ","顯色":"Color ","主題":"Themes ",
-                            "風險":"Risk ","介面":"GUI ","核心":"Core ","服務":"Services ",
-                            "管理":"Manage","所有":"All ","全盤":"Full ","取消":"Cancel ",
-                            "垃圾":"Junk ","位元":"byte ","本機":"Computer ","停止":"Stop ",
-                            "增加到":"Add File To ","白名單":"White List ","軟體":"Software ",
-                            "增加":"Add","移除":"Delete","項目":"Items ","生成":"Create ","稱":"",
-                            "開啟":"Open ","您":"You ","此":"This ","並":"And ","未":"Not ",
-                            "高":"High ","新":"New ","和":"And ","可能會":"Possible ","會":"Can ",
-                            "日誌":"Log","已":"Has ","彈窗":"Window ","要":"To ","驅動":"Drive ",
-                            "的":"","顯示":"","個":"","嗎":"","中":"","了":"","行":"","立即":"",
-                            "密鑰":"Key ","檢測":"Detect ","誤報":"False Positive ","  ":" ",
-                            "複製":"Copy "}}
+translate_dict = {
+    "zh_TW": {"已开启":"已開啟","已关闭":"已關閉","On":"已開啟","Off":"已關閉"},
+    "zh_CN": {"已開啟":"已开启","已關閉":"已关闭","On":"已开启","Off":"已关闭",
+              "軟體":"软件","檔案":"文件","確定":"确认","設定":"设置","關於":"关于",
+              "錯誤":"错误","實時":"实时","發現":"发现","關閉":"关闭","裝置":"设备",
+              "總數":"总数","註冊":"注册","圖標":"图标","路徑":"路径","網路":"网络",
+              "位元":"字节","彈窗":"弹窗","靈敏":"敏感","誤報":"误报","複製":"复制",
+              "訊":"讯","項":"项","掃":"扫","試":"试","黃":"黄","嗎":"吗","防":"保",
+              "鑰":"钥","當":"当","後":"后","護":"护","導":"导","區":"区","復":"复",
+              "攔":"拦","請":"请","統":"统","刪":"删","進":"进","個":"个","選":"选",
+              "啟":"启","監":"监","並":"并","徑":"径","盤":"盘","隱":"隐","動":"动",
+              "狀":"状","誌":"志","這":"这","會":"会","題":"题","擇":"择","語":"语",
+              "無":"无","顯":"显","變":"变","戶":"户","碼":"码","雲":"云","訂":"订",
+              "稱":"称","類":"类","輸":"输","執":"执","數":"数","尋":"寻","網":"网",
+              "詢":"询","製":"制","員":"员","許":"许","條":"条","藍":"蓝","紅":"红",
+              "綠":"绿","態":"态","務":"务","連":"连","檢":"检","測":"测","關":"关",
+              "強":"强","驅":"驱","為":"为","腳":"脚","增":"增","單":"单","開":"开",
+              "惡":"恶","介":"界"},
+    "en_US": {"已開啟":"On","已關閉":"Off","已开启":"On","已关闭":"Off","，":", ",
+              "立即刪除":"Delete Now","安全模式":"Safe Mode","官方網站":"Website",
+              "惡意檔案":"Virus ","惡意軟體":"Malware ","勒索軟體":"Ransom ",
+              "惡意腳本":"Script ","惡意行為":"Behavior ","可疑軟體":"Suspicious ",
+              "許可條款":"License Terms","惡意":"Malicious","可疑":"Suspicious",
+              "啟用":"Enable ","禁用":"Disable ","正在初始化中":"Initializing ",
+              "正在掃描":"Scanning ","正在刪除":"Deleting ","最小化到":"Minimize To ",
+              "確定要":"Sure Want To ","靈敏度":"Sensitivity ","更新":"Update",
+              "白色主題":"White","黑色主題":"Black","黃色主題":"Yellow","紅色主題":"Red",
+              "藍色主題":"Blue","綠色主題":"Green","語言":"Language ","製作":"Make",
+              "測試人員":"Testers","版本資訊":"Information","通訊":"Communication ",
+              "註冊表":"Registry ","退出":"Quit ","檔案":"File ","當前":"Currently ",
+              "狀態":"State ","掃描":"Scan ","工具":"Tools ","防護":"Protect ",
+              "失敗":"Failed ","請":"Please ","路徑":"Path ","方式":"Method ",
+              "需要":"Need ","病毒":"Virus ","選擇":"Select ","返回":"Back",
+              "受到":"Been ","錯誤":"Error","完成":"Complete ","引擎":"Engine ",
+              "托盤":"Tray ","圖標":"Icon ","不安全":"Unsafe","增強":"Enhanced ",
+              "系統":"System ","雲端":"Cloud ","重啟":"Reboot ","修復":"Repair ",
+              "裝置":"Device ","發現":"Found ","成功":"Success ","清理":"Cleanup ",
+              "確定":"OK","執行":"Execute ","警告":"Warning ","使用":"Use ",
+              "網路":"Network ","變更":"Change ","總數":"Total","結束":"End ",
+              "選項":"Options ","監控":"Monitor ","連接":"Connect ","可以":"To ",
+              "進程":"Process ","攔截":"Block ","引導":"Boot ","刪除":"Delete ",
+              "分區":"Partition ","設定":"Settings","關於":"About","模式":"Mode ",
+              "提高":"Improve ","安全":"Security ","顯色":"Color ","主題":"Themes ",
+              "風險":"Risk ","介面":"GUI ","核心":"Core ","服務":"Services ",
+              "管理":"Manage","所有":"All ","您":"You ","取消":"Cancel ","未":"Not ",
+              "垃圾":"Junk ","位元":"byte ","本機":"Computer ","停止":"Stop ",
+              "增加到":"Add File To ","白名單":"White List ","軟體":"Software ",
+              "增加":"Add","移除":"Delete","項目":"Items ","生成":"Create ","稱":"",
+              "複製":"Copy ","開啟":"Open ","此":"This ","並":"And ","全盤":"Full ",
+              "高":"High ","新":"New ","和":"And ","可能會":"Possible ","會":"Can ",
+              "日誌":"Log","已":"Has ","彈窗":"Window ","要":"To ","驅動":"Drive ",
+              "的":"","顯示":"","個":"","嗎":"","中":"","了":"","行":"","立即":"",
+              "密鑰":"Key ","檢測":"Detect ","誤報":"False Positive ","  ":" "}}

PYAS_Version.py

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

PYML.py

@@ -0,0 +1,166 @@
+from collections import defaultdict
+import math, json, time, types, random
+
+class ListClassifier:
+    def __init__(self):
+        self.word_weights = defaultdict(lambda: {'ones': 0, 'zero': 0})
+        self.total_counts = {'ones': 0, 'zero': 0}
+        self.prior_probs = {'ones': 0, 'zero': 0}
+        self.class_weights = {'ones': 1, 'zero': 1}
+        self.min_quantity = 0
+        self.max_quantity = 1.0
+        self.learning_rate = 0.1
+        self.max_sequence = 100
+        self.test_data = []
+        self.train_data = []
+
+    def load_model(self, model_data):
+        if isinstance(model_data, str):
+            with open(model_data, 'r') as f:
+                model_data = json.load(f)
+        self.word_weights = defaultdict(lambda: {'ones': 0, 'zero': 0})
+        self.word_weights.update(model_data['word_weights'])
+        self.total_counts = model_data['total_counts']
+        self.prior_probs = model_data['prior_probs']
+        self.max_sequence = model_data.get('max_sequence')
+
+    def save_model(self, filename):
+        model_data = {
+            'word_weights': dict(self.word_weights),
+            'total_counts': self.total_counts,
+            'prior_probs': self.prior_probs,
+            'max_sequence': self.max_sequence}
+        with open(filename, 'w') as f:
+            json.dump(model_data, f)
+        print(f"\nModel Is Safe In {filename}")
+
+    def train_model(self, ones_data, zero_data, epochs=1, batch_size=32, min_quantity=0, max_quantity=1.0, max_sequence=10, test_size=0.2, learning_rate=None, random_seed=None, class_weights=None):
+        self.min_quantity = min_quantity
+        self.max_quantity = max_quantity
+        self.max_sequence = max_sequence
+        self.learning_rate = learning_rate if learning_rate is not None else self.learning_rate
+        self.class_weights = class_weights if class_weights is not None else self.class_weights
+        is_generator = isinstance(ones_data[0], types.GeneratorType)
+        all_data = ones_data + zero_data
+        random.seed(random_seed)
+        random.shuffle(all_data)
+        split_index = int((1 - test_size) * len(all_data))
+        self.train_data = all_data[:split_index]
+        self.test_data = all_data[split_index:]
+        train_accuracies, train_losses = [], []
+        test_accuracies, test_losses = [], []
+        for epoch in range(epochs):
+            print(f'Epoch: {epoch + 1}/{epochs}')
+            train_accuracy, train_loss = self._train_epoch(self.train_data, batch_size, is_generator, prefix='Training:')
+            test_accuracy, test_loss = self._test_epoch(self.test_data, batch_size, is_generator, prefix='Testing: ')
+            train_accuracies.append(train_accuracy)
+            train_losses.append(train_loss)
+            test_accuracies.append(test_accuracy)
+            test_losses.append(test_loss)
+        self._print_table(train_accuracies, train_losses, test_accuracies, test_losses)
+        total_lists = len(self.train_data) + len(self.test_data)
+        self.prior_probs['ones'] = len(ones_data) / total_lists
+        self.prior_probs['zero'] = len(zero_data) / total_lists
+        self._filter_word_weights()
+
+    def _train_epoch(self, data, batch_size, is_generator, prefix=''):
+        start_time = time.time()
+        correct_predictions, total_samples, total_loss = 0, 0, 0
+        for i in range(0, len(data), batch_size):
+            batch = data[i:i + batch_size]
+            for lst_gen in batch:
+                lst = list(lst_gen) if is_generator else lst_gen
+                correct_predictions += self._update_weights(lst, 'ones')
+                total_samples += len(lst)
+                total_loss += self._calculate_loss(lst, 'ones')
+            elapsed_time = time.time() - start_time
+            suffix = f'(Accuracy: {correct_predictions/total_samples:.2%}, Loss: {total_loss/total_samples:.4f}, Elapsed: {elapsed_time:.2f}s)'
+            self._print_progress_bar(i + batch_size, len(data), prefix=prefix, suffix=suffix)
+        print('')
+        accuracy = correct_predictions / total_samples
+        average_loss = total_loss / total_samples
+        return accuracy, average_loss
+
+    def _test_epoch(self, data, batch_size, is_generator, prefix=''):
+        start_time = time.time()
+        correct_predictions, total_samples, total_loss = 0, 0, 0
+        for i in range(0, len(data), batch_size):
+            batch = data[i:i + batch_size]
+            for lst_gen in batch:
+                lst = list(lst_gen) if is_generator else lst_gen
+                correct_predictions += (len(lst) - self._update_weights(lst, 'zero'))
+                total_samples += len(lst)
+                total_loss += self._calculate_loss(lst, 'zero')
+            elapsed_time = time.time() - start_time
+            suffix = f'(Accuracy: {correct_predictions/total_samples:.2%}, Loss: {total_loss/total_samples:.4f}, Elapsed: {elapsed_time:.2f}s)'
+            self._print_progress_bar(i + batch_size, len(data), prefix=prefix, suffix=suffix)
+        print('')
+        accuracy = correct_predictions / total_samples
+        average_loss = total_loss / total_samples
+        return accuracy, average_loss
+
+    def _print_table(self, train_accuracies, train_losses, test_accuracies, test_losses):
+        print("\nEpoch\tTrain Accuracy\tTrain Loss\tTest Accuracy\tTest Loss (Table):")
+        for i, (train_acc, train_loss, test_acc, test_loss) in enumerate(zip(train_accuracies, train_losses, test_accuracies, test_losses), 1):
+            print(f"{i}\t{train_acc:.2%}\t\t{train_loss:.4f}\t\t{test_acc:.2%}\t\t{test_loss:.4f}")
+
+    def _print_progress_bar(self, iteration, total, prefix='', suffix='', length=30, fill='█'):
+        percent = min(100.0, max(0.0, 100 * (iteration / float(total))))
+        percent_string = "{0:.2f}".format(percent)
+        filled_length = int(length * iteration // total)
+        bar = fill * filled_length + ' ' * (length - filled_length)
+        print(f'\r    {prefix} |{bar}| {percent_string}% {suffix}', end='\r')
+
+    def _filter_word_weights(self):
+        for word, weights in list(self.word_weights.items()):
+            total_weight = weights['ones'] + weights['zero']
+            if (total_weight < self.min_quantity or total_weight > self.max_quantity * max(self.total_counts.values())):
+                del self.word_weights[word]
+
+    def _get_data_for_word(self, word, category):
+        data = []
+        if isinstance(self.word_weights[word][category], list):
+            for lst in self.word_weights[word][category]:
+                data.append(lst)
+        return data
+
+    def predict(self, new_list):
+        ones_log_prob = self._calculate_category_log_prob(new_list, 'ones')
+        zero_log_prob = self._calculate_category_log_prob(new_list, 'zero')
+        if ones_log_prob > zero_log_prob:
+            return 1
+        else:
+            return 0
+
+    def _update_weights(self, lst, category):
+        correct_prediction = 0
+        for i, word in enumerate(lst):
+            position_encoding = min(i / len(lst), 1.0)
+            word_frequency = self.word_weights[word][category]
+            learning_rate = self.learning_rate / (word_frequency + 1)
+            self.word_weights[word][category] += learning_rate * self.class_weights[category] * position_encoding
+            self.total_counts[category] += learning_rate
+            prediction = self.predict([word])
+            correct_prediction += 1 if prediction == int(category == 'ones') else 0
+        return correct_prediction
+
+    def _calculate_loss(self, lst, category):
+        loss = 0.0
+        for word in lst:
+            if word in self.word_weights:
+                word_weight_category = self.word_weights[word][category]
+                total_weight_category = self.total_counts[category]
+                prob = (word_weight_category + 1) / (total_weight_category + len(self.word_weights))
+                loss += -math.log(prob)
+        return loss
+
+    def _calculate_category_log_prob(self, lst, category):
+        log_prob = 0.0
+        for word in lst:
+            if word in self.word_weights:
+                word_weight_category = self.word_weights[word][category]
+                total_weight_category = self.total_counts[category]
+                prob = (word_weight_category + 1) / (total_weight_category + len(self.word_weights))
+                log_prob += math.log(prob)
+        log_prob += math.log(self.prior_probs[category] + 1e-10)
+        return log_prob

README.md

@@ -14,9 +14,9 @@ Windows 8.1, 10, 11 (64-bit), Ram 1GB, Rom 1GB or higher
 psutil==5.9.5, pefile==2023.2.7, requests==2.31.0, pyperclip==1.8.2, pywin32==306, PyQt5==5.15.9
 
 ## Import Module
-os, gc, sys, time, json, psutil, hashlib, pefile, socket, msvcrt, requests, pyperclip, win32file, win32gui, win32api, win32con, xml, PYAS_Scripts, PYAS_Function, PYAS_Extension, PYAS_Language, PYAS_Interface, threading, PyQt5
+os, gc, sys, time, json, psutil, hashlib, pefile, socket, msvcrt, requests, pyperclip, win32file, win32gui, win32api, win32con, xml, PYAS_Function, PYAS_Function_Safe, PYAS_Extension, PYAS_Language, PYAS_Interface, threading, PyQt5
 
 ## Special Thanks
-360, Wix, VirusShare, mtkiao129, LenStevens
+360, Wix, VirusShare, mtkiao129, LenStevens, AV-T Team of LisectGroup
 
 Copyright© 2020~2023 PYAS Security By 87owo