Support for multiple PCI roots / IOMMUs

BUILD

@@ -10,10 +10,35 @@
 # bazel test //:rustfmt-all
 # bazel test //:test-all
 
+load("@bazel_skylib//rules:common_settings.bzl", "bool_flag")
 load("@rules_rust//rust:defs.bzl", "rust_binary", "rust_clippy", "rust_doc", "rust_test", "rustfmt_test")
 load("//:objcopy.bzl", "objcopy_to_object")
 load("//:lds.bzl", "lds_rule")
 
+bool_flag(
+    name = "enable_unsafe_enhanced_allocation",
+    build_setting_default = 0,
+)
+
+config_setting(
+    name = "unsafe_enhanced_allocation",
+    flag_values = {
+      "enable_unsafe_enhanced_allocation": "true",
+    },
+)
+
+bool_flag(
+    name = "enable_hardware_ad_updates",
+    build_setting_default = 1,
+)
+
+config_setting(
+    name = "hardware_ad_updates",
+    flag_values = {
+      "enable_hardware_ad_updates": "true",
+    },
+)
+
 filegroup(
     name = "salus-all",
     srcs = [
@@ -191,6 +216,10 @@ rust_binary(
         "-Clink-arg=-T$(location //:l_rule)",
     ],
     deps = salus_deps,
+    crate_features = select({
+      ":hardware_ad_updates": ["hardware_ad_updates"],
+      "//conditions:default": [],
+    }),
 )
 
 rust_clippy(

device-tree/src/device_tree.rs

@@ -46,6 +46,8 @@ pub struct DeviceTreeNode {
 }
 
 pub type NodeArena = Arena<DeviceTreeNode, Global>;
+
+/// An ID for a device tree node. The node itself can be obtained via `DeviceTree::get_node`.
 pub type NodeId = ArenaId<DeviceTreeNode>;
 
 /// A tree representation of the hardware in a system based on v0.3 of the Devicetree Specification.

device-tree/src/lib.rs

@@ -13,7 +13,7 @@ mod error;
 mod fdt;
 mod serialize;
 
-pub use crate::device_tree::{DeviceTree, DeviceTreeIter, DeviceTreeNode};
+pub use crate::device_tree::{DeviceTree, DeviceTreeIter, DeviceTreeNode, NodeId};
 pub use error::Error as DeviceTreeError;
 pub use error::Result as DeviceTreeResult;
 pub use fdt::{Cpu, Fdt, FdtMemoryRegion, ImsicInfo};

drivers/BUILD

@@ -27,6 +27,13 @@ rust_library(
         "@salus-index//:static_assertions",
         "@salus-index//:tock-registers",
     ],
+    crate_features = select({
+      "//:unsafe_enhanced_allocation": ["unsafe_enhanced_allocation"],
+      "//conditions:default": [],
+    }) + select({
+      "//:hardware_ad_updates": ["hardware_ad_updates"],
+      "//conditions:default": [],
+    }),
 )
 
 rust_clippy(