Segwit support: parse transactions

znort987#87
dvlop · Apr 21, 2021 · 7b6f4d1 · 7b6f4d1
1 parent 81b0e4f
commit 7b6f4d1
Showing 1 changed file with 55 additions and 9 deletions.
diff --git a/parser.cpp b/parser.cpp
@@ -358,12 +358,26 @@ static void parseInputs(
     const Block   *block,
     const uint8_t *&p,
     const uint8_t *txHash
+    #if defined(BITCOIN)
+        , uint64_t* witnessNum
+    #endif
 ) {
     if(!skip) {
         startInputs(p);
     }
 
     LOAD_VARINT(nbInputs, p);
+
+    #if defined(BITCOIN) // segwit support
+    *witnessNum = 0;
+    if(nbInputs == 0)
+    {
+        SKIP(unsigned char, flag, p); // flag is always 1 in current implementation
+        LOAD_VARINT(nbRealInputs, p);
+        nbInputs = nbRealInputs;
+        *witnessNum = nbRealInputs;
+    }
+    #endif
     for(uint64_t inputIndex=0; inputIndex<nbInputs; ++inputIndex) {
         parseInput<skip>(
             block,
@@ -378,6 +392,28 @@ static void parseInputs(
     }
 }
 
+#if defined(BITCOIN)
+static void parseWitness(
+    const uint8_t* &p
+) {
+    LOAD_VARINT(num, p);
+    for(uint64_t i=0; i<num; ++i)
+    {
+        LOAD_VARINT(num2, p);
+        p += num2; // just skip witness data for now
+    }
+}
+
+
+static void parseWitnesses(
+    const uint64_t witnessNum,
+    const uint8_t* &p
+) {
+    for(uint64_t i=0; i<witnessNum; ++i)
+        parseWitness(p);
+}
+#endif
+
 template<
     bool skip
 >
@@ -409,7 +445,12 @@ static void parseTX(
             SKIP(uint32_t, nTime, p);
         #endif
 
+        #if defined(BITCOIN)
+        uint64_t witnessNum = 0;
+        parseInputs<skip>(block, p, txHash, &witnessNum);
+        #else
         parseInputs<skip>(block, p, txHash);
+        #endif
 
         Chunk *txo = 0;
         size_t txoOffset = -1;
@@ -422,6 +463,11 @@ static void parseTX(
 
         parseOutputs<skip, false>(p, txHash);
 
+        #if defined(BITCOIN)
+        if(witnessNum)
+            parseWitnesses(witnessNum, p);
+        #endif
+
         if(txo) {
             size_t txoSize = p - outputsStart;
             txo->init(
@@ -978,15 +1024,15 @@ static void findBlockFiles() {
         }
 
         auto fileSize = statBuf.st_size;
-	#if !defined(_WIN64)
-	    auto r1 = posix_fadvise(fd, 0, fileSize, POSIX_FADV_NOREUSE);
-	    if(r1<0) {
-		warning(
-		    "failed to posix_fadvise on block chain file %s",
-		    fileName.c_str()
-		);
-	    }
-	#endif
+    #if !defined(_WIN64)
+        auto r1 = posix_fadvise(fd, 0, fileSize, POSIX_FADV_NOREUSE);
+        if(r1<0) {
+        warning(
+            "failed to posix_fadvise on block chain file %s",
+            fileName.c_str()
+        );
+        }
+    #endif
 
         BlockFile blockFile;
         blockFile.fd = fd;