# win32.py - utility functions that use win32 API

#

# Copyright 2005-2009 Matt Mackall <mpm@selenic.com> and others

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

from __future__ import absolute_import

import ctypes

import ctypes.wintypes as wintypes

import errno

import msvcrt

import os

import random

import subprocess

from . import (

    encoding,

    pycompat,

)

_kernel32 = ctypes.windll.kernel32

_advapi32 = ctypes.windll.advapi32

_user32 = ctypes.windll.user32

_crypt32 = ctypes.windll.crypt32

_BOOL = ctypes.c_long

_WORD = ctypes.c_ushort

_DWORD = ctypes.c_ulong

_UINT = ctypes.c_uint

_LONG = ctypes.c_long

_LPCSTR = _LPSTR = ctypes.c_char_p

_HANDLE = ctypes.c_void_p

_HWND = _HANDLE

_PCCERT_CONTEXT = ctypes.c_void_p

_MAX_PATH = wintypes.MAX_PATH

_INVALID_HANDLE_VALUE = _HANDLE(-1).value

# GetLastError

_ERROR_SUCCESS = 0

_ERROR_NO_MORE_FILES = 18

_ERROR_INVALID_PARAMETER = 87

_ERROR_BROKEN_PIPE = 109

_ERROR_INSUFFICIENT_BUFFER = 122

_ERROR_NO_DATA = 232

# WPARAM is defined as UINT_PTR (unsigned type)

# LPARAM is defined as LONG_PTR (signed type)

if ctypes.sizeof(ctypes.c_long) == ctypes.sizeof(ctypes.c_void_p):

    _WPARAM = ctypes.c_ulong

    _LPARAM = ctypes.c_long

elif ctypes.sizeof(ctypes.c_longlong) == ctypes.sizeof(ctypes.c_void_p):

    _WPARAM = ctypes.c_ulonglong

    _LPARAM = ctypes.c_longlong

class _FILETIME(ctypes.Structure):

    _fields_ = [('dwLowDateTime', _DWORD), ('dwHighDateTime', _DWORD)]

class _BY_HANDLE_FILE_INFORMATION(ctypes.Structure):

    _fields_ = [

        ('dwFileAttributes', _DWORD),

        ('ftCreationTime', _FILETIME),

        ('ftLastAccessTime', _FILETIME),

        ('ftLastWriteTime', _FILETIME),

        ('dwVolumeSerialNumber', _DWORD),

        ('nFileSizeHigh', _DWORD),

        ('nFileSizeLow', _DWORD),

        ('nNumberOfLinks', _DWORD),

        ('nFileIndexHigh', _DWORD),

        ('nFileIndexLow', _DWORD),

    ]

# CreateFile

_FILE_SHARE_READ = 0x00000001

_FILE_SHARE_WRITE = 0x00000002

_FILE_SHARE_DELETE = 0x00000004

_OPEN_EXISTING = 3

_FILE_FLAG_BACKUP_SEMANTICS = 0x02000000

# SetFileAttributes

_FILE_ATTRIBUTE_NORMAL = 0x80

_FILE_ATTRIBUTE_NOT_CONTENT_INDEXED = 0x2000

# Process Security and Access Rights

_PROCESS_QUERY_INFORMATION = 0x0400

# GetExitCodeProcess

_STILL_ACTIVE = 259

class _STARTUPINFO(ctypes.Structure):

    _fields_ = [

        ('cb', _DWORD),

        ('lpReserved', _LPSTR),

        ('lpDesktop', _LPSTR),

        ('lpTitle', _LPSTR),

        ('dwX', _DWORD),

        ('dwY', _DWORD),

        ('dwXSize', _DWORD),

        ('dwYSize', _DWORD),

        ('dwXCountChars', _DWORD),

        ('dwYCountChars', _DWORD),

        ('dwFillAttribute', _DWORD),

        ('dwFlags', _DWORD),

        ('wShowWindow', _WORD),

        ('cbReserved2', _WORD),

        ('lpReserved2', ctypes.c_char_p),

        ('hStdInput', _HANDLE),

        ('hStdOutput', _HANDLE),

        ('hStdError', _HANDLE),

    ]

class _PROCESS_INFORMATION(ctypes.Structure):

    _fields_ = [

        ('hProcess', _HANDLE),

        ('hThread', _HANDLE),

        ('dwProcessId', _DWORD),

        ('dwThreadId', _DWORD),

    ]

_CREATE_NO_WINDOW = 0x08000000

_SW_HIDE = 0

class _COORD(ctypes.Structure):

    _fields_ = [('X', ctypes.c_short), ('Y', ctypes.c_short)]

class _SMALL_RECT(ctypes.Structure):

    _fields_ = [

        ('Left', ctypes.c_short),

        ('Top', ctypes.c_short),

        ('Right', ctypes.c_short),

        ('Bottom', ctypes.c_short),

    ]

