REDROOM

Navigation

Breadcrumbs

/ / lib64 / python3.12 / asyncio

Create

Create File | Create Folder | Upload Files | Run Command

Preview: windows_utils.py Size: 4.94 KB

/lib64/python3.12/asyncio/windows_utils.py

"""Various Windows specific bits and pieces."""

import sys

if sys.platform != 'win32':  # pragma: no cover
    raise ImportError('win32 only')

import _winapi
import itertools
import msvcrt
import os
import subprocess
import tempfile
import warnings


__all__ = 'pipe', 'Popen', 'PIPE', 'PipeHandle'


# Constants/globals


BUFSIZE = 8192
PIPE = subprocess.PIPE
STDOUT = subprocess.STDOUT
_mmap_counter = itertools.count()


# Replacement for os.pipe() using handles instead of fds


def pipe(*, duplex=False, overlapped=(True, True), bufsize=BUFSIZE):
    """Like os.pipe() but with overlapped support and using handles not fds."""
    address = tempfile.mktemp(
        prefix=r'\\.\pipe\python-pipe-{:d}-{:d}-'.format(
            os.getpid(), next(_mmap_counter)))

    if duplex:
        openmode = _winapi.PIPE_ACCESS_DUPLEX
        access = _winapi.GENERIC_READ | _winapi.GENERIC_WRITE
        obsize, ibsize = bufsize, bufsize
    else:
        openmode = _winapi.PIPE_ACCESS_INBOUND
        access = _winapi.GENERIC_WRITE
        obsize, ibsize = 0, bufsize

    openmode |= _winapi.FILE_FLAG_FIRST_PIPE_INSTANCE

    if overlapped[0]:
        openmode |= _winapi.FILE_FLAG_OVERLAPPED

    if overlapped[1]:
        flags_and_attribs = _winapi.FILE_FLAG_OVERLAPPED
    else:
        flags_and_attribs = 0

    h1 = h2 = None
    try:
        h1 = _winapi.CreateNamedPipe(
            address, openmode, _winapi.PIPE_WAIT,
            1, obsize, ibsize, _winapi.NMPWAIT_WAIT_FOREVER, _winapi.NULL)

        h2 = _winapi.CreateFile(
            address, access, 0, _winapi.NULL, _winapi.OPEN_EXISTING,
            flags_and_attribs, _winapi.NULL)

        ov = _winapi.ConnectNamedPipe(h1, overlapped=True)
        ov.GetOverlappedResult(True)
        return h1, h2
    except:
        if h1 is not None:
            _winapi.CloseHandle(h1)
        if h2 is not None:
            _winapi.CloseHandle(h2)
        raise


# Wrapper for a pipe handle


class PipeHandle:
    """Wrapper for an overlapped pipe handle which is vaguely file-object like.

    The IOCP event loop can use these instead of socket objects.
    """
    def __init__(self, handle):
        self._handle = handle

    def __repr__(self):
        if self._handle is not None:
            handle = f'handle={self._handle!r}'
        else:
            handle = 'closed'
        return f'<{self.__class__.__name__} {handle}>'

    @property
    def handle(self):
        return self._handle

    def fileno(self):
        if self._handle is None:
            raise ValueError("I/O operation on closed pipe")
        return self._handle

    def close(self, *, CloseHandle=_winapi.CloseHandle):
        if self._handle is not None:
            CloseHandle(self._handle)
            self._handle = None

    def __del__(self, _warn=warnings.warn):
        if self._handle is not None:
            _warn(f"unclosed {self!r}", ResourceWarning, source=self)
            self.close()

    def __enter__(self):
        return self

    def __exit__(self, t, v, tb):
        self.close()


# Replacement for subprocess.Popen using overlapped pipe handles


class Popen(subprocess.Popen):
    """Replacement for subprocess.Popen using overlapped pipe handles.

    The stdin, stdout, stderr are None or instances of PipeHandle.
    """
    def __init__(self, args, stdin=None, stdout=None, stderr=None, **kwds):
        assert not kwds.get('universal_newlines')
        assert kwds.get('bufsize', 0) == 0
        stdin_rfd = stdout_wfd = stderr_wfd = None
        stdin_wh = stdout_rh = stderr_rh = None
        if stdin == PIPE:
            stdin_rh, stdin_wh = pipe(overlapped=(False, True), duplex=True)
            stdin_rfd = msvcrt.open_osfhandle(stdin_rh, os.O_RDONLY)
        else:
            stdin_rfd = stdin
        if stdout == PIPE:
            stdout_rh, stdout_wh = pipe(overlapped=(True, False))
            stdout_wfd = msvcrt.open_osfhandle(stdout_wh, 0)
        else:
            stdout_wfd = stdout
        if stderr == PIPE:
            stderr_rh, stderr_wh = pipe(overlapped=(True, False))
            stderr_wfd = msvcrt.open_osfhandle(stderr_wh, 0)
        elif stderr == STDOUT:
            stderr_wfd = stdout_wfd
        else:
            stderr_wfd = stderr
        try:
            super().__init__(args, stdin=stdin_rfd, stdout=stdout_wfd,
                             stderr=stderr_wfd, **kwds)
        except:
            for h in (stdin_wh, stdout_rh, stderr_rh):
                if h is not None:
                    _winapi.CloseHandle(h)
            raise
        else:
            if stdin_wh is not None:
                self.stdin = PipeHandle(stdin_wh)
            if stdout_rh is not None:
                self.stdout = PipeHandle(stdout_rh)
            if stderr_rh is not None:
                self.stderr = PipeHandle(stderr_rh)
        finally:
            if stdin == PIPE:
                os.close(stdin_rfd)
            if stdout == PIPE:
                os.close(stdout_wfd)
            if stderr == PIPE:
                os.close(stderr_wfd)

Directory Contents

Dirs: 1 × Files: 33

Sort:

Enable drag-to-move

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2026-01-08 23:02:23	Edit Download recursive recursive
base_events.py	76.73 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
base_futures.py	1.93 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
base_subprocess.py	8.66 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
base_tasks.py	2.61 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
constants.py	1.38 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
coroutines.py	3.26 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
events.py	28.65 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
exceptions.py	1.71 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
format_helpers.py	2.35 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
futures.py	14.00 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
locks.py	18.55 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
log.py	124 B	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
mixins.py	481 B	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
proactor_events.py	32.71 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
protocols.py	6.79 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
queues.py	7.79 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
runners.py	7.06 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
selector_events.py	47.20 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
sslproto.py	31.15 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
staggered.py	6.91 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
streams.py	26.97 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
subprocess.py	7.56 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
taskgroups.py	9.33 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
tasks.py	36.49 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
threads.py	790 B	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
timeouts.py	5.20 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
transports.py	10.47 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
trsock.py	2.42 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
unix_events.py	51.88 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
windows_events.py	31.82 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
windows_utils.py	4.94 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
__init__.py	1.19 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive
__main__.py	3.41 KB	lrw-r--r--	2025-10-09 11:07:00	Edit Download recursive recursive