# Autodetecting setup.py script for building the Python extensions

#

import sys, os, importlib.machinery, re, optparse

from glob import glob

import sysconfig

from distutils import log

from distutils import text_file

from distutils.errors import *

from distutils.core import Extension, setup

from distutils.command.build_ext import build_ext

from distutils.command.install import install

from distutils.command.install_lib import install_lib

from distutils.command.build_scripts import build_scripts

from distutils.spawn import find_executable

cross_compiling = "_PYTHON_HOST_PLATFORM" in os.environ

def get_platform():

# cross build

if "_PYTHON_HOST_PLATFORM" in os.environ:

return os.environ["_PYTHON_HOST_PLATFORM"]

# Get value of sys.platform

if sys.platform.startswith('osf1'):

return 'osf1'

return sys.platform

host_platform = get_platform()

# Were we compiled --with-pydebug or with #define Py_DEBUG?

COMPILED_WITH_PYDEBUG = ('--with-pydebug' in sysconfig.get_config_var("CONFIG_ARGS"))

# This global variable is used to hold the list of modules to be disabled.

disabled_module_list = []

def add_dir_to_list(dirlist, dir):

"""Add the directory 'dir' to the list 'dirlist' (after any relative

directories) if:

1) 'dir' is not already in 'dirlist'

2) 'dir' actually exists, and is a directory.

"""

if dir is None or not os.path.isdir(dir) or dir in dirlist:

return

for i, path in enumerate(dirlist):

if not os.path.isabs(path):

dirlist.insert(i + 1, dir)

return

dirlist.insert(0, dir)

def macosx_sdk_root():

"""

Return the directory of the current OSX SDK,

or '/' if no SDK was specified.

"""

cflags = sysconfig.get_config_var('CFLAGS')

m = re.search(r'-isysroot\s+(\S+)', cflags)

if m is None:

sysroot = '/'

else:

sysroot = m.group(1)

return sysroot

def is_macosx_sdk_path(path):

"""

Returns True if 'path' can be located in an OSX SDK

"""

return ( (path.startswith('/usr/') and not path.startswith('/usr/local'))

or path.startswith('/System/')

or path.startswith('/Library/') )

def find_file(filename, std_dirs, paths):

"""Searches for the directory where a given file is located,

and returns a possibly-empty list of additional directories, or None

if the file couldn't be found at all.

'filename' is the name of a file, such as readline.h or libcrypto.a.

'std_dirs' is the list of standard system directories; if the

file is found in one of them, no additional directives are needed.

'paths' is a list of additional locations to check; if the file is

found in one of them, the resulting list will contain the directory.

"""

if host_platform == 'darwin':

# Honor the MacOSX SDK setting when one was specified.

# An SDK is a directory with the same structure as a real

# system, but with only header files and libraries.

sysroot = macosx_sdk_root()

# Check the standard locations

for dir in std_dirs:

f = os.path.join(dir, filename)

if host_platform == 'darwin' and is_macosx_sdk_path(dir):

f = os.path.join(sysroot, dir[1:], filename)

if os.path.exists(f): return []

# Check the additional directories

for dir in paths:

f = os.path.join(dir, filename)

if host_platform == 'darwin' and is_macosx_sdk_path(dir):

f = os.path.join(sysroot, dir[1:], filename)

if os.path.exists(f):

return [dir]

# Not found anywhere

return None

def find_library_file(compiler, libname, std_dirs, paths):

result = compiler.find_library_file(std_dirs + paths, libname)

if result is None:

return None

if host_platform == 'darwin':

sysroot = macosx_sdk_root()

# Check whether the found file is in one of the standard directories

dirname = os.path.dirname(result)

for p in std_dirs:

# Ensure path doesn't end with path separator

p = p.rstrip(os.sep)

if host_platform == 'darwin' and is_macosx_sdk_path(p):

if os.path.join(sysroot, p[1:]) == dirname:

return [ ]

if p == dirname:

return [ ]

# Otherwise, it must have been in one of the additional directories,

# so we have to figure out which one.

for p in paths:

# Ensure path doesn't end with path separator

p = p.rstrip(os.sep)

if host_platform == 'darwin' and is_macosx_sdk_path(p):

if os.path.join(sysroot, p[1:]) == dirname:

return [ p ]

if p == dirname:

return [p]

else:

assert False, "Internal error: Path not found in std_dirs or paths"

def module_enabled(extlist, modname):

