Skip to content

Commit

Permalink
Optimize transaction
Browse files Browse the repository at this point in the history
  • Loading branch information
winsvega committed Mar 9, 2021
1 parent 47c69eb commit 931a191
Show file tree
Hide file tree
Showing 5 changed files with 49 additions and 58 deletions.
3 changes: 2 additions & 1 deletion retesteth/session/ToolBackend/ToolChain.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -183,7 +183,8 @@ ToolResponse ToolChain::mineBlockOnTool(EthereumBlockState const& _block, SealEn
if (tr.getCContent().gasLimit().asU256() <= c_maxGasLimit) // tool fails on limits here.
txs.addArrayObject(tr.getCContent().asDataObject(ExportOrder::ToolStyle));
else
ETH_WARNING("Retesteth rejecting tx with gasLimit > 64 bits for tool");
ETH_WARNING(
"Retesteth rejecting tx with gasLimit > 64 bits for tool" + TestOutputHelper::get().testInfo().errorDebug());
}
writeFile(txsPath.string(), txs.asJson());

Expand Down
40 changes: 21 additions & 19 deletions retesteth/testStructures/types/Ethereum/Transaction.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,7 @@ void Transaction::fromDataObject(DataObject const& _data)
throw test::UpwardsException("Incorrect transaction `v` value: " + m_v.getCContent().asString());
m_r = spVALUE(new VALUE(_data.atKey("r")));
m_s = spVALUE(new VALUE(_data.atKey("s")));
rebuildRLP();
}
requireJsonFields(_data, "Transaction " + _data.getKey(),
{
Expand Down Expand Up @@ -109,7 +110,6 @@ Transaction::Transaction(BYTES const& _rlp)
fromRLP(rlp);
}


void Transaction::streamHeader(dev::RLPStream& _s) const
{
_s << nonce().asU256();
Expand Down Expand Up @@ -141,32 +141,21 @@ void Transaction::buildVRS(VALUE const& _secret)
assignV(spVALUE(new VALUE(v)));
assignR(spVALUE(new VALUE(r)));
assignS(spVALUE(new VALUE(s)));
rebuildRLP();
}

void Transaction::assignV(spVALUE const _v) { m_v = _v; }
void Transaction::assignR(spVALUE const _r) { m_r = _r; }
void Transaction::assignS(spVALUE const _s) { m_s = _s; }

BYTES const Transaction::getSignedRLP() const
BYTES const& Transaction::getSignedRLP() const
{
dev::RLPStream sWithSignature;
sWithSignature.appendList(9);
streamHeader(sWithSignature);
sWithSignature << v().asU256().convert_to<dev::byte>();
sWithSignature << r().asU256();
sWithSignature << s().asU256();
return BYTES(dev::toHexPrefixed(sWithSignature.out()));
return m_signedRLPdata.getCContent();
}

dev::RLPStream const Transaction::asRLPStream() const
dev::RLPStream const& Transaction::asRLPStream() const
{
dev::RLPStream out;
out.appendList(9);
streamHeader(out);
out << v().asU256().convert_to<dev::byte>();
out << r().asU256();
out << s().asU256();
return out;
return m_outRlpStream;
}

const DataObject Transaction::asDataObject(ExportOrder _order) const
Expand Down Expand Up @@ -202,9 +191,22 @@ const DataObject Transaction::asDataObject(ExportOrder _order) const
return out;
}

FH32 Transaction::hash() const
FH32 const& Transaction::hash() const
{
return FH32("0x" + dev::toString(dev::sha3(asRLPStream().out())));
return m_hash.getCContent();
}

void Transaction::rebuildRLP()
{
dev::RLPStream out;
out.appendList(9);
streamHeader(out);
out << v().asU256().convert_to<dev::byte>();
out << r().asU256();
out << s().asU256();
m_outRlpStream = out;
m_signedRLPdata = spBYTES(new BYTES(dev::toHexPrefixed(out.out())));
m_hash = spFH32(new FH32("0x" + dev::toString(dev::sha3(out.out()))));
}

} // namespace teststruct
Expand Down
12 changes: 9 additions & 3 deletions retesteth/testStructures/types/Ethereum/Transaction.h
Original file line number Diff line number Diff line change
Expand Up @@ -36,11 +36,11 @@ struct Transaction : GCP_SPointerBase
VALUE const& v() const { return m_v.getCContent(); }
VALUE const& r() const { return m_r.getCContent(); }
VALUE const& s() const { return m_s.getCContent(); }
virtual FH32 hash() const;
FH32 const& hash() const;
virtual TransactionType type() const { return TransactionType::LEGACY; }

