Use byte[] instead of IEnumerable<byte> if possible
[NovacoinLibrary.git] / Novacoin / CBlock.cs
index 4402480..3a5a36d 100644 (file)
@@ -1,4 +1,22 @@
-\feffusing System;
+\feff/**
+ *  Novacoin classes library
+ *  Copyright (C) 2015 Alex D. (balthazar.ad@gmail.com)
+
+ *  This program is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU Affero General Public License as
+ *  published by the Free Software Foundation, either version 3 of the
+ *  License, or (at your option) any later version.
+
+ *  This program 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 Affero General Public License for more details.
+
+ *  You should have received a copy of the GNU Affero General Public License
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+ using System;
 using System.Text;
 using System.Collections.Generic;
 
@@ -19,72 +37,152 @@ namespace Novacoin
                /// </summary>
                public CTransaction[] vtx;
 
-               /// <summary>
-               /// Block header signature.
-               /// </summary>
-               public byte[] signature;
+        /// <summary>
+        /// Block header signature.
+        /// </summary>
+        public byte[] signature = new byte[0];
+
+        public CBlock(CBlock b)
+        {
+            header = new CBlockHeader(b.header);
+
+            for (int i = 0; i < b.vtx.Length; i++)
+            {
+                vtx[i] = new CTransaction(b.vtx[i]);
+            }
+
+            b.signature.CopyTo(signature, 0);
+        }
 
         /// <summary>
         /// Parse byte sequence and initialize new block instance
         /// </summary>
         /// <param name="blockBytes"></param>
-               public CBlock (List<byte> blockBytes)
+               public CBlock (IList<byte> blockBytes)
                {
-            header = new CBlockHeader();
-
-            WrappedList<byte> wBytes = new WrappedList<byte>(blockBytes);
+            ByteQueue wBytes = new ByteQueue(blockBytes);
 
             // Fill the block header fields
-            header.nVersion = Interop.LEBytesToUInt32(wBytes.GetItems(4));
-            header.prevHash = new Hash256(wBytes.GetItems(32));
-            header.merkleRoot = new Hash256(wBytes.GetItems(32));
-            header.nTime = Interop.LEBytesToUInt32(wBytes.GetItems(4));
-            header.nBits = Interop.LEBytesToUInt32(wBytes.GetItems(4));
-            header.nNonce = Interop.LEBytesToUInt32(wBytes.GetItems(4));
+            header = new CBlockHeader(wBytes.Get(80));
 
             // Parse transactions list
             vtx = CTransaction.ReadTransactionsList(ref wBytes);
 
             // Read block signature
-            signature = wBytes.GetItems((int)VarInt.ReadVarInt(ref wBytes));
+            signature = wBytes.Get((int)wBytes.GetVarInt());
                }
 
+        public CBlock()
+        {
+            // Initialize empty array of transactions. Please note that such 
+            // configuration is not valid real block since it has to provide 
+            // at least one transaction.
+            vtx = new CTransaction[0];
+        }
+
         /// <summary>
-        /// Convert current instance into sequence of bytes
+        /// Is this a Proof-of-Stake block?
         /// </summary>
-        /// <returns>Byte sequence</returns>
-        public IList<byte> ToBytes()
+        public bool IsProofOfStake
         {
-            List<byte> r = new List<byte>();
-
-            r.AddRange(header.ToBytes());
-            r.AddRange(VarInt.EncodeVarInt(vtx.LongLength)); // transactions count
+            get
+            {
+                return (vtx.Length > 1 && vtx[1].IsCoinStake);
+            }
+        }
 
-            foreach (CTransaction tx in vtx)
+        /// <summary>
+        /// Was this signed correctly?
+        /// </summary>
+        public bool SignatureOK
+        {
+            get
             {
-                r.AddRange(tx.ToBytes());
+                if (IsProofOfStake)
+                {
+                    if (signature.Length == 0)
+                    {
+                        return false; // No signature
+                    }
+
+                    txnouttype whichType;
+                    IList<byte[]> solutions;
+
+                    if (!ScriptCode.Solver(vtx[1].vout[1].scriptPubKey, out whichType, out solutions))
+                    {
+                        return false; // No solutions found
+                    }
+
+                    if (whichType == txnouttype.TX_PUBKEY)
+                    {
+                        CPubKey pubkey;
+
+                        try
+                        {
+                            pubkey = new CPubKey(solutions[0]);
+                        }
+                        catch (Exception)
+                        {
+                            return false; // Error while loading public key
+                        }
+
+                        return pubkey.VerifySignature(header.Hash, signature);
+                    }
+                }
+                else
+                {
+                    // Proof-of-Work blocks have no signature
+
+                    return true;
+                }
+
+                return false;
             }
+        }
+
+        /// <summary>
+        /// Get current instance as sequence of bytes
+        /// </summary>
+        /// <returns>Byte sequence</returns>
+        public IList<byte> Bytes 
+        {
+            get
+            {
+                var r = new List<byte>();
+
+                r.AddRange(header.Bytes);
+                r.AddRange(VarInt.EncodeVarInt(vtx.LongLength)); // transactions count
 
-            r.AddRange(VarInt.EncodeVarInt(signature.LongLength));
-            r.AddRange(signature);
+                foreach (var tx in vtx)
+                {
+                    r.AddRange(tx.Bytes);
+                }
 
-            return r;
+                r.AddRange(VarInt.EncodeVarInt(signature.LongLength));
+                r.AddRange(signature);
+
+                return r;
+            }
         }
 
         public override string ToString()
         {
-            StringBuilder sb = new StringBuilder();
+            var sb = new StringBuilder();
 
             sb.AppendFormat("CBlock(\n header={0},\n", header.ToString());
 
-            foreach(CTransaction tx in vtx)
+            foreach(var tx in vtx)
+            {
+                sb.AppendFormat("{0}", tx.ToString());
+            }
+
+            if (IsProofOfStake)
             {
-                sb.AppendFormat("{0},\n", tx.ToString());
+                sb.AppendFormat(", signature={0}, signatureOK={1}\n", Interop.ToHex(signature), SignatureOK);
             }
 
-            sb.AppendFormat("signature={0})\n", Interop.ToHex(signature));
+            sb.Append(")");
             
-            // TODO
             return sb.ToString();
         }
        }