"""Returns whether the module 'modname' is present in the list

of extensions 'extlist'."""

extlist = [ext for ext in extlist if ext.name == modname]

return len(extlist)

def find_module_file(module, dirlist):

"""Find a module in a set of possible folders. If it is not found

return the unadorned filename"""

list = find_file(module, [], dirlist)

if not list:

return module

if len(list) > 1:

log.info("WARNING: multiple copies of %s found"%module)

return os.path.join(list[0], module)

class PyBuildExt(build_ext):

def __init__(self, dist):

build_ext.__init__(self, dist)

self.failed = []

def build_extensions(self):

# Detect which modules should be compiled

missing = self.detect_modules()

# Remove modules that are present on the disabled list

extensions = [ext for ext in self.extensions

if ext.name not in disabled_module_list]

# move ctypes to the end, it depends on other modules

ext_map = dict((ext.name, i) for i, ext in enumerate(extensions))

if "_ctypes" in ext_map:

ctypes = extensions.pop(ext_map["_ctypes"])

extensions.append(ctypes)

self.extensions = extensions

# Fix up the autodetected modules, prefixing all the source files

# with Modules/.

srcdir = sysconfig.get_config_var('srcdir')

if not srcdir:

# Maybe running on Windows but not using CYGWIN?

raise ValueError("No source directory; cannot proceed.")

srcdir = os.path.abspath(srcdir)

moddirlist = [os.path.join(srcdir, 'Modules')]

# Fix up the paths for scripts, too

self.distribution.scripts = [os.path.join(srcdir, filename)

for filename in self.distribution.scripts]

# Python header files

headers = [sysconfig.get_config_h_filename()]

headers += glob(os.path.join(sysconfig.get_path('include'), "*.h"))

for ext in self.extensions[:]:

ext.sources = [ find_module_file(filename, moddirlist)

for filename in ext.sources ]

if ext.depends is not None:

ext.depends = [find_module_file(filename, moddirlist)

for filename in ext.depends]

else:

ext.depends = []

# re-compile extensions if a header file has been changed

ext.depends.extend(headers)

# If a module has already been built statically,

# don't build it here

if ext.name in sys.builtin_module_names:

self.extensions.remove(ext)

# Parse Modules/Setup and Modules/Setup.local to figure out which

# modules are turned on in the file.

remove_modules = []

for filename in ('Modules/Setup', 'Modules/Setup.local'):

input = text_file.TextFile(filename, join_lines=1)

while 1:

line = input.readline()

if not line: break

line = line.split()

remove_modules.append(line[0])

input.close()

for ext in self.extensions[:]:

if ext.name in remove_modules:

self.extensions.remove(ext)

# When you run "make CC=altcc" or something similar, you really want

# those environment variables passed into the setup.py phase. Here's

# a small set of useful ones.

compiler = os.environ.get('CC')

args = {}

# unfortunately, distutils doesn't let us provide separate C and C++

# compilers

if compiler is not None:

(ccshared,cflags) = sysconfig.get_config_vars('CCSHARED','CFLAGS')

args['compiler_so'] = compiler + ' ' + ccshared + ' ' + cflags

self.compiler.set_executables(**args)

build_ext.build_extensions(self)

longest = max([len(e.name) for e in self.extensions])

if self.failed:

longest = max(longest, max([len(name) for name in self.failed]))

def print_three_column(lst):

lst.sort(key=str.lower)

# guarantee zip() doesn't drop anything

while len(lst) % 3:

lst.append("")

for e, f, g in zip(lst[::3], lst[1::3], lst[2::3]):

print("%-*s %-*s %-*s" % (longest, e, longest, f,

longest, g))

if missing:

print()

print("Python build finished successfully!")

print("The necessary bits to build these optional modules were not "

"found:")

print_three_column(missing)

print("To find the necessary bits, look in setup.py in"

" detect_modules() for the module's name.")

print()

if self.failed:

failed = self.failed[:]

print()

print("Failed to build these modules:")

print_three_column(failed)

print()

def build_extension(self, ext):

if ext.name == '_ctypes':

if not self.configure_ctypes(ext):

return

try:

build_ext.build_extension(self, ext)

except (CCompilerError, DistutilsError) as why:

self.announce('WARNING: building of extension "%s" failed: %s' %

(ext.name, sys.exc_info()[1]))

self.failed.append(ext.name)

return

# Workaround for Mac OS X: The Carbon-based modules cannot be

# reliably imported into a command-line Python

