OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2013 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. |
| 5 |
| 6 """ |
| 7 Tries to compile given code, produces different output depending on success. |
| 8 |
| 9 This is similar to checks done by ./configure scripts. |
| 10 """ |
| 11 |
| 12 |
| 13 import optparse |
| 14 import os |
| 15 import shutil |
| 16 import subprocess |
| 17 import sys |
| 18 import tempfile |
| 19 |
| 20 |
| 21 def DoMain(argv): |
| 22 parser = optparse.OptionParser() |
| 23 parser.add_option('--code') |
| 24 parser.add_option('--on-success', default='') |
| 25 parser.add_option('--on-failure', default='') |
| 26 |
| 27 options, args = parser.parse_args(argv) |
| 28 |
| 29 if not options.code: |
| 30 parser.error('Missing required --code switch.') |
| 31 |
| 32 cxx = os.environ.get('CXX', 'g++') |
| 33 |
| 34 tmpdir = tempfile.mkdtemp() |
| 35 try: |
| 36 cxx_path = os.path.join(tmpdir, 'test.cc') |
| 37 with open(cxx_path, 'w') as f: |
| 38 f.write(options.code.decode('string-escape')) |
| 39 |
| 40 o_path = os.path.join(tmpdir, 'test.o') |
| 41 |
| 42 cxx_popen = subprocess.Popen([cxx, cxx_path, '-o', o_path, '-c'], |
| 43 stdout=subprocess.PIPE, |
| 44 stderr=subprocess.PIPE) |
| 45 cxx_stdout, cxx_stderr = cxx_popen.communicate() |
| 46 if cxx_popen.returncode == 0: |
| 47 print options.on_success |
| 48 else: |
| 49 print options.on_failure |
| 50 finally: |
| 51 shutil.rmtree(tmpdir) |
| 52 |
| 53 return 0 |
| 54 |
| 55 |
| 56 if __name__ == '__main__': |
| 57 sys.exit(DoMain(sys.argv[1:])) |
OLD | NEW |