Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 1 | import sys, os |
Eric Smith | 5ccfa29 | 2009-10-23 12:56:11 +0000 | [diff] [blame] | 2 | import contextlib |
Antoine Pitrou | 8c52027 | 2011-04-23 17:51:04 +0200 | [diff] [blame] | 3 | import subprocess |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 4 | |
| 5 | # find_library(name) returns the pathname of a library, or None. |
| 6 | if os.name == "nt": |
Thomas Heller | 3eaaeb4 | 2008-05-23 17:26:46 +0000 | [diff] [blame] | 7 | |
| 8 | def _get_build_version(): |
| 9 | """Return the version of MSVC that was used to build Python. |
| 10 | |
| 11 | For Python 2.3 and up, the version number is included in |
| 12 | sys.version. For earlier versions, assume the compiler is MSVC 6. |
| 13 | """ |
| 14 | # This function was copied from Lib/distutils/msvccompiler.py |
| 15 | prefix = "MSC v." |
| 16 | i = sys.version.find(prefix) |
| 17 | if i == -1: |
| 18 | return 6 |
| 19 | i = i + len(prefix) |
| 20 | s, rest = sys.version[i:].split(" ", 1) |
| 21 | majorVersion = int(s[:-2]) - 6 |
| 22 | minorVersion = int(s[2:3]) / 10.0 |
| 23 | # I don't think paths are affected by minor version in version 6 |
| 24 | if majorVersion == 6: |
| 25 | minorVersion = 0 |
| 26 | if majorVersion >= 6: |
| 27 | return majorVersion + minorVersion |
| 28 | # else we don't know what version of the compiler this is |
| 29 | return None |
| 30 | |
| 31 | def find_msvcrt(): |
| 32 | """Return the name of the VC runtime dll""" |
| 33 | version = _get_build_version() |
| 34 | if version is None: |
| 35 | # better be safe than sorry |
| 36 | return None |
| 37 | if version <= 6: |
| 38 | clibname = 'msvcrt' |
| 39 | else: |
| 40 | clibname = 'msvcr%d' % (version * 10) |
| 41 | |
| 42 | # If python was built with in debug mode |
Brett Cannon | cb66eb0 | 2012-05-11 12:58:42 -0400 | [diff] [blame] | 43 | import importlib.machinery |
| 44 | if '_d.pyd' in importlib.machinery.EXTENSION_SUFFIXES: |
Thomas Heller | 3eaaeb4 | 2008-05-23 17:26:46 +0000 | [diff] [blame] | 45 | clibname += 'd' |
| 46 | return clibname+'.dll' |
| 47 | |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 48 | def find_library(name): |
Thomas Heller | 3eaaeb4 | 2008-05-23 17:26:46 +0000 | [diff] [blame] | 49 | if name in ('c', 'm'): |
| 50 | return find_msvcrt() |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 51 | # See MSDN for the REAL search order. |
| 52 | for directory in os.environ['PATH'].split(os.pathsep): |
| 53 | fname = os.path.join(directory, name) |
Thomas Heller | 00cfc37 | 2009-05-05 19:04:40 +0000 | [diff] [blame] | 54 | if os.path.isfile(fname): |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 55 | return fname |
| 56 | if fname.lower().endswith(".dll"): |
| 57 | continue |
| 58 | fname = fname + ".dll" |
Thomas Heller | 00cfc37 | 2009-05-05 19:04:40 +0000 | [diff] [blame] | 59 | if os.path.isfile(fname): |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 60 | return fname |
| 61 | return None |
| 62 | |
| 63 | if os.name == "ce": |
| 64 | # search path according to MSDN: |
| 65 | # - absolute path specified by filename |
| 66 | # - The .exe launch directory |
| 67 | # - the Windows directory |
| 68 | # - ROM dll files (where are they?) |
| 69 | # - OEM specified search path: HKLM\Loader\SystemPath |
| 70 | def find_library(name): |
| 71 | return name |
| 72 | |
| 73 | if os.name == "posix" and sys.platform == "darwin": |
| 74 | from ctypes.macholib.dyld import dyld_find as _dyld_find |
| 75 | def find_library(name): |
| 76 | possible = ['lib%s.dylib' % name, |
| 77 | '%s.dylib' % name, |
| 78 | '%s.framework/%s' % (name, name)] |
| 79 | for name in possible: |
| 80 | try: |
| 81 | return _dyld_find(name) |
| 82 | except ValueError: |
| 83 | continue |
| 84 | return None |
| 85 | |
| 86 | elif os.name == "posix": |
| 87 | # Andreas Degert's find functions, using gcc, /sbin/ldconfig, objdump |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 88 | import re, tempfile, errno |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 89 | |
| 90 | def _findLib_gcc(name): |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 91 | expr = r'[^\(\)\s]*lib%s\.[^\(\)\s]*' % re.escape(name) |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 92 | fdout, ccout = tempfile.mkstemp() |
| 93 | os.close(fdout) |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 94 | cmd = 'if type gcc >/dev/null 2>&1; then CC=gcc; elif type cc >/dev/null 2>&1; then CC=cc;else exit 10; fi;' \ |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 95 | '$CC -Wl,-t -o ' + ccout + ' 2>&1 -l' + name |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 96 | try: |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 97 | f = os.popen(cmd) |
Guido van Rossum | f4d4f8b | 2007-12-12 20:26:00 +0000 | [diff] [blame] | 98 | try: |
| 99 | trace = f.read() |
| 100 | finally: |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 101 | rv = f.close() |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 102 | finally: |
| 103 | try: |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 104 | os.unlink(ccout) |
Giampaolo Rodola' | 0166a28 | 2013-02-12 15:14:17 +0100 | [diff] [blame^] | 105 | except FileNotFoundError: |
| 106 | pass |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 107 | if rv == 10: |
Benjamin Peterson | 17bf082 | 2009-01-10 22:37:11 +0000 | [diff] [blame] | 108 | raise OSError('gcc or cc command not found') |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 109 | res = re.search(expr, trace) |
| 110 | if not res: |
| 111 | return None |
| 112 | return res.group(0) |
| 113 | |
Thomas Wouters | 1b7f891 | 2007-09-19 03:06:30 +0000 | [diff] [blame] | 114 | |
| 115 | if sys.platform == "sunos5": |
| 116 | # use /usr/ccs/bin/dump on solaris |
| 117 | def _get_soname(f): |
| 118 | if not f: |
| 119 | return None |
| 120 | cmd = "/usr/ccs/bin/dump -Lpv 2>/dev/null " + f |
Eric Smith | 5ccfa29 | 2009-10-23 12:56:11 +0000 | [diff] [blame] | 121 | with contextlib.closing(os.popen(cmd)) as f: |
Guido van Rossum | f4d4f8b | 2007-12-12 20:26:00 +0000 | [diff] [blame] | 122 | data = f.read() |
Guido van Rossum | f4d4f8b | 2007-12-12 20:26:00 +0000 | [diff] [blame] | 123 | res = re.search(r'\[.*\]\sSONAME\s+([^\s]+)', data) |
Thomas Wouters | 1b7f891 | 2007-09-19 03:06:30 +0000 | [diff] [blame] | 124 | if not res: |
| 125 | return None |
| 126 | return res.group(1) |
| 127 | else: |
| 128 | def _get_soname(f): |
| 129 | # assuming GNU binutils / ELF |
| 130 | if not f: |
| 131 | return None |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 132 | cmd = 'if ! type objdump >/dev/null 2>&1; then exit 10; fi;' \ |
| 133 | "objdump -p -j .dynamic 2>/dev/null " + f |
| 134 | f = os.popen(cmd) |
| 135 | dump = f.read() |
| 136 | rv = f.close() |
| 137 | if rv == 10: |
Benjamin Peterson | 17bf082 | 2009-01-10 22:37:11 +0000 | [diff] [blame] | 138 | raise OSError('objdump command not found') |
Antoine Pitrou | b46004c | 2011-05-25 18:17:25 +0200 | [diff] [blame] | 139 | res = re.search(r'\sSONAME\s+([^\s]+)', dump) |
Thomas Wouters | 1b7f891 | 2007-09-19 03:06:30 +0000 | [diff] [blame] | 140 | if not res: |
| 141 | return None |
| 142 | return res.group(1) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 143 | |
Victor Stinner | e674747 | 2011-08-21 00:39:18 +0200 | [diff] [blame] | 144 | if sys.platform.startswith(("freebsd", "openbsd", "dragonfly")): |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 145 | |
| 146 | def _num_version(libname): |
| 147 | # "libxyz.so.MAJOR.MINOR" => [ MAJOR, MINOR ] |
| 148 | parts = libname.split(".") |
| 149 | nums = [] |
| 150 | try: |
| 151 | while parts: |
| 152 | nums.insert(0, int(parts.pop())) |
| 153 | except ValueError: |
| 154 | pass |
Christian Heimes | a37d4c6 | 2007-12-04 23:02:19 +0000 | [diff] [blame] | 155 | return nums or [ sys.maxsize ] |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 156 | |
| 157 | def find_library(name): |
| 158 | ename = re.escape(name) |
| 159 | expr = r':-l%s\.\S+ => \S*/(lib%s\.\S+)' % (ename, ename) |
Eric Smith | 5ccfa29 | 2009-10-23 12:56:11 +0000 | [diff] [blame] | 160 | with contextlib.closing(os.popen('/sbin/ldconfig -r 2>/dev/null')) as f: |
Guido van Rossum | f4d4f8b | 2007-12-12 20:26:00 +0000 | [diff] [blame] | 161 | data = f.read() |
Guido van Rossum | f4d4f8b | 2007-12-12 20:26:00 +0000 | [diff] [blame] | 162 | res = re.findall(expr, data) |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 163 | if not res: |
| 164 | return _get_soname(_findLib_gcc(name)) |
Raymond Hettinger | d4cb56d | 2008-01-30 02:55:10 +0000 | [diff] [blame] | 165 | res.sort(key=_num_version) |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 166 | return res[-1] |
| 167 | |
Benjamin Peterson | cfe3474 | 2013-02-03 19:25:11 -0500 | [diff] [blame] | 168 | elif sys.platform == "sunos5": |
| 169 | |
| 170 | def _findLib_crle(name, is64): |
| 171 | if not os.path.exists('/usr/bin/crle'): |
| 172 | return None |
| 173 | |
| 174 | if is64: |
| 175 | cmd = 'env LC_ALL=C /usr/bin/crle -64 2>/dev/null' |
| 176 | else: |
| 177 | cmd = 'env LC_ALL=C /usr/bin/crle 2>/dev/null' |
| 178 | |
| 179 | for line in os.popen(cmd).readlines(): |
| 180 | line = line.strip() |
| 181 | if line.startswith('Default Library Path (ELF):'): |
| 182 | paths = line.split()[4] |
| 183 | |
| 184 | if not paths: |
| 185 | return None |
| 186 | |
| 187 | for dir in paths.split(":"): |
| 188 | libfile = os.path.join(dir, "lib%s.so" % name) |
| 189 | if os.path.exists(libfile): |
| 190 | return libfile |
| 191 | |
| 192 | return None |
| 193 | |
| 194 | def find_library(name, is64 = False): |
| 195 | return _get_soname(_findLib_crle(name, is64) or _findLib_gcc(name)) |
| 196 | |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 197 | else: |
| 198 | |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 199 | def _findSoname_ldconfig(name): |
| 200 | import struct |
| 201 | if struct.calcsize('l') == 4: |
Larry Hastings | 605a62d | 2012-06-24 04:33:36 -0700 | [diff] [blame] | 202 | machine = os.uname().machine + '-32' |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 203 | else: |
Larry Hastings | 605a62d | 2012-06-24 04:33:36 -0700 | [diff] [blame] | 204 | machine = os.uname().machine + '-64' |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 205 | mach_map = { |
| 206 | 'x86_64-64': 'libc6,x86-64', |
| 207 | 'ppc64-64': 'libc6,64bit', |
| 208 | 'sparc64-64': 'libc6,64bit', |
| 209 | 's390x-64': 'libc6,64bit', |
| 210 | 'ia64-64': 'libc6,IA-64', |
| 211 | } |
| 212 | abi_type = mach_map.get(machine, 'libc6') |
| 213 | |
| 214 | # XXX assuming GLIBC's ldconfig (with option -p) |
Antoine Pitrou | 8c52027 | 2011-04-23 17:51:04 +0200 | [diff] [blame] | 215 | regex = os.fsencode( |
| 216 | '\s+(lib%s\.[^\s]+)\s+\(%s' % (re.escape(name), abi_type)) |
| 217 | try: |
| 218 | with subprocess.Popen(['/sbin/ldconfig', '-p'], |
| 219 | stdin=subprocess.DEVNULL, |
| 220 | stderr=subprocess.DEVNULL, |
| 221 | stdout=subprocess.PIPE, |
| 222 | env={'LC_ALL': 'C', 'LANG': 'C'}) as p: |
| 223 | res = re.search(regex, p.stdout.read()) |
| 224 | if res: |
| 225 | return os.fsdecode(res.group(1)) |
| 226 | except OSError: |
| 227 | pass |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 228 | |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 229 | def find_library(name): |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 230 | return _findSoname_ldconfig(name) or _get_soname(_findLib_gcc(name)) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 231 | |
| 232 | ################################################################ |
| 233 | # test code |
| 234 | |
| 235 | def test(): |
| 236 | from ctypes import cdll |
| 237 | if os.name == "nt": |
Guido van Rossum | be19ed7 | 2007-02-09 05:37:30 +0000 | [diff] [blame] | 238 | print(cdll.msvcrt) |
| 239 | print(cdll.load("msvcrt")) |
| 240 | print(find_library("msvcrt")) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 241 | |
| 242 | if os.name == "posix": |
| 243 | # find and load_version |
Guido van Rossum | be19ed7 | 2007-02-09 05:37:30 +0000 | [diff] [blame] | 244 | print(find_library("m")) |
| 245 | print(find_library("c")) |
| 246 | print(find_library("bz2")) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 247 | |
| 248 | # getattr |
| 249 | ## print cdll.m |
| 250 | ## print cdll.bz2 |
| 251 | |
| 252 | # load |
| 253 | if sys.platform == "darwin": |
Guido van Rossum | be19ed7 | 2007-02-09 05:37:30 +0000 | [diff] [blame] | 254 | print(cdll.LoadLibrary("libm.dylib")) |
| 255 | print(cdll.LoadLibrary("libcrypto.dylib")) |
| 256 | print(cdll.LoadLibrary("libSystem.dylib")) |
| 257 | print(cdll.LoadLibrary("System.framework/System")) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 258 | else: |
Guido van Rossum | be19ed7 | 2007-02-09 05:37:30 +0000 | [diff] [blame] | 259 | print(cdll.LoadLibrary("libm.so")) |
| 260 | print(cdll.LoadLibrary("libcrypt.so")) |
| 261 | print(find_library("crypt")) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 262 | |
| 263 | if __name__ == "__main__": |
| 264 | test() |