claim/bytom/protocolbc/types/block_header.go

   1 package types
   2
   3 import (
   4         "encoding/hex"
   5         "fmt"
   6         "io"
   7         "time"
   8
   9         "github.com/vapor/encoding/blockchain"
  10         "github.com/vapor/encoding/bufpool"
  11         "github.com/vapor/errors"
  12         "github.com/vapor/protocol/bc"
  13 )
  14
  15 // BlockHeader defines information about a block and is used in the Bytom
  16 type BlockHeader struct {
  17         Version           uint64  // The version of the block.
  18         Height            uint64  // The height of the block.
  19         PreviousBlockHash bc.Hash // The hash of the previous block.
  20         Timestamp         uint64  // The time of the block in seconds.
  21         Nonce             uint64  // Nonce used to generate the block.
  22         Bits              uint64  // Difficulty target for the block.
  23         BlockCommitment
  24 }
  25
  26 // Time returns the time represented by the Timestamp in block header.
  27 func (bh *BlockHeader) Time() time.Time {
  28         return time.Unix(int64(bh.Timestamp), 0).UTC()
  29 }
  30
  31 // Hash returns complete hash of the block header.
  32 func (bh *BlockHeader) Hash() bc.Hash {
  33         h, _ := mapBlockHeader(bh)
  34         return h
  35 }
  36
  37 // MarshalText fulfills the json.Marshaler interface. This guarantees that
  38 // block headers will get deserialized correctly when being parsed from HTTP
  39 // requests.
  40 func (bh *BlockHeader) MarshalText() ([]byte, error) {
  41         buf := bufpool.Get()
  42         defer bufpool.Put(buf)
  43
  44         if _, err := bh.WriteTo(buf); err != nil {
  45                 return nil, err
  46         }
  47
  48         enc := make([]byte, hex.EncodedLen(buf.Len()))
  49         hex.Encode(enc, buf.Bytes())
  50         return enc, nil
  51 }
  52
  53 // UnmarshalText fulfills the encoding.TextUnmarshaler interface.
  54 func (bh *BlockHeader) UnmarshalText(text []byte) error {
  55         decoded := make([]byte, hex.DecodedLen(len(text)))
  56         if _, err := hex.Decode(decoded, text); err != nil {
  57                 return err
  58         }
  59
  60         _, err := bh.readFrom(blockchain.NewReader(decoded))
  61         return err
  62 }
  63
  64 func (bh *BlockHeader) readFrom(r *blockchain.Reader) (serflag uint8, err error) {
  65         var serflags [1]byte
  66         io.ReadFull(r, serflags[:])
  67         serflag = serflags[0]
  68         switch serflag {
  69         case SerBlockHeader, SerBlockFull:
  70         default:
  71                 return 0, fmt.Errorf("unsupported serialization flags 0x%x", serflags)
  72         }
  73
  74         if bh.Version, err = blockchain.ReadVarint63(r); err != nil {
  75                 return 0, err
  76         }
  77         if bh.Height, err = blockchain.ReadVarint63(r); err != nil {
  78                 return 0, err
  79         }
  80         if _, err = bh.PreviousBlockHash.ReadFrom(r); err != nil {
  81                 return 0, err
  82         }
  83         if bh.Timestamp, err = blockchain.ReadVarint63(r); err != nil {
  84                 return 0, err
  85         }
  86         if _, err = blockchain.ReadExtensibleString(r, bh.BlockCommitment.readFrom); err != nil {
  87                 return 0, err
  88         }
  89         if bh.Nonce, err = blockchain.ReadVarint63(r); err != nil {
  90                 return 0, err
  91         }
  92         if bh.Bits, err = blockchain.ReadVarint63(r); err != nil {
  93                 return 0, err
  94         }
  95         return
  96 }
  97
  98 // WriteTo writes the block header to the input io.Writer
  99 func (bh *BlockHeader) WriteTo(w io.Writer) (int64, error) {
 100         ew := errors.NewWriter(w)
 101         if err := bh.writeTo(ew, SerBlockHeader); err != nil {
 102                 return 0, err
 103         }
 104         return ew.Written(), ew.Err()
 105 }
 106
 107 func (bh *BlockHeader) writeTo(w io.Writer, serflags uint8) (err error) {
 108         w.Write([]byte{serflags})
 109         if _, err = blockchain.WriteVarint63(w, bh.Version); err != nil {
 110                 return err
 111         }
 112         if _, err = blockchain.WriteVarint63(w, bh.Height); err != nil {
 113                 return err
 114         }
 115         if _, err = bh.PreviousBlockHash.WriteTo(w); err != nil {
 116                 return err
 117         }
 118         if _, err = blockchain.WriteVarint63(w, bh.Timestamp); err != nil {
 119                 return err
 120         }
 121         if _, err = blockchain.WriteExtensibleString(w, nil, bh.BlockCommitment.writeTo); err != nil {
 122                 return err
 123         }
 124         if _, err = blockchain.WriteVarint63(w, bh.Nonce); err != nil {
 125                 return err
 126         }
 127         if _, err = blockchain.WriteVarint63(w, bh.Bits); err != nil {
 128                 return err
 129         }
 130         return nil
 131 }