class _CONSOLE_SCREEN_BUFFER_INFO(ctypes.Structure):

    _fields_ = [

        ('dwSize', _COORD),

        ('dwCursorPosition', _COORD),

        ('wAttributes', _WORD),

        ('srWindow', _SMALL_RECT),

        ('dwMaximumWindowSize', _COORD),

    ]

_STD_OUTPUT_HANDLE = _DWORD(-11).value

_STD_ERROR_HANDLE = _DWORD(-12).value

# CERT_TRUST_STATUS dwErrorStatus

CERT_TRUST_IS_PARTIAL_CHAIN = 0x10000

# CertCreateCertificateContext encodings

X509_ASN_ENCODING = 0x00000001

PKCS_7_ASN_ENCODING = 0x00010000

# These structs are only complete enough to achieve what we need.

class CERT_CHAIN_CONTEXT(ctypes.Structure):

    _fields_ = (

        ("cbSize", _DWORD),

        # CERT_TRUST_STATUS struct

        ("dwErrorStatus", _DWORD),

        ("dwInfoStatus", _DWORD),

        ("cChain", _DWORD),

        ("rgpChain", ctypes.c_void_p),

        ("cLowerQualityChainContext", _DWORD),

        ("rgpLowerQualityChainContext", ctypes.c_void_p),

        ("fHasRevocationFreshnessTime", _BOOL),

        ("dwRevocationFreshnessTime", _DWORD),

    )

class CERT_USAGE_MATCH(ctypes.Structure):

    _fields_ = (

        ("dwType", _DWORD),

        # CERT_ENHKEY_USAGE struct

        ("cUsageIdentifier", _DWORD),

        ("rgpszUsageIdentifier", ctypes.c_void_p),  # LPSTR *

    )

class CERT_CHAIN_PARA(ctypes.Structure):

    _fields_ = (

        ("cbSize", _DWORD),

        ("RequestedUsage", CERT_USAGE_MATCH),

        ("RequestedIssuancePolicy", CERT_USAGE_MATCH),

        ("dwUrlRetrievalTimeout", _DWORD),

        ("fCheckRevocationFreshnessTime", _BOOL),

        ("dwRevocationFreshnessTime", _DWORD),

        ("pftCacheResync", ctypes.c_void_p),  # LPFILETIME

        ("pStrongSignPara", ctypes.c_void_p),  # PCCERT_STRONG_SIGN_PARA

        ("dwStrongSignFlags", _DWORD),

    )

# types of parameters of C functions used (required by pypy)

_crypt32.CertCreateCertificateContext.argtypes = [

    _DWORD,  # cert encoding

    ctypes.c_char_p,  # cert

    _DWORD,

]  # cert size

_crypt32.CertCreateCertificateContext.restype = _PCCERT_CONTEXT

_crypt32.CertGetCertificateChain.argtypes = [

    ctypes.c_void_p,  # HCERTCHAINENGINE

    _PCCERT_CONTEXT,

    ctypes.c_void_p,  # LPFILETIME

    ctypes.c_void_p,  # HCERTSTORE

    ctypes.c_void_p,  # PCERT_CHAIN_PARA

    _DWORD,

    ctypes.c_void_p,  # LPVOID

    ctypes.c_void_p,  # PCCERT_CHAIN_CONTEXT *

]

_crypt32.CertGetCertificateChain.restype = _BOOL

_crypt32.CertFreeCertificateContext.argtypes = [_PCCERT_CONTEXT]

_crypt32.CertFreeCertificateContext.restype = _BOOL

_kernel32.CreateFileA.argtypes = [

    _LPCSTR,

    _DWORD,

    _DWORD,

    ctypes.c_void_p,

    _DWORD,

    _DWORD,

    _HANDLE,

]

_kernel32.CreateFileA.restype = _HANDLE

_kernel32.GetFileInformationByHandle.argtypes = [_HANDLE, ctypes.c_void_p]

_kernel32.GetFileInformationByHandle.restype = _BOOL

_kernel32.CloseHandle.argtypes = [_HANDLE]

_kernel32.CloseHandle.restype = _BOOL

try:

    _kernel32.CreateHardLinkA.argtypes = [_LPCSTR, _LPCSTR, ctypes.c_void_p]

    _kernel32.CreateHardLinkA.restype = _BOOL

except AttributeError:

    pass

_kernel32.SetFileAttributesA.argtypes = [_LPCSTR, _DWORD]

_kernel32.SetFileAttributesA.restype = _BOOL

_DRIVE_UNKNOWN = 0

_DRIVE_NO_ROOT_DIR = 1

_DRIVE_REMOVABLE = 2

_DRIVE_FIXED = 3

_DRIVE_REMOTE = 4

_DRIVE_CDROM = 5

_DRIVE_RAMDISK = 6

_kernel32.GetDriveTypeA.argtypes = [_LPCSTR]

_kernel32.GetDriveTypeA.restype = _UINT

_kernel32.GetVolumeInformationA.argtypes = [

    _LPCSTR,

    ctypes.c_void_p,

    _DWORD,

    ctypes.c_void_p,

    ctypes.c_void_p,

    ctypes.c_void_p,

    ctypes.c_void_p,

    _DWORD,

]

