UNPKG

@polkadot/util-crypto

Version:
83 lines (82 loc) 3.1 kB
import { stringToU8a, u8aToU8a } from '@polkadot/util'; import { pbkdf2Encode } from '../pbkdf2/index.js'; import { randomAsU8a } from '../random/index.js'; import { sha256AsU8a } from '../sha/index.js'; import DEFAULT_WORDLIST from './wordlists/en.js'; const INVALID_MNEMONIC = 'Invalid mnemonic'; const INVALID_ENTROPY = 'Invalid entropy'; const INVALID_CHECKSUM = 'Invalid mnemonic checksum'; /** @internal */ function normalize(str) { return (str || '').normalize('NFKD'); } /** @internal */ function binaryToByte(bin) { return parseInt(bin, 2); } /** @internal */ function bytesToBinary(bytes) { return bytes.map((x) => x.toString(2).padStart(8, '0')).join(''); } /** @internal */ function deriveChecksumBits(entropyBuffer) { return bytesToBinary(Array.from(sha256AsU8a(entropyBuffer))).slice(0, (entropyBuffer.length * 8) / 32); } export function mnemonicToSeedSync(mnemonic, password) { return pbkdf2Encode(stringToU8a(normalize(mnemonic)), stringToU8a(`mnemonic${normalize(password)}`)).password; } export function mnemonicToEntropy(mnemonic, wordlist = DEFAULT_WORDLIST) { const words = normalize(mnemonic).split(' '); if (words.length % 3 !== 0) { throw new Error(INVALID_MNEMONIC); } // convert word indices to 11 bit binary strings const bits = words .map((word) => { const index = wordlist.indexOf(word); if (index === -1) { throw new Error(INVALID_MNEMONIC); } return index.toString(2).padStart(11, '0'); }) .join(''); // split the binary string into ENT/CS const dividerIndex = Math.floor(bits.length / 33) * 32; const entropyBits = bits.slice(0, dividerIndex); const checksumBits = bits.slice(dividerIndex); // calculate the checksum and compare const matched = entropyBits.match(/(.{1,8})/g); const entropyBytes = matched?.map(binaryToByte); if (!entropyBytes || (entropyBytes.length % 4 !== 0) || (entropyBytes.length < 16) || (entropyBytes.length > 32)) { throw new Error(INVALID_ENTROPY); } const entropy = u8aToU8a(entropyBytes); if (deriveChecksumBits(entropy) !== checksumBits) { throw new Error(INVALID_CHECKSUM); } return entropy; } export function entropyToMnemonic(entropy, wordlist = DEFAULT_WORDLIST) { // 128 <= ENT <= 256 if ((entropy.length % 4 !== 0) || (entropy.length < 16) || (entropy.length > 32)) { throw new Error(INVALID_ENTROPY); } const matched = `${bytesToBinary(Array.from(entropy))}${deriveChecksumBits(entropy)}`.match(/(.{1,11})/g); const mapped = matched?.map((b) => wordlist[binaryToByte(b)]); if (!mapped || (mapped.length < 12)) { throw new Error('Unable to map entropy to mnemonic'); } return mapped.join(' '); } export function generateMnemonic(numWords, wordlist) { return entropyToMnemonic(randomAsU8a((numWords / 3) * 4), wordlist); } export function validateMnemonic(mnemonic, wordlist) { try { mnemonicToEntropy(mnemonic, wordlist); } catch { return false; } return true; }