Add miri_test

Author: dzbarsky

rs/experimental/miri/BUILD.bazel

@@ -0,0 +1,43 @@
+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",
+    ],
+)
+
+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/miri_test.bzl

@@ -0,0 +1,148 @@
+"""Miri test rule."""
+
+load("//rs/experimental/miri/private:compile.bzl", "alias_for_dep", "miri_compile_aspect")
+load("//rs/experimental/miri/private:providers.bzl", "MiriCrateInfo")
+load("//rs/experimental/miri/private:toolchain.bzl", "MIRI_TOOLCHAIN_TYPE")
+
+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):
+    return ["--extern", "{}={}".format(extern.name, extern.output.path)]
+
+def _dependency_search_dir_arg(file):
+    return "-Ldependency={}".format(file.dirname)
+
+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)
+    direct_deps = []
+    for dep in ctx.attr.deps:
+        if MiriCrateInfo not in dep:
+            continue
+        miri_dep = dep[MiriCrateInfo]
+        direct_deps.append(struct(
+            name = alias_for_dep(ctx.attr.aliases, dep, miri_dep.crate_info),
+            output = miri_dep.output,
+            transitive_inputs = miri_dep.transitive_inputs,
+            transitive_outputs = miri_dep.transitive_outputs,
+        ))
+
+    test_runner_name = ctx.label.name
+    if ctx.executable._test_runner.extension:
+        test_runner_name += "." + ctx.executable._test_runner.extension
+
+    test_runner = ctx.actions.declare_file(test_runner_name)
+    stdout_file = ctx.actions.declare_file(test_runner.basename + ".stdout", sibling = test_runner)
+    stderr_file = ctx.actions.declare_file(test_runner.basename + ".stderr", sibling = test_runner)
+    exit_code_file = ctx.actions.declare_file(test_runner.basename + ".exit_code", sibling = test_runner)
+    outputs = [stdout_file, stderr_file, exit_code_file]
+
+    miri_args = ctx.actions.args()
+    miri_args.set_param_file_format("multiline")
+    miri_args.use_param_file("@%s", use_always = True)
+    miri_args.add_all([
+        "--sysroot",
+        toolchain.miri_sysroot.path,
+        crate_root.path,
+        "--crate-name",
+        crate_name,
+        "--crate-type",
+        "bin",
+        "--edition",
+        ctx.attr.edition,
+        "--test",
+        "--target",
+        toolchain.target_triple,
+        "--cfg=miri",
+    ])
+
+    miri_args.add_all(direct_deps, map_each = _miri_extern_arg)
+    miri_args.add_all(depset(transitive = [dep.transitive_outputs for dep in direct_deps]), 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)
+
+    process_wrapper_args = ctx.actions.args()
+    process_wrapper_args.add("--stdout-file", stdout_file)
+    process_wrapper_args.add("--stderr-file", stderr_file)
+    process_wrapper_args.add("--captured-exit-code-file", exit_code_file)
+
+    miri_path = ctx.actions.args()
+    miri_path.add("--")
+    miri_path.add(toolchain.miri)
+
+    ctx.actions.run(
+        executable = toolchain.process_wrapper,
+        arguments = [process_wrapper_args, miri_path, miri_args],
+        env = ctx.attr.env | {"MIRI_SYSROOT": toolchain.miri_sysroot.path},
+        inputs = depset(
+            direct = ctx.files.srcs + ctx.files.data + [crate_root],
+            transitive = [toolchain.all_files] + [dep.transitive_inputs for dep in direct_deps],
+        ),
+        outputs = outputs,
+        mnemonic = "MiriTest",
+        progress_message = "Running Miri test %{label}",
+        toolchain = MIRI_TOOLCHAIN_TYPE,
+    )
+
+    ctx.actions.symlink(
+        output = test_runner,
+        target_file = ctx.executable._test_runner,
+        is_executable = True,
+    )
+
+    runfiles = ctx.runfiles(files = outputs)
+    runfiles = runfiles.merge(ctx.attr._test_runner[DefaultInfo].default_runfiles)
+
+    return [
+        DefaultInfo(
+            executable = test_runner,
+            files = depset([test_runner]),
+            runfiles = runfiles,
+        ),
+        OutputGroupInfo(
+            miri_test_exit_code = depset([exit_code_file]),
+            miri_test_outputs = depset(outputs),
+            miri_test_stderr = depset([stderr_file]),
+            miri_test_stdout = depset([stdout_file]),
+        ),
+    ]
+
+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_runner": attr.label(
+            default = Label("@rules_rust//rust/private:captured_output_test_runner"),
+            executable = True,
+            cfg = "exec",
+        ),
+    },
+    test = True,
+    toolchains = [MIRI_TOOLCHAIN_TYPE],
+)