File: generate_wrap_sh.py

package info (click to toggle)
chromium 139.0.7258.127-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 6,122,156 kB
  • sloc: cpp: 35,100,771; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (76 lines) | stat: -rwxr-xr-x 2,184 bytes parent folder | download | duplicates (3)
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#! /usr/bin/env python3
# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import argparse
import pathlib


def _generate(arch, hwasan, asan, env):
  ret = """\
#!/system/bin/sh
"""

  if hwasan:
    # https://developer.android.com/ndk/guides/hwasan#wrapsh
    ret += """
# import options file
_HWASAN_OPTIONS=$(cat /data/local/tmp/hwasan.options 2> /dev/null || true)

log -t cr_wrap.sh -- "Launching with HWASAN enabled."
"""
    env['HWASAN_OPTIONS'] = '$_HWASAN_OPTIONS'
    env['LD_HWASAN'] = '1'

  if asan:
    # https://github.com/google/sanitizers/wiki/AddressSanitizerOnAndroid/01f8df1ac1a447a8475cdfcb03e8b13140042dbd#running-with-wrapsh-recommended
    ret += f"""
HERE="$(cd "$(dirname "$0")" && pwd)"
# Options suggested by wiki docs:
_ASAN_OPTIONS="log_to_syslog=false,allow_user_segv_handler=1"
# Chromium-specific option (supposedly for graphics drivers):
_ASAN_OPTIONS="$_ASAN_OPTIONS,strict_memcmp=0,use_sigaltstack=1"
_LD_PRELOAD="$HERE/libclang_rt.asan-{arch}-android.so"

log -t cr_wrap.sh -- "Launching with ASAN enabled."
"""
    env['LD_PRELOAD'] = '$_LD_PRELOAD'
    env['ASAN_OPTIONS'] = '$_ASAN_OPTIONS'

  ret += 'log -t cr_wrap.sh -- "Command: $0 $@"\n'

  # Cannot set env vars before calling "log" commands, because they would
  # affect the "log" executable.
  for key, value in env.items():
    ret += f'log -t cr_wrap.sh -- "{key}={value}"\n'

  for key, value in env.items():
    ret += f'export {key}="{value}"\n'

  ret += """
exec "$@"
"""
  return ret


def main():
  parser = argparse.ArgumentParser()
  parser.add_argument('--output', required=True)
  parser.add_argument('--arch', required=True)
  parser.add_argument('--hwasan', action='store_true', default=False)
  parser.add_argument('--asan', action='store_true', default=False)
  parser.add_argument('--env')
  args = parser.parse_args()

  env = {}
  if args.env:
    for prop in args.env.split():
      key, value = prop.split('=', 1)
      env[key] = value
  pathlib.Path(args.output).write_text(
      _generate(args.arch, args.hwasan, args.asan, env))


if __name__ == '__main__':
  main()