Craig Raw
4 years ago
6 changed files with 340 additions and 5 deletions
@ -1 +1 @@ |
|||||
Subproject commit 6b4b2529c803581b13e1604acd230dff24c839b4 |
Subproject commit ee49ddd94bdfec1e87d334f17833a63382958790 |
||||
@ -0,0 +1,286 @@ |
|||||
|
package com.sparrowwallet.sparrow.control; |
||||
|
|
||||
|
import com.google.common.eventbus.Subscribe; |
||||
|
import com.sparrowwallet.drongo.SecureString; |
||||
|
import com.sparrowwallet.drongo.address.Address; |
||||
|
import com.sparrowwallet.drongo.address.InvalidAddressException; |
||||
|
import com.sparrowwallet.drongo.crypto.ECKey; |
||||
|
import com.sparrowwallet.drongo.wallet.Keystore; |
||||
|
import com.sparrowwallet.drongo.wallet.Wallet; |
||||
|
import com.sparrowwallet.drongo.wallet.WalletNode; |
||||
|
import com.sparrowwallet.sparrow.AppController; |
||||
|
import com.sparrowwallet.sparrow.EventManager; |
||||
|
import com.sparrowwallet.sparrow.event.OpenWalletsEvent; |
||||
|
import com.sparrowwallet.sparrow.event.RequestOpenWalletsEvent; |
||||
|
import com.sparrowwallet.sparrow.event.StorageEvent; |
||||
|
import com.sparrowwallet.sparrow.event.TimedEvent; |
||||
|
import com.sparrowwallet.sparrow.io.Storage; |
||||
|
import javafx.application.Platform; |
||||
|
import javafx.scene.control.*; |
||||
|
import javafx.scene.image.Image; |
||||
|
import javafx.scene.image.ImageView; |
||||
|
import javafx.scene.layout.VBox; |
||||
|
import org.controlsfx.validation.ValidationResult; |
||||
|
import org.controlsfx.validation.ValidationSupport; |
||||
|
import org.controlsfx.validation.decoration.StyleClassValidationDecoration; |
||||
|
import org.slf4j.Logger; |
||||
|
import org.slf4j.LoggerFactory; |
||||
|
import tornadofx.control.Field; |
||||
|
import tornadofx.control.Fieldset; |
||||
|
import tornadofx.control.Form; |
||||
|
|
||||
|
import java.security.SignatureException; |
||||
|
import java.util.Arrays; |
||||
|
import java.util.Optional; |
||||
|
|
||||
|
import static com.sparrowwallet.sparrow.AppController.setStageIcon; |
||||
|
|
||||
|
public class MessageSignDialog extends Dialog<ButtonBar.ButtonData> { |
||||
|
private static final Logger log = LoggerFactory.getLogger(MessageSignDialog.class); |
||||
|
|
||||
|
private final TextField address; |
||||
|
private final TextArea message; |
||||
|
private final TextArea signature; |
||||
|
private final Wallet wallet; |
||||
|
private WalletNode walletNode; |
||||
|
|
||||
|
/** |
||||
|
* Verification only constructor |
||||
|
*/ |
||||
|
public MessageSignDialog() { |
||||
|
this(null, null); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Sign and verify with user entered address |
||||
|
* |
||||
|
* @param wallet Wallet to sign with |
||||
|
*/ |
||||
|
public MessageSignDialog(Wallet wallet) { |
||||
|
this(wallet, null); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Sign and verify with preset address |
||||
|
* |
||||
|
* @param wallet Wallet to sign with |
||||
|
* @param walletNode Wallet node to derive address from |
||||
|
*/ |
||||
|
public MessageSignDialog(Wallet wallet, WalletNode walletNode) { |
||||
|
if(wallet != null) { |
||||
|
if(wallet.getKeystores().size() != 1) { |
||||
|
throw new IllegalArgumentException("Cannot sign messages using a wallet with multiple keystores - a single key is required"); |
||||
|
} |
||||
|
if(!wallet.getKeystores().get(0).hasSeed()) { |
||||
|
throw new IllegalArgumentException("Cannot sign messages using a wallet without a seed"); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
this.wallet = wallet; |
||||
|
this.walletNode = walletNode; |
||||
|
|
||||
|
final DialogPane dialogPane = getDialogPane(); |
||||
|
dialogPane.getStylesheets().add(AppController.class.getResource("general.css").toExternalForm()); |
||||
|
AppController.setStageIcon(dialogPane.getScene().getWindow()); |
||||
|
dialogPane.setHeaderText(wallet == null ? "Verify Message" : "Sign/Verify Message"); |
||||
|
|
||||
|
Image image = new Image("image/seed.png", 50, 50, false, false); |
||||
|
if (!image.isError()) { |
||||
|
ImageView imageView = new ImageView(); |
||||
|
imageView.setSmooth(false); |
||||
|
imageView.setImage(image); |
||||
|
dialogPane.setGraphic(imageView); |
||||
|
} |
||||
|
|
||||
|
VBox vBox = new VBox(); |
||||
|
vBox.setSpacing(20); |
||||
|
|
||||
|
Form form = new Form(); |
||||
|
Fieldset fieldset = new Fieldset(); |
||||
|
fieldset.setText(""); |
||||
|
|
||||
|
Field addressField = new Field(); |
||||
|
addressField.setText("Address:"); |
||||
|
address = new TextField(); |
||||
|
address.getStyleClass().add("id"); |
||||
|
address.setEditable(walletNode == null); |
||||
|
addressField.getInputs().add(address); |
||||
|
|
||||
|
if(walletNode != null) { |
||||
|
address.setText(wallet.getAddress(walletNode).toString()); |
||||
|
} |
||||
|
|
||||
|
Field messageField = new Field(); |
||||
|
messageField.setText("Message:"); |
||||
|
message = new TextArea(); |
||||
|
message.setWrapText(true); |
||||
|
message.setPrefRowCount(10); |
||||
|
message.setStyle("-fx-pref-height: 180px"); |
||||
|
messageField.getInputs().add(message); |
||||
|
|
||||
|
Field signatureField = new Field(); |
||||
|
signatureField.setText("Signature:"); |
||||
|
signature = new TextArea(); |
||||
|
signature.getStyleClass().add("id"); |
||||
|
signature.setPrefRowCount(2); |
||||
|
signature.setStyle("-fx-pref-height: 60px"); |
||||
|
signature.setWrapText(true); |
||||
|
signatureField.getInputs().add(signature); |
||||
|
|
||||
|
fieldset.getChildren().addAll(addressField, messageField, signatureField); |
||||
|
form.getChildren().add(fieldset); |
||||
|
dialogPane.setContent(form); |
||||
|
|
||||
|
ButtonType signButtonType = new javafx.scene.control.ButtonType("Sign", ButtonBar.ButtonData.BACK_PREVIOUS); |
||||
|
ButtonType verifyButtonType = new javafx.scene.control.ButtonType("Verify", ButtonBar.ButtonData.NEXT_FORWARD); |
||||
|
ButtonType doneButtonType = new javafx.scene.control.ButtonType("Done", ButtonBar.ButtonData.OK_DONE); |
||||
|
dialogPane.getButtonTypes().addAll(signButtonType, verifyButtonType, doneButtonType); |
||||
|
|
||||
|
Button signButton = (Button)dialogPane.lookupButton(signButtonType); |
||||
|
signButton.setDisable(wallet == null); |
||||
|
signButton.setOnAction(event -> { |
||||
|
signMessage(); |
||||
|
}); |
||||
|
|
||||
|
Button verifyButton = (Button)dialogPane.lookupButton(verifyButtonType); |
||||
|
verifyButton.setDefaultButton(false); |
||||
|
verifyButton.setOnAction(event -> { |
||||
|
verifyMessage(); |
||||
|
}); |
||||
|
|
||||
|
boolean validAddress = isValidAddress(); |
||||
|
signButton.setDisable(!validAddress || (wallet == null)); |
||||
|
verifyButton.setDisable(!validAddress); |
||||
|
|
||||
|
ValidationSupport validationSupport = new ValidationSupport(); |
||||
|
Platform.runLater(() -> { |
||||
|
validationSupport.registerValidator(address, (Control c, String newValue) -> ValidationResult.fromErrorIf(c, "Invalid address", !isValidAddress())); |
||||
|
validationSupport.setValidationDecorator(new StyleClassValidationDecoration()); |
||||
|
}); |
||||
|
|
||||
|
address.textProperty().addListener((observable, oldValue, newValue) -> { |
||||
|
boolean valid = isValidAddress(); |
||||
|
signButton.setDisable(!valid || (wallet == null)); |
||||
|
verifyButton.setDisable(!valid); |
||||
|
|
||||
|
if(valid && wallet != null) { |
||||
|
try { |
||||
|
Address address = getAddress(); |
||||
|
setWalletNodeFromAddress(wallet, address); |
||||
|
} catch(InvalidAddressException e) { |
||||
|
//can't happen
|
||||
|
} |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
EventManager.get().register(this); |
||||
|
setOnCloseRequest(event -> { |
||||
|
if(ButtonBar.ButtonData.APPLY.equals(getResult())) { |
||||
|
event.consume(); |
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
EventManager.get().unregister(this); |
||||
|
}); |
||||
|
|
||||
|
setResultConverter(dialogButton -> dialogButton == signButtonType || dialogButton == verifyButtonType ? ButtonBar.ButtonData.APPLY : ButtonBar.ButtonData.OK_DONE); |
||||
|
} |
||||
|
|
||||
|
private Address getAddress()throws InvalidAddressException { |
||||
|
return Address.fromString(address.getText()); |
||||
|
} |
||||
|
|
||||
|
private boolean isValidAddress() { |
||||
|
try { |
||||
|
getAddress(); |
||||
|
return true; |
||||
|
} catch (InvalidAddressException e) { |
||||
|
return false; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void setWalletNodeFromAddress(Wallet wallet, Address address) { |
||||
|
walletNode = wallet.getWalletAddresses().get(address); |
||||
|
} |
||||
|
|
||||
|
private void signMessage() { |
||||
|
if(walletNode == null) { |
||||
|
AppController.showErrorDialog("Address not in wallet", "The provided address is not present in the currently selected wallet."); |
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
if(wallet.isEncrypted()) { |
||||
|
EventManager.get().post(new RequestOpenWalletsEvent()); |
||||
|
} else { |
||||
|
signUnencryptedKeystore(wallet); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void signUnencryptedKeystore(Wallet decryptedWallet) { |
||||
|
try { |
||||
|
//Note we can expect a single keystore due to the check above
|
||||
|
Keystore keystore = decryptedWallet.getKeystores().get(0); |
||||
|
ECKey privKey = keystore.getKey(walletNode); |
||||
|
String signatureText = privKey.signMessage(message.getText(), null); |
||||
|
signature.clear(); |
||||
|
signature.appendText(signatureText); |
||||
|
} catch(Exception e) { |
||||
|
log.error("Could not sign message", e); |
||||
|
AppController.showErrorDialog("Could not sign message", e.getMessage()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void verifyMessage() { |
||||
|
try { |
||||
|
//Find ECKey from message and signature
|
||||
|
//http://www.secg.org/download/aid-780/sec1-v2.pdf section 4.1.6
|
||||
|
|
||||
|
ECKey signedMessageKey = ECKey.signedMessageToKey(message.getText(), signature.getText()); |
||||
|
Address address = getAddress(); |
||||
|
|
||||
|
byte[] pubKeyHash = address.getOutputScriptData(); |
||||
|
byte[] signedKeyHash = signedMessageKey.getPubKeyHash(); |
||||
|
|
||||
|
if(!Arrays.equals(pubKeyHash, signedKeyHash)) { |
||||
|
throw new SignatureException("Address pubkey hash did not match signed pubkey hash"); |
||||
|
} |
||||
|
|
||||
|
Alert alert = new Alert(Alert.AlertType.INFORMATION); |
||||
|
setStageIcon(alert.getDialogPane().getScene().getWindow()); |
||||
|
alert.setTitle("Verification Succeeded"); |
||||
|
alert.setHeaderText("Verification Succeeded"); |
||||
|
alert.setContentText("The signature verified against the message."); |
||||
|
alert.showAndWait(); |
||||
|
} catch(SignatureException e) { |
||||
|
AppController.showErrorDialog("Verification failed", "The provided signature did not match the message for this address."); |
||||
|
} catch(Exception e) { |
||||
|
log.error("Could not verify message", e); |
||||
|
AppController.showErrorDialog("Could not verify message", e.getMessage()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Subscribe |
||||
|
public void openWallets(OpenWalletsEvent event) { |
||||
|
Storage storage = event.getStorage(wallet); |
||||
|
if(storage == null) { |
||||
|
throw new IllegalStateException("Wallet " + wallet + " without Storage"); |
||||
|
} |
||||
|
|
||||
|
WalletPasswordDialog dlg = new WalletPasswordDialog(WalletPasswordDialog.PasswordRequirement.LOAD); |
||||
|
Optional<SecureString> password = dlg.showAndWait(); |
||||
|
if(password.isPresent()) { |
||||
|
Storage.DecryptWalletService decryptWalletService = new Storage.DecryptWalletService(wallet.copy(), password.get()); |
||||
|
decryptWalletService.setOnSucceeded(workerStateEvent -> { |
||||
|
EventManager.get().post(new StorageEvent(storage.getWalletFile(), TimedEvent.Action.END, "Done")); |
||||
|
Wallet decryptedWallet = decryptWalletService.getValue(); |
||||
|
signUnencryptedKeystore(decryptedWallet); |
||||
|
}); |
||||
|
decryptWalletService.setOnFailed(workerStateEvent -> { |
||||
|
EventManager.get().post(new StorageEvent(storage.getWalletFile(), TimedEvent.Action.END, "Failed")); |
||||
|
AppController.showErrorDialog("Incorrect Password", decryptWalletService.getException().getMessage()); |
||||
|
}); |
||||
|
EventManager.get().post(new StorageEvent(storage.getWalletFile(), TimedEvent.Action.START, "Decrypting wallet...")); |
||||
|
decryptWalletService.start(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue