evm: Add net_peerCount RPC (#2569)

* Add net_peerCount RPC * Add new line to end of file
DeFiCh · Oct 11, 2023 · aeaa1cd · aeaa1cd
1 parent 1815ada
commit aeaa1cd
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 0 deletions.
diff --git a/lib/ain-cpp-imports/src/bridge.rs b/lib/ain-cpp-imports/src/bridge.rs
@@ -49,5 +49,6 @@ pub mod ffi {
         fn getClientVersion() -> String;
         fn getNumCores() -> i32;
         fn getCORSAllowedOrigin() -> String;
+        fn getNumConnections() -> i32;
     }
 }
diff --git a/lib/ain-cpp-imports/src/lib.rs b/lib/ain-cpp-imports/src/lib.rs
@@ -97,6 +97,9 @@ mod ffi {
     pub fn getCORSAllowedOrigin() -> String {
         unimplemented!("{}", UNIMPL_MSG)
     }
+    pub fn getNumConnections() -> i32 {
+        unimplemented!("{}", UNIMPL_MSG)
+    }
 }
 
 pub use ffi::Attributes;
@@ -204,5 +207,9 @@ pub fn get_cors_allowed_origin() -> String {
     ffi::getCORSAllowedOrigin()
 }
 
+pub fn get_num_connections() -> i32 {
+    ffi::getNumConnections()
+}
+
 #[cfg(test)]
 mod tests {}
diff --git a/lib/ain-grpc/src/rpc/net.rs b/lib/ain-grpc/src/rpc/net.rs
@@ -11,6 +11,9 @@ pub trait MetachainNetRPC {
     /// Returns the current network ID as a string.
     #[method(name = "version")]
     fn net_version(&self) -> RpcResult<String>;
+
+    #[method(name = "peerCount")]
+    fn peer_count(&self) -> RpcResult<String>;
 }
 
 pub struct MetachainNetRPCModule {
@@ -31,4 +34,10 @@ impl MetachainNetRPCServer for MetachainNetRPCModule {
 
         Ok(format!("{chain_id}"))
     }
+
+    fn peer_count(&self) -> RpcResult<String> {
+        let peer_count = ain_cpp_imports::get_num_connections();
+
+        Ok(format!("{:#x}", peer_count))
+    }
 }
diff --git a/src/ffi/ffiexports.cpp b/src/ffi/ffiexports.cpp
@@ -1,5 +1,6 @@
 #include <ffi/ffiexports.h>
 #include <util/system.h>
+#include <net.h>
 #include <dfi/mn_rpc.h>
 #include <key_io.h>
 #include <logging.h>
@@ -295,3 +296,7 @@ int32_t getNumCores() {
 rust::string getCORSAllowedOrigin() {
     return gArgs.GetArg("-rpcallowcors", "");
 }
+
+int32_t getNumConnections() {
+    return (int32_t)g_connman->GetNodeCount(CConnman::CONNECTIONS_ALL);
+}
diff --git a/src/ffi/ffiexports.h b/src/ffi/ffiexports.h
@@ -70,5 +70,6 @@ rust::vec<DST20Token> getDST20Tokens(std::size_t mnview_ptr);
 rust::string getClientVersion();
 int32_t getNumCores();
 rust::string getCORSAllowedOrigin();
+int32_t getNumConnections();
 
 #endif  // DEFI_FFI_FFIEXPORTS_H