如何在Python中判断文件在Windows上是否可执行?

时间:2009-03-14 23:48:40

标签: c++ python windows winapi

我正在编写grepath实用程序,用于查找%PATH%中与模式匹配的可执行文件。 我需要定义路径中给定的文件名是否可执行(重点是命令行脚本)。

基于"Tell if a file is executable"我得到了:

import os
from pywintypes import error
from win32api   import FindExecutable, GetLongPathName

def is_executable_win(path):
    try:
        _, executable = FindExecutable(path)
        ext = lambda p: os.path.splitext(p)[1].lower()
        if (ext(path) == ext(executable) # reject *.cmd~, *.bat~ cases
            and samefile(GetLongPathName(executable), path)):
            return True
        # path is a document with assoc. check whether it has extension
        # from %PATHEXT% 
        pathexts = os.environ.get('PATHEXT', '').split(os.pathsep)
        return any(ext(path) == e.lower() for e in pathexts)
    except error:
        return None # not an exe or a document with assoc.

samefile的位置:

try: samefile = os.path.samefile
except AttributeError:    
    def samefile(path1, path2):
        rp = lambda p: os.path.realpath(os.path.normcase(p))
        return rp(path1) == rp(path2)

如何在给定的上下文中改进is_executable_win? Win32 API的哪些功能可以提供帮助?

P.S。

  • 时间表现并不重要
  • subst驱动器和UNC,unicode路径未被考虑
  • 如果C ++使用Windows XP上提供的功能
  • ,则答案是可以的

实施例

  • notepad.exe是可执行的(通常)
  • which.py是可执行的,如果它与某些可执行文件(例如,python.exe)相关联,并且.PY%PATHEXT%中,即'C:\> which'可以启动:

    some\path\python.exe another\path\in\PATH\which.py
    
  • somefile.doc很可能是可执行文件(例如,当它与Word关联时)

  • another_file.txt 可执行文件(通常)
  • 如果
  • ack.pl与某些可执行文件(最可能是perl.exe)相关联且.PL位于%PATHEXT%中,则
  • ack是可执行的(即我可以在不指定扩展名的情况下运行def is_executable_win_destructive(path): #NOTE: it assumes `path` <-> `barename` for the sake of example barename = os.path.splitext(os.path.basename(path))[0] p = Popen(barename, stdout=PIPE, stderr=PIPE, shell=True) stdout, stderr = p.communicate() return p.poll() != 1 or stdout != '' or stderr != error_message(barename) 如果它在路径中)

此问题中的“可执行文件”是什么

error_message()

def error_message(barename): return "'%(barename)s' is not recognized as an internal" \ " or external\r\ncommand, operable program or batch file.\r\n" \ % dict(barename=barename) 取决于语言。英文版是:

is_executable_win_destructive()

如果>>> path = r"c:\docs\somefile.doc" >>> barename = "somefile" 返回时它定义路径是否指向此问题的可执行文件。

示例:

c:\cwd> cmd.exe /c somefile

之后执行%COMSPEC%(默认为cmd.exe):

path

如果输出如下:

'somefile' is not recognized as an internal or external
command, operable program or batch file.

然后path不是其他可执行文件(假设示例中barename>>> path = r'c:\bin\grepath.py' >>> barename = 'grepath' 之间存在一对一的对应关系。)

另一个例子:

.PY

如果%PATHEXT%c:\bin中的%PATH%位于c:\docs> grepath Usage: grepath.py [options] PATTERN grepath.py [options] -e PATTERN grepath.py: error: incorrect number of arguments ,那么:

error_message(barename)

以上输出不等于'c:\bin\grepath.py'因此path是“可执行文件”。

所以问题是如何在不实际运行错误的情况下找出{{1}}是否会产生错误?什么Win32 API函数和用于触发'的条件不被识别为内部..'错误?

6 个答案:

答案 0 :(得分:3)

shoosh打败了我:)

如果我没记错的话,你应该尝试读取文件中的前2个字符。如果你回到“MZ”,你就有了一个exe。


hnd = open(file,"rb")
if hnd.read(2) == "MZ":
  print "exe"

答案 1 :(得分:3)

