We also need to catch OSError here.
[oota-llvm.git] / test / Bindings / Go / lit.local.cfg
1 import os
2 import pipes
3 import shlex
4 import sys
5
6 if not 'go' in config.root.llvm_bindings:
7     config.unsupported = True
8
9 def find_executable(executable, path=None):
10     if path is None:
11         path = os.environ['PATH']
12     paths = path.split(os.pathsep)
13     base, ext = os.path.splitext(executable)
14
15     if (sys.platform == 'win32' or os.name == 'os2') and (ext != '.exe'):
16         executable = executable + '.exe'
17
18     if not os.path.isfile(executable):
19         for p in paths:
20             f = os.path.join(p, executable)
21             if os.path.isfile(f):
22                 return f
23         return None
24     else:
25         return executable
26
27 # Resolve certain symlinks in the first word of compiler.
28 #
29 # This is a Go-specific hack. cgo and other Go tools check $CC and $CXX for the
30 # substring 'clang' to determine if the compiler is Clang. This won't work if
31 # $CC is cc and cc is a symlink pointing to clang, as it is on Darwin.
32 def fixup_compiler_path(compiler):
33     args = shlex.split(compiler)
34     path = find_executable(args[0])
35
36     try:
37         if path.endswith('/cc') and os.readlink(path) == 'clang':
38             args[0] = path[:len(path)-2] + 'clang'
39     except (AttributeError, OSError):
40         pass
41
42     try:
43         if path.endswith('/c++') and os.readlink(path) == 'clang++':
44             args[0] = path[:len(path)-3] + 'clang++'
45     except (AttributeError, OSError):
46         pass
47
48     return ' '.join([pipes.quote(arg) for arg in args])
49
50 config.environment['CC'] = fixup_compiler_path(config.host_cc)
51 config.environment['CXX'] = fixup_compiler_path(config.host_cxx)