Implement handle_open_init function

Author: majecty

core/src/ibc/connection_03/manager.rs

@@ -0,0 +1,77 @@
+// Copyright 2020 Kodebox, Inc.
+// This file is part of CodeChain.
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as
+// published by the Free Software Foundation, either version 3 of the
+// License, or (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+use super::path as connection_path;
+use super::types::{CommitmentPrefix, Identifier};
+use crate::ibc;
+use crate::ibc::connection_03::client_connections_path;
+use crate::ibc::connection_03::types::{ConnectionEnd, ConnectionIdentifiersInClient, ConnectionState};
+use rlp::{Encodable, Rlp};
+
+#[derive(Default)]
+pub struct Manager {}
+
+impl Manager {
+    pub fn new() -> Self {
+        Manager {}
+    }
+
+    pub fn handle_open_init(
+        &self,
+        ctx: &mut dyn ibc::Context,
+        identifier: Identifier,
+        desired_counterparty_connection_identifier: Identifier,
+        counterparty_prefix: CommitmentPrefix,
+        client_identifier: Identifier,
+        counterparty_client_identifier: Identifier,
+    ) -> Result<(), String> {
+        let kv_store = ctx.get_kv_store();
+        if kv_store.has(&connection_path(&identifier)) {
+            return Err("Connection exist".to_owned())
+        }
+        let state = ConnectionState::INIT;
+        let connection = ConnectionEnd {
+            state,
+            counterparty_connection_identifier: desired_counterparty_connection_identifier,
+            counterparty_prefix,
+            client_identifier: client_identifier.clone(),
+            counterparty_client_identifier,
+        };
+        kv_store.set(&connection_path(&identifier), &connection.rlp_bytes());
+        self.add_connection_to_client(ctx, client_identifier, identifier)?;
+        Ok(())
+    }
+
+    fn add_connection_to_client(
+        &self,
+        ctx: &mut dyn ibc::Context,
+        client_identifier: Identifier,
+        connection_identifier: Identifier,
+    ) -> Result<(), String> {
+        let kv_store = ctx.get_kv_store();
+        if kv_store.has(&connection_path(&connection_identifier)) {
+            return Err("Connection exist".to_owned())
+        }
+        let bytes = kv_store.get(&client_connections_path(&client_identifier));
+        let rlp = Rlp::new(&bytes);
+        let mut conns: ConnectionIdentifiersInClient = rlp.as_val().expect("data from DB");
+
+        conns.add(connection_identifier);
+
+        kv_store.set(&client_connections_path(&client_identifier), &rlp::encode(&conns));
+        Ok(())
+    }
+}

core/src/ibc/connection_03/mod.rs

@@ -0,0 +1,26 @@
+// Copyright 2020 Kodebox, Inc.
+// This file is part of CodeChain.
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as
+// published by the Free Software Foundation, either version 3 of the
+// License, or (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+mod manager;
+mod types;
+
+pub fn path(id: &str) -> String {
+    format!("connections/{}", id)
+}
+
+pub fn client_connections_path(client_id: &str) -> String {
+    format!("clients/{}/connections", client_id)
+}

core/src/ibc/connection_03/types.rs

@@ -0,0 +1,76 @@
+// Copyright 2020 Kodebox, Inc.
+// This file is part of CodeChain.
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as
+// published by the Free Software Foundation, either version 3 of the
+// License, or (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+use rlp;
+use rlp::{DecoderError, Rlp, RlpStream};
+
+#[repr(u8)]
+#[derive(Copy, Clone)]
+pub enum ConnectionState {
+    INIT = 0,
+    TRYOPEN = 1,
+    OPEN = 2,
+}
+
+impl rlp::Encodable for ConnectionState {
+    fn rlp_append(&self, s: &mut RlpStream) {
+        s.append(&(*self as u8));
+    }
+}
+
+// FIXME: current commitment_23::Prefix is too generic.
+pub type CommitmentPrefix = String;
+pub type Identifier = String;
+
+pub struct ConnectionEnd {
+    pub state: ConnectionState,
+    pub counterparty_connection_identifier: Identifier,
+    pub counterparty_prefix: CommitmentPrefix,
+    pub client_identifier: Identifier,
+    pub counterparty_client_identifier: Identifier,
+    // FIXME: implement version
+}
+
+impl rlp::Encodable for ConnectionEnd {
+    fn rlp_append(&self, s: &mut rlp::RlpStream) {
+        s.begin_list(5);
+        s.append(&self.state);
+        s.append(&self.counterparty_connection_identifier);
+        s.append(&self.counterparty_prefix);
+        s.append(&self.client_identifier);
+        s.append(&self.counterparty_client_identifier);
+    }
+}
+
+pub struct ConnectionIdentifiersInClient(Vec<Identifier>);
+
+impl ConnectionIdentifiersInClient {
+    pub fn add(&mut self, identifier: Identifier) {
+        self.0.push(identifier);
+    }
+}
+
+impl rlp::Encodable for ConnectionIdentifiersInClient {
+    fn rlp_append(&self, s: &mut RlpStream) {
+        s.append_list::<String, String>(&self.0);
+    }
+}
+
+impl rlp::Decodable for ConnectionIdentifiersInClient {
+    fn decode<'a>(rlp: &Rlp<'a>) -> Result<Self, DecoderError> {
+        Ok(Self(rlp.as_list()?))
+    }
+}

core/src/ibc/mod.rs

@@ -18,6 +18,9 @@ pub mod client_02;
 #[allow(dead_code)]
 #[allow(unused_variables)]
 mod commitment_23;
+#[allow(dead_code)]
+#[allow(unused_variables)]
+mod connection_03;
 pub mod context;
 mod kv_store;
 mod transaction_handler;