if 'Carbon' in ext.extra_link_args:

self.announce(

'WARNING: skipping import check for Carbon-based "%s"' %

ext.name)

return

if host_platform == 'darwin' and (

sys.maxsize > 2**32 and '-arch' in ext.extra_link_args):

# Don't bother doing an import check when an extension was

# build with an explicit '-arch' flag on OSX. That's currently

# only used to build 32-bit only extensions in a 4-way

# universal build and loading 32-bit code into a 64-bit

# process will fail.

self.announce(

'WARNING: skipping import check for "%s"' %

ext.name)

return

# Workaround for Cygwin: Cygwin currently has fork issues when many

# modules have been imported

if host_platform == 'cygwin':

self.announce('WARNING: skipping import check for Cygwin-based "%s"'

% ext.name)

return

ext_filename = os.path.join(

self.build_lib,

self.get_ext_filename(self.get_ext_fullname(ext.name)))

# If the build directory didn't exist when setup.py was

# started, sys.path_importer_cache has a negative result

# cached. Clear that cache before trying to import.

sys.path_importer_cache.clear()

# Don't try to load extensions for cross builds

if cross_compiling:

return

loader = importlib.machinery.ExtensionFileLoader(ext.name, ext_filename)

try:

loader.load_module()

except ImportError as why:

self.failed.append(ext.name)

self.announce('*** WARNING: renaming "%s" since importing it'

' failed: %s' % (ext.name, why), level=3)

assert not self.inplace

basename, tail = os.path.splitext(ext_filename)

newname = basename + "_failed" + tail

if os.path.exists(newname):

os.remove(newname)

os.rename(ext_filename, newname)

# XXX -- This relies on a Vile HACK in

# distutils.command.build_ext.build_extension(). The

# _built_objects attribute is stored there strictly for

# use here.

# If there is a failure, _built_objects may not be there,

# so catch the AttributeError and move on.

try:

for filename in self._built_objects:

os.remove(filename)

except AttributeError:

self.announce('unable to remove files (ignored)')

except:

exc_type, why, tb = sys.exc_info()

self.announce('*** WARNING: importing extension "%s" '

'failed with %s: %s' % (ext.name, exc_type, why),

level=3)

self.failed.append(ext.name)

def add_multiarch_paths(self):

# Debian/Ubuntu multiarch support.

# https://wiki.ubuntu.com/MultiarchSpec

cc = sysconfig.get_config_var('CC')

tmpfile = os.path.join(self.build_temp, 'multiarch')

if not os.path.exists(self.build_temp):

os.makedirs(self.build_temp)

ret = os.system(

'%s -print-multiarch > %s 2> /dev/null' % (cc, tmpfile))

multiarch_path_component = ''

try:

if ret >> 8 == 0:

with open(tmpfile) as fp:

multiarch_path_component = fp.readline().strip()

finally:

os.unlink(tmpfile)

if multiarch_path_component != '':

add_dir_to_list(self.compiler.library_dirs,

'/usr/lib/' + multiarch_path_component)

add_dir_to_list(self.compiler.include_dirs,

'/usr/include/' + multiarch_path_component)

return

if not find_executable('dpkg-architecture'):

return

opt = ''

if cross_compiling:

opt = '-t' + sysconfig.get_config_var('HOST_GNU_TYPE')

tmpfile = os.path.join(self.build_temp, 'multiarch')

if not os.path.exists(self.build_temp):

os.makedirs(self.build_temp)

ret = os.system(

'dpkg-architecture %s -qDEB_HOST_MULTIARCH > %s 2> /dev/null' %

(opt, tmpfile))

try:

if ret >> 8 == 0:

with open(tmpfile) as fp:

multiarch_path_component = fp.readline().strip()

add_dir_to_list(self.compiler.library_dirs,

'/usr/lib/' + multiarch_path_component)

add_dir_to_list(self.compiler.include_dirs,

'/usr/include/' + multiarch_path_component)

finally:

os.unlink(tmpfile)

def add_gcc_paths(self):

gcc = sysconfig.get_config_var('CC')

tmpfile = os.path.join(self.build_temp, 'gccpaths')

if not os.path.exists(self.build_temp):

os.makedirs(self.build_temp)

ret = os.system('%s -E -v - </dev/null 2>%s 1>/dev/null' % (gcc, tmpfile))

is_gcc = False

in_incdirs = False

inc_dirs = []

lib_dirs = []

try:

if ret >> 8 == 0:

with open(tmpfile) as fp:

