Merge pull request #70 from kinode-dao/develop

Develop

Author: dr-frmr

Cargo.lock

@@ -1,7 +1,7 @@
 [package]
 name = "kinode_process_lib"
 description = "A library for writing Kinode processes in Rust."
-version = "0.6.0"
+version = "0.6.1"
 edition = "2021"
 license-file = "LICENSE"
 homepage = "https://kinode.org"
@@ -15,10 +15,11 @@ alloy-json-rpc = { git = "https://github.com/alloy-rs/alloy.git", rev = "6f8ebb4
 anyhow = "1.0"
 bincode = "1.3.3"
 http = "1.0.0"
+mime_guess = "2.0"
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
 rand = "0.8"
+rmp-serde = "1.1.2"
 thiserror = "1.0"
 url = "2.4.1"
-mime_guess = "2.0"
-wit-bindgen = { git = "https://github.com/bytecodealliance/wit-bindgen", rev = "21a46c7" }
+wit-bindgen = "0.24.0"

Cargo.toml

@@ -33,6 +33,9 @@ pub mod http;
 pub mod kernel_types;
 /// Interact with the key_value module
 pub mod kv;
+/// Interact with the networking module
+/// For configuration, debugging, and creating signatures with networking key.
+pub mod net;
 /// Interact with the sqlite module
 pub mod sqlite;
 /// Interact with the timer runtime module.
@@ -74,6 +77,7 @@ macro_rules! call_init {
                 $init_func(our);
             }
         }
+        export!(Component);
     };
 }
 

src/lib.rs

@@ -0,0 +1,124 @@
+use crate::*;
+
+//
+// Networking protocol types
+//
+
+#[derive(Clone, Debug, Serialize, Deserialize)]
+pub struct Identity {
+    pub name: NodeId,
+    pub networking_key: String,
+    pub ws_routing: Option<(String, u16)>,
+    pub allowed_routers: Vec<NodeId>,
+}
+
+/// Must be parsed from message pack vector.
+/// all Get actions must be sent from local process. used for debugging
+#[derive(Clone, Debug, Serialize, Deserialize)]
+pub enum NetAction {
+    /// Received from a router of ours when they have a new pending passthrough for us.
+    /// We should respond (if we desire) by using them to initialize a routed connection
+    /// with the NodeId given.
+    ConnectionRequest(NodeId),
+    /// can only receive from trusted source, for now just ourselves locally,
+    /// in the future could get from remote provider
+    KnsUpdate(KnsUpdate),
+    KnsBatchUpdate(Vec<KnsUpdate>),
+    /// get a list of peers we are connected to
+    GetPeers,
+    /// get the [`Identity`] struct for a single peer
+    GetPeer(String),
+    /// get the [`NodeId`] associated with a given namehash, if any
+    GetName(String),
+    /// get a user-readable diagnostics string containing networking inforamtion
+    GetDiagnostics,
+    /// sign the attached blob payload, sign with our node's networking key.
+    /// **only accepted from our own node**
+    /// **the source [`Address`] will always be prepended to the payload**
+    Sign,
+    /// given a message in blob payload, verify the message is signed by
+    /// the given source. if the signer is not in our representation of
+    /// the PKI, will not verify.
+    /// **the `from` [`Address`] will always be prepended to the payload**
+    Verify {
+        from: Address,
+        signature: Vec<u8>,
+    },
+}
+
+/// For now, only sent in response to a ConnectionRequest.
+/// Must be parsed from message pack vector
+#[derive(Clone, Debug, Serialize, Deserialize)]
+pub enum NetResponse {
+    Accepted(NodeId),
+    Rejected(NodeId),
+    /// response to [`NetAction::GetPeers`]
+    Peers(Vec<Identity>),
+    /// response to [`NetAction::GetPeer`]
+    Peer(Option<Identity>),
+    /// response to [`NetAction::GetName`]
+    Name(Option<String>),
+    /// response to [`NetAction::GetDiagnostics`]. a user-readable string.
+    Diagnostics(String),
+    /// response to [`NetAction::Sign`]. contains the signature in blob
+    Signed,
+    /// response to [`NetAction::Verify`]. boolean indicates whether
+    /// the signature was valid or not. note that if the signer node
+    /// cannot be found in our representation of PKI, this will return false,
+    /// because we cannot find the networking public key to verify with.
+    Verified(bool),
+}
+
+#[derive(Clone, Debug, Serialize, Deserialize)]
+pub struct KnsUpdate {
+    pub name: String, // actual username / domain name
+    pub owner: String,
+    pub node: String, // hex namehash of node
+    pub public_key: String,
+    pub ip: String,
+    pub port: u16,
+    pub routers: Vec<String>,
+}
+
+//
+// Helpers
+//
+
+pub fn sign<T>(message: T) -> Result<Vec<u8>, SendError>
+where
+    T: Into<Vec<u8>>,
+{
+    Request::to(("our", "net", "distro", "sys"))
+        .body(rmp_serde::to_vec(&NetAction::Sign).unwrap())
+        .blob_bytes(message.into())
+        .send_and_await_response(30)
+        .unwrap()
+        .map(|_resp| get_blob().unwrap().bytes)
+}
+
+pub fn verify<T, U, V>(from: T, message: U, signature: V) -> Result<bool, SendError>
+where
+    T: Into<Address>,
+    U: Into<Vec<u8>>,
+    V: Into<Vec<u8>>,
+{
+    Request::to(("our", "net", "distro", "sys"))
+        .body(
+            rmp_serde::to_vec(&NetAction::Verify {
+                from: from.into(),
+                signature: signature.into(),
+            })
+            .unwrap(),
+        )
+        .blob_bytes(message.into())
+        .send_and_await_response(30)
+        .unwrap()
+        .map(|resp| {
+            let Ok(NetResponse::Verified(valid)) =
+                rmp_serde::from_slice::<NetResponse>(resp.body())
+            else {
+                return false;
+            };
+            valid
+        })
+}