mirror of
https://github.com/fluencelabs/js-libp2p-noise
synced 2025-06-13 02:31:39 +00:00
Create fallback test
This commit is contained in:
83
test/xx-fallback-handshake.test.ts
Normal file
83
test/xx-fallback-handshake.test.ts
Normal file
@ -0,0 +1,83 @@
|
||||
import Wrap from "it-pb-rpc";
|
||||
import {Buffer} from "buffer";
|
||||
import Duplex from 'it-pair/duplex';
|
||||
|
||||
import {
|
||||
createHandshakePayload,
|
||||
generateKeypair,
|
||||
getHandshakePayload,
|
||||
signEarlyDataPayload,
|
||||
signPayload
|
||||
} from "../src/utils";
|
||||
import {generateEd25519Keys, getKeyPairFromPeerId} from "./utils";
|
||||
import {Handshake} from "../src/handshake-xx-fallback";
|
||||
import {createPeerIdsFromFixtures} from "./fixtures/peer";
|
||||
import {assert} from "chai";
|
||||
import {encodeMessageBuffer} from "../src/encoder";
|
||||
|
||||
describe("XX Fallback Handshake", () => {
|
||||
let peerA, peerB, fakePeer;
|
||||
|
||||
before(async () => {
|
||||
[peerA, peerB] = await createPeerIdsFromFixtures(2);
|
||||
});
|
||||
|
||||
it("should make handshake with received ephemeral key (from initial IK message)", async () => {
|
||||
try {
|
||||
const duplex = Duplex();
|
||||
const connectionFrom = Wrap(duplex[0]);
|
||||
const connectionTo = Wrap(duplex[1]);
|
||||
|
||||
const prologue = Buffer.from('/noise');
|
||||
const staticKeysInitiator = generateKeypair();
|
||||
const staticKeysResponder = generateKeypair();
|
||||
|
||||
const {privateKey: initiatorPrivKey, publicKey: initiatorPubKey} = getKeyPairFromPeerId(peerA);
|
||||
const {privateKey: responderPrivKey, publicKey: responderPubKey} = getKeyPairFromPeerId(peerB);
|
||||
|
||||
const signedPayload = signPayload(initiatorPrivKey, getHandshakePayload(staticKeysInitiator.publicKey));
|
||||
const signedEarlyDataPayload = signEarlyDataPayload(initiatorPrivKey, Buffer.alloc(0));
|
||||
const handshakePayload = await createHandshakePayload(
|
||||
initiatorPubKey,
|
||||
initiatorPrivKey,
|
||||
signedPayload,
|
||||
signedEarlyDataPayload,
|
||||
);
|
||||
const initialMsg = encodeMessageBuffer({
|
||||
ne: staticKeysInitiator.publicKey,
|
||||
ns: Buffer.alloc(32),
|
||||
ciphertext: handshakePayload,
|
||||
});
|
||||
|
||||
const handshakeInit =
|
||||
new Handshake(true, initiatorPrivKey, initiatorPubKey, prologue, staticKeysInitiator, connectionFrom, peerB, staticKeysInitiator, initialMsg);
|
||||
|
||||
const handshakeResp =
|
||||
new Handshake(false, responderPrivKey, responderPubKey, prologue, staticKeysResponder, connectionTo, peerA, staticKeysInitiator, initialMsg);
|
||||
|
||||
|
||||
await handshakeInit.propose();
|
||||
await handshakeResp.propose();
|
||||
|
||||
await handshakeResp.exchange();
|
||||
await handshakeInit.exchange();
|
||||
|
||||
await handshakeInit.finish();
|
||||
await handshakeResp.finish();
|
||||
|
||||
const sessionInitator = handshakeInit.session;
|
||||
const sessionResponder = handshakeResp.session;
|
||||
|
||||
// Test shared key
|
||||
if (sessionInitator.cs1 && sessionResponder.cs1 && sessionInitator.cs2 && sessionResponder.cs2) {
|
||||
assert(sessionInitator.cs1.k.equals(sessionResponder.cs1.k));
|
||||
assert(sessionInitator.cs2.k.equals(sessionResponder.cs2.k));
|
||||
} else {
|
||||
assert(false);
|
||||
}
|
||||
} catch (e) {
|
||||
console.error(e);
|
||||
assert(false, e.message);
|
||||
}
|
||||
});
|
||||
})
|
@ -9,7 +9,7 @@ import {createPeerIdsFromFixtures} from "./fixtures/peer";
|
||||
import {getKeyPairFromPeerId} from "./utils";
|
||||
|
||||
|
||||
describe("Handshake", () => {
|
||||
describe("XX Handshake", () => {
|
||||
let peerA, peerB, fakePeer;
|
||||
|
||||
before(async () => {
|
Reference in New Issue
Block a user