12d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#!/usr/bin/python
22d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines
32d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesimport os, sys, subprocess, tempfile
42d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesfrom android_common import *
52d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines
62d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen HinesANDROID_TMPDIR = '/data/local/tmp/Output'
72d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines
82d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hineshere = os.path.abspath(os.path.dirname(sys.argv[0]))
92d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesdevice_binary = os.path.join(ANDROID_TMPDIR, os.path.basename(sys.argv[0]))
102d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines
112d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesdef build_env():
122d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines    args = []
132d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines    # Android linker ignores RPATH. Set LD_LIBRARY_PATH to Output dir.
142d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines    args.append('LD_LIBRARY_PATH=%s:%s' %
152d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines                (ANDROID_TMPDIR, os.environ.get('LD_LIBRARY_PATH', '')))
162d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines    for (key, value) in os.environ.items():
1786277eb844c4983c81de62d7c050e92fe7155788Stephen Hines        if key in ['ASAN_OPTIONS', 'ASAN_ACTIVATION_OPTIONS']:
182d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines            args.append('%s="%s"' % (key, value))
192d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines    return ' '.join(args)
202d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines
212d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesdevice_env = build_env()
222d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesdevice_args = ' '.join(sys.argv[1:]) # FIXME: escape?
232d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesdevice_stdout = device_binary + '.stdout'
242d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesdevice_stderr = device_binary + '.stderr'
252d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesdevice_exitcode = device_binary + '.exitcode'
266a211c5814e25d6745a5058cc0e499e5235d3821Stephen Hinesret = adb(['shell', 'cd %s && %s asanwrapper %s %s >%s 2>%s ; echo $? >%s' %
272d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines           (ANDROID_TMPDIR, device_env, device_binary, device_args,
282d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines            device_stdout, device_stderr, device_exitcode)])
292d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesif ret != 0:
302d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines    sys.exit(ret)
312d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines
322d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinessys.stdout.write(pull_from_device(device_stdout))
332d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinessys.stderr.write(pull_from_device(device_stderr))
342d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinessys.exit(int(pull_from_device(device_exitcode)))
35