_kernel32.GetVolumeInformationA.restype = _BOOL

_kernel32.GetVolumePathNameA.argtypes = [_LPCSTR, ctypes.c_void_p, _DWORD]

_kernel32.GetVolumePathNameA.restype = _BOOL

_kernel32.OpenProcess.argtypes = [_DWORD, _BOOL, _DWORD]

_kernel32.OpenProcess.restype = _HANDLE

_kernel32.GetExitCodeProcess.argtypes = [_HANDLE, ctypes.c_void_p]

_kernel32.GetExitCodeProcess.restype = _BOOL

_kernel32.GetLastError.argtypes = []

_kernel32.GetLastError.restype = _DWORD

_kernel32.GetModuleFileNameA.argtypes = [_HANDLE, ctypes.c_void_p, _DWORD]

_kernel32.GetModuleFileNameA.restype = _DWORD

_kernel32.CreateProcessA.argtypes = [

    _LPCSTR,

    _LPCSTR,

    ctypes.c_void_p,

    ctypes.c_void_p,

    _BOOL,

    _DWORD,

    ctypes.c_void_p,

    _LPCSTR,

    ctypes.c_void_p,

    ctypes.c_void_p,

]

_kernel32.CreateProcessA.restype = _BOOL

_kernel32.ExitProcess.argtypes = [_UINT]

_kernel32.ExitProcess.restype = None

_kernel32.GetCurrentProcessId.argtypes = []

_kernel32.GetCurrentProcessId.restype = _DWORD

_SIGNAL_HANDLER = ctypes.WINFUNCTYPE(_BOOL, _DWORD)

_kernel32.SetConsoleCtrlHandler.argtypes = [_SIGNAL_HANDLER, _BOOL]

_kernel32.SetConsoleCtrlHandler.restype = _BOOL

_kernel32.SetConsoleMode.argtypes = [_HANDLE, _DWORD]

_kernel32.SetConsoleMode.restype = _BOOL

_kernel32.GetConsoleMode.argtypes = [_HANDLE, ctypes.c_void_p]

_kernel32.GetConsoleMode.restype = _BOOL

_kernel32.GetStdHandle.argtypes = [_DWORD]

_kernel32.GetStdHandle.restype = _HANDLE

_kernel32.GetConsoleScreenBufferInfo.argtypes = [_HANDLE, ctypes.c_void_p]

_kernel32.GetConsoleScreenBufferInfo.restype = _BOOL

_advapi32.GetUserNameA.argtypes = [ctypes.c_void_p, ctypes.c_void_p]

_advapi32.GetUserNameA.restype = _BOOL

_user32.GetWindowThreadProcessId.argtypes = [_HANDLE, ctypes.c_void_p]

_user32.GetWindowThreadProcessId.restype = _DWORD

_user32.ShowWindow.argtypes = [_HANDLE, ctypes.c_int]

_user32.ShowWindow.restype = _BOOL

_WNDENUMPROC = ctypes.WINFUNCTYPE(_BOOL, _HWND, _LPARAM)

_user32.EnumWindows.argtypes = [_WNDENUMPROC, _LPARAM]

_user32.EnumWindows.restype = _BOOL

_kernel32.PeekNamedPipe.argtypes = [

    _HANDLE,

    ctypes.c_void_p,

    _DWORD,

    ctypes.c_void_p,

    ctypes.c_void_p,

    ctypes.c_void_p,

]

_kernel32.PeekNamedPipe.restype = _BOOL

def _raiseoserror(name):

    # Force the code to a signed int to avoid an 'int too large' error.

    # See https://bugs.python.org/issue28474

    code = _kernel32.GetLastError()

    if code > 0x7FFFFFFF:

        code -= 2 ** 32

    err = ctypes.WinError(code=code)

    raise OSError(

        err.errno, '%s: %s' % (encoding.strfromlocal(name), err.strerror)

    )

def _getfileinfo(name):

    fh = _kernel32.CreateFileA(

        name,

        0,

        _FILE_SHARE_READ | _FILE_SHARE_WRITE | _FILE_SHARE_DELETE,

        None,

        _OPEN_EXISTING,

        _FILE_FLAG_BACKUP_SEMANTICS,

        None,

    )

    if fh == _INVALID_HANDLE_VALUE:

        _raiseoserror(name)

    try:

        fi = _BY_HANDLE_FILE_INFORMATION()

        if not _kernel32.GetFileInformationByHandle(fh, ctypes.byref(fi)):

            _raiseoserror(name)

        return fi

    finally:

        _kernel32.CloseHandle(fh)

def checkcertificatechain(cert, build=True):

    '''Tests the given certificate to see if there is a complete chain to a

       trusted root certificate.  As a side effect, missing certificates are

       downloaded and installed unless ``build=False``.  True is returned if a

       chain to a trusted root exists (even if built on the fly), otherwise