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
|
# RUN: rm -rf %t && mkdir -p %t && cd %t
# RUN: llvm-mc -triple=x86_64-pc-windows-msvc -filetype=obj -o a.obj %S/Inputs/a.s
# RUN: llvm-mc -triple=x86_64-pc-windows-msvc -filetype=obj -o b.obj %S/Inputs/b.s
# RUN: llvm-lib /out:xfghashmap.lib b.obj a.obj
## Replace a section in the library file with /<XFGHASHMAP>/ emulating
## a library from the Windows SDK for Windows 11.
# RUN: %python %s xfghashmap.lib b.obj/
## This should print the /<XFGHASHMAP>/ section as well as an .obj one.
# RUN: llvm-lib /list %t/xfghashmap.lib | FileCheck %s
# CHECK: /<XFGHASHMAP>/
# CHECK-NOT: b.obj
# CHECK: a.obj
import sys
if len(sys.argv) < 3:
print("Use: python3 xfghashmap-list.test <LIBRARY_FILE> <TEMPLATE>")
exit(1)
template = bytes(sys.argv[2], 'utf-8')
xfghashmap = b'/<XFGHASHMAP>/'
data = None
with open(sys.argv[1], "rb") as inp:
data = inp.read()
with open(sys.argv[1], "wb") as outp:
pos = data.find(template)
outp.write(data[:pos])
outp.write(xfghashmap)
outp.write(data[pos + len(xfghashmap):])
|
