feat: modified bitcoin_transaction_script to infer flags

keep-starknet-strange · Nov 3, 2024 · faf2752 · faf2752
1 parent d509dfe
commit faf2752
Showing 1 changed file with 111 additions and 31 deletions.
diff --git a/scripts/run_bitcoin_transaction.sh b/scripts/run_bitcoin_transaction.sh
@@ -5,6 +5,23 @@
 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 BASE_DIR=$SCRIPT_DIR/..
 
+# BIP activation heights (mainnet)
+BIP_16_BLOCK_HEIGHT=173805  # P2SH
+BIP_66_BLOCK_HEIGHT=363725  # Strict DER signatures
+BIP_65_BLOCK_HEIGHT=388381  # CHECKLOCKTIMEVERIFY
+BIP_112_BLOCK_HEIGHT=419328 # CHECKSEQUENCEVERIFY
+BIP_141_BLOCK_HEIGHT=481824 # SegWit
+BIP_341_BLOCK_HEIGHT=709632 # Taproot
+
+# Script flags values
+SCRIPT_BIP16=1              # (1 << 0)
+SCRIPT_VERIFY_DER_SIG=2     # (1 << 1)
+SCRIPT_VERIFY_CLTV=4        # (1 << 2)
+SCRIPT_VERIFY_CSV=8         # (1 << 3)
+SCRIPT_VERIFY_WITNESS=16    # (1 << 4)
+SCRIPT_STRICT_MULTISIG=32   # (1 << 5)
+SCRIPT_VERIFY_TAPROOT=64    # (1 << 6)
+
 TXID=$1
 
 RPC_API="https://bitcoin-mainnet.public.blastapi.io"
@@ -17,7 +34,7 @@ RPC_API="https://bitcoin-mainnet.public.blastapi.io"
 #    pub pubkey_script: ByteArray,
 #    pub block_height: i32,
 #    // TODO: flags?
-#} 
+#}
 
 # Fetch the transaction
 RES=$(curl -s -X POST -H "Content-Type: application/json" -d "{\"jsonrpc\":\"1.0\",\"id\":0,\"method\":\"getrawtransaction\",\"params\":[\"$TXID\", true]}" $RPC_API)
@@ -34,38 +51,101 @@ RAW_TX_INPUT=$(sed 's/^\[\(.*\)\]$/\1/' <<< $RAW_TX_TEXT)
 VINS=$(echo $RES | jq '.result.vin')
 # echo "VINS: $VINS"
 
-BLOCK_HEIGHT=0 # TODO?
+# Get block hash from transaction
+BLOCK_HASH=$(echo $RES | jq -r '.result.blockhash')
+# Fetch block info to get height
+BLOCK_INFO=$(curl -s -X POST -H "Content-Type: application/json" -d "{\"jsonrpc\":\"1.0\",\"id\":0,\"method\":\"getblock\",\"params\":[\"$BLOCK_HASH\"]}" $RPC_API)
+BLOCK_HEIGHT=$(echo $BLOCK_INFO | jq -r '.result.height')
+echo "BLOCK_HEIGHT: $BLOCK_HEIGHT"
+BLOCK_VERSION=$(echo $BLOCK_INFO | jq -r '.result.version')
+
+# Check if this is a coinbase transaction
+IS_COINBASE=false
+VIN_TXID=$(echo $RES | jq -r '.result.vin[0].txid')
+if [ "$VIN_TXID" = "null" ] || [ "$VIN_TXID" = "0000000000000000000000000000000000000000000000000000000000000000" ]; then
+    IS_COINBASE=true
+    echo "Detected coinbase transaction"
+    SCRIPT_FLAGS=0  # No script verification needed for coinbase
+else
+    # Calculate script flags based on block height and version for normal transactions
+    SCRIPT_FLAGS=0
+
+    # BIP16 (P2SH)
+    if [ $BLOCK_HEIGHT -ge $BIP_16_BLOCK_HEIGHT ]; then
+        SCRIPT_FLAGS=$((SCRIPT_FLAGS | SCRIPT_BIP16))
+    fi
+
+    # BIP66 (Strict DER signatures)
+    if [ $BLOCK_VERSION -ge 3 ] && [ $BLOCK_HEIGHT -ge $BIP_66_BLOCK_HEIGHT ]; then
+        SCRIPT_FLAGS=$((SCRIPT_FLAGS | SCRIPT_VERIFY_DER_SIG))
+    fi
+
+    # BIP65 (CHECKLOCKTIMEVERIFY)
+    if [ $BLOCK_VERSION -ge 4 ] && [ $BLOCK_HEIGHT -ge $BIP_65_BLOCK_HEIGHT ]; then
+        SCRIPT_FLAGS=$((SCRIPT_FLAGS | SCRIPT_VERIFY_CLTV))
+    fi
+
+    # BIP112 (CHECKSEQUENCEVERIFY)
+    if [ $BLOCK_HEIGHT -ge $BIP_112_BLOCK_HEIGHT ]; then
+        SCRIPT_FLAGS=$((SCRIPT_FLAGS | SCRIPT_VERIFY_CSV))
+    fi
+
+    # BIP141 (SegWit)
+    if [ $BLOCK_HEIGHT -ge $BIP_141_BLOCK_HEIGHT ]; then
+        SCRIPT_FLAGS=$((SCRIPT_FLAGS | SCRIPT_VERIFY_WITNESS))
+        SCRIPT_FLAGS=$((SCRIPT_FLAGS | SCRIPT_STRICT_MULTISIG))
+    fi
+
+    # BIP341 (Taproot)
+    if [ $BLOCK_HEIGHT -ge $BIP_341_BLOCK_HEIGHT ]; then
+        SCRIPT_FLAGS=$((SCRIPT_FLAGS | SCRIPT_VERIFY_TAPROOT))
+    fi
+fi
+
+echo "Transaction type: $([ "$IS_COINBASE" = true ] && echo 'Coinbase' || echo 'Regular')"
+echo "Block height: $BLOCK_HEIGHT"
+echo "Script flags: $SCRIPT_FLAGS"
+echo "UTXO construction: $UTXOS"
+
 AMOUNT=0 # TODO?
 UTXOS=""
-for vin in $(echo $VINS | jq -r '.[] | @base64'); do
-    _jq() {
-     echo ${vin} | base64 --decode | jq -r ${1}
-    }
-
-    TXID=$(echo $(_jq '.txid'))
-    VOUT=$(echo $(_jq '.vout'))
-
-    # Fetch the transaction
-    RES=$(curl -s -X POST -H "Content-Type: application/json" -d "{\"jsonrpc\":\"1.0\",\"id\":0,\"method\":\"getrawtransaction\",\"params\":[\"$TXID\", true]}" $RPC_API)
-
-    # AMOUNT=$(echo $RES | jq ".result.vout[$VOUT].value")
-    # echo "AMOUNT: $AMOUNT"
-
-    PUBKEY_SCRIPT=$(echo $RES | jq ".result.vout[$VOUT].scriptPubKey.hex" | tr -d '"')
-    PUBKEY_SCRIPT="0x$PUBKEY_SCRIPT"
-    # echo "PUBKEY_SCRIPT: $PUBKEY_SCRIPT"
-
-    PUBKEY_SCRIPT_TEXT=$($SCRIPT_DIR/text_to_byte_array.sh $PUBKEY_SCRIPT)
-    PUBKEY_SCRIPT_INPUT=$(sed 's/^\[\(.*\)\]$/\1/' <<< $PUBKEY_SCRIPT_TEXT)
-    # echo "PUBKEY_SCRIPT_INPUT: $PUBKEY_SCRIPT_INPUT"
-
-    # Construct UTXO
-    UTXO="{\"amount\":$AMOUNT,\"pubkey_script\":\"$PUBKEY_SCRIPT\",\"block_height\":$BLOCK_HEIGHT}"
-    # echo "UTXO: $UTXO"
-
-    UTXOS="$UTXOS$AMOUNT,$PUBKEY_SCRIPT_INPUT,$BLOCK_HEIGHT,"
-done
-UTXOS=$(sed 's/,$//' <<< $UTXOS)
+if [ "$IS_COINBASE" = true ]; then
+    echo "Setting up coinbase UTXO"
+    # Get the coinbase output amount
+    COINBASE_AMOUNT=$(echo $RES | jq '.result.vout[0].value * 100000000 | floor')
+    # Get the output script
+    COINBASE_SCRIPT=$(echo $RES | jq -r '.result.vout[0].scriptPubKey.hex')
+    COINBASE_SCRIPT="0x$COINBASE_SCRIPT"
+
+    SCRIPT_TEXT=$($SCRIPT_DIR/text_to_byte_array.sh $COINBASE_SCRIPT)
+    SCRIPT_INPUT=$(sed 's/^\[\(.*\)\]$/\1/' <<< $SCRIPT_TEXT)
+
+    # Construct coinbase UTXO with proper format
+    UTXOS="$COINBASE_AMOUNT,$SCRIPT_INPUT,$BLOCK_HEIGHT"
+else
+    UTXOS=""
+    for vin in $(echo $VINS | jq -r '.[] | @base64'); do
+            _jq() {
+                echo ${vin} | base64 --decode | jq -r ${1}
+            }
+
+            TXID=$(echo $(_jq '.txid'))
+            VOUT=$(echo $(_jq '.vout'))
+
+            # Fetch the transaction
+            RES=$(curl -s -X POST -H "Content-Type: application/json" -d "{\"jsonrpc\":\"1.0\",\"id\":0,\"method\":\"getrawtransaction\",\"params\":[\"$TXID\", true]}" $RPC_API)
+
+            AMOUNT=$(echo $RES | jq ".result.vout[$VOUT].value * 100000000 | floor")
+            PUBKEY_SCRIPT=$(echo $RES | jq ".result.vout[$VOUT].scriptPubKey.hex" | tr -d '"')
+            PUBKEY_SCRIPT="0x$PUBKEY_SCRIPT"
+
+            PUBKEY_SCRIPT_TEXT=$($SCRIPT_DIR/text_to_byte_array.sh $PUBKEY_SCRIPT)
+            PUBKEY_SCRIPT_INPUT=$(sed 's/^\[\(.*\)\]$/\1/' <<< $PUBKEY_SCRIPT_TEXT)
+
+            UTXOS="$UTXOS$AMOUNT,$PUBKEY_SCRIPT_INPUT,$BLOCK_HEIGHT,"
+        done
+    UTXOS=$(sed 's/,$//' <<< $UTXOS)
+fi
 
 JOINED_INPUT="[$RAW_TX_INPUT,[$UTXOS]]"
 # echo "JOINED_INPUT: $JOINED_INPUT"