Add files via upload

PYAS.py

@@ -10,7 +10,7 @@
 #
 ##################################### 載入模組套件 ###################################
 
-import os, sys, time, psutil, subprocess
+import os, sys, time, json, psutil, subprocess
 from configparser import RawConfigParser
 from pefile import PE, DIRECTORY_ENTRY
 from hashlib import md5, sha1, sha256
@@ -63,58 +63,48 @@ def protect_autorun():
 
 def pyas_library():
     try:
-        if not os.path.isdir('Library/PYAS/Temp'):
-            os.makedirs('Library/PYAS/Temp')
-        if not os.path.isdir('Library/PYAS/Setup'):
-            os.makedirs('Library/PYAS/Setup')
-        if not os.path.isdir('Library/PYAS/Icon'):
-            os.makedirs('Library/PYAS/Icon')
-        if not os.path.isdir('Library/PYAE/Function'):
-            os.makedirs('Library/PYAE/Function')
-        if not os.path.isdir('Library/PYAE/Hashes'):
-            os.makedirs('Library/PYAE/Hashes')
+        checkPath = ['Library/PYAS/Temp',
+                     'Library/PYAS/Setup',
+                     'Library/PYAS/Icon',
+                     'Library/PYAE/Hashes']
+        for i in checkPath:
+            if not os.path.isdir(i):
+                os.makedirs(i)
     except Exception as e:
         pyas_bug_log(e)
 
 ##################################### 更新資料庫 ####################################
-    
+
 def pyas_vl_update():
     try:
-        v = open('Library/PYAE/Hashes/Viruslist.num','r').read()
+        v = open('Library/PYAE/Hashes/Viruslist.num', 'r').read()
     except:
         v = 0
     try:
         import requests as req
-        for i in range(int(v),10000):
+        for i in range(int(v), 10000):
             try:
                 x = 5 - len(str(i))
-                y = '0'*x+str(i)
-                file = req.get('https://virusshare.com/hashfiles/VirusShare_'+y+'.md5', allow_redirects=True)
-                open('Library/PYAE/Hashes/'+y+'.md5', 'wb').write(file.content)
-                with open('Library/PYAE/Hashes/'+str(y)+'.md5',"rb") as f:
-                    bytes = f.read()
-                    readable_hash = str(md5(bytes).hexdigest())
-                f.close()
+                y = '0' * x + str(i)
+                file = req.get(f'https://virusshare.com/hashfiles/VirusShare_{y}.md5', allow_redirects=True)
+                readable_hash = str(md5(file.content).hexdigest())
                 if '449a34c34a7507dffe1d39afad3eeac9' == readable_hash:
-                    v = open('Library/PYAE/Hashes/Viruslist.num','w').write(str(i))
-                    os.remove('Library/PYAE/Hashes/'+str(y)+'.md5')
-                    print('[INFO] Hashes Update Complete (V'+str(i-1)+')')
+                    v = open('Library/PYAE/Hashes/Viruslist.num', 'w').write(str(i))
+                    print(f'[INFO] Hashes Update Complete (V{i - 1})')
                     break
                 else:
-                    try:
-                        with open('Library/PYAE/Hashes/'+str(y)+'.md5','r') as fi:
-                            rfpl = fi.readlines()
-                            with open('Library/PYAE/Hashes/Viruslist.md5','a') as vm:
-                                for j in range(6,len(rfpl)):
-                                    vm.write(str(rfpl[j])[:10]+'\n')
-                        vm.close()
-                        fi.close()
-                    except:
-                        pass
+                    open(f'Library/PYAE/Hashes/{y}.md5', 'wb').write(file.content)
+                    with open(f'Library/PYAE/Hashes/{y}.md5','r') as fi:
+                        rfpl = fi.readlines()
+                        with open('Library/PYAE/Hashes/Viruslist.md5','a') as vm:
+                            for j in range(6, len(rfpl)):
+                                vm.write(str(rfpl[j])[:10]+'\n')
+                    vm.close()
+                    fi.close()
                     v = open('Library/PYAE/Hashes/Viruslist.num','w').write(str(i+1))
-                    os.remove('Library/PYAE/Hashes/'+str(y)+'.md5')
+                    os.remove(f'Library/PYAE/Hashes/{y}.md5')
             except:
-                print('[INFO] Hashes Update Fail (V'+str(i-1)+')')
+                print(f'[INFO] Hashes Update Fail (V{i - 1})')
                 break
     except Exception as e:
         pyas_bug_log(e)
@@ -162,9 +152,13 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
         
     def resizeEvent(self, event):
         width, height = event.size().width(), event.size().height()
-        print('[INFO] '+str(event.size()))
+        print(f'[INFO] {event.size()}')
 
-    def setup_control(self):# TODO
+    def writeConfig(self, config):
+        with open('Library/PYAS/Setup/PYAS.json', 'w', encoding='utf-8') as f:
+            f.write(json.dumps(config, indent=4, ensure_ascii=False))
+ 
+    def setup_control(self):
         self.init()#調用本地函數"init"
         self.ui.Close_Button.clicked.connect(self.close)#讓物件名稱"Close_Button"連接到函數"close"
         self.ui.Minimize_Button.clicked.connect(self.showMinimized)
