Rename ethdm*key to encryption*key

The naming was just confusing.
This commit is contained in:
Franck Royer 2021-08-12 14:45:40 +10:00
parent a2b6947b73
commit 0bbe52ebb1
No known key found for this signature in database
GPG Key ID: A82ED75A8DFC50A4
9 changed files with 79 additions and 66 deletions

View File

@ -68,7 +68,9 @@ const useStyles = makeStyles({
function App() {
const [waku, setWaku] = useState<Waku>();
const [signer, setSigner] = useState<Signer>();
const [ethDmKeyPair, setEthDmKeyPair] = useState<KeyPair | undefined>();
const [EncryptionKeyPair, setEncryptionKeyPair] = useState<
KeyPair | undefined
>();
const [publicKeys, setPublicKeys] = useState<Map<string, string>>(new Map());
const [messages, setMessages] = useState<Message[]>([]);
const [address, setAddress] = useState<string>();
@ -110,21 +112,21 @@ function App() {
useEffect(() => {
if (!waku) return;
if (!ethDmKeyPair) return;
if (!EncryptionKeyPair) return;
waku.relay.addDecryptionKey(ethDmKeyPair.privateKey);
waku.relay.addDecryptionKey(EncryptionKeyPair.privateKey);
return function cleanUp() {
if (!waku) return;
if (!ethDmKeyPair) return;
if (!EncryptionKeyPair) return;
waku.relay.deleteDecryptionKey(ethDmKeyPair.privateKey);
waku.relay.deleteDecryptionKey(EncryptionKeyPair.privateKey);
};
}, [waku, ethDmKeyPair]);
}, [waku, EncryptionKeyPair]);
useEffect(() => {
if (!waku) return;
if (!ethDmKeyPair) return;
if (!EncryptionKeyPair) return;
if (!address) return;
const observerDirectMessage = handleDirectMessage.bind(
@ -142,7 +144,7 @@ function App() {
DirectMessageContentTopic,
]);
};
}, [waku, address, ethDmKeyPair]);
}, [waku, address, EncryptionKeyPair]);
let relayPeers = 0;
let lightPushPeers = 0;
@ -189,14 +191,14 @@ function App() {
<ConnectWallet setAddress={setAddress} setSigner={setSigner} />
</fieldset>
<fieldset>
<legend>Eth-DM Key Pair</legend>
<legend>Encryption Key Pair</legend>
<KeyPairHandling
ethDmKeyPair={ethDmKeyPair}
setEthDmKeyPair={setEthDmKeyPair}
encryptionKeyPair={EncryptionKeyPair}
setEncryptionKeyPair={setEncryptionKeyPair}
/>
<BroadcastPublicKey
signer={signer}
ethDmKeyPair={ethDmKeyPair}
EncryptionKeyPair={EncryptionKeyPair}
waku={waku}
/>
</fieldset>

View File

@ -7,20 +7,20 @@ import { Signer } from '@ethersproject/abstract-signer';
import { PublicKeyContentTopic } from './waku';
interface Props {
ethDmKeyPair: KeyPair | undefined;
EncryptionKeyPair: KeyPair | undefined;
waku: Waku | undefined;
signer: Signer | undefined;
}
export default function BroadcastPublicKey({
signer,
ethDmKeyPair,
EncryptionKeyPair,
waku,
}: Props) {
const [publicKeyMsg, setPublicKeyMsg] = useState<PublicKeyMessage>();
const broadcastPublicKey = () => {
if (!ethDmKeyPair) return;
if (!EncryptionKeyPair) return;
if (!signer) return;
if (!waku) return;
@ -35,7 +35,7 @@ export default function BroadcastPublicKey({
console.log('Failed to encode Public Key Message in Waku Message');
});
} else {
createPublicKeyMessage(signer, ethDmKeyPair.publicKey)
createPublicKeyMessage(signer, EncryptionKeyPair.publicKey)
.then((msg) => {
setPublicKeyMsg(msg);
encodePublicKeyWakuMessage(msg)
@ -64,9 +64,9 @@ export default function BroadcastPublicKey({
variant="contained"
color="primary"
onClick={broadcastPublicKey}
disabled={!ethDmKeyPair || !waku}
disabled={!EncryptionKeyPair || !waku}
>
Broadcast Eth-DM Public Key
Broadcast Encryption Public Key
</Button>
);
}

View File

@ -16,7 +16,7 @@ export interface KeyPair {
* the entropy for the EthCrypto keypair. Note that the entropy is hashed with keccak256
* to make the private key.
*/
export async function generateEthDmKeyPair(): Promise<KeyPair> {
export async function generateEncryptionKeyPair(): Promise<KeyPair> {
const privateKey = generatePrivateKey();
const publicKey = getPublicKey(privateKey);
return { privateKey, publicKey };
@ -29,25 +29,25 @@ export async function generateEthDmKeyPair(): Promise<KeyPair> {
*/
export async function createPublicKeyMessage(
web3Signer: Signer,
ethDmPublicKey: Uint8Array
encryptionPublicKey: Uint8Array
): Promise<PublicKeyMessage> {
const ethAddress = await web3Signer.getAddress();
const signature = await web3Signer.signMessage(
formatPublicKeyForSignature(ethDmPublicKey)
formatPublicKeyForSignature(encryptionPublicKey)
);
return new PublicKeyMessage({
ethDmPublicKey: ethDmPublicKey,
encryptionPublicKey: encryptionPublicKey,
ethAddress: hexToBuf(ethAddress),
signature: hexToBuf(signature),
});
}
/**
* Validate that the EthDm Public Key was signed by the holder of the given Ethereum address.
* Validate that the Encryption Public Key was signed by the holder of the given Ethereum address.
*/
export function validatePublicKeyMessage(msg: PublicKeyMessage): boolean {
const formattedMsg = formatPublicKeyForSignature(msg.ethDmPublicKey);
const formattedMsg = formatPublicKeyForSignature(msg.encryptionPublicKey);
try {
const sigAddress = ethers.utils.verifyMessage(formattedMsg, msg.signature);
return equalByteArrays(sigAddress, msg.ethAddress);
@ -63,13 +63,13 @@ export function validatePublicKeyMessage(msg: PublicKeyMessage): boolean {
/**
* Prepare Eth-Dm Public key to be signed for publication.
* The public key is set in on Object `{ ethDmPublicKey: string; }`, converted
* The public key is set in on Object `{ encryptionPublicKey: string; }`, converted
* to JSON and then hashed with Keccak256.
* The usage of the object helps ensure the signature is only used in an Eth-DM
* context.
*/
function formatPublicKeyForSignature(ethDmPublicKey: Uint8Array): string {
function formatPublicKeyForSignature(encryptionPublicKey: Uint8Array): string {
return JSON.stringify({
ethDmPublicKey: bufToHex(ethDmPublicKey),
encryptionPublicKey: bufToHex(encryptionPublicKey),
});
}

View File

@ -2,7 +2,7 @@ import { Button } from '@material-ui/core';
import { LoadKeyPair } from './LoadKeyPair';
import { SaveKeyPair } from './SaveKeyPair';
import React, { useState } from 'react';
import { generateEthDmKeyPair, KeyPair } from '../crypto';
import { generateEncryptionKeyPair, KeyPair } from '../crypto';
import { makeStyles } from '@material-ui/core/styles';
import PasswordInput from './PasswordInput';
@ -29,24 +29,24 @@ const useStyles = makeStyles({
});
export interface Props {
ethDmKeyPair: KeyPair | undefined;
setEthDmKeyPair: (keyPair: KeyPair) => void;
encryptionKeyPair: KeyPair | undefined;
setEncryptionKeyPair: (keyPair: KeyPair) => void;
}
export default function KeyPairHandling({
ethDmKeyPair,
setEthDmKeyPair,
encryptionKeyPair,
setEncryptionKeyPair,
}: Props) {
const classes = useStyles();
const [password, setPassword] = useState<string>();
const generateKeyPair = () => {
if (ethDmKeyPair) return;
if (encryptionKeyPair) return;
generateEthDmKeyPair()
generateEncryptionKeyPair()
.then((keyPair) => {
setEthDmKeyPair(keyPair);
setEncryptionKeyPair(keyPair);
})
.catch((e) => {
console.error('Failed to generate Key Pair', e);
@ -60,9 +60,9 @@ export default function KeyPairHandling({
variant="contained"
color="primary"
onClick={generateKeyPair}
disabled={!!ethDmKeyPair}
disabled={!!encryptionKeyPair}
>
Generate Eth-DM Key Pair
Generate Encryption Key Pair
</Button>
<div className={classes.storage}>
<PasswordInput
@ -72,13 +72,16 @@ export default function KeyPairHandling({
<div className={classes.loadSave}>
<div className={classes.loadSaveButton}>
<LoadKeyPair
setEthDmKeyPair={(keyPair) => setEthDmKeyPair(keyPair)}
disabled={!!ethDmKeyPair}
setEncryptionKeyPair={(keyPair) => setEncryptionKeyPair(keyPair)}
disabled={!!encryptionKeyPair}
password={password}
/>
</div>
<div className={classes.loadSaveButton}>
<SaveKeyPair ethDmKeyPair={ethDmKeyPair} password={password} />
<SaveKeyPair
EncryptionKeyPair={encryptionKeyPair}
password={password}
/>
</div>
</div>
</div>

View File

@ -4,19 +4,23 @@ import { loadKeyPairFromStorage } from './key_pair_storage';
import { KeyPair } from '../crypto';
export interface Props {
setEthDmKeyPair: (keyPair: KeyPair) => void;
setEncryptionKeyPair: (keyPair: KeyPair) => void;
disabled: boolean;
password: string | undefined;
}
export function LoadKeyPair({ password, disabled, setEthDmKeyPair }: Props) {
export function LoadKeyPair({
password,
disabled,
setEncryptionKeyPair,
}: Props) {
const loadKeyPair = () => {
if (disabled) return;
if (!password) return;
loadKeyPairFromStorage(password).then((keyPair: KeyPair | undefined) => {
if (!keyPair) return;
console.log('EthDm KeyPair loaded from storage');
setEthDmKeyPair(keyPair);
console.log('Encryption KeyPair loaded from storage');
setEncryptionKeyPair(keyPair);
});
};
@ -27,7 +31,7 @@ export function LoadKeyPair({ password, disabled, setEthDmKeyPair }: Props) {
onClick={loadKeyPair}
disabled={!password || disabled}
>
Load Eth-DM Key Pair from storage
Load Encryption Key Pair from storage
</Button>
);
}

View File

@ -4,16 +4,16 @@ import { KeyPair } from '../crypto';
import { saveKeyPairToStorage } from './key_pair_storage';
export interface Props {
ethDmKeyPair: KeyPair | undefined;
EncryptionKeyPair: KeyPair | undefined;
password: string | undefined;
}
export function SaveKeyPair({ password, ethDmKeyPair }: Props) {
export function SaveKeyPair({ password, EncryptionKeyPair }: Props) {
const saveKeyPair = () => {
if (!ethDmKeyPair) return;
if (!EncryptionKeyPair) return;
if (!password) return;
saveKeyPairToStorage(ethDmKeyPair, password).then(() => {
console.log('EthDm KeyPair saved to storage');
saveKeyPairToStorage(EncryptionKeyPair, password).then(() => {
console.log('Encryption KeyPair saved to storage');
});
};
@ -22,9 +22,9 @@ export function SaveKeyPair({ password, ethDmKeyPair }: Props) {
variant="contained"
color="primary"
onClick={saveKeyPair}
disabled={!password || !ethDmKeyPair}
disabled={!password || !EncryptionKeyPair}
>
Save Eth-DM Key Pair to storage
Save Encryption Key Pair to storage
</Button>
);
}

View File

@ -5,10 +5,10 @@ import { bufToHex, hexToBuf } from 'js-waku/lib/utils';
* Save keypair to storage, encrypted with password
*/
export async function saveKeyPairToStorage(
ethDmKeyPair: KeyPair,
EncryptionKeyPair: KeyPair,
password: string
) {
const { salt, iv, cipher } = await encryptKey(ethDmKeyPair, password);
const { salt, iv, cipher } = await encryptKey(EncryptionKeyPair, password);
const data = {
salt: bufToHex(salt),
@ -16,7 +16,7 @@ export async function saveKeyPairToStorage(
cipher: bufToHex(cipher),
};
localStorage.setItem('cipherEthDmKeyPair', JSON.stringify(data));
localStorage.setItem('cipherEncryptionKeyPair', JSON.stringify(data));
}
/**
@ -25,7 +25,7 @@ export async function saveKeyPairToStorage(
export async function loadKeyPairFromStorage(
password: string
): Promise<KeyPair | undefined> {
const str = localStorage.getItem('cipherEthDmKeyPair');
const str = localStorage.getItem('cipherEncryptionKeyPair');
if (!str) return;
const data = JSON.parse(str);
@ -71,13 +71,13 @@ function getWrapKey(keyMaterial: CryptoKey, salt: Uint8Array) {
/**
* Encrypt Eth-DM KeyPair using provided password
*/
async function encryptKey(ethDmKeyPair: KeyPair, password: string) {
async function encryptKey(encryptionKeyPair: KeyPair, password: string) {
const keyMaterial = await getKeyMaterial(password);
const salt = window.crypto.getRandomValues(new Uint8Array(16));
const wrappingKey = await getWrapKey(keyMaterial, salt);
const enc = new TextEncoder();
const encodedKeyPair = enc.encode(JSON.stringify(ethDmKeyPair));
const encodedKeyPair = enc.encode(JSON.stringify(encryptionKeyPair));
const iv = window.crypto.getRandomValues(new Uint8Array(12));
const cipher = await window.crypto.subtle.encrypt(

View File

@ -1,7 +1,7 @@
import * as protobuf from 'protobufjs/light';
export interface PublicKeyMessagePayload {
ethDmPublicKey: Uint8Array;
encryptionPublicKey: Uint8Array;
ethAddress: Uint8Array;
signature: Uint8Array;
}
@ -15,7 +15,7 @@ const Root = protobuf.Root,
*/
export class PublicKeyMessage {
private static Type = new Type('PublicKeyMessage')
.add(new Field('ethDmPublicKey', 1, 'bytes'))
.add(new Field('encryptionPublicKey', 1, 'bytes'))
.add(new Field('ethAddress', 2, 'bytes'))
.add(new Field('signature', 3, 'bytes'));
private static Root = new Root()
@ -35,15 +35,19 @@ export class PublicKeyMessage {
const payload = PublicKeyMessage.Type.decode(
bytes
) as unknown as PublicKeyMessagePayload;
if (!payload.signature || !payload.ethDmPublicKey || !payload.ethAddress) {
if (
!payload.signature ||
!payload.encryptionPublicKey ||
!payload.ethAddress
) {
console.log('Field missing on decoded Public Key Message', payload);
return;
}
return new PublicKeyMessage(payload);
}
get ethDmPublicKey(): Uint8Array {
return this.payload.ethDmPublicKey;
get encryptionPublicKey(): Uint8Array {
return this.payload.encryptionPublicKey;
}
get ethAddress(): Uint8Array {

View File

@ -41,8 +41,8 @@ export function handlePublicKeyMessage(
if (!msg.payload) return;
const publicKeyMsg = PublicKeyMessage.decode(msg.payload);
if (!publicKeyMsg) return;
const ethDmPublicKey = bufToHex(publicKeyMsg.ethDmPublicKey);
console.log(ethDmPublicKey, myAddress);
const encryptionPublicKey = bufToHex(publicKeyMsg.encryptionPublicKey);
console.log(encryptionPublicKey, myAddress);
if (myAddress && equalByteArrays(publicKeyMsg.ethAddress, myAddress)) return;
const res = validatePublicKeyMessage(publicKeyMsg);
@ -50,7 +50,7 @@ export function handlePublicKeyMessage(
if (res) {
setter((prevPks: Map<string, string>) => {
prevPks.set(bufToHex(publicKeyMsg.ethAddress), ethDmPublicKey);
prevPks.set(bufToHex(publicKeyMsg.ethAddress), encryptionPublicKey);
return new Map(prevPks);
});
}