Add miri_test

Author: dzbarsky

MODULE.bazel

@@ -52,6 +52,7 @@ use_repo(
 bazel_dep(name = "bazel_lib", version = "3.0.0")
 bazel_dep(name = "bazel_features", version = "1.45.0")
 bazel_dep(name = "bazel_skylib", version = "1.4.1")
+bazel_dep(name = "hermetic_launcher", version = "0.0.5")
 bazel_dep(name = "package_metadata", version = "0.0.7")
 bazel_dep(name = "platforms", version = "1.1.0")
 bazel_dep(name = "protobuf", version = "34.0.bcr.1", repo_name = "com_google_protobuf")

MODULE.bazel.lock

@@ -0,0 +1,44 @@
+load("@bazel_lib//:bzl_library.bzl", "bzl_library")
+
+package(default_visibility = ["//visibility:private"])
+
+toolchain_type(
+    name = "toolchain_type",
+    visibility = ["//visibility:public"],
+)
+
+toolchain_type(
+    name = "sysroot_toolchain_type",
+    visibility = ["//visibility:public"],
+)
+
+bzl_library(
+    name = "miri_test",
+    srcs = ["miri_test.bzl"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//rs/experimental/miri/private:compile",
+        "//rs/experimental/miri/private:providers",
+        "//rs/experimental/miri/private:toolchain",
+        "@bazel_features//:features",
+        "@hermetic_launcher//launcher:lib_bzl",
+    ],
+)
+
+bzl_library(
+    name = "module_extension",
+    srcs = ["module_extension.bzl"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//rs/platforms:triples",
+        "//rs/private:cargo_repository",
+        "//rs/private:miri_repository",
+        "//rs/private:module_extension_utils",
+        "//rs/private:rustc_repository",
+        "//rs/private:rustc_src_repository",
+        "//rs/private:stdlib_repository",
+        "//rs/toolchains:toolchain_utils",
+        "@rules_rust//rust/platform:bzl_lib",
+        "@rules_rust//rust/private:bzl_lib",
+    ],
+)

rs/experimental/miri/BUILD.bazel

@@ -0,0 +1,138 @@
+"""Miri test rule."""
+
+load("@hermetic_launcher//launcher:lib.bzl", "launcher")
+load("//rs/experimental/miri/private:compile.bzl", "alias_for_dep", "dependency_search_dir_arg", "miri_compile_aspect")
+load("//rs/experimental/miri/private:providers.bzl", "MiriCrateInfo")
+load("//rs/experimental/miri/private:toolchain.bzl", "MIRI_TOOLCHAIN_TYPE")
+
+def _dirname(path):
+    if "/" not in path:
+        return "."
+    return path.rsplit("/", 1)[0]
+
+def _default_crate_name(label):
+    return label.name.replace("-", "_")
+
+def _crate_root(ctx):
+    if ctx.file.crate_root:
+        return ctx.file.crate_root
+
+    for src in ctx.files.srcs:
+        if src.basename in ("main.rs", "lib.rs"):
+            return src
+
+    if len(ctx.files.srcs) == 1:
+        return ctx.files.srcs[0]
+
+    fail("miri_test requires crate_root when srcs does not contain main.rs or lib.rs")
+
+def _miri_extern_arg(extern):
+    name, output = extern
+    return ["--extern", "{}={}".format(name, output.short_path)]
+
+def _miri_extern_args(deps, aliases):
+    externs = []
+    dependency_search_dirs = set()
+    dependency_inputs = []
+    for dep in deps:
+        if MiriCrateInfo not in dep:
+            continue
+        miri_dep = dep[MiriCrateInfo]
+        name = alias_for_dep(aliases, dep, miri_dep.crate_info)
+        externs.append((name, miri_dep.output))
+        dependency_search_dirs.add(_dirname(miri_dep.output.short_path))
+        dependency_inputs.append(miri_dep.transitive_inputs)
+        for transitive_output in miri_dep.transitive_outputs.to_list():
+            dependency_search_dirs.add(_dirname(transitive_output.short_path))
+
+    return struct(
+        dependency_inputs = dependency_inputs,
+        dependency_search_dirs = sorted(dependency_search_dirs),
+        externs = externs,
+    )
+
+def _miri_test_impl(ctx):
+    toolchain = ctx.toolchains[MIRI_TOOLCHAIN_TYPE]
+    crate_root = _crate_root(ctx)
+    crate_name = ctx.attr.crate_name or _default_crate_name(ctx.label)
+    deps = _miri_extern_args(ctx.attr.deps, ctx.attr.aliases)
+
+    miri_args = ctx.actions.args()
+    miri_args.set_param_file_format("multiline")
+    miri_args.add_all([
+        "--sysroot",
+        toolchain.miri_sysroot.short_path,
+        crate_root.short_path,
+        "--crate-name",
+        crate_name,
+        "--crate-type",
+        "bin",
+        "--edition",
+        ctx.attr.edition,
+        "--test",
+        "--target",
+        toolchain.target_triple,
+        "--cfg=miri",
+    ])
+
+    miri_args.add_all(deps.externs, map_each = _miri_extern_arg)
+    miri_args.add_all(deps.dependency_search_dirs, map_each = dependency_search_dir_arg)
+    miri_args.add_all(ctx.attr.rustc_flags)
+    miri_args.add_all(ctx.attr.miri_flags)
+    miri_args.add("--")
+    miri_args.add_all(ctx.attr.args)
+
+    args_file = ctx.actions.declare_file(ctx.label.name + ".miri_args")
+    ctx.actions.write(
+        output = args_file,
+        content = miri_args,
+    )
+
+    executable = ctx.actions.declare_file(ctx.label.name)
+    embedded_args, transformed_args = launcher.args_from_entrypoint(toolchain.miri)
+    embedded_args = embedded_args + ["@" + args_file.short_path]
+    launcher.compile_stub(
+        ctx = ctx,
+        embedded_args = embedded_args,
+        transformed_args = transformed_args,
+        output_file = executable,
+    )
+
+    runfiles = ctx.runfiles(
+        files = ctx.files.srcs + ctx.files.data + [args_file, crate_root],
+        transitive_files = depset(transitive = [
+            toolchain.all_files,
+            depset([output for _, output in deps.externs]),
+            depset(transitive = deps.dependency_inputs),
+        ]),
+    )
+
+    return [
+        DefaultInfo(
+            executable = executable,
+            runfiles = runfiles,
+        ),
+        RunEnvironmentInfo(environment = ctx.attr.env | {"MIRI_SYSROOT": toolchain.miri_sysroot.short_path}),
+    ]
+
+miri_test = rule(
+    implementation = _miri_test_impl,
+    attrs = {
+        "crate_name": attr.string(doc = "Rust crate name. Defaults to the target name with '-' replaced by '_'."),
+        "crate_root": attr.label(allow_single_file = [".rs"]),
+        "aliases": attr.label_keyed_string_dict(doc = "Remap direct dependencies to another extern crate name."),
+        "data": attr.label_list(allow_files = True),
+        "deps": attr.label_list(aspects = [miri_compile_aspect]),
+        "edition": attr.string(default = "2021"),
+        "env": attr.string_dict(doc = "Environment variables set while running Miri."),
+        "miri_flags": attr.string_list(doc = "Extra flags passed to the Miri driver."),
+        "rustc_flags": attr.string_list(doc = "Extra rustc-compatible flags passed to Miri."),
+        "srcs": attr.label_list(allow_files = [".rs"], mandatory = True),
+    },
+    test = True,
+    toolchains = [
+        MIRI_TOOLCHAIN_TYPE,
+        launcher.finalizer_toolchain_type,
+        launcher.template_toolchain_type,
+    ],
+)