@@ -244,34 +238,33 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
         self.Process_quantity = []
         self.Virus_Scan = 0
         self.Safe = True
-        self.Process_Timer=QTimer()
+        self.Process_Timer = QTimer()
         self.Process_Timer.timeout.connect(self.Process_list)
-        self.ini_config = RawConfigParser()
+        if (not os.path.exists('Library/PYAS/Setup/PYAS.json')): 
+            config = {"high_sensitivity":0,"language":"english"}
+            self.writeConfig(config)
+        with open('Library/PYAS/Setup/PYAS.json', 'r', encoding='utf-8') as f:
+            self.pyasConfig = json.load(f)
         try:
-            self.ini_config.read(r"Library/PYAS/Setup/PYAS.ini")
-            if self.ini_config.get("Setting","language") == "zh_TW":
-                self.language = "zh_TW"
+            if self.pyasConfig['language'] == "zh_TW":
                 self.lang_init_zh_tw()
                 self.ui.Language_Traditional_Chinese.setChecked(True)
-            elif self.ini_config.get("Setting","language") == "zh_CN":
-                self.language = "zh_CN"
+            elif self.pyasConfig['language'] == "zh_CN":
                 self.lang_init_zh_cn()
                 self.ui.Language_Simplified_Chinese.setChecked(True)
             else:
-                self.language = "english"
                 self.lang_init_en()
                 self.ui.Languahe_English.setChecked(True)
         except:
-            with open(r"Library/PYAS/Setup/PYAS.ini",mode="w",encoding="utf-8") as file:
-                file.write("[Setting]\nhigh_sensitivity = 0\nlanguage = english")
-            file.close()
-            self.language = "english"
+            config = {"high_sensitivity":0,"language":"english"}
+            self.writeConfig(config)
+            self.pyasConfig['language'] = "english"
             self.lang_init_en()
             self.ui.Languahe_English.setChecked(True)
         try:
             Thread(target = pyas_vl_update).start()
             Thread(target = self.pyas_protect_init_zh).start()
