aboutsummaryrefslogtreecommitdiff
path: root/lldb/test/Shell/DAP/TestSTDINConsole.test
blob: 0d6d25400260e9792dee28c39db79449d5930d44 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# REQUIRES: python, system-windows
# RUN: %python %s lldb-dap > %t.out 2>&1
# RUN: FileCheck %s --check-prefix=ERROR --allow-empty < %t.out

# ERROR-NOT: DAP session error:

# Test that we can successfully start the dap server from the console on Windows.

import sys
import subprocess
import os

lldb_dap_path = sys.argv[1]
if lldb_dap_path.startswith('%'):
    lldb_dap_path = lldb_dap_path[1:]
lldb_dap_path = os.path.normpath(lldb_dap_path)

import ctypes
from ctypes import wintypes
import msvcrt

GENERIC_READ = 0x80000000
GENERIC_WRITE = 0x40000000
OPEN_EXISTING = 3
FILE_SHARE_READ = 0x00000001
FILE_SHARE_WRITE = 0x00000002

kernel32 = ctypes.WinDLL('kernel32', use_last_error=True)

conin_handle = kernel32.CreateFileW(
    "CONIN$",
    GENERIC_READ | GENERIC_WRITE,
    FILE_SHARE_READ | FILE_SHARE_WRITE,
    None,
    OPEN_EXISTING,
    0,
    None
)

if conin_handle == -1:
    print("Failed to open CONIN$", file=sys.stderr)
    sys.exit(1)

conin_fd = msvcrt.open_osfhandle(conin_handle, os.O_RDONLY)

proc = subprocess.Popen(
    [lldb_dap_path],
    stdin=conin_fd,
    stdout=subprocess.PIPE,
    stderr=subprocess.STDOUT
)

os.close(conin_fd)

try:
    output, _ = proc.communicate(timeout=2)
except subprocess.TimeoutExpired:
    proc.kill()
    output, _ = proc.communicate()

sys.stdout.buffer.write(output)
sys.stdout.flush()