Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: replace node buffers with uint8arrays #127

Merged
merged 2 commits into from
Aug 7, 2020
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions .aegir.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
'use strict'

module.exports = {
bundlesize: {
maxSize: '140kB'
}
}
19 changes: 8 additions & 11 deletions package.json
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
"release-minor": "aegir release --type minor",
"release-major": "aegir release --type major",
"coverage": "aegir coverage",
"size": "bundlesize -f dist/index.min.js -s 140kB"
"size": "aegir build -b"
},
"files": [
"src",
Expand All @@ -38,20 +38,17 @@
"devDependencies": {
"@types/chai": "^4.2.7",
"@types/dirty-chai": "^2.0.2",
"@types/mocha": "^7.0.2",
"aegir": "^22.0.0",
"bundlesize": "~0.18.0",
"chai": "^4.2.0",
"dirty-chai": "^2.0.1"
"@types/mocha": "^8.0.1",
"aegir": "^25.0.0"
},
"dependencies": {
"buffer": "^5.5.0",
"cids": "^0.8.0",
"cids": "^1.0.0",
"class-is": "^1.1.0",
"libp2p-crypto": "^0.17.7",
"libp2p-crypto": "^0.18.0",
"minimist": "^1.2.5",
"multihashes": "^1.0.1",
"protons": "^1.0.2"
"multihashes": "^3.0.1",
"protons": "^2.0.0",
"uint8arrays": "^1.1.0"
},
"repository": {
"type": "git",
Expand Down
38 changes: 19 additions & 19 deletions src/index.d.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import crypto, { PrivateKey, PublicKey, KeyType } from "libp2p-crypto";
import { PrivateKey, PublicKey, KeyType } from "libp2p-crypto";
import CID from 'cids'

declare namespace PeerId {
Expand Down Expand Up @@ -51,36 +51,36 @@ declare namespace PeerId {
* @param str The input hex string.
*/
function createFromHexString(str: string): PeerId;

/**
* Create PeerId from raw bytes.
* @param buf The raw bytes.
*/
function createFromBytes(buf: Buffer): PeerId;
function createFromBytes(buf: Uint8Array): PeerId;

/**
* Create PeerId from base58-encoded string.
* @param str The base58-encoded string.
*/
function createFromB58String(str: string): PeerId;

/**
* Create PeerId from CID.
* @param cid The CID.
*/
function createFromCID(cid: CID | Buffer | string | object): PeerId;
function createFromCID(cid: CID | Uint8Array | string | object): PeerId;

/**
* Create PeerId from public key.
* @param key Public key, as Buffer or base64-encoded string.
* @param key Public key, as Uint8Array or base64-encoded string.
*/
function createFromPubKey(key: Buffer | string): Promise<PeerId>;
function createFromPubKey(key: Uint8Array | string): Promise<PeerId>;

/**
* Create PeerId from private key.
* @param key Private key, as Buffer or base64-encoded string.
* @param key Private key, as Uint8Array or base64-encoded string.
*/
function createFromPrivKey(key: Buffer | string): Promise<PeerId>;
function createFromPrivKey(key: Uint8Array | string): Promise<PeerId>;

/**
* Create PeerId from PeerId JSON formatted object.
Expand All @@ -91,21 +91,21 @@ declare namespace PeerId {

/**
* Create PeerId from Protobuf bytes.
* @param buf Protobuf bytes, as Buffer or hex-encoded string.
* @param buf Protobuf bytes, as Uint8Array or hex-encoded string.
*/
function createFromProtobuf(buf: Buffer | string): Promise<PeerId>;
function createFromProtobuf(buf: Uint8Array | string): Promise<PeerId>;
}

/**
* PeerId is an object representation of a peer identifier.
*/
declare class PeerId {
constructor(id: Buffer | string, privKey?: PrivateKey, pubKey?: PublicKey);
constructor(id: Uint8Array | string, privKey?: PrivateKey, pubKey?: PublicKey);

/**
* Raw id.
*/
readonly id: Buffer;
readonly id: Uint8Array;

/**
* Private key.
Expand All @@ -120,18 +120,18 @@ declare class PeerId {
/**
* Return the protobuf version of the public key, matching go ipfs formatting.
*/
marshalPubKey(): Buffer;
marshalPubKey(): Uint8Array;

/**
* Return the protobuf version of the private key, matching go ipfs formatting.
*/
marshalPrivKey(): Buffer;
marshalPrivKey(): Uint8Array;

/**
* Return the protobuf version of the peer-id.
* @param excludePriv Whether to exclude the private key information from the output.
*/
marshal(excludePriv?: boolean): Buffer;
marshal(excludePriv?: boolean): Uint8Array;

/**
* String representation.
Expand All @@ -153,7 +153,7 @@ declare class PeerId {
/**
* Return raw id bytes.
*/
toBytes(): Buffer;
toBytes(): Uint8Array;

/**
* Encode to base58 string.
Expand All @@ -170,14 +170,14 @@ declare class PeerId {
* Checks the equality of `this` peer against a given PeerId.
* @param id The other PeerId.
*/
equals(id: PeerId | Buffer): boolean;
equals(id: PeerId | Uint8Array): boolean;

/**
* Checks the equality of `this` peer against a given PeerId.
* @deprecated Use {.equals}
* @param id The other PeerId.
*/
isEqual(id: PeerId | Buffer): boolean;
isEqual(id: PeerId | Uint8Array): boolean;

/**
* Check if this PeerId instance is valid (privKey -> pubKey -> Id)
Expand Down
56 changes: 28 additions & 28 deletions src/index.js
Original file line number Diff line number Diff line change
Expand Up @@ -4,20 +4,22 @@

'use strict'

const { Buffer } = require('buffer')
const mh = require('multihashes')
const CID = require('cids')
const cryptoKeys = require('libp2p-crypto/src/keys')
const withIs = require('class-is')
const { PeerIdProto } = require('./proto')
const uint8ArrayEquals = require('uint8arrays/equals')
const uint8ArrayFromString = require('uint8arrays/from-string')
const uint8ArrayToString = require('uint8arrays/to-string')

class PeerId {
constructor (id, privKey, pubKey) {
if (!Buffer.isBuffer(id)) {
if (!(id instanceof Uint8Array)) {
throw new Error('invalid id provided')
}

if (privKey && pubKey && !privKey.public.bytes.equals(pubKey.bytes)) {
if (privKey && pubKey && !uint8ArrayEquals(privKey.public.bytes, pubKey.bytes)) {
throw new Error('inconsistent arguments')
}

Expand Down Expand Up @@ -142,14 +144,14 @@ class PeerId {

/**
* Checks the equality of `this` peer against a given PeerId.
* @param {Buffer|PeerId} id
* @param {Uint8Array|PeerId} id
* @returns {boolean}
*/
equals (id) {
if (Buffer.isBuffer(id)) {
return this.id.equals(id)
if (id instanceof Uint8Array) {
return uint8ArrayEquals(this.id, id)
} else if (id.id) {
return this.id.equals(id.id)
return uint8ArrayEquals(this.id, id.id)
} else {
throw new Error('not valid Id')
}
Expand All @@ -158,7 +160,7 @@ class PeerId {
/**
* Checks the equality of `this` peer against a given PeerId.
* @deprecated Use `.equals`
* @param {Buffer|PeerId} id
* @param {Uint8Array|PeerId} id
* @returns {boolean}
*/
isEqual (id) {
Expand All @@ -173,8 +175,8 @@ class PeerId {
return Boolean(this.privKey &&
this.privKey.public &&
this.privKey.public.bytes &&
Buffer.isBuffer(this.pubKey.bytes) &&
this.privKey.public.bytes.equals(this.pubKey.bytes))
this.pubKey.bytes instanceof Uint8Array &&
uint8ArrayEquals(this.privKey.public.bytes, this.pubKey.bytes))
}
}

Expand Down Expand Up @@ -231,16 +233,16 @@ exports.createFromCID = (cid) => {
return new PeerIdWithIs(cid.multihash)
}

// Public Key input will be a buffer
// Public Key input will be a Uint8Array
exports.createFromPubKey = async (key) => {
let buf = key

if (typeof buf === 'string') {
buf = Buffer.from(key, 'base64')
buf = uint8ArrayFromString(key, 'base64pad')
}

if (!Buffer.isBuffer(buf)) {
throw new Error('Supplied key is neither a base64 string nor a buffer')
if (!(buf instanceof Uint8Array)) {
throw new Error('Supplied key is neither a base64 string nor a Uint8Array')
}

const pubKey = await cryptoKeys.unmarshalPublicKey(buf)
Expand All @@ -249,24 +251,22 @@ exports.createFromPubKey = async (key) => {

// Private key input will be a string
exports.createFromPrivKey = async (key) => {
let buf = key

if (typeof buf === 'string') {
buf = Buffer.from(key, 'base64')
if (typeof key === 'string') {
key = uint8ArrayFromString(key, 'base64pad')
}

if (!Buffer.isBuffer(buf)) {
throw new Error('Supplied key is neither a base64 string nor a buffer')
if (!(key instanceof Uint8Array)) {
throw new Error('Supplied key is neither a base64 string nor a Uint8Array')
}

const privKey = await cryptoKeys.unmarshalPrivateKey(buf)
const privKey = await cryptoKeys.unmarshalPrivateKey(key)
return computePeerId(privKey, privKey.public)
}

exports.createFromJSON = async (obj) => {
const id = mh.fromB58String(obj.id)
const rawPrivKey = obj.privKey && Buffer.from(obj.privKey, 'base64')
const rawPubKey = obj.pubKey && Buffer.from(obj.pubKey, 'base64')
const rawPrivKey = obj.privKey && uint8ArrayFromString(obj.privKey, 'base64pad')
const rawPubKey = obj.pubKey && uint8ArrayFromString(obj.pubKey, 'base64pad')
const pub = rawPubKey && await cryptoKeys.unmarshalPublicKey(rawPubKey)

if (!rawPrivKey) {
Expand All @@ -282,11 +282,11 @@ exports.createFromJSON = async (obj) => {
pubDigest = await computeDigest(pub)
}

if (pub && !privDigest.equals(pubDigest)) {
if (pub && !uint8ArrayEquals(privDigest, pubDigest)) {
throw new Error('Public and private key do not match')
}

if (id && !privDigest.equals(id)) {
if (id && !uint8ArrayEquals(privDigest, id)) {
throw new Error('Id and private key do not match')
}

Expand All @@ -295,7 +295,7 @@ exports.createFromJSON = async (obj) => {

exports.createFromProtobuf = async (buf) => {
if (typeof buf === 'string') {
buf = Buffer.from(buf, 'hex')
buf = uint8ArrayFromString(buf, 'base16')
}

let { id, privKey, pubKey } = PeerIdProto.decode(buf)
Expand All @@ -316,7 +316,7 @@ exports.createFromProtobuf = async (buf) => {

if (privKey) {
if (pubKey) {
if (!privDigest.equals(pubDigest)) {
if (!uint8ArrayEquals(privDigest, pubDigest)) {
throw new Error('Public and private key do not match')
}
}
Expand Down Expand Up @@ -344,6 +344,6 @@ exports.isPeerId = (peerId) => {

function toB64Opt (val) {
if (val) {
return val.toString('base64')
return uint8ArrayToString(val, 'base64pad')
}
}
Loading