我认为,这应该足够了:

  1. 检查PATHEXT中的文件扩展名 - 文件是否可直接执行
  2. 使用cmd.exe命令“assoc .ext”可以查看文件是否与某些可执行文件相关联(启动此文件时将启动某些可执行文件)。您可以解析不带参数的assoc的捕获输出,并收集所有关联的扩展并检查测试的文件扩展名。
  3. 其他文件扩展名将触发错误“命令无法识别...”因此您可以认为此类文件不可执行。
  4. 我真的不明白你如何分辨somefile.py和somefile.txt之间的区别,因为关联可以真的相同。您可以将系统配置为以与.py文件相同的方式运行.txt文件。

答案 2 :(得分:2)

窗口PE始终以字符“MZ”开头。但是,这也包括任何类型的DLL,它们不一定是可执行文件 要检查这个,你必须打开文件并阅读标题,这可能不是你想要的。

答案 3 :(得分:1)

以下是我在问题中链接的grepath.py

#!/usr/bin/env python
"""Find executables in %PATH% that match PATTERN.

"""
#XXX: remove --use-pathext option

import fnmatch, itertools, os, re, sys, warnings
from optparse import OptionParser
from stat import S_IMODE, S_ISREG, ST_MODE
from subprocess import PIPE, Popen


def warn_import(*args):
    """pass '-Wd' option to python interpreter to see these warnings."""
    warnings.warn("%r" % (args,), ImportWarning, stacklevel=2)


class samefile_win:
    """
http://timgolden.me.uk/python/win32_how_do_i/see_if_two_files_are_the_same_file.html
"""
    @staticmethod
    def get_read_handle (filename):
        return win32file.CreateFile (
            filename,
            win32file.GENERIC_READ,
            win32file.FILE_SHARE_READ,
            None,
            win32file.OPEN_EXISTING,
            0,
            None
            )

    @staticmethod
    def get_unique_id (hFile):
        (attributes,
         created_at, accessed_at, written_at,
         volume,
         file_hi, file_lo,
         n_links,
         index_hi, index_lo
         ) = win32file.GetFileInformationByHandle (hFile)
        return volume, index_hi, index_lo

    @staticmethod
    def samefile_win(filename1, filename2):
        """Whether filename1 and filename2 represent the same file.

It works for subst, ntfs hardlinks, junction points.
It works unreliably for network drives.

Based on GetFileInformationByHandle() Win32 API call.
http://timgolden.me.uk/python/win32_how_do_i/see_if_two_files_are_the_same_file.html
"""
        if samefile_generic(filename1, filename2): return True
        try:
            hFile1 = samefile_win.get_read_handle (filename1)
            hFile2 = samefile_win.get_read_handle (filename2)
            are_equal = (samefile_win.get_unique_id (hFile1)
                         == samefile_win.get_unique_id (hFile2))
            hFile2.Close ()
            hFile1.Close ()
            return are_equal
        except win32file.error:
            return None


def canonical_path(path):
    """NOTE: it might return wrong path for paths with symbolic links."""
    return os.path.realpath(os.path.normcase(path))


def samefile_generic(path1, path2):
    return canonical_path(path1) == canonical_path(path2)


class is_executable_destructive:
    @staticmethod
    def error_message(barename):
        r"""
"'%(barename)s' is not recognized as an internal or external\r\n
command, operable program or batch file.\r\n"

in Russian:
"""
        return '"%(barename)s" \xad\xa5 \xef\xa2\xab\xef\xa5\xe2\xe1\xef \xa2\xad\xe3\xe2\xe0\xa5\xad\xad\xa5\xa9 \xa8\xab\xa8 \xa2\xad\xa5\xe8\xad\xa5\xa9\r\n\xaa\xae\xac\xa0\xad\xa4\xae\xa9, \xa8\xe1\xaf\xae\xab\xad\xef\xa5\xac\xae\xa9 \xaf\xe0\xae\xa3\xe0\xa0\xac\xac\xae\xa9 \xa8\xab\xa8 \xaf\xa0\xaa\xa5\xe2\xad\xeb\xac \xe4\xa0\xa9\xab\xae\xac.\r\n' % dict(barename=barename)

    @staticmethod
    def is_executable_win_destructive(path):
        # assume path <-> barename that is false in general
        barename = os.path.splitext(os.path.basename(path))[0]
        p = Popen(barename, stdout=PIPE, stderr=PIPE, shell=True)
        stdout, stderr = p.communicate()
        return p.poll() != 1 or stdout != '' or stderr != error_message(barename)


def is_executable_win(path):
    """Based on:
http://timgolden.me.uk/python/win32_how_do_i/tell-if-a-file-is-executable.html

Known bugs: treat some "*~" files as executable, e.g. some "*.bat~" files
"""
    try:
        _, executable = FindExecutable(path)
        return bool(samefile(GetLongPathName(executable), path))
    except error:
        return None # not an exe or a document with assoc.


