Add wallet method for finding spendable notes in a CTransaction

2016-08-24 15:50:45 +12:00
parent 5db5e42ec3
commit 02e674555e
8 changed files with 264 additions and 2 deletions
--- a/src/wallet/gtest/test_wallet.cpp
+++ b/src/wallet/gtest/test_wallet.cpp
@@ -0,0 +1,98 @@
+#include <gtest/gtest.h>
+#include <sodium.h>
+
+#include "base58.h"
+#include "chainparams.h"
+#include "random.h"
+#include "wallet/wallet.h"
+#include "zcash/JoinSplit.hpp"
+#include "zcash/Note.hpp"
+#include "zcash/NoteEncryption.hpp"
+
+ZCJoinSplit* params = ZCJoinSplit::Unopened();
+
+CWalletTx GetValidReceive(const libzcash::SpendingKey& sk, CAmount value, bool randomInputs) {
+    CMutableTransaction mtx;
+    mtx.nVersion = 2; // Enable JoinSplits
+    mtx.vin.resize(2);
+    if (randomInputs) {
+        mtx.vin[0].prevout.hash = GetRandHash();
+        mtx.vin[1].prevout.hash = GetRandHash();
+    } else {
+        mtx.vin[0].prevout.hash = uint256S("0000000000000000000000000000000000000000000000000000000000000001");
+        mtx.vin[1].prevout.hash = uint256S("0000000000000000000000000000000000000000000000000000000000000002");
+    }
+    mtx.vin[0].prevout.n = 0;
+    mtx.vin[1].prevout.n = 0;
+
+    // Generate an ephemeral keypair.
+    uint256 joinSplitPubKey;
+    unsigned char joinSplitPrivKey[crypto_sign_SECRETKEYBYTES];
+    crypto_sign_keypair(joinSplitPubKey.begin(), joinSplitPrivKey);
+    mtx.joinSplitPubKey = joinSplitPubKey;
+
+    boost::array<libzcash::JSInput, 2> inputs = {
+        libzcash::JSInput(), // dummy input
+        libzcash::JSInput() // dummy input
+    };
+
+    boost::array<libzcash::JSOutput, 2> outputs = {
+        libzcash::JSOutput(), // dummy output
+        libzcash::JSOutput(sk.address(), value)
+    };
+
+    boost::array<libzcash::Note, 2> output_notes;
+
+    // Prepare JoinSplits
+    uint256 rt;
+    JSDescription jsdesc {*params, mtx.joinSplitPubKey, rt,
+                          inputs, outputs, value, 0, false};
+    mtx.vjoinsplit.push_back(jsdesc);
+
+    // Empty output script.
+    CScript scriptCode;
+    CTransaction signTx(mtx);
+    uint256 dataToBeSigned = SignatureHash(scriptCode, signTx, NOT_AN_INPUT, SIGHASH_ALL);
+
+    // Add the signature
+    assert(crypto_sign_detached(&mtx.joinSplitSig[0], NULL,
+                         dataToBeSigned.begin(), 32,
+                         joinSplitPrivKey
+                        ) == 0);
+
+    CTransaction tx {mtx};
+    CWalletTx wtx {NULL, tx};
+    return wtx;
+}
+
+libzcash::Note GetNote(const libzcash::SpendingKey& sk,
+                       const CTransaction& tx, size_t js, size_t n) {
+    ZCNoteDecryption decryptor {sk.viewing_key()};
+    auto hSig = tx.vjoinsplit[js].h_sig(*params, tx.joinSplitPubKey);
+    auto note_pt = libzcash::NotePlaintext::decrypt(
+        decryptor,
+        tx.vjoinsplit[js].ciphertexts[n],
+        tx.vjoinsplit[js].ephemeralKey,
+        hSig,
+        (unsigned char) n);
+    return note_pt.note(sk.address());
+}
+
+TEST(wallet_tests, find_note_in_tx) {
+    CWallet wallet;
+
+    auto sk = libzcash::SpendingKey::random();
+    wallet.AddSpendingKey(sk);
+
+    auto wtx = GetValidReceive(sk, 10, true);
+    auto note = GetNote(sk, wtx, 0, 1);
+    auto nullifier = note.nullifier(sk);
+
+    auto noteMap = wallet.FindMyNotes(wtx);
+    EXPECT_EQ(1, noteMap.size());
+
+    JSOutPoint jsoutpt {wtx.GetTxid(), 0, 1};
+    CNoteData nd {sk.address(), nullifier};
+    EXPECT_EQ(1, noteMap.count(jsoutpt));
+    EXPECT_EQ(nd, noteMap[jsoutpt]);
+}