virtual BYTES const getSignedRLP() const;
virtual dev::RLPStream const asRLPStream() const;
BYTES const& getSignedRLP() const;
dev::RLPStream const& asRLPStream() const;
virtual DataObject const asDataObject(ExportOrder _order = ExportOrder::Default) const;

virtual ~Transaction(){};
Expand Down Expand Up @@ -76,6 +76,12 @@ struct Transaction : GCP_SPointerBase
void assignV(spVALUE const);
void assignR(spVALUE const);
void assignS(spVALUE const);

// Optimization
spFH32 m_hash;
dev::RLPStream m_outRlpStream;
spBYTES m_signedRLPdata;
virtual void rebuildRLP();
};

typedef GCP_SPointer<Transaction> spTransaction;
Expand Down
45 changes: 15 additions & 30 deletions retesteth/testStructures/types/Ethereum/TransactionAccessList.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,8 @@ TransactionAccessList::TransactionAccessList(DataObject const& _data, string con
m_accessList = spAccessList(new AccessList(_data.atKey("accessList")));
if (_data.count("secretKey"))
buildVRS(VALUE(_data.atKey("secretKey"))); // Build without v + 27
else
rebuildRLP();
}

TransactionAccessList::TransactionAccessList(dev::RLP const& _rlp)
Expand Down Expand Up @@ -82,6 +84,7 @@ void TransactionAccessList::buildVRS(VALUE const& _secret)
assignV(spVALUE(new VALUE(v)));
assignR(spVALUE(new VALUE(r)));
assignS(spVALUE(new VALUE(s)));
rebuildRLP();
}

void TransactionAccessList::streamHeader(dev::RLPStream& _s) const
Expand All @@ -106,31 +109,6 @@ void TransactionAccessList::streamHeader(dev::RLPStream& _s) const
_s.appendRaw(accessList.out());
}

BYTES const TransactionAccessList::getSignedRLP() const
{
return BYTES("0x" + dev::toHex(asRLPStream().out()));
}

dev::RLPStream const TransactionAccessList::asRLPStream() const
{
// RLP(01 + tr.rlp)
dev::RLPStream wrapper;
dev::RLPStream out;
out.appendList(11);
streamHeader(out);
out << v().asU256().convert_to<dev::byte>();
out << r().asU256();
out << s().asU256();

// Alter output with prefixed 01 byte + tr.rlp
dev::bytes outa = out.out();
outa.insert(outa.begin(), dev::byte(1));

// Encode bytearray into rlp
wrapper << outa;
return wrapper;
}

DataObject const TransactionAccessList::asDataObject(ExportOrder _order) const
{
DataObject out = Transaction::asDataObject(_order);
Expand All @@ -146,19 +124,26 @@ DataObject const TransactionAccessList::asDataObject(ExportOrder _order) const
return out;
}

FH32 TransactionAccessList::hash() const
void TransactionAccessList::rebuildRLP()
{
// HASH (01 + tr.rlp)
// RLP(01 + tr.rlp)
dev::RLPStream wrapper;
dev::RLPStream out;
out.appendList(11);
streamHeader(out);
out << v().asU256().convert_to<dev::byte>();
out << r().asU256();
out << s().asU256();

dev::bytes trRLP = out.out();
trRLP.insert(trRLP.begin(), dev::byte(1));
return FH32("0x" + dev::toString(dev::sha3(trRLP)));
// Alter output with prefixed 01 byte + tr.rlp
dev::bytes outa = out.out();
outa.insert(outa.begin(), dev::byte(1));

// Encode bytearray into rlp
wrapper << outa;
m_outRlpStream = wrapper;
m_signedRLPdata = spBYTES(new BYTES(dev::toHexPrefixed(m_outRlpStream.out())));
m_hash = spFH32(new FH32("0x" + dev::toString(dev::sha3(outa))));
}


Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,19 +13,16 @@ struct TransactionAccessList : Transaction
TransactionAccessList(BYTES const&);
TransactionAccessList(dev::RLP const&);

BYTES const getSignedRLP() const override;
dev::RLPStream const asRLPStream() const override;
DataObject const asDataObject(ExportOrder _order = ExportOrder::Default) const override;

FH32 hash() const override;
TransactionType type() const override { return TransactionType::ACCESSLIST; }

private:
// void fromDataObject(DataObject const&) override;
void fromRLP(dev::RLP const&) override;

void buildVRS(VALUE const& _secret) override;
void streamHeader(dev::RLPStream& _stream) const override;

void rebuildRLP() override;
spAccessList m_accessList;
};

Expand Down

0 comments on commit 931a191

Please sign in to comment.