mirror of
https://github.com/fluencelabs/js-libp2p-noise
synced 2025-04-24 19:52:26 +00:00
77 lines
2.5 KiB
TypeScript
77 lines
2.5 KiB
TypeScript
import Wrap from 'it-pb-rpc'
|
|
import { Buffer } from 'buffer'
|
|
import Duplex from 'it-pair/duplex'
|
|
|
|
import {
|
|
generateKeypair,
|
|
getPayload
|
|
} from '../src/utils'
|
|
import { XXFallbackHandshake } from '../src/handshake-xx-fallback'
|
|
import { createPeerIdsFromFixtures } from './fixtures/peer'
|
|
import { assert } from 'chai'
|
|
import { encode0 } from '../src/encoder'
|
|
|
|
describe('XX Fallback Handshake', () => {
|
|
let peerA, peerB
|
|
|
|
before(async () => {
|
|
[peerA, peerB] = await createPeerIdsFromFixtures(2)
|
|
})
|
|
|
|
it('should test that both parties can fallback to XX and finish handshake', async () => {
|
|
try {
|
|
const duplex = Duplex()
|
|
const connectionFrom = Wrap(duplex[0])
|
|
const connectionTo = Wrap(duplex[1])
|
|
|
|
const prologue = Buffer.alloc(0)
|
|
const staticKeysInitiator = generateKeypair()
|
|
const staticKeysResponder = generateKeypair()
|
|
const ephemeralKeys = generateKeypair()
|
|
|
|
// Initial msg for responder is IK first message from initiator
|
|
const handshakePayload = await getPayload(peerA, staticKeysInitiator.publicKey)
|
|
const initialMsgR = encode0({
|
|
ne: ephemeralKeys.publicKey,
|
|
ns: Buffer.alloc(0),
|
|
ciphertext: handshakePayload
|
|
})
|
|
|
|
const respPayload = await getPayload(peerB, staticKeysResponder.publicKey)
|
|
const handshakeResp =
|
|
new XXFallbackHandshake(false, respPayload, prologue, staticKeysResponder, connectionTo, initialMsgR, peerA)
|
|
|
|
await handshakeResp.propose()
|
|
await handshakeResp.exchange()
|
|
|
|
// Initial message for initiator is XX Message B from responder
|
|
// This is the point where initiator falls back from IK
|
|
const initialMsgI = await connectionFrom.readLP()
|
|
const handshakeInit =
|
|
new XXFallbackHandshake(true, handshakePayload, prologue, staticKeysInitiator, connectionFrom, initialMsgI.slice(0), peerB, ephemeralKeys)
|
|
|
|
await handshakeInit.propose()
|
|
await handshakeInit.exchange()
|
|
|
|
await handshakeInit.finish()
|
|
await handshakeResp.finish()
|
|
|
|
const sessionInitator = handshakeInit.session
|
|
const sessionResponder = handshakeResp.session
|
|
|
|
// Test shared key
|
|
if (sessionInitator.cs1 !== undefined &&
|
|
sessionResponder.cs1 !== undefined &&
|
|
sessionInitator.cs2 !== undefined &&
|
|
sessionResponder.cs2 !== undefined) {
|
|
assert(sessionInitator.cs1.k.equals(sessionResponder.cs1.k))
|
|
assert(sessionInitator.cs2.k.equals(sessionResponder.cs2.k))
|
|
} else {
|
|
assert(false)
|
|
}
|
|
} catch (e) {
|
|
assert(false, e.message)
|
|
}
|
|
})
|
|
})
|