for line in fp.readlines():

if line.startswith("gcc version"):

is_gcc = True

elif line.startswith("#include <...>"):

in_incdirs = True

elif line.startswith("End of search list"):

in_incdirs = False

elif is_gcc and line.startswith("LIBRARY_PATH"):

for d in line.strip().split("=")[1].split(":"):

d = os.path.normpath(d)

if '/gcc/' not in d:

add_dir_to_list(self.compiler.library_dirs,

d)

elif is_gcc and in_incdirs and '/gcc/' not in line:

add_dir_to_list(self.compiler.include_dirs,

line.strip())

finally:

os.unlink(tmpfile)

def detect_modules(self):

# Ensure that /usr/local is always used, but the local build

# directories (i.e. '.' and 'Include') must be first. See issue

# 10520.

if not cross_compiling:

add_dir_to_list(self.compiler.library_dirs, '/usr/local/lib')

add_dir_to_list(self.compiler.include_dirs, '/usr/local/include')

# only change this for cross builds for 3.3, issues on Mageia

if cross_compiling:

self.add_gcc_paths()

self.add_multiarch_paths()

# Add paths specified in the environment variables LDFLAGS and

# CPPFLAGS for header and library files.

# We must get the values from the Makefile and not the environment

# directly since an inconsistently reproducible issue comes up where

# the environment variable is not set even though the value were passed

# into configure and stored in the Makefile (issue found on OS X 10.3).

for env_var, arg_name, dir_list in (

('LDFLAGS', '-R', self.compiler.runtime_library_dirs),

('LDFLAGS', '-L', self.compiler.library_dirs),

('CPPFLAGS', '-I', self.compiler.include_dirs)):

env_val = sysconfig.get_config_var(env_var)

if env_val:

# To prevent optparse from raising an exception about any

# options in env_val that it doesn't know about we strip out

# all double dashes and any dashes followed by a character

# that is not for the option we are dealing with.

#

# Please note that order of the regex is important! We must

# strip out double-dashes first so that we don't end up with

# substituting "--Long" to "-Long" and thus lead to "ong" being

# used for a library directory.

env_val = re.sub(r'(^|\s+)-(-|(?!%s))' % arg_name[1],

' ', env_val)

parser = optparse.OptionParser()

# Make sure that allowing args interspersed with options is

# allowed

parser.allow_interspersed_args = True

parser.error = lambda msg: None

parser.add_option(arg_name, dest="dirs", action="append")

options = parser.parse_args(env_val.split())[0]

if options.dirs:

for directory in reversed(options.dirs):

add_dir_to_list(dir_list, directory)

if os.path.normpath(sys.base_prefix) != '/usr' \

and not sysconfig.get_config_var('PYTHONFRAMEWORK'):

# OSX note: Don't add LIBDIR and INCLUDEDIR to building a framework

# (PYTHONFRAMEWORK is set) to avoid # linking problems when

# building a framework with different architectures than

# the one that is currently installed (issue #7473)

add_dir_to_list(self.compiler.library_dirs,

sysconfig.get_config_var("LIBDIR"))

add_dir_to_list(self.compiler.include_dirs,

sysconfig.get_config_var("INCLUDEDIR"))

# lib_dirs and inc_dirs are used to search for files;

# if a file is found in one of those directories, it can

# be assumed that no additional -I,-L directives are needed.

if not cross_compiling:

lib_dirs = self.compiler.library_dirs + [

'/lib64', '/usr/lib64',

'/lib', '/usr/lib',

]

inc_dirs = self.compiler.include_dirs + ['/usr/include']

else:

lib_dirs = self.compiler.library_dirs[:]

inc_dirs = self.compiler.include_dirs[:]

exts = []

missing = []

config_h = sysconfig.get_config_h_filename()

with open(config_h) as file:

config_h_vars = sysconfig.parse_config_h(file)

srcdir = sysconfig.get_config_var('srcdir')

# OSF/1 and Unixware have some stuff in /usr/ccs/lib (like -ldb)

if host_platform in ['osf1', 'unixware7', 'openunix8']:

lib_dirs += ['/usr/ccs/lib']

# HP-UX11iv3 keeps files in lib/hpux folders.

if host_platform == 'hp-ux11':

lib_dirs += ['/usr/lib/hpux64', '/usr/lib/hpux32']

if host_platform == 'darwin':

# This should work on any unixy platform ;-)

# If the user has bothered specifying additional -I and -L flags

