|
|
@ -1,106 +0,0 @@ |
|
|
import pefile |
|
|
|
|
|
import sys |
|
|
|
|
|
import re |
|
|
|
|
|
import os |
|
|
|
|
|
import queue |
|
|
|
|
|
import shutil |
|
|
|
|
|
|
|
|
|
|
|
# constant definitions |
|
|
|
|
|
KNOWN_SYS_DLLS = ['WINMM.DLL', 'MSVCRT.DLL', 'VERSION.DLL', 'MPR.DLL', |
|
|
|
|
|
'DWMAPI.DLL', 'UXTHEME.DLL', 'DNSAPI.DLL', 'IPHLPAPI.DLL'] |
|
|
|
|
|
# below is for Ubuntu 18.04 with specified PPA enabled, if you are using |
|
|
|
|
|
# other distro or different repositories, change the following accordingly |
|
|
|
|
|
DLL_PATH = [ |
|
|
|
|
|
'/usr/x86_64-w64-mingw32/bin/', |
|
|
|
|
|
'/usr/x86_64-w64-mingw32/lib/', |
|
|
|
|
|
'/usr/lib/gcc/x86_64-w64-mingw32/7.3-posix/' |
|
|
|
|
|
] |
|
|
|
|
|
|
|
|
|
|
|
missing = [] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def parse_imports(file_name): |
|
|
|
|
|
results = [] |
|
|
|
|
|
pe = pefile.PE(file_name, fast_load=True) |
|
|
|
|
|
pe.parse_data_directories() |
|
|
|
|
|
|
|
|
|
|
|
for entry in pe.DIRECTORY_ENTRY_IMPORT: |
|
|
|
|
|
current = entry.dll.decode() |
|
|
|
|
|
current_u = current.upper() # b/c Windows is often case insensitive |
|
|
|
|
|
# here we filter out system dlls |
|
|
|
|
|
# dll w/ names like *32.dll are likely to be system dlls |
|
|
|
|
|
if current_u.upper() not in KNOWN_SYS_DLLS and not re.match(string=current_u, pattern=r'.*32\.DLL'): |
|
|
|
|
|
results.append(current) |
|
|
|
|
|
|
|
|
|
|
|
return results |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def parse_imports_recursive(file_name, path_list=[]): |
|
|
|
|
|
q = queue.Queue() # create a FIFO queue |
|
|
|
|
|
# file_name can be a string or a list for the convience |
|
|
|
|
|
if isinstance(file_name, str): |
|
|
|
|
|
q.put(file_name) |
|
|
|
|
|
elif isinstance(file_name, list): |
|
|
|
|
|
for i in file_name: |
|
|
|
|
|
q.put(i) |
|
|
|
|
|
full_list = [] |
|
|
|
|
|
while q.qsize(): |
|
|
|
|
|
current = q.get_nowait() |
|
|
|
|
|
print('> %s' % current) |
|
|
|
|
|
deps = parse_imports(current) |
|
|
|
|
|
# if this dll does not have any import, ignore it |
|
|
|
|
|
if not deps: |
|
|
|
|
|
continue |
|
|
|
|
|
for dep in deps: |
|
|
|
|
|
# the dependency already included in the list, skip |
|
|
|
|
|
if dep in full_list: |
|
|
|
|
|
continue |
|
|
|
|
|
# find the requested dll in the provided paths |
|
|
|
|
|
full_path = find_dll(dep) |
|
|
|
|
|
if not full_path: |
|
|
|
|
|
missing.append(dep) |
|
|
|
|
|
continue |
|
|
|
|
|
full_list.append(dep) |
|
|
|
|
|
q.put(full_path) |
|
|
|
|
|
path_list.append(full_path) |
|
|
|
|
|
return full_list |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def find_dll(name): |
|
|
|
|
|
for path in DLL_PATH: |
|
|
|
|
|
for root, _, files in os.walk(path): |
|
|
|
|
|
for f in files: |
|
|
|
|
|
if name.lower() == f.lower(): |
|
|
|
|
|
return os.path.join(root, f) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def deploy(name, dst, dry_run=False): |
|
|
|
|
|
dlls_path = [] |
|
|
|
|
|
parse_imports_recursive(name, dlls_path) |
|
|
|
|
|
for dll_entry in dlls_path: |
|
|
|
|
|
if not dry_run: |
|
|
|
|
|
shutil.copy(dll_entry, dst) |
|
|
|
|
|
else: |
|
|
|
|
|
print('[Dry-Run] Copy %s to %s' % (dll_entry, dst)) |
|
|
|
|
|
print('Deploy completed.') |
|
|
|
|
|
return dlls_path |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def main(): |
|
|
|
|
|
if len(sys.argv) < 3: |
|
|
|
|
|
print('Usage: %s [files to examine ...] [target deploy directory]') |
|
|
|
|
|
return 1 |
|
|
|
|
|
to_deploy = sys.argv[1:-1] |
|
|
|
|
|
tgt_dir = sys.argv[-1] |
|
|
|
|
|
if not os.path.isdir(tgt_dir): |
|
|
|
|
|
print('%s is not a directory.' % tgt_dir) |
|
|
|
|
|
return 1 |
|
|
|
|
|
print('Scanning dependencies...') |
|
|
|
|
|
deploy(to_deploy, tgt_dir) |
|
|
|
|
|
if missing: |
|
|
|
|
|
print('Following DLLs are not found: %s' % ('\n'.join(missing))) |
|
|
|
|
|
return 0 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
|
|
|
main() |
|
|
|
FearlessTobi
4 years ago