Support IB equivocations (#428)

* sim-rs: add ib-equivocations to schema

* sim-rs: support IB equivocation

* sim-rs: equivocation protection during propagation

* sim-rs: output VRFs for IBs

* sim-rs: transmit proofs of equivocation

* sim-rs: support "adversarial" property for compat with old Haskell config

* sim-rs: newline at end of config file

* sim-rs: only vote on IBs which arrived quickly enough

* sim-rs: account for IB generation time in equivocation guard

* sim-rs: never vote for equivocated IBs

Author: SupernaviX

data/simulation/config.d.ts

@@ -49,6 +49,12 @@ export interface Config {
    * This is Δhdr from the Leios paper.
    * */
   "leios-header-diffusion-time-ms": number;
+  /**
+   * The expected time it takes a node to generate an IB.
+   * This is used as part of a validation rule to protect against equivocation attacks.
+   * In practice, it should probably always equal ib-generation-cpu-time-ms.
+   */
+  "leios-ib-generation-time-ms": number;
   /**
    * The strategy to use when selecting TXs from the Leios mempool.
    */

data/simulation/config.default.yaml

@@ -29,6 +29,7 @@ leios-stage-active-voting-slots: 1
 leios-vote-send-recv-stages: false
 leios-late-ib-inclusion: true
 leios-header-diffusion-time-ms: 1000.0
+leios-ib-generation-time-ms: 130.0
 leios-mempool-sampling-strategy: ordered-by-id
 leios-mempool-aggressive-pruning: false
 # TODO: revise default

data/simulation/config.schema.json

@@ -269,8 +269,12 @@
       "description": "The expected time it takes a header to fully diffuse across the network.\nThis is Δhdr from the Leios paper.",
       "type": "number"
     },
+    "leios-ib-generation-time-ms": {
+      "description": "The expected time it takes a node to generate an IB.\nThis is used as part of a validation rule to protect against equivocation attacks.\nIn practice, it should probably always equal ib-generation-cpu-time-ms.",
+      "type": "number"
+    },
     "leios-late-ib-inclusion": {
-      "description": "Extends Leios so that EB producers include IBs directly from previous pipelines\nwhere no certified EB was observed.\n\nOnly supported by Rust simulation.",
+      "description": "Extends Leios so that EB producers include IBs directly from previous pipelines.\nDue to casuality, the EB must always include them, even if those IBs end up being\ncertified in their own pipeline.",
       "type": "boolean"
     },
     "leios-mempool-aggressive-pruning": {
@@ -375,7 +379,7 @@
     },
     "tx-overcollateralization-factor-distribution": {
       "$ref": "#/definitions/Distribution",
-      "description": "Distribution used to choose the \"over-collateralization factor\" for a transaction.\n0 means the transaction is not over-collateralized, n means it has enough extra collateral to be included in n shards.\nOnly supported by Rust simulation."
+      "description": "Distribution used to choose the \"over-collateralization factor\" for a transaction.\nAn \"overcollateralization factor\" of n means the TX has enough collateral to be included in n+1 shards.\nA factor of 0 means the TX has enough collateral to be include in 1 shard; i.e. it is not over-collateralized.\nOnly supported by Rust simulation."
     },
     "tx-size-bytes-distribution": {
       "$ref": "#/definitions/Distribution",

data/simulation/topology.d.ts

@@ -33,11 +33,14 @@ export interface Node<Location> {
    * Only supported by Rust simulation.
    */
   "tx-generation-weight"?: number | null;
-  /** If not null, the node will behave according to the given Behaviour.
-  *
-  * Only supported by Haskell simulation.
-  */
+  /** If not null, the node will behave according to the given Behaviour. */
   adversarial?: Behaviour | null;
+  /** If not null, the node will follow the given behaviours.
+   * 
+   * 
+   * Only supported by Rust simulation.
+   */
+  behaviours?: Behaviour[] | null;
 }
 
 /** Link information. */
@@ -70,3 +73,11 @@ export interface UnboundedIbs {
   "slot-of-generated-ibs": number;
   "ibs-per-slot": number;
 }
+
+/** A node which generates equivocated IBs at any possible opportunity.
+ * 
+ * Only supported by Rust simulation.
+ */
+export interface IbEquivocation {
+  behaviour: "ib-equivocation";
+}

data/simulation/topology.schema.json

@@ -28,7 +28,14 @@
       "properties": {
         "adversarial": {
           "$ref": "#/definitions/UnboundedIbs",
-          "description": "If not null, the node will behave according to the given Behaviour.\n\nOnly supported by Haskell simulation."
+          "description": "If not null, the node will behave according to the given Behaviour."
+        },
+        "behaviours": {
+          "description": "If not null, the node will follow the given behaviours.\n\n\nOnly supported by Rust simulation.",
+          "items": {
+            "$ref": "#/definitions/UnboundedIbs"
+          },
+          "type": "array"
         },
         "cpu-core-count": {
           "additionalProperties": false,
@@ -65,7 +72,14 @@
       "properties": {
         "adversarial": {
           "$ref": "#/definitions/UnboundedIbs",
-          "description": "If not null, the node will behave according to the given Behaviour.\n\nOnly supported by Haskell simulation."
+          "description": "If not null, the node will behave according to the given Behaviour."
+        },
+        "behaviours": {
+          "description": "If not null, the node will follow the given behaviours.\n\n\nOnly supported by Rust simulation.",
+          "items": {
+            "$ref": "#/definitions/UnboundedIbs"
+          },
+          "type": "array"
         },
         "cpu-core-count": {
           "additionalProperties": false,

data/simulation/trace.haskell.schema.json

@@ -99,6 +99,12 @@
     },
     "GeneratedEndorserBlock": {
       "properties": {
+        "endorser_blocks": {
+          "items": {
+            "$ref": "#/definitions/BlockRef"
+          },
+          "type": "array"
+        },
         "id": {
           "type": "string"
         },
@@ -125,6 +131,7 @@
         }
       },
       "required": [
+        "endorser_blocks",
         "id",
         "input_blocks",
         "pipeline",

data/simulation/trace.rust.schema.json

@@ -94,6 +94,12 @@
     },
     "GeneratedEndorserBlock": {
       "properties": {
+        "endorser_blocks": {
+          "items": {
+            "$ref": "#/definitions/BlockRef"
+          },
+          "type": "array"
+        },
         "id": {
           "type": "string"
         },
@@ -120,6 +126,7 @@
         }
       },
       "required": [
+        "endorser_blocks",
         "id",
         "input_blocks",
         "pipeline",

sim-rs/sim-cli/src/bin/gen-test-data/strategy/utils.rs

@@ -173,6 +173,8 @@ impl GraphBuilder {
                     tx_conflict_fraction: None,
                     tx_generation_weight: None,
                     producers: BTreeMap::new(),
+                    adversarial: None,
+                    behaviours: vec![],
                 };
                 (name, node)
             })

sim-rs/sim-core/src/config.rs

@@ -67,6 +67,7 @@ pub struct RawParameters {
     pub leios_stage_active_voting_slots: u64,
     pub leios_late_ib_inclusion: bool,
     pub leios_header_diffusion_time_ms: f64,
+    pub leios_ib_generation_time_ms: f64,
     pub leios_mempool_sampling_strategy: MempoolSamplingStrategy,
     pub leios_mempool_aggressive_pruning: bool,
     pub praos_chain_quality: u64,
@@ -185,6 +186,16 @@ pub struct RawNode {
     #[serde(skip_serializing_if = "Option::is_none")]
     pub tx_generation_weight: Option<u64>,
     pub producers: BTreeMap<String, RawLinkInfo>,
+    #[serde(default, skip_serializing_if = "Option::is_none")]
+    pub adversarial: Option<RawNodeBehaviour>,
+    #[serde(default, skip_serializing_if = "Vec::is_empty")]
+    pub behaviours: Vec<RawNodeBehaviour>,
+}
+
+#[derive(Debug, Serialize, Deserialize)]
+#[serde(rename_all = "kebab-case", tag = "behaviour")]
+pub enum RawNodeBehaviour {
+    IbEquivocation,
 }
 
 #[derive(Debug, Serialize, Deserialize)]
@@ -253,6 +264,7 @@ impl From<RawTopology> for Topology {
         for (index, (name, node)) in value.nodes.iter().enumerate() {
             let id = NodeId::new(index);
             node_ids.insert(name.clone(), id);
+            let behaviours = NodeBehaviours::parse(&node.adversarial, &node.behaviours);
             nodes.insert(
                 id,
                 NodeConfiguration {
@@ -264,6 +276,7 @@ impl From<RawTopology> for Topology {
                     tx_conflict_fraction: node.tx_conflict_fraction,
                     tx_generation_weight: node.tx_generation_weight,
                     consumers: vec![],
+                    behaviours,
                 },
             );
         }
@@ -483,6 +496,7 @@ pub struct SimConfiguration {
     pub vote_threshold: u64,
     pub(crate) praos_fallback: bool,
     pub(crate) header_diffusion_time: Duration,
+    pub(crate) ib_generation_time: Duration,
     pub(crate) relay_strategy: RelayStrategy,
     pub(crate) mempool_strategy: MempoolSamplingStrategy,
     pub(crate) mempool_aggressive_pruning: bool,
@@ -540,6 +554,7 @@ impl SimConfiguration {
             variant: params.leios_variant,
             praos_fallback: params.praos_fallback_enabled,
             header_diffusion_time: duration_ms(params.leios_header_diffusion_time_ms),
+            ib_generation_time: duration_ms(params.leios_ib_generation_time_ms),
             relay_strategy: params.relay_strategy,
             mempool_strategy: params.leios_mempool_sampling_strategy,
             mempool_aggressive_pruning: params.leios_mempool_aggressive_pruning,
@@ -579,6 +594,7 @@ pub struct NodeConfiguration {
     pub tx_conflict_fraction: Option<f64>,
     pub tx_generation_weight: Option<u64>,
     pub consumers: Vec<NodeId>,
+    pub behaviours: NodeBehaviours,
 }
 
 #[derive(Debug, Clone)]
@@ -587,3 +603,22 @@ pub struct LinkConfiguration {
     pub latency: Duration,
     pub bandwidth_bps: Option<u64>,
 }
+
+#[derive(Debug, Clone, Default)]
+pub struct NodeBehaviours {
+    pub ib_equivocation: bool,
+}
+
+impl NodeBehaviours {
+    fn parse(adversarial: &Option<RawNodeBehaviour>, behaviours: &[RawNodeBehaviour]) -> Self {
+        let mut result = NodeBehaviours::default();
+        for behaviour in adversarial.iter().chain(behaviours) {
+            match behaviour {
+                RawNodeBehaviour::IbEquivocation => {
+                    result.ib_equivocation = true;
+                }
+            }
+        }
+        result
+    }
+}

sim-rs/sim-core/src/events.rs

@@ -165,6 +165,7 @@ pub enum Event {
         pipeline: u64,
         producer: Node,
         index: u64,
+        vrf: u64,
         shard: u64,
         header_bytes: u64,
         tx_payload_bytes: u64,
@@ -477,6 +478,7 @@ impl EventTracker {
             pipeline: block.header.id.pipeline,
             producer: self.to_node(block.header.id.producer),
             index: block.header.id.index,
+            vrf: block.header.vrf,
             shard: block.header.shard,
             header_bytes,
             tx_payload_bytes,