# in OPT and LDFLAGS we might as well use them here.

#

# NOTE: using shlex.split would technically be more correct, but

# also gives a bootstrap problem. Let's hope nobody uses

# directories with whitespace in the name to store libraries.

cflags, ldflags = sysconfig.get_config_vars(

'CFLAGS', 'LDFLAGS')

for item in cflags.split():

if item.startswith('-I'):

inc_dirs.append(item[2:])

for item in ldflags.split():

if item.startswith('-L'):

lib_dirs.append(item[2:])

# Check for MacOS X, which doesn't need libm.a at all

math_libs = ['m']

if host_platform == 'darwin':

math_libs = []

# XXX Omitted modules: gl, pure, dl, SGI-specific modules

#

# The following modules are all pretty straightforward, and compile

# on pretty much any POSIXish platform.

#

# array objects

exts.append( Extension('array', ['arraymodule.c']) )

# complex math library functions

exts.append( Extension('cmath', ['cmathmodule.c', '_math.c'],

depends=['_math.h'],

libraries=math_libs) )

# math library functions, e.g. sin()

exts.append( Extension('math', ['mathmodule.c', '_math.c'],

depends=['_math.h'],

libraries=math_libs) )

# time libraries: librt may be needed for clock_gettime()

time_libs = []

lib = sysconfig.get_config_var('TIMEMODULE_LIB')

if lib:

time_libs.append(lib)

# time operations and variables

exts.append( Extension('time', ['timemodule.c'],

libraries=time_libs) )

exts.append( Extension('_datetime', ['_datetimemodule.c']) )

# random number generator implemented in C

exts.append( Extension("_random", ["_randommodule.c"]) )

# bisect

exts.append( Extension("_bisect", ["_bisectmodule.c"]) )

# heapq

exts.append( Extension("_heapq", ["_heapqmodule.c"]) )

# C-optimized pickle replacement

exts.append( Extension("_pickle", ["_pickle.c"]) )

# atexit

exts.append( Extension("atexit", ["atexitmodule.c"]) )

# _json speedups

exts.append( Extension("_json", ["_json.c"]) )

# Python C API test module

exts.append( Extension('_testcapi', ['_testcapimodule.c'],

depends=['testcapi_long.h']) )

# Python PEP-3118 (buffer protocol) test module

exts.append( Extension('_testbuffer', ['_testbuffer.c']) )

# Test loading multiple modules from one compiled file (http://bugs.python.org/issue16421)

exts.append( Extension('_testimportmultiple', ['_testimportmultiple.c']) )

# profiler (_lsprof is for cProfile.py)

exts.append( Extension('_lsprof', ['_lsprof.c', 'rotatingtree.c']) )

# static Unicode character database

exts.append( Extension('unicodedata', ['unicodedata.c']) )

# Modules with some UNIX dependencies -- on by default:

# (If you have a really backward UNIX, select and socket may not be

# supported...)

# fcntl(2) and ioctl(2)

libs = []

if (config_h_vars.get('FLOCK_NEEDS_LIBBSD', False)):

# May be necessary on AIX for flock function

libs = ['bsd']

exts.append( Extension('fcntl', ['fcntlmodule.c'], libraries=libs) )

# pwd(3)

exts.append( Extension('pwd', ['pwdmodule.c']) )

# grp(3)

exts.append( Extension('grp', ['grpmodule.c']) )

# spwd, shadow passwords

if (config_h_vars.get('HAVE_GETSPNAM', False) or

config_h_vars.get('HAVE_GETSPENT', False)):

exts.append( Extension('spwd', ['spwdmodule.c']) )

else:

missing.append('spwd')

# select(2); not on ancient System V

exts.append( Extension('select', ['selectmodule.c']) )

# Fred Drake's interface to the Python parser

exts.append( Extension('parser', ['parsermodule.c']) )

# Memory-mapped files (also works on Win32).

exts.append( Extension('mmap', ['mmapmodule.c']) )

# Lance Ellinghaus's syslog module

# syslog daemon interface

exts.append( Extension('syslog', ['syslogmodule.c']) )

#

# Here ends the simple stuff. From here on, modules need certain

# libraries, are platform-specific, or present other surprises.

#

# Multimedia modules

# These don't work for 64-bit platforms!!!

# These represent audio samples or images as strings:

# Operations on audio samples

# According to #993173, this one should actually work fine on

# 64-bit platforms.

exts.append( Extension('audioop', ['audioop.c']) )

