From: CryptoManiac <balthazar@yandex.ru>
Date: Sun, 3 Apr 2016 23:42:25 +0000 (+0300)
Subject: RPC: now decryptdata/decryptmessage methods are able to understand WIF formatted... 
X-Git-Tag: nvc-v0.5.8~1
X-Git-Url: https://git.novaco.in/?p=novacoin.git;a=commitdiff_plain;h=05df71c9c292877e7469884227fd3ba84d02908b;hp=396877e60214f233b88100fa2831a97c5600e3f4

RPC: now decryptdata/decryptmessage methods are able to understand WIF formatted private key.
---

diff --git a/src/rpccrypt.cpp b/src/rpccrypt.cpp
index 69e14c0..91f0302 100644
--- a/src/rpccrypt.cpp
+++ b/src/rpccrypt.cpp
@@ -30,17 +30,26 @@ Value decryptdata(const Array& params, bool fHelp)
 {
     if (fHelp || params.size() != 2)
         throw runtime_error(
-            "decryptdata <novacoin address> <encrypted stream>\n"
+            "decryptdata <novacoin address or private key> <encrypted stream>\n"
             "Decrypt octet stream.\n");
 
     EnsureWalletIsUnlocked();
-    CBitcoinAddress addr(params[0].get_str());
-
-    CKeyID keyID;
-    addr.GetKeyID(keyID);
-
     CKey key;
-    pwalletMain->GetKey(keyID, key);
+    CBitcoinAddress addr(params[0].get_str());
+    if (addr.IsValid()) {
+        CKeyID keyID;
+        addr.GetKeyID(keyID);
+        if (!pwalletMain->GetKey(keyID, key))
+            throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "We have no private key for this address");
+    }
+    else {
+        CBitcoinSecret vchSecret;
+        if (!vchSecret.SetString(params[0].get_str()))
+            throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Provided private key is inconsistent.");
+        bool fCompressed;
+        CSecret secret = vchSecret.GetSecret(fCompressed);
+        key.SetSecret(secret, fCompressed);
+    }
 
     vector<unsigned char> vchDecrypted;
     key.DecryptData(ParseHex(params[1].get_str()), vchDecrypted);
@@ -68,17 +77,27 @@ Value decryptmessage(const Array& params, bool fHelp)
 {
     if (fHelp || params.size() != 2)
         throw runtime_error(
-            "decryptmessage <novacoin address> <encrypted message>\n"
+            "decryptmessage <novacoin address or private key> <encrypted message>\n"
             "Decrypt message string.\n");
 
     EnsureWalletIsUnlocked();
-    CBitcoinAddress addr(params[0].get_str());
-
-    CKeyID keyID;
-    addr.GetKeyID(keyID);
 
     CKey key;
-    pwalletMain->GetKey(keyID, key);
+    CBitcoinAddress addr(params[0].get_str());
+    if (addr.IsValid()) {
+        CKeyID keyID;
+        addr.GetKeyID(keyID);
+        if (!pwalletMain->GetKey(keyID, key))
+            throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "We have no private key for this address");
+    }
+    else {
+        CBitcoinSecret vchSecret;
+        if (!vchSecret.SetString(params[0].get_str()))
+            throw JSONRPCError(RPC_INVALID_ADDRESS_OR_KEY, "Provided private key is inconsistent.");
+        bool fCompressed;
+        CSecret secret = vchSecret.GetSecret(fCompressed);
+        key.SetSecret(secret, fCompressed);
+    }
 
     vector<unsigned char> vchEncrypted;
     if (!DecodeBase58Check(params[1].get_str(), vchEncrypted))