llvm/test/tools/llvm-lib/xfghashmap-list.test

   1 # RUN: rm -rf %t && mkdir -p %t && cd %t
   2 # RUN: llvm-mc -triple=x86_64-pc-windows-msvc -filetype=obj -o a.obj %S/Inputs/a.s
   3 # RUN: llvm-mc -triple=x86_64-pc-windows-msvc -filetype=obj -o b.obj %S/Inputs/b.s
   4 # RUN: llvm-lib /out:xfghashmap.lib b.obj a.obj
   5
   6 ## Replace a section in the library file with /<XFGHASHMAP>/ emulating
   7 ## a library from the Windows SDK for Windows 11.
   8 # RUN: %python %s xfghashmap.lib b.obj/
   9
  10 ## This should print the /<XFGHASHMAP>/ section as well as an .obj one.
  11 # RUN: llvm-lib /list %t/xfghashmap.lib | FileCheck %s
  12
  13 # CHECK: /<XFGHASHMAP>/
  14 # CHECK-NOT: b.obj
  15 # CHECK: a.obj
  16
  17 import sys
  18
  19 if len(sys.argv) < 3:
  20   print("Use: python3 xfghashmap-list.test <LIBRARY_FILE> <TEMPLATE>")
  21   exit(1)
  22
  23 template = bytes(sys.argv[2], 'utf-8')
  24 xfghashmap = b'/<XFGHASHMAP>/'
  25
  26 data = None
  27 with open(sys.argv[1], "rb") as inp:
  28   data = inp.read()
  29 with open(sys.argv[1], "wb") as outp:
  30   pos = data.find(template)
  31   outp.write(data[:pos])
  32   outp.write(xfghashmap)
  33   outp.write(data[pos + len(xfghashmap):])