Add RPC method for base64 encoding

blur-network · Jul 31, 2019 · 4cd30f5 · 4cd30f5
1 parent e7a8561
commit 4cd30f5
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 3 deletions.
diff --git a/src/rpc/core_rpc_server.cpp b/src/rpc/core_rpc_server.cpp
@@ -31,6 +31,7 @@
 
 #include "include_base_utils.h"
 #include "string_tools.h"
+#include "string_coding.h"
 using namespace epee;
 
 #include "core_rpc_server.h"
@@ -1918,7 +1919,31 @@ namespace cryptonote
      res.tree_hash = tree_hash_s;
      res.status = "OK";
      return true;
-}
+  }
+  //------------------------------------------------------------------------------------------------------------------------------
+  bool core_rpc_server::on_base64_encode(const COMMAND_RPC_BASE64_ENCODE::request& req, COMMAND_RPC_BASE64_ENCODE::response& res, epee::json_rpc::error& error_resp)
+  {
+
+    unsigned char const bytes = req.bytes;
+
+    if (sizeof(bytes) < 1) {
+      res.status = "Minimum length of 1 byte needed to encode to base 64";
+      return false;
+    }
+
+    std::string base_64_string = epee::string_encoding::base64_encode(&bytes, sizeof(bytes));
+    res.base_64_string = base_64_string;
+
+    bool r = base_64_string.empty();
+    if (r) {
+      res.status = "Error encoding to base64 string";
+      return false;
+    }
+
+    res.status = CORE_RPC_STATUS_OK;
+    return true;
+  }
+  //------------------------------------------------------------------------------------------------------------------------------
   bool core_rpc_server::on_flush_txpool(const COMMAND_RPC_FLUSH_TRANSACTION_POOL::request& req, COMMAND_RPC_FLUSH_TRANSACTION_POOL::response& res, epee::json_rpc::error& error_resp)
   {
     PERF_TIMER(on_flush_txpool);

diff --git a/src/rpc/core_rpc_server.h b/src/rpc/core_rpc_server.h
@@ -152,8 +152,9 @@ namespace cryptonote
         MAP_JON_RPC_WE_IF("relay_tx",            on_relay_tx,                   COMMAND_RPC_RELAY_TX, !m_restricted)
         MAP_JON_RPC_WE_IF("sync_info",           on_sync_info,                  COMMAND_RPC_SYNC_INFO, !m_restricted)
         MAP_JON_RPC_WE("get_txpool_backlog",     on_get_txpool_backlog,         COMMAND_RPC_GET_TRANSACTION_POOL_BACKLOG)
-        MAP_JON_RPC_WE("get_output_distribution", on_get_output_distribution, COMMAND_RPC_GET_OUTPUT_DISTRIBUTION)
-        MAP_JON_RPC_WE_IF("get_merkle_root",       on_get_merkle_root,            COMMAND_RPC_GET_MERKLE_ROOT, !m_restricted)
+        MAP_JON_RPC_WE("get_output_distribution",on_get_output_distribution, COMMAND_RPC_GET_OUTPUT_DISTRIBUTION)
+        MAP_JON_RPC_WE_IF("get_merkle_root",     on_get_merkle_root,            COMMAND_RPC_GET_MERKLE_ROOT, !m_restricted)
+        MAP_JON_RPC_WE("base64_encode",          on_base64_encode,            COMMAND_RPC_BASE64_ENCODE)
       END_JSON_RPC_MAP()
     END_URI_MAP2()
 
@@ -215,6 +216,7 @@ namespace cryptonote
     bool on_get_txpool_backlog(const COMMAND_RPC_GET_TRANSACTION_POOL_BACKLOG::request& req, COMMAND_RPC_GET_TRANSACTION_POOL_BACKLOG::response& res, epee::json_rpc::error& error_resp);
     bool on_get_output_distribution(const COMMAND_RPC_GET_OUTPUT_DISTRIBUTION::request& req, COMMAND_RPC_GET_OUTPUT_DISTRIBUTION::response& res, epee::json_rpc::error& error_resp);
     bool on_get_merkle_root(const COMMAND_RPC_GET_MERKLE_ROOT::request& req, COMMAND_RPC_GET_MERKLE_ROOT::response& res, epee::json_rpc::error& error);
+    bool on_base64_encode(const COMMAND_RPC_BASE64_ENCODE::request& req, COMMAND_RPC_BASE64_ENCODE::response& res, epee::json_rpc::error& error);
     //-----------------------
 
 private:

diff --git a/src/rpc/core_rpc_server_commands_defs.h b/src/rpc/core_rpc_server_commands_defs.h
@@ -979,6 +979,29 @@ namespace cryptonote
     };
   };
   //-----------------------------------------------
+  struct COMMAND_RPC_BASE64_ENCODE
+  {
+    struct request
+    {
+      unsigned char bytes;
+
+      BEGIN_KV_SERIALIZE_MAP()
+        KV_SERIALIZE(bytes)
+      END_KV_SERIALIZE_MAP()
+    };
+
+    struct response
+    {
+      std::string status;
+      std::string base_64_string;
+
+      BEGIN_KV_SERIALIZE_MAP()
+        KV_SERIALIZE(base_64_string)
+        KV_SERIALIZE(status)
+      END_KV_SERIALIZE_MAP()
+    };
+  };
+  //-----------------------------------------------
   struct COMMAND_RPC_GET_MERKLE_ROOT
   {
     struct request