From 5bd39214352105c28f07cbb352fbde21b3b73d41 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=81=8D=E5=85=AE=E6=83=9A=E5=85=AE?=
 <101191390+HIllya51@users.noreply.github.com>
Date: Wed, 31 Jan 2024 20:40:45 +0800
Subject: [PATCH] clear up api-ms-win

---
 LunaTranslator/pack.py | 44 +++++++++++++++++++++++++++++++++---------
 1 file changed, 35 insertions(+), 9 deletions(-)

diff --git a/LunaTranslator/pack.py b/LunaTranslator/pack.py
index 7741b83a..88e39583 100644
--- a/LunaTranslator/pack.py
+++ b/LunaTranslator/pack.py
@@ -53,23 +53,49 @@ for f in ['imageformats','platforms','styles']:
 
 remove(rf'{targetdir}\files\plugins\{baddll}')
 
+shutil.copy(rf'{downlevel}\ucrtbase.dll',targetdir_in)
 collect=[]
 for _dir,_,fs in os.walk(targetdir):
     for f in fs:
         collect.append(os.path.join(_dir,f))
-dlls=[]
 for f in collect:
     if f.endswith('.pyc') or f.endswith('Thumbs.db'):
         os.remove(f)
     elif f.endswith('.exe') or f.endswith('.pyd') or f.endswith('.dll'):
-        print(f)
-        dlls+=get_import_table(f) 
-dlls+=get_import_table(rf'{downlevel}\ucrtbase.dll')+['ucrtbase.dll']
-for f in set(dlls):
-    if os.path.exists(rf'{targetdir_in}\{f}'):
-        continue
-    elif os.path.exists(rf'{downlevel}\{f}'):
-        shutil.copy(rf'{downlevel}\{f}',targetdir_in)
+        
+        try:
+            pe = pefile.PE(f)
+            import_table = pe.DIRECTORY_ENTRY_IMPORT
+            imports=[]
+            for entry in import_table:
+                if entry.dll.decode("utf-8").lower().startswith('api'):
+                    imports.append(entry.dll.decode("utf-8"))
+            pe.close()
+        except:
+            continue
+        if f.endswith('Magpie.Core.exe'):continue
+        if f.endswith('QtWidgets.pyd'):
+            imports+=['api-ms-win-crt-runtime-l1-1-0.dll','api-ms-win-crt-heap-l1-1-0.dll'] 
+            #pefile好像有bug，仅对于QtWidgets.pyd这个文件，只能读取到导入了Qt5Widgets.dll
+        print(f,imports)
+        if len(imports)==0:continue
+        with open(f,'rb') as ff:
+            bs=bytearray(ff.read())
+        for _dll in imports:
+            if _dll.lower().startswith('api-ms-win-core'):
+                #其实对于api-ms-win-core-winrt-XXX实际上是到ComBase.dll之类的，不过此项目中不包含这些
+                _target='kernel32.dll'
+            elif _dll.lower().startswith('api-ms-win-crt'):
+                _target='ucrtbase.dll'
+            _dll=_dll.encode()
+            _target=_target.encode()
+            idx=bs.find(_dll)
+            #print(len(bs))
+            bs[idx:idx+len(_dll)]=_target+b'\0'*(len(_dll)-len(_target))
+            #print(len(bs))
+        with open(f,'wb') as ff:
+            ff.write(bs)
+
 if os.path.exists(rf'{targetdir}\..\{target}'):
     os.remove(rf'{targetdir}\..\{target}')
 os.system(rf'"C:\Program Files\7-Zip\7z.exe" a -m0=LZMA -mx9 {targetdir}\..\{target} {targetdir}')
\ No newline at end of file