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,162 @@
+"""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_deps(deps, aliases):
+    externs = []
+    dependency_inputs = []
+    dependency_outputs = []
+    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(struct(
+            name = name,
+            output = miri_dep.output,
+        ))
+        dependency_inputs.append(miri_dep.transitive_inputs)
+        dependency_outputs.append(miri_dep.transitive_outputs)
+
+    return struct(
+        externs = externs,
+        inputs = depset(transitive = dependency_inputs),
+        outputs = depset(transitive = dependency_outputs),
+    )
+
+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_deps(ctx.attr.deps, ctx.attr.aliases)
+
+    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 + ".miri_test.stdout", sibling = test_runner)
+    stderr_file = ctx.actions.declare_file(test_runner.basename + ".miri_test.stderr", sibling = test_runner)
+    exit_code_file = ctx.actions.declare_file(test_runner.basename + ".miri_test.exit_code", sibling = test_runner)
+
+    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(deps.externs, map_each = _miri_extern_arg)
+    miri_args.add_all(deps.outputs, 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)
+    process_wrapper_args.add("--subst", "pwd=${pwd}")
+    process_wrapper_args.add("--subst", "exec_root=${exec_root}")
+    process_wrapper_args.add("--subst", "output_base=${output_base}")
+
+    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 = [
+                deps.inputs,
+                deps.outputs,
+                toolchain.all_files,
+            ],
+        ),
+        outputs = [
+            stdout_file,
+            stderr_file,
+            exit_code_file,
+        ],
+        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 = [stdout_file, stderr_file, exit_code_file])
+    runfiles = runfiles.merge(ctx.attr._test_runner[DefaultInfo].default_runfiles)
+
+    return [DefaultInfo(
+        executable = test_runner,
+        files = depset([test_runner]),
+        runfiles = runfiles,
+    )]
+
+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("//rs/experimental/miri/private:miri_test_runner"),
+            executable = True,
+            cfg = "exec",
+        ),
+    },
+    test = True,
+    toolchains = [MIRI_TOOLCHAIN_TYPE],
+)