56 lines
2.1 KiB
JavaScript
56 lines
2.1 KiB
JavaScript
|
"use strict";
|
||
|
|
import { ethers } from "ethers";
|
||
|
|
import scrypt from "scrypt-js";
|
||
|
|
import { version } from "./_version";
|
||
|
|
const logger = new ethers.utils.Logger(version);
|
||
|
|
let warned = false;
|
||
|
|
export class BrainWallet extends ethers.Wallet {
|
||
|
|
static _generate(username, password, legacy, progressCallback) {
|
||
|
|
if (!warned) {
|
||
|
|
logger.warn("Warning: using Brain Wallets should be considered insecure (this warning will not be repeated)");
|
||
|
|
warned = true;
|
||
|
|
}
|
||
|
|
let usernameBytes = null;
|
||
|
|
let passwordBytes = null;
|
||
|
|
if (typeof (username) === 'string') {
|
||
|
|
logger.checkNormalize();
|
||
|
|
usernameBytes = ethers.utils.toUtf8Bytes(username.normalize('NFKC'));
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
usernameBytes = ethers.utils.arrayify(username);
|
||
|
|
}
|
||
|
|
if (typeof (password) === 'string') {
|
||
|
|
logger.checkNormalize();
|
||
|
|
passwordBytes = ethers.utils.toUtf8Bytes(password.normalize('NFKC'));
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
passwordBytes = ethers.utils.arrayify(password);
|
||
|
|
}
|
||
|
|
return new Promise((resolve, reject) => {
|
||
|
|
scrypt(passwordBytes, usernameBytes, (1 << 18), 8, 1, 32, (error, progress, key) => {
|
||
|
|
if (error) {
|
||
|
|
reject(error);
|
||
|
|
}
|
||
|
|
else if (key) {
|
||
|
|
if (legacy) {
|
||
|
|
resolve(new BrainWallet(key));
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
let mnemonic = ethers.utils.entropyToMnemonic(ethers.utils.arrayify(key).slice(0, 16));
|
||
|
|
resolve(new BrainWallet(ethers.Wallet.fromMnemonic(mnemonic)));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (progressCallback) {
|
||
|
|
return progressCallback(progress);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
});
|
||
|
|
}
|
||
|
|
static generate(username, password, progressCallback) {
|
||
|
|
return BrainWallet._generate(username, password, false, progressCallback);
|
||
|
|
}
|
||
|
|
static generateLegacy(username, password, progressCallback) {
|
||
|
|
return BrainWallet._generate(username, password, true, progressCallback);
|
||
|
|
}
|
||
|
|
}
|