From 98d23c1aaf2bfe1b1133c0f6adf9bceb586f31b2 Mon Sep 17 00:00:00 2001
From: CryptoManiac <balthazar@yandex.ru>
Date: Wed, 24 Feb 2016 01:25:40 +0300
Subject: [PATCH] Experimental: use Solver to find out all solutions for txout.scriptPublicKey

---
 src/qt/transactionrecord.cpp |   73 +++++++++++++++++++++++++++++++++++------
 1 files changed, 62 insertions(+), 11 deletions(-)

diff --git a/src/qt/transactionrecord.cpp b/src/qt/transactionrecord.cpp
index 34d3edf..ec4ba10 100644
--- a/src/qt/transactionrecord.cpp
+++ b/src/qt/transactionrecord.cpp
@@ -3,6 +3,8 @@
 #include "wallet.h"
 #include "base58.h"
 
+extern CWallet* pwalletMain;
+
 /* Return positive answer if transaction should be shown in list.
  */
 bool TransactionRecord::showTransaction(const CWalletTx &wtx)
@@ -41,20 +43,69 @@ QList<TransactionRecord> TransactionRecord::decomposeTransaction(const CWallet *
             if(wallet->IsMine(txout))
             {
                 TransactionRecord sub(hash, nTime);
-                CTxDestination address;
                 sub.idx = parts.size(); // sequence number
                 sub.credit = txout.nValue;
-                if (ExtractDestination(txout.scriptPubKey, address) && IsMine(*wallet, address))
-                {
-                    // Received by Bitcoin Address
-                    sub.type = TransactionRecord::RecvWithAddress;
-                    sub.address = CBitcoinAddress(address).ToString();
-                }
-                else
+
+                if (!fTestNet)
                 {
-                    // Received by IP connection (deprecated features), or a multisignature or other non-simple transaction
-                    sub.type = TransactionRecord::RecvFromOther;
-                    sub.address = mapValue["from"];
+                    CTxDestination address;
+                    if (ExtractDestination(txout.scriptPubKey, address) && IsMine(*wallet, address))
+                    {
+                        // Received by Bitcoin Address
+                        sub.type = TransactionRecord::RecvWithAddress;
+                        sub.address = CBitcoinAddress(address).ToString();
+                    }
+                    else
+                    {
+                        // Received by IP connection (deprecated features), or a multisignature or other non-simple transaction
+                        sub.type = TransactionRecord::RecvFromOther;
+                        sub.address = mapValue["from"];
+                    }
+                    }
+                    else
+                    {
+                    txnouttype whichType;
+                    std::vector<valtype> vSolutions;
+                    if (Solver(txout.scriptPubKey, whichType, vSolutions))
+                    {
+                        CTxDestination address;
+                        if (whichType == TX_PUBKEY)
+                        {
+                            // Pay-to-Pubkey
+                            address = CPubKey(vSolutions[0]).GetID();
+                            sub.type = TransactionRecord::RecvWithAddress;
+                            sub.address = CBitcoinAddress(address).ToString();
+                        }
+                        else if (whichType == TX_PUBKEYHASH)
+                        {
+                            // Pay-to-PubkeyHash
+                            address = CKeyID(uint160(vSolutions[0]));
+                            sub.type = TransactionRecord::RecvWithAddress;
+                            sub.address = CBitcoinAddress(address).ToString();
+                        }
+                        else if (whichType == TX_SCRIPTHASH)
+                        {
+                            // Pay-to-ScriptHash
+                            address = CScriptID(uint160(vSolutions[0]));
+                            sub.type = TransactionRecord::RecvWithAddress;
+                            sub.address = CBitcoinAddress(address).ToString();
+                        }
+                        else if (whichType == TX_PUBKEY_DROP)
+                        {
+                            // Pay-to-Pubkey-R
+                            sub.type = TransactionRecord::RecvWithAddress;
+
+                            CMalleableKeyView view;
+                            pwalletMain->CheckOwnership(CPubKey(vSolutions[0]), CPubKey(vSolutions[1]), view);
+                            sub.address = view.GetMalleablePubKey().ToString();
+                        }
+                        else
+                        {
+                            // Received by IP connection (deprecated features), or a multisignature or other non-simple transaction
+                            sub.type = TransactionRecord::RecvFromOther;
+                            sub.address = mapValue["from"];
+                        }
+                    }
                 }
                 if (wtx.IsCoinBase())
                 {
-- 
1.7.1