danicoin/tests/unit_tests/TransactionApiHelpers.h
2015-05-27 13:28:09 +01:00

105 lines
3.4 KiB
C++

// Copyright (c) 2012-2015, The CryptoNote developers, The Bytecoin developers
//
// This file is part of Bytecoin.
//
// Bytecoin is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Bytecoin is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with Bytecoin. If not, see <http://www.gnu.org/licenses/>.
#pragma once
#include "ITransaction.h"
#include "crypto/crypto.h"
#include "cryptonote_core/cryptonote_format_utils.h"
#include "transfers/TransfersContainer.h"
namespace {
using namespace CryptoNote;
inline TransactionTypes::KeyPair generateKeys() {
TransactionTypes::KeyPair kp;
crypto::generate_keys(
reinterpret_cast<crypto::public_key&>(kp.publicKey),
reinterpret_cast<crypto::secret_key&>(kp.secretKey));
return kp;
}
inline AccountKeys accountKeysFromKeypairs(
const TransactionTypes::KeyPair& viewKeys,
const TransactionTypes::KeyPair& spendKeys) {
AccountKeys ak;
ak.address.spendPublicKey = spendKeys.publicKey;
ak.address.viewPublicKey = viewKeys.publicKey;
ak.spendSecretKey = spendKeys.secretKey;
ak.viewSecretKey = viewKeys.secretKey;
return ak;
}
inline AccountKeys generateAccountKeys() {
return accountKeysFromKeypairs(generateKeys(), generateKeys());
}
KeyImage generateKeyImage() {
return crypto::rand<KeyImage>();
}
KeyImage generateKeyImage(const AccountKeys& keys, size_t idx, const PublicKey& txPubKey) {
KeyImage keyImage;
CryptoNote::KeyPair in_ephemeral;
CryptoNote::generate_key_image_helper(
reinterpret_cast<const CryptoNote::account_keys&>(keys),
reinterpret_cast<const crypto::public_key&>(txPubKey),
idx,
in_ephemeral,
reinterpret_cast<crypto::key_image&>(keyImage));
return keyImage;
}
void addTestInput(ITransaction& transaction, uint64_t amount) {
TransactionTypes::InputKey input;
input.amount = amount;
input.keyImage = generateKeyImage();
input.keyOffsets.emplace_back(1);
transaction.addInput(input);
}
TransactionOutputInformationIn addTestKeyOutput(ITransaction& transaction, uint64_t amount,
uint64_t globalOutputIndex, const AccountKeys& senderKeys = generateAccountKeys()) {
uint32_t index = static_cast<uint32_t>(transaction.addOutput(amount, senderKeys.address));
TransactionTypes::OutputKey output;
transaction.getOutput(index, output);
TransactionOutputInformationIn outputInfo;
outputInfo.type = TransactionTypes::OutputType::Key;
outputInfo.amount = output.amount;
outputInfo.globalOutputIndex = globalOutputIndex;
outputInfo.outputInTransaction = index;
outputInfo.transactionPublicKey = transaction.getTransactionPublicKey();
outputInfo.outputKey = output.key;
outputInfo.keyImage = generateKeyImage(senderKeys, index, transaction.getTransactionPublicKey());
return outputInfo;
}
}
namespace CryptoNote {
inline bool operator == (const AccountKeys& a, const AccountKeys& b) {
return memcmp(&a, &b, sizeof(a)) == 0;
}
}