def is_executable_posix(path):
    """Whether the file is executable.

Based on which.py from stdlib
"""
    #XXX it ignores effective uid, guid?
    try: st = os.stat(path)
    except os.error:
        return None

    isregfile = S_ISREG(st[ST_MODE])
    isexemode = (S_IMODE(st[ST_MODE]) & 0111)
    return bool(isregfile and isexemode)

try:
    #XXX replace with ctypes?
    from win32api import FindExecutable, GetLongPathName, error
    is_executable = is_executable_win
except ImportError, e:
    warn_import("is_executable: fall back on posix variant", e)
    is_executable = is_executable_posix

try: samefile = os.path.samefile
except AttributeError, e:
    warn_import("samefile: fallback to samefile_win", e)
    try:
        import win32file
        samefile = samefile_win.samefile_win
    except ImportError, e:
        warn_import("samefile: fallback to generic", e)
        samefile = samefile_generic

def main():
    parser = OptionParser(usage="""
%prog [options] PATTERN
%prog [options] -e PATTERN""", description=__doc__)
    opt = parser.add_option
    opt("-e", "--regex", metavar="PATTERN",
        help="use PATTERN as a regular expression")
    opt("--ignore-case", action="store_true", default=True,
        help="""[default] ignore case when --regex is present; for \
non-regex PATTERN both FILENAME and PATTERN are first \
case-normalized if the operating system requires it otherwise \
unchanged.""")
    opt("--no-ignore-case", dest="ignore_case", action="store_false")
    opt("--use-pathext", action="store_true", default=True,
        help="[default] whether to use %PATHEXT% environment variable")
    opt("--no-use-pathext", dest="use_pathext", action="store_false")
    opt("--show-non-executable", action="store_true", default=False,
        help="show non executable files")

    (options, args) = parser.parse_args()

    if len(args) != 1 and not options.regex:
       parser.error("incorrect number of arguments")
    if not options.regex:
       pattern = args[0]
    del args

    if options.regex:
       filepred = re.compile(options.regex, options.ignore_case and re.I).search
    else:
       fnmatch_ = fnmatch.fnmatch if options.ignore_case else fnmatch.fnmatchcase
       for file_pattern_symbol in "*?":
           if file_pattern_symbol in pattern:
               break
       else: # match in any place if no explicit file pattern symbols supplied
           pattern = "*" + pattern + "*"
       filepred = lambda fn: fnmatch_(fn, pattern)

    if not options.regex and options.ignore_case:
       filter_files = lambda files: fnmatch.filter(files, pattern)
    else:
       filter_files = lambda files: itertools.ifilter(filepred, files)

    if options.use_pathext:
       pathexts = frozenset(map(str.upper,
            os.environ.get('PATHEXT', '').split(os.pathsep)))

    seen = set()
    for dirpath in os.environ.get('PATH', '').split(os.pathsep):
        if os.path.isdir(dirpath): # assume no expansion needed
           # visit "each" directory only once
           # it is unaware of subst drives, junction points, symlinks, etc
           rp = canonical_path(dirpath)
           if rp in seen: continue
           seen.add(rp); del rp

           for filename in filter_files(os.listdir(dirpath)):
               path = os.path.join(dirpath, filename)
               isexe = is_executable(path)

               if isexe == False and is_executable == is_executable_win:
                  # path is a document with associated program
                  # check whether it is a script (.pl, .rb, .py, etc)
                  if not isexe and options.use_pathext:
                     ext = os.path.splitext(path)[1]
                     isexe = ext.upper() in pathexts

               if isexe:
                  print path
               elif options.show_non_executable:
                  print "non-executable:", path


if __name__=="__main__":
   main()

答案 4 :(得分:0)

解析PE格式。

http://code.google.com/p/pefile/

这可能是除了使用python实际尝试运行程序之外的最佳解决方案。

编辑:我看到你也想要有关联的文件。这将需要在注册表中输入,我没有这些信息。

Edit2:我也看到你区分.doc和.py。这是一个相当随意的区分,必须使用手动规则来指定,因为对于Windows,它们都是程序读取的文件扩展名。

答案 5 :(得分:-1)

您的问题无法解答。 Windows无法区分与脚本语言相关的文件与其他任意程序之间的区别。就Windows而言,.PY文件只是一个由python.exe打开的文档。