*/
using System;
-using System.Linq;
using System.Text;
using System.Collections.Generic;
-using System.Security.Cryptography;
+using System.Diagnostics.Contracts;
namespace Novacoin
{
- /// <summary>
- /// Represents the block. Block consists of header, transaction array and header signature.
- /// </summary>
- public class CBlock
+ [Serializable]
+ public class BlockException : Exception
+ {
+ public BlockException()
+ {
+ }
+
+ public BlockException(string message)
+ : base(message)
+ {
+ }
+
+ public BlockException(string message, Exception inner)
+ : base(message, inner)
+ {
+ }
+ }
+
+ /// <summary>
+ /// Represents the block. Block consists of header, transaction array and header signature.
+ /// </summary>
+ public class CBlock
{
/// <summary>
/// Block header.
/// </summary>
public byte[] signature = new byte[0];
+ /// <summary>
+ /// Copy constructor.
+ /// </summary>
+ /// <param name="b">CBlock instance.</param>
public CBlock(CBlock b)
{
header = new CBlockHeader(b.header);
+ vtx = new CTransaction[b.vtx.Length];
for (int i = 0; i < b.vtx.Length; i++)
{
vtx[i] = new CTransaction(b.vtx[i]);
}
+ signature = new byte[b.signature.Length];
b.signature.CopyTo(signature, 0);
}
/// <summary>
/// Parse byte sequence and initialize new block instance
/// </summary>
- /// <param name="blockBytes"></param>
- public CBlock (IList<byte> blockBytes)
+ /// <param name="blockBytes">Bytes sequence.</param>
+ public CBlock (byte[] blockBytes)
{
- ByteQueue wBytes = new ByteQueue(blockBytes);
+ try
+ {
+ ByteQueue wBytes = new ByteQueue(ref blockBytes);
- // Fill the block header fields
- header = new CBlockHeader(wBytes.Get(80));
+ // Fill the block header fields
+ header = new CBlockHeader(wBytes.Get(80));
- // Parse transactions list
- vtx = CTransaction.ReadTransactionsList(ref wBytes);
+ // Parse transactions list
+ vtx = CTransaction.ReadTransactionsList(ref wBytes);
- // Read block signature
- signature = wBytes.Get((int)wBytes.GetVarInt());
+ // Read block signature
+ signature = wBytes.Get((int)wBytes.GetVarInt());
+ }
+ catch (Exception e)
+ {
+ throw new BlockException("Deserialization failed", e);
+ }
}
public CBlock()
vtx = new CTransaction[0];
}
+ public bool CheckBlock(bool fCheckPOW = true, bool fCheckMerkleRoot = true, bool fCheckSig = true)
+ {
+ var uniqueTX = new List<Hash256>(); // tx hashes
+ uint nSigOps = 0; // total sigops
+
+ // Basic sanity checkings
+ if (vtx.Length == 0 || Size > 1000000)
+ {
+ return false;
+ }
+
+ bool fProofOfStake = IsProofOfStake;
+
+ // First transaction must be coinbase, the rest must not be
+ if (!vtx[0].IsCoinBase)
+ {
+ return false;
+ }
+
+ if (!vtx[0].CheckTransaction())
+ {
+ return false;
+ }
+
+ uniqueTX.Add(vtx[0].Hash);
+ nSigOps += vtx[0].LegacySigOpCount;
+
+ if (fProofOfStake)
+ {
+ // Proof-of-STake related checkings. Note that we know here that 1st transactions is coinstake. We don't need
+ // check the type of 1st transaction because it's performed earlier by IsProofOfStake()
+
+ // nNonce must be zero for proof-of-stake blocks
+ if (header.nNonce != 0)
+ {
+ return false;
+ }
+
+ // Coinbase output should be empty if proof-of-stake block
+ if (vtx[0].vout.Length != 1 || !vtx[0].vout[0].IsEmpty)
+ {
+ return false;
+ }
+
+ // Check coinstake timestamp
+ if (header.nTime != vtx[1].nTime)
+ {
+ return false;
+ }
+
+ // Check proof-of-stake block signature
+ if (fCheckSig && !SignatureOK)
+ {
+ return false;
+ }
+
+ if (!vtx[1].CheckTransaction())
+ {
+ return false;
+ }
+
+ uniqueTX.Add(vtx[1].Hash);
+ nSigOps += vtx[1].LegacySigOpCount;
+ }
+ else
+ {
+ // Check proof of work matches claimed amount
+ if (fCheckPOW && !CheckProofOfWork(header.Hash, header.nBits))
+ {
+ return false;
+ }
+
+ // Check timestamp
+ if (header.nTime > NetUtils.FutureDrift(NetUtils.GetAdjustedTime()))
+ {
+ return false;
+ }
+
+ // Check coinbase timestamp
+ if (header.nTime < NetUtils.PastDrift(vtx[0].nTime))
+ {
+ return false;
+ }
+ }
+
+ // Iterate all transactions starting from second for proof-of-stake block
+ // or first for proof-of-work block
+ for (int i = fProofOfStake ? 2 : 1; i < vtx.Length; i++)
+ {
+ var tx = vtx[i];
+
+ // Reject coinbase transactions at non-zero index
+ if (tx.IsCoinBase)
+ {
+ return false;
+ }
+
+ // Reject coinstake transactions at index != 1
+ if (tx.IsCoinStake)
+ {
+ return false;
+ }
+
+ // Check transaction timestamp
+ if (header.nTime < tx.nTime)
+ {
+ return false;
+ }
+
+ // Check transaction consistency
+ if (!tx.CheckTransaction())
+ {
+ return false;
+ }
+
+ // Add transaction hash into list of unique transaction IDs
+ uniqueTX.Add(tx.Hash);
+
+ // Calculate sigops count
+ nSigOps += tx.LegacySigOpCount;
+ }
+
+ // Check for duplicate txids.
+ if (uniqueTX.Count != vtx.Length)
+ {
+ return false;
+ }
+
+ // Reject block if validation would consume too much resources.
+ if (nSigOps > 50000)
+ {
+ return false;
+ }
+
+ // Check merkle root
+ if (fCheckMerkleRoot && hashMerkleRoot != header.merkleRoot)
+ {
+ return false;
+ }
+
+ return true;
+ }
+
+ private bool CheckProofOfWork(ScryptHash256 hash, uint nBits)
+ {
+ // TODO: stub!
+
+ return true;
+ }
+
/// <summary>
/// Is this a Proof-of-Stake block?
/// </summary>
}
/// <summary>
- /// Get current instance as sequence of bytes
+ /// Get instance as sequence of bytes
/// </summary>
/// <returns>Byte sequence</returns>
- public IList<byte> Bytes
+ public static implicit operator byte[] (CBlock b)
{
- get
+ var r = new List<byte>();
+
+ r.AddRange((byte[])b.header);
+ r.AddRange(VarInt.EncodeVarInt(b.vtx.LongLength)); // transactions count
+
+ foreach (var tx in b.vtx)
{
- var r = new List<byte>();
+ r.AddRange((byte[])tx);
+ }
+
+ r.AddRange(VarInt.EncodeVarInt(b.signature.LongLength));
+ r.AddRange(b.signature);
+
+ return r.ToArray();
+ }
- r.AddRange(header.Bytes);
- r.AddRange(VarInt.EncodeVarInt(vtx.LongLength)); // transactions count
+ /// <summary>
+ /// Serialized size
+ /// </summary>
+ public int Size
+ {
+ get
+ {
+ int nSize = 80 + VarInt.GetEncodedSize(vtx.Length); // CBlockHeader + NumTx
foreach (var tx in vtx)
{
- r.AddRange(tx.Bytes);
+ nSize += tx.Size;
}
- r.AddRange(VarInt.EncodeVarInt(signature.LongLength));
- r.AddRange(signature);
+ nSize += VarInt.GetEncodedSize(signature.Length) + signature.Length;
- return r;
+ return nSize;
}
}
/// <summary>
- /// MErkle root
+ /// Get transaction offset inside block.
+ /// </summary>
+ /// <param name="nTx">Transaction index.</param>
+ /// <returns>Offset in bytes from the beginning of block header.</returns>
+ public int GetTxOffset(int nTx)
+ {
+ Contract.Requires<ArgumentException>(nTx >= 0 && nTx < vtx.Length, "Transaction index you've specified is incorrect.");
+
+ int nOffset = 80 + VarInt.GetEncodedSize(vtx.Length); // CBlockHeader + NumTx
+
+ for (int i = 0; i < nTx; i++)
+ {
+ nOffset += vtx[i].Size;
+ }
+
+ return nOffset;
+ }
+
+ /// <summary>
+ /// Merkle root
/// </summary>
public Hash256 hashMerkleRoot
{
foreach (var tx in vtx)
{
- merkleTree.AddRange(tx.Hash.hashBytes);
+ merkleTree.AddRange(Hash256.ComputeRaw256(tx));
}
- var hasher = new SHA256Managed();
- hasher.Initialize();
-
- int j = 0;
- for (int nSize = vtx.Length; nSize > 1; nSize = (nSize + 1) / 2)
+ int levelOffset = 0;
+ for (int nLevelSize = vtx.Length; nLevelSize > 1; nLevelSize = (nLevelSize + 1) / 2)
{
- for (int i = 0; i < nSize; i += 2)
+ for (int nLeft = 0; nLeft < nLevelSize; nLeft += 2)
{
- int i2 = Math.Min(i + 1, nSize - 1);
-
- var pair = new List<byte>();
-
- pair.AddRange(merkleTree.GetRange((j + i)*32, 32));
- pair.AddRange(merkleTree.GetRange((j + i2)*32, 32));
+ int nRight = Math.Min(nLeft + 1, nLevelSize - 1);
- var digest1 = hasher.ComputeHash(pair.ToArray());
- var digest2 = hasher.ComputeHash(digest1);
+ var left = merkleTree.GetRange((levelOffset + nLeft) * 32, 32).ToArray();
+ var right = merkleTree.GetRange((levelOffset + nRight) * 32, 32).ToArray();
- merkleTree.AddRange(digest2);
+ merkleTree.AddRange(Hash256.ComputeRaw256(ref left, ref right));
}
- j += nSize;
+ levelOffset += nLevelSize;
}
- return (merkleTree.Count == 0) ? new Hash256() : new Hash256(merkleTree.GetRange(merkleTree.Count-32, 32));
+ return (merkleTree.Count == 0) ? new Hash256() : new Hash256(merkleTree.GetRange(merkleTree.Count-32, 32).ToArray());
}
}