forked from OSchip/llvm-project
72 lines
2.2 KiB
Plaintext
72 lines
2.2 KiB
Plaintext
## On Windows co-operative applications can be expected to open LLD's output
|
|
## with FILE_SHARE_DELETE included in the sharing mode. This allows us to link
|
|
## over the top of an existing file even if it is in use by another application.
|
|
|
|
# REQUIRES: system-windows, x86
|
|
# RUN: echo '.globl _start; _start:' > %t.s
|
|
# RUN: llvm-mc -filetype=obj -triple=x86_64-unknown-unknown %t.s -o %t.o
|
|
|
|
## FILE_SHARE_READ = 1
|
|
## FILE_SHARE_WRITE = 2
|
|
## FILE_SHARE_DELETE = 4
|
|
|
|
# RUN: %python %s %t.o 7
|
|
# RUN: not %python %s %t.o 3 2>&1 | FileCheck %s
|
|
# CHECK: error: failed to write to the output file
|
|
|
|
import contextlib
|
|
import ctypes
|
|
from ctypes import wintypes as w
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
import platform
|
|
import sys
|
|
import time
|
|
|
|
object_file = sys.argv[1]
|
|
share_flags = int(sys.argv[2])
|
|
|
|
@contextlib.contextmanager
|
|
def open_with_share_flags(filename, share_flags):
|
|
GENERIC_READ = 0x80000000
|
|
FILE_ATTRIBUTE_NORMAL = 0x80
|
|
OPEN_EXISTING = 0x3
|
|
INVALID_HANDLE_VALUE = w.HANDLE(-1).value
|
|
|
|
CreateFileA = ctypes.windll.kernel32.CreateFileA
|
|
CreateFileA.restype = w.HANDLE
|
|
h = CreateFileA(filename.encode('mbcs'), GENERIC_READ, share_flags,
|
|
None, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, None)
|
|
|
|
assert h != INVALID_HANDLE_VALUE, 'Failed to open ' + filename
|
|
try:
|
|
yield
|
|
finally:
|
|
ctypes.windll.kernel32.CloseHandle(h)
|
|
|
|
## Ensure we have an empty directory for the output.
|
|
outdir = os.path.basename(__file__) + '.dir'
|
|
if os.path.exists(outdir):
|
|
shutil.rmtree(outdir)
|
|
os.makedirs(outdir)
|
|
|
|
## Link on top of an open file.
|
|
elf = os.path.join(outdir, 'output_file.elf')
|
|
open(elf, 'wb').close()
|
|
with open_with_share_flags(elf, share_flags):
|
|
subprocess.check_call(['ld.lld.exe', object_file, '-o', elf])
|
|
|
|
## Check the linker wrote the output file.
|
|
with open(elf, 'rb') as f:
|
|
assert f.read(4) == b'\x7fELF', "linker did not write output file correctly"
|
|
|
|
## Check no temp files are left around.
|
|
## It might take a while for Windows to remove them, so loop.
|
|
deleted = lambda: len(os.listdir(outdir)) == 1
|
|
for _ in range(10):
|
|
if not deleted():
|
|
time.sleep (1)
|
|
|
|
assert deleted(), "temp file(s) not deleted after grace period"
|