Question

我正在寻求在Windows dll中添加功能来检测调用Python脚本的名称。

我使用ctypes通过Python调用dll，如How can I call a DLL from a scripting language?

的答案中所述

在dll中，我能够使用WINAPI GetModuleFileName（）http://msdn.microsoft.com/en-us/library/windows/desktop/ms683197(v=vs.85).aspx成功确定调用进程。但是，由于这是一个Python脚本，它是通过Python可执行文件运行的，因此返回的模块文件名是“C：/Python33/Python.exe”。我需要执行调用的实际脚本文件的名称。这可能吗？

有关原因的一些背景知识：此dll用于身份验证。它使用共享密钥生成哈希，脚本用于验证HTTP请求。它嵌入在dll中，因此使用该脚本的人将无法看到该密钥。我们希望确保调用脚本的python文件已签名，这样不仅任何人都可以使用此dll生成签名，因此获取调用脚本的文件路径是第一步。

Answer 1

通常，在不使用Python C-API的情况下，您可以使用Win32 GetCommandLine和CommandLineToArgvW获取流程命令行并将其解析为argv数组。然后检查argv[1]是否是.py文件。

Python演示，使用ctypes：

import ctypes
from ctypes import wintypes

GetCommandLine = ctypes.windll.kernel32.GetCommandLineW
GetCommandLine.restype = wintypes.LPWSTR
GetCommandLine.argtypes = []

CommandLineToArgvW = ctypes.windll.shell32.CommandLineToArgvW
CommandLineToArgvW.restype = ctypes.POINTER(wintypes.LPWSTR)
CommandLineToArgvW.argtypes = [
    wintypes.LPCWSTR,  # lpCmdLine,
    ctypes.POINTER(ctypes.c_int),  # pNumArgs
]

if __name__ == '__main__':
    cmdline = GetCommandLine()
    argc = ctypes.c_int()
    argv = CommandLineToArgvW(cmdline, ctypes.byref(argc))
    argc = argc.value
    argv = argv[:argc]
    print(argv)

使用CType时检测从Windows DLL调用Python脚本

1 个答案: