Compare commits
4 Commits
f167b704bf
...
6c781accb0
Author | SHA1 | Date | |
---|---|---|---|
6c781accb0 | |||
4fd4bb407b | |||
dc6179dcd0 | |||
ce7bd27de0 |
|
@ -1,148 +0,0 @@
|
||||||
package io.rudefox.burrow;
|
|
||||||
|
|
||||||
import com.bjdweck.bitcoin.mnemonic.Entropy;
|
|
||||||
import com.diogonunes.jcolor.AnsiFormat;
|
|
||||||
|
|
||||||
import java.util.Scanner;
|
|
||||||
import java.util.regex.Pattern;
|
|
||||||
|
|
||||||
import static com.diogonunes.jcolor.Attribute.*;
|
|
||||||
|
|
||||||
public class Dice8EntropyGenerator {
|
|
||||||
|
|
||||||
public static final AnsiFormat RED_STYLE = new AnsiFormat(WHITE_TEXT(), RED_BACK(), BOLD());
|
|
||||||
public static final AnsiFormat GREEN_STYLE = new AnsiFormat(BLACK_TEXT(), GREEN_BACK(), BOLD());
|
|
||||||
public static final AnsiFormat BLUE_STYLE = new AnsiFormat(WHITE_TEXT(), BLUE_BACK(), BOLD());
|
|
||||||
|
|
||||||
public static final int DICE_PER_ROLL = 11;
|
|
||||||
|
|
||||||
private final int targetBitsOfEntropy;
|
|
||||||
private final Scanner inputScanner;
|
|
||||||
private final boolean ansiColorOutput;
|
|
||||||
|
|
||||||
private Entropy entropy = new Entropy();
|
|
||||||
|
|
||||||
public Dice8EntropyGenerator(int targetBitsOfEntropy, Scanner inputScanner, boolean ansiColorOutput) {
|
|
||||||
|
|
||||||
this.targetBitsOfEntropy = targetBitsOfEntropy;
|
|
||||||
this.inputScanner = inputScanner;
|
|
||||||
this.ansiColorOutput = ansiColorOutput;
|
|
||||||
}
|
|
||||||
|
|
||||||
Entropy generate() {
|
|
||||||
|
|
||||||
entropy = new Entropy();
|
|
||||||
|
|
||||||
for (int rollSetCount = 0; entropy.getBitLength() < targetBitsOfEntropy; rollSetCount++)
|
|
||||||
doDiceRoll(rollSetCount);
|
|
||||||
|
|
||||||
return entropy.truncate(targetBitsOfEntropy).appendChecksum();
|
|
||||||
}
|
|
||||||
|
|
||||||
private void doDiceRoll(int currentRollSet) {
|
|
||||||
|
|
||||||
String eventString = scanNextRollSetString();
|
|
||||||
|
|
||||||
StringBuilder rollValuesLine = new StringBuilder("|");
|
|
||||||
|
|
||||||
for (int rollNumber = 0; rollNumber < DICE_PER_ROLL; rollNumber++) {
|
|
||||||
|
|
||||||
int rollValue = Integer.parseInt(eventString.charAt(rollNumber) + "");
|
|
||||||
|
|
||||||
String formatString =
|
|
||||||
rollNumber == 3 || rollNumber == 7 ? " %d |" : " %d |";
|
|
||||||
|
|
||||||
rollValuesLine.append(String.format(formatString, rollValue));
|
|
||||||
|
|
||||||
entropy = entropy.appendBits(rollValue - 1, 3);
|
|
||||||
}
|
|
||||||
|
|
||||||
System.out.printf("%n%s%n%s%n%s%n%n",
|
|
||||||
rollValuesLine,
|
|
||||||
getBitsLine(),
|
|
||||||
getSeedWordsLine(currentRollSet)
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
private String scanNextRollSetString() {
|
|
||||||
|
|
||||||
String rollSetString = "";
|
|
||||||
Pattern rollSetPattern = Pattern.compile(String.format("[1-8]{%d}", DICE_PER_ROLL));
|
|
||||||
|
|
||||||
while (!rollSetPattern.matcher(rollSetString).matches()) {
|
|
||||||
System.out.print("Input 11 x 8-sided dice rolls [1-8]: ");
|
|
||||||
rollSetString = inputScanner.next();
|
|
||||||
}
|
|
||||||
|
|
||||||
return rollSetString;
|
|
||||||
}
|
|
||||||
|
|
||||||
private StringBuilder getBitsLine() {
|
|
||||||
|
|
||||||
String entropyBitString = entropy.toString();
|
|
||||||
|
|
||||||
String currentRollSetBitString =
|
|
||||||
entropyBitString.substring(entropyBitString.length() - (DICE_PER_ROLL * 3));
|
|
||||||
|
|
||||||
StringBuilder rollSetBitsLine = new StringBuilder("|");
|
|
||||||
for (int rollSetBitIndex = 0; rollSetBitIndex < currentRollSetBitString.length(); rollSetBitIndex++) {
|
|
||||||
|
|
||||||
int entropyBitIndex = entropyBitString.length() - currentRollSetBitString.length() + rollSetBitIndex;
|
|
||||||
|
|
||||||
if (entropyBitIndex < targetBitsOfEntropy)
|
|
||||||
rollSetBitsLine.append(styleBit("" + currentRollSetBitString.charAt(rollSetBitIndex), entropyBitIndex, rollSetBitIndex));
|
|
||||||
else
|
|
||||||
rollSetBitsLine.append("-");
|
|
||||||
|
|
||||||
if (rollSetBitIndex == 32)
|
|
||||||
rollSetBitsLine.append("|");
|
|
||||||
else if (rollSetBitIndex % 3 == 2)
|
|
||||||
rollSetBitsLine.append(styleBit(" ", entropyBitIndex, rollSetBitIndex));
|
|
||||||
else if (rollSetBitIndex % 11 == 10)
|
|
||||||
rollSetBitsLine.append(" | ");
|
|
||||||
}
|
|
||||||
|
|
||||||
return rollSetBitsLine;
|
|
||||||
}
|
|
||||||
|
|
||||||
private String styleBit(String bit, int globalEntropyBitIndex, int currentRollSetBitIndex) {
|
|
||||||
|
|
||||||
if (!ansiColorOutput || globalEntropyBitIndex >= targetBitsOfEntropy)
|
|
||||||
return bit;
|
|
||||||
|
|
||||||
return getBitFormat(currentRollSetBitIndex).format(bit);
|
|
||||||
}
|
|
||||||
|
|
||||||
private AnsiFormat getBitFormat(int rollSetBitIndex) {
|
|
||||||
|
|
||||||
int wordBitIndex = rollSetBitIndex % 11;
|
|
||||||
|
|
||||||
if (wordBitIndex == 0)
|
|
||||||
return RED_STYLE;
|
|
||||||
|
|
||||||
if (wordBitIndex <= 6)
|
|
||||||
return GREEN_STYLE;
|
|
||||||
|
|
||||||
return BLUE_STYLE;
|
|
||||||
}
|
|
||||||
|
|
||||||
private String getSeedWordsLine(int rollSet) {
|
|
||||||
|
|
||||||
int baseIndex = rollSet * 3;
|
|
||||||
|
|
||||||
return String.format(
|
|
||||||
"|%-15s|%-17s|%-15s|",
|
|
||||||
getFormattedSeedWord(baseIndex),
|
|
||||||
getFormattedSeedWord(baseIndex + 1),
|
|
||||||
getFormattedSeedWord(baseIndex + 2));
|
|
||||||
}
|
|
||||||
|
|
||||||
private String getFormattedSeedWord(int index) {
|
|
||||||
|
|
||||||
int lastWordIndex = (3 * targetBitsOfEntropy / 32) - 1;
|
|
||||||
|
|
||||||
String seedWord = index != lastWordIndex ? entropy.getWord(index) : "CHECKWORD";
|
|
||||||
|
|
||||||
return String.format(" %2d. %s", index + 1, seedWord);
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -3,31 +3,43 @@ package io.rudefox.burrow;
|
||||||
import com.bjdweck.bitcoin.mnemonic.Entropy;
|
import com.bjdweck.bitcoin.mnemonic.Entropy;
|
||||||
import com.bjdweck.math.UnsignedInt;
|
import com.bjdweck.math.UnsignedInt;
|
||||||
|
|
||||||
class DiceEventBuffer {
|
class EventBuffer {
|
||||||
|
|
||||||
private final int diceBase;
|
private final int eventBase;
|
||||||
private final StringBuilder buffer;
|
private final StringBuilder buffer;
|
||||||
private final int targetBitsOfEntropy;
|
private final int targetBitsOfEntropy;
|
||||||
|
|
||||||
DiceEventBuffer(int targetBitsOfEntropy, int diceBase) {
|
EventBuffer(int targetBitsOfEntropy, int eventBase) {
|
||||||
|
|
||||||
this.diceBase = diceBase;
|
this.eventBase = eventBase;
|
||||||
this.targetBitsOfEntropy = targetBitsOfEntropy;
|
this.targetBitsOfEntropy = targetBitsOfEntropy;
|
||||||
this.buffer = new StringBuilder(getRequiredEvents());
|
this.buffer = new StringBuilder(getRequiredEvents());
|
||||||
}
|
}
|
||||||
|
|
||||||
int getRequiredEvents() {
|
int getRequiredEvents() {
|
||||||
return (int) Math.ceil(this.targetBitsOfEntropy * Math.log(2) / Math.log(diceBase));
|
return (int) Math.ceil(this.targetBitsOfEntropy * Math.log(2) / Math.log(eventBase));
|
||||||
}
|
}
|
||||||
|
|
||||||
void appendEvents(String eventString) {
|
void appendBinaryEvents(String eventString) {
|
||||||
|
for (char inChar : eventString.toCharArray())
|
||||||
|
if (inChar >= '0' && inChar <= '1' && events() < getRequiredEvents())
|
||||||
|
append(inChar);
|
||||||
|
}
|
||||||
|
|
||||||
|
void appendD6Events(String eventString) {
|
||||||
for (char inChar : eventString.toCharArray())
|
for (char inChar : eventString.toCharArray())
|
||||||
if (inChar >= '1' && inChar <= '6' && events() < getRequiredEvents())
|
if (inChar >= '1' && inChar <= '6' && events() < getRequiredEvents())
|
||||||
append((char) (inChar - 1));
|
append((char) (inChar - 1));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void appendD8Events(String eventString) {
|
||||||
|
for (char inChar : eventString.toCharArray())
|
||||||
|
if (inChar >= '1' && inChar <= '8' && events() < getRequiredEvents())
|
||||||
|
append((char) (inChar - 1));
|
||||||
|
}
|
||||||
|
|
||||||
Entropy toEntropy() {
|
Entropy toEntropy() {
|
||||||
UnsignedInt entropy = new UnsignedInt(toString(), diceBase);
|
UnsignedInt entropy = new UnsignedInt(toString(), eventBase);
|
||||||
byte[] entropyBytes = entropy.getLowestOrderBits(this.targetBitsOfEntropy).toBigEndianByteArray();
|
byte[] entropyBytes = entropy.getLowestOrderBits(this.targetBitsOfEntropy).toBigEndianByteArray();
|
||||||
return Entropy.fromRawEntropy(entropyBytes);
|
return Entropy.fromRawEntropy(entropyBytes);
|
||||||
}
|
}
|
|
@ -0,0 +1,74 @@
|
||||||
|
package io.rudefox.burrow;
|
||||||
|
|
||||||
|
import java.util.Scanner;
|
||||||
|
import java.util.regex.Pattern;
|
||||||
|
|
||||||
|
public class InteractiveBinaryEntropyGenerator extends InteractiveEntropyGenerator {
|
||||||
|
|
||||||
|
public InteractiveBinaryEntropyGenerator(int targetBitsOfEntropy, Scanner inputScanner, boolean ansiColorOutput) {
|
||||||
|
super(inputScanner, ansiColorOutput, targetBitsOfEntropy, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected String readNextEventSetString() {
|
||||||
|
|
||||||
|
StringBuilder eventSetBuffer = new StringBuilder();
|
||||||
|
int remainingEvents = eventsPerEventSet;
|
||||||
|
Pattern eventSetPattern = Pattern.compile("[01]+");
|
||||||
|
|
||||||
|
while (remainingEvents > 0) {
|
||||||
|
|
||||||
|
System.out.printf("Input %d coin tosses [0-1]: ", remainingEvents);
|
||||||
|
|
||||||
|
String nextInput = inputScanner.next();
|
||||||
|
|
||||||
|
if (!eventSetPattern.matcher(nextInput).matches()) {
|
||||||
|
System.out.printf(
|
||||||
|
"Invalid input! Enter %d-digit sequence of 1's and 0's (for example, 1011101010)" + System.lineSeparator(),
|
||||||
|
remainingEvents);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
eventSetBuffer.append(nextInput);
|
||||||
|
remainingEvents -= nextInput.length();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (remainingEvents < 0)
|
||||||
|
System.out.printf("NOTE: Discarding the %d extra coin toss(es) entered" + System.lineSeparator(), -remainingEvents);
|
||||||
|
|
||||||
|
return eventSetBuffer.substring(0, eventsPerEventSet);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected String getEventValueFormatString(int eventIndexWithinSet) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected int toZeroBasedInteger(int eventDecimalValue) {
|
||||||
|
return eventDecimalValue;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void outputEventValuesLine(StringBuilder eventValuesLine) { }
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected StringBuilder getInitialBitsLine() {
|
||||||
|
return new StringBuilder("| ");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void padBitsLine(StringBuilder eventSetBitsLine, int bitIndexWithinEventSet, int bitIndexWithinEntropy) {
|
||||||
|
if (bitIndexWithinEventSet == BITS_PER_EVENT_SET - 1)
|
||||||
|
eventSetBitsLine.append(" |");
|
||||||
|
else if (bitIndexWithinEventSet % BITS_PER_WORD == 0 || bitIndexWithinEventSet % BITS_PER_WORD == 6)
|
||||||
|
eventSetBitsLine.append(styleBit(" ", bitIndexWithinEntropy, bitIndexWithinEventSet));
|
||||||
|
else if (bitIndexWithinEventSet % BITS_PER_WORD == BITS_PER_WORD - 1)
|
||||||
|
eventSetBitsLine.append(" | ");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected String getSeedWordLineFormatString() {
|
||||||
|
return "|%-15s|%-15s|%-15s|";
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,76 @@
|
||||||
|
package io.rudefox.burrow;
|
||||||
|
|
||||||
|
import java.util.Scanner;
|
||||||
|
import java.util.regex.Pattern;
|
||||||
|
|
||||||
|
public class InteractiveDice8EntropyGenerator extends InteractiveEntropyGenerator {
|
||||||
|
|
||||||
|
public InteractiveDice8EntropyGenerator(int targetBitsOfEntropy, Scanner inputScanner, boolean ansiColorOutput) {
|
||||||
|
super(inputScanner, ansiColorOutput, targetBitsOfEntropy, 3);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected String readNextEventSetString() {
|
||||||
|
|
||||||
|
StringBuilder eventSetBuffer = new StringBuilder();
|
||||||
|
int remainingEvents = eventsPerEventSet;
|
||||||
|
Pattern eventSetPattern = Pattern.compile("[1-8]+");
|
||||||
|
|
||||||
|
while (remainingEvents > 0) {
|
||||||
|
|
||||||
|
System.out.printf("Input %d x 8-sided dice rolls [1-8]: ", remainingEvents);
|
||||||
|
|
||||||
|
String nextInput = inputScanner.next();
|
||||||
|
|
||||||
|
if (!eventSetPattern.matcher(nextInput).matches()) {
|
||||||
|
System.out.printf(
|
||||||
|
"Invalid input! Enter %d-digit sequence of digits from 1 to 8 (for example, 8765375812)" + System.lineSeparator(),
|
||||||
|
remainingEvents);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
eventSetBuffer.append(nextInput);
|
||||||
|
remainingEvents -= nextInput.length();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (remainingEvents < 0)
|
||||||
|
System.out.printf("NOTE: Discarding the %d extra dice roll(s) entered" + System.lineSeparator(), -remainingEvents);
|
||||||
|
|
||||||
|
return eventSetBuffer.substring(0, eventsPerEventSet);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected String getEventValueFormatString(int eventIndexWithinSet) {
|
||||||
|
return eventIndexWithinSet == bitsPerEvent || eventIndexWithinSet == 7 ? " %d |" : " %d |";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected int toZeroBasedInteger(int eventDecimalValue) {
|
||||||
|
return eventDecimalValue - 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void outputEventValuesLine(StringBuilder eventValuesLine) {
|
||||||
|
System.out.println(eventValuesLine);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected StringBuilder getInitialBitsLine() {
|
||||||
|
return new StringBuilder("|");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void padBitsLine(StringBuilder eventSetBitsLine, int bitIndexWithinEventSet, int bitIndexWithinEntropy) {
|
||||||
|
if (bitIndexWithinEventSet == BITS_PER_EVENT_SET - 1)
|
||||||
|
eventSetBitsLine.append("|");
|
||||||
|
else if (bitIndexWithinEventSet % bitsPerEvent == bitsPerEvent - 1)
|
||||||
|
eventSetBitsLine.append(styleBit(" ", bitIndexWithinEntropy, bitIndexWithinEventSet));
|
||||||
|
else if (bitIndexWithinEventSet % BITS_PER_WORD == BITS_PER_WORD - 1)
|
||||||
|
eventSetBitsLine.append(" | ");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected String getSeedWordLineFormatString() {
|
||||||
|
return "|%-15s|%-17s|%-15s|";
|
||||||
|
}
|
||||||
|
}
|
154
src/main/java/io/rudefox/burrow/InteractiveEntropyGenerator.java
Normal file
154
src/main/java/io/rudefox/burrow/InteractiveEntropyGenerator.java
Normal file
|
@ -0,0 +1,154 @@
|
||||||
|
package io.rudefox.burrow;
|
||||||
|
|
||||||
|
import com.bjdweck.bitcoin.mnemonic.Entropy;
|
||||||
|
import com.diogonunes.jcolor.AnsiFormat;
|
||||||
|
|
||||||
|
import java.util.Scanner;
|
||||||
|
|
||||||
|
import static com.diogonunes.jcolor.Attribute.*;
|
||||||
|
|
||||||
|
public abstract class InteractiveEntropyGenerator {
|
||||||
|
|
||||||
|
public static final int WORDS_PER_EVENT_SET = 3;
|
||||||
|
public static final int BITS_PER_WORD = 11;
|
||||||
|
public static final int BITS_PER_EVENT_SET = WORDS_PER_EVENT_SET * BITS_PER_WORD;
|
||||||
|
|
||||||
|
public static final AnsiFormat RED_STYLE = new AnsiFormat(WHITE_TEXT(), RED_BACK(), BOLD());
|
||||||
|
public static final AnsiFormat GREEN_STYLE = new AnsiFormat(BLACK_TEXT(), GREEN_BACK(), BOLD());
|
||||||
|
public static final AnsiFormat BLUE_STYLE = new AnsiFormat(WHITE_TEXT(), BLUE_BACK(), BOLD());
|
||||||
|
|
||||||
|
private Entropy entropy = new Entropy();
|
||||||
|
|
||||||
|
protected final int targetBitsOfEntropy;
|
||||||
|
protected final int bitsPerEvent;
|
||||||
|
protected final int eventsPerEventSet;
|
||||||
|
|
||||||
|
protected final Scanner inputScanner;
|
||||||
|
protected final boolean ansiColorOutput;
|
||||||
|
|
||||||
|
protected InteractiveEntropyGenerator(Scanner inputScanner, boolean ansiColorOutput, int targetBitsOfEntropy, int bitsPerEvent) {
|
||||||
|
|
||||||
|
this.targetBitsOfEntropy = targetBitsOfEntropy;
|
||||||
|
this.bitsPerEvent = bitsPerEvent;
|
||||||
|
this.eventsPerEventSet = BITS_PER_EVENT_SET / this.bitsPerEvent;
|
||||||
|
|
||||||
|
this.inputScanner = inputScanner;
|
||||||
|
this.ansiColorOutput = ansiColorOutput;
|
||||||
|
}
|
||||||
|
|
||||||
|
Entropy generate() {
|
||||||
|
|
||||||
|
entropy = new Entropy();
|
||||||
|
|
||||||
|
for (int eventSetIndex = 0; entropy.getBitLength() < targetBitsOfEntropy; eventSetIndex++) {
|
||||||
|
|
||||||
|
String eventString = readNextEventSetString();
|
||||||
|
StringBuilder eventValuesLine = processEventSetString(eventString);
|
||||||
|
|
||||||
|
System.out.println();
|
||||||
|
|
||||||
|
outputEventValuesLine(eventValuesLine);
|
||||||
|
|
||||||
|
System.out.println(getBitsLine());
|
||||||
|
System.out.println(getSeedWordsLine(eventSetIndex));
|
||||||
|
System.out.println();
|
||||||
|
}
|
||||||
|
|
||||||
|
Entropy truncatedEntropy = entropy.truncate(targetBitsOfEntropy);
|
||||||
|
return truncatedEntropy.appendChecksum();
|
||||||
|
}
|
||||||
|
|
||||||
|
private StringBuilder processEventSetString(String eventString) {
|
||||||
|
|
||||||
|
StringBuilder eventValuesLine = new StringBuilder("|");
|
||||||
|
|
||||||
|
for (int eventIndexWithinSet = 0; eventIndexWithinSet < eventsPerEventSet; eventIndexWithinSet++) {
|
||||||
|
|
||||||
|
int eventDecimalValue = Integer.parseInt(eventString.charAt(eventIndexWithinSet) + "");
|
||||||
|
|
||||||
|
eventValuesLine.append(String.format(getEventValueFormatString(eventIndexWithinSet), eventDecimalValue));
|
||||||
|
|
||||||
|
entropy = entropy.appendBits(toZeroBasedInteger(eventDecimalValue), bitsPerEvent);
|
||||||
|
}
|
||||||
|
|
||||||
|
return eventValuesLine;
|
||||||
|
}
|
||||||
|
|
||||||
|
private StringBuilder getBitsLine() {
|
||||||
|
|
||||||
|
String entropyBitString = entropy.toString();
|
||||||
|
|
||||||
|
String eventSetBitString = entropyBitString.substring(entropyBitString.length() - BITS_PER_EVENT_SET);
|
||||||
|
|
||||||
|
StringBuilder bitsLine = getInitialBitsLine();
|
||||||
|
|
||||||
|
for (int bitIndexWithinEventSet = 0; bitIndexWithinEventSet < eventSetBitString.length(); bitIndexWithinEventSet++) {
|
||||||
|
|
||||||
|
int binIndexWithinEntropy = entropyBitString.length() - eventSetBitString.length() + bitIndexWithinEventSet;
|
||||||
|
|
||||||
|
if (binIndexWithinEntropy < targetBitsOfEntropy)
|
||||||
|
bitsLine.append(styleBit("" + eventSetBitString.charAt(bitIndexWithinEventSet), binIndexWithinEntropy, bitIndexWithinEventSet));
|
||||||
|
else
|
||||||
|
bitsLine.append("-");
|
||||||
|
|
||||||
|
padBitsLine(bitsLine, bitIndexWithinEventSet, binIndexWithinEntropy);
|
||||||
|
}
|
||||||
|
|
||||||
|
return bitsLine;
|
||||||
|
}
|
||||||
|
|
||||||
|
String styleBit(String bit, int bitIndexWithinEntropy, int bitIndexWithinEventSet) {
|
||||||
|
|
||||||
|
if (!ansiColorOutput || bitIndexWithinEntropy >= targetBitsOfEntropy)
|
||||||
|
return bit;
|
||||||
|
|
||||||
|
return getBitColor(bitIndexWithinEventSet).format(bit);
|
||||||
|
}
|
||||||
|
|
||||||
|
private AnsiFormat getBitColor(int bitIndexWithinEventSet) {
|
||||||
|
|
||||||
|
int bitIndexWithinWord = bitIndexWithinEventSet % BITS_PER_WORD;
|
||||||
|
|
||||||
|
if (bitIndexWithinWord == 0)
|
||||||
|
return RED_STYLE;
|
||||||
|
|
||||||
|
if (bitIndexWithinWord <= 6)
|
||||||
|
return GREEN_STYLE;
|
||||||
|
|
||||||
|
return BLUE_STYLE;
|
||||||
|
}
|
||||||
|
|
||||||
|
private String getSeedWordsLine(int eventSetIndex) {
|
||||||
|
|
||||||
|
int firstWordIndex = eventSetIndex * WORDS_PER_EVENT_SET;
|
||||||
|
|
||||||
|
return String.format(
|
||||||
|
getSeedWordLineFormatString(),
|
||||||
|
getFormattedSeedWord(firstWordIndex),
|
||||||
|
getFormattedSeedWord(firstWordIndex + 1),
|
||||||
|
getFormattedSeedWord(firstWordIndex + 2));
|
||||||
|
}
|
||||||
|
|
||||||
|
private String getFormattedSeedWord(int wordIndex) {
|
||||||
|
|
||||||
|
int lastWordIndex = WORDS_PER_EVENT_SET * targetBitsOfEntropy / (BITS_PER_EVENT_SET - 1) - 1;
|
||||||
|
|
||||||
|
String seedWord = wordIndex != lastWordIndex ? entropy.getWord(wordIndex) : "CHECKWORD";
|
||||||
|
|
||||||
|
return String.format(" %2d. %s", wordIndex + 1, seedWord);
|
||||||
|
}
|
||||||
|
|
||||||
|
protected abstract String readNextEventSetString();
|
||||||
|
|
||||||
|
protected abstract String getEventValueFormatString(int eventIndexWithinSet);
|
||||||
|
|
||||||
|
protected abstract int toZeroBasedInteger(int eventDecimalValue);
|
||||||
|
|
||||||
|
protected abstract void outputEventValuesLine(StringBuilder eventValuesLine);
|
||||||
|
|
||||||
|
protected abstract void padBitsLine(StringBuilder eventSetBitsLine, int bitIndexWithinEventSet, int bitIndexWithinEntropy);
|
||||||
|
|
||||||
|
protected abstract StringBuilder getInitialBitsLine();
|
||||||
|
|
||||||
|
protected abstract String getSeedWordLineFormatString();
|
||||||
|
}
|
|
@ -11,14 +11,19 @@ public class MnemonicCommand implements Runnable {
|
||||||
|
|
||||||
private static final int DEFAULT_BITS_OF_ENTROPY = 256;
|
private static final int DEFAULT_BITS_OF_ENTROPY = 256;
|
||||||
|
|
||||||
private boolean isDiceEntropy() {
|
private boolean isCustomEntropy() {
|
||||||
return entropyOptions != null && (entropyOptions.isDice6Entropy || entropyOptions.isDice8Entropy);
|
return entropyOptions != null &&
|
||||||
|
(entropyOptions.isDice6Entropy || entropyOptions.isDice8Entropy || entropyOptions.isBinaryEntropy);
|
||||||
}
|
}
|
||||||
|
|
||||||
private boolean isInteractiveMode() {
|
private boolean isInteractiveMode() {
|
||||||
return entropyOptions != null && entropyOptions.eventMethod.isInteractiveMode;
|
return entropyOptions != null && entropyOptions.eventMethod.isInteractiveMode;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private boolean isBinaryInteractiveMode() {
|
||||||
|
return isInteractiveMode() && entropyOptions.isBinaryEntropy;
|
||||||
|
}
|
||||||
|
|
||||||
private boolean isDice6InteractiveMode() {
|
private boolean isDice6InteractiveMode() {
|
||||||
return isInteractiveMode() && entropyOptions.isDice6Entropy;
|
return isInteractiveMode() && entropyOptions.isDice6Entropy;
|
||||||
}
|
}
|
||||||
|
@ -32,6 +37,10 @@ public class MnemonicCommand implements Runnable {
|
||||||
|
|
||||||
static class EntropyOptions {
|
static class EntropyOptions {
|
||||||
|
|
||||||
|
@CommandLine.Option(names = {"-2", "--binary-entropy"},
|
||||||
|
description = "use a coin or other binary entropy source")
|
||||||
|
boolean isBinaryEntropy;
|
||||||
|
|
||||||
@CommandLine.Option(names = {"-6", "--dice6-entropy"},
|
@CommandLine.Option(names = {"-6", "--dice6-entropy"},
|
||||||
description = "use 6-sided dice entropy source")
|
description = "use 6-sided dice entropy source")
|
||||||
boolean isDice6Entropy;
|
boolean isDice6Entropy;
|
||||||
|
@ -45,7 +54,7 @@ public class MnemonicCommand implements Runnable {
|
||||||
|
|
||||||
static class EventMethod {
|
static class EventMethod {
|
||||||
|
|
||||||
@CommandLine.Option(names = {"-e", "--events"}, paramLabel = "[1-6]{100}|[1-8]{86}",
|
@CommandLine.Option(names = {"-e", "--events"}, paramLabel = "[0-1]{256} | [1-6]{100} | [1-8]{86}",
|
||||||
description = "string representing events from entropy source",
|
description = "string representing events from entropy source",
|
||||||
required = true)
|
required = true)
|
||||||
String getEventString;
|
String getEventString;
|
||||||
|
@ -71,50 +80,58 @@ public class MnemonicCommand implements Runnable {
|
||||||
|
|
||||||
Entropy getEntropy() {
|
Entropy getEntropy() {
|
||||||
|
|
||||||
if (!isDiceEntropy())
|
if (!isCustomEntropy())
|
||||||
return getGeneratedEntropy();
|
return generateEntropy();
|
||||||
|
|
||||||
DiceEventBuffer diceEventBuffer;
|
if (isBinaryInteractiveMode())
|
||||||
|
return new InteractiveBinaryEntropyGenerator(targetBitsOfEntropy, new Scanner(System.in), CommandLine.Help.Ansi.AUTO.enabled()).generate();
|
||||||
if (entropyOptions.isDice6Entropy)
|
|
||||||
diceEventBuffer = new DiceEventBuffer(this.targetBitsOfEntropy, 6);
|
|
||||||
else
|
|
||||||
diceEventBuffer = new DiceEventBuffer(this.targetBitsOfEntropy, 8);
|
|
||||||
|
|
||||||
if (isDice6InteractiveMode())
|
if (isDice6InteractiveMode())
|
||||||
return getDice6EntropyInteractive(diceEventBuffer);
|
return getDice6EntropyInteractive(new EventBuffer(this.targetBitsOfEntropy, 6));
|
||||||
|
|
||||||
if (isDice8InteractiveMode())
|
if (isDice8InteractiveMode())
|
||||||
return new Dice8EntropyGenerator(targetBitsOfEntropy, new Scanner(System.in), CommandLine.Help.Ansi.AUTO.enabled()).generate();
|
return new InteractiveDice8EntropyGenerator(targetBitsOfEntropy, new Scanner(System.in), CommandLine.Help.Ansi.AUTO.enabled()).generate();
|
||||||
|
|
||||||
diceEventBuffer.appendEvents(entropyOptions.eventMethod.getEventString);
|
EventBuffer eventBuffer;
|
||||||
return diceEventBuffer.toEntropy();
|
|
||||||
|
if (entropyOptions.isBinaryEntropy) {
|
||||||
|
eventBuffer = new EventBuffer(this.targetBitsOfEntropy, 2);
|
||||||
|
eventBuffer.appendBinaryEvents(entropyOptions.eventMethod.getEventString);
|
||||||
|
} else if (entropyOptions.isDice6Entropy) {
|
||||||
|
eventBuffer = new EventBuffer(this.targetBitsOfEntropy, 6);
|
||||||
|
eventBuffer.appendD6Events(entropyOptions.eventMethod.getEventString);
|
||||||
|
} else {
|
||||||
|
eventBuffer = new EventBuffer(this.targetBitsOfEntropy, 8);
|
||||||
|
eventBuffer.appendD8Events(entropyOptions.eventMethod.getEventString);
|
||||||
}
|
}
|
||||||
|
|
||||||
private Entropy getDice6EntropyInteractive(DiceEventBuffer diceEventBuffer) {
|
return eventBuffer.toEntropy();
|
||||||
|
}
|
||||||
|
|
||||||
int requiredEvents = diceEventBuffer.getRequiredEvents();
|
private Entropy getDice6EntropyInteractive(EventBuffer eventBuffer) {
|
||||||
|
|
||||||
|
int requiredEvents = eventBuffer.getRequiredEvents();
|
||||||
|
|
||||||
Scanner scanner = new Scanner(System.in);
|
Scanner scanner = new Scanner(System.in);
|
||||||
|
|
||||||
boolean firstIteration = true;
|
boolean firstIteration = true;
|
||||||
|
|
||||||
while (diceEventBuffer.events() < requiredEvents) {
|
while (eventBuffer.events() < requiredEvents) {
|
||||||
|
|
||||||
int remainingEvents = requiredEvents - diceEventBuffer.events();
|
int remainingEvents = requiredEvents - eventBuffer.events();
|
||||||
System.out.print(String.format("Input %d %sdice rolls [1-6]: ", remainingEvents, (firstIteration ? "" : "more ")));
|
System.out.printf("Input %d %sdice rolls [1-6]: ", remainingEvents, (firstIteration ? "" : "more "));
|
||||||
|
|
||||||
String inputString = scanner.next();
|
String inputString = scanner.next();
|
||||||
|
|
||||||
diceEventBuffer.appendEvents(inputString);
|
eventBuffer.appendD6Events(inputString);
|
||||||
|
|
||||||
firstIteration = false;
|
firstIteration = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
return diceEventBuffer.toEntropy();
|
return eventBuffer.toEntropy();
|
||||||
}
|
}
|
||||||
|
|
||||||
private Entropy getGeneratedEntropy() {
|
private Entropy generateEntropy() {
|
||||||
|
|
||||||
SecureRandom random = new SecureRandom();
|
SecureRandom random = new SecureRandom();
|
||||||
int byteLength = targetBitsOfEntropy / 8;
|
int byteLength = targetBitsOfEntropy / 8;
|
||||||
|
|
62
src/test/java/io/rudefox/burrow/coin_entropy_tests.java
Normal file
62
src/test/java/io/rudefox/burrow/coin_entropy_tests.java
Normal file
|
@ -0,0 +1,62 @@
|
||||||
|
package io.rudefox.burrow;
|
||||||
|
|
||||||
|
import com.bjdweck.test.CliTestFixture;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
import java.io.UnsupportedEncodingException;
|
||||||
|
|
||||||
|
class coin_entropy_tests extends CliTestFixture {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void with_arguments_interactive_coin_should_generate_mnemonic_sentence() throws UnsupportedEncodingException {
|
||||||
|
|
||||||
|
System.setProperty("picocli.ansi", "false");
|
||||||
|
|
||||||
|
withArgs("mnemonic -i2 --bits 128");
|
||||||
|
|
||||||
|
expectedOutput("Input 33 coin tosses [0-1]: ");
|
||||||
|
provideInput("000001010011100101110111000001010" + EOL);
|
||||||
|
|
||||||
|
expectedOutput(EOL);
|
||||||
|
expectedOutput("| 0 000010 1001 | 1 100101 1101 | 1 100000 1010 |" + EOL);
|
||||||
|
expectedOutput("| 1. ahead | 2. slight | 3. scout |" + EOL);
|
||||||
|
expectedOutput(EOL);
|
||||||
|
|
||||||
|
expectedOutput("Input 33 coin tosses [0-1]: ");
|
||||||
|
provideInput("000001010011100101110111000001010" + EOL);
|
||||||
|
|
||||||
|
expectedOutput(EOL);
|
||||||
|
expectedOutput("| 0 000010 1001 | 1 100101 1101 | 1 100000 1010 |" + EOL);
|
||||||
|
expectedOutput("| 4. ahead | 5. slight | 6. scout |" + EOL);
|
||||||
|
expectedOutput(EOL);
|
||||||
|
|
||||||
|
expectedOutput("Input 33 coin tosses [0-1]: ");
|
||||||
|
provideInput("000001010011100101110111000001010" + EOL);
|
||||||
|
|
||||||
|
expectedOutput(EOL);
|
||||||
|
expectedOutput("| 0 000010 1001 | 1 100101 1101 | 1 100000 1010 |" + EOL);
|
||||||
|
expectedOutput("| 7. ahead | 8. slight | 9. scout |" + EOL);
|
||||||
|
expectedOutput(EOL);
|
||||||
|
|
||||||
|
expectedOutput("Input 33 coin tosses [0-1]: ");
|
||||||
|
provideInput("000001010011100101110111000001010" + EOL);
|
||||||
|
|
||||||
|
expectedOutput(EOL);
|
||||||
|
expectedOutput("| 0 000010 1001 | 1 100101 1101 | 1 100000 ---- |" + EOL);
|
||||||
|
expectedOutput("| 10. ahead | 11. slight | 12. CHECKWORD |" + EOL);
|
||||||
|
expectedOutput(EOL);
|
||||||
|
|
||||||
|
expectedOutput("ahead slight scout ahead slight scout ahead slight scout ahead slight scan" + EOL);
|
||||||
|
|
||||||
|
doMain();
|
||||||
|
|
||||||
|
verifyOutput();
|
||||||
|
}
|
||||||
|
|
||||||
|
private void doMain() {
|
||||||
|
|
||||||
|
setInput();
|
||||||
|
|
||||||
|
RudefoxBurrow.main(getArgs());
|
||||||
|
}
|
||||||
|
}
|
|
@ -49,6 +49,18 @@ class mnemonic_command_tests extends CliTestFixture {
|
||||||
verifyOutput();
|
verifyOutput();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void with_arguments_non_interactive_binary_entropy_should_generate_mnemonic_sentence() throws UnsupportedEncodingException {
|
||||||
|
|
||||||
|
withArgs("mnemonic --binary-entropy --events 10100101010010101010101000101010101001010100101010101010001010101010010101001010101010100010101010100101010010101010101000101010 --bits 128");
|
||||||
|
|
||||||
|
expectedOutput("pipe fetch melt enhance primary best news fetch click clean pride feed" + EOL);
|
||||||
|
|
||||||
|
doMain();
|
||||||
|
|
||||||
|
verifyOutput();
|
||||||
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
void with_arguments_without_interactive_dice_should_generate_mnemonic_sentence() throws UnsupportedEncodingException {
|
void with_arguments_without_interactive_dice_should_generate_mnemonic_sentence() throws UnsupportedEncodingException {
|
||||||
|
|
||||||
|
@ -62,6 +74,18 @@ class mnemonic_command_tests extends CliTestFixture {
|
||||||
verifyOutput();
|
verifyOutput();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void with_arguments_without_interactive_d8_dice_should_generate_mnemonic_sentence() throws UnsupportedEncodingException {
|
||||||
|
|
||||||
|
withArgs("mnemonic --dice8-entropy --events 23438688738737812541514768125415147632873218 --bits 128");
|
||||||
|
|
||||||
|
expectedOutput("fashion wave tourist notice alert marine vote alert marine vocal dish aware" + EOL);
|
||||||
|
|
||||||
|
doMain();
|
||||||
|
|
||||||
|
verifyOutput();
|
||||||
|
}
|
||||||
|
|
||||||
private void doMain() {
|
private void doMain() {
|
||||||
|
|
||||||
setInput();
|
setInput();
|
||||||
|
|
Loading…
Reference in New Issue
Block a user