Add ConnOpenInit datagram

Author: majecty

core/src/ibc/connection_03/mod.rs

@@ -24,3 +24,5 @@ pub fn path(id: &str) -> String {
 pub fn client_connections_path(client_id: &str) -> String {
     format!("clients/{}/connections", client_id)
 }
+
+pub use manager::Manager;

core/src/ibc/transaction_handler/datagrams.rs

@@ -21,6 +21,7 @@ use rlp::{Decodable, DecoderError, Encodable, Rlp, RlpStream};
 enum DatagramTag {
     CreateClient = 1,
     UpdateClient = 2,
+    ConnOpenInit = 3,
 }
 
 impl Encodable for DatagramTag {
@@ -35,6 +36,7 @@ impl Decodable for DatagramTag {
         match byte {
             1 => Ok(DatagramTag::CreateClient),
             2 => Ok(DatagramTag::UpdateClient),
+            3 => Ok(DatagramTag::ConnOpenInit),
             _ => Err(DecoderError::Custom("Unexpected DatagramTag Value")),
         }
     }
@@ -51,6 +53,13 @@ pub enum Datagram {
         id: String,
         header: Vec<u8>,
     },
+    ConnOpenInit {
+        identifier: String,
+        desired_counterparty_connection_identifier: String,
+        counterparty_prefix: String,
+        client_identifier: String,
+        counterparty_client_identifier: String,
+    },
 }
 
 impl Encodable for Datagram {
@@ -69,6 +78,21 @@ impl Encodable for Datagram {
             } => {
                 s.begin_list(3).append(&DatagramTag::UpdateClient).append(id).append(header);
             }
+            Datagram::ConnOpenInit {
+                identifier,
+                desired_counterparty_connection_identifier,
+                counterparty_prefix,
+                client_identifier,
+                counterparty_client_identifier,
+            } => {
+                s.begin_list(6);
+                s.append(&DatagramTag::ConnOpenInit)
+                    .append(identifier)
+                    .append(desired_counterparty_connection_identifier)
+                    .append(counterparty_prefix)
+                    .append(client_identifier)
+                    .append(counterparty_client_identifier);
+            }
         };
     }
 }
@@ -104,6 +128,22 @@ impl Decodable for Datagram {
                     header: rlp.val_at(2)?,
                 })
             }
+            DatagramTag::ConnOpenInit => {
+                let item_count = rlp.item_count()?;
+                if item_count != 6 {
+                    return Err(DecoderError::RlpInvalidLength {
+                        expected: 6,
+                        got: item_count,
+                    })
+                }
+                Ok(Datagram::ConnOpenInit {
+                    identifier: rlp.val_at(1)?,
+                    desired_counterparty_connection_identifier: rlp.val_at(2)?,
+                    counterparty_prefix: rlp.val_at(3)?,
+                    client_identifier: rlp.val_at(4)?,
+                    counterparty_client_identifier: rlp.val_at(5)?,
+                })
+            }
         }
     }
 }

core/src/ibc/transaction_handler/mod.rs

@@ -23,6 +23,7 @@ use cstate::{StateResult, TopLevelState};
 use ctypes::errors::RuntimeError;
 use ibc::client_02 as ibc_client;
 use ibc::client_02::foundry as ibc_foundry;
+use ibc::connection_03 as ibc_connection;
 use ibc::context as ibc_context;
 use rlp::{Decodable, Rlp};
 
@@ -43,6 +44,26 @@ pub fn execute(
             id,
             header,
         } => update_client(state, &id, &header),
+        Datagram::ConnOpenInit {
+            identifier,
+            desired_counterparty_connection_identifier,
+            counterparty_prefix,
+            client_identifier,
+            counterparty_client_identifier,
+        } => {
+            let mut context = ibc_context::TopLevelContext::new(state);
+            let connection_manager = ibc_connection::Manager::new();
+            connection_manager
+                .handle_open_init(
+                    &mut context,
+                    identifier,
+                    desired_counterparty_connection_identifier,
+                    counterparty_prefix,
+                    client_identifier,
+                    counterparty_client_identifier,
+                )
+                .map_err(|err| RuntimeError::IBC(format!("ConnOpenInit: {}", err)).into())
+        }
     }
 }