# readline

do_readline = self.compiler.find_library_file(lib_dirs, 'readline')

readline_termcap_library = ""

curses_library = ""

# Cannot use os.popen here in py3k.

tmpfile = os.path.join(self.build_temp, 'readline_termcap_lib')

if not os.path.exists(self.build_temp):

os.makedirs(self.build_temp)

# Determine if readline is already linked against curses or tinfo.

if do_readline:

if cross_compiling:

ret = os.system("%s -d %s | grep '(NEEDED)' > %s" \

% (sysconfig.get_config_var('READELF'),

do_readline, tmpfile))

elif find_executable('ldd'):

ret = os.system("ldd %s > %s" % (do_readline, tmpfile))

else:

ret = 256

if ret >> 8 == 0:

with open(tmpfile) as fp:

for ln in fp:

if 'curses' in ln:

readline_termcap_library = re.sub(

r'.*lib(n?cursesw?)\.so.*', r'\1', ln

).rstrip()

break

# termcap interface split out from ncurses

if 'tinfo' in ln:

readline_termcap_library = 'tinfo'

break

if os.path.exists(tmpfile):

os.unlink(tmpfile)

# Issue 7384: If readline is already linked against curses,

# use the same library for the readline and curses modules.

if 'curses' in readline_termcap_library:

curses_library = readline_termcap_library

elif self.compiler.find_library_file(lib_dirs, 'ncursesw'):

curses_library = 'ncursesw'

elif self.compiler.find_library_file(lib_dirs, 'ncurses'):

curses_library = 'ncurses'

elif self.compiler.find_library_file(lib_dirs, 'curses'):

curses_library = 'curses'

if host_platform == 'darwin':

os_release = int(os.uname()[2].split('.')[0])

dep_target = sysconfig.get_config_var('MACOSX_DEPLOYMENT_TARGET')

if dep_target and dep_target.split('.') < ['10', '5']:

os_release = 8

if os_release < 9:

# MacOSX 10.4 has a broken readline. Don't try to build

# the readline module unless the user has installed a fixed

# readline package

if find_file('readline/rlconf.h', inc_dirs, []) is None:

do_readline = False

if do_readline:

if host_platform == 'darwin' and os_release < 9:

# In every directory on the search path search for a dynamic

# library and then a static library, instead of first looking

# for dynamic libraries on the entire path.

# This way a staticly linked custom readline gets picked up

# before the (possibly broken) dynamic library in /usr/lib.

readline_extra_link_args = ('-Wl,-search_paths_first',)

else:

readline_extra_link_args = ()

readline_libs = ['readline']

if readline_termcap_library:

pass # Issue 7384: Already linked against curses or tinfo.

elif curses_library:

readline_libs.append(curses_library)

elif self.compiler.find_library_file(lib_dirs +

['/usr/lib/termcap'],

'termcap'):

readline_libs.append('termcap')

exts.append( Extension('readline', ['readline.c'],

library_dirs=['/usr/lib/termcap'],

extra_link_args=readline_extra_link_args,

libraries=readline_libs) )

else:

missing.append('readline')

# crypt module.

if self.compiler.find_library_file(lib_dirs, 'crypt'):

libs = ['crypt']

else:

libs = []

exts.append( Extension('_crypt', ['_cryptmodule.c'], libraries=libs) )

# CSV files

exts.append( Extension('_csv', ['_csv.c']) )

# POSIX subprocess module helper.

exts.append( Extension('_posixsubprocess', ['_posixsubprocess.c']) )

# socket(2)

exts.append( Extension('_socket', ['socketmodule.c'],

depends = ['socketmodule.h']) )

# Detect SSL support for the socket module (via _ssl)

search_for_ssl_incs_in = [

'/usr/local/ssl/include',

'/usr/contrib/ssl/include/'

]

ssl_incs = find_file('openssl/ssl.h', inc_dirs,

search_for_ssl_incs_in

)

if ssl_incs is not None:

krb5_h = find_file('krb5.h', inc_dirs,

['/usr/kerberos/include'])

if krb5_h:

ssl_incs += krb5_h

ssl_libs = find_library_file(self.compiler, 'ssl',lib_dirs,

['/usr/local/ssl/lib',

'/usr/contrib/ssl/lib/'

] )

if (ssl_incs is not None and

ssl_libs is not None):

exts.append( Extension('_ssl', ['_ssl.c'],

include_dirs = ssl_incs,

library_dirs = ssl_libs,

libraries = ['ssl', 'crypto'],

depends = ['socketmodule.h']), )

