| # Copyright 2018 The Bazel Authors. All rights reserved. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| """Functionality for constructing actions that invoke the Rust compiler""" |
| |
| load("@bazel_skylib//rules:common_settings.bzl", "BuildSettingInfo") |
| load( |
| "@bazel_tools//tools/build_defs/cc:action_names.bzl", |
| "CPP_LINK_DYNAMIC_LIBRARY_ACTION_NAME", |
| "CPP_LINK_EXECUTABLE_ACTION_NAME", |
| "CPP_LINK_NODEPS_DYNAMIC_LIBRARY_ACTION_NAME", |
| "CPP_LINK_STATIC_LIBRARY_ACTION_NAME", |
| ) |
| load("//rust/private:common.bzl", "rust_common") |
| load("//rust/private:providers.bzl", "RustcOutputDiagnosticsInfo", _BuildInfo = "BuildInfo") |
| load("//rust/private:stamp.bzl", "is_stamping_enabled") |
| load( |
| "//rust/private:utils.bzl", |
| "abs", |
| "expand_dict_value_locations", |
| "expand_list_element_locations", |
| "find_cc_toolchain", |
| "get_lib_name_default", |
| "get_lib_name_for_windows", |
| "get_preferred_artifact", |
| "is_exec_configuration", |
| "make_static_lib_symlink", |
| "relativize", |
| ) |
| load(":utils.bzl", "is_std_dylib") |
| |
| # This feature is disabled unless one of the dependencies is a cc_library. |
| # Authors of C++ toolchains can place linker flags that should only be applied |
| # when linking with C objects in a feature with this name, or require this |
| # feature from other features which needs to be disabled together. |
| RUST_LINK_CC_FEATURE = "rules_rust_link_cc" |
| |
| BuildInfo = _BuildInfo |
| |
| AliasableDepInfo = provider( |
| doc = "A provider mapping an alias name to a Crate's information.", |
| fields = { |
| "dep": "CrateInfo", |
| "name": "str", |
| }, |
| ) |
| |
| _error_format_values = ["human", "json", "short"] |
| |
| ErrorFormatInfo = provider( |
| doc = "Set the --error-format flag for all rustc invocations", |
| fields = {"error_format": "(string) [" + ", ".join(_error_format_values) + "]"}, |
| ) |
| |
| ExtraRustcFlagsInfo = provider( |
| doc = "Pass each value as an additional flag to non-exec rustc invocations", |
| fields = {"extra_rustc_flags": "List[string] Extra flags to pass to rustc in non-exec configuration"}, |
| ) |
| |
| ExtraExecRustcFlagsInfo = provider( |
| doc = "Pass each value as an additional flag to exec rustc invocations", |
| fields = {"extra_exec_rustc_flags": "List[string] Extra flags to pass to rustc in exec configuration"}, |
| ) |
| |
| PerCrateRustcFlagsInfo = provider( |
| doc = "Pass each value as an additional flag to non-exec rustc invocations for crates matching the provided filter", |
| fields = {"per_crate_rustc_flags": "List[string] Extra flags to pass to rustc in non-exec configuration"}, |
| ) |
| |
| IsProcMacroDepInfo = provider( |
| doc = "Records if this is a transitive dependency of a proc-macro.", |
| fields = {"is_proc_macro_dep": "Boolean"}, |
| ) |
| |
| def _is_proc_macro_dep_impl(ctx): |
| return IsProcMacroDepInfo(is_proc_macro_dep = ctx.build_setting_value) |
| |
| is_proc_macro_dep = rule( |
| doc = "Records if this is a transitive dependency of a proc-macro.", |
| implementation = _is_proc_macro_dep_impl, |
| build_setting = config.bool(flag = True), |
| ) |
| |
| IsProcMacroDepEnabledInfo = provider( |
| doc = "Enables the feature to record if a library is a transitive dependency of a proc-macro.", |
| fields = {"enabled": "Boolean"}, |
| ) |
| |
| def _is_proc_macro_dep_enabled_impl(ctx): |
| return IsProcMacroDepEnabledInfo(enabled = ctx.build_setting_value) |
| |
| is_proc_macro_dep_enabled = rule( |
| doc = "Enables the feature to record if a library is a transitive dependency of a proc-macro.", |
| implementation = _is_proc_macro_dep_enabled_impl, |
| build_setting = config.bool(flag = True), |
| ) |
| |
| def _get_rustc_env(attr, toolchain, crate_name): |
| """Gathers rustc environment variables |
| |
| Args: |
| attr (struct): The current target's attributes |
| toolchain (rust_toolchain): The current target's rust toolchain context |
| crate_name (str): The name of the crate to be compiled |
| |
| Returns: |
| dict: Rustc environment variables |
| """ |
| version = attr.version if hasattr(attr, "version") else "0.0.0" |
| major, minor, patch = version.split(".", 2) |
| if "-" in patch: |
| patch, pre = patch.split("-", 1) |
| else: |
| pre = "" |
| |
| result = { |
| "CARGO_CFG_TARGET_ARCH": "" if toolchain.target_arch == None else toolchain.target_arch, |
| "CARGO_CFG_TARGET_OS": "" if toolchain.target_os == None else toolchain.target_os, |
| "CARGO_CRATE_NAME": crate_name, |
| "CARGO_PKG_AUTHORS": "", |
| "CARGO_PKG_DESCRIPTION": "", |
| "CARGO_PKG_HOMEPAGE": "", |
| "CARGO_PKG_NAME": attr.name, |
| "CARGO_PKG_VERSION": version, |
| "CARGO_PKG_VERSION_MAJOR": major, |
| "CARGO_PKG_VERSION_MINOR": minor, |
| "CARGO_PKG_VERSION_PATCH": patch, |
| "CARGO_PKG_VERSION_PRE": pre, |
| } |
| if hasattr(attr, "_is_proc_macro_dep_enabled") and attr._is_proc_macro_dep_enabled[IsProcMacroDepEnabledInfo].enabled: |
| is_proc_macro_dep = "0" |
| if hasattr(attr, "_is_proc_macro_dep") and attr._is_proc_macro_dep[IsProcMacroDepInfo].is_proc_macro_dep: |
| is_proc_macro_dep = "1" |
| result["BAZEL_RULES_RUST_IS_PROC_MACRO_DEP"] = is_proc_macro_dep |
| return result |
| |
| def get_compilation_mode_opts(ctx, toolchain): |
| """Gathers rustc flags for the current compilation mode (opt/debug) |
| |
| Args: |
| ctx (ctx): The current rule's context object |
| toolchain (rust_toolchain): The current rule's `rust_toolchain` |
| |
| Returns: |
| struct: See `_rust_toolchain_impl` for more details |
| """ |
| comp_mode = ctx.var["COMPILATION_MODE"] |
| if not comp_mode in toolchain.compilation_mode_opts: |
| fail("Unrecognized compilation mode {} for toolchain.".format(comp_mode)) |
| |
| return toolchain.compilation_mode_opts[comp_mode] |
| |
| def _are_linkstamps_supported(feature_configuration, has_grep_includes): |
| # Are linkstamps supported by the C++ toolchain? |
| return (cc_common.is_enabled(feature_configuration = feature_configuration, feature_name = "linkstamps") and |
| # Is Bazel recent enough to support Starlark linkstamps? |
| hasattr(cc_common, "register_linkstamp_compile_action") and |
| # The current rule doesn't define _grep_includes attribute; this |
| # attribute is required for compiling linkstamps. |
| has_grep_includes) |
| |
| def _should_use_pic(cc_toolchain, feature_configuration, crate_type, compilation_mode): |
| """Whether or not [PIC][pic] should be enabled |
| |
| [pic]: https://en.wikipedia.org/wiki/Position-independent_code |
| |
| Args: |
| cc_toolchain (CcToolchainInfo): The current `cc_toolchain`. |
| feature_configuration (FeatureConfiguration): Feature configuration to be queried. |
| crate_type (str): A Rust target's crate type. |
| compilation_mode: The compilation mode. |
| |
| Returns: |
| bool: Whether or not [PIC][pic] should be enabled. |
| """ |
| |
| # We use the same logic to select between `pic` and `nopic` outputs as the C++ rules: |
| # - For shared libraries - we use `pic`. This covers `dylib`, `cdylib` and `proc-macro` crate types. |
| # - In `fastbuild` and `dbg` mode we use `pic` by default. |
| # - In `opt` mode we use `nopic` outputs to build binaries. |
| if crate_type in ("cdylib", "dylib", "proc-macro"): |
| return cc_toolchain.needs_pic_for_dynamic_libraries(feature_configuration = feature_configuration) |
| elif compilation_mode in ("fastbuild", "dbg"): |
| return True |
| return False |
| |
| def _is_proc_macro(crate_info): |
| return "proc-macro" in (crate_info.type, crate_info.wrapped_crate_type) |
| |
| def collect_deps( |
| deps, |
| proc_macro_deps, |
| aliases): |
| """Walks through dependencies and collects the transitive dependencies. |
| |
| Args: |
| deps (list): The deps from ctx.attr.deps. |
| proc_macro_deps (list): The proc_macro deps from ctx.attr.proc_macro_deps. |
| aliases (dict): A dict mapping aliased targets to their actual Crate information. |
| |
| Returns: |
| tuple: Returns a tuple of: |
| DepInfo, |
| BuildInfo, |
| linkstamps (depset[CcLinkstamp]): A depset of CcLinkstamps that need to be compiled and linked into all linked binaries when applicable. |
| |
| """ |
| direct_crates = [] |
| transitive_crates = [] |
| transitive_data = [] |
| transitive_proc_macro_data = [] |
| transitive_noncrates = [] |
| transitive_build_infos = [] |
| transitive_link_search_paths = [] |
| build_info = None |
| linkstamps = [] |
| transitive_crate_outputs = [] |
| transitive_metadata_outputs = [] |
| |
| crate_deps = [] |
| for dep in depset(transitive = [deps, proc_macro_deps]).to_list(): |
| crate_group = None |
| |
| if type(dep) == "Target" and rust_common.crate_group_info in dep: |
| crate_group = dep[rust_common.crate_group_info] |
| elif type(dep) == "struct" and hasattr(dep, "crate_group_info") and dep.crate_group_info != None: |
| crate_group = dep.crate_group_info |
| else: |
| crate_deps.append(dep) |
| |
| if crate_group: |
| for dep_variant_info in crate_group.dep_variant_infos.to_list(): |
| crate_deps.append(struct( |
| crate_info = dep_variant_info.crate_info, |
| dep_info = dep_variant_info.dep_info, |
| cc_info = dep_variant_info.cc_info, |
| )) |
| |
| aliases = {k.label: v for k, v in aliases.items()} |
| for dep in crate_deps: |
| (crate_info, dep_info) = _get_crate_and_dep_info(dep) |
| cc_info = _get_cc_info(dep) |
| dep_build_info = _get_build_info(dep) |
| |
| if cc_info: |
| linkstamps.append(cc_info.linking_context.linkstamps()) |
| |
| if crate_info: |
| # This dependency is a rust_library |
| |
| # When crate_info.owner is set, we use it. When the dep type is Target we get the |
| # label from dep.label |
| owner = getattr(crate_info, "owner", dep.label if type(dep) == "Target" else None) |
| |
| direct_crates.append(AliasableDepInfo( |
| name = aliases.get(owner, crate_info.name), |
| dep = crate_info, |
| )) |
| |
| transitive_crates.append( |
| depset( |
| [crate_info], |
| transitive = [] if _is_proc_macro(crate_info) else [dep_info.transitive_crates], |
| ), |
| ) |
| |
| if _is_proc_macro(crate_info): |
| # This crate's data and its non-macro dependencies' data are proc macro data. |
| transitive_proc_macro_data.append(crate_info.data) |
| transitive_proc_macro_data.append(dep_info.transitive_data) |
| else: |
| # This crate's proc macro dependencies' data are proc macro data. |
| transitive_proc_macro_data.append(dep_info.transitive_proc_macro_data) |
| |
| # Track transitive non-macro data in case a proc macro depends on this crate. |
| transitive_data.append(crate_info.data) |
| transitive_data.append(dep_info.transitive_data) |
| |
| # If this dependency produces metadata, add it to the metadata outputs. |
| # If it doesn't (for example a custom library that exports crate_info), |
| # we depend on crate_info.output. |
| depend_on = crate_info.metadata |
| if not crate_info.metadata: |
| depend_on = crate_info.output |
| |
| # If this dependency is a proc_macro, it still can be used for lib crates |
| # that produce metadata. |
| # In that case, we don't depend on its metadata dependencies. |
| transitive_metadata_outputs.append( |
| depset( |
| [depend_on], |
| transitive = [] if _is_proc_macro(crate_info) else [dep_info.transitive_metadata_outputs], |
| ), |
| ) |
| |
| transitive_crate_outputs.append( |
| depset( |
| [crate_info.output], |
| transitive = [] if _is_proc_macro(crate_info) else [dep_info.transitive_crate_outputs], |
| ), |
| ) |
| |
| if "proc-macro" not in [crate_info.type, crate_info.wrapped_crate_type]: |
| transitive_noncrates.append(dep_info.transitive_noncrates) |
| transitive_link_search_paths.append(dep_info.link_search_path_files) |
| |
| transitive_build_infos.append(dep_info.transitive_build_infos) |
| elif cc_info or dep_build_info: |
| if cc_info: |
| # This dependency is a cc_library |
| transitive_noncrates.append(cc_info.linking_context.linker_inputs) |
| |
| if dep_build_info: |
| if build_info: |
| fail("Several deps are providing build information, " + |
| "only one is allowed in the dependencies") |
| build_info = dep_build_info |
| transitive_build_infos.append(depset([build_info])) |
| if build_info.link_search_paths: |
| transitive_link_search_paths.append(depset([build_info.link_search_paths])) |
| else: |
| fail("rust targets can only depend on rust_library, rust_*_library or cc_library " + |
| "targets.") |
| |
| transitive_crates_depset = depset(transitive = transitive_crates) |
| transitive_data_depset = depset(transitive = transitive_data) |
| transitive_proc_macro_data_depset = depset(transitive = transitive_proc_macro_data) |
| |
| return ( |
| rust_common.dep_info( |
| direct_crates = depset(direct_crates), |
| transitive_crates = transitive_crates_depset, |
| transitive_data = transitive_data_depset, |
| transitive_proc_macro_data = transitive_proc_macro_data_depset, |
| transitive_noncrates = depset( |
| transitive = transitive_noncrates, |
| order = "topological", # dylib link flag ordering matters. |
| ), |
| transitive_crate_outputs = depset(transitive = transitive_crate_outputs), |
| transitive_metadata_outputs = depset(transitive = transitive_metadata_outputs), |
| transitive_build_infos = depset(transitive = transitive_build_infos), |
| link_search_path_files = depset(transitive = transitive_link_search_paths), |
| dep_env = build_info.dep_env if build_info else None, |
| ), |
| build_info, |
| depset(transitive = linkstamps), |
| ) |
| |
| def _collect_libs_from_linker_inputs(linker_inputs, use_pic): |
| # TODO: We could let the user choose how to link, instead of always preferring to link static libraries. |
| return [ |
| get_preferred_artifact(lib, use_pic) |
| for li in linker_inputs |
| for lib in li.libraries |
| ] |
| |
| def _get_crate_and_dep_info(dep): |
| if type(dep) == "Target" and rust_common.crate_info in dep: |
| return (dep[rust_common.crate_info], dep[rust_common.dep_info]) |
| elif type(dep) == "struct" and hasattr(dep, "crate_info"): |
| return (dep.crate_info, dep.dep_info) |
| return (None, None) |
| |
| def _get_cc_info(dep): |
| if type(dep) == "Target" and CcInfo in dep: |
| return dep[CcInfo] |
| elif type(dep) == "struct" and hasattr(dep, "cc_info"): |
| return dep.cc_info |
| return None |
| |
| def _get_build_info(dep): |
| if type(dep) == "Target" and BuildInfo in dep: |
| return dep[BuildInfo] |
| elif type(dep) == "struct" and hasattr(dep, "build_info"): |
| return dep.build_info |
| return None |
| |
| def get_cc_user_link_flags(ctx): |
| """Get the current target's linkopt flags |
| |
| Args: |
| ctx (ctx): The current rule's context object |
| |
| Returns: |
| depset: The flags passed to Bazel by --linkopt option. |
| """ |
| return ctx.fragments.cpp.linkopts |
| |
| def get_linker_and_args(ctx, attr, crate_type, cc_toolchain, feature_configuration, rpaths, add_flags_for_binary = False): |
| """Gathers cc_common linker information |
| |
| Args: |
| ctx (ctx): The current target's context object |
| attr (struct): Attributes to use in gathering linker args |
| crate_type (str): The target crate's type (i.e. "bin", "proc-macro", etc.). |
| cc_toolchain (CcToolchain): cc_toolchain for which we are creating build variables. |
| feature_configuration (FeatureConfiguration): Feature configuration to be queried. |
| rpaths (depset): Depset of directories where loader will look for libraries at runtime. |
| add_flags_for_binary (bool, optional): Whether to add "bin" link flags to the command regardless of `crate_type`. |
| |
| |
| Returns: |
| tuple: A tuple of the following items: |
| - (str): The tool path for given action. |
| - (sequence): A flattened command line flags for given action. |
| - (dict): Environment variables to be set for given action. |
| """ |
| user_link_flags = get_cc_user_link_flags(ctx) |
| |
| if crate_type in ("bin") or add_flags_for_binary: |
| is_linking_dynamic_library = False |
| action_name = CPP_LINK_EXECUTABLE_ACTION_NAME |
| elif crate_type in ("dylib"): |
| is_linking_dynamic_library = True |
| action_name = CPP_LINK_NODEPS_DYNAMIC_LIBRARY_ACTION_NAME |
| elif crate_type in ("staticlib"): |
| is_linking_dynamic_library = False |
| action_name = CPP_LINK_STATIC_LIBRARY_ACTION_NAME |
| elif crate_type in ("cdylib", "proc-macro"): |
| # Proc macros get compiled as shared libraries to be loaded by the compiler. |
| is_linking_dynamic_library = True |
| action_name = CPP_LINK_DYNAMIC_LIBRARY_ACTION_NAME |
| elif crate_type in ("lib", "rlib"): |
| fail("Invalid `crate_type` for linking action: {}".format(crate_type)) |
| else: |
| fail("Unknown `crate_type`: {}".format(crate_type)) |
| |
| # Add linkopts from dependencies. This includes linkopts from transitive |
| # dependencies since they get merged up. |
| for dep in getattr(attr, "deps", []): |
| if CcInfo in dep and dep[CcInfo].linking_context: |
| for linker_input in dep[CcInfo].linking_context.linker_inputs.to_list(): |
| for flag in linker_input.user_link_flags: |
| user_link_flags.append(flag) |
| link_variables = cc_common.create_link_variables( |
| feature_configuration = feature_configuration, |
| cc_toolchain = cc_toolchain, |
| is_linking_dynamic_library = is_linking_dynamic_library, |
| runtime_library_search_directories = rpaths, |
| user_link_flags = user_link_flags, |
| ) |
| link_args = cc_common.get_memory_inefficient_command_line( |
| feature_configuration = feature_configuration, |
| action_name = action_name, |
| variables = link_variables, |
| ) |
| link_env = cc_common.get_environment_variables( |
| feature_configuration = feature_configuration, |
| action_name = action_name, |
| variables = link_variables, |
| ) |
| ld = cc_common.get_tool_for_action( |
| feature_configuration = feature_configuration, |
| action_name = action_name, |
| ) |
| |
| return ld, link_args, link_env |
| |
| def _process_build_scripts( |
| build_info, |
| dep_info, |
| compile_inputs, |
| include_link_flags = True): |
| """Gathers the outputs from a target's `cargo_build_script` action. |
| |
| Args: |
| build_info (BuildInfo): The target Build's dependency info. |
| dep_info (DepInfo): The Depinfo provider form the target Crate's set of inputs. |
| compile_inputs (depset): A set of all files that will participate in the build. |
| include_link_flags (bool, optional): Whether to include flags like `-l` that instruct the linker to search for a library. |
| |
| Returns: |
| tuple: A tuple: A tuple of the following items: |
| - (depset[File]): A list of all build info `OUT_DIR` File objects |
| - (str): The `OUT_DIR` of the current build info |
| - (File): An optional path to a generated environment file from a `cargo_build_script` target |
| - (depset[File]): All direct and transitive build flags from the current build info. |
| """ |
| extra_inputs, out_dir, build_env_file, build_flags_files = _create_extra_input_args(build_info, dep_info, include_link_flags = include_link_flags) |
| compile_inputs = depset(transitive = [extra_inputs, compile_inputs]) |
| return compile_inputs, out_dir, build_env_file, build_flags_files |
| |
| def _symlink_for_ambiguous_lib(actions, toolchain, crate_info, lib): |
| """Constructs a disambiguating symlink for a library dependency. |
| |
| Args: |
| actions (Actions): The rule's context actions object. |
| toolchain: The Rust toolchain object. |
| crate_info (CrateInfo): The target crate's info. |
| lib (File): The library to symlink to. |
| |
| Returns: |
| (File): The disambiguating symlink for the library. |
| """ |
| # FIXME: Once the relative order part of the native-link-modifiers rustc |
| # feature is stable, we should be able to eliminate the need to construct |
| # symlinks by passing the full paths to the libraries. |
| # https://github.com/rust-lang/rust/issues/81490. |
| |
| # Take the absolute value of hash() since it could be negative. |
| path_hash = abs(hash(lib.path)) |
| lib_name = get_lib_name_for_windows(lib) if toolchain.target_os.startswith("windows") else get_lib_name_default(lib) |
| |
| if toolchain.target_os.startswith("windows"): |
| prefix = "" |
| extension = ".lib" |
| elif lib_name.endswith(".pic"): |
| # Strip the .pic suffix |
| lib_name = lib_name[:-4] |
| prefix = "lib" |
| extension = ".pic.a" |
| else: |
| prefix = "lib" |
| extension = ".a" |
| |
| # Ensure the symlink follows the lib<name>.a pattern on Unix-like platforms |
| # or <name>.lib on Windows. |
| # Add a hash of the original library path to disambiguate libraries with the same basename. |
| symlink_name = "{}{}-{}{}".format(prefix, lib_name, path_hash, extension) |
| |
| # Add the symlink to a target crate-specific _ambiguous_libs/ subfolder, |
| # to avoid possible collisions with sibling crates that may depend on the |
| # same ambiguous libraries. |
| symlink = actions.declare_file("_ambiguous_libs/" + crate_info.output.basename + "/" + symlink_name) |
| actions.symlink( |
| output = symlink, |
| target_file = lib, |
| progress_message = "Creating symlink to ambiguous lib: {}".format(lib.path), |
| ) |
| return symlink |
| |
| def _disambiguate_libs(actions, toolchain, crate_info, dep_info, use_pic): |
| """Constructs disambiguating symlinks for ambiguous library dependencies. |
| |
| The symlinks are all created in a _ambiguous_libs/ subfolder specific to |
| the target crate to avoid possible collisions with sibling crates that may |
| depend on the same ambiguous libraries. |
| |
| Args: |
| actions (Actions): The rule's context actions object. |
| toolchain: The Rust toolchain object. |
| crate_info (CrateInfo): The target crate's info. |
| dep_info: (DepInfo): The target crate's dependency info. |
| use_pic: (boolean): Whether the build should use PIC. |
| |
| Returns: |
| dict[String, File]: A mapping from ambiguous library paths to their |
| disambiguating symlink. |
| """ |
| # FIXME: Once the relative order part of the native-link-modifiers rustc |
| # feature is stable, we should be able to eliminate the need to construct |
| # symlinks by passing the full paths to the libraries. |
| # https://github.com/rust-lang/rust/issues/81490. |
| |
| # A dictionary from file paths of ambiguous libraries to the corresponding |
| # symlink. |
| ambiguous_libs = {} |
| |
| # A dictionary maintaining a mapping from preferred library name to the |
| # last visited artifact with that name. |
| visited_libs = {} |
| for link_input in dep_info.transitive_noncrates.to_list(): |
| for lib in link_input.libraries: |
| # FIXME: Dynamic libs are not disambiguated right now, there are |
| # cases where those have a non-standard name with version (e.g., |
| # //test/unit/versioned_libs). We hope that the link modifiers |
| # stabilization will come before we need to make this work. |
| if _is_dylib(lib): |
| continue |
| artifact = get_preferred_artifact(lib, use_pic) |
| name = get_lib_name_for_windows(artifact) if toolchain.target_os.startswith("windows") else get_lib_name_default(artifact) |
| |
| # On Linux-like platforms, normally library base names start with |
| # `lib`, following the pattern `lib[name].(a|lo)` and we pass |
| # -lstatic=name. |
| # On Windows, the base name looks like `name.lib` and we pass |
| # -lstatic=name. |
| # FIXME: Under the native-link-modifiers unstable rustc feature, |
| # we could use -lstatic:+verbatim instead. |
| needs_symlink_to_standardize_name = ( |
| toolchain.target_os.startswith(("linux", "mac", "darwin")) and |
| artifact.basename.endswith(".a") and not artifact.basename.startswith("lib") |
| ) or ( |
| toolchain.target_os.startswith("windows") and not artifact.basename.endswith(".lib") |
| ) |
| |
| # Detect cases where we need to disambiguate library dependencies |
| # by constructing symlinks. |
| if ( |
| needs_symlink_to_standardize_name or |
| # We have multiple libraries with the same name. |
| (name in visited_libs and visited_libs[name].path != artifact.path) |
| ): |
| # Disambiguate the previously visited library (if we just detected |
| # that it is ambiguous) and the current library. |
| if name in visited_libs: |
| old_path = visited_libs[name].path |
| if old_path not in ambiguous_libs: |
| ambiguous_libs[old_path] = _symlink_for_ambiguous_lib(actions, toolchain, crate_info, visited_libs[name]) |
| ambiguous_libs[artifact.path] = _symlink_for_ambiguous_lib(actions, toolchain, crate_info, artifact) |
| |
| visited_libs[name] = artifact |
| return ambiguous_libs |
| |
| def _depend_on_metadata(crate_info, force_depend_on_objects): |
| """Determines if we can depend on metadata for this crate. |
| |
| By default (when pipelining is disabled or when the crate type needs to link against |
| objects) we depend on the set of object files (.rlib). |
| When pipelining is enabled and the crate type supports depending on metadata, |
| we depend on metadata files only (.rmeta). |
| In some rare cases, even if both of those conditions are true, we still want to |
| depend on objects. This is what force_depend_on_objects is. |
| |
| Args: |
| crate_info (CrateInfo): The Crate to determine this for. |
| force_depend_on_objects (bool): if set we will not depend on metadata. |
| |
| Returns: |
| Whether we can depend on metadata for this crate. |
| """ |
| if force_depend_on_objects: |
| return False |
| |
| return crate_info.type in ("rlib", "lib") |
| |
| def collect_inputs( |
| ctx, |
| file, |
| files, |
| linkstamps, |
| toolchain, |
| cc_toolchain, |
| feature_configuration, |
| crate_info, |
| dep_info, |
| build_info, |
| stamp = False, |
| force_depend_on_objects = False, |
| experimental_use_cc_common_link = False, |
| include_link_flags = True): |
| """Gather's the inputs and required input information for a rustc action |
| |
| Args: |
| ctx (ctx): The rule's context object. |
| file (struct): A struct containing files defined in label type attributes marked as `allow_single_file`. |
| files (list): A list of all inputs (`ctx.files`). |
| linkstamps (depset): A depset of CcLinkstamps that need to be compiled and linked into all linked binaries. |
| toolchain (rust_toolchain): The current `rust_toolchain`. |
| cc_toolchain (CcToolchainInfo): The current `cc_toolchain`. |
| feature_configuration (FeatureConfiguration): Feature configuration to be queried. |
| crate_info (CrateInfo): The Crate information of the crate to process build scripts for. |
| dep_info (DepInfo): The target Crate's dependency information. |
| build_info (BuildInfo): The target Crate's build settings. |
| stamp (bool, optional): Whether or not workspace status stamping is enabled. For more details see |
| https://docs.bazel.build/versions/main/user-manual.html#flag--stamp |
| force_depend_on_objects (bool, optional): Forces dependencies of this rule to be objects rather than |
| metadata, even for libraries. This is used in rustdoc tests. |
| experimental_use_cc_common_link (bool, optional): Whether rules_rust uses cc_common.link to link |
| rust binaries. |
| include_link_flags (bool, optional): Whether to include flags like `-l` that instruct the linker to search for a library. |
| |
| Returns: |
| tuple: A tuple: A tuple of the following items: |
| - (list): A list of all build info `OUT_DIR` File objects |
| - (str): The `OUT_DIR` of the current build info |
| - (File): An optional path to a generated environment file from a `cargo_build_script` target |
| - (depset[File]): All direct and transitive build flag files from the current build info |
| - (list[File]): Linkstamp outputs |
| - (dict[String, File]): Ambiguous libs, see `_disambiguate_libs`. |
| """ |
| linker_script = getattr(file, "linker_script") if hasattr(file, "linker_script") else None |
| |
| # TODO: As of writing this comment Bazel used Java CcToolchainInfo. |
| # However there is ongoing work to rewrite provider in Starlark. |
| # rules_rust is not coupled with Bazel release. Remove conditional and change to |
| # _linker_files once Starlark CcToolchainInfo is visible to Bazel. |
| # https://github.com/bazelbuild/rules_rust/issues/2425 |
| if hasattr(cc_toolchain, "_linker_files"): |
| linker_depset = cc_toolchain._linker_files |
| else: |
| linker_depset = cc_toolchain.linker_files() |
| compilation_mode = ctx.var["COMPILATION_MODE"] |
| |
| use_pic = _should_use_pic(cc_toolchain, feature_configuration, crate_info.type, compilation_mode) |
| |
| # Pass linker inputs only for linking-like actions, not for example where |
| # the output is rlib. This avoids quadratic behavior where transitive noncrates are |
| # flattened on each transitive rust_library dependency. |
| additional_transitive_inputs = [] |
| ambiguous_libs = {} |
| if crate_info.type not in ("lib", "rlib"): |
| linker_inputs = dep_info.transitive_noncrates.to_list() |
| ambiguous_libs = _disambiguate_libs(ctx.actions, toolchain, crate_info, dep_info, use_pic) |
| additional_transitive_inputs = _collect_libs_from_linker_inputs(linker_inputs, use_pic) + [ |
| additional_input |
| for linker_input in linker_inputs |
| for additional_input in linker_input.additional_inputs |
| ] + ambiguous_libs.values() |
| |
| # Compute linkstamps. Use the inputs of the binary as inputs to the |
| # linkstamp action to ensure linkstamps are rebuilt whenever binary inputs |
| # change. |
| linkstamp_outs = [] |
| |
| transitive_crate_outputs = dep_info.transitive_crate_outputs |
| if _depend_on_metadata(crate_info, force_depend_on_objects): |
| transitive_crate_outputs = dep_info.transitive_metadata_outputs |
| |
| build_info_inputs = [] |
| if build_info: |
| if build_info.rustc_env: |
| build_info_inputs.append(build_info.rustc_env) |
| if build_info.flags: |
| build_info_inputs.append(build_info.flags) |
| |
| nolinkstamp_compile_inputs = depset( |
| getattr(files, "data", []) + |
| build_info_inputs + |
| ([toolchain.target_json] if toolchain.target_json else []) + |
| ([] if linker_script == None else [linker_script]), |
| transitive = [ |
| linker_depset, |
| crate_info.srcs, |
| transitive_crate_outputs, |
| depset(additional_transitive_inputs), |
| crate_info.compile_data, |
| dep_info.transitive_proc_macro_data, |
| toolchain.all_files, |
| ], |
| ) |
| |
| # Register linkstamps when linking with rustc (when linking with |
| # cc_common.link linkstamps are handled by cc_common.link itself). |
| if not experimental_use_cc_common_link and crate_info.type in ("bin", "cdylib"): |
| # There is no other way to register an action for each member of a depset than |
| # flattening the depset as of 2021-10-12. Luckily, usually there is only one linkstamp |
| # in a build, and we only flatten the list on binary targets that perform transitive linking, |
| # so it's extremely unlikely that this call to `to_list()` will ever be a performance |
| # problem. |
| for linkstamp in linkstamps.to_list(): |
| # The linkstamp output path is based on the binary crate |
| # name and the input linkstamp path. This is to disambiguate |
| # the linkstamp outputs produced by multiple binary crates |
| # that depend on the same linkstamp. We use the same pattern |
| # for the output name as the one used by native cc rules. |
| out_name = "_objs/" + crate_info.output.basename + "/" + linkstamp.file().path[:-len(linkstamp.file().extension)] + "o" |
| linkstamp_out = ctx.actions.declare_file(out_name) |
| linkstamp_outs.append(linkstamp_out) |
| cc_common.register_linkstamp_compile_action( |
| actions = ctx.actions, |
| cc_toolchain = cc_toolchain, |
| feature_configuration = feature_configuration, |
| source_file = linkstamp.file(), |
| output_file = linkstamp_out, |
| compilation_inputs = linkstamp.hdrs(), |
| inputs_for_validation = nolinkstamp_compile_inputs, |
| label_replacement = str(ctx.label), |
| output_replacement = crate_info.output.path, |
| ) |
| |
| # If stamping is enabled include the volatile and stable status info file |
| stamp_info = [ctx.version_file, ctx.info_file] if stamp else [] |
| |
| compile_inputs = depset( |
| linkstamp_outs + stamp_info, |
| transitive = [ |
| nolinkstamp_compile_inputs, |
| ], |
| ) |
| |
| # For backwards compatibility, we also check the value of the `rustc_env_files` attribute when |
| # `crate_info.rustc_env_files` is not populated. |
| build_env_files = crate_info.rustc_env_files if crate_info.rustc_env_files else getattr(files, "rustc_env_files", []) |
| compile_inputs, out_dir, build_env_file, build_flags_files = _process_build_scripts(build_info, dep_info, compile_inputs, include_link_flags = include_link_flags) |
| if build_env_file: |
| build_env_files = [f for f in build_env_files] + [build_env_file] |
| compile_inputs = depset(build_env_files, transitive = [compile_inputs]) |
| return compile_inputs, out_dir, build_env_files, build_flags_files, linkstamp_outs, ambiguous_libs |
| |
| def construct_arguments( |
| ctx, |
| attr, |
| file, |
| toolchain, |
| tool_path, |
| cc_toolchain, |
| feature_configuration, |
| crate_info, |
| dep_info, |
| linkstamp_outs, |
| ambiguous_libs, |
| output_hash, |
| rust_flags, |
| out_dir, |
| build_env_files, |
| build_flags_files, |
| emit = ["dep-info", "link"], |
| force_all_deps_direct = False, |
| add_flags_for_binary = False, |
| include_link_flags = True, |
| stamp = False, |
| remap_path_prefix = "", |
| use_json_output = False, |
| build_metadata = False, |
| force_depend_on_objects = False, |
| skip_expanding_rustc_env = False): |
| """Builds an Args object containing common rustc flags |
| |
| Args: |
| ctx (ctx): The rule's context object |
| attr (struct): The attributes for the target. These may be different from ctx.attr in an aspect context. |
| file (struct): A struct containing files defined in label type attributes marked as `allow_single_file`. |
| toolchain (rust_toolchain): The current target's `rust_toolchain` |
| tool_path (str): Path to rustc |
| cc_toolchain (CcToolchain): The CcToolchain for the current target. |
| feature_configuration (FeatureConfiguration): Class used to construct command lines from CROSSTOOL features. |
| crate_info (CrateInfo): The CrateInfo provider of the target crate |
| dep_info (DepInfo): The DepInfo provider of the target crate |
| linkstamp_outs (list): Linkstamp outputs of native dependencies |
| ambiguous_libs (dict): Ambiguous libs, see `_disambiguate_libs` |
| output_hash (str): The hashed path of the crate root |
| rust_flags (list): Additional flags to pass to rustc |
| out_dir (str): The path to the output directory for the target Crate. |
| build_env_files (list): Files containing rustc environment variables, for instance from `cargo_build_script` actions. |
| build_flags_files (depset): The output files of a `cargo_build_script` actions containing rustc build flags |
| emit (list): Values for the --emit flag to rustc. |
| force_all_deps_direct (bool, optional): Whether to pass the transitive rlibs with --extern |
| to the commandline as opposed to -L. |
| add_flags_for_binary (bool, optional): Whether to add "bin" link flags to the command regardless of `emit` and `crate_type`. |
| include_link_flags (bool, optional): Whether to include flags like `-l` that instruct the linker to search for a library. |
| stamp (bool, optional): Whether or not workspace status stamping is enabled. For more details see |
| https://docs.bazel.build/versions/main/user-manual.html#flag--stamp |
| remap_path_prefix (str, optional): A value used to remap `${pwd}` to. If set to None, no prefix will be set. |
| use_json_output (bool): Have rustc emit json and process_wrapper parse json messages to output rendered output. |
| build_metadata (bool): Generate CLI arguments for building *only* .rmeta files. This requires use_json_output. |
| force_depend_on_objects (bool): Force using `.rlib` object files instead of metadata (`.rmeta`) files even if they are available. |
| skip_expanding_rustc_env (bool): Whether to skip expanding CrateInfo.rustc_env_attr |
| |
| Returns: |
| tuple: A tuple of the following items |
| - (struct): A struct of arguments used to run the `Rustc` action |
| - process_wrapper_flags (Args): Arguments for the process wrapper |
| - rustc_path (Args): Arguments for invoking rustc via the process wrapper |
| - rustc_flags (Args): Rust flags for the Rust compiler |
| - all (list): A list of all `Args` objects in the order listed above. |
| This is to be passed to the `arguments` parameter of actions |
| - (dict): Common rustc environment variables |
| """ |
| if build_metadata and not use_json_output: |
| fail("build_metadata requires parse_json_output") |
| |
| output_dir = getattr(crate_info.output, "dirname", None) |
| linker_script = getattr(file, "linker_script", None) |
| |
| env = _get_rustc_env(attr, toolchain, crate_info.name) |
| |
| # Wrapper args first |
| process_wrapper_flags = ctx.actions.args() |
| |
| for build_env_file in build_env_files: |
| process_wrapper_flags.add("--env-file", build_env_file) |
| |
| process_wrapper_flags.add_all(build_flags_files, before_each = "--arg-file") |
| |
| # Certain rust build processes expect to find files from the environment |
| # variable `$CARGO_MANIFEST_DIR`. Examples of this include pest, tera, |
| # asakuma. |
| # |
| # The compiler and by extension proc-macros see the current working |
| # directory as the Bazel exec root. This is what `$CARGO_MANIFEST_DIR` |
| # would default to but is often the wrong value (e.g. if the source is in a |
| # sub-package or if we are building something in an external repository). |
| # Hence, we need to set `CARGO_MANIFEST_DIR` explicitly. |
| # |
| # Since we cannot get the `exec_root` from starlark, we cheat a little and |
| # use `${pwd}` which resolves the `exec_root` at action execution time. |
| process_wrapper_flags.add("--subst", "pwd=${pwd}") |
| |
| # If stamping is enabled, enable the functionality in the process wrapper |
| if stamp: |
| process_wrapper_flags.add("--volatile-status-file", ctx.version_file) |
| process_wrapper_flags.add("--stable-status-file", ctx.info_file) |
| |
| # Both ctx.label.workspace_root and ctx.label.package are relative paths |
| # and either can be empty strings. Avoid trailing/double slashes in the path. |
| components = "${{pwd}}/{}/{}".format(ctx.label.workspace_root, ctx.label.package).split("/") |
| env["CARGO_MANIFEST_DIR"] = "/".join([c for c in components if c]) |
| |
| if out_dir != None: |
| env["OUT_DIR"] = "${pwd}/" + out_dir |
| |
| # Arguments for launching rustc from the process wrapper |
| rustc_path = ctx.actions.args() |
| rustc_path.add("--") |
| rustc_path.add(tool_path) |
| |
| # Rustc arguments |
| rustc_flags = ctx.actions.args() |
| rustc_flags.set_param_file_format("multiline") |
| rustc_flags.use_param_file("@%s", use_always = False) |
| rustc_flags.add(crate_info.root) |
| rustc_flags.add(crate_info.name, format = "--crate-name=%s") |
| rustc_flags.add(crate_info.type, format = "--crate-type=%s") |
| |
| error_format = "human" |
| if hasattr(attr, "_error_format"): |
| error_format = attr._error_format[ErrorFormatInfo].error_format |
| |
| if use_json_output: |
| # If --error-format was set to json, we just pass the output through |
| # Otherwise process_wrapper uses the "rendered" field. |
| process_wrapper_flags.add("--rustc-output-format", "json" if error_format == "json" else "rendered") |
| |
| # Configure rustc json output by adding artifact notifications. |
| # These will always be filtered out by process_wrapper and will be use to terminate |
| # rustc when appropriate. |
| json = ["artifacts"] |
| if error_format == "short": |
| json.append("diagnostic-short") |
| elif error_format == "human" and toolchain.target_os != "windows": |
| # If the os is not windows, we can get colorized output. |
| json.append("diagnostic-rendered-ansi") |
| |
| rustc_flags.add_joined(json, format_joined = "--json=%s", join_with = ",") |
| |
| error_format = "json" |
| |
| if build_metadata: |
| # Configure process_wrapper to terminate rustc when metadata are emitted |
| process_wrapper_flags.add("--rustc-quit-on-rmeta", "true") |
| if crate_info.rustc_rmeta_output: |
| process_wrapper_flags.add("--output-file", crate_info.rustc_rmeta_output.path) |
| elif crate_info.rustc_output: |
| process_wrapper_flags.add("--output-file", crate_info.rustc_output.path) |
| |
| rustc_flags.add(error_format, format = "--error-format=%s") |
| |
| # Mangle symbols to disambiguate crates with the same name. This could |
| # happen only for non-final artifacts where we compute an output_hash, |
| # e.g., rust_library. |
| # |
| # For "final" artifacts and ones intended for distribution outside of |
| # Bazel, such as rust_binary, rust_static_library and rust_shared_library, |
| # where output_hash is None we don't need to add these flags. |
| if output_hash: |
| rustc_flags.add(output_hash, format = "--codegen=metadata=-%s") |
| rustc_flags.add(output_hash, format = "--codegen=extra-filename=-%s") |
| |
| if output_dir: |
| rustc_flags.add(output_dir, format = "--out-dir=%s") |
| |
| compilation_mode = get_compilation_mode_opts(ctx, toolchain) |
| rustc_flags.add(compilation_mode.opt_level, format = "--codegen=opt-level=%s") |
| rustc_flags.add(compilation_mode.debug_info, format = "--codegen=debuginfo=%s") |
| rustc_flags.add(compilation_mode.strip_level, format = "--codegen=strip=%s") |
| |
| # For determinism to help with build distribution and such |
| if remap_path_prefix != None: |
| rustc_flags.add("--remap-path-prefix=${{pwd}}={}".format(remap_path_prefix)) |
| |
| emit_without_paths = [] |
| for kind in emit: |
| if kind == "link" and crate_info.type == "bin" and crate_info.output != None: |
| rustc_flags.add(crate_info.output, format = "--emit=link=%s") |
| else: |
| emit_without_paths.append(kind) |
| |
| if emit_without_paths: |
| rustc_flags.add_joined(emit_without_paths, format_joined = "--emit=%s", join_with = ",") |
| if error_format != "json": |
| # Color is not compatible with json output. |
| rustc_flags.add("--color=always") |
| rustc_flags.add(toolchain.target_flag_value, format = "--target=%s") |
| if hasattr(attr, "crate_features"): |
| rustc_flags.add_all(getattr(attr, "crate_features"), before_each = "--cfg", format_each = 'feature="%s"') |
| if linker_script: |
| rustc_flags.add(linker_script, format = "--codegen=link-arg=-T%s") |
| |
| # Tell Rustc where to find the standard library (or libcore) |
| rustc_flags.add_all(toolchain.rust_std_paths, before_each = "-L", format_each = "%s") |
| rustc_flags.add_all(rust_flags) |
| |
| # Gather data path from crate_info since it is inherited from real crate for rust_doc and rust_test |
| # Deduplicate data paths due to https://github.com/bazelbuild/bazel/issues/14681 |
| data_paths = depset(direct = getattr(attr, "data", []), transitive = [crate_info.compile_data_targets]).to_list() |
| |
| rustc_flags.add_all( |
| expand_list_element_locations( |
| ctx, |
| getattr(attr, "rustc_flags", []), |
| data_paths, |
| ), |
| ) |
| add_edition_flags(rustc_flags, crate_info) |
| |
| # Link! |
| if ("link" in emit and crate_info.type not in ["rlib", "lib"]) or add_flags_for_binary: |
| # Rust's built-in linker can handle linking wasm files. We don't want to attempt to use the cc |
| # linker since it won't understand. |
| compilation_mode = ctx.var["COMPILATION_MODE"] |
| if toolchain.target_arch != "wasm32": |
| if output_dir: |
| use_pic = _should_use_pic(cc_toolchain, feature_configuration, crate_info.type, compilation_mode) |
| rpaths = _compute_rpaths(toolchain, output_dir, dep_info, use_pic) |
| else: |
| rpaths = depset() |
| |
| ld, link_args, link_env = get_linker_and_args(ctx, attr, crate_info.type, cc_toolchain, feature_configuration, rpaths, add_flags_for_binary = add_flags_for_binary) |
| |
| env.update(link_env) |
| rustc_flags.add(ld, format = "--codegen=linker=%s") |
| |
| # Split link args into individual "--codegen=link-arg=" flags to handle nested spaces. |
| # Additional context: https://github.com/rust-lang/rust/pull/36574 |
| rustc_flags.add_all(link_args, format_each = "--codegen=link-arg=%s") |
| |
| _add_native_link_flags(rustc_flags, dep_info, linkstamp_outs, ambiguous_libs, crate_info.type, toolchain, cc_toolchain, feature_configuration, compilation_mode, include_link_flags = include_link_flags) |
| |
| use_metadata = _depend_on_metadata(crate_info, force_depend_on_objects) |
| |
| # These always need to be added, even if not linking this crate. |
| add_crate_link_flags(rustc_flags, dep_info, force_all_deps_direct, use_metadata) |
| |
| needs_extern_proc_macro_flag = _is_proc_macro(crate_info) and crate_info.edition != "2015" |
| if needs_extern_proc_macro_flag: |
| rustc_flags.add("--extern") |
| rustc_flags.add("proc_macro") |
| |
| if toolchain.llvm_cov and ctx.configuration.coverage_enabled: |
| # https://doc.rust-lang.org/rustc/instrument-coverage.html |
| rustc_flags.add("--codegen=instrument-coverage") |
| |
| if toolchain._experimental_link_std_dylib: |
| rustc_flags.add("--codegen=prefer-dynamic") |
| |
| # Make bin crate data deps available to tests. |
| for data in getattr(attr, "data", []): |
| if rust_common.crate_info in data: |
| dep_crate_info = data[rust_common.crate_info] |
| if dep_crate_info.type == "bin": |
| # Trying to make CARGO_BIN_EXE_{} canonical across platform by strip out extension if exists |
| env_basename = dep_crate_info.output.basename[:-(1 + len(dep_crate_info.output.extension))] if len(dep_crate_info.output.extension) > 0 else dep_crate_info.output.basename |
| env["CARGO_BIN_EXE_" + env_basename] = dep_crate_info.output.short_path |
| |
| # Add environment variables from the Rust toolchain. |
| env.update(toolchain.env) |
| |
| # Update environment with user provided variables. |
| if skip_expanding_rustc_env: |
| env.update(crate_info.rustc_env) |
| else: |
| env.update(expand_dict_value_locations( |
| ctx, |
| crate_info.rustc_env, |
| data_paths, |
| )) |
| |
| # Ensure the sysroot is set for the target platform |
| if toolchain._toolchain_generated_sysroot: |
| rustc_flags.add(toolchain.sysroot, format = "--sysroot=%s") |
| |
| if toolchain._rename_first_party_crates: |
| env["RULES_RUST_THIRD_PARTY_DIR"] = toolchain._third_party_dir |
| |
| if crate_info.type in toolchain.extra_rustc_flags_for_crate_types.keys(): |
| rustc_flags.add_all(toolchain.extra_rustc_flags_for_crate_types[crate_info.type]) |
| |
| if is_exec_configuration(ctx): |
| rustc_flags.add_all(toolchain.extra_exec_rustc_flags) |
| else: |
| rustc_flags.add_all(toolchain.extra_rustc_flags) |
| |
| # extra_rustc_flags apply to the target configuration, not the exec configuration. |
| if hasattr(ctx.attr, "_extra_rustc_flags") and not is_exec_configuration(ctx): |
| rustc_flags.add_all(ctx.attr._extra_rustc_flags[ExtraRustcFlagsInfo].extra_rustc_flags) |
| |
| if hasattr(ctx.attr, "_extra_rustc_flag") and not is_exec_configuration(ctx): |
| rustc_flags.add_all(ctx.attr._extra_rustc_flag[ExtraRustcFlagsInfo].extra_rustc_flags) |
| |
| if hasattr(ctx.attr, "_per_crate_rustc_flag") and not is_exec_configuration(ctx): |
| per_crate_rustc_flags = ctx.attr._per_crate_rustc_flag[PerCrateRustcFlagsInfo].per_crate_rustc_flags |
| _add_per_crate_rustc_flags(ctx, rustc_flags, crate_info, per_crate_rustc_flags) |
| |
| if hasattr(ctx.attr, "_extra_exec_rustc_flags") and is_exec_configuration(ctx): |
| rustc_flags.add_all(ctx.attr._extra_exec_rustc_flags[ExtraExecRustcFlagsInfo].extra_exec_rustc_flags) |
| |
| if hasattr(ctx.attr, "_extra_exec_rustc_flag") and is_exec_configuration(ctx): |
| rustc_flags.add_all(ctx.attr._extra_exec_rustc_flag[ExtraExecRustcFlagsInfo].extra_exec_rustc_flags) |
| |
| if _is_no_std(ctx, toolchain, crate_info): |
| rustc_flags.add('--cfg=feature="no_std"') |
| |
| # Needed for bzlmod-aware runfiles resolution. |
| env["REPOSITORY_NAME"] = ctx.label.workspace_name |
| |
| # Create a struct which keeps the arguments separate so each may be tuned or |
| # replaced where necessary |
| args = struct( |
| process_wrapper_flags = process_wrapper_flags, |
| rustc_path = rustc_path, |
| rustc_flags = rustc_flags, |
| all = [process_wrapper_flags, rustc_path, rustc_flags], |
| ) |
| |
| return args, env |
| |
| def rustc_compile_action( |
| ctx, |
| attr, |
| toolchain, |
| rust_flags = [], |
| output_hash = None, |
| force_all_deps_direct = False, |
| crate_info_dict = None, |
| skip_expanding_rustc_env = False, |
| include_coverage = True): |
| """Create and run a rustc compile action based on the current rule's attributes |
| |
| Args: |
| ctx (ctx): The rule's context object |
| attr (struct): Attributes to use for the rust compile action |
| toolchain (rust_toolchain): The current `rust_toolchain` |
| output_hash (str, optional): The hashed path of the crate root. Defaults to None. |
| rust_flags (list, optional): Additional flags to pass to rustc. Defaults to []. |
| force_all_deps_direct (bool, optional): Whether to pass the transitive rlibs with --extern |
| to the commandline as opposed to -L. |
| crate_info_dict: A mutable dict used to create CrateInfo provider |
| skip_expanding_rustc_env (bool, optional): Whether to expand CrateInfo.rustc_env |
| include_coverage (bool, optional): Whether to generate coverage information or not. |
| |
| Returns: |
| list: A list of the following providers: |
| - (CrateInfo): info for the crate we just built; same as `crate_info` parameter. |
| - (DepInfo): The transitive dependencies of this crate. |
| - (DefaultInfo): The output file for this crate, and its runfiles. |
| """ |
| crate_info = rust_common.create_crate_info(**crate_info_dict) |
| |
| build_metadata = crate_info_dict.get("metadata", None) |
| rustc_output = crate_info_dict.get("rustc_output", None) |
| rustc_rmeta_output = crate_info_dict.get("rustc_rmeta_output", None) |
| |
| # Determine whether to use cc_common.link: |
| # * either if experimental_use_cc_common_link is 1, |
| # * or if experimental_use_cc_common_link is -1 and |
| # the toolchain experimental_use_cc_common_link is true. |
| experimental_use_cc_common_link = False |
| if hasattr(ctx.attr, "experimental_use_cc_common_link"): |
| if ctx.attr.experimental_use_cc_common_link == 0: |
| experimental_use_cc_common_link = False |
| elif ctx.attr.experimental_use_cc_common_link == 1: |
| experimental_use_cc_common_link = True |
| elif ctx.attr.experimental_use_cc_common_link == -1: |
| experimental_use_cc_common_link = toolchain._experimental_use_cc_common_link |
| |
| dep_info, build_info, linkstamps = collect_deps( |
| deps = crate_info_dict["deps"], |
| proc_macro_deps = crate_info_dict["proc_macro_deps"], |
| aliases = crate_info_dict["aliases"], |
| ) |
| extra_disabled_features = [RUST_LINK_CC_FEATURE] |
| if crate_info.type in ["bin", "cdylib"] and dep_info.transitive_noncrates.to_list(): |
| # One or more of the transitive deps is a cc_library / cc_import |
| extra_disabled_features = [] |
| cc_toolchain, feature_configuration = find_cc_toolchain(ctx, extra_disabled_features) |
| if not _are_linkstamps_supported( |
| feature_configuration = feature_configuration, |
| has_grep_includes = hasattr(ctx.attr, "_use_grep_includes"), |
| ): |
| linkstamps = depset([]) |
| |
| # Determine if the build is currently running with --stamp |
| stamp = is_stamping_enabled(attr) |
| |
| compile_inputs, out_dir, build_env_files, build_flags_files, linkstamp_outs, ambiguous_libs = collect_inputs( |
| ctx = ctx, |
| file = ctx.file, |
| files = ctx.files, |
| linkstamps = linkstamps, |
| toolchain = toolchain, |
| cc_toolchain = cc_toolchain, |
| feature_configuration = feature_configuration, |
| crate_info = crate_info, |
| dep_info = dep_info, |
| build_info = build_info, |
| stamp = stamp, |
| experimental_use_cc_common_link = experimental_use_cc_common_link, |
| ) |
| |
| # The types of rustc outputs to emit. |
| # If we build metadata, we need to keep the command line of the two invocations |
| # (rlib and rmeta) as similar as possible, otherwise rustc rejects the rmeta as |
| # a candidate. |
| # Because of that we need to add emit=metadata to both the rlib and rmeta invocation. |
| # |
| # When cc_common linking is enabled, emit a `.o` file, which is later |
| # passed to the cc_common.link action. |
| emit = ["dep-info", "link"] |
| if build_metadata: |
| emit.append("metadata") |
| if experimental_use_cc_common_link: |
| emit = ["obj"] |
| |
| args, env_from_args = construct_arguments( |
| ctx = ctx, |
| attr = attr, |
| file = ctx.file, |
| toolchain = toolchain, |
| tool_path = toolchain.rustc.path, |
| cc_toolchain = cc_toolchain, |
| emit = emit, |
| feature_configuration = feature_configuration, |
| crate_info = crate_info, |
| dep_info = dep_info, |
| linkstamp_outs = linkstamp_outs, |
| ambiguous_libs = ambiguous_libs, |
| output_hash = output_hash, |
| rust_flags = rust_flags, |
| out_dir = out_dir, |
| build_env_files = build_env_files, |
| build_flags_files = build_flags_files, |
| force_all_deps_direct = force_all_deps_direct, |
| stamp = stamp, |
| use_json_output = bool(build_metadata) or bool(rustc_output) or bool(rustc_rmeta_output), |
| skip_expanding_rustc_env = skip_expanding_rustc_env, |
| ) |
| |
| args_metadata = None |
| if build_metadata: |
| args_metadata, _ = construct_arguments( |
| ctx = ctx, |
| attr = attr, |
| file = ctx.file, |
| toolchain = toolchain, |
| tool_path = toolchain.rustc.path, |
| cc_toolchain = cc_toolchain, |
| emit = emit, |
| feature_configuration = feature_configuration, |
| crate_info = crate_info, |
| dep_info = dep_info, |
| linkstamp_outs = linkstamp_outs, |
| ambiguous_libs = ambiguous_libs, |
| output_hash = output_hash, |
| rust_flags = rust_flags, |
| out_dir = out_dir, |
| build_env_files = build_env_files, |
| build_flags_files = build_flags_files, |
| force_all_deps_direct = force_all_deps_direct, |
| stamp = stamp, |
| use_json_output = True, |
| build_metadata = True, |
| ) |
| |
| env = dict(ctx.configuration.default_shell_env) |
| |
| # this is the final list of env vars |
| env.update(env_from_args) |
| |
| if hasattr(attr, "version") and attr.version != "0.0.0": |
| formatted_version = " v{}".format(attr.version) |
| else: |
| formatted_version = "" |
| |
| # Declares the outputs of the rustc compile action. |
| # By default this is the binary output; if cc_common.link is used, this is |
| # the main `.o` file (`output_o` below). |
| outputs = [crate_info.output] |
| |
| # The `.o` output file, only used for linking via cc_common.link. |
| output_o = None |
| if experimental_use_cc_common_link: |
| obj_ext = ".o" |
| output_o = ctx.actions.declare_file(crate_info.name + obj_ext, sibling = crate_info.output) |
| outputs = [output_o] |
| |
| # For a cdylib that might be added as a dependency to a cc_* target on Windows, it is important to include the |
| # interface library that rustc generates in the output files. |
| interface_library = None |
| if toolchain.target_os == "windows" and crate_info.type == "cdylib": |
| # Rustc generates the import library with a `.dll.lib` extension rather than the usual `.lib` one that msvc |
| # expects (see https://github.com/rust-lang/rust/pull/29520 for more context). |
| interface_library = ctx.actions.declare_file(crate_info.output.basename + ".lib", sibling = crate_info.output) |
| outputs.append(interface_library) |
| |
| # The action might generate extra output that we don't want to include in the `DefaultInfo` files. |
| action_outputs = list(outputs) |
| if rustc_output: |
| action_outputs.append(rustc_output) |
| |
| # Get the compilation mode for the current target. |
| compilation_mode = get_compilation_mode_opts(ctx, toolchain) |
| |
| # Rustc generates a pdb file (on Windows) or a dsym folder (on macos) so provide it in an output group for crate |
| # types that benefit from having debug information in a separate file. |
| pdb_file = None |
| dsym_folder = None |
| if crate_info.type in ("cdylib", "bin"): |
| if toolchain.target_os == "windows" and compilation_mode.strip_level == "none": |
| pdb_file = ctx.actions.declare_file(crate_info.output.basename[:-len(crate_info.output.extension)] + "pdb", sibling = crate_info.output) |
| action_outputs.append(pdb_file) |
| elif toolchain.target_os == "darwin": |
| dsym_folder = ctx.actions.declare_directory(crate_info.output.basename + ".dSYM", sibling = crate_info.output) |
| action_outputs.append(dsym_folder) |
| |
| if ctx.executable._process_wrapper: |
| # Run as normal |
| ctx.actions.run( |
| executable = ctx.executable._process_wrapper, |
| inputs = compile_inputs, |
| outputs = action_outputs, |
| env = env, |
| arguments = args.all, |
| mnemonic = "Rustc", |
| progress_message = "Compiling Rust {} {}{} ({} files)".format( |
| crate_info.type, |
| ctx.label.name, |
| formatted_version, |
| len(crate_info.srcs.to_list()), |
| ), |
| toolchain = "@rules_rust//rust:toolchain_type", |
| ) |
| if args_metadata: |
| ctx.actions.run( |
| executable = ctx.executable._process_wrapper, |
| inputs = compile_inputs, |
| outputs = [build_metadata] + [x for x in [rustc_rmeta_output] if x], |
| env = env, |
| arguments = args_metadata.all, |
| mnemonic = "RustcMetadata", |
| progress_message = "Compiling Rust metadata {} {}{} ({} files)".format( |
| crate_info.type, |
| ctx.label.name, |
| formatted_version, |
| len(crate_info.srcs.to_list()), |
| ), |
| toolchain = "@rules_rust//rust:toolchain_type", |
| ) |
| elif hasattr(ctx.executable, "_bootstrap_process_wrapper"): |
| # Run without process_wrapper |
| if build_env_files or build_flags_files or stamp or build_metadata: |
| fail("build_env_files, build_flags_files, stamp, build_metadata are not supported when building without process_wrapper") |
| ctx.actions.run( |
| executable = ctx.executable._bootstrap_process_wrapper, |
| inputs = compile_inputs, |
| outputs = action_outputs, |
| env = env, |
| arguments = [args.rustc_path, args.rustc_flags], |
| mnemonic = "Rustc", |
| progress_message = "Compiling Rust (without process_wrapper) {} {}{} ({} files)".format( |
| crate_info.type, |
| ctx.label.name, |
| formatted_version, |
| len(crate_info.srcs.to_list()), |
| ), |
| toolchain = "@rules_rust//rust:toolchain_type", |
| ) |
| else: |
| fail("No process wrapper was defined for {}".format(ctx.label)) |
| |
| if experimental_use_cc_common_link: |
| # Wrap the main `.o` file into a compilation output suitable for |
| # cc_common.link. The main `.o` file is useful in both PIC and non-PIC |
| # modes. |
| compilation_outputs = cc_common.create_compilation_outputs( |
| objects = depset([output_o]), |
| pic_objects = depset([output_o]), |
| ) |
| |
| malloc_library = ctx.attr._custom_malloc or ctx.attr.malloc |
| |
| # Collect the linking contexts of the standard library and dependencies. |
| linking_contexts = [ |
| malloc_library[CcInfo].linking_context, |
| _get_std_and_alloc_info(ctx, toolchain, crate_info).linking_context, |
| toolchain.stdlib_linkflags.linking_context, |
| ] |
| |
| for dep in crate_info.deps.to_list(): |
| if dep.cc_info: |
| linking_contexts.append(dep.cc_info.linking_context) |
| |
| # In the cc_common.link action we need to pass the name of the final |
| # binary (output) relative to the package of this target. |
| # We compute it by stripping the path to the package directory, |
| # which is a prefix of the path of `crate_info.output`. |
| |
| # The path to the package dir, including a trailing "/". |
| package_dir = ctx.bin_dir.path + "/" |
| |
| # For external repositories, workspace root is not part of the output |
| # path when sibling repository layout is used (the repository name is |
| # part of the bin_dir). This scenario happens when the workspace root |
| # starts with "../" |
| if ctx.label.workspace_root and not ctx.label.workspace_root.startswith("../"): |
| package_dir = package_dir + ctx.label.workspace_root + "/" |
| if ctx.label.package: |
| package_dir = package_dir + ctx.label.package + "/" |
| |
| if not crate_info.output.path.startswith(package_dir): |
| fail("The package dir path", package_dir, "should be a prefix of the crate_info.output.path", crate_info.output.path) |
| |
| output_relative_to_package = crate_info.output.path[len(package_dir):] |
| |
| # Compile actions that produce shared libraries create output of the form "libfoo.so" for linux and macos; |
| # cc_common.link expects us to pass "foo" to the name parameter. We cannot simply use crate_info.name because |
| # the name of the crate does not always match the name of output file, e.g a crate named foo-bar will produce |
| # a (lib)foo_bar output file. |
| if crate_info.type == "cdylib": |
| output_lib = crate_info.output.basename |
| if toolchain.target_os != "windows": |
| # Strip the leading "lib" prefix |
| output_lib = output_lib[3:] |
| |
| # Strip the file extension |
| output_lib = output_lib[:-(1 + len(crate_info.output.extension))] |
| |
| # Remove the basename (which contains the undesired 'lib' prefix and the file extension) |
| output_relative_to_package = output_relative_to_package[:-len(crate_info.output.basename)] |
| |
| # Append the name of the library |
| output_relative_to_package = output_relative_to_package + output_lib |
| |
| cc_common.link( |
| actions = ctx.actions, |
| feature_configuration = feature_configuration, |
| cc_toolchain = cc_toolchain, |
| linking_contexts = linking_contexts, |
| compilation_outputs = compilation_outputs, |
| name = output_relative_to_package, |
| stamp = ctx.attr.stamp, |
| output_type = "executable" if crate_info.type == "bin" else "dynamic_library", |
| ) |
| |
| outputs = [crate_info.output] |
| |
| coverage_runfiles = [] |
| if toolchain.llvm_cov and ctx.configuration.coverage_enabled and crate_info.is_test: |
| coverage_runfiles = [toolchain.llvm_cov, toolchain.llvm_profdata] |
| |
| experimental_use_coverage_metadata_files = toolchain._experimental_use_coverage_metadata_files |
| |
| dynamic_libraries = [ |
| library_to_link.dynamic_library |
| for dep in getattr(ctx.attr, "deps", []) |
| if CcInfo in dep |
| for linker_input in dep[CcInfo].linking_context.linker_inputs.to_list() |
| for library_to_link in linker_input.libraries |
| if _is_dylib(library_to_link) |
| ] |
| runfiles = ctx.runfiles( |
| files = getattr(ctx.files, "data", []) + |
| ([] if experimental_use_coverage_metadata_files else coverage_runfiles) + |
| dynamic_libraries, |
| collect_data = True, |
| ) |
| if getattr(ctx.attr, "crate", None): |
| runfiles = runfiles.merge(ctx.attr.crate[DefaultInfo].default_runfiles) |
| runfiles = runfiles.merge(ctx.attr.crate[DefaultInfo].data_runfiles) |
| |
| # TODO: Remove after some resolution to |
| # https://github.com/bazelbuild/rules_rust/issues/771 |
| out_binary = getattr(attr, "out_binary", False) |
| |
| executable = crate_info.output if crate_info.type == "bin" or crate_info.is_test or out_binary else None |
| |
| instrumented_files_kwargs = { |
| "dependency_attributes": ["deps", "crate"], |
| "extensions": ["rs"], |
| "source_attributes": ["srcs"], |
| } |
| |
| if experimental_use_coverage_metadata_files: |
| instrumented_files_kwargs.update({ |
| "metadata_files": coverage_runfiles + [executable] if executable else [], |
| }) |
| |
| providers = [ |
| DefaultInfo( |
| # nb. This field is required for cc_library to depend on our output. |
| files = depset(outputs), |
| runfiles = runfiles, |
| executable = executable, |
| ), |
| ] |
| |
| # When invoked by aspects (and when running `bazel coverage`), the |
| # baseline_coverage.dat created here will conflict with the baseline_coverage.dat of the |
| # underlying target, which is a build failure. So we add an option to disable it so that this |
| # function can be invoked from aspects for rules that have its own InstrumentedFilesInfo. |
| if include_coverage: |
| providers.append( |
| coverage_common.instrumented_files_info( |
| ctx, |
| **instrumented_files_kwargs |
| ), |
| ) |
| |
| if crate_info_dict != None: |
| crate_info_dict.update({ |
| "rustc_env": env, |
| }) |
| crate_info = rust_common.create_crate_info(**crate_info_dict) |
| |
| if crate_info.type in ["staticlib", "cdylib"]: |
| # These rules are not supposed to be depended on by other rust targets, and |
| # as such they shouldn't provide a CrateInfo. However, one may still want to |
| # write a rust_test for them, so we provide the CrateInfo wrapped in a provider |
| # that rust_test understands. |
| providers.extend([rust_common.test_crate_info(crate = crate_info), dep_info]) |
| else: |
| providers.extend([crate_info, dep_info]) |
| |
| providers += establish_cc_info(ctx, attr, crate_info, toolchain, cc_toolchain, feature_configuration, interface_library) |
| |
| output_group_info = {} |
| |
| if pdb_file: |
| output_group_info["pdb_file"] = depset([pdb_file]) |
| if dsym_folder: |
| output_group_info["dsym_folder"] = depset([dsym_folder]) |
| if build_metadata: |
| output_group_info["build_metadata"] = depset([build_metadata]) |
| if build_metadata: |
| output_group_info["build_metadata"] = depset([build_metadata]) |
| if rustc_rmeta_output: |
| output_group_info["rustc_rmeta_output"] = depset([rustc_rmeta_output]) |
| if rustc_output: |
| output_group_info["rustc_output"] = depset([rustc_output]) |
| |
| if output_group_info: |
| providers.append(OutputGroupInfo(**output_group_info)) |
| |
| return providers |
| |
| def _is_no_std(ctx, toolchain, crate_info): |
| if is_exec_configuration(ctx) or crate_info.is_test: |
| return False |
| if toolchain._no_std == "off": |
| return False |
| return True |
| |
| def _get_std_and_alloc_info(ctx, toolchain, crate_info): |
| if is_exec_configuration(ctx): |
| return toolchain.libstd_and_allocator_ccinfo |
| if toolchain._experimental_use_global_allocator: |
| if _is_no_std(ctx, toolchain, crate_info): |
| return toolchain.nostd_and_global_allocator_cc_info |
| else: |
| return toolchain.libstd_and_global_allocator_ccinfo |
| else: |
| return toolchain.libstd_and_allocator_ccinfo |
| |
| def _is_dylib(dep): |
| return not bool(dep.static_library or dep.pic_static_library) |
| |
| def _collect_nonstatic_linker_inputs(cc_info): |
| shared_linker_inputs = [] |
| for linker_input in cc_info.linking_context.linker_inputs.to_list(): |
| dylibs = [ |
| lib |
| for lib in linker_input.libraries |
| if _is_dylib(lib) |
| ] |
| if dylibs: |
| shared_linker_inputs.append(cc_common.create_linker_input( |
| owner = linker_input.owner, |
| libraries = depset(dylibs), |
| )) |
| return shared_linker_inputs |
| |
| def establish_cc_info(ctx, attr, crate_info, toolchain, cc_toolchain, feature_configuration, interface_library): |
| """If the produced crate is suitable yield a CcInfo to allow for interop with cc rules |
| |
| Args: |
| ctx (ctx): The rule's context object |
| attr (struct): Attributes to use in gathering CcInfo |
| crate_info (CrateInfo): The CrateInfo provider of the target crate |
| toolchain (rust_toolchain): The current `rust_toolchain` |
| cc_toolchain (CcToolchainInfo): The current `CcToolchainInfo` |
| feature_configuration (FeatureConfiguration): Feature configuration to be queried. |
| interface_library (File): Optional interface library for cdylib crates on Windows. |
| |
| Returns: |
| list: A list containing the CcInfo provider |
| """ |
| |
| # A test will not need to produce CcInfo as nothing can depend on test targets |
| if crate_info.is_test: |
| return [] |
| |
| # Only generate CcInfo for particular crate types |
| if crate_info.type not in ("staticlib", "cdylib", "rlib", "lib"): |
| return [] |
| |
| # TODO: Remove after some resolution to |
| # https://github.com/bazelbuild/rules_rust/issues/771 |
| if getattr(attr, "out_binary", False): |
| return [] |
| |
| if crate_info.type == "staticlib": |
| library_to_link = cc_common.create_library_to_link( |
| actions = ctx.actions, |
| feature_configuration = feature_configuration, |
| cc_toolchain = cc_toolchain, |
| static_library = crate_info.output, |
| # TODO(hlopko): handle PIC/NOPIC correctly |
| pic_static_library = crate_info.output, |
| alwayslink = getattr(attr, "alwayslink", False), |
| ) |
| elif crate_info.type in ("rlib", "lib"): |
| # bazel hard-codes a check for endswith((".a", ".pic.a", |
| # ".lib")) in create_library_to_link, so we work around that |
| # by creating a symlink to the .rlib with a .a extension. |
| dot_a = make_static_lib_symlink(ctx.label.package, ctx.actions, crate_info.output) |
| |
| # TODO(hlopko): handle PIC/NOPIC correctly |
| library_to_link = cc_common.create_library_to_link( |
| actions = ctx.actions, |
| feature_configuration = feature_configuration, |
| cc_toolchain = cc_toolchain, |
| static_library = dot_a, |
| # TODO(hlopko): handle PIC/NOPIC correctly |
| pic_static_library = dot_a, |
| alwayslink = getattr(attr, "alwayslink", False), |
| ) |
| elif crate_info.type == "cdylib": |
| library_to_link = cc_common.create_library_to_link( |
| actions = ctx.actions, |
| feature_configuration = feature_configuration, |
| cc_toolchain = cc_toolchain, |
| dynamic_library = crate_info.output, |
| interface_library = interface_library, |
| ) |
| else: |
| fail("Unexpected case") |
| |
| link_input = cc_common.create_linker_input( |
| owner = ctx.label, |
| libraries = depset([library_to_link]), |
| ) |
| |
| linking_context = cc_common.create_linking_context( |
| # TODO - What to do for no_std? |
| linker_inputs = depset([link_input]), |
| ) |
| |
| cc_infos = [ |
| CcInfo(linking_context = linking_context), |
| toolchain.stdlib_linkflags, |
| ] |
| |
| # Flattening is okay since crate_info.deps only records direct deps. |
| for dep in crate_info.deps.to_list(): |
| if dep.cc_info: |
| # A Rust staticlib or shared library doesn't need to propagate linker inputs |
| # of its dependencies, except for shared libraries. |
| if crate_info.type in ["cdylib", "staticlib"]: |
| shared_linker_inputs = _collect_nonstatic_linker_inputs(dep.cc_info) |
| if shared_linker_inputs: |
| linking_context = cc_common.create_linking_context( |
| linker_inputs = depset(shared_linker_inputs), |
| ) |
| cc_infos.append(CcInfo(linking_context = linking_context)) |
| else: |
| cc_infos.append(dep.cc_info) |
| |
| if crate_info.type in ("rlib", "lib"): |
| libstd_and_allocator_cc_info = _get_std_and_alloc_info(ctx, toolchain, crate_info) |
| if libstd_and_allocator_cc_info: |
| # TODO: if we already have an rlib in our deps, we could skip this |
| cc_infos.append(libstd_and_allocator_cc_info) |
| |
| return [cc_common.merge_cc_infos(cc_infos = cc_infos)] |
| |
| def add_edition_flags(args, crate): |
| """Adds the Rust edition flag to an arguments object reference |
| |
| Args: |
| args (Args): A reference to an Args object |
| crate (CrateInfo): A CrateInfo provider |
| """ |
| if crate.edition != "2015": |
| args.add(crate.edition, format = "--edition=%s") |
| |
| def _create_extra_input_args(build_info, dep_info, include_link_flags = True): |
| """Gather additional input arguments from transitive dependencies |
| |
| Args: |
| build_info (BuildInfo): The BuildInfo provider from the target Crate's set of inputs. |
| dep_info (DepInfo): The Depinfo provider form the target Crate's set of inputs. |
| include_link_flags (bool, optional): Whether to include flags like `-l` that instruct the linker to search for a library. |
| |
| Returns: |
| tuple: A tuple of the following items: |
| - (depset[File]): A list of all build info `OUT_DIR` File objects |
| - (str): The `OUT_DIR` of the current build info |
| - (File): An optional generated environment file from a `cargo_build_script` target |
| - (depset[File]): All direct and transitive build flag files from the current build info to be passed to rustc. |
| """ |
| input_files = [] |
| input_depsets = [] |
| |
| # Arguments to the commandline line wrapper that are going to be used |
| # to create the final command line |
| out_dir = None |
| build_env_file = None |
| build_flags_files = [] |
| |
| if build_info: |
| if build_info.out_dir: |
| out_dir = build_info.out_dir.path |
| input_files.append(build_info.out_dir) |
| build_env_file = build_info.rustc_env |
| if build_info.flags: |
| build_flags_files.append(build_info.flags) |
| if build_info.linker_flags and include_link_flags: |
| build_flags_files.append(build_info.linker_flags) |
| input_files.append(build_info.linker_flags) |
| |
| input_depsets.append(build_info.compile_data) |
| |
| return ( |
| depset(input_files, transitive = [dep_info.link_search_path_files] + input_depsets), |
| out_dir, |
| build_env_file, |
| depset(build_flags_files, transitive = [dep_info.link_search_path_files]), |
| ) |
| |
| def _compute_rpaths(toolchain, output_dir, dep_info, use_pic): |
| """Determine the artifact's rpaths relative to the bazel root for runtime linking of shared libraries. |
| |
| Args: |
| toolchain (rust_toolchain): The current `rust_toolchain` |
| output_dir (str): The output directory of the current target |
| dep_info (DepInfo): The current target's dependency info |
| use_pic: If set, prefers pic_static_library over static_library. |
| |
| Returns: |
| depset: A set of relative paths from the output directory to each dependency |
| """ |
| |
| # Windows has no rpath equivalent, so always return an empty depset. |
| # Fuchsia assembles shared libraries during packaging. |
| if toolchain.target_os == "windows" or toolchain.target_os == "fuchsia": |
| return depset([]) |
| |
| dylibs = [ |
| get_preferred_artifact(lib, use_pic) |
| for linker_input in dep_info.transitive_noncrates.to_list() |
| for lib in linker_input.libraries |
| if _is_dylib(lib) |
| ] |
| |
| # Include std dylib if dylib linkage is enabled |
| if toolchain._experimental_link_std_dylib: |
| # TODO: Make toolchain.rust_std to only include libstd.so |
| # When dylib linkage is enabled, toolchain.rust_std should only need to |
| # include libstd.so. Hence, no filtering needed. |
| for file in toolchain.rust_std.to_list(): |
| if is_std_dylib(file): |
| dylibs.append(file) |
| |
| if not dylibs: |
| return depset([]) |
| |
| # For darwin, dylibs compiled by Bazel will fail to be resolved at runtime |
| # without a version of Bazel that includes |
| # https://github.com/bazelbuild/bazel/pull/13427. This is known to not be |
| # included in Bazel 4.1 and below. |
| if toolchain.target_os not in ["linux", "darwin", "android"]: |
| fail("Runtime linking is not supported on {}, but found {}".format( |
| toolchain.target_os, |
| dep_info.transitive_noncrates, |
| )) |
| |
| # Multiple dylibs can be present in the same directory, so deduplicate them. |
| return depset([ |
| relativize(lib_dir, output_dir) |
| for lib_dir in _get_dir_names(dylibs) |
| ]) |
| |
| def _get_dir_names(files): |
| """Returns a list of directory names from the given list of File objects |
| |
| Args: |
| files (list): A list of File objects |
| |
| Returns: |
| list: A list of directory names for all files |
| """ |
| dirs = {} |
| for f in files: |
| dirs[f.dirname] = None |
| return dirs.keys() |
| |
| def add_crate_link_flags(args, dep_info, force_all_deps_direct = False, use_metadata = False): |
| """Adds link flags to an Args object reference |
| |
| Args: |
| args (Args): An arguments object reference |
| dep_info (DepInfo): The current target's dependency info |
| force_all_deps_direct (bool, optional): Whether to pass the transitive rlibs with --extern |
| to the commandline as opposed to -L. |
| use_metadata (bool, optional): Build command line arugments using metadata for crates that provide it. |
| """ |
| |
| direct_crates = depset( |
| transitive = [ |
| dep_info.direct_crates, |
| dep_info.transitive_crates, |
| ], |
| ) if force_all_deps_direct else dep_info.direct_crates |
| |
| crate_to_link_flags = _crate_to_link_flag_metadata if use_metadata else _crate_to_link_flag |
| args.add_all(direct_crates, uniquify = True, map_each = crate_to_link_flags) |
| |
| args.add_all( |
| dep_info.transitive_crates, |
| map_each = _get_crate_dirname, |
| uniquify = True, |
| format_each = "-Ldependency=%s", |
| ) |
| |
| def _crate_to_link_flag_metadata(crate): |
| """A helper macro used by `add_crate_link_flags` for adding crate link flags to a Arg object |
| |
| Args: |
| crate (CrateInfo|AliasableDepInfo): A CrateInfo or an AliasableDepInfo provider |
| |
| Returns: |
| list: Link flags for the given provider |
| """ |
| |
| # This is AliasableDepInfo, we should use the alias as a crate name |
| if hasattr(crate, "dep"): |
| name = crate.name |
| crate_info = crate.dep |
| else: |
| name = crate.name |
| crate_info = crate |
| |
| lib_or_meta = crate_info.metadata |
| if not crate_info.metadata: |
| lib_or_meta = crate_info.output |
| return ["--extern={}={}".format(name, lib_or_meta.path)] |
| |
| def _crate_to_link_flag(crate): |
| """A helper macro used by `add_crate_link_flags` for adding crate link flags to a Arg object |
| |
| Args: |
| crate (CrateInfo|AliasableDepInfo): A CrateInfo or an AliasableDepInfo provider |
| |
| Returns: |
| list: Link flags for the given provider |
| """ |
| |
| # This is AliasableDepInfo, we should use the alias as a crate name |
| if hasattr(crate, "dep"): |
| name = crate.name |
| crate_info = crate.dep |
| else: |
| name = crate.name |
| crate_info = crate |
| return ["--extern={}={}".format(name, crate_info.output.path)] |
| |
| def _get_crate_dirname(crate): |
| """A helper macro used by `add_crate_link_flags` for getting the directory name of the current crate's output path |
| |
| Args: |
| crate (CrateInfo): A CrateInfo provider from the current rule |
| |
| Returns: |
| str: The directory name of the the output File that will be produced. |
| """ |
| return crate.output.dirname |
| |
| def _portable_link_flags(lib, use_pic, ambiguous_libs, get_lib_name, for_windows = False, for_darwin = False, flavor_msvc = False): |
| artifact = get_preferred_artifact(lib, use_pic) |
| if ambiguous_libs and artifact.path in ambiguous_libs: |
| artifact = ambiguous_libs[artifact.path] |
| if lib.static_library or lib.pic_static_library: |
| # To ensure appropriate linker library argument order, in the presence |
| # of both native libraries that depend on rlibs and rlibs that depend |
| # on native libraries, we use an approach where we "sandwich" the |
| # rust libraries between two similar sections of all of native |
| # libraries: |
| # n1 n2 ... r1 r2 ... n1 n2 ... |
| # A B C |
| # This way any dependency from a native library to a rust library |
| # is resolved from A to B, and any dependency from a rust library to |
| # a native one is resolved from B to C. |
| # The question of resolving dependencies from a native library from A |
| # to any rust library is addressed in a different place, where we |
| # create symlinks to the rlibs, pretending they are native libraries, |
| # and adding references to these symlinks in the native section A. |
| # We rely in the behavior of -Clink-arg to put the linker args |
| # at the end of the linker invocation constructed by rustc. |
| |
| # We skip adding `-Clink-arg=-l` for libstd and libtest from the standard library, as |
| # these two libraries are present both as an `.rlib` and a `.so` format. |
| # On linux, Rustc adds a -Bdynamic to the linker command line before the libraries specified |
| # with `-Clink-arg`, which leads to us linking against the `.so`s but not putting the |
| # corresponding value to the runtime library search paths, which results in a |
| # "cannot open shared object file: No such file or directory" error at exectuion time. |
| # We can fix this by adding a `-Clink-arg=-Bstatic` on linux, but we don't have that option for |
| # macos. The proper solution for this issue would be to remove `libtest-{hash}.so` and `libstd-{hash}.so` |
| # from the toolchain. However, it is not enough to change the toolchain's `rust_std_{...}` filegroups |
| # here: https://github.com/bazelbuild/rules_rust/blob/a9d5d894ad801002d007b858efd154e503796b9f/rust/private/repository_utils.bzl#L144 |
| # because rustc manages to escape the sandbox and still finds them at linking time. |
| # We need to modify the repository rules to erase those files completely. |
| if "lib/rustlib" in artifact.path and ( |
| artifact.basename.startswith("libtest-") or artifact.basename.startswith("libstd-") or |
| artifact.basename.startswith("test-") or artifact.basename.startswith("std-") |
| ): |
| return [] if for_darwin else ["-lstatic=%s" % get_lib_name(artifact)] |
| |
| if for_windows: |
| if flavor_msvc: |
| return [ |
| "-lstatic=%s" % get_lib_name(artifact), |
| "-Clink-arg={}".format(artifact.basename), |
| ] |
| else: |
| return [ |
| "-lstatic=%s" % get_lib_name(artifact), |
| "-Clink-arg=-l{}".format(artifact.basename), |
| ] |
| else: |
| return [ |
| "-lstatic=%s" % get_lib_name(artifact), |
| "-Clink-arg=-l{}".format(get_lib_name(artifact)), |
| ] |
| elif _is_dylib(lib): |
| return [ |
| "-ldylib=%s" % get_lib_name(artifact), |
| ] |
| |
| return [] |
| |
| def _make_link_flags_windows(make_link_flags_args, flavor_msvc): |
| linker_input, use_pic, ambiguous_libs, include_link_flags = make_link_flags_args |
| ret = [] |
| for lib in linker_input.libraries: |
| if lib.alwayslink: |
| if flavor_msvc: |
| ret.extend(["-C", "link-arg=/WHOLEARCHIVE:%s" % get_preferred_artifact(lib, use_pic).path]) |
| else: |
| ret.extend([ |
| "-C", |
| "link-arg=-Wl,--whole-archive", |
| "-C", |
| ("link-arg=%s" % get_preferred_artifact(lib, use_pic).path), |
| "-C", |
| "link-arg=-Wl,--no-whole-archive", |
| ]) |
| elif include_link_flags: |
| ret.extend(_portable_link_flags(lib, use_pic, ambiguous_libs, get_lib_name_for_windows, for_windows = True, flavor_msvc = flavor_msvc)) |
| return ret |
| |
| def _make_link_flags_windows_msvc(make_link_flags_args): |
| return _make_link_flags_windows(make_link_flags_args, flavor_msvc = True) |
| |
| def _make_link_flags_windows_gnu(make_link_flags_args): |
| return _make_link_flags_windows(make_link_flags_args, flavor_msvc = False) |
| |
| def _make_link_flags_darwin(make_link_flags_args): |
| linker_input, use_pic, ambiguous_libs, include_link_flags = make_link_flags_args |
| ret = [] |
| for lib in linker_input.libraries: |
| if lib.alwayslink: |
| ret.extend([ |
| "-C", |
| ("link-arg=-Wl,-force_load,%s" % get_preferred_artifact(lib, use_pic).path), |
| ]) |
| elif include_link_flags: |
| ret.extend(_portable_link_flags(lib, use_pic, ambiguous_libs, get_lib_name_default, for_darwin = True)) |
| return ret |
| |
| def _make_link_flags_default(make_link_flags_args): |
| linker_input, use_pic, ambiguous_libs, include_link_flags = make_link_flags_args |
| ret = [] |
| for lib in linker_input.libraries: |
| if lib.alwayslink: |
| ret.extend([ |
| "-C", |
| "link-arg=-Wl,--whole-archive", |
| "-C", |
| ("link-arg=%s" % get_preferred_artifact(lib, use_pic).path), |
| "-C", |
| "link-arg=-Wl,--no-whole-archive", |
| ]) |
| elif include_link_flags: |
| ret.extend(_portable_link_flags(lib, use_pic, ambiguous_libs, get_lib_name_default)) |
| return ret |
| |
| def _libraries_dirnames(make_link_flags_args): |
| link_input, use_pic, _, _ = make_link_flags_args |
| |
| # De-duplicate names. |
| return depset([get_preferred_artifact(lib, use_pic).dirname for lib in link_input.libraries]).to_list() |
| |
| def _add_native_link_flags(args, dep_info, linkstamp_outs, ambiguous_libs, crate_type, toolchain, cc_toolchain, feature_configuration, compilation_mode, include_link_flags = True): |
| """Adds linker flags for all dependencies of the current target. |
| |
| Args: |
| args (Args): The Args struct for a ctx.action |
| dep_info (DepInfo): Dependency Info provider |
| linkstamp_outs (list): Linkstamp outputs of native dependencies |
| ambiguous_libs (dict): Ambiguous libs, see `_disambiguate_libs` |
| crate_type: Crate type of the current target |
| toolchain (rust_toolchain): The current `rust_toolchain` |
| cc_toolchain (CcToolchainInfo): The current `cc_toolchain` |
| feature_configuration (FeatureConfiguration): feature configuration to use with cc_toolchain |
| compilation_mode (bool): The compilation mode for this build. |
| include_link_flags (bool, optional): Whether to include flags like `-l` that instruct the linker to search for a library. |
| """ |
| if crate_type in ["lib", "rlib"]: |
| return |
| |
| use_pic = _should_use_pic(cc_toolchain, feature_configuration, crate_type, compilation_mode) |
| |
| if toolchain.target_os == "windows": |
| make_link_flags = _make_link_flags_windows_msvc if toolchain.target_triple.abi == "msvc" else _make_link_flags_windows_gnu |
| get_lib_name = get_lib_name_for_windows |
| elif toolchain.target_os.startswith(("mac", "darwin", "ios")): |
| make_link_flags = _make_link_flags_darwin |
| get_lib_name = get_lib_name_default |
| else: |
| make_link_flags = _make_link_flags_default |
| get_lib_name = get_lib_name_default |
| |
| # TODO(hlopko): Remove depset flattening by using lambdas once we are on >=Bazel 5.0 |
| make_link_flags_args = [(arg, use_pic, ambiguous_libs, include_link_flags) for arg in dep_info.transitive_noncrates.to_list()] |
| args.add_all(make_link_flags_args, map_each = _libraries_dirnames, uniquify = True, format_each = "-Lnative=%s") |
| if ambiguous_libs: |
| # If there are ambiguous libs, the disambiguation symlinks to them are |
| # all created in the same directory. Add it to the library search path. |
| ambiguous_libs_dirname = ambiguous_libs.values()[0].dirname |
| args.add(ambiguous_libs_dirname, format = "-Lnative=%s") |
| |
| args.add_all(make_link_flags_args, map_each = make_link_flags) |
| |
| args.add_all(linkstamp_outs, before_each = "-C", format_each = "link-args=%s") |
| |
| if crate_type in ["dylib", "cdylib"]: |
| # For shared libraries we want to link C++ runtime library dynamically |
| # (for example libstdc++.so or libc++.so). |
| args.add_all( |
| cc_toolchain.dynamic_runtime_lib(feature_configuration = feature_configuration), |
| map_each = _get_dirname, |
| format_each = "-Lnative=%s", |
| ) |
| if include_link_flags: |
| args.add_all( |
| cc_toolchain.dynamic_runtime_lib(feature_configuration = feature_configuration), |
| map_each = get_lib_name, |
| format_each = "-ldylib=%s", |
| ) |
| else: |
| # For all other crate types we want to link C++ runtime library statically |
| # (for example libstdc++.a or libc++.a). |
| args.add_all( |
| cc_toolchain.static_runtime_lib(feature_configuration = feature_configuration), |
| map_each = _get_dirname, |
| format_each = "-Lnative=%s", |
| ) |
| if include_link_flags: |
| args.add_all( |
| cc_toolchain.static_runtime_lib(feature_configuration = feature_configuration), |
| map_each = get_lib_name, |
| format_each = "-lstatic=%s", |
| ) |
| |
| def _get_dirname(file): |
| """A helper function for `_add_native_link_flags`. |
| |
| Args: |
| file (File): The target file |
| |
| Returns: |
| str: Directory name of `file` |
| """ |
| return file.dirname |
| |
| def _add_per_crate_rustc_flags(ctx, args, crate_info, per_crate_rustc_flags): |
| """Adds matching per-crate rustc flags to an arguments object reference |
| |
| Args: |
| ctx (ctx): The source rule's context object |
| args (Args): A reference to an Args object |
| crate_info (CrateInfo): A CrateInfo provider |
| per_crate_rustc_flags (list): A list of per_crate_rustc_flag values |
| """ |
| for per_crate_rustc_flag in per_crate_rustc_flags: |
| at_index = per_crate_rustc_flag.find("@") |
| if at_index == -1: |
| fail("per_crate_rustc_flag '{}' does not follow the expected format: prefix_filter@flag".format(per_crate_rustc_flag)) |
| |
| prefix_filter = per_crate_rustc_flag[:at_index] |
| flag = per_crate_rustc_flag[at_index + 1:] |
| if not flag: |
| fail("per_crate_rustc_flag '{}' does not follow the expected format: prefix_filter@flag".format(per_crate_rustc_flag)) |
| |
| label_string = str(ctx.label) |
| label = label_string[1:] if label_string.startswith("@//") else label_string |
| execution_path = crate_info.root.path |
| |
| if label.startswith(prefix_filter) or execution_path.startswith(prefix_filter): |
| args.add(flag) |
| |
| def _error_format_impl(ctx): |
| """Implementation of the `error_format` rule |
| |
| Args: |
| ctx (ctx): The rule's context object |
| |
| Returns: |
| list: A list containing the ErrorFormatInfo provider |
| """ |
| raw = ctx.build_setting_value |
| if raw not in _error_format_values: |
| fail("{} expected a value in `{}` but got `{}`".format( |
| ctx.label, |
| _error_format_values, |
| raw, |
| )) |
| return [ErrorFormatInfo(error_format = raw)] |
| |
| error_format = rule( |
| doc = ( |
| "Change the [--error-format](https://doc.rust-lang.org/rustc/command-line-arguments.html#option-error-format) " + |
| "flag from the command line with `--@rules_rust//:error_format`. See rustc documentation for valid values." |
| ), |
| implementation = _error_format_impl, |
| build_setting = config.string(flag = True), |
| ) |
| |
| def _rustc_output_diagnostics_impl(ctx): |
| """Implementation of the `rustc_output_diagnostics` rule |
| |
| Args: |
| ctx (ctx): The rule's context object |
| |
| Returns: |
| list: A list containing the RustcOutputDiagnosticsInfo provider |
| """ |
| return [RustcOutputDiagnosticsInfo( |
| rustc_output_diagnostics = ctx.build_setting_value, |
| )] |
| |
| rustc_output_diagnostics = rule( |
| doc = ( |
| "Setting this flag from the command line with `--@rules_rust//:rustc_output_diagnostics` " + |
| "makes rules_rust save rustc json output(suitable for consumption by rust-analyzer) in a file. " + |
| "These are accessible via the " + |
| "`rustc_rmeta_output`(for pipelined compilation) and `rustc_output` output groups. " + |
| "You can find these using `bazel cquery`" |
| ), |
| implementation = _rustc_output_diagnostics_impl, |
| build_setting = config.bool(flag = True), |
| ) |
| |
| def _extra_rustc_flags_impl(ctx): |
| return ExtraRustcFlagsInfo(extra_rustc_flags = ctx.build_setting_value) |
| |
| extra_rustc_flags = rule( |
| doc = ( |
| "Add additional rustc_flags from the command line with `--@rules_rust//:extra_rustc_flags`. " + |
| "This flag should only be used for flags that need to be applied across the entire build. For options that " + |
| "apply to individual crates, use the rustc_flags attribute on the individual crate's rule instead. NOTE: " + |
| "These flags not applied to the exec configuration (proc-macros, cargo_build_script, etc); " + |
| "use `--@rules_rust//:extra_exec_rustc_flags` to apply flags to the exec configuration." |
| ), |
| implementation = _extra_rustc_flags_impl, |
| build_setting = config.string_list(flag = True), |
| ) |
| |
| def _extra_rustc_flag_impl(ctx): |
| return ExtraRustcFlagsInfo(extra_rustc_flags = [f for f in ctx.build_setting_value if f != ""]) |
| |
| extra_rustc_flag = rule( |
| doc = ( |
| "Add additional rustc_flag from the command line with `--@rules_rust//:extra_rustc_flag`. " + |
| "Multiple uses are accumulated and appended after the extra_rustc_flags." |
| ), |
| implementation = _extra_rustc_flag_impl, |
| build_setting = config.string(flag = True, allow_multiple = True), |
| ) |
| |
| def _extra_exec_rustc_flags_impl(ctx): |
| return ExtraExecRustcFlagsInfo(extra_exec_rustc_flags = ctx.build_setting_value) |
| |
| extra_exec_rustc_flags = rule( |
| doc = ( |
| "Add additional rustc_flags in the exec configuration from the command line with `--@rules_rust//:extra_exec_rustc_flags`. " + |
| "This flag should only be used for flags that need to be applied across the entire build. " + |
| "These flags only apply to the exec configuration (proc-macros, cargo_build_script, etc)." |
| ), |
| implementation = _extra_exec_rustc_flags_impl, |
| build_setting = config.string_list(flag = True), |
| ) |
| |
| def _extra_exec_rustc_flag_impl(ctx): |
| return ExtraExecRustcFlagsInfo(extra_exec_rustc_flags = [f for f in ctx.build_setting_value if f != ""]) |
| |
| extra_exec_rustc_flag = rule( |
| doc = ( |
| "Add additional rustc_flags in the exec configuration from the command line with `--@rules_rust//:extra_exec_rustc_flag`. " + |
| "Multiple uses are accumulated and appended after the extra_exec_rustc_flags." |
| ), |
| implementation = _extra_exec_rustc_flag_impl, |
| build_setting = config.string(flag = True, allow_multiple = True), |
| ) |
| |
| def _per_crate_rustc_flag_impl(ctx): |
| return PerCrateRustcFlagsInfo(per_crate_rustc_flags = [f for f in ctx.build_setting_value if f != ""]) |
| |
| per_crate_rustc_flag = rule( |
| doc = ( |
| "Add additional rustc_flag to matching crates from the command line with `--@rules_rust//:experimental_per_crate_rustc_flag`. " + |
| "The expected flag format is prefix_filter@flag, where any crate with a label or execution path starting with the prefix filter will be built with the given flag." + |
| "The label matching uses the canonical form of the label (i.e //package:label_name)." + |
| "The execution path is the relative path to your workspace directory including the base name (including extension) of the crate root." + |
| "This flag is only applied to the exec configuration (proc-macros, cargo_build_script, etc)." + |
| "Multiple uses are accumulated." |
| ), |
| implementation = _per_crate_rustc_flag_impl, |
| build_setting = config.string(flag = True, allow_multiple = True), |
| ) |
| |
| def _no_std_impl(ctx): |
| value = str(ctx.attr._no_std[BuildSettingInfo].value) |
| if is_exec_configuration(ctx): |
| return [config_common.FeatureFlagInfo(value = "off")] |
| return [config_common.FeatureFlagInfo(value = value)] |
| |
| no_std = rule( |
| doc = ( |
| "No std; we need this so that we can distinguish between host and exec" |
| ), |
| attrs = { |
| "_no_std": attr.label(default = "//:no_std"), |
| }, |
| implementation = _no_std_impl, |
| ) |