python-psutil.Process(pid).name的替代

2019-11-02 12:57:40 阅读：557 来源： 互联网

我测量了psutil.Process(pid).name的性能,结果发现它比psutil.Process(pid).exe慢十倍以上.由于这些功能的最后一个功能要求对路径具有不同的特权,因此我不能仅从路径中提取文件名.我的问题是：psutil.Process(pid).name是否有其他替代方法？

解决方法:

您提到这是用于Windows的.我看了一下psutil对Windows的作用.看起来psutil.Process().name正在使用Windows工具帮助API.如果查看psutil的Process代码并跟踪.name,它将转到process_info.c中的get_name().它将遍历系统上的所有pid,直至找到所需的pid.我认为这可能是toolhelp API的局限性.但这就是为什么它比.exe慢的原因,后者使用不同的API路径(如您所指出的那样),需要额外的特权.

我想出的解决方案是使用ctypes和ctypes.windll直接调用Windows ntapi.它只需要PROCESS_QUERY_INFORMATION,与PROCESS_ALL_ACCESS不同：

import ctypes
import os.path

# duplicate the UNICODE_STRING structure from the windows API

class UNICODE_STRING(ctypes.Structure):
    _fields_ = [
      ('Length', ctypes.c_short),
      ('MaximumLength', ctypes.c_short),
      ('Buffer', ctypes.c_wchar_p)
    ]

# args

pid = 8000 # put your pid here

# define some constants; from windows API reference

MAX_TOTAL_PATH_CHARS = 32767
PROCESS_QUERY_INFORMATION = 0x0400 
PROCESS_IMAGE_FILE_NAME = 27

# open handles

ntdll = ctypes.windll.LoadLibrary('ntdll.dll')
process = ctypes.windll.kernel32.OpenProcess(PROCESS_QUERY_INFORMATION,
  False, pid)

# allocate memory

buflen = (((MAX_TOTAL_PATH_CHARS + 1) * ctypes.sizeof(ctypes.c_wchar)) +
  ctypes.sizeof(UNICODE_STRING))
buffer = ctypes.c_char_p(' ' * buflen) 

# query process image filename and parse for process "name"

ntdll.NtQueryInformationProcess(process, PROCESS_IMAGE_FILE_NAME, buffer,
  buflen, None)
pustr = ctypes.cast(buffer, ctypes.POINTER(UNICODE_STRING))
print os.path.split(pustr.contents.Buffer)[-1]

# cleanup

ctypes.windll.kernel32.CloseHandle(process)
ctypes.windll.kernel32.FreeLibrary(ntdll._handle)

标签：pywin32,performance,python
来源： https://codeday.me/bug/20191102/1991506.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

python-psutil.Process(pid).name的替代