else:

missing.append('_ssl')

# find out which version of OpenSSL we have

openssl_ver = 0

openssl_ver_re = re.compile(

'^\s*#\s*define\s+OPENSSL_VERSION_NUMBER\s+(0x[0-9a-fA-F]+)' )

# look for the openssl version header on the compiler search path.

opensslv_h = find_file('openssl/opensslv.h', [],

inc_dirs + search_for_ssl_incs_in)

if opensslv_h:

name = os.path.join(opensslv_h[0], 'openssl/opensslv.h')

if host_platform == 'darwin' and is_macosx_sdk_path(name):

name = os.path.join(macosx_sdk_root(), name[1:])

try:

with open(name, 'r') as incfile:

for line in incfile:

m = openssl_ver_re.match(line)

if m:

openssl_ver = int(m.group(1), 16)

break

except IOError as msg:

print("IOError while reading opensshv.h:", msg)

#print('openssl_ver = 0x%08x' % openssl_ver)

min_openssl_ver = 0x00907000

have_any_openssl = ssl_incs is not None and ssl_libs is not None

have_usable_openssl = (have_any_openssl and

openssl_ver >= min_openssl_ver)

if have_any_openssl:

if have_usable_openssl:

# The _hashlib module wraps optimized implementations

# of hash functions from the OpenSSL library.

exts.append( Extension('_hashlib', ['_hashopenssl.c'],

depends = ['hashlib.h'],

include_dirs = ssl_incs,

library_dirs = ssl_libs,

libraries = ['ssl', 'crypto']) )

else:

print("warning: openssl 0x%08x is too old for _hashlib" %

openssl_ver)

missing.append('_hashlib')

# We always compile these even when OpenSSL is available (issue #14693).

# It's harmless and the object code is tiny (40-50 KB per module,

# only loaded when actually used).

exts.append( Extension('_sha256', ['sha256module.c'],

depends=['hashlib.h']) )

exts.append( Extension('_sha512', ['sha512module.c'],

depends=['hashlib.h']) )

exts.append( Extension('_md5', ['md5module.c'],

depends=['hashlib.h']) )

exts.append( Extension('_sha1', ['sha1module.c'],

depends=['hashlib.h']) )

# SHA-3 (Keccak) module

sha3_depends = ['hashlib.h']

keccak = os.path.join(os.getcwd(), srcdir, 'Modules', '_sha3',

'keccak')

for pattern in ('*.c', '*.h', '*.macros'):

sha3_depends.extend(glob(os.path.join(keccak, pattern)))

exts.append(Extension("_sha3", ["_sha3/sha3module.c"],

depends=sha3_depends))

# Modules that provide persistent dictionary-like semantics. You will

# probably want to arrange for at least one of them to be available on

# your machine, though none are defined by default because of library

# dependencies. The Python module dbm/__init__.py provides an

# implementation independent wrapper for these; dbm/dumb.py provides

# similar functionality (but slower of course) implemented in Python.

# Sleepycat^WOracle Berkeley DB interface.

# http://www.oracle.com/database/berkeley-db/db/index.html

#

# This requires the Sleepycat^WOracle DB code. The supported versions

# are set below. Visit the URL above to download

# a release. Most open source OSes come with one or more

# versions of BerkeleyDB already installed.

max_db_ver = (5, 3)

min_db_ver = (3, 3)

db_setup_debug = False # verbose debug prints from this script?

def allow_db_ver(db_ver):

"""Returns a boolean if the given BerkeleyDB version is acceptable.

Args:

db_ver: A tuple of the version to verify.

"""

if not (min_db_ver <= db_ver <= max_db_ver):

return False

return True

def gen_db_minor_ver_nums(major):

if major == 4:

for x in range(max_db_ver[1]+1):

if allow_db_ver((4, x)):

yield x

elif major == 3:

for x in (3,):

if allow_db_ver((3, x)):

yield x

else:

raise ValueError("unknown major BerkeleyDB version", major)

# construct a list of paths to look for the header file in on

# top of the normal inc_dirs.

db_inc_paths = [

'/usr/include/db4',

'/usr/local/include/db4',

'/opt/sfw/include/db4',

'/usr/include/db3',

'/usr/local/include/db3',

'/opt/sfw/include/db3',

# Fink defaults (http://fink.sourceforge.net/)

'/sw/include/db4',

'/sw/include/db3',

]

