This repository has been archived by the owner on Feb 24, 2021. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 22
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #8 from libp2p/pull
[WIP] Refactor to use pull streams
- Loading branch information
Showing
13 changed files
with
493 additions
and
448 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,163 @@ | ||
| 'use strict' | ||
|
|
||
| const protobuf = require('protocol-buffers') | ||
| const path = require('path') | ||
| const fs = require('fs') | ||
| const PeerId = require('peer-id') | ||
| const crypto = require('libp2p-crypto') | ||
| const debug = require('debug') | ||
| const log = debug('libp2p:secio') | ||
| log.error = debug('libp2p:secio:error') | ||
|
|
||
| const pbm = protobuf(fs.readFileSync(path.join(__dirname, 'secio.proto'))) | ||
|
|
||
| const support = require('../support') | ||
|
|
||
| // nonceSize is the size of our nonces (in bytes) | ||
| const nonceSize = 16 | ||
|
|
||
| exports.createProposal = (state) => { | ||
| state.proposal.out = { | ||
| rand: support.randomBytes(nonceSize), | ||
| pubkey: state.key.local.public.bytes, | ||
| exchanges: support.exchanges.join(','), | ||
| ciphers: support.ciphers.join(','), | ||
| hashes: support.hashes.join(',') | ||
| } | ||
|
|
||
| state.proposalEncoded.out = pbm.Propose.encode(state.proposal.out) | ||
| return state.proposalEncoded.out | ||
| } | ||
|
|
||
| exports.createExchange = (state) => { | ||
| const res = crypto.generateEphemeralKeyPair(state.protocols.local.curveT) | ||
| state.ephemeralKey.local = res.key | ||
| state.shared.generate = res.genSharedKey | ||
|
|
||
| // Gather corpus to sign. | ||
| const selectionOut = Buffer.concat([ | ||
| state.proposalEncoded.out, | ||
| state.proposalEncoded.in, | ||
| state.ephemeralKey.local | ||
| ]) | ||
|
|
||
| state.exchange.out = { | ||
| epubkey: state.ephemeralKey.local, | ||
| signature: new Buffer(state.key.local.sign(selectionOut), 'binary') | ||
| } | ||
|
|
||
| return pbm.Exchange.encode(state.exchange.out) | ||
| } | ||
|
|
||
| exports.identify = (state, msg) => { | ||
| log('1.1 identify') | ||
|
|
||
| state.proposalEncoded.in = msg | ||
| state.proposal.in = pbm.Propose.decode(msg) | ||
| const pubkey = state.proposal.in.pubkey | ||
|
|
||
| console.log(state.proposal.in) | ||
|
|
||
| state.key.remote = crypto.unmarshalPublicKey(pubkey) | ||
| state.id.remote = PeerId.createFromPubKey(pubkey.toString('base64')) | ||
|
|
||
| log('1.1 identify - %s - identified remote peer as %s', state.id.local.toB58String(), state.id.remote.toB58String()) | ||
| } | ||
|
|
||
| exports.selectProtocols = (state) => { | ||
| log('1.2 selection') | ||
|
|
||
| const local = { | ||
| pubKeyBytes: state.key.local.public.bytes, | ||
| exchanges: support.exchanges, | ||
| hashes: support.hashes, | ||
| ciphers: support.ciphers, | ||
| nonce: state.proposal.out.rand | ||
| } | ||
|
|
||
| const remote = { | ||
| pubKeyBytes: state.proposal.in.pubkey, | ||
| exchanges: state.proposal.in.exchanges.split(','), | ||
| hashes: state.proposal.in.hashes.split(','), | ||
| ciphers: state.proposal.in.ciphers.split(','), | ||
| nonce: state.proposal.in.rand | ||
| } | ||
|
|
||
| let selected = support.selectBest(local, remote) | ||
| // we use the same params for both directions (must choose same curve) | ||
| // WARNING: if they dont SelectBest the same way, this won't work... | ||
| state.protocols.remote = { | ||
| order: selected.order, | ||
| curveT: selected.curveT, | ||
| cipherT: selected.cipherT, | ||
| hashT: selected.hashT | ||
| } | ||
|
|
||
| state.protocols.local = { | ||
| order: selected.order, | ||
| curveT: selected.curveT, | ||
| cipherT: selected.cipherT, | ||
| hashT: selected.hashT | ||
| } | ||
| } | ||
|
|
||
| exports.verify = (state, msg) => { | ||
| log('2.1. verify') | ||
|
|
||
| state.exchange.in = pbm.Exchange.decode(msg) | ||
| state.ephemeralKey.remote = state.exchange.in.epubkey | ||
|
|
||
| const selectionIn = Buffer.concat([ | ||
| state.proposalEncoded.in, | ||
| state.proposalEncoded.out, | ||
| state.ephemeralKey.remote | ||
| ]) | ||
|
|
||
| const sigOk = state.key.remote.verify(selectionIn, state.exchange.in.signature) | ||
|
|
||
| if (!sigOk) { | ||
| throw new Error('Bad signature') | ||
| } | ||
|
|
||
| log('2.1. verify - signature verified') | ||
| } | ||
|
|
||
| exports.generateKeys = (state) => { | ||
| log('2.2. keys') | ||
|
|
||
| state.shared.secret = state.shared.generate(state.exchange.in.epubkey) | ||
|
|
||
| const keys = crypto.keyStretcher( | ||
| state.protocols.local.cipherT, | ||
| state.protocols.local.hashT, | ||
| state.shared.secret | ||
| ) | ||
|
|
||
| // use random nonces to decide order. | ||
| if (state.protocols.local.order > 0) { | ||
| state.protocols.local.keys = keys.k1 | ||
| state.protocols.remote.keys = keys.k2 | ||
| } else if (state.protocols.local.order < 0) { | ||
| // swap | ||
| state.protocols.local.keys = keys.k2 | ||
| state.protocols.remote.keys = keys.k1 | ||
| } else { | ||
| // we should've bailed before state. but if not, bail here. | ||
| throw new Error('you are trying to talk to yourself') | ||
| } | ||
|
|
||
| log('2.3. mac + cipher') | ||
|
|
||
| support.makeMacAndCipher(state.protocols.local) | ||
| support.makeMacAndCipher(state.protocols.remote) | ||
| } | ||
|
|
||
| exports.verifyNonce = (state, n2) => { | ||
| const n1 = state.proposal.out.rand | ||
|
|
||
| if (n1.equals(n2)) return | ||
|
|
||
| throw new Error( | ||
| `Failed to read our encrypted nonce: ${n1.toString('hex')} != ${n2.toString('hex')}` | ||
| ) | ||
| } |
Oops, something went wrong.