diff options
Diffstat (limited to 'pw_toolchain/rust/defs.bzl')
| -rw-r--r-- | pw_toolchain/rust/defs.bzl | 216 |
1 files changed, 216 insertions, 0 deletions
diff --git a/pw_toolchain/rust/defs.bzl b/pw_toolchain/rust/defs.bzl new file mode 100644 index 000000000..f3f58d218 --- /dev/null +++ b/pw_toolchain/rust/defs.bzl @@ -0,0 +1,216 @@ +# Copyright 2023 The Pigweed Authors +# +# 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 +# +# https://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. +"""Utilities for declaring Rust toolchains that are compatible with Arm gcc.""" + +load("@rules_rust//rust:toolchain.bzl", "rust_analyzer_toolchain", "rust_toolchain") +load("//pw_env_setup/bazel/cipd_setup:cipd_rules.bzl", "cipd_repository") + +HOSTS = [ + { + "cpu": "aarch64", + "cipd_arch": "arm64", + "os": "linux", + "triple": "aarch64-unknown-linux-gnu", + "dylib_ext": ".so", + }, + { + "cpu": "x86_64", + "cipd_arch": "amd64", + "os": "linux", + "triple": "x86_64-unknown-linux-gnu", + "dylib_ext": ".so", + }, + { + "cpu": "aarch64", + "cipd_arch": "arm64", + "os": "macos", + "triple": "aarch64-apple-darwin", + "dylib_ext": ".dylib", + }, + { + "cpu": "x86_64", + "cipd_arch": "amd64", + "os": "macos", + "triple": "x86_64-apple-darwin", + "dylib_ext": ".dylib", + }, +] + +EXTRA_TARGETS = [ + { + "cpu": "armv6-m", + "triple": "thumbv6m-none-eabi", + }, + { + "cpu": "armv7-m", + "triple": "thumbv7m-none-eabi", + }, + { + "cpu": "armv7e-m", + "triple": "thumbv7m-none-eabi", + }, +] + +# buildifier: disable=unnamed-macro +def pw_rust_register_toolchain_and_target_repos(cipd_tag): + """Declare and register CIPD repos for Rust toolchain and target rupport. + + Args: + cipd_tag: Tag with which to select specific package versions. + """ + for host in HOSTS: + cipd_os = host["os"] + if cipd_os == "macos": + cipd_os = "mac" + + cipd_repository( + name = "rust_toolchain_host_{}_{}".format(host["os"], host["cpu"]), + build_file = "//pw_toolchain/rust:rust_toolchain.BUILD", + path = "fuchsia/third_party/rust/host/{}-{}".format(cipd_os, host["cipd_arch"]), + tag = cipd_tag, + ) + + cipd_repository( + name = "rust_toolchain_target_{}".format(host["triple"]), + build_file = "//pw_toolchain/rust:rust_stdlib.BUILD", + path = "fuchsia/third_party/rust/target/{}".format(host["triple"]), + tag = cipd_tag, + ) + + for target in EXTRA_TARGETS: + cipd_repository( + name = "rust_toolchain_target_{}".format(target["triple"]), + build_file = "//pw_toolchain/rust:rust_stdlib.BUILD", + path = "fuchsia/third_party/rust/target/{}".format(target["triple"]), + tag = cipd_tag, + ) + +# buildifier: disable=unnamed-macro +def pw_rust_register_toolchains(): + """Register Rust Toolchains + + For this registration to be valid one must + 1. Call `pw_rust_register_toolchain_and_target_repos(tag)` pervisouly in the + WORKSPACE file. + 2. Call `pw_rust_declare_toolchain_targets()` from + `//pw_toolchain/rust/BUILD.bazel`. + """ + for host in HOSTS: + native.register_toolchains( + "//pw_toolchain/rust:host_rust_toolchain_{}_{}".format(host["os"], host["cpu"]), + "//pw_toolchain/rust:host_rust_analyzer_toolchain_{}_{}".format(host["os"], host["cpu"]), + ) + for target in EXTRA_TARGETS: + native.register_toolchains( + "//pw_toolchain/rust:{}_{}_rust_toolchain_{}_{}".format(host["os"], host["cpu"], target["triple"], target["cpu"]), + ) + +# buildifier: disable=unnamed-macro +def pw_rust_declare_toolchain_targets(): + """Declare rust toolchain targets""" + for host in HOSTS: + _pw_rust_host_toolchain( + name = "host_rust_toolchain_{}_{}".format(host["os"], host["cpu"]), + analyzer_toolchain_name = "host_rust_analyzer_toolchain_{}_{}".format(host["os"], host["cpu"]), + compatible_with = [ + "@platforms//cpu:{}".format(host["cpu"]), + "@platforms//os:{}".format(host["os"]), + ], + dylib_ext = host["dylib_ext"], + target_repo = "@rust_toolchain_target_{}".format(host["triple"]), + toolchain_repo = "@rust_toolchain_host_{}_{}".format(host["os"], host["cpu"]), + triple = host["triple"], + ) + for target in EXTRA_TARGETS: + _pw_rust_toolchain( + name = "{}_{}_rust_toolchain_{}_{}".format(host["os"], host["cpu"], target["triple"], target["cpu"]), + exec_triple = host["triple"], + target_triple = target["triple"], + target_repo = "@rust_toolchain_target_{}".format(target["triple"]), + toolchain_repo = "@rust_toolchain_host_{}_{}".format(host["os"], host["cpu"]), + dylib_ext = "*.so", + exec_compatible_with = [ + "@platforms//cpu:{}".format(host["cpu"]), + "@platforms//os:{}".format(host["os"]), + ], + target_compatible_with = [ + "@platforms//cpu:{}".format(target["cpu"]), + ], + ) + +def _pw_rust_toolchain( + name, + exec_triple, + target_triple, + toolchain_repo, + target_repo, + dylib_ext, + exec_compatible_with, + target_compatible_with): + rust_toolchain( + name = "{}_rust_toolchain".format(name), + binary_ext = "", + default_edition = "2021", + dylib_ext = dylib_ext, + exec_triple = exec_triple, + rust_doc = "{}//:bin/rustdoc".format(toolchain_repo), + rust_std = "{}//:rust_std".format(target_repo), + rustc = "{}//:bin/rustc".format(toolchain_repo), + rustc_lib = "{}//:rustc_lib".format(toolchain_repo), + staticlib_ext = ".a", + stdlib_linkflags = [], + target_triple = target_triple, + ) + native.toolchain( + name = name, + exec_compatible_with = exec_compatible_with, + target_compatible_with = target_compatible_with, + toolchain = ":{}_rust_toolchain".format(name), + toolchain_type = "@rules_rust//rust:toolchain", + ) + +def _pw_rust_host_toolchain( + name, + analyzer_toolchain_name, + triple, + toolchain_repo, + target_repo, + dylib_ext, + compatible_with): + _pw_rust_toolchain( + name = name, + exec_triple = triple, + target_triple = triple, + toolchain_repo = toolchain_repo, + target_repo = target_repo, + dylib_ext = dylib_ext, + exec_compatible_with = compatible_with, + target_compatible_with = compatible_with, + ) + + rust_analyzer_toolchain( + name = "{}_rust_analyzer_toolchain".format(analyzer_toolchain_name), + proc_macro_srv = "{}//:libexec/rust-analyzer-proc-macro-srv".format(toolchain_repo), + rustc = "{}//:bin/rustc".format(toolchain_repo), + rustc_srcs = "{}//:rustc_srcs".format(toolchain_repo), + visibility = ["//visibility:public"], + ) + + native.toolchain( + name = analyzer_toolchain_name, + exec_compatible_with = compatible_with, + target_compatible_with = compatible_with, + toolchain = ":{}_rust_analyzer_toolchain".format(analyzer_toolchain_name), + toolchain_type = "@rules_rust//rust/rust_analyzer:toolchain_type", + ) |