-            if self.ini_config.getint("Setting","high_sensitivity") == 1:
+            if self.pyasConfig['high_sensitivity'] == 1:
                 self.show_virus_scan_progress_bar = 1
                 self.ui.Show_Virus_Scan_Progress_Bar_switch_Button.setText(self.text_Translate("已開啟"))
                 self.ui.Show_Virus_Scan_Progress_Bar_switch_Button.setStyleSheet("""
@@ -399,11 +392,11 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
         self.ui.About_Back.setText(_translate("MainWindow", "Back"))
         self.ui.PYAS_Version.setText(_translate("MainWindow", "PYAS V"+pyas_version))
         self.ui.GUI_Made_title.setText(_translate("MainWindow", "GUI Make:"))
-        self.ui.GUI_Made_Name.setText(_translate("MainWindow", "mtkiao129"))
+        self.ui.GUI_Made_Name.setText(_translate("MainWindow", "87owo"))
         self.ui.Core_Made_title.setText(_translate("MainWindow", "Core Make:"))
         self.ui.Core_Made_Name.setText(_translate("MainWindow", "87owo"))
         self.ui.Testers_title.setText(_translate("MainWindow", "Testers:"))
-        self.ui.Testers_Name.setText(_translate("MainWindow", "yang5487"))
+        self.ui.Testers_Name.setText(_translate("MainWindow", "87owo"))
         self.ui.PYAS_URL_title.setText(_translate("MainWindow", "Website:"))
         self.ui.PYAS_URL.setText(_translate("MainWindow", "<html><head/><body><p><a href=\"https://pyantivirus.wixsite.com/pyas?lang=en\"><span style= \" text-decoration: underline; color:#000000;\">https://pyantivirus.wixsite.com/pyas</span></a></p></body></html>"))
         cy = time.strftime('%Y')
@@ -528,11 +521,11 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
         self.ui.About_Back.setText(_translate("MainWindow", "返回"))
         self.ui.PYAS_Version.setText(_translate("MainWindow", "PYAS V"+pyas_version))
         self.ui.GUI_Made_title.setText(_translate("MainWindow", "介面制作:"))
-        self.ui.GUI_Made_Name.setText(_translate("MainWindow", "mtkiao129"))
+        self.ui.GUI_Made_Name.setText(_translate("MainWindow", "87owo"))
         self.ui.Core_Made_title.setText(_translate("MainWindow", "核心制作:"))
         self.ui.Core_Made_Name.setText(_translate("MainWindow", "87owo"))
         self.ui.Testers_title.setText(_translate("MainWindow", "测试人员:"))
-        self.ui.Testers_Name.setText(_translate("MainWindow", "yang5487"))
+        self.ui.Testers_Name.setText(_translate("MainWindow", "87owo"))
         self.ui.PYAS_URL_title.setText(_translate("MainWindow", "官方网站:"))
         self.ui.PYAS_URL.setText(_translate("MainWindow", "<html><head/><body><p><a href=\"https://pyantivirus.wixsite.com/pyas\"><span style=\" text-decoration: underline; color:#000000;\">https://pyantivirus.wixsite.com/pyas</span></a></p></body></html>"))
         cy = time.strftime('%Y')
@@ -657,11 +650,11 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
         self.ui.About_Back.setText(_translate("MainWindow", "返回"))
         self.ui.PYAS_Version.setText(_translate("MainWindow", "PYAS V"+pyas_version))
         self.ui.GUI_Made_title.setText(_translate("MainWindow", "介面製作:"))
-        self.ui.GUI_Made_Name.setText(_translate("MainWindow", "mtkiao129"))
+        self.ui.GUI_Made_Name.setText(_translate("MainWindow", "87owo"))
         self.ui.Core_Made_title.setText(_translate("MainWindow", "核心製作:"))
         self.ui.Core_Made_Name.setText(_translate("MainWindow", "87owo"))
         self.ui.Testers_title.setText(_translate("MainWindow", "測試人員:"))
-        self.ui.Testers_Name.setText(_translate("MainWindow", "yang5487"))
+        self.ui.Testers_Name.setText(_translate("MainWindow", "87owo"))
         self.ui.PYAS_URL_title.setText(_translate("MainWindow", "官方網站:"))
         self.ui.PYAS_URL.setText(_translate("MainWindow", "<html><head/><body><p><a href=\"https://pyantivirus.wixsite.com/pyas\"><span style=\" text-decoration: underline; color:#000000;\">https://pyantivirus.wixsite.com/pyas</span></a></p></body></html>"))
         cy = time.strftime('%Y')
@@ -708,8 +701,8 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
 
 ####################################### 翻譯 #####################################
     
-    def text_Translate(self,text):
-        if self.language == "zh_CN":
+    def text_Translate(self, text):
+        if self.pyasConfig['language'] == "zh_CN":
             text = text.replace("嗎","吗")
             text = text.replace("項","项")
             text = text.replace("護","护")
@@ -733,7 +726,7 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
             text = text.replace("開啟","开启")
             text = text.replace("關閉","关闭")
             return text
-        elif self.language == "english":
+        elif self.pyasConfig['language'] == "english":
             return english_list[text]
         else:
             return text
@@ -805,7 +798,7 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
             self.opacity.setOpacity(self.opacity.i/100)
             widget.setGraphicsEffect(self.opacity)
             self.opacity.i += 1
-            if self.opacity.i >= 100:
+            if self.opacity.i > 100:
                 self.timer.stop()
                 self.timer.deleteLater()
         self.timer = QTimer()
@@ -987,7 +980,7 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
                     return False#回傳不是病毒
         except:
             return False#回傳不是病毒
-        
+
     def pyas_sign_start(self,file):#簽名檢查
         try:
             pe = PE(file,fast_load=True)
@@ -1074,19 +1067,29 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
                 self.ui.Virus_Scan_choose_Button.hide()
                 QApplication.processEvents()
                 print('[INFO] Reading Viruslist Database')#讀取
-                with open('Library/PYAE/Hashes/Viruslist.md5','r') as fp:
-                    rfp = fp.read()
-                fp.close()
-                if self.show_virus_scan_progress_bar != 0:
-                    with open('Library/PYAE/Function/Viruslist.func','r') as fn:
-                        rfn = fn.read()
-                    fn.close()
-                else:
-                    rfn = ''
+                try:
+                    with open('Library/PYAE/Hashes/Viruslist.md5','r') as fp:
+                        rfp = fp.read()
+                    fp.close()
+                except:
+                    rfp = ''
                 if self.show_virus_scan_progress_bar == 0:#確認是否高靈敏
                     if self.pyas_sign_start(file):#簽名檢查
                         if self.pyas_scan_start(file,rfp):#MD5較驗
                             self.pyas_scan_write_en(file)#寫入發現病毒
+                        else:
+                            fts = False
+                            pe = PE(file)
+                            for entry in pe.DIRECTORY_ENTRY_IMPORT:#函數檢測
+                                for function in entry.imports:
+                                    if fts != True:
+                                        if '_CorExeMain' in str(function.name):
+                                            fts = True
+                            pe.close()
+                            if fts:
+                                self.pyas_scan_write_en(file)#寫入發現病毒
+                                fts = False
+                                print('[SCAN] Malware has been detected')
                 else:
                     try:
                         if self.pyas_scan_start(file,rfp):#MD5較驗
@@ -1097,7 +1100,7 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
                             for entry in pe.DIRECTORY_ENTRY_IMPORT:#函數檢測
                                 for function in entry.imports:
                                     if fts != True:
-                                        if str(function.name.decode('utf-8')) in rfn:
+                                        if '_CorExeMain' in str(function.name):
                                             fts = True
                             pe.close()
                             if fts:
@@ -1121,11 +1124,6 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
         self.ui.Virus_Scan_choose_widget.hide()
         self.ui.Virus_Scan_Solve_Button.hide()
         self.ui.Virus_Scan_ProgressBar.hide()
-        self.ini_config = RawConfigParser()
-        try:
-            self.ini_config.read(r"Library/PYAS/Setup/PYAS.ini")
-        except:
-            pass
         self.Virus_List = []
         self.Virus_List_output=QStringListModel()
         self.Virus_List_output.setStringList(self.Virus_List)
@@ -1139,16 +1137,13 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
                 self.ui.Virus_Scan_Break_Button.show()
                 QApplication.processEvents()
                 print('[INFO] Reading Viruslist Database')#讀取
-                with open('Library/PYAE/Hashes/Viruslist.md5','r') as fp:
-                    rfp = fp.read()
-                fp.close()
-                if self.show_virus_scan_progress_bar != 0:
-                    with open('Library/PYAE/Function/Viruslist.func','r') as fn:
-                        rfn = fn.read()
-                    fn.close()
-                else:
-                    rfn = ''
-                self.pyas_scan_path_en(file,rfp,rfn,False)
+                try:
+                    with open('Library/PYAE/Hashes/Viruslist.md5','r') as fp:
+                        rfp = fp.read()
+                    fp.close()
+                except:
+                    rfp = ''
+                self.pyas_scan_path_en(file,rfp,False)
                 self.pyas_scan_answer_en()
             else:
                 self.ui.Virus_Scan_text.setText(self.text_Translate("請選擇掃描方式"))
@@ -1156,7 +1151,7 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
             pyas_bug_log(e)
             QMessageBox.critical(self,self.text_Translate('錯誤'),self.text_Translate('錯誤: ')+str(e),QMessageBox.Ok)
 
-    def pyas_scan_path_en(self,path,rfp,rfn,fts):
+    def pyas_scan_path_en(self,path,rfp,fts):
         try:
             for fd in os.listdir(path):#遍歷檔案
                 try:
@@ -1166,7 +1161,7 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
                     else:
                         fullpath = str(os.path.join(path,fd)).replace("\\", "/")
                         if os.path.isdir(fullpath):#資料夾 深入遍歷
-                            self.pyas_scan_path_en(fullpath,rfp,rfn,fts)
+                            self.pyas_scan_path_en(fullpath,rfp,fts)
                         else:#檔案
                             try:
                                 self.ui.Virus_Scan_text.setText(self.text_Translate("正在掃描: ")+fullpath)
@@ -1175,6 +1170,17 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
                                     if self.pyas_sign_start(fullpath):#簽名檢查
                                         if self.pyas_scan_start(fullpath,rfp):#MD5較驗
                                             self.pyas_scan_write_en(fullpath)#寫入發現病毒
+                                        else:
+                                            pe = PE(fullpath)#函數檢測
+                                            for entry in pe.DIRECTORY_ENTRY_IMPORT:
+                                                for function in entry.imports:
+                                                    if fts != True:
+                                                        if '_CorExeMain' in str(function.name):
+                                                            fts = True
+                                            pe.close()
+                                            if fts:
+                                                self.pyas_scan_write_en(fullpath)#寫入發現病毒
+                                                fts = False
                                 else:
                                     if self.pyas_scan_start(fullpath,rfp):#MD5較驗
                                         self.pyas_scan_write_en(fullpath)#寫入發現病毒
@@ -1183,7 +1189,7 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
                                         for entry in pe.DIRECTORY_ENTRY_IMPORT:
                                             for function in entry.imports:
                                                 if fts != True:
-                                                    if str(function.name.decode('utf-8')) in rfn:
+                                                    if '_CorExeMain' in str(function.name):
                                                         fts = True
                                         pe.close()
                                         if fts:
@@ -1200,30 +1206,31 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
     
     def pyas_scan_disk_init_en(self):
         print('[SCAN] Start Scan Action (Disk Scan)')
-        self.pyas_del_virus_temp()
-        self.Virus_Scan = 1
-        self.ui.Virus_Scan_choose_widget.hide()
-        self.ui.Virus_Scan_Solve_Button.hide()
-        self.ui.Virus_Scan_ProgressBar.hide()
-        self.ui.Virus_Scan_choose_Button.hide()
-        self.ui.Virus_Scan_Break_Button.show()
-        self.Virus_List = []
-        self.Virus_List_output=QStringListModel()
-        self.Virus_List_output.setStringList(self.Virus_List)
-        self.ui.Virus_Scan_output.setModel(self.Virus_List_output)
-        self.scaning = self.text_Translate("正在掃描: ")
         try:
+            self.ui.Virus_Scan_choose_widget.hide()
             self.ui.Virus_Scan_text.setText(self.text_Translate("正在初始化中，請稍後..."))
             QApplication.processEvents()
             print('[INFO] Reading Viruslist Database')
             with open('Library/PYAE/Hashes/Viruslist.md5','r') as self.fp:
                 rfp = self.fp.read()
             self.fp.close()
+            self.pyas_del_virus_temp()
+            self.Virus_Scan = 1
+            self.ui.Virus_Scan_Solve_Button.hide()
+            self.ui.Virus_Scan_ProgressBar.hide()
+            self.ui.Virus_Scan_choose_Button.hide()
+            self.ui.Virus_Scan_Break_Button.show()
+            self.Virus_List = []
+            self.Virus_List_output=QStringListModel()
+            self.Virus_List_output.setStringList(self.Virus_List)
+            self.ui.Virus_Scan_output.setModel(self.Virus_List_output)
+            self.scaning = self.text_Translate("正在掃描: ")
             for d in range(26):
                 self.pyas_scan_disk_en(str(chr(65+d))+':/',rfp)
             self.pyas_scan_answer_en()
         except Exception as e:
             pyas_bug_log(e)
+            self.ui.Virus_Scan_text.setText(self.text_Translate("請選擇掃描方式"))
             QMessageBox.critical(self,self.text_Translate('錯誤'),self.text_Translate('錯誤: ')+str(e),QMessageBox.Ok)
 
     def pyas_scan_disk_en(self,path,rfp):
@@ -1485,15 +1492,6 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
                         except:
                             self.ui.Customize_CMD_Command_output.setText("Error: Add MD5 database failed")
                         return
-                    if CMD_Command.split()[2].lower() == "--func" or CMD_Command.split()[2].lower() == "-f":
-                        try:
-                            value = CMD_Command.split()[3]
-                            with open('Library/PYAE/Function/Viruslist.func','a') as vm:
-                                vm.write(str(value)+'\n')
-                            self.ui.Customize_CMD_Command_output.setText("Successfully added Function to the database!")
-                        except:
-                            self.ui.Customize_CMD_Command_output.setText("Error: Add Function database failed")
-                        return
             else:
                 try:
                     si = subprocess.STARTUPINFO()
@@ -1614,6 +1612,9 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
     def Look_for_File(self):
         try:
             os.remove('Library/PYAS/Temp/PYASF.tmp')
+        except:
+            pass
+        try:
             File_Name = self.ui.Look_for_File_input.text()
             if File_Name != "":
                 self.find_files_info_zh(File_Name)
@@ -1803,45 +1804,74 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
 
     def pyas_protect_init_zh(self):
         try:
+            import win32api,win32con
             ft = open('Library/PYAS/Temp/PYASP.tmp','w',encoding='utf-8')
-            ft.write('real-time protect ini file, dont remove it')
+            ft.write('Protect temp file, dont remove it')
             ft.close()
-        except:
-            pass
-        self.Virus_Scan = 1
-        print('[INFO] Start Action (Real-time Protect)')
-        if self.ui.Protection_switch_Button.text() == self.text_Translate("已關閉"):
-            self.ui.Protection_illustrate.setText(self.text_Translate("啟用該選項可以實時監控進程中的惡意軟體並清除。"))
-            self.ui.Protection_switch_Button.setText(self.text_Translate("已開啟"))
-            self.ui.Protection_switch_Button.setStyleSheet("""
-            QPushButton{border:none;background-color:rgba(20,200,20,100);border-radius: 15px;}
-            QPushButton:hover{background-color:rgba(20,200,20,120);}""")
-        if self.ui.Protection_switch_Button.text() == self.text_Translate("已关闭"):
-            self.ui.Protection_illustrate.setText(self.text_Translate("啟用該選項可以實時監控進程中的惡意軟體並清除。"))
-            self.ui.Protection_switch_Button.setText(self.text_Translate("已开启"))
-            self.ui.Protection_switch_Button.setStyleSheet("""
-            QPushButton{border:none;background-color:rgba(20,200,20,100);border-radius: 15px;}
-            QPushButton:hover{background-color:rgba(20,200,20,120);}""")
-        while os.path.isfile('Library/PYAS/Temp/PYASP.tmp'):
-            for p in psutil.process_iter():
-                try:
-                    time.sleep(0.0001)
-                    QApplication.processEvents()
-                    file = str(p.exe())
-                    if '' == file or 'C:\Windows' in file or 'C:\Program' in file or 'AppData' in file or 'MemCompression' in file or 'Registry' in file or 'PYAS' in file:
-                        pass
-                    else:
-                        if self.pyas_sign_start(file):
-                            if subprocess.call('taskkill /f /im "'+str(p.name())+'" /t',shell=True) == 0:
-                                print('[INFO] Malware blocking success: '+str(p.name()))
-                                self.ui.State_output.append(self.text_Translate('{} > [實時防護] 成功攔截了一個惡意軟體:').format(str(time.strftime('%Y/%m/%d %H:%M:%S')))+str(p.name()))
-                                ToastNotifier().show_toast("PYAS Security",self.text_Translate('{} > [實時防護] 成功攔截了一個惡意軟體:').format(str(time.strftime('%Y/%m/%d %H:%M:%S')))+str(p.name()),icon_path="Library/PYAS/Icon/ICON.ico",threaded=True)
-                            else:
-                                print('[INFO] Malware blocking failed: '+str(p.name()))
-                                self.ui.State_output.append(self.text_Translate('{} > [實時防護] 惡意軟體攔截失敗:').format(time.strftime('%Y/%m/%d %H:%M:%S'))+str(p.name()))
-                                ToastNotifier().show_toast("PYAS Security",self.text_Translate('{} > [實時防護] 惡意軟體攔截失敗:').format(str(time.strftime('%Y/%m/%d %H:%M:%S')))+str(p.name()),icon_path="Library/PYAS/Icon/ICON.ico",threaded=True)
-                except:
-                    continue
+            self.Virus_Scan = 1
+            print('[INFO] Start Action (Real-time Protect)')
+            if self.ui.Protection_switch_Button.text() == self.text_Translate("已關閉"):
+                self.ui.Protection_illustrate.setText(self.text_Translate("啟用該選項可以實時監控進程中的惡意軟體並清除。"))
+                self.ui.Protection_switch_Button.setText(self.text_Translate("已開啟"))
+                self.ui.Protection_switch_Button.setStyleSheet("""
+                QPushButton{border:none;background-color:rgba(20,200,20,100);border-radius: 15px;}
+                QPushButton:hover{background-color:rgba(20,200,20,120);}""")
+            if self.ui.Protection_switch_Button.text() == self.text_Translate("已关闭"):
+                self.ui.Protection_illustrate.setText(self.text_Translate("啟用該選項可以實時監控進程中的惡意軟體並清除。"))
+                self.ui.Protection_switch_Button.setText(self.text_Translate("已开启"))
+                self.ui.Protection_switch_Button.setStyleSheet("""
+                QPushButton{border:none;background-color:rgba(20,200,20,100);border-radius: 15px;}
+                QPushButton:hover{background-color:rgba(20,200,20,120);}""")
+            while os.path.isfile('Library/PYAS/Temp/PYASP.tmp'):
+                for p in psutil.process_iter():
+                    try:
+                        time.sleep(0.0001)
+                        QApplication.processEvents()
+                        file = str(p.exe())
+                        if '' == file or ':\Windows' in file or ':\Program' in file or 'AppData' in file:
+                            pass
+                        elif 'mem' in file.lower() or 'Registry' in file or 'PYAS' in file:
+                            pass
+                        else:
+                            if self.pyas_sign_start(file):
+                                if subprocess.call('taskkill /f /im "'+str(p.name())+'" /t',shell=True) == 0:
+                                    print('[INFO] Malware blocking success: '+str(p.name()))
+                                    self.ui.State_output.append(self.text_Translate('{} > [實時防護] 成功攔截了一個惡意軟體:').format(str(time.strftime('%Y/%m/%d %H:%M:%S')))+str(p.name()))
+                                    ToastNotifier().show_toast("PYAS Security",self.text_Translate('{} > [實時防護] 成功攔截了一個惡意軟體:').format(str(time.strftime('%Y/%m/%d %H:%M:%S')))+str(p.name()),icon_path="Library/PYAS/Icon/ICON.ico",threaded=True)
+                                else:
+                                    rp_reg = False
+                                    try:
+                                        key = win32api.RegOpenKey(win32con.HKEY_CURRENT_USER,'SOFTWARE\Policies\Microsoft\Windows\System',0,win32con.KEY_ALL_ACCESS)
+                                    except:
+                                        key = win32api.RegOpenKey(win32con.HKEY_CURRENT_USER,'SOFTWARE\Policies\Microsoft\Windows',0,win32con.KEY_ALL_ACCESS)
+                                        win32api.RegCreateKey(key,'System')
+                                        key = win32api.RegOpenKey(win32con.HKEY_CURRENT_USER,'SOFTWARE\Policies\Microsoft\Windows\System',0,win32con.KEY_ALL_ACCESS)
+                                    try:
+                                        win32api.RegDeleteValue(key, 'DisableCMD')
+                                        rp_reg = True
+                                    except:
+                                        pass
+                                    win32api.RegCloseKey(key)
+                                    try:
+                                        key = win32api.RegOpenKey(win32con.HKEY_LOCAL_MACHINE,'SOFTWARE\Policies\Microsoft\Windows\System',0,win32con.KEY_ALL_ACCESS)
+                                    except:
+                                        key = win32api.RegOpenKey(win32con.HKEY_LOCAL_MACHINE,'SOFTWARE\Policies\Microsoft\Windows',0,win32con.KEY_ALL_ACCESS)
+                                        win32api.RegCreateKey(key,'System')
+                                        key = win32api.RegOpenKey(win32con.HKEY_LOCAL_MACHINE,'SOFTWARE\Policies\Microsoft\Windows\System',0,win32con.KEY_ALL_ACCESS)
+                                    try:
+                                        win32api.RegDeleteValue(key, 'DisableCMD')
+                                        rp_reg = True
+                                    except:
+                                        pass
+                                    win32api.RegCloseKey(key)
+                                    if rp_reg == False:
+                                        print('[INFO] Malware blocking failed: '+str(p.name()))
+                                        self.ui.State_output.append(self.text_Translate('{} > [實時防護] 惡意軟體攔截失敗:').format(time.strftime('%Y/%m/%d %H:%M:%S'))+str(p.name()))
+                                        ToastNotifier().show_toast("PYAS Security",self.text_Translate('{} > [實時防護] 惡意軟體攔截失敗:').format(str(time.strftime('%Y/%m/%d %H:%M:%S')))+str(p.name()),icon_path="Library/PYAS/Icon/ICON.ico",threaded=True)
+                    except:
+                        continue
+        except Exception as e:
+            pyas_bug_log(e)
 
 ##################################### 系統設置 #####################################
 
@@ -1873,48 +1903,29 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
 
     def Show_Virus_Scan_Progress_Bar_switch(self):
         try:
-            self.ini_config = RawConfigParser()
-            self.ini_config.read(r"Library/PYAS/Setup/PYAS.ini")
             sw_state = self.ui.Show_Virus_Scan_Progress_Bar_switch_Button.text()
             if sw_state == self.text_Translate("已關閉"):
-                self.ini_config.set("Setting", "high_sensitivity", 1)
-                self.ini_config.write(open(r"Library/PYAS/Setup/PYAS.ini", 'w'))
+                self.pyasConfig['high_sensitivity'] = 1
+                self.writeConfig(self.pyasConfig)
                 self.ui.Show_Virus_Scan_Progress_Bar_switch_Button.setText(self.text_Translate("已開啟"))
                 self.ui.Show_Virus_Scan_Progress_Bar_switch_Button.setStyleSheet("""
                 QPushButton{border:none;background-color:rgba(20,200,20,100);border-radius: 15px;}
                 QPushButton:hover{background-color:rgba(20,200,20,120);}""")
                 self.show_virus_scan_progress_bar = 1
             elif sw_state == self.text_Translate("已開啟"):
-                self.ini_config.set("Setting", "high_sensitivity", 0)
-                self.ini_config.write(open(r"Library/PYAS/Setup/PYAS.ini", 'w'))
+                self.pyasConfig['high_sensitivity'] = 0
+                self.writeConfig(self.pyasConfig)
                 self.ui.Show_Virus_Scan_Progress_Bar_switch_Button.setText(self.text_Translate("已關閉"))
                 self.ui.Show_Virus_Scan_Progress_Bar_switch_Button.setStyleSheet("""
                 QPushButton{border:none;background-color:rgba(20,20,20,30);border-radius: 15px;}
                 QPushButton:hover{background-color:rgba(20,20,20,50);}""")
                 self.show_virus_scan_progress_bar = 0
-            if sw_state == self.text_Translate("已关闭"):
-                self.ini_config.set("Setting", "high_sensitivity", 1)
-                self.ini_config.write(open(r"Library/PYAS/Setup/PYAS.ini", 'w'))
-                self.ui.Show_Virus_Scan_Progress_Bar_switch_Button.setText(self.text_Translate("已开启"))
-                self.ui.Show_Virus_Scan_Progress_Bar_switch_Button.setStyleSheet("""
-                QPushButton{border:none;background-color:rgba(20,200,20,100);border-radius: 15px;}
-                QPushButton:hover{background-color:rgba(20,200,20,120);}""")
-                self.show_virus_scan_progress_bar = 1
-            elif sw_state == self.text_Translate("已开启"):
-                self.ini_config.set("Setting", "high_sensitivity", 0)
-                self.ini_config.write(open(r"Library/PYAS/Setup/PYAS.ini", 'w'))
-                self.ui.Show_Virus_Scan_Progress_Bar_switch_Button.setText(self.text_Translate("已关闭"))
-                self.ui.Show_Virus_Scan_Progress_Bar_switch_Button.setStyleSheet("""
-                QPushButton{border:none;background-color:rgba(20,20,20,30);border-radius: 15px;}
-                QPushButton:hover{background-color:rgba(20,20,20,50);}""")
-                self.show_virus_scan_progress_bar = 0
             else:
                 pass
         except:
             try:
-                with open(r"Library/PYAS/Setup/PYAS.ini",mode="w",encoding="utf-8") as file:
-                    file.write("[Setting]\nhigh_sensitivity = 0\nlanguage = english")
-                file.close()
+                config = {'high_sensitivity': 0,'language': 'english'}
+                self.writeConfig(config)
             except Exception as e:
                 pyas_bug_log(e)
 
@@ -1922,24 +1933,18 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
     
     def Change_language(self):
         try:
-            self.ini_config = RawConfigParser()
-            self.ini_config.read(r"Library/PYAS/Setup/PYAS.ini")
             if self.ui.Language_Traditional_Chinese.isChecked():
                 self.lang_init_zh_tw()
-                self.language = "zh_TW"
-                self.ini_config.set("Setting", "language", "zh_TW") 
+                self.pyasConfig['language'] = "zh_TW"
+                self.writeConfig(self.pyasConfig)
             elif self.ui.Language_Simplified_Chinese.isChecked():
                 self.lang_init_zh_cn()
-                self.language = "zh_CN"
-                self.ini_config.set("Setting", "language", "zh_CN") 
+                self.pyasConfig['language'] = "zh_CN"
+                self.writeConfig(self.pyasConfig)
             else:
                 self.lang_init_en()
-                self.language = "english"
-                self.ini_config.set("Setting", "language", "english") 
-        except:
-            pass
-        try:
-            self.ini_config.write(open(r"Library/PYAS/Setup/PYAS.ini", 'w'))
+                self.pyasConfig['language'] = "english"
+                self.writeConfig(self.pyasConfig)
         except Exception as e:
             pyas_bug_log(e)
 
@@ -1983,8 +1988,8 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
             self.m_Position=event.globalPos()-self.pos() #獲取鼠標相對窗口的位置
             event.accept()
             while self.ui.pyas_opacity > 60 and self.m_flag == True:
-                time.sleep(0.01)
-                self.ui.pyas_opacity -= 2
+                time.sleep(0.003)
+                self.ui.pyas_opacity -= 1
                 window.setWindowOpacity(self.ui.pyas_opacity/100)
                 QApplication.processEvents()
         
@@ -2001,8 +2006,8 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
         self.m_flag=False
         self.setCursor(QCursor(Qt.ArrowCursor))
         while self.ui.pyas_opacity < 100 and self.m_flag == False:
-            time.sleep(0.01)
-            self.ui.pyas_opacity += 2
+            time.sleep(0.003)
+            self.ui.pyas_opacity += 1
             window.setWindowOpacity(self.ui.pyas_opacity/100)
             QApplication.processEvents()
 
@@ -2018,6 +2023,13 @@ class MainWindow_Controller(QtWidgets.QMainWindow):
         rect.setHeight(rect.height()-10)
         pat2.drawRoundedRect(rect, 1, 1)
 
+    def closeEvent(self, event):
+        while self.ui.pyas_opacity > 0:
+            time.sleep(0.001)
+            self.ui.pyas_opacity -= 1
+            window.setWindowOpacity(self.ui.pyas_opacity/100)
+            QApplication.processEvents()
+
 ##################################### 主初始化 #####################################
 
 if __name__ == '__main__':
@@ -2025,15 +2037,19 @@ if __name__ == '__main__':
         pyas_library()
         pyasp_remove()
         pyasb_remove()
-        pyas_version = "2.5.5"
-        pyae_version = "2.2.1"
+        pyas_version = "2.5.8"
+        pyae_version = "2.2.2"
         QtCore.QCoreApplication.setAttribute(Qt.AA_EnableHighDpiScaling)# 自適應窗口縮放
         QtGui.QGuiApplication.setAttribute(Qt.HighDpiScaleFactorRoundingPolicy.PassThrough)# 自適應窗口縮放
         app = QtWidgets.QApplication(sys.argv)
         window = MainWindow_Controller()
         window.show()
-        print('[INFO] PYAS V'+pyas_version+' , PYAE V'+pyae_version)
-        print('[LOAD] Process Time: '+str(time.process_time()-start_m)+' sec')
+        for i in range(101):
+            time.sleep(0.001)
+            window.setWindowOpacity(i/100)
+            QApplication.processEvents()
+        print(f'[INFO] PYAS V{pyas_version} , PYAE V{pyae_version}')
+        print(f'[LOAD] Process Time: {time.process_time()-start_m} sec')
         sys.exit(app.exec_())
     except Exception as e:
         pyas_bug_log(e)

PYAS_UI.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-# Form implementation generated from reading ui file 'C:\Users\Windows\Documents\Project\Python\PYDT開發團隊\PYAS防毒軟體\PYAS_程式碼\PYAS_UI.ui'
+# Form implementation generated from reading ui file 'C:\Users\Windows\Documents\Project\Python\PYDT開發團隊\PYAS防毒軟體\PYAS_程式碼\Library\PYAS\Qtui\PYAS_UI.ui'
 #
 # Created by: PyQt5 UI code generator 5.15.7
 #
@@ -569,7 +569,7 @@ class Ui_MainWindow(object):
         self.Encryption_Text_output.setFont(font)
         self.Encryption_Text_output.setObjectName("Encryption_Text_output")
         self.Encryption_Text_Password_input = QtWidgets.QLineEdit(self.Encryption_Text_widget)
-        self.Encryption_Text_Password_input.setGeometry(QtCore.QRect(140, 400, 281, 31))
+        self.Encryption_Text_Password_input.setGeometry(QtCore.QRect(150, 400, 271, 31))
         font = QtGui.QFont()
         font.setFamily("Microsoft YaHei")
         font.setPointSize(11)
@@ -1853,7 +1853,7 @@ class Ui_MainWindow(object):
         self.PYAE_Version.setAlignment(QtCore.Qt.AlignLeading|QtCore.Qt.AlignLeft|QtCore.Qt.AlignVCenter)
         self.PYAE_Version.setObjectName("PYAE_Version")
         self.License_terms = QtWidgets.QTextEdit(self.About_widget)
-        self.License_terms.setGeometry(QtCore.QRect(30, 300, 611, 131))
+        self.License_terms.setGeometry(QtCore.QRect(30, 310, 611, 121))
         self.License_terms.setStyleSheet("")
         self.License_terms.setReadOnly(True)
         self.License_terms.setObjectName("License_terms")

PYAS_UI.ui

@@ -1103,9 +1103,9 @@ QPushButton:pressed
      <widget class="QLineEdit" name="Encryption_Text_Password_input">
       <property name="geometry">
        <rect>
-        <x>140</x>
+        <x>150</x>
         <y>400</y>
-        <width>281</width>
+        <width>271</width>
         <height>31</height>
        </rect>
       </property>
@@ -3719,9 +3719,9 @@ QPushButton:hover
       <property name="geometry">
        <rect>
         <x>30</x>
-        <y>300</y>
+        <y>310</y>
         <width>611</width>
-        <height>131</height>
+        <height>121</height>
        </rect>
       </property>
       <property name="styleSheet">

PYAS_Version_File.py

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