OLD | NEW |
1 #!/usr/bin/env python2 | 1 #!/usr/bin/env python2 |
2 | 2 |
3 import argparse | 3 import argparse |
| 4 import errno |
4 import os | 5 import os |
| 6 import shutil |
5 import tempfile | 7 import tempfile |
6 from utils import shellcmd | 8 from utils import shellcmd |
7 from utils import FindBaseNaCl | 9 from utils import FindBaseNaCl |
8 | 10 |
9 if __name__ == '__main__': | 11 if __name__ == '__main__': |
10 argparser = argparse.ArgumentParser() | 12 argparser = argparse.ArgumentParser() |
11 argparser.add_argument('cfile', nargs='+', type=str, | 13 argparser.add_argument('cfile', nargs='+', type=str, |
12 help='C file(s) to convert') | 14 help='C file(s) to convert') |
13 argparser.add_argument('--dir', nargs='?', type=str, default='.', | 15 argparser.add_argument('--dir', nargs='?', type=str, default='.', |
14 help='Output directory. Default "%(default)s".') | 16 help='Output directory. Default "%(default)s".') |
15 argparser.add_argument('--disable-verify', action='store_true') | 17 argparser.add_argument('--disable-verify', action='store_true') |
16 args = argparser.parse_args() | 18 args = argparser.parse_args() |
17 | 19 |
18 nacl_root = FindBaseNaCl() | 20 nacl_root = FindBaseNaCl() |
19 # Prepend bin to $PATH. | 21 # Prepend bin to $PATH. |
20 os.environ['PATH'] = ( | 22 os.environ['PATH'] = ( |
21 nacl_root + '/toolchain/linux_x86/pnacl_newlib/bin' + os.pathsep + | 23 nacl_root + '/toolchain/linux_x86/pnacl_newlib/bin' + os.pathsep + |
22 os.pathsep + os.environ['PATH']) | 24 os.pathsep + os.environ['PATH']) |
23 | 25 |
24 tempdir = tempfile.mkdtemp() | 26 try: |
| 27 tempdir = tempfile.mkdtemp() |
25 | 28 |
26 for cname in args.cfile: | 29 for cname in args.cfile: |
27 basename = os.path.splitext(cname)[0] | 30 basename = os.path.splitext(cname)[0] |
28 llname = os.path.join(tempdir, basename + '.ll') | 31 llname = os.path.join(tempdir, basename + '.ll') |
29 pnaclname = basename + '.pnacl.ll' | 32 pnaclname = basename + '.pnacl.ll' |
30 pnaclname = os.path.join(args.dir, pnaclname) | 33 pnaclname = os.path.join(args.dir, pnaclname) |
31 | 34 |
32 shellcmd('pnacl-clang -O2 -c {0} -o {1}'.format(cname, llname)) | 35 shellcmd('pnacl-clang -O2 -c {0} -o {1}'.format(cname, llname)) |
33 shellcmd('pnacl-opt ' + | 36 shellcmd('pnacl-opt ' + |
34 '-pnacl-abi-simplify-preopt -pnacl-abi-simplify-postopt' + | 37 '-pnacl-abi-simplify-preopt -pnacl-abi-simplify-postopt' + |
35 ('' if args.disable_verify else | 38 ('' if args.disable_verify else |
36 ' -verify-pnaclabi-module -verify-pnaclabi-functions') + | 39 ' -verify-pnaclabi-module -verify-pnaclabi-functions') + |
37 ' -pnaclabi-allow-debug-metadata' | 40 ' -pnaclabi-allow-debug-metadata' |
38 ' {0} -S -o {1}'.format(llname, pnaclname)) | 41 ' {0} -S -o {1}'.format(llname, pnaclname)) |
| 42 finally: |
| 43 try: |
| 44 shutil.rmtree(tempdir) |
| 45 except OSError as exc: |
| 46 if exc.errno != errno.ENOENT: # ENOENT - no such file or directory |
| 47 raise # re-raise exception |
OLD | NEW |