# 4.x minor number specific paths

for x in gen_db_minor_ver_nums(4):

db_inc_paths.append('/usr/include/db4%d' % x)

db_inc_paths.append('/usr/include/db4.%d' % x)

db_inc_paths.append('/usr/local/BerkeleyDB.4.%d/include' % x)

db_inc_paths.append('/usr/local/include/db4%d' % x)

db_inc_paths.append('/pkg/db-4.%d/include' % x)

db_inc_paths.append('/opt/db-4.%d/include' % x)

# MacPorts default (http://www.macports.org/)

db_inc_paths.append('/opt/local/include/db4%d' % x)

# 3.x minor number specific paths

for x in gen_db_minor_ver_nums(3):

db_inc_paths.append('/usr/include/db3%d' % x)

db_inc_paths.append('/usr/local/BerkeleyDB.3.%d/include' % x)

db_inc_paths.append('/usr/local/include/db3%d' % x)

db_inc_paths.append('/pkg/db-3.%d/include' % x)

db_inc_paths.append('/opt/db-3.%d/include' % x)

if cross_compiling:

db_inc_paths = []

# Add some common subdirectories for Sleepycat DB to the list,

# based on the standard include directories. This way DB3/4 gets

# picked up when it is installed in a non-standard prefix and

# the user has added that prefix into inc_dirs.

std_variants = []

for dn in inc_dirs:

std_variants.append(os.path.join(dn, 'db3'))

std_variants.append(os.path.join(dn, 'db4'))

for x in gen_db_minor_ver_nums(4):

std_variants.append(os.path.join(dn, "db4%d"%x))

std_variants.append(os.path.join(dn, "db4.%d"%x))

for x in gen_db_minor_ver_nums(3):

std_variants.append(os.path.join(dn, "db3%d"%x))

std_variants.append(os.path.join(dn, "db3.%d"%x))

db_inc_paths = std_variants + db_inc_paths

db_inc_paths = [p for p in db_inc_paths if os.path.exists(p)]

db_ver_inc_map = {}

if host_platform == 'darwin':

sysroot = macosx_sdk_root()

class db_found(Exception): pass

try:

# See whether there is a Sleepycat header in the standard

# search path.

for d in inc_dirs + db_inc_paths:

f = os.path.join(d, "db.h")

if host_platform == 'darwin' and is_macosx_sdk_path(d):

f = os.path.join(sysroot, d[1:], "db.h")

if db_setup_debug: print("db: looking for db.h in", f)

if os.path.exists(f):

with open(f, 'rb') as file:

f = file.read()

m = re.search(br"#define\WDB_VERSION_MAJOR\W(\d+)", f)

if m:

db_major = int(m.group(1))

m = re.search(br"#define\WDB_VERSION_MINOR\W(\d+)", f)

db_minor = int(m.group(1))

db_ver = (db_major, db_minor)

# Avoid 4.6 prior to 4.6.21 due to a BerkeleyDB bug

if db_ver == (4, 6):

m = re.search(br"#define\WDB_VERSION_PATCH\W(\d+)", f)

db_patch = int(m.group(1))

if db_patch < 21:

print("db.h:", db_ver, "patch", db_patch,

"being ignored (4.6.x must be >= 4.6.21)")

continue

if ( (db_ver not in db_ver_inc_map) and

allow_db_ver(db_ver) ):

# save the include directory with the db.h version

# (first occurrence only)

db_ver_inc_map[db_ver] = d

if db_setup_debug:

print("db.h: found", db_ver, "in", d)

else:

# we already found a header for this library version

if db_setup_debug: print("db.h: ignoring", d)

else:

# ignore this header, it didn't contain a version number

if db_setup_debug:

print("db.h: no version number version in", d)

db_found_vers = list(db_ver_inc_map.keys())

db_found_vers.sort()

while db_found_vers:

db_ver = db_found_vers.pop()

db_incdir = db_ver_inc_map[db_ver]

# check lib directories parallel to the location of the header

db_dirs_to_check = [

db_incdir.replace("include", 'lib64'),

db_incdir.replace("include", 'lib'),

]

if host_platform != 'darwin':

db_dirs_to_check = list(filter(os.path.isdir, db_dirs_to_check))

else:

# Same as other branch, but takes OSX SDK into account

tmp = []

for dn in db_dirs_to_check:

if is_macosx_sdk_path(dn):

if os.path.isdir(os.path.join(sysroot, dn[1:])):