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 |
Victor Stinner | 7fa767e | 2014-03-20 09:16:38 +0100 | [diff] [blame^] | 88 | import re, tempfile |
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;' \ |
doko@ubuntu.com | ef53558 | 2013-05-15 18:02:13 +0200 | [diff] [blame] | 95 | 'LANG=C LC_ALL=C $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) |
Nick Coghlan | 1889623 | 2013-11-24 12:53:50 +1000 | [diff] [blame] | 135 | try: |
| 136 | dump = f.read() |
| 137 | finally: |
| 138 | rv = f.close() |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 139 | if rv == 10: |
Benjamin Peterson | 17bf082 | 2009-01-10 22:37:11 +0000 | [diff] [blame] | 140 | raise OSError('objdump command not found') |
Antoine Pitrou | b46004c | 2011-05-25 18:17:25 +0200 | [diff] [blame] | 141 | res = re.search(r'\sSONAME\s+([^\s]+)', dump) |
Thomas Wouters | 1b7f891 | 2007-09-19 03:06:30 +0000 | [diff] [blame] | 142 | if not res: |
| 143 | return None |
| 144 | return res.group(1) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 145 | |
Victor Stinner | e674747 | 2011-08-21 00:39:18 +0200 | [diff] [blame] | 146 | if sys.platform.startswith(("freebsd", "openbsd", "dragonfly")): |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 147 | |
| 148 | def _num_version(libname): |
| 149 | # "libxyz.so.MAJOR.MINOR" => [ MAJOR, MINOR ] |
| 150 | parts = libname.split(".") |
| 151 | nums = [] |
| 152 | try: |
| 153 | while parts: |
| 154 | nums.insert(0, int(parts.pop())) |
| 155 | except ValueError: |
| 156 | pass |
Christian Heimes | a37d4c6 | 2007-12-04 23:02:19 +0000 | [diff] [blame] | 157 | return nums or [ sys.maxsize ] |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 158 | |
| 159 | def find_library(name): |
| 160 | ename = re.escape(name) |
| 161 | expr = r':-l%s\.\S+ => \S*/(lib%s\.\S+)' % (ename, ename) |
Eric Smith | 5ccfa29 | 2009-10-23 12:56:11 +0000 | [diff] [blame] | 162 | 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] | 163 | data = f.read() |
Guido van Rossum | f4d4f8b | 2007-12-12 20:26:00 +0000 | [diff] [blame] | 164 | res = re.findall(expr, data) |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 165 | if not res: |
| 166 | return _get_soname(_findLib_gcc(name)) |
Raymond Hettinger | d4cb56d | 2008-01-30 02:55:10 +0000 | [diff] [blame] | 167 | res.sort(key=_num_version) |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 168 | return res[-1] |
| 169 | |
Benjamin Peterson | cfe3474 | 2013-02-03 19:25:11 -0500 | [diff] [blame] | 170 | elif sys.platform == "sunos5": |
| 171 | |
| 172 | def _findLib_crle(name, is64): |
| 173 | if not os.path.exists('/usr/bin/crle'): |
| 174 | return None |
| 175 | |
| 176 | if is64: |
| 177 | cmd = 'env LC_ALL=C /usr/bin/crle -64 2>/dev/null' |
| 178 | else: |
| 179 | cmd = 'env LC_ALL=C /usr/bin/crle 2>/dev/null' |
| 180 | |
Nick Coghlan | 1889623 | 2013-11-24 12:53:50 +1000 | [diff] [blame] | 181 | with contextlib.closing(os.popen(cmd)) as f: |
| 182 | for line in f.readlines(): |
| 183 | line = line.strip() |
| 184 | if line.startswith('Default Library Path (ELF):'): |
| 185 | paths = line.split()[4] |
Benjamin Peterson | cfe3474 | 2013-02-03 19:25:11 -0500 | [diff] [blame] | 186 | |
| 187 | if not paths: |
| 188 | return None |
| 189 | |
| 190 | for dir in paths.split(":"): |
| 191 | libfile = os.path.join(dir, "lib%s.so" % name) |
| 192 | if os.path.exists(libfile): |
| 193 | return libfile |
| 194 | |
| 195 | return None |
| 196 | |
| 197 | def find_library(name, is64 = False): |
| 198 | return _get_soname(_findLib_crle(name, is64) or _findLib_gcc(name)) |
| 199 | |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 200 | else: |
| 201 | |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 202 | def _findSoname_ldconfig(name): |
| 203 | import struct |
| 204 | if struct.calcsize('l') == 4: |
Larry Hastings | 605a62d | 2012-06-24 04:33:36 -0700 | [diff] [blame] | 205 | machine = os.uname().machine + '-32' |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 206 | else: |
Larry Hastings | 605a62d | 2012-06-24 04:33:36 -0700 | [diff] [blame] | 207 | machine = os.uname().machine + '-64' |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 208 | mach_map = { |
| 209 | 'x86_64-64': 'libc6,x86-64', |
| 210 | 'ppc64-64': 'libc6,64bit', |
| 211 | 'sparc64-64': 'libc6,64bit', |
| 212 | 's390x-64': 'libc6,64bit', |
| 213 | 'ia64-64': 'libc6,IA-64', |
| 214 | } |
| 215 | abi_type = mach_map.get(machine, 'libc6') |
| 216 | |
| 217 | # XXX assuming GLIBC's ldconfig (with option -p) |
Antoine Pitrou | 8c52027 | 2011-04-23 17:51:04 +0200 | [diff] [blame] | 218 | regex = os.fsencode( |
| 219 | '\s+(lib%s\.[^\s]+)\s+\(%s' % (re.escape(name), abi_type)) |
| 220 | try: |
| 221 | with subprocess.Popen(['/sbin/ldconfig', '-p'], |
| 222 | stdin=subprocess.DEVNULL, |
| 223 | stderr=subprocess.DEVNULL, |
| 224 | stdout=subprocess.PIPE, |
| 225 | env={'LC_ALL': 'C', 'LANG': 'C'}) as p: |
| 226 | res = re.search(regex, p.stdout.read()) |
| 227 | if res: |
| 228 | return os.fsdecode(res.group(1)) |
| 229 | except OSError: |
| 230 | pass |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 231 | |
Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 232 | def find_library(name): |
Matthias Klose | 2c7e3ee | 2009-01-10 17:08:25 +0000 | [diff] [blame] | 233 | return _findSoname_ldconfig(name) or _get_soname(_findLib_gcc(name)) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 234 | |
| 235 | ################################################################ |
| 236 | # test code |
| 237 | |
| 238 | def test(): |
| 239 | from ctypes import cdll |
| 240 | if os.name == "nt": |
Guido van Rossum | be19ed7 | 2007-02-09 05:37:30 +0000 | [diff] [blame] | 241 | print(cdll.msvcrt) |
| 242 | print(cdll.load("msvcrt")) |
| 243 | print(find_library("msvcrt")) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 244 | |
| 245 | if os.name == "posix": |
| 246 | # find and load_version |
Guido van Rossum | be19ed7 | 2007-02-09 05:37:30 +0000 | [diff] [blame] | 247 | print(find_library("m")) |
| 248 | print(find_library("c")) |
| 249 | print(find_library("bz2")) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 250 | |
| 251 | # getattr |
| 252 | ## print cdll.m |
| 253 | ## print cdll.bz2 |
| 254 | |
| 255 | # load |
| 256 | if sys.platform == "darwin": |
Guido van Rossum | be19ed7 | 2007-02-09 05:37:30 +0000 | [diff] [blame] | 257 | print(cdll.LoadLibrary("libm.dylib")) |
| 258 | print(cdll.LoadLibrary("libcrypto.dylib")) |
| 259 | print(cdll.LoadLibrary("libSystem.dylib")) |
| 260 | print(cdll.LoadLibrary("System.framework/System")) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 261 | else: |
Guido van Rossum | be19ed7 | 2007-02-09 05:37:30 +0000 | [diff] [blame] | 262 | print(cdll.LoadLibrary("libm.so")) |
| 263 | print(cdll.LoadLibrary("libcrypt.so")) |
| 264 | print(find_library("crypt")) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 265 | |
| 266 | if __name__ == "__main__": |
| 267 | test() |