fluencelabs/js-libp2p-noise
1
0
Fork 0
mirror of https://github.com/fluencelabs/js-libp2p-noise synced 2025-07-02 23:21:43 +00:00
Code Issues Projects Releases Wiki Activity

add noise spec length prefix encoding

Browse Source
This commit is contained in:
Marin Petrunić
2020-02-13 22:51:36 +01:00
parent 66b9e76352
commit 87f641f650
5 changed files with 47 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/encoder.ts
View File

@ -4,7 +4,8 @@ import {MessageBuffer} from "./@types/handshake";
export const uint16BEEncode = (value, target, offset) => {
target = target || Buffer.allocUnsafe(2);
return target.writeUInt16BE(value, offset);
target.writeUInt16BE(value, offset);
return target;
};
uint16BEEncode.bytes = 2;

25
src/noise.ts
View File

@ -4,7 +4,7 @@ import Wrap from 'it-pb-rpc';
import DuplexPair from 'it-pair/duplex';
import ensureBuffer from 'it-buffer';
import pipe from 'it-pipe';
import lp from 'it-length-prefixed';
import {encode, decode} from 'it-length-prefixed';
import {XXHandshake} from "./handshake-xx";
import {IKHandshake} from "./handshake-ik";
@ -19,6 +19,7 @@ import {IHandshake} from "./@types/handshake-interface";
import {KeyCache} from "./keycache";
import {logger} from "./logger";
import PeerId from "peer-id";
import {NOISE_MSG_MAX_LENGTH_BYTES} from "./constants";
export type WrappedConnection = ReturnType<typeof Wrap>;
@ -66,7 +67,14 @@ export class Noise implements INoiseConnection {
* @returns {Promise<SecureOutbound>}
*/
public async secureOutbound(localPeer: PeerId, connection: any, remotePeer: PeerId): Promise<SecureOutbound> {
const wrappedConnection = Wrap(connection);
const wrappedConnection = Wrap(
connection,
{
lengthEncoder: uint16BEEncode,
lengthDecoder: uint16BEDecode,
maxDataLength: NOISE_MSG_MAX_LENGTH_BYTES
}
);
const handshake = await this.performHandshake({
connection: wrappedConnection,
isInitiator: true,
@ -89,7 +97,14 @@ export class Noise implements INoiseConnection {
* @returns {Promise<SecureOutbound>}
*/
public async secureInbound(localPeer: PeerId, connection: any, remotePeer?: PeerId): Promise<SecureOutbound> {
const wrappedConnection = Wrap(connection);
const wrappedConnection = Wrap(
connection,
{
lengthEncoder: uint16BEEncode,
lengthDecoder: uint16BEDecode,
maxDataLength: NOISE_MSG_MAX_LENGTH_BYTES
}
);
const handshake = await this.performHandshake({
connection: wrappedConnection,
isInitiator: false,
@ -213,9 +228,9 @@ export class Noise implements INoiseConnection {
secure, // write to wrapper
ensureBuffer, // ensure any type of data is converted to buffer
encryptStream(handshake), // data is encrypted
lp.encode({ lengthEncoder: uint16BEEncode }), // prefix with message length
encode({ lengthEncoder: uint16BEEncode }), // prefix with message length
network, // send to the remote peer
lp.decode({ lengthDecoder: uint16BEDecode }), // read message length prefix
decode({ lengthDecoder: uint16BEDecode, maxDataLength: NOISE_MSG_MAX_LENGTH_BYTES }), // read message length prefix
ensureBuffer, // ensure any type of data is converted to buffer
decryptStream(handshake), // decrypt the incoming data
secure // pipe to the wrapper