mirror of
https://github.com/dobin/SuperMega
synced 2026-06-02 17:27:10 +00:00
Dobin Rutishauser
66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
import sys
|
|
import ctypes
|
|
import os
|
|
from typing import List
|
|
|
|
from pe.superpe import SuperPe
|
|
|
|
|
|
class DllResolve():
|
|
def __init__(self, dllname, cdll_res, path_res):
|
|
self.dllname = dllname
|
|
self.cdll_res = cdll_res
|
|
self.path_res = path_res
|
|
|
|
|
|
def all_dll_exist(superpe) -> bool:
|
|
for dll_name in superpe.get_iat_entries():
|
|
if not check_dll_availability(dll_name):
|
|
return False
|
|
return True
|
|
|
|
|
|
def unresolved_dlls(superpe) -> List[str]:
|
|
res = []
|
|
for dll_name in superpe.get_iat_entries():
|
|
if not check_dll_availability(dll_name):
|
|
res.append(dll_name)
|
|
return res
|
|
|
|
|
|
def resolve_dlls(superpe) -> List[DllResolve]:
|
|
res = []
|
|
for dll_name in superpe.get_iat_entries():
|
|
res.append(resolve_dll(dll_name))
|
|
return res
|
|
|
|
|
|
def resolve_dll(dllname) -> DllResolve:
|
|
cdll_res = check_dll_availability(dllname)
|
|
path_res = search_for_dll(dllname)
|
|
return DllResolve(dllname, cdll_res, path_res)
|
|
|
|
|
|
def check_dll_availability(dll_name) -> bool:
|
|
"""Check if a DLL is available for loading by attempting to load it with ctypes."""
|
|
try:
|
|
_ = ctypes.CDLL(dll_name)
|
|
return True
|
|
except OSError:
|
|
return False
|
|
|
|
|
|
def search_for_dll(dll_name) -> str:
|
|
"""Search for a DLL in the system directories and PATH."""
|
|
paths = [
|
|
os.getcwd(), # Current directory
|
|
os.environ.get('SYSTEMROOT', '') + '\\System32', # System directory
|
|
os.environ.get('SYSTEMROOT', ''), # Windows directory
|
|
] + os.environ.get('PATH', '').split(';') # PATH directories
|
|
|
|
for path in paths:
|
|
full_path = os.path.join(path, dll_name)
|
|
if os.path.exists(full_path):
|
|
return full_path
|
|
return "<not found>"
|