Merge pull request #1096 from tronprotocol/add_message

Add message
tronprotocol · Jun 24, 2018 · 07236ce · 07236ce
2 parents c50e14f + 6e3099d
commit 07236ce
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/main/java/org/tron/core/capsule/BlockCapsule.java b/src/main/java/org/tron/core/capsule/BlockCapsule.java
@@ -140,6 +140,7 @@ public BlockCapsule(long number, Sha256Hash hash, long when, ByteString witnessA
     initTxs();
   }
 
+
   public BlockCapsule(long timestamp, ByteString parentHash, long number,
       List<Transaction> transactionList) {
     // blockheader raw
@@ -248,6 +249,15 @@ public void setMerkleRoot() {
     this.block = this.block.toBuilder().setBlockHeader(
         this.block.getBlockHeader().toBuilder().setRawData(blockHeaderRaw)).build();
   }
+  /* only for genisis */
+  public void  setWitness(String witness) {
+    BlockHeader.raw blockHeaderRaw =
+        this.block.getBlockHeader().getRawData().toBuilder().setWitnessAddress(
+            ByteString.copyFrom(witness.getBytes())).build();
+
+    this.block = this.block.toBuilder().setBlockHeader(
+        this.block.getBlockHeader().toBuilder().setRawData(blockHeaderRaw)).build();
+  }
 
   public Sha256Hash getMerkleRoot() {
     return Sha256Hash.wrap(this.block.getBlockHeader().getRawData().getTxTrieRoot());

diff --git a/src/main/java/org/tron/core/capsule/utils/BlockUtil.java b/src/main/java/org/tron/core/capsule/utils/BlockUtil.java
@@ -50,7 +50,7 @@ public static BlockCapsule newGenesisBlockCapsule() {
     BlockCapsule blockCapsule = new BlockCapsule(timestamp, parentHash, number, transactionList);
 
     blockCapsule.setMerkleRoot();
-
+    blockCapsule.setWitness("A new system must allow existing systems to be linked together without requiring any central control or coordination");
     blockCapsule.generatedByMyself = true;
 
     return blockCapsule;