harald/opretj
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

testapp: Harden against ctlr-c

Browse source
This commit is contained in:
Harald Hoyer 2016-08-31 18:25:44 +02:00
parent 7555de6184
commit d3b1b3c0de
1 changed files with 24 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
opret-testapp/src/main/java/org/tcpid/opretj/testapp/App.java
View file

@ -21,6 +21,7 @@ import org.bitcoinj.core.TransactionOutput;
import org.bitcoinj.core.Utils; import org.bitcoinj.core.Utils;
import org.bitcoinj.script.Script; import org.bitcoinj.script.Script;
import org.bitcoinj.script.ScriptBuilder; import org.bitcoinj.script.ScriptBuilder;
import org.bitcoinj.utils.Threading;
import org.bitcoinj.wallet.KeyChain.KeyPurpose; import org.bitcoinj.wallet.KeyChain.KeyPurpose;
import org.bitcoinj.wallet.SendRequest; import org.bitcoinj.wallet.SendRequest;
import org.bitcoinj.wallet.Wallet.SendResult; import org.bitcoinj.wallet.Wallet.SendResult;
@ -31,6 +32,7 @@ import org.tcpid.opretj.OPRETWallet;
import org.tcpid.opretj.OPRETWalletAppKit; import org.tcpid.opretj.OPRETWalletAppKit;
import com.google.common.primitives.Bytes; import com.google.common.primitives.Bytes;
import com.google.common.util.concurrent.Service;
import joptsimple.OptionParser; import joptsimple.OptionParser;
import joptsimple.OptionSet; import joptsimple.OptionSet;
@ -46,7 +48,7 @@ public class App {
private static byte[] pkhash = Sha256Hash.hash(v.toBytes()); private static byte[] pkhash = Sha256Hash.hash(v.toBytes());
static byte[] revokemsg = Bytes.concat("Revoke ".getBytes(), pkhash); static byte[] revokemsg = Bytes.concat("Revoke ".getBytes(), pkhash);
public static void check(final byte[] pkhash, final byte[] sig) { public static void checkKey(final byte[] pkhash, final byte[] sig) {
logger.warn("CHECKING REVOKE PK {} - SIG {}", Utils.HEX.encode(pkhash), Utils.HEX.encode(sig)); logger.warn("CHECKING REVOKE PK {} - SIG {}", Utils.HEX.encode(pkhash), Utils.HEX.encode(sig));
if (!Arrays.equals(App.pkhash, pkhash)) { if (!Arrays.equals(App.pkhash, pkhash)) {
@ -101,7 +103,7 @@ public class App {
final OPRETECParser bs = new OPRETECParser(); final OPRETECParser bs = new OPRETECParser();
bs.addOPRETECRevokeEventListener((pkhash, sig) -> check(pkhash, sig)); bs.addOPRETECRevokeEventListener((pkhash, sig) -> checkKey(pkhash, sig));
long earliestTime; long earliestTime;
if (params.getId().equals(NetworkParameters.ID_REGTEST)) { if (params.getId().equals(NetworkParameters.ID_REGTEST)) {
@ -125,6 +127,14 @@ public class App {
// https://github.com/bitcoinj/bitcoinj/blob/master/core/src/main/java/org/bitcoinj/kits/WalletAppKit.java // https://github.com/bitcoinj/bitcoinj/blob/master/core/src/main/java/org/bitcoinj/kits/WalletAppKit.java
final OPRETWalletAppKit kit = new OPRETWalletAppKit(params, new File("."), "opretwallet" + params.getId(), bs); final OPRETWalletAppKit kit = new OPRETWalletAppKit(params, new File("."), "opretwallet" + params.getId(), bs);
kit.addListener(new Service.Listener() {
@Override
public void failed(final Service.State from, final Throwable failure) {
logger.error(failure.getMessage());
System.exit(-1);
}
}, Threading.SAME_THREAD);
// In case you want to connect with your local bitcoind tell the kit to // In case you want to connect with your local bitcoind tell the kit to
// connect to localhost. // connect to localhost.
// You must do that in reg test mode. // You must do that in reg test mode.
@ -138,7 +148,16 @@ public class App {
// introduction to Guava services: // introduction to Guava services:
// https://github.com/google/guava/wiki/ServiceExplained // https://github.com/google/guava/wiki/ServiceExplained
kit.startAsync(); kit.startAsync();
System.out.println("Please wait for the blockchain to be downloaded!");
try {
kit.awaitRunning(); kit.awaitRunning();
} catch (final Exception e) {
System.err.println("Aborting - shutting down");
// e.printStackTrace();
kit.stopAsync();
kit.awaitTerminated();
System.exit(-1);
}
final OPRETWallet wallet = kit.opretwallet(); final OPRETWallet wallet = kit.opretwallet();
@ -159,7 +178,6 @@ public class App {
final TransactionConfidence confidence = tx.getConfidence(); final TransactionConfidence confidence = tx.getConfidence();
System.out.println("new block depth: " + confidence.getDepthInBlocks()); System.out.println("new block depth: " + confidence.getDepthInBlocks());
}); });
// wallet.allowSpendingUnconfirmedTransactions(); // wallet.allowSpendingUnconfirmedTransactions();
// Ready to run. The kit syncs the blockchain and our wallet event // Ready to run. The kit syncs the blockchain and our wallet event
@ -187,7 +205,7 @@ public class App {
case 2: case 2:
System.out.println("send money to: " + receiveStr); System.out.println("send money to: " + receiveStr);
try { try {
System.out.print(executeCommand("qrencodes -t UTF8 -o - " + receiveStr)); System.out.print(executeCommand("qrencode -t UTF8 -o - " + receiveStr));
} catch (final Exception e) { } catch (final Exception e) {
; ;
} }
@ -211,7 +229,7 @@ public class App {
// manually want to stop the kit. The WalletAppKit registers a runtime // manually want to stop the kit. The WalletAppKit registers a runtime
// ShutdownHook so we actually do not need to worry about that when our // ShutdownHook so we actually do not need to worry about that when our
// application is stopping. // application is stopping.
System.out.println("shutting down again"); System.out.println("shutting down");
kit.stopAsync(); kit.stopAsync();
kit.awaitTerminated(); kit.awaitTerminated();
} }