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

Posted 2023-02-23

【中文标题】使用 CType 时检测从 Windows DLL 调用 Python 脚本

【英文标题】：Detecting Calling Python Script from Windows DLL When Using CTypes

【发布时间】：2013-05-21 15:22:59

【问题描述】：

我正在寻求在 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 来生成签名，所以第一步是获取调用脚本的文件路径。

【参考方案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)

【讨论】：

啊，当然。我希望不在 Python 脚本中进行检查，而是在 Python 脚本调用的 dll 中进行检查，但您的答案提供了足够的信息。通过在 dll 中使用 GetCommandLine()，我将能够识别脚本的文件名以检查它是否已签名，谢谢！