7 files changed, 138 insertions, 32 deletions

diff --git a/tests/performance_tests/CMakeLists.txt b/tests/performance_tests/CMakeLists.txt
index 37accb393..5ec53cd2b 100644
--- a/tests/performance_tests/CMakeLists.txt
+++ b/tests/performance_tests/CMakeLists.txt
@@ -30,11 +30,12 @@ set(performance_tests_sources
   main.cpp)
 
 set(performance_tests_headers
-  check_ring_signature.h
+  check_tx_signature.h
   cn_slow_hash.h
   construct_tx.h
   derive_public_key.h
   derive_secret_key.h
+  ge_frombytes_vartime.h
   generate_key_derivation.h
   generate_key_image.h
   generate_key_image_helper.h
diff --git a/tests/performance_tests/check_ring_signature.h b/tests/performance_tests/check_tx_signature.h
index a55849742..fe595a4da 100644
--- a/tests/performance_tests/check_ring_signature.h
+++ b/tests/performance_tests/check_tx_signature.h
@@ -36,17 +36,19 @@
 #include "cryptonote_core/cryptonote_basic.h"
 #include "cryptonote_core/cryptonote_format_utils.h"
 #include "crypto/crypto.h"
+#include "ringct/rctSigs.h"
 
 #include "multi_tx_test_base.h"
 
-template<size_t a_ring_size>
-class test_check_ring_signature : private multi_tx_test_base<a_ring_size>
+template<size_t a_ring_size, bool a_rct>
+class test_check_tx_signature : private multi_tx_test_base<a_ring_size>
 {
   static_assert(0 < a_ring_size, "ring_size must be greater than 0");
 
 public:
-  static const size_t loop_count = a_ring_size < 100 ? 100 : 10;
+  static const size_t loop_count = a_rct ? 10 : a_ring_size < 100 ? 100 : 10;
   static const size_t ring_size = a_ring_size;
+  static const bool rct = a_rct;
 
   typedef multi_tx_test_base<a_ring_size> base_class;
 
@@ -62,7 +64,8 @@ public:
     std::vector<tx_destination_entry> destinations;
     destinations.push_back(tx_destination_entry(this->m_source_amount, m_alice.get_keys().m_account_address));
 
-    if (!construct_tx(this->m_miners[this->real_source_idx].get_keys(), this->m_sources, destinations, std::vector<uint8_t>(), m_tx, 0))
+    crypto::secret_key tx_key;
+    if (!construct_tx_and_get_tx_key(this->m_miners[this->real_source_idx].get_keys(), this->m_sources, destinations, std::vector<uint8_t>(), m_tx, 0, tx_key, rct))
       return false;
 
     get_transaction_prefix_hash(m_tx, m_tx_prefix_hash);
@@ -72,8 +75,18 @@ public:
 
   bool test()
   {
-    const cryptonote::txin_to_key& txin = boost::get<cryptonote::txin_to_key>(m_tx.vin[0]);
-    return crypto::check_ring_signature(m_tx_prefix_hash, txin.k_image, this->m_public_key_ptrs, ring_size, m_tx.signatures[0].data());
+    if (rct)
+    {
+      if (m_tx.rct_signatures.type == rct::RCTTypeFull)
+        return rct::verRct(m_tx.rct_signatures);
+      else
+        return rct::verRctSimple(m_tx.rct_signatures);
+    }
+    else
+    {
+      const cryptonote::txin_to_key& txin = boost::get<cryptonote::txin_to_key>(m_tx.vin[0]);
+      return crypto::check_ring_signature(m_tx_prefix_hash, txin.k_image, this->m_public_key_ptrs, ring_size, m_tx.signatures[0].data());
+    }
   }
 
 private:
diff --git a/tests/performance_tests/construct_tx.h b/tests/performance_tests/construct_tx.h
index d3409c0f3..aef455eaa 100644
--- a/tests/performance_tests/construct_tx.h
+++ b/tests/performance_tests/construct_tx.h
@@ -36,7 +36,7 @@
 
 #include "multi_tx_test_base.h"
 
-template<size_t a_in_count, size_t a_out_count>
+template<size_t a_in_count, size_t a_out_count, bool a_rct>
 class test_construct_tx : private multi_tx_test_base<a_in_count>
 {
   static_assert(0 < a_in_count, "in_count must be greater than 0");
@@ -46,6 +46,7 @@ public:
   static const size_t loop_count = (a_in_count + a_out_count < 100) ? 100 : 10;
   static const size_t in_count  = a_in_count;
   static const size_t out_count = a_out_count;
+  static const bool rct = a_rct;
 
   typedef multi_tx_test_base<a_in_count> base_class;
 
@@ -68,7 +69,8 @@ public:
 
   bool test()
   {
-    return cryptonote::construct_tx(this->m_miners[this->real_source_idx].get_keys(), this->m_sources, m_destinations, std::vector<uint8_t>(), m_tx, 0);
+    crypto::secret_key tx_key;
+    return cryptonote::construct_tx_and_get_tx_key(this->m_miners[this->real_source_idx].get_keys(), this->m_sources, m_destinations, std::vector<uint8_t>(), m_tx, 0, tx_key, rct);
   }
 
 private:
diff --git a/tests/performance_tests/ge_frombytes_vartime.h b/tests/performance_tests/ge_frombytes_vartime.h
new file mode 100644
index 000000000..c815422f9
--- /dev/null
+++ b/tests/performance_tests/ge_frombytes_vartime.h
@@ -0,0 +1,70 @@
+// Copyright (c) 2014-2016, The Monero Project
+// 
+// All rights reserved.
+// 
+// Redistribution and use in source and binary forms, with or without modification, are
+// permitted provided that the following conditions are met:
+// 
+// 1. Redistributions of source code must retain the above copyright notice, this list of
+//    conditions and the following disclaimer.
+// 
+// 2. Redistributions in binary form must reproduce the above copyright notice, this list
+//    of conditions and the following disclaimer in the documentation and/or other
+//    materials provided with the distribution.
+// 
+// 3. Neither the name of the copyright holder nor the names of its contributors may be
+//    used to endorse or promote products derived from this software without specific
+//    prior written permission.
+// 
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY
+// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
+// THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
+// STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
+// THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+// 
+// Parts of this file are originally copyright (c) 2012-2013 The Cryptonote developers
+
+#pragma once
+
+#include "crypto/crypto.h"
+#include "cryptonote_core/cryptonote_basic.h"
+
+#include "single_tx_test_base.h"
+
+class test_ge_frombytes_vartime : public multi_tx_test_base<1>
+{
+public:
+  static const size_t loop_count = 10000;
+
+  typedef multi_tx_test_base<1> base_class;
+
+  bool init()
+  {
+    using namespace cryptonote;
+
+    if (!base_class::init())
+      return false;
+
+    m_alice.generate();
+
+    std::vector<tx_destination_entry> destinations;
+    destinations.push_back(tx_destination_entry(1, m_alice.get_keys().m_account_address));
+
+    return construct_tx(this->m_miners[this->real_source_idx].get_keys(), this->m_sources, destinations, std::vector<uint8_t>(), m_tx, 0);
+  }
+
+  bool test()
+  {
+    ge_p3 unp;
+    const cryptonote::txin_to_key& txin = boost::get<cryptonote::txin_to_key>(m_tx.vin[0]);
+    return ge_frombytes_vartime(&unp, (const unsigned char*) &txin.k_image) == 0;
+  }
+
+private:
+  cryptonote::account_base m_alice;
+  cryptonote::transaction m_tx;
+};
diff --git a/tests/performance_tests/main.cpp b/tests/performance_tests/main.cpp
index 84a51aa96..d09276230 100644
--- a/tests/performance_tests/main.cpp
+++ b/tests/performance_tests/main.cpp
@@ -33,10 +33,11 @@
 
 // tests
 #include "construct_tx.h"
-#include "check_ring_signature.h"
+#include "check_tx_signature.h"
 #include "cn_slow_hash.h"
 #include "derive_public_key.h"
 #include "derive_secret_key.h"
+#include "ge_frombytes_vartime.h"
 #include "generate_key_derivation.h"
 #include "generate_key_image.h"
 #include "generate_key_image_helper.h"
@@ -50,31 +51,47 @@ int main(int argc, char** argv)
   performance_timer timer;
   timer.start();
 
-  TEST_PERFORMANCE2(test_construct_tx, 1, 1);
-  TEST_PERFORMANCE2(test_construct_tx, 1, 2);
-  TEST_PERFORMANCE2(test_construct_tx, 1, 10);
-  TEST_PERFORMANCE2(test_construct_tx, 1, 100);
-  TEST_PERFORMANCE2(test_construct_tx, 1, 1000);
+  TEST_PERFORMANCE3(test_construct_tx, 1, 1, false);
+  TEST_PERFORMANCE3(test_construct_tx, 1, 2, false);
+  TEST_PERFORMANCE3(test_construct_tx, 1, 10, false);
+  TEST_PERFORMANCE3(test_construct_tx, 1, 100, false);
+  TEST_PERFORMANCE3(test_construct_tx, 1, 1000, false);
 
-  TEST_PERFORMANCE2(test_construct_tx, 2, 1);
-  TEST_PERFORMANCE2(test_construct_tx, 2, 2);
-  TEST_PERFORMANCE2(test_construct_tx, 2, 10);
-  TEST_PERFORMANCE2(test_construct_tx, 2, 100);
+  TEST_PERFORMANCE3(test_construct_tx, 2, 1, false);
+  TEST_PERFORMANCE3(test_construct_tx, 2, 2, false);
+  TEST_PERFORMANCE3(test_construct_tx, 2, 10, false);
+  TEST_PERFORMANCE3(test_construct_tx, 2, 100, false);
 
-  TEST_PERFORMANCE2(test_construct_tx, 10, 1);
-  TEST_PERFORMANCE2(test_construct_tx, 10, 2);
-  TEST_PERFORMANCE2(test_construct_tx, 10, 10);
-  TEST_PERFORMANCE2(test_construct_tx, 10, 100);
+  TEST_PERFORMANCE3(test_construct_tx, 10, 1, false);
+  TEST_PERFORMANCE3(test_construct_tx, 10, 2, false);
+  TEST_PERFORMANCE3(test_construct_tx, 10, 10, false);
+  TEST_PERFORMANCE3(test_construct_tx, 10, 100, false);
 
-  TEST_PERFORMANCE2(test_construct_tx, 100, 1);
-  TEST_PERFORMANCE2(test_construct_tx, 100, 2);
-  TEST_PERFORMANCE2(test_construct_tx, 100, 10);
-  TEST_PERFORMANCE2(test_construct_tx, 100, 100);
+  TEST_PERFORMANCE3(test_construct_tx, 100, 1, false);
+  TEST_PERFORMANCE3(test_construct_tx, 100, 2, false);
+  TEST_PERFORMANCE3(test_construct_tx, 100, 10, false);
+  TEST_PERFORMANCE3(test_construct_tx, 100, 100, false);
 
-  TEST_PERFORMANCE1(test_check_ring_signature, 1);
-  TEST_PERFORMANCE1(test_check_ring_signature, 2);
-  TEST_PERFORMANCE1(test_check_ring_signature, 10);
-  TEST_PERFORMANCE1(test_check_ring_signature, 100);
+  TEST_PERFORMANCE3(test_construct_tx, 2, 1, true);
+  TEST_PERFORMANCE3(test_construct_tx, 2, 2, true);
+  TEST_PERFORMANCE3(test_construct_tx, 2, 10, true);
+
+  TEST_PERFORMANCE3(test_construct_tx, 10, 1, true);
+  TEST_PERFORMANCE3(test_construct_tx, 10, 2, true);
+  TEST_PERFORMANCE3(test_construct_tx, 10, 10, true);
+
+  TEST_PERFORMANCE3(test_construct_tx, 100, 1, true);
+  TEST_PERFORMANCE3(test_construct_tx, 100, 2, true);
+  TEST_PERFORMANCE3(test_construct_tx, 100, 10, true);
+
+  TEST_PERFORMANCE2(test_check_tx_signature, 1, false);
+  TEST_PERFORMANCE2(test_check_tx_signature, 2, false);
+  TEST_PERFORMANCE2(test_check_tx_signature, 10, false);
+  TEST_PERFORMANCE2(test_check_tx_signature, 100, false);
+
+  TEST_PERFORMANCE2(test_check_tx_signature, 2, true);
+  TEST_PERFORMANCE2(test_check_tx_signature, 10, true);
+  TEST_PERFORMANCE2(test_check_tx_signature, 100, true);
 
   TEST_PERFORMANCE0(test_is_out_to_acc);
   TEST_PERFORMANCE0(test_generate_key_image_helper);
@@ -82,6 +99,7 @@ int main(int argc, char** argv)
   TEST_PERFORMANCE0(test_generate_key_image);
   TEST_PERFORMANCE0(test_derive_public_key);
   TEST_PERFORMANCE0(test_derive_secret_key);
+  TEST_PERFORMANCE0(test_ge_frombytes_vartime);
 
   TEST_PERFORMANCE0(test_cn_slow_hash);
 
diff --git a/tests/performance_tests/multi_tx_test_base.h b/tests/performance_tests/multi_tx_test_base.h
index d8898b60d..feabab022 100644
--- a/tests/performance_tests/multi_tx_test_base.h
+++ b/tests/performance_tests/multi_tx_test_base.h
@@ -59,7 +59,7 @@ public:
         return false;
 
       txout_to_key tx_out = boost::get<txout_to_key>(m_miner_txs[i].vout[0].target);
-      output_entries.push_back(std::make_pair(i, rct::ctkey({rct::pk2rct(tx_out.key), rct::identity()})));
+      output_entries.push_back(std::make_pair(i, rct::ctkey({rct::pk2rct(tx_out.key), rct::zeroCommit(m_miner_txs[i].vout[0].amount)})));
       m_public_keys[i] = tx_out.key;
       m_public_key_ptrs[i] = &m_public_keys[i];
     }
@@ -72,6 +72,7 @@ public:
     source_entry.real_output_in_tx_index = 0;
     source_entry.outputs.swap(output_entries);
     source_entry.real_output = real_source_idx;
+    source_entry.mask = rct::identity();
     source_entry.rct = false;
 
     m_sources.push_back(source_entry);
diff --git a/tests/performance_tests/performance_tests.h b/tests/performance_tests/performance_tests.h
index 9781349bf..77707148b 100644
--- a/tests/performance_tests/performance_tests.h
+++ b/tests/performance_tests/performance_tests.h
@@ -142,3 +142,4 @@ void run_test(const char* test_name)
 #define TEST_PERFORMANCE0(test_class)         run_test< test_class >(QUOTEME(test_class))
 #define TEST_PERFORMANCE1(test_class, a0)     run_test< test_class<a0> >(QUOTEME(test_class<a0>))
 #define TEST_PERFORMANCE2(test_class, a0, a1) run_test< test_class<a0, a1> >(QUOTEME(test_class) "<" QUOTEME(a0) ", " QUOTEME(a1) ">")
+#define TEST_PERFORMANCE3(test_class, a0, a1, a2) run_test< test_class<a0, a1, a2> >(QUOTEME(test_class) "<" QUOTEME(a0) ", " QUOTEME(a1) ", " QUOTEME(a2) ">")