mirror of
https://github.com/mongodb/mongo.git
synced 2024-11-22 13:07:39 +01:00
99aea5ef94
GitOrigin-RevId: 809708bcf3de2dbdb85a8d770dfe0fa2ca36d6e5
1479 lines
56 KiB
Python
1479 lines
56 KiB
Python
"""Extension to SCons providing advanced static library dependency tracking.
|
|
|
|
These modifications to a build environment, which can be attached to
|
|
StaticLibrary and Program builders via a call to setup_environment(env),
|
|
cause the build system to track library dependencies through static libraries,
|
|
and to add them to the link command executed when building programs.
|
|
|
|
For example, consider a program 'try' that depends on a lib 'tc', which in
|
|
turn uses a symbol from a lib 'tb' which in turn uses a library from 'ta'.
|
|
|
|
Without this package, the Program declaration for "try" looks like this:
|
|
|
|
Program('try', ['try.c', 'path/to/${LIBPREFIX}tc${LIBSUFFIX}',
|
|
'path/to/${LIBPREFIX}tb${LIBSUFFIX}',
|
|
'path/to/${LIBPREFIX}ta${LIBSUFFIX}',])
|
|
|
|
With this library, we can instead write the following
|
|
|
|
Program('try', ['try.c'], LIBDEPS=['path/to/tc'])
|
|
StaticLibrary('tc', ['c.c'], LIBDEPS=['path/to/tb'])
|
|
StaticLibrary('tb', ['b.c'], LIBDEPS=['path/to/ta'])
|
|
StaticLibrary('ta', ['a.c'])
|
|
|
|
And the build system will figure out that it needs to link libta.a and libtb.a
|
|
when building 'try'.
|
|
|
|
A StaticLibrary S may also declare programs or libraries, [L1, ...] to be dependent
|
|
upon S by setting LIBDEPS_DEPENDENTS=[L1, ...], using the same syntax as is used
|
|
for LIBDEPS, except that the libraries and programs will not have LIBPREFIX/LIBSUFFIX
|
|
automatically added when missing.
|
|
"""
|
|
|
|
# Copyright (c) 2010, Corensic Inc., All Rights Reserved.
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining
|
|
# a copy of this software and associated documentation files (the
|
|
# "Software"), to deal in the Software without restriction, including
|
|
# without limitation the rights to use, copy, modify, merge, publish,
|
|
# distribute, sublicense, and/or sell copies of the Software, and to
|
|
# permit persons to whom the Software is furnished to do so, subject to
|
|
# the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included
|
|
# in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
|
|
# KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
|
|
# WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
|
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
from collections import defaultdict
|
|
from functools import partial
|
|
import enum
|
|
import copy
|
|
import json
|
|
import os
|
|
import sys
|
|
import glob
|
|
import textwrap
|
|
import hashlib
|
|
import json
|
|
import fileinput
|
|
import subprocess
|
|
|
|
try:
|
|
import networkx
|
|
from buildscripts.libdeps.libdeps.graph import EdgeProps, NodeProps, LibdepsGraph
|
|
except ImportError:
|
|
pass
|
|
|
|
import SCons.Errors
|
|
import SCons.Scanner
|
|
import SCons.Util
|
|
import SCons
|
|
from SCons.Script import COMMAND_LINE_TARGETS
|
|
|
|
|
|
class Constants:
|
|
Libdeps = "LIBDEPS"
|
|
LibdepsCached = "LIBDEPS_cached"
|
|
LibdepsDependents = "LIBDEPS_DEPENDENTS"
|
|
LibdepsGlobal = "LIBDEPS_GLOBAL"
|
|
LibdepsNoInherit = "LIBDEPS_NO_INHERIT"
|
|
LibdepsInterface = "LIBDEPS_INTERFACE"
|
|
LibdepsPrivate = "LIBDEPS_PRIVATE"
|
|
LibdepsTags = "LIBDEPS_TAGS"
|
|
LibdepsTagExpansion = "LIBDEPS_TAG_EXPANSIONS"
|
|
MissingLibdep = "MISSING_LIBDEP_"
|
|
ProgdepsDependents = "PROGDEPS_DEPENDENTS"
|
|
SysLibdeps = "SYSLIBDEPS"
|
|
SysLibdepsCached = "SYSLIBDEPS_cached"
|
|
SysLibdepsPrivate = "SYSLIBDEPS_PRIVATE"
|
|
|
|
|
|
class deptype(tuple, enum.Enum):
|
|
Global: tuple = (0, 'GLOBAL')
|
|
Public: tuple = (1, 'PUBLIC')
|
|
Private: tuple = (2, 'PRIVATE')
|
|
Interface: tuple = (3, 'INTERFACE')
|
|
|
|
def __lt__(self, other):
|
|
if self.__class__ is other.__class__:
|
|
return self.value[0] < other.value[0]
|
|
return NotImplemented
|
|
|
|
def __str__(self):
|
|
return self.value[1]
|
|
|
|
def __int__(self):
|
|
return self.value[0]
|
|
|
|
|
|
class dependency:
|
|
def __init__(self, value, deptype, listed_name):
|
|
self.target_node = value
|
|
self.dependency_type = deptype
|
|
self.listed_name = listed_name
|
|
|
|
def __str__(self):
|
|
return str(self.target_node)
|
|
|
|
|
|
class FlaggedLibdep:
|
|
"""
|
|
Utility class used for processing prefix and postfix flags on libdeps. The class
|
|
can keep track of separate lists for prefix and postfix as well separators,
|
|
allowing for modifications to the lists and then re-application of the flags with
|
|
modifications to a larger list representing the link line.
|
|
"""
|
|
|
|
def __init__(self, libnode=None, env=None, start_index=None):
|
|
"""
|
|
The libnode should be a Libdep SCons node, and the env is the target env in
|
|
which the target has a dependency on the libdep. The start_index is important as
|
|
it determines where this FlaggedLibdep starts in the larger list of libdeps.
|
|
|
|
The start_index will cut the larger list, and then re-apply this libdep with flags
|
|
at that location. This class will exract the prefix and postfix flags
|
|
from the Libdep nodes env.
|
|
"""
|
|
self.libnode = libnode
|
|
self.env = env
|
|
|
|
# We need to maintain our own copy so as not to disrupt the env's original list.
|
|
try:
|
|
self.prefix_flags = copy.copy(getattr(libnode.attributes, 'libdeps_prefix_flags', []))
|
|
self.postfix_flags = copy.copy(getattr(libnode.attributes, 'libdeps_postfix_flags', []))
|
|
except AttributeError:
|
|
self.prefix_flags = []
|
|
self.postfix_flags = []
|
|
|
|
self.start_index = start_index
|
|
|
|
def __str__(self):
|
|
return str(self.libnode)
|
|
|
|
def add_lib_to_result_list(self, result):
|
|
"""
|
|
This function takes in the current list of libdeps for a given target, and will
|
|
apply the libdep taking care of the prefix, postfix and any required separators when
|
|
adding to the list.
|
|
"""
|
|
if self.start_index != None:
|
|
result[:] = result[:self.start_index]
|
|
self._add_lib_and_flags(result)
|
|
|
|
def _get_separators(self, flags):
|
|
|
|
separated_list = []
|
|
|
|
for flag in flags:
|
|
separators = self.env.get('LIBDEPS_FLAG_SEPARATORS', {}).get(flag, {})
|
|
separated_list.append(separators.get('prefix', ' '))
|
|
separated_list.append(flag)
|
|
separated_list.append(separators.get('suffix', ' '))
|
|
|
|
return separated_list
|
|
|
|
def _get_lib_with_flags(self):
|
|
|
|
lib_and_flags = []
|
|
|
|
lib_and_flags += self._get_separators(self.prefix_flags)
|
|
lib_and_flags += [str(self)]
|
|
lib_and_flags += self._get_separators(self.postfix_flags)
|
|
|
|
return lib_and_flags
|
|
|
|
def _add_lib_and_flags(self, result):
|
|
"""
|
|
This function will clean up the flags for the link line after extracting everything
|
|
from the environment. This will mostly look for separators that are just a space, and
|
|
remove them from the list, as the final link line will add spaces back for each item
|
|
in the list. It will take to concat flags where the separators don't allow for a space.
|
|
"""
|
|
next_contig_str = ''
|
|
|
|
for item in self._get_lib_with_flags():
|
|
if item != ' ':
|
|
next_contig_str += item
|
|
else:
|
|
if next_contig_str:
|
|
result.append(next_contig_str)
|
|
next_contig_str = ''
|
|
|
|
if next_contig_str:
|
|
result.append(next_contig_str)
|
|
|
|
|
|
class LibdepLinter:
|
|
"""
|
|
This class stores the rules for linting the libdeps. Using a decorator,
|
|
new rules can easily be added to the class, and will be called when
|
|
linting occurs. Each rule is run on each libdep.
|
|
|
|
When a rule is broken, a LibdepLinterError exception will be raised.
|
|
Optionally the class can be configured to print the error message and
|
|
keep going with the build.
|
|
|
|
Each rule should provide a method to skip that rule on a given node,
|
|
by supplying the correct flag in the LIBDEPS_TAG environment var for
|
|
that node.
|
|
|
|
"""
|
|
|
|
skip_linting = False
|
|
print_linter_errors = False
|
|
|
|
linting_time = 0
|
|
linting_infractions = 0
|
|
linting_rules_run = 0
|
|
registered_linting_time = False
|
|
|
|
dangling_dep_dependents = set()
|
|
|
|
@staticmethod
|
|
def _make_linter_decorator():
|
|
"""
|
|
This is used for gathering the functions
|
|
by decorator that will be used for linting a given libdep.
|
|
"""
|
|
|
|
funcs = {}
|
|
|
|
def linter_rule_func(func):
|
|
funcs[func.__name__] = func
|
|
return func
|
|
|
|
linter_rule_func.all = funcs
|
|
return linter_rule_func
|
|
|
|
linter_rule = _make_linter_decorator.__func__()
|
|
linter_final_check = _make_linter_decorator.__func__()
|
|
|
|
@classmethod
|
|
def _skip_linting(cls):
|
|
return cls.skip_linting
|
|
|
|
@classmethod
|
|
def _start_timer(cls):
|
|
# Record time spent linting if we are in print mode.
|
|
if cls.print_linter_errors:
|
|
from timeit import default_timer as timer
|
|
return timer()
|
|
|
|
@classmethod
|
|
def _stop_timer(cls, start, num_rules):
|
|
# Record time spent linting if we are in print mode.
|
|
if cls.print_linter_errors:
|
|
from timeit import default_timer as timer
|
|
cls.linting_time += timer() - start
|
|
cls.linting_rules_run += num_rules
|
|
|
|
def __init__(self, env, target=None):
|
|
self.env = env
|
|
self.target = target
|
|
self.unique_libs = set()
|
|
self._libdeps_types_previous = dict()
|
|
|
|
# If we are in print mode, we will record some linting metrics,
|
|
# and print the results at the end of the build.
|
|
if self.__class__.print_linter_errors and not self.__class__.registered_linting_time:
|
|
import atexit
|
|
|
|
def print_linting_time():
|
|
print(f"Spent {self.__class__.linting_time} seconds linting libdeps.")
|
|
print(
|
|
f"Found {self.__class__.linting_infractions} issues out of {self.__class__.linting_rules_run} libdeps rules checked."
|
|
)
|
|
|
|
atexit.register(print_linting_time)
|
|
self.__class__.registered_linting_time = True
|
|
|
|
def lint_libdeps(self, libdeps):
|
|
"""
|
|
Lint the given list of libdeps for all
|
|
rules.
|
|
"""
|
|
|
|
# Build performance optimization if you
|
|
# are sure your build is clean.
|
|
if self._skip_linting():
|
|
return
|
|
start = self._start_timer()
|
|
|
|
linter_rules = [getattr(self, linter_rule) for linter_rule in self.linter_rule.all]
|
|
|
|
for libdep in libdeps:
|
|
for linter_rule in linter_rules:
|
|
linter_rule(libdep)
|
|
|
|
self._stop_timer(start, len(linter_rules) * len(libdeps))
|
|
|
|
def final_checks(self):
|
|
# Build performance optimization if you
|
|
# are sure your build is clean.
|
|
if self._skip_linting():
|
|
return
|
|
start = self._start_timer()
|
|
|
|
linter_rules = [
|
|
getattr(self.__class__, rule) for rule in self.__class__.linter_final_check.all
|
|
]
|
|
|
|
for linter_rule in linter_rules:
|
|
linter_rule(self)
|
|
|
|
self._stop_timer(start, len(linter_rules))
|
|
|
|
def _raise_libdep_lint_exception(self, message):
|
|
"""
|
|
Raises the LibdepLinterError exception or if configure
|
|
to do so, just prints the error.
|
|
"""
|
|
prefix = "LibdepLinter: \n\t"
|
|
message = prefix + message.replace('\n', '\n\t') + '\n'
|
|
if self.__class__.print_linter_errors:
|
|
self.__class__.linting_infractions += 1
|
|
print(message)
|
|
else:
|
|
raise LibdepLinterError(message)
|
|
|
|
def _check_for_lint_tags(self, lint_tag, env=None, inclusive_tag=False):
|
|
"""
|
|
Used to get the lint tag from the environment,
|
|
and if printing instead of raising exceptions,
|
|
will ignore the tags.
|
|
"""
|
|
|
|
# If print mode is on, we want to make sure to bypass checking
|
|
# exclusive tags so we can make sure the exceptions are not excluded
|
|
# and are printed. If it's an inclusive tag, we want to ignore this
|
|
# early return completely, because we want to make sure the node
|
|
# gets included for checking, and the exception gets printed.
|
|
if not inclusive_tag and self.__class__.print_linter_errors:
|
|
return False
|
|
|
|
target_env = env if env else self.env
|
|
|
|
if lint_tag in target_env.get(Constants.LibdepsTags, []):
|
|
return True
|
|
|
|
def _get_deps_dependents(self, env=None):
|
|
""" util function to get all types of DEPS_DEPENDENTS"""
|
|
target_env = env if env else self.env
|
|
deps_dependents = target_env.get(Constants.LibdepsDependents, []).copy()
|
|
deps_dependents += target_env.get(Constants.ProgdepsDependents, [])
|
|
return deps_dependents
|
|
|
|
def _get_deps_dependents_with_types(self, builder, type):
|
|
return [(dependent[0], builder) if isinstance(dependent, tuple) else (dependent, builder)
|
|
for dependent in self.env.get(type, [])]
|
|
|
|
@linter_rule
|
|
def linter_rule_leaf_node_no_deps(self, libdep):
|
|
"""
|
|
LIBDEP RULE:
|
|
Nodes marked explicitly as a leaf node should not have any dependencies,
|
|
unless those dependencies are explicitly marked as allowed as leaf node
|
|
dependencies.
|
|
"""
|
|
if not self._check_for_lint_tags('lint-leaf-node-no-deps', inclusive_tag=True):
|
|
return
|
|
|
|
# Ignore dependencies that explicitly exempt themselves.
|
|
if self._check_for_lint_tags('lint-leaf-node-allowed-dep', libdep.target_node.env):
|
|
return
|
|
|
|
# Global dependencies will apply to leaf nodes, so they should
|
|
# be automatically exempted.
|
|
if libdep.dependency_type == deptype.Global:
|
|
return
|
|
|
|
target_type = self.target[0].builder.get_name(self.env)
|
|
lib = os.path.basename(str(libdep))
|
|
self._raise_libdep_lint_exception(
|
|
textwrap.dedent(f"""\
|
|
{target_type} '{self.target[0]}' has dependency '{lib}' and is marked explicitly as a leaf node,
|
|
and '{lib}' does not exempt itself as an exception to the rule."""))
|
|
|
|
@linter_rule
|
|
def linter_rule_no_dangling_deps(self, libdep):
|
|
"""
|
|
LIBDEP RULE:
|
|
All reverse dependency edges must point to a node which will be built.
|
|
"""
|
|
if self._check_for_lint_tags('lint-allow-dangling-dep-dependent'):
|
|
return
|
|
|
|
# Gather the DEPS_DEPENDENTS and store them for a final check to make sure they were
|
|
# eventually defined as being built by some builder
|
|
libdep_libbuilder = self.target[0].builder.get_name(self.env)
|
|
deps_depends = self._get_deps_dependents_with_types(libdep_libbuilder,
|
|
Constants.LibdepsDependents)
|
|
deps_depends += self._get_deps_dependents_with_types("Program",
|
|
Constants.ProgdepsDependents)
|
|
deps_depends = [(_get_node_with_ixes(self.env, dep[0], dep[1]), dep[1])
|
|
for dep in deps_depends]
|
|
self.__class__.dangling_dep_dependents.update(deps_depends)
|
|
|
|
@linter_final_check
|
|
def linter_rule_no_dangling_dep_final_check(self):
|
|
# At this point the SConscripts have defined all the build items,
|
|
# and so we can go check any DEPS_DEPENDENTS listed and make sure a builder
|
|
# was instantiated to build them.
|
|
for dep_dependent in self.__class__.dangling_dep_dependents:
|
|
if not dep_dependent[0].has_builder():
|
|
self._raise_libdep_lint_exception(
|
|
textwrap.dedent(f"""\
|
|
Found reverse dependency linked to node '{dep_dependent[0]}'
|
|
which will never be built by any builder.
|
|
Remove the reverse dependency or add a way to build it."""))
|
|
|
|
@linter_rule
|
|
def linter_rule_no_public_deps(self, libdep):
|
|
"""
|
|
LIBDEP RULE:
|
|
Nodes explicitly marked as not allowed to have public dependencies, should not
|
|
have public dependencies, unless the dependency is explicitly marked as allowed.
|
|
"""
|
|
if not self._check_for_lint_tags('lint-no-public-deps', inclusive_tag=True):
|
|
return
|
|
|
|
if libdep.dependency_type not in (deptype.Global, deptype.Private):
|
|
# Check if the libdep exempts itself from this rule.
|
|
if self._check_for_lint_tags('lint-public-dep-allowed', libdep.target_node.env):
|
|
return
|
|
|
|
target_type = self.target[0].builder.get_name(self.env)
|
|
lib = os.path.basename(str(libdep))
|
|
self._raise_libdep_lint_exception(
|
|
textwrap.dedent(f"""\
|
|
{target_type} '{self.target[0]}' has public dependency '{lib}'
|
|
while being marked as not allowed to have public dependencies
|
|
and '{lib}' does not exempt itself."""))
|
|
|
|
@linter_rule
|
|
def linter_rule_no_dups(self, libdep):
|
|
"""
|
|
LIBDEP RULE:
|
|
A given node shall not link the same LIBDEP across public, private
|
|
or interface dependency types because it is ambiguous and unnecessary.
|
|
"""
|
|
if self._check_for_lint_tags('lint-allow-dup-libdeps'):
|
|
return
|
|
|
|
if str(libdep) in self.unique_libs:
|
|
target_type = self.target[0].builder.get_name(self.env)
|
|
lib = os.path.basename(str(libdep))
|
|
self._raise_libdep_lint_exception(
|
|
f"{target_type} '{self.target[0]}' links '{lib}' multiple times.")
|
|
|
|
self.unique_libs.add(str(libdep))
|
|
|
|
@linter_rule
|
|
def linter_rule_alphabetic_deps(self, libdep):
|
|
"""
|
|
LIBDEP RULE:
|
|
Libdeps shall be listed alphabetically by type in the SCons files.
|
|
"""
|
|
|
|
if self._check_for_lint_tags('lint-allow-non-alphabetic'):
|
|
return
|
|
|
|
# Start checking order after the first item in the list is recorded to compare with.
|
|
if libdep.dependency_type in self._libdeps_types_previous:
|
|
if self._libdeps_types_previous[libdep.dependency_type] > libdep.listed_name:
|
|
target_type = self.target[0].builder.get_name(self.env)
|
|
self._raise_libdep_lint_exception(
|
|
f"{target_type} '{self.target[0]}' has '{libdep.listed_name}' listed in {dep_type_to_env_var[libdep.dependency_type]} out of alphabetical order."
|
|
)
|
|
|
|
self._libdeps_types_previous[libdep.dependency_type] = libdep.listed_name
|
|
|
|
@linter_rule
|
|
def linter_rule_programs_link_private(self, libdep):
|
|
"""
|
|
LIBDEP RULE:
|
|
All Programs shall only have public dependency's
|
|
because a Program will never be a dependency of another Program
|
|
or Library, and LIBDEPS transitiveness does not apply. Public
|
|
transitiveness has no meaning in this case and is used just as default.
|
|
"""
|
|
if self._check_for_lint_tags('lint-allow-program-links-private'):
|
|
return
|
|
|
|
if (self.target[0].builder.get_name(self.env) == "Program"
|
|
and libdep.dependency_type not in (deptype.Global, deptype.Public)):
|
|
|
|
lib = os.path.basename(str(libdep))
|
|
self._raise_libdep_lint_exception(
|
|
textwrap.dedent(f"""\
|
|
Program '{self.target[0]}' links non-public library '{lib}'
|
|
A 'Program' can only have {Constants.Libdeps} libs,
|
|
not {Constants.LibdepsPrivate} or {Constants.LibdepsInterface}."""))
|
|
|
|
@linter_rule
|
|
def linter_rule_no_bidirectional_deps(self, libdep):
|
|
"""
|
|
LIBDEP RULE:
|
|
And Library which issues reverse dependencies, shall not be directly
|
|
linked to by another node, to prevent forward and reverse linkages existing
|
|
at the same node. Instead the content of the library that needs to issue reverse
|
|
dependency needs to be separated from content that needs direct linkage into two
|
|
separate libraries, which can be linked correctly respectively.
|
|
"""
|
|
|
|
if not libdep.target_node.env:
|
|
return
|
|
elif self._check_for_lint_tags('lint-allow-bidirectional-edges', libdep.target_node.env):
|
|
return
|
|
elif len(self._get_deps_dependents(libdep.target_node.env)) > 0:
|
|
|
|
target_type = self.target[0].builder.get_name(self.env)
|
|
lib = os.path.basename(str(libdep))
|
|
self._raise_libdep_lint_exception(
|
|
textwrap.dedent(f"""\
|
|
{target_type} '{self.target[0]}' links directly to a reverse dependency node '{lib}'
|
|
No node can link directly to a node that has {Constants.LibdepsDependents} or {Constants.ProgdepsDependents}."""
|
|
))
|
|
|
|
@linter_rule
|
|
def linter_rule_nonprivate_on_deps_dependents(self, libdep):
|
|
"""
|
|
LIBDEP RULE:
|
|
A Library that issues reverse dependencies, shall not link libraries
|
|
with any kind of transitiveness, and will only link libraries privately.
|
|
This is because functionality that requires reverse dependencies should
|
|
not be transitive.
|
|
"""
|
|
if self._check_for_lint_tags('lint-allow-nonprivate-on-deps-dependents'):
|
|
return
|
|
|
|
if (libdep.dependency_type != deptype.Private and libdep.dependency_type != deptype.Global
|
|
and len(self._get_deps_dependents()) > 0):
|
|
|
|
target_type = self.target[0].builder.get_name(self.env)
|
|
lib = os.path.basename(str(libdep))
|
|
self._raise_libdep_lint_exception(
|
|
textwrap.dedent(f"""\
|
|
{target_type} '{self.target[0]}' links non-private libdep '{lib}' and has a reverse dependency.
|
|
A {target_type} can only have {Constants.LibdepsPrivate} depends if it has {Constants.LibdepsDependents} or {Constants.ProgdepsDependents}."""
|
|
))
|
|
|
|
@linter_rule
|
|
def linter_rule_libdeps_must_be_list(self, libdep):
|
|
"""
|
|
LIBDEP RULE:
|
|
LIBDEPS, LIBDEPS_PRIVATE, and LIBDEPS_INTERFACE must be set as lists in the
|
|
environment.
|
|
"""
|
|
if self._check_for_lint_tags('lint-allow-nonlist-libdeps'):
|
|
return
|
|
|
|
libdeps_vars = list(dep_type_to_env_var.values()) + [
|
|
Constants.LibdepsDependents,
|
|
Constants.ProgdepsDependents,
|
|
]
|
|
|
|
for dep_type_val in libdeps_vars:
|
|
|
|
libdeps_list = self.env.get(dep_type_val, [])
|
|
if not SCons.Util.is_List(libdeps_list):
|
|
|
|
target_type = self.target[0].builder.get_name(self.env)
|
|
self._raise_libdep_lint_exception(
|
|
textwrap.dedent(f"""\
|
|
Found non-list type '{libdeps_list}' while evaluating {dep_type_val[1]} for {target_type} '{self.target[0]}'
|
|
{dep_type_val[1]} must be setup as a list."""))
|
|
|
|
|
|
dependency_visibility_ignored = {
|
|
deptype.Global: deptype.Public,
|
|
deptype.Interface: deptype.Public,
|
|
deptype.Public: deptype.Public,
|
|
deptype.Private: deptype.Public,
|
|
}
|
|
|
|
dependency_visibility_honored = {
|
|
deptype.Global: deptype.Private,
|
|
deptype.Interface: deptype.Interface,
|
|
deptype.Public: deptype.Public,
|
|
deptype.Private: deptype.Private,
|
|
}
|
|
|
|
dep_type_to_env_var = {
|
|
deptype.Global: Constants.LibdepsGlobal,
|
|
deptype.Interface: Constants.LibdepsInterface,
|
|
deptype.Public: Constants.Libdeps,
|
|
deptype.Private: Constants.LibdepsPrivate,
|
|
}
|
|
|
|
|
|
class DependencyCycleError(SCons.Errors.UserError):
|
|
"""Exception representing a cycle discovered in library dependencies."""
|
|
|
|
def __init__(self, first_node):
|
|
super(DependencyCycleError, self).__init__()
|
|
self.cycle_nodes = [first_node]
|
|
|
|
def __str__(self):
|
|
return "Library dependency cycle detected: " + " => ".join(str(n) for n in self.cycle_nodes)
|
|
|
|
|
|
class LibdepLinterError(SCons.Errors.UserError):
|
|
"""Exception representing a discongruent usages of libdeps"""
|
|
|
|
|
|
class MissingSyslibdepError(SCons.Errors.UserError):
|
|
"""Exception representing a discongruent usages of libdeps"""
|
|
|
|
|
|
def _get_sorted_direct_libdeps(node):
|
|
direct_sorted = getattr(node.attributes, "libdeps_direct_sorted", None)
|
|
if direct_sorted is None:
|
|
direct = getattr(node.attributes, "libdeps_direct", [])
|
|
direct_sorted = sorted(direct, key=lambda t: str(t.target_node))
|
|
setattr(node.attributes, "libdeps_direct_sorted", direct_sorted)
|
|
return direct_sorted
|
|
|
|
|
|
class LibdepsVisitationMark(enum.IntEnum):
|
|
UNMARKED = 0
|
|
MARKED_PRIVATE = 1
|
|
MARKED_PUBLIC = 2
|
|
|
|
|
|
def _libdeps_visit_private(n, marked, walking, debug=False):
|
|
if marked[n.target_node] >= LibdepsVisitationMark.MARKED_PRIVATE:
|
|
return
|
|
|
|
if n.target_node in walking:
|
|
raise DependencyCycleError(n.target_node)
|
|
|
|
walking.add(n.target_node)
|
|
|
|
try:
|
|
for child in _get_sorted_direct_libdeps(n.target_node):
|
|
_libdeps_visit_private(child, marked, walking)
|
|
|
|
marked[n.target_node] = LibdepsVisitationMark.MARKED_PRIVATE
|
|
|
|
except DependencyCycleError as e:
|
|
if len(e.cycle_nodes) == 1 or e.cycle_nodes[0] != e.cycle_nodes[-1]:
|
|
e.cycle_nodes.insert(0, n.target_node)
|
|
raise
|
|
|
|
finally:
|
|
walking.remove(n.target_node)
|
|
|
|
|
|
def _libdeps_visit(n, tsorted, marked, walking, debug=False):
|
|
# The marked dictionary tracks which sorts of visitation a node
|
|
# has received. Values for a given node can be UNMARKED/absent,
|
|
# MARKED_PRIVATE, or MARKED_PUBLIC. These are to be interpreted as
|
|
# follows:
|
|
#
|
|
# 0/UNMARKED: Node is not not marked.
|
|
#
|
|
# MARKED_PRIVATE: Node has only been explored as part of looking
|
|
# for cycles under a LIBDEPS_PRIVATE edge.
|
|
#
|
|
# MARKED_PUBLIC: Node has been explored and any of its transiive
|
|
# dependencies have been incorporated into `tsorted`.
|
|
#
|
|
# The __libdeps_visit_private function above will only mark things
|
|
# at with MARKED_PRIVATE, while __libdeps_visit will mark things
|
|
# MARKED_PUBLIC.
|
|
if marked[n.target_node] == LibdepsVisitationMark.MARKED_PUBLIC:
|
|
return
|
|
|
|
# The walking set is used for cycle detection. We record all our
|
|
# predecessors in our depth-first search, and if we observe one of
|
|
# our predecessors as a child, we know we have a cycle.
|
|
if n.target_node in walking:
|
|
raise DependencyCycleError(n.target_node)
|
|
|
|
walking.add(n.target_node)
|
|
|
|
if debug:
|
|
print(f" * {n.dependency_type} => {n.listed_name}")
|
|
|
|
try:
|
|
children = _get_sorted_direct_libdeps(n.target_node)
|
|
|
|
# We first walk all of our public dependencies so that we can
|
|
# put full marks on anything that is in our public transitive
|
|
# graph. We then do a second walk into any private nodes to
|
|
# look for cycles. While we could do just one walk over the
|
|
# children, it is slightly faster to do two passes, since if
|
|
# the algorithm walks into a private edge early, it would do a
|
|
# lot of non-productive (except for cycle checking) walking
|
|
# and marking, but if another public path gets into that same
|
|
# subtree, then it must walk and mark it again to raise it to
|
|
# the public mark level. Whereas, if the algorithm first walks
|
|
# the whole public tree, then those are all productive marks
|
|
# and add to tsorted, and then the private walk will only need
|
|
# to examine those things that are only reachable via private
|
|
# edges.
|
|
|
|
for child in children:
|
|
if child.dependency_type != deptype.Private:
|
|
_libdeps_visit(child, tsorted, marked, walking, debug)
|
|
|
|
for child in children:
|
|
if child.dependency_type == deptype.Private:
|
|
_libdeps_visit_private(child, marked, walking, debug)
|
|
|
|
marked[n.target_node] = LibdepsVisitationMark.MARKED_PUBLIC
|
|
tsorted.append(n.target_node)
|
|
|
|
except DependencyCycleError as e:
|
|
if len(e.cycle_nodes) == 1 or e.cycle_nodes[0] != e.cycle_nodes[-1]:
|
|
e.cycle_nodes.insert(0, n.target_node)
|
|
raise
|
|
|
|
finally:
|
|
walking.remove(n.target_node)
|
|
|
|
|
|
def _get_libdeps(node, debug=False):
|
|
"""Given a SCons Node, return its library dependencies, topologically sorted.
|
|
|
|
Computes the dependencies if they're not already cached.
|
|
"""
|
|
|
|
cache = getattr(node.attributes, Constants.LibdepsCached, None)
|
|
if cache is not None:
|
|
if debug:
|
|
print(" Cache:")
|
|
for dep in cache:
|
|
print(f" * {str(dep)}")
|
|
return cache
|
|
|
|
if debug:
|
|
print(f" Edges:")
|
|
|
|
tsorted = []
|
|
|
|
marked = defaultdict(lambda: LibdepsVisitationMark.UNMARKED)
|
|
walking = set()
|
|
|
|
for child in _get_sorted_direct_libdeps(node):
|
|
if child.dependency_type != deptype.Interface:
|
|
_libdeps_visit(child, tsorted, marked, walking, debug=debug)
|
|
tsorted.reverse()
|
|
|
|
setattr(node.attributes, Constants.LibdepsCached, tsorted)
|
|
return tsorted
|
|
|
|
|
|
def _missing_syslib(name):
|
|
return Constants.MissingLibdep + name
|
|
|
|
|
|
def update_scanner(env, builder_name=None, debug=False):
|
|
"""Update the scanner for "builder" to also scan library dependencies."""
|
|
|
|
builder = env["BUILDERS"][builder_name]
|
|
old_scanner = builder.target_scanner
|
|
|
|
if old_scanner:
|
|
path_function = old_scanner.path_function
|
|
else:
|
|
path_function = None
|
|
|
|
def new_scanner(node, env, path=()):
|
|
if debug:
|
|
print(f"LIBDEPS SCANNER: {str(node)}")
|
|
print(f" Declared dependencies:")
|
|
print(f" global: {env.get(Constants.LibdepsGlobal, None)}")
|
|
print(f" private: {env.get(Constants.LibdepsPrivate, None)}")
|
|
print(f" public: {env.get(Constants.Libdeps, None)}")
|
|
print(f" interface: {env.get(Constants.LibdepsInterface, None)}")
|
|
print(f" no_inherit: {env.get(Constants.LibdepsNoInherit, None)}")
|
|
|
|
if old_scanner:
|
|
result = old_scanner.function(node, env, path)
|
|
else:
|
|
result = []
|
|
result.extend(_get_libdeps(node, debug=debug))
|
|
if debug:
|
|
print(f" Build dependencies:")
|
|
print('\n'.join([' * ' + str(t) for t in result]))
|
|
print('\n')
|
|
return result
|
|
|
|
builder.target_scanner = SCons.Scanner.Scanner(function=new_scanner,
|
|
path_function=path_function)
|
|
|
|
|
|
def get_libdeps(source, target, env, for_signature, debug=False):
|
|
"""Implementation of the special _LIBDEPS environment variable.
|
|
|
|
Expands to the library dependencies for a target.
|
|
"""
|
|
|
|
target = env.Flatten([target])
|
|
return _get_libdeps(target[0], debug=debug)
|
|
|
|
|
|
def get_libdeps_objs(source, target, env, for_signature, debug=False):
|
|
objs = []
|
|
for lib in get_libdeps(source, target, env, for_signature, debug=debug):
|
|
# This relies on Node.sources being order stable build-to-build.
|
|
objs.extend(lib.sources)
|
|
return objs
|
|
|
|
|
|
def stringify_deps(env, deps):
|
|
lib_link_prefix = env.subst("$LIBLINKPREFIX")
|
|
lib_link_suffix = env.subst("$LIBLINKSUFFIX")
|
|
|
|
# Elements of libdeps are either strings (str or unicode), or they're File objects.
|
|
# If they're File objects, they can be passed straight through. If they're strings,
|
|
# they're believed to represent library short names, that should be prefixed with -l
|
|
# or the compiler-specific equivalent. I.e., 'm' becomes '-lm', but 'File("m.a") is passed
|
|
# through whole cloth.
|
|
return [f"{lib_link_prefix}{d}{lib_link_suffix}" if isinstance(d, str) else d for d in deps]
|
|
|
|
|
|
def get_syslibdeps(source, target, env, for_signature, debug=False, shared=True):
|
|
""" Given a SCons Node, return its system library dependencies.
|
|
|
|
These are the dependencies listed with SYSLIBDEPS, and are linked using -l.
|
|
"""
|
|
|
|
deps = getattr(target[0].attributes, Constants.SysLibdepsCached, None)
|
|
if deps is None:
|
|
|
|
# Get the syslibdeps for the current node
|
|
deps = target[0].get_env().Flatten(
|
|
copy.copy(target[0].get_env().get(Constants.SysLibdepsPrivate)) or [])
|
|
deps += target[0].get_env().Flatten(target[0].get_env().get(Constants.SysLibdeps) or [])
|
|
|
|
for lib in _get_libdeps(target[0]):
|
|
|
|
# For each libdep get its syslibdeps, and then check to see if we can
|
|
# add it to the deps list. For static build we will also include private
|
|
# syslibdeps to be transitive. For a dynamic build we will only make
|
|
# public libdeps transitive.
|
|
syslibs = []
|
|
if not shared:
|
|
syslibs += lib.get_env().get(Constants.SysLibdepsPrivate) or []
|
|
syslibs += lib.get_env().get(Constants.SysLibdeps) or []
|
|
|
|
# Validate the libdeps, a configure check has already checked what
|
|
# syslibdeps are available so we can hard fail here if a syslibdep
|
|
# is being attempted to be linked with.
|
|
for syslib in syslibs:
|
|
if not syslib:
|
|
continue
|
|
|
|
if isinstance(syslib, str) and syslib.startswith(Constants.MissingLibdep):
|
|
raise MissingSyslibdepError(
|
|
textwrap.dedent(f"""\
|
|
LibdepsError:
|
|
Target '{str(target[0])}' depends on the availability of a
|
|
system provided library for '{syslib[len(Constants.MissingLibdep):]}',
|
|
but no suitable library was found during configuration."""))
|
|
|
|
deps.append(syslib)
|
|
|
|
setattr(target[0].attributes, Constants.SysLibdepsCached, deps)
|
|
return stringify_deps(env, deps)
|
|
|
|
|
|
def _append_direct_libdeps(node, prereq_nodes):
|
|
# We do not bother to decorate nodes that are not actual Objects
|
|
if type(node) == str:
|
|
return
|
|
if getattr(node.attributes, "libdeps_direct", None) is None:
|
|
node.attributes.libdeps_direct = []
|
|
node.attributes.libdeps_direct.extend(prereq_nodes)
|
|
|
|
|
|
def _get_libdeps_with_link_flags(source, target, env, for_signature):
|
|
for lib in get_libdeps(source, target, env, for_signature):
|
|
# Make sure lib is a Node so we can get the env to check for flags.
|
|
libnode = lib
|
|
if not isinstance(lib, (str, SCons.Node.FS.File, SCons.Node.FS.Entry)):
|
|
libnode = env.File(lib)
|
|
|
|
# Virtual libdeps don't appear on the link line
|
|
if 'virtual-libdep' in libnode.get_env().get('LIBDEPS_TAGS', []):
|
|
continue
|
|
|
|
# Create a libdep and parse the prefix and postfix (and separators if any)
|
|
# flags from the environment.
|
|
cur_lib = FlaggedLibdep(libnode, env)
|
|
yield cur_lib
|
|
|
|
|
|
def _get_node_with_ixes(env, node, node_builder_type):
|
|
"""
|
|
Gets the node passed in node with the correct ixes applied
|
|
for the given builder type.
|
|
"""
|
|
|
|
if not node:
|
|
return node
|
|
|
|
node_builder = env["BUILDERS"][node_builder_type]
|
|
node_factory = node_builder.target_factory or env.File
|
|
|
|
# Cache the 'ixes' in a function scope global so we don't need
|
|
# to run SCons performance intensive 'subst' each time
|
|
cache_key = (id(env), node_builder_type)
|
|
try:
|
|
prefix, suffix = _get_node_with_ixes.node_type_ixes[cache_key]
|
|
except KeyError:
|
|
prefix = node_builder.get_prefix(env)
|
|
suffix = node_builder.get_suffix(env)
|
|
|
|
# TODO(SERVER-50681): Find a way to do this that doesn't hard
|
|
# code these extensions. See the code review for SERVER-27507
|
|
# for additional discussion.
|
|
if suffix == ".dll":
|
|
suffix = ".lib"
|
|
|
|
_get_node_with_ixes.node_type_ixes[cache_key] = (prefix, suffix)
|
|
|
|
node_with_ixes = SCons.Util.adjustixes(node, prefix, suffix)
|
|
return node_factory(node_with_ixes)
|
|
|
|
|
|
_get_node_with_ixes.node_type_ixes = dict()
|
|
|
|
|
|
def add_node_from(env, node):
|
|
|
|
env.GetLibdepsGraph().add_nodes_from([(
|
|
node.abspath,
|
|
{
|
|
NodeProps.bin_type.name:
|
|
node.builder.get_name(env) if node.has_builder() else "Unknown",
|
|
},
|
|
)])
|
|
|
|
|
|
def add_edge_from(env, from_node, to_node, visibility, direct):
|
|
|
|
add_node_from(env, from_node)
|
|
add_node_from(env, to_node)
|
|
|
|
env.GetLibdepsGraph().add_edges_from([(
|
|
from_node.abspath,
|
|
to_node.abspath,
|
|
{
|
|
EdgeProps.direct.name: direct,
|
|
EdgeProps.visibility.name: int(visibility),
|
|
},
|
|
)])
|
|
|
|
|
|
def add_libdeps_node(env, target, libdeps):
|
|
|
|
if str(target).endswith(env["SHLIBSUFFIX"]):
|
|
node = _get_node_with_ixes(env, str(target.abspath), target.get_builder().get_name(env))
|
|
add_node_from(env, node)
|
|
|
|
for libdep in libdeps:
|
|
if str(libdep.target_node).endswith(env["SHLIBSUFFIX"]):
|
|
add_edge_from(
|
|
env,
|
|
node,
|
|
libdep.target_node,
|
|
visibility=libdep.dependency_type,
|
|
direct=True,
|
|
)
|
|
|
|
|
|
def get_libdeps_nodes(env, target, builder, debug=False, visibility_map=None):
|
|
if visibility_map is None:
|
|
visibility_map = dependency_visibility_ignored
|
|
|
|
if not SCons.Util.is_List(target):
|
|
target = [target]
|
|
|
|
# Get the current list of nodes not to inherit on each target
|
|
no_inherit = set(env.get(Constants.LibdepsNoInherit, []))
|
|
|
|
# Get all the libdeps from the env so we can
|
|
# can append them to the current target_node.
|
|
libdeps = []
|
|
for dep_type in sorted(visibility_map.keys()):
|
|
|
|
if dep_type == deptype.Global:
|
|
if any("conftest" in str(t) for t in target):
|
|
# Ignore global dependencies for conftests
|
|
continue
|
|
|
|
# Libraries may not be stored as a list in the env,
|
|
# so we must convert single library strings to a list.
|
|
libs = env.get(dep_type_to_env_var[dep_type], []).copy()
|
|
if not SCons.Util.is_List(libs):
|
|
libs = [libs]
|
|
|
|
for lib in libs:
|
|
if not lib:
|
|
continue
|
|
|
|
lib_with_ixes = _get_node_with_ixes(env, lib, builder)
|
|
|
|
if lib in no_inherit:
|
|
if debug and not any("conftest" in str(t) for t in target):
|
|
print(f" {dep_type[1]} =/> {lib}")
|
|
|
|
else:
|
|
if debug and not any("conftest" in str(t) for t in target):
|
|
print(f" {dep_type[1]} => {lib}")
|
|
|
|
libdeps.append(dependency(lib_with_ixes, dep_type, lib))
|
|
|
|
return libdeps
|
|
|
|
|
|
def libdeps_emitter(target, source, env, debug=False, builder=None, visibility_map=None,
|
|
ignore_progdeps=False):
|
|
"""SCons emitter that takes values from the LIBDEPS environment variable and
|
|
converts them to File node objects, binding correct path information into
|
|
those File objects.
|
|
|
|
Emitters run on a particular "target" node during the initial execution of
|
|
the SConscript file, rather than during the later build phase. When they
|
|
run, the "env" environment's working directory information is what you
|
|
expect it to be -- that is, the working directory is considered to be the
|
|
one that contains the SConscript file. This allows specification of
|
|
relative paths to LIBDEPS elements.
|
|
|
|
This emitter also adds LIBSUFFIX and LIBPREFIX appropriately.
|
|
|
|
NOTE: For purposes of LIBDEPS_DEPENDENTS propagation, only the first member
|
|
of the "target" list is made a prerequisite of the elements of LIBDEPS_DEPENDENTS.
|
|
"""
|
|
|
|
if visibility_map is None:
|
|
visibility_map = dependency_visibility_ignored
|
|
|
|
if debug and not any("conftest" in str(t) for t in target):
|
|
print(f"LIBDEPS EMITTER: {str(target[0])}")
|
|
print(f" Declared dependencies:")
|
|
print(f" global: {env.get(Constants.LibdepsGlobal, None)}")
|
|
print(f" private: {env.get(Constants.LibdepsPrivate, None)}")
|
|
print(f" public: {env.get(Constants.Libdeps, None)}")
|
|
print(f" interface: {env.get(Constants.LibdepsInterface, None)}")
|
|
print(f" no_inherit: {env.get(Constants.LibdepsNoInherit, None)}")
|
|
print(f" Edges:")
|
|
|
|
libdeps = get_libdeps_nodes(env, target, builder, debug, visibility_map)
|
|
|
|
if debug and not any("conftest" in str(t) for t in target):
|
|
print(f"\n")
|
|
|
|
# Lint the libdeps to make sure they are following the rules.
|
|
# This will skip some or all of the checks depending on the options
|
|
# and LIBDEPS_TAGS used.
|
|
if not any("conftest" in str(t) for t in target):
|
|
LibdepLinter(env, target).lint_libdeps(libdeps)
|
|
|
|
if env.get('SYMBOLDEPSSUFFIX', None):
|
|
for t in target:
|
|
add_libdeps_node(env, t, libdeps)
|
|
|
|
# We ignored the visibility_map until now because we needed to use
|
|
# original dependency value for linting. Now go back through and
|
|
# use the map to convert to the desired dependencies, for example
|
|
# all Public in the static linking case.
|
|
for libdep in libdeps:
|
|
libdep.dependency_type = visibility_map[libdep.dependency_type]
|
|
|
|
for t in target:
|
|
# target[0] must be a Node and not a string, or else libdeps will fail to
|
|
# work properly.
|
|
_append_direct_libdeps(t, libdeps)
|
|
|
|
for dependent in env.get(Constants.LibdepsDependents, []):
|
|
if dependent is None:
|
|
continue
|
|
|
|
visibility = deptype.Private
|
|
if isinstance(dependent, tuple):
|
|
visibility = dependent[1]
|
|
dependent = dependent[0]
|
|
|
|
dependentNode = _get_node_with_ixes(env, dependent, builder)
|
|
_append_direct_libdeps(dependentNode,
|
|
[dependency(target[0], visibility_map[visibility], dependent)])
|
|
|
|
if not ignore_progdeps:
|
|
for dependent in env.get(Constants.ProgdepsDependents, []):
|
|
if dependent is None:
|
|
continue
|
|
|
|
visibility = deptype.Public
|
|
if isinstance(dependent, tuple):
|
|
# TODO: Error here? Non-public PROGDEPS_DEPENDENTS probably are meaningless
|
|
visibility = dependent[1]
|
|
dependent = dependent[0]
|
|
|
|
dependentNode = _get_node_with_ixes(env, dependent, "Program")
|
|
_append_direct_libdeps(dependentNode,
|
|
[dependency(target[0], visibility_map[visibility], dependent)])
|
|
|
|
return target, source
|
|
|
|
|
|
def expand_libdeps_tags(source, target, env, for_signature):
|
|
results = []
|
|
for expansion in env.get(Constants.LibdepsTagExpansion, []):
|
|
results.append(expansion(source, target, env, for_signature))
|
|
return results
|
|
|
|
|
|
def expand_libdeps_for_link(source, target, env, for_signature):
|
|
|
|
libdeps_with_flags = []
|
|
|
|
# Used to make modifications to the previous libdep on the link line
|
|
# if needed. An empty class here will make the switch_flag conditionals
|
|
# below a bit cleaner.
|
|
prev_libdep = None
|
|
|
|
for flagged_libdep in _get_libdeps_with_link_flags(source, target, env, for_signature):
|
|
|
|
# If there are no flags to process we can move on to the next lib.
|
|
# start_index wont mater in the case because if there are no flags
|
|
# on the previous lib, then we will never need to do the chopping
|
|
# mechanism on the next iteration.
|
|
if not flagged_libdep.prefix_flags and not flagged_libdep.postfix_flags:
|
|
libdeps_with_flags.append(str(flagged_libdep))
|
|
prev_libdep = flagged_libdep
|
|
continue
|
|
|
|
# This for loop will go through the previous results and remove the 'off'
|
|
# flag as well as removing the new 'on' flag. For example, let libA and libB
|
|
# both use on and off flags which would normally generate on the link line as:
|
|
# -Wl--on-flag libA.a -Wl--off-flag -Wl--on-flag libA.a -Wl--off-flag
|
|
# This loop below will spot the cases were the flag was turned off and then
|
|
# immediately turned back on
|
|
for switch_flag in getattr(flagged_libdep.libnode.attributes, 'libdeps_switch_flags', []):
|
|
if (prev_libdep and switch_flag['on'] in flagged_libdep.prefix_flags
|
|
and switch_flag['off'] in prev_libdep.postfix_flags):
|
|
|
|
flagged_libdep.prefix_flags.remove(switch_flag['on'])
|
|
prev_libdep.postfix_flags.remove(switch_flag['off'])
|
|
|
|
# prev_lib has had its list modified, and it has a start index
|
|
# from the last iteration, so it will chop of the end the current
|
|
# list and reapply the end with the new flags.
|
|
prev_libdep.add_lib_to_result_list(libdeps_with_flags)
|
|
|
|
# Store the information of the len of the current list before adding
|
|
# the next set of flags as that will be the start index for the previous
|
|
# lib next time around in case there are any switch flags to chop off.
|
|
start_index = len(libdeps_with_flags)
|
|
flagged_libdep.add_lib_to_result_list(libdeps_with_flags)
|
|
|
|
# Done processing the current lib, so set it to previous for the next iteration.
|
|
prev_libdep = flagged_libdep
|
|
prev_libdep.start_index = start_index
|
|
|
|
return libdeps_with_flags
|
|
|
|
|
|
def generate_libdeps_graph(env):
|
|
if env.get('SYMBOLDEPSSUFFIX', None):
|
|
|
|
find_symbols = env.Dir("$BUILD_DIR").path + "/libdeps/find_symbols"
|
|
libdeps_graph = env.GetLibdepsGraph()
|
|
|
|
symbol_deps = []
|
|
for symbols_file, target_node in env.get('LIBDEPS_SYMBOL_DEP_FILES', []):
|
|
|
|
direct_libdeps = []
|
|
for direct_libdep in _get_sorted_direct_libdeps(target_node):
|
|
add_node_from(env, direct_libdep.target_node)
|
|
add_edge_from(
|
|
env,
|
|
target_node,
|
|
direct_libdep.target_node,
|
|
visibility=int(direct_libdep.dependency_type),
|
|
direct=True,
|
|
)
|
|
direct_libdeps.append(direct_libdep.target_node.abspath)
|
|
|
|
for libdep in _get_libdeps(target_node):
|
|
if libdep.abspath not in direct_libdeps:
|
|
add_node_from(env, libdep)
|
|
add_edge_from(
|
|
env,
|
|
target_node,
|
|
libdep,
|
|
visibility=int(deptype.Public),
|
|
direct=False,
|
|
)
|
|
if env['PLATFORM'] == 'darwin':
|
|
sep = ' '
|
|
else:
|
|
sep = ':'
|
|
ld_path = sep.join(
|
|
[os.path.dirname(str(libdep)) for libdep in _get_libdeps(target_node)])
|
|
symbol_deps.append(
|
|
env.Command(
|
|
target=symbols_file,
|
|
source=target_node,
|
|
action=SCons.Action.Action(
|
|
f'{find_symbols} $SOURCE "{ld_path}" $TARGET',
|
|
"Generating $SOURCE symbol dependencies" if not env['VERBOSE'] else ""),
|
|
))
|
|
|
|
def write_graph_hash(env, target, source):
|
|
|
|
with open(target[0].path, 'w') as f:
|
|
json_str = json.dumps(
|
|
networkx.readwrite.json_graph.node_link_data(env.GetLibdepsGraph()),
|
|
sort_keys=True).encode('utf-8')
|
|
f.write(hashlib.sha256(json_str).hexdigest())
|
|
|
|
graph_hash = env.Command(
|
|
target="$BUILD_DIR/libdeps/graph_hash.sha256",
|
|
source=symbol_deps,
|
|
action=SCons.Action.FunctionAction(
|
|
write_graph_hash,
|
|
{"cmdstr": None},
|
|
),
|
|
)
|
|
env.Depends(
|
|
graph_hash,
|
|
[env.File("#SConstruct")] + glob.glob("**/SConscript", recursive=True) +
|
|
[os.path.abspath(__file__),
|
|
env.File('$BUILD_DIR/mongo/util/version_constants.h')],
|
|
)
|
|
|
|
graph_node = env.Command(
|
|
target=env.get('LIBDEPS_GRAPH_FILE', None),
|
|
source=symbol_deps,
|
|
action=SCons.Action.FunctionAction(
|
|
generate_graph,
|
|
{"cmdstr": "Generating libdeps graph"},
|
|
),
|
|
)
|
|
|
|
env.Depends(graph_node, [graph_hash] + env.Glob("#buildscripts/libdeps/libdeps/*"))
|
|
|
|
|
|
def generate_graph(env, target, source):
|
|
|
|
libdeps_graph = env.GetLibdepsGraph()
|
|
|
|
demangled_symbols = {}
|
|
for symbol_deps_file in source:
|
|
|
|
with open(str(symbol_deps_file)) as f:
|
|
symbols = {}
|
|
try:
|
|
for symbol, lib in json.load(f).items():
|
|
# ignore symbols from external libraries,
|
|
# they will just clutter the graph
|
|
if lib.startswith(env.Dir("$BUILD_DIR").path):
|
|
if lib not in symbols:
|
|
symbols[lib] = []
|
|
symbols[lib].append(symbol)
|
|
except json.JSONDecodeError:
|
|
env.FatalError(f"Failed processing json file: {str(symbol_deps_file)}")
|
|
|
|
demangled_symbols[str(symbol_deps_file)] = symbols
|
|
|
|
p1 = subprocess.Popen(['c++filt', '-n'], stdin=subprocess.PIPE, stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT)
|
|
stdout, stderr = p1.communicate(json.dumps(demangled_symbols).encode('utf-8'))
|
|
demangled_symbols = json.loads(stdout.decode("utf-8"))
|
|
|
|
for deps_file in demangled_symbols:
|
|
|
|
for libdep in demangled_symbols[deps_file]:
|
|
|
|
from_node = os.path.abspath(str(deps_file)[:-len(env['SYMBOLDEPSSUFFIX'])])
|
|
to_node = os.path.abspath(libdep).strip()
|
|
libdeps_graph.add_edges_from([(
|
|
from_node,
|
|
to_node,
|
|
{EdgeProps.symbols.name: "\n".join(demangled_symbols[deps_file][libdep])},
|
|
)])
|
|
node = env.File(str(deps_file)[:-len(env['SYMBOLDEPSSUFFIX'])])
|
|
add_node_from(env, node)
|
|
|
|
libdeps_graph_file = f"{env.Dir('$BUILD_DIR').path}/libdeps/libdeps.graphml"
|
|
networkx.write_graphml(libdeps_graph, libdeps_graph_file, named_key_ids=True)
|
|
with fileinput.FileInput(libdeps_graph_file, inplace=True) as file:
|
|
for line in file:
|
|
print(line.replace(str(env.Dir("$BUILD_DIR").abspath + os.sep), ''), end='')
|
|
|
|
|
|
def setup_environment(env, emitting_shared=False, debug='off', linting='on'):
|
|
"""Set up the given build environment to do LIBDEPS tracking."""
|
|
|
|
LibdepLinter.skip_linting = linting == 'off'
|
|
LibdepLinter.print_linter_errors = linting == 'print'
|
|
|
|
try:
|
|
env["_LIBDEPS"]
|
|
except KeyError:
|
|
env["_LIBDEPS"] = "$_LIBDEPS_LIBS"
|
|
|
|
env["_LIBDEPS_TAGS"] = expand_libdeps_tags
|
|
env["_LIBDEPS_GET_LIBS"] = partial(get_libdeps, debug=debug)
|
|
env["_LIBDEPS_OBJS"] = partial(get_libdeps_objs, debug=debug)
|
|
env["_SYSLIBDEPS"] = partial(get_syslibdeps, debug=debug, shared=emitting_shared)
|
|
|
|
env[Constants.Libdeps] = SCons.Util.CLVar()
|
|
env[Constants.SysLibdeps] = SCons.Util.CLVar()
|
|
|
|
# Create the alias for graph generation, the existence of this alias
|
|
# on the command line will cause the libdeps-graph generation to be
|
|
# configured.
|
|
env['LIBDEPS_GRAPH_ALIAS'] = env.Alias(
|
|
'generate-libdeps-graph',
|
|
"${BUILD_DIR}/libdeps/libdeps.graphml",
|
|
)[0]
|
|
|
|
if str(env['LIBDEPS_GRAPH_ALIAS']) in COMMAND_LINE_TARGETS:
|
|
|
|
# Detect if the current system has the tools to perform the generation.
|
|
if env.GetOption('ninja') != 'disabled':
|
|
env.FatalError("Libdeps graph generation is not supported with ninja builds.")
|
|
if not emitting_shared:
|
|
env.FatalError("Libdeps graph generation currently only supports dynamic builds.")
|
|
|
|
if env['PLATFORM'] == 'darwin':
|
|
required_bins = ['awk', 'sed', 'otool', 'nm']
|
|
else:
|
|
required_bins = ['awk', 'grep', 'ldd', 'nm']
|
|
for bin in required_bins:
|
|
if not env.WhereIs(bin):
|
|
env.FatalError(f"'{bin}' not found, Libdeps graph generation requires {bin}.")
|
|
|
|
# The find_symbols binary is a small fast C binary which will extract the missing
|
|
# symbols from the target library, and discover what linked libraries supply it. This
|
|
# setups the binary to be built.
|
|
find_symbols_env = env.Clone()
|
|
find_symbols_env.VariantDir('${BUILD_DIR}/libdeps', 'buildscripts/libdeps', duplicate=0)
|
|
find_symbols_node = find_symbols_env.Program(
|
|
target='${BUILD_DIR}/libdeps/find_symbols',
|
|
source=['${BUILD_DIR}/libdeps/find_symbols.c'],
|
|
CFLAGS=['-O3'],
|
|
)
|
|
|
|
# Here we are setting up some functions which will return single instance of the
|
|
# network graph and symbol deps list. We also setup some environment variables
|
|
# which are used along side the functions.
|
|
symbol_deps = []
|
|
|
|
def append_symbol_deps(env, symbol_deps_file):
|
|
env.Depends(env['LIBDEPS_GRAPH_FILE'], symbol_deps_file[0])
|
|
symbol_deps.append(symbol_deps_file)
|
|
|
|
env.AddMethod(append_symbol_deps, "AppendSymbolDeps")
|
|
|
|
env['LIBDEPS_SYMBOL_DEP_FILES'] = symbol_deps
|
|
env['LIBDEPS_GRAPH_FILE'] = env.File("${BUILD_DIR}/libdeps/libdeps.graphml")
|
|
env['LIBDEPS_GRAPH_SCHEMA_VERSION'] = 5
|
|
env["SYMBOLDEPSSUFFIX"] = '.symbol_deps'
|
|
|
|
libdeps_graph = LibdepsGraph()
|
|
libdeps_graph.graph['invocation'] = " ".join([env['ESCAPE'](str(sys.executable))] +
|
|
[env['ESCAPE'](arg) for arg in sys.argv])
|
|
libdeps_graph.graph['git_hash'] = env['MONGO_GIT_HASH']
|
|
libdeps_graph.graph['graph_schema_version'] = env['LIBDEPS_GRAPH_SCHEMA_VERSION']
|
|
libdeps_graph.graph['build_dir'] = env.Dir('$BUILD_DIR').path
|
|
libdeps_graph.graph['deptypes'] = json.dumps({
|
|
key: value[0]
|
|
for key, value in deptype.__members__.items() if isinstance(value, tuple)
|
|
})
|
|
|
|
def get_libdeps_graph(env):
|
|
return libdeps_graph
|
|
|
|
env.AddMethod(get_libdeps_graph, "GetLibdepsGraph")
|
|
|
|
# Now we will setup an emitter, and an additional action for several
|
|
# of the builder involved with dynamic builds.
|
|
def libdeps_graph_emitter(target, source, env):
|
|
if "conftest" not in str(target[0]):
|
|
symbol_deps_file = env.File(str(target[0]) + env['SYMBOLDEPSSUFFIX'])
|
|
env.Depends(symbol_deps_file, '${BUILD_DIR}/libdeps/find_symbols')
|
|
env.AppendSymbolDeps((symbol_deps_file, target[0]))
|
|
|
|
return target, source
|
|
|
|
for builder_name in ("Program", "SharedLibrary", "LoadableModule"):
|
|
builder = env['BUILDERS'][builder_name]
|
|
base_emitter = builder.emitter
|
|
new_emitter = SCons.Builder.ListEmitter([base_emitter, libdeps_graph_emitter])
|
|
builder.emitter = new_emitter
|
|
|
|
env.Append(
|
|
LIBDEPS_LIBEMITTER=partial(
|
|
libdeps_emitter,
|
|
debug=debug,
|
|
builder="StaticLibrary",
|
|
),
|
|
LIBEMITTER=lambda target, source, env: env["LIBDEPS_LIBEMITTER"](target, source, env),
|
|
LIBDEPS_SHAREMITTER=partial(
|
|
libdeps_emitter,
|
|
debug=debug,
|
|
builder="SharedArchive",
|
|
ignore_progdeps=True,
|
|
),
|
|
SHAREMITTER=lambda target, source, env: env["LIBDEPS_SHAREMITTER"](target, source, env),
|
|
LIBDEPS_SHLIBEMITTER=partial(
|
|
libdeps_emitter,
|
|
debug=debug,
|
|
builder="SharedLibrary",
|
|
visibility_map=dependency_visibility_honored,
|
|
),
|
|
SHLIBEMITTER=lambda target, source, env: env["LIBDEPS_SHLIBEMITTER"](target, source, env),
|
|
LIBDEPS_PROGEMITTER=partial(
|
|
libdeps_emitter,
|
|
debug=debug,
|
|
builder="SharedLibrary" if emitting_shared else "StaticLibrary",
|
|
),
|
|
PROGEMITTER=lambda target, source, env: env["LIBDEPS_PROGEMITTER"](target, source, env),
|
|
)
|
|
|
|
env["_LIBDEPS_LIBS_FOR_LINK"] = expand_libdeps_for_link
|
|
|
|
env["_LIBDEPS_LIBS"] = ("$LINK_LIBGROUP_START "
|
|
"$_LIBDEPS_LIBS_FOR_LINK "
|
|
"$LINK_LIBGROUP_END ")
|
|
|
|
env.Prepend(_LIBFLAGS="$_LIBDEPS_TAGS $_LIBDEPS $_SYSLIBDEPS ")
|
|
for builder_name in ("Program", "SharedLibrary", "LoadableModule", "SharedArchive"):
|
|
try:
|
|
update_scanner(env, builder_name, debug=debug)
|
|
except KeyError:
|
|
pass
|
|
|
|
|
|
def setup_conftests(conf):
|
|
def FindSysLibDep(context, name, libs, **kwargs):
|
|
var = "LIBDEPS_" + name.upper() + "_SYSLIBDEP"
|
|
kwargs["autoadd"] = False
|
|
for lib in libs:
|
|
result = context.sconf.CheckLib(lib, **kwargs)
|
|
context.did_show_result = 1
|
|
if result:
|
|
context.env[var] = lib
|
|
context.Result(result)
|
|
return result
|
|
context.env[var] = _missing_syslib(name)
|
|
context.Result(result)
|
|
return result
|
|
|
|
conf.AddTest("FindSysLibDep", FindSysLibDep)
|