123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131 |
- # -*- coding: utf-8 -*-
- #
- # Copyright 2014 Thomas Amland <thomas.amland@gmail.com>
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- """
- :module: watchdog.utils.win32stat
- :synopsis: Implementation of stat with st_ino and st_dev support.
- Functions
- ---------
- .. autofunction:: stat
- """
- import ctypes
- import ctypes.wintypes
- import stat as stdstat
- from collections import namedtuple
- INVALID_HANDLE_VALUE = ctypes.c_void_p(-1).value
- OPEN_EXISTING = 3
- FILE_READ_ATTRIBUTES = 0x80
- FILE_ATTRIBUTE_NORMAL = 0x80
- FILE_ATTRIBUTE_READONLY = 0x1
- FILE_ATTRIBUTE_DIRECTORY = 0x10
- FILE_FLAG_BACKUP_SEMANTICS = 0x02000000
- FILE_FLAG_OPEN_REPARSE_POINT = 0x00200000
- class FILETIME(ctypes.Structure):
- _fields_ = [("dwLowDateTime", ctypes.wintypes.DWORD),
- ("dwHighDateTime", ctypes.wintypes.DWORD)]
- class BY_HANDLE_FILE_INFORMATION(ctypes.Structure):
- _fields_ = [('dwFileAttributes', ctypes.wintypes.DWORD),
- ('ftCreationTime', FILETIME),
- ('ftLastAccessTime', FILETIME),
- ('ftLastWriteTime', FILETIME),
- ('dwVolumeSerialNumber', ctypes.wintypes.DWORD),
- ('nFileSizeHigh', ctypes.wintypes.DWORD),
- ('nFileSizeLow', ctypes.wintypes.DWORD),
- ('nNumberOfLinks', ctypes.wintypes.DWORD),
- ('nFileIndexHigh', ctypes.wintypes.DWORD),
- ('nFileIndexLow', ctypes.wintypes.DWORD)]
- kernel32 = ctypes.WinDLL("kernel32")
- CreateFile = kernel32.CreateFileW
- CreateFile.restype = ctypes.wintypes.HANDLE
- CreateFile.argtypes = (
- ctypes.c_wchar_p,
- ctypes.wintypes.DWORD,
- ctypes.wintypes.DWORD,
- ctypes.c_void_p,
- ctypes.wintypes.DWORD,
- ctypes.wintypes.DWORD,
- ctypes.wintypes.HANDLE,
- )
- GetFileInformationByHandle = kernel32.GetFileInformationByHandle
- GetFileInformationByHandle.restype = ctypes.wintypes.BOOL
- GetFileInformationByHandle.argtypes = (
- ctypes.wintypes.HANDLE,
- ctypes.wintypes.POINTER(BY_HANDLE_FILE_INFORMATION),
- )
- CloseHandle = kernel32.CloseHandle
- CloseHandle.restype = ctypes.wintypes.BOOL
- CloseHandle.argtypes = (ctypes.wintypes.HANDLE,)
- StatResult = namedtuple('StatResult', 'st_dev st_ino st_mode st_mtime st_size')
- def _to_mode(attr):
- m = 0
- if (attr & FILE_ATTRIBUTE_DIRECTORY):
- m |= stdstat.S_IFDIR | 0o111
- else:
- m |= stdstat.S_IFREG
- if (attr & FILE_ATTRIBUTE_READONLY):
- m |= 0o444
- else:
- m |= 0o666
- return m
- def _to_unix_time(ft):
- t = (ft.dwHighDateTime) << 32 | ft.dwLowDateTime
- return (t / 10000000) - 11644473600
- def stat(path):
- hfile = CreateFile(path,
- FILE_READ_ATTRIBUTES,
- 0,
- None,
- OPEN_EXISTING,
- FILE_ATTRIBUTE_NORMAL
- | FILE_FLAG_BACKUP_SEMANTICS
- | FILE_FLAG_OPEN_REPARSE_POINT,
- None)
- if hfile == INVALID_HANDLE_VALUE:
- raise ctypes.WinError()
- info = BY_HANDLE_FILE_INFORMATION()
- r = GetFileInformationByHandle(hfile, info)
- CloseHandle(hfile)
- if not r:
- raise ctypes.WinError()
- return StatResult(st_dev=info.dwVolumeSerialNumber,
- st_ino=(info.nFileIndexHigh << 32) + info.nFileIndexLow,
- st_mode=_to_mode(info.dwFileAttributes),
- st_mtime=_to_unix_time(info.ftLastWriteTime),
- st_size=(info.nFileSizeHigh << 32) + info.nFileSizeLow
- )
|