Craig Raw
2 years ago
6 changed files with 540 additions and 72 deletions
@ -0,0 +1,270 @@ |
|||||
|
package com.sparrowwallet.sparrow.terminal.wallet; |
||||
|
|
||||
|
import com.googlecode.lanterna.TerminalPosition; |
||||
|
import com.googlecode.lanterna.TerminalSize; |
||||
|
import com.googlecode.lanterna.gui2.*; |
||||
|
import com.googlecode.lanterna.gui2.dialogs.DialogWindow; |
||||
|
import com.sparrowwallet.drongo.policy.Policy; |
||||
|
import com.sparrowwallet.drongo.policy.PolicyType; |
||||
|
import com.sparrowwallet.drongo.protocol.ScriptType; |
||||
|
import com.sparrowwallet.drongo.wallet.DeterministicSeed; |
||||
|
import com.sparrowwallet.drongo.wallet.Keystore; |
||||
|
import com.sparrowwallet.drongo.wallet.Wallet; |
||||
|
import com.sparrowwallet.sparrow.io.Bip39; |
||||
|
import com.sparrowwallet.sparrow.io.ImportException; |
||||
|
import com.sparrowwallet.sparrow.terminal.SparrowTerminal; |
||||
|
import org.slf4j.Logger; |
||||
|
import org.slf4j.LoggerFactory; |
||||
|
|
||||
|
import java.security.NoSuchAlgorithmException; |
||||
|
import java.security.SecureRandom; |
||||
|
import java.util.ArrayList; |
||||
|
import java.util.Arrays; |
||||
|
import java.util.List; |
||||
|
|
||||
|
public class Bip39Dialog extends NewWalletDialog { |
||||
|
private static final Logger log = LoggerFactory.getLogger(Bip39Dialog.class); |
||||
|
public static final int MAX_COLUMNS = 40; |
||||
|
|
||||
|
private final Bip39 importer = new Bip39(); |
||||
|
|
||||
|
private Wallet wallet; |
||||
|
|
||||
|
private final String walletName; |
||||
|
private final ComboBox<DisplayScriptType> scriptType; |
||||
|
private final TextBox seedWords; |
||||
|
private final TextBox passphrase; |
||||
|
private final Button createWallet; |
||||
|
|
||||
|
public Bip39Dialog(String walletName) { |
||||
|
super("Create BIP39 Wallet - " + walletName); |
||||
|
|
||||
|
setHints(List.of(Hint.CENTERED)); |
||||
|
|
||||
|
this.walletName = walletName; |
||||
|
|
||||
|
Panel mainPanel = new Panel(); |
||||
|
mainPanel.setLayoutManager(new GridLayout(2).setHorizontalSpacing(5).setVerticalSpacing(1)); |
||||
|
|
||||
|
mainPanel.addComponent(new EmptySpace(TerminalSize.ONE)); |
||||
|
mainPanel.addComponent(new EmptySpace(TerminalSize.ONE)); |
||||
|
|
||||
|
mainPanel.addComponent(new Label("Script type")); |
||||
|
scriptType = new ComboBox<>(); |
||||
|
mainPanel.addComponent(scriptType); |
||||
|
|
||||
|
mainPanel.addComponent(new Label("Seed words")); |
||||
|
seedWords = new TextBox(new TerminalSize(MAX_COLUMNS, 5)); |
||||
|
mainPanel.addComponent(seedWords); |
||||
|
|
||||
|
mainPanel.addComponent(new Label("Passphrase")); |
||||
|
passphrase = new TextBox(new TerminalSize(25, 1)); |
||||
|
mainPanel.addComponent(passphrase); |
||||
|
|
||||
|
Panel buttonPanel = new Panel(); |
||||
|
buttonPanel.setLayoutManager(new GridLayout(2).setHorizontalSpacing(1)); |
||||
|
buttonPanel.addComponent(new Button("Cancel", this::onCancel)); |
||||
|
createWallet = new Button("Create Wallet", this::createWallet).setLayoutData(GridLayout.createLayoutData(GridLayout.Alignment.CENTER, GridLayout.Alignment.CENTER, true, false)); |
||||
|
createWallet.setEnabled(false); |
||||
|
buttonPanel.addComponent(createWallet); |
||||
|
|
||||
|
mainPanel.addComponent(new Button("Generate New", () -> generateNew())); |
||||
|
|
||||
|
buttonPanel.setLayoutData(GridLayout.createLayoutData(GridLayout.Alignment.END, GridLayout.Alignment.CENTER,false,false)).addTo(mainPanel); |
||||
|
setComponent(mainPanel); |
||||
|
|
||||
|
ScriptType.getAddressableScriptTypes(PolicyType.SINGLE).stream().map(DisplayScriptType::new).forEach(scriptType::addItem); |
||||
|
scriptType.setSelectedItem(new DisplayScriptType(ScriptType.P2WPKH)); |
||||
|
|
||||
|
seedWords.setTextChangeListener((newText, changedByUserInteraction) -> { |
||||
|
try { |
||||
|
String[] words = newText.split("[ \n]"); |
||||
|
importer.getKeystore(scriptType.getSelectedItem().scriptType.getDefaultDerivation(), Arrays.asList(words), passphrase.getText()); |
||||
|
createWallet.setEnabled(true); |
||||
|
} catch(ImportException e) { |
||||
|
createWallet.setEnabled(false); |
||||
|
} |
||||
|
|
||||
|
if(changedByUserInteraction) { |
||||
|
List<String> lines = splitString(newText, MAX_COLUMNS); |
||||
|
String splitText = lines.stream().reduce((s1, s2) -> s1 + "\n" + s2).get(); |
||||
|
if(!newText.equals(splitText)) { |
||||
|
seedWords.setText(splitText); |
||||
|
|
||||
|
TerminalPosition pos = seedWords.getCaretPosition(); |
||||
|
if(pos.getRow() == lines.size() - 2 && pos.getColumn() == lines.get(lines.size() - 2).length()) { |
||||
|
seedWords.setCaretPosition(lines.size() - 1, lines.get(lines.size() - 1).length()); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
private void generateNew() { |
||||
|
WordNumberDialog wordNumberDialog = new WordNumberDialog(); |
||||
|
Integer numberOfWords = wordNumberDialog.showDialog(SparrowTerminal.get().getGui()); |
||||
|
|
||||
|
if(numberOfWords != null) { |
||||
|
int mnemonicSeedLength = numberOfWords * 11; |
||||
|
int entropyLength = mnemonicSeedLength - (mnemonicSeedLength/33); |
||||
|
|
||||
|
SecureRandom secureRandom; |
||||
|
try { |
||||
|
secureRandom = SecureRandom.getInstanceStrong(); |
||||
|
} catch(NoSuchAlgorithmException e) { |
||||
|
secureRandom = new SecureRandom(); |
||||
|
} |
||||
|
|
||||
|
DeterministicSeed deterministicSeed = new DeterministicSeed(secureRandom, entropyLength, ""); |
||||
|
List<String> words = deterministicSeed.getMnemonicCode(); |
||||
|
setWords(words); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private List<String> getWords() { |
||||
|
return List.of(seedWords.getText().split("[ \n]")); |
||||
|
} |
||||
|
|
||||
|
private void setWords(List<String> words) { |
||||
|
String text = words.stream().reduce((s1, s2) -> s1 + " " + s2).get(); |
||||
|
List<String> splitText = splitString(text, MAX_COLUMNS); |
||||
|
seedWords.setText(splitText.stream().reduce((s1, s2) -> s1 + "\n" + s2).get()); |
||||
|
} |
||||
|
|
||||
|
public static List<String> splitString(String stringToSplit, int maxLength) { |
||||
|
int splitLength = maxLength - 1; |
||||
|
String text = stringToSplit; |
||||
|
List<String> lines = new ArrayList<>(); |
||||
|
while (text.length() > splitLength) { |
||||
|
int spaceAt = splitLength - 1; |
||||
|
// the text is too long.
|
||||
|
// find the last space before the maxLength
|
||||
|
for (int i = splitLength - 1; i > 0; i--) { |
||||
|
if (Character.isWhitespace(text.charAt(i))) { |
||||
|
spaceAt = i; |
||||
|
break; |
||||
|
} |
||||
|
} |
||||
|
lines.add(text.substring(0, spaceAt)); |
||||
|
text = text.substring(spaceAt + 1); |
||||
|
} |
||||
|
lines.add(text); |
||||
|
return lines; |
||||
|
} |
||||
|
|
||||
|
private void createWallet() { |
||||
|
close(); |
||||
|
|
||||
|
try { |
||||
|
wallet = getWallet(); |
||||
|
discoverAndSaveWallet(wallet); |
||||
|
} catch(ImportException e) { |
||||
|
log.error("Cannot import wallet", e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private Wallet getWallet() throws ImportException { |
||||
|
Wallet wallet = new Wallet(walletName); |
||||
|
wallet.setPolicyType(PolicyType.SINGLE); |
||||
|
wallet.setScriptType(scriptType.getSelectedItem().scriptType); |
||||
|
Keystore keystore = importer.getKeystore(wallet.getScriptType().getDefaultDerivation(), getWords(), passphrase.getText()); |
||||
|
wallet.getKeystores().add(keystore); |
||||
|
wallet.setDefaultPolicy(Policy.getPolicy(PolicyType.SINGLE, wallet.getScriptType(), wallet.getKeystores(), 1)); |
||||
|
return wallet; |
||||
|
} |
||||
|
|
||||
|
private void onCancel() { |
||||
|
close(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public Wallet showDialog(WindowBasedTextGUI textGUI) { |
||||
|
super.showDialog(textGUI); |
||||
|
return wallet; |
||||
|
} |
||||
|
|
||||
|
private static final class DisplayScriptType { |
||||
|
private final ScriptType scriptType; |
||||
|
|
||||
|
public DisplayScriptType(ScriptType scriptType) { |
||||
|
this.scriptType = scriptType; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public String toString() { |
||||
|
return scriptType.getDescription(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public boolean equals(Object o) { |
||||
|
if(this == o) { |
||||
|
return true; |
||||
|
} |
||||
|
if(o == null || getClass() != o.getClass()) { |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
DisplayScriptType that = (DisplayScriptType) o; |
||||
|
|
||||
|
return scriptType == that.scriptType; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public int hashCode() { |
||||
|
return scriptType.hashCode(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private static final class WordNumberDialog extends DialogWindow { |
||||
|
ComboBox<Integer> wordCount; |
||||
|
private Integer numberOfWords; |
||||
|
|
||||
|
public WordNumberDialog() { |
||||
|
super("Generate Seed Words"); |
||||
|
|
||||
|
setHints(List.of(Hint.CENTERED)); |
||||
|
|
||||
|
Panel mainPanel = new Panel(); |
||||
|
mainPanel.setLayoutManager(new GridLayout(2).setHorizontalSpacing(5).setVerticalSpacing(1)); |
||||
|
|
||||
|
mainPanel.addComponent(new EmptySpace(TerminalSize.ONE)); |
||||
|
mainPanel.addComponent(new EmptySpace(TerminalSize.ONE)); |
||||
|
|
||||
|
mainPanel.addComponent(new Label("Number of words")); |
||||
|
wordCount = new ComboBox<>(); |
||||
|
mainPanel.addComponent(wordCount); |
||||
|
|
||||
|
wordCount.addItem(24); |
||||
|
wordCount.addItem(21); |
||||
|
wordCount.addItem(18); |
||||
|
wordCount.addItem(15); |
||||
|
wordCount.addItem(12); |
||||
|
|
||||
|
Panel buttonPanel = new Panel(); |
||||
|
buttonPanel.setLayoutManager(new GridLayout(2).setHorizontalSpacing(1)); |
||||
|
buttonPanel.addComponent(new Button("Cancel", this::onCancel)); |
||||
|
Button okButton = new Button("Ok", this::onOk).setLayoutData(GridLayout.createLayoutData(GridLayout.Alignment.CENTER, GridLayout.Alignment.CENTER, true, false)); |
||||
|
buttonPanel.addComponent(okButton); |
||||
|
|
||||
|
mainPanel.addComponent(new EmptySpace(TerminalSize.ONE)); |
||||
|
|
||||
|
buttonPanel.setLayoutData(GridLayout.createLayoutData(GridLayout.Alignment.END, GridLayout.Alignment.CENTER,false,false)).addTo(mainPanel); |
||||
|
setComponent(mainPanel); |
||||
|
} |
||||
|
|
||||
|
private void onOk() { |
||||
|
numberOfWords = wordCount.getSelectedItem(); |
||||
|
close(); |
||||
|
} |
||||
|
|
||||
|
private void onCancel() { |
||||
|
close(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public Integer showDialog(WindowBasedTextGUI textGUI) { |
||||
|
super.showDialog(textGUI); |
||||
|
return numberOfWords; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,163 @@ |
|||||
|
package com.sparrowwallet.sparrow.terminal.wallet; |
||||
|
|
||||
|
import com.googlecode.lanterna.TerminalSize; |
||||
|
import com.googlecode.lanterna.gui2.EmptySpace; |
||||
|
import com.googlecode.lanterna.gui2.Label; |
||||
|
import com.googlecode.lanterna.gui2.LinearLayout; |
||||
|
import com.googlecode.lanterna.gui2.Panel; |
||||
|
import com.googlecode.lanterna.gui2.dialogs.DialogWindow; |
||||
|
import com.googlecode.lanterna.gui2.dialogs.TextInputDialogBuilder; |
||||
|
import com.sparrowwallet.drongo.SecureString; |
||||
|
import com.sparrowwallet.drongo.crypto.ECKey; |
||||
|
import com.sparrowwallet.drongo.crypto.EncryptionType; |
||||
|
import com.sparrowwallet.drongo.crypto.Key; |
||||
|
import com.sparrowwallet.drongo.wallet.MnemonicException; |
||||
|
import com.sparrowwallet.drongo.wallet.Wallet; |
||||
|
import com.sparrowwallet.sparrow.AppServices; |
||||
|
import com.sparrowwallet.sparrow.EventManager; |
||||
|
import com.sparrowwallet.sparrow.event.StorageEvent; |
||||
|
import com.sparrowwallet.sparrow.event.TimedEvent; |
||||
|
import com.sparrowwallet.sparrow.io.Storage; |
||||
|
import com.sparrowwallet.sparrow.io.StorageException; |
||||
|
import com.sparrowwallet.sparrow.net.ElectrumServer; |
||||
|
import com.sparrowwallet.sparrow.terminal.SparrowTerminal; |
||||
|
import javafx.application.Platform; |
||||
|
import org.slf4j.Logger; |
||||
|
import org.slf4j.LoggerFactory; |
||||
|
|
||||
|
import java.io.IOException; |
||||
|
import java.util.List; |
||||
|
|
||||
|
import static com.sparrowwallet.sparrow.AppServices.showErrorDialog; |
||||
|
|
||||
|
public class NewWalletDialog extends DialogWindow { |
||||
|
private static final Logger log = LoggerFactory.getLogger(NewWalletDialog.class); |
||||
|
|
||||
|
public NewWalletDialog(String title) { |
||||
|
super(title); |
||||
|
} |
||||
|
|
||||
|
protected void discoverAndSaveWallet(Wallet wallet) { |
||||
|
if(AppServices.onlineProperty().get()) { |
||||
|
discoverAccounts(wallet); |
||||
|
} else { |
||||
|
saveWallet(wallet); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void discoverAccounts(Wallet wallet) { |
||||
|
DiscoveringDialog discoveringDialog = new DiscoveringDialog(wallet); |
||||
|
SparrowTerminal.get().getGui().addWindow(discoveringDialog); |
||||
|
|
||||
|
Platform.runLater(() -> { |
||||
|
ElectrumServer.WalletDiscoveryService walletDiscoveryService = new ElectrumServer.WalletDiscoveryService(List.of(wallet)); |
||||
|
walletDiscoveryService.setOnSucceeded(successEvent -> { |
||||
|
SparrowTerminal.get().getGuiThread().invokeLater(() -> { |
||||
|
SparrowTerminal.get().getGui().removeWindow(discoveringDialog); |
||||
|
saveWallet(wallet); |
||||
|
}); |
||||
|
}); |
||||
|
walletDiscoveryService.setOnFailed(failedEvent -> { |
||||
|
SparrowTerminal.get().getGuiThread().invokeLater(() -> { |
||||
|
SparrowTerminal.get().getGui().removeWindow(discoveringDialog); |
||||
|
saveWallet(wallet); |
||||
|
}); |
||||
|
log.error("Failed to discover accounts", failedEvent.getSource().getException()); |
||||
|
}); |
||||
|
walletDiscoveryService.start(); |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
private void saveWallet(Wallet wallet) { |
||||
|
Storage storage = new Storage(Storage.getWalletFile(wallet.getName())); |
||||
|
|
||||
|
TextInputDialogBuilder builder = new TextInputDialogBuilder().setTitle("Wallet Password"); |
||||
|
builder.setDescription(SettingsDialog.PasswordRequirement.UPDATE_NEW.getDescription()); |
||||
|
builder.setPasswordInput(true); |
||||
|
|
||||
|
String password = builder.build().showDialog(SparrowTerminal.get().getGui()); |
||||
|
if(password != null) { |
||||
|
Platform.runLater(() -> { |
||||
|
if(password.length() == 0) { |
||||
|
try { |
||||
|
storage.setEncryptionPubKey(Storage.NO_PASSWORD_KEY); |
||||
|
storage.saveWallet(wallet); |
||||
|
storage.restorePublicKeysFromSeed(wallet, null); |
||||
|
SparrowTerminal.addWallet(storage, wallet); |
||||
|
|
||||
|
for(Wallet childWallet : wallet.getChildWallets()) { |
||||
|
storage.saveWallet(childWallet); |
||||
|
storage.restorePublicKeysFromSeed(childWallet, null); |
||||
|
SparrowTerminal.addWallet(storage, childWallet); |
||||
|
} |
||||
|
|
||||
|
SparrowTerminal.get().getGuiThread().invokeLater(() -> LoadWallet.getOpeningDialog(storage, wallet).showDialog(SparrowTerminal.get().getGui())); |
||||
|
} catch(IOException | StorageException | MnemonicException e) { |
||||
|
log.error("Error saving imported wallet", e); |
||||
|
} |
||||
|
} else { |
||||
|
Storage.KeyDerivationService keyDerivationService = new Storage.KeyDerivationService(storage, new SecureString(password)); |
||||
|
keyDerivationService.setOnSucceeded(workerStateEvent -> { |
||||
|
EventManager.get().post(new StorageEvent(Storage.getWalletFile(wallet.getName()).getAbsolutePath(), TimedEvent.Action.END, "Done")); |
||||
|
ECKey encryptionFullKey = keyDerivationService.getValue(); |
||||
|
Key key = null; |
||||
|
|
||||
|
try { |
||||
|
ECKey encryptionPubKey = ECKey.fromPublicOnly(encryptionFullKey); |
||||
|
key = new Key(encryptionFullKey.getPrivKeyBytes(), storage.getKeyDeriver().getSalt(), EncryptionType.Deriver.ARGON2); |
||||
|
wallet.encrypt(key); |
||||
|
storage.setEncryptionPubKey(encryptionPubKey); |
||||
|
storage.saveWallet(wallet); |
||||
|
storage.restorePublicKeysFromSeed(wallet, key); |
||||
|
SparrowTerminal.addWallet(storage, wallet); |
||||
|
|
||||
|
for(Wallet childWallet : wallet.getChildWallets()) { |
||||
|
if(!childWallet.isNested()) { |
||||
|
childWallet.encrypt(key); |
||||
|
} |
||||
|
storage.saveWallet(childWallet); |
||||
|
storage.restorePublicKeysFromSeed(childWallet, key); |
||||
|
SparrowTerminal.addWallet(storage, childWallet); |
||||
|
} |
||||
|
|
||||
|
SparrowTerminal.get().getGuiThread().invokeLater(() -> LoadWallet.getOpeningDialog(storage, wallet).showDialog(SparrowTerminal.get().getGui())); |
||||
|
} catch(IOException | StorageException | MnemonicException e) { |
||||
|
log.error("Error saving imported wallet", e); |
||||
|
} finally { |
||||
|
encryptionFullKey.clear(); |
||||
|
if(key != null) { |
||||
|
key.clear(); |
||||
|
} |
||||
|
} |
||||
|
}); |
||||
|
keyDerivationService.setOnFailed(workerStateEvent -> { |
||||
|
EventManager.get().post(new StorageEvent(Storage.getWalletFile(wallet.getName()).getAbsolutePath(), TimedEvent.Action.END, "Failed")); |
||||
|
showErrorDialog("Error encrypting wallet", keyDerivationService.getException().getMessage()); |
||||
|
}); |
||||
|
EventManager.get().post(new StorageEvent(Storage.getWalletFile(wallet.getName()).getAbsolutePath(), TimedEvent.Action.START, "Encrypting wallet...")); |
||||
|
keyDerivationService.start(); |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private static final class DiscoveringDialog extends DialogWindow { |
||||
|
public DiscoveringDialog(Wallet wallet) { |
||||
|
super(wallet.getName()); |
||||
|
|
||||
|
setHints(List.of(Hint.CENTERED)); |
||||
|
setFixedSize(new TerminalSize(30, 5)); |
||||
|
|
||||
|
Panel mainPanel = new Panel(); |
||||
|
mainPanel.setLayoutManager(new LinearLayout()); |
||||
|
mainPanel.addComponent(new EmptySpace(), LinearLayout.createLayoutData(LinearLayout.Alignment.Beginning, LinearLayout.GrowPolicy.CanGrow)); |
||||
|
|
||||
|
Label label = new Label("Discovering Accounts..."); |
||||
|
mainPanel.addComponent(label, LinearLayout.createLayoutData(LinearLayout.Alignment.Center)); |
||||
|
|
||||
|
mainPanel.addComponent(new EmptySpace(), LinearLayout.createLayoutData(LinearLayout.Alignment.Beginning, LinearLayout.GrowPolicy.CanGrow)); |
||||
|
|
||||
|
setComponent(mainPanel); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue