[bitcoin-tx] add rbfoptin command

jonasschnelli · jonasschnelli · commit e17853f5fb5b · 2015-12-22T13:59:16.000+01:00
diff --git a/src/bitcoin-tx.cpp b/src/bitcoin-tx.cpp
@@ -70,6 +70,7 @@ static bool AppInitRawTx(int argc, char* argv[])
         strUsage += HelpMessageOpt("in=TXID:VOUT(:SEQUENCE_NUMBER)", _("Add input to TX"));
         strUsage += HelpMessageOpt("locktime=N", _("Set TX lock time to N"));
         strUsage += HelpMessageOpt("nversion=N", _("Set TX version to N"));
+        strUsage += HelpMessageOpt("rbfoptin(=N)", _("Set RBF opt-in sequence number for input N (if not provided, opt-in all available inputs)"));
         strUsage += HelpMessageOpt("outaddr=VALUE:ADDRESS", _("Add address-based output to TX"));
         strUsage += HelpMessageOpt("outdata=[VALUE:]DATA", _("Add data-based output to TX"));
         strUsage += HelpMessageOpt("outscript=VALUE:SCRIPT", _("Add raw script output to TX"));
@@ -175,6 +176,24 @@ static void MutateTxLocktime(CMutableTransaction& tx, const string& cmdVal)
     tx.nLockTime = (unsigned int) newLocktime;
 }
 
+static void MutateTxRBFOptIn(CMutableTransaction& tx, const string& strInIdx)
+{
+    // parse requested index
+    int inIdx = atoi(strInIdx);
+    if (inIdx < 0 || inIdx >= (int)tx.vin.size()) {
+        string strErr = "Invalid TX input index '" + strInIdx + "'";
+        throw runtime_error(strErr.c_str());
+    }
+
+    // set the nSequence to MAX_INT - 2 (= RBF opt in flag)
+    int cnt = 0;
+    BOOST_FOREACH(CTxIn& txin, tx.vin) {
+        if (strInIdx == "" || cnt == inIdx)
+            txin.nSequence = std::numeric_limits<unsigned int>::max() - 2;
+        cnt++;
+    }
+}
+
 static void MutateTxAddInput(CMutableTransaction& tx, const string& strInput)
 {
     std::vector<std::string> vStrInputParts;
@@ -502,6 +521,8 @@ static void MutateTx(CMutableTransaction& tx, const string& command,
         MutateTxVersion(tx, commandVal);
     else if (command == "locktime")
         MutateTxLocktime(tx, commandVal);
+    else if (command == "rbfoptin")
+        MutateTxRBFOptIn(tx, commandVal);
 
     else if (command == "delin")
         MutateTxDelInput(tx, commandVal);
