6ea78398aa
The infrastructure runs everything already in Python3, so this is mostly a clean-up. For MB, a python2 holdover was removed and new lint errors were fixed. The renames were automated with: git grep -e "/usr/bin/python$" | cut -d':' -f1 | xargs sed -i 's/#!\/usr\/bin\/python$/#!\/usr\/bin\/python3/1' and git grep -e "/usr/bin/env python$" | cut -d':' -f1 | xargs sed -i 's/#!\/usr\/bin\/env python$/#!\/usr\/bin\/env python3/1' Bug: v8:13148 Change-Id: If4f3c7635e72fa134798d55314ac1aa92ddd01bf Reviewed-on: https://chromium-review.googlesource.com/c/v8/v8/+/3811499 Reviewed-by: Liviu Rau <liviurau@google.com> Commit-Queue: Michael Achenbach <machenbach@chromium.org> Cr-Commit-Position: refs/heads/main@{#82231}
61 lines
1.9 KiB
Python
Executable File
61 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# Copyright 2019 the V8 project authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be found
|
|
# in the LICENSE file.
|
|
|
|
import argparse
|
|
import io
|
|
import sys
|
|
|
|
from wasm import *
|
|
|
|
FUNCTION_SECTION_ID = 3
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(\
|
|
description="Inject compilation hints into a Wasm module.")
|
|
parser.add_argument("-i", "--in-wasm-file", \
|
|
type=str, \
|
|
help="original wasm module")
|
|
parser.add_argument("-o", "--out-wasm-file", \
|
|
type=str, \
|
|
help="wasm module with injected hints")
|
|
parser.add_argument("-x", "--hints-file", \
|
|
type=str, required=True, \
|
|
help="binary hints file to be injected as a custom section " + \
|
|
"'compilationHints'")
|
|
return parser.parse_args()
|
|
|
|
if __name__ == "__main__":
|
|
args = parse_args()
|
|
in_wasm_file = args.in_wasm_file if args.in_wasm_file else sys.stdin.fileno()
|
|
out_wasm_file = args.out_wasm_file if args.out_wasm_file else sys.stdout.fileno()
|
|
hints_bs = open(args.hints_file, "rb").read()
|
|
with io.open(in_wasm_file, "rb") as fin:
|
|
with io.open(out_wasm_file, "wb") as fout:
|
|
magic_number, bs = read_magic_number(fin);
|
|
fout.write(bs)
|
|
version, bs = read_version(fin);
|
|
fout.write(bs)
|
|
num_declared_functions = None
|
|
while True:
|
|
id, bs = read_varuintN(fin)
|
|
fout.write(bs)
|
|
if id == None:
|
|
break
|
|
payload_length, bs = read_varuintN(fin)
|
|
fout.write(bs)
|
|
|
|
# Peek into function section for upcoming validity check.
|
|
if id == FUNCTION_SECTION_ID:
|
|
num_declared_functions, bs = peek_varuintN(fin)
|
|
|
|
bs = fin.read(payload_length)
|
|
fout.write(bs)
|
|
|
|
# Instert hint section after function section.
|
|
if id == FUNCTION_SECTION_ID:
|
|
assert len(hints_bs) == num_declared_functions, "unexpected number of hints"
|
|
write_compilation_hints_section(fout, hints_bs)
|