2014-06-11 19:39:13 -07:00
|
|
|
'use strict';
|
|
|
|
|
2014-06-24 08:36:32 -07:00
|
|
|
var chai = chai || require('chai');
|
|
|
|
var should = chai.should();
|
|
|
|
var expect = chai.expect;
|
|
|
|
var sinon = sinon || require('sinon');
|
|
|
|
var bitcore = bitcore || require('bitcore');
|
2014-06-16 14:41:33 -07:00
|
|
|
var WebRTC = require('../js/models/network/WebRTC');
|
2014-06-11 19:39:13 -07:00
|
|
|
|
|
|
|
describe('Network / WebRTC', function() {
|
|
|
|
|
2014-06-24 08:36:32 -07:00
|
|
|
it('should create an instance', function() {
|
2014-06-16 14:41:33 -07:00
|
|
|
var n = new WebRTC();
|
2014-06-11 19:39:13 -07:00
|
|
|
should.exist(n);
|
|
|
|
});
|
|
|
|
|
2014-06-16 14:41:33 -07:00
|
|
|
describe('#WebRTC constructor', function() {
|
2014-06-16 15:35:54 -07:00
|
|
|
|
2014-06-16 14:41:33 -07:00
|
|
|
it('should set reconnect attempts', function() {
|
|
|
|
var n = new WebRTC();
|
|
|
|
n.reconnectAttempts.should.equal(3);
|
|
|
|
});
|
|
|
|
|
2014-06-11 19:39:13 -07:00
|
|
|
it('should call cleanUp', function() {
|
2014-06-16 14:41:33 -07:00
|
|
|
var save = WebRTC.prototype.cleanUp;
|
|
|
|
WebRTC.prototype.cleanUp = sinon.spy();
|
|
|
|
var n = new WebRTC();
|
2014-06-11 19:39:13 -07:00
|
|
|
n.cleanUp.calledOnce.should.equal(true);
|
2014-06-16 14:41:33 -07:00
|
|
|
WebRTC.prototype.cleanUp = save;
|
2014-06-11 19:39:13 -07:00
|
|
|
});
|
2014-06-24 08:36:32 -07:00
|
|
|
|
2014-06-11 19:39:13 -07:00
|
|
|
});
|
|
|
|
|
|
|
|
describe('#cleanUp', function() {
|
2014-06-16 15:35:54 -07:00
|
|
|
|
|
|
|
it('should not set netKey', function() {
|
|
|
|
var n = new WebRTC();
|
|
|
|
(n.netKey === undefined).should.equal(true);
|
|
|
|
});
|
|
|
|
|
2014-06-11 19:39:13 -07:00
|
|
|
it('should set privkey to null', function() {
|
2014-06-16 14:41:33 -07:00
|
|
|
var n = new WebRTC();
|
2014-06-11 19:39:13 -07:00
|
|
|
n.cleanUp();
|
|
|
|
expect(n.privkey).to.equal(null);
|
|
|
|
});
|
2014-06-16 15:35:54 -07:00
|
|
|
|
|
|
|
});
|
|
|
|
|
2014-06-26 08:49:22 -07:00
|
|
|
describe('#_encode', function() {
|
2014-06-16 15:35:54 -07:00
|
|
|
|
2014-06-26 08:49:22 -07:00
|
|
|
it('should encode data successfully', function() {
|
2014-06-16 15:35:54 -07:00
|
|
|
var n = new WebRTC();
|
2014-06-26 08:49:22 -07:00
|
|
|
var data = new bitcore.Buffer('my data to encode');
|
2014-06-16 15:35:54 -07:00
|
|
|
var privkeystr = new bitcore.Buffer('test privkey');
|
|
|
|
var privkey = bitcore.util.sha256(privkeystr);
|
|
|
|
var key = new bitcore.Key();
|
|
|
|
key.private = privkey;
|
|
|
|
key.regenerateSync();
|
2014-06-26 08:49:22 -07:00
|
|
|
var encoded = n._encode(key.public, key, data);
|
|
|
|
should.exist(encoded);
|
|
|
|
encoded.sig.length.should.not.equal(0);
|
|
|
|
encoded.pubkey.length.should.not.equal(0);
|
|
|
|
encoded.encrypted.length.should.not.equal(0);
|
2014-06-16 15:35:54 -07:00
|
|
|
});
|
|
|
|
|
|
|
|
});
|
|
|
|
|
2014-06-26 08:49:22 -07:00
|
|
|
describe('#_decode', function() {
|
2014-06-16 15:35:54 -07:00
|
|
|
|
2014-06-26 08:49:22 -07:00
|
|
|
it('should decode that which was encoded', function() {
|
2014-06-16 15:35:54 -07:00
|
|
|
var n = new WebRTC();
|
|
|
|
var data = new bitcore.Buffer('my data to encrypt');
|
|
|
|
var privkeystr = new bitcore.Buffer('test privkey');
|
|
|
|
var privkey = bitcore.util.sha256(privkeystr);
|
|
|
|
var key = new bitcore.Key();
|
|
|
|
key.private = privkey;
|
|
|
|
key.regenerateSync();
|
2014-06-26 08:49:22 -07:00
|
|
|
var encoded = n._encode(key.public, key, data);
|
|
|
|
var decoded = n._decode(key, encoded);
|
|
|
|
encoded.sig.should.not.equal(0);
|
|
|
|
decoded.toString().should.equal('my data to encrypt');
|
2014-06-16 15:35:54 -07:00
|
|
|
});
|
|
|
|
|
|
|
|
});
|
|
|
|
|
|
|
|
describe('#send', function() {
|
|
|
|
|
|
|
|
it('should call _sendToOne for a copayer', function(done) {
|
|
|
|
var n = new WebRTC();
|
2014-06-26 08:49:22 -07:00
|
|
|
n.privkey = bitcore.util.sha256('test');
|
2014-06-16 15:35:54 -07:00
|
|
|
|
|
|
|
var data = new bitcore.Buffer('my data to send');
|
|
|
|
|
|
|
|
var privkeystr = new bitcore.Buffer('test privkey');
|
|
|
|
var privkey = bitcore.util.sha256(privkeystr);
|
|
|
|
var key = new bitcore.Key();
|
|
|
|
key.private = privkey;
|
|
|
|
key.regenerateSync();
|
|
|
|
|
|
|
|
var copayerId = key.public.toString('hex');
|
2014-06-26 13:58:45 -07:00
|
|
|
n._sendToOne = function(a1, a2, cb) {
|
2014-06-24 08:36:32 -07:00
|
|
|
cb();
|
|
|
|
};
|
2014-06-16 15:35:54 -07:00
|
|
|
var sig = undefined;
|
|
|
|
n.send(copayerId, data, function() {
|
|
|
|
done();
|
|
|
|
});
|
|
|
|
|
|
|
|
});
|
|
|
|
|
2014-06-16 15:55:40 -07:00
|
|
|
it('should call _sendToOne with encrypted data for a copayer', function(done) {
|
|
|
|
var n = new WebRTC();
|
2014-06-26 08:49:22 -07:00
|
|
|
n.privkey = bitcore.util.sha256('test');
|
2014-06-16 15:55:40 -07:00
|
|
|
|
|
|
|
var data = new bitcore.Buffer('my data to send');
|
|
|
|
|
|
|
|
var privkeystr = new bitcore.Buffer('test privkey');
|
|
|
|
var privkey = bitcore.util.sha256(privkeystr);
|
|
|
|
var key = new bitcore.Key();
|
|
|
|
key.private = privkey;
|
|
|
|
key.regenerateSync();
|
|
|
|
|
|
|
|
var copayerId = key.public.toString('hex');
|
2014-06-26 14:26:58 -07:00
|
|
|
n._sendToOne = function(a1, enc, cb) {
|
|
|
|
var encPayload = JSON.parse(enc.toString());
|
2014-06-26 08:49:22 -07:00
|
|
|
encPayload.sig.length.should.be.greaterThan(0);
|
2014-06-16 15:55:40 -07:00
|
|
|
cb();
|
|
|
|
};
|
|
|
|
var sig = undefined;
|
|
|
|
n.send(copayerId, data, function() {
|
|
|
|
done();
|
|
|
|
});
|
|
|
|
|
|
|
|
});
|
|
|
|
|
2014-06-16 15:35:54 -07:00
|
|
|
it('should call _sendToOne for a list of copayers', function(done) {
|
|
|
|
var n = new WebRTC();
|
2014-06-26 08:49:22 -07:00
|
|
|
n.privkey = bitcore.util.sha256('test');
|
2014-06-16 15:35:54 -07:00
|
|
|
|
|
|
|
var data = new bitcore.Buffer('my data to send');
|
|
|
|
|
|
|
|
var privkeystr = new bitcore.Buffer('test privkey');
|
|
|
|
var privkey = bitcore.util.sha256(privkeystr);
|
|
|
|
var key = new bitcore.Key();
|
|
|
|
key.private = privkey;
|
|
|
|
key.regenerateSync();
|
|
|
|
|
|
|
|
var copayerIds = [key.public.toString('hex')];
|
2014-06-26 13:58:45 -07:00
|
|
|
n._sendToOne = function(a1, a2, cb) {
|
2014-06-24 08:36:32 -07:00
|
|
|
cb();
|
|
|
|
};
|
2014-06-16 15:35:54 -07:00
|
|
|
var sig = undefined;
|
|
|
|
n.send(copayerIds, data, function() {
|
|
|
|
done();
|
|
|
|
});
|
|
|
|
|
|
|
|
});
|
2014-06-11 19:39:13 -07:00
|
|
|
});
|
|
|
|
|
2014-06-26 13:58:45 -07:00
|
|
|
describe('#_onData', function() {
|
|
|
|
var privkey1 = bitcore.util.sha256('test privkey 1');
|
|
|
|
var privkey2 = bitcore.util.sha256('test privkey 2');
|
|
|
|
var privkey3 = bitcore.util.sha256('test privkey 2');
|
|
|
|
|
|
|
|
var key1 = new bitcore.Key();
|
|
|
|
key1.private = privkey1;
|
|
|
|
key1.regenerateSync();
|
|
|
|
|
|
|
|
var key2 = new bitcore.Key();
|
|
|
|
key2.private = privkey2;
|
|
|
|
key2.regenerateSync();
|
|
|
|
|
|
|
|
var key3 = new bitcore.Key();
|
|
|
|
key3.private = privkey3;
|
|
|
|
key3.regenerateSync();
|
|
|
|
|
|
|
|
it('should not reject data sent from a peer with hijacked pubkey', function() {
|
|
|
|
var n = new WebRTC();
|
|
|
|
n.privkey = key2.private.toString('hex');
|
|
|
|
|
|
|
|
var message = {
|
|
|
|
type: 'hello',
|
|
|
|
copayerId: key1.public.toString('hex')
|
|
|
|
};
|
|
|
|
var messagestr = JSON.stringify(message);
|
|
|
|
var messagebuf = new Buffer(messagestr);
|
|
|
|
|
|
|
|
var encoded = n._encode(key2.public, key1, messagebuf);
|
|
|
|
var encodedstr = JSON.stringify(encoded);
|
|
|
|
var encodeduint = new Buffer(encodedstr);
|
|
|
|
|
|
|
|
var isInbound = true;
|
|
|
|
var peerId = new bitcore.SIN(key1.public);
|
|
|
|
|
|
|
|
n._deletePeer = sinon.spy();
|
|
|
|
|
|
|
|
n._onData(encodeduint, isInbound, peerId);
|
|
|
|
n._deletePeer.calledOnce.should.equal(false);
|
|
|
|
});
|
|
|
|
|
|
|
|
it('should reject data sent from a peer with hijacked pubkey', function() {
|
|
|
|
var n = new WebRTC();
|
|
|
|
n.privkey = key2.private.toString('hex');
|
|
|
|
|
|
|
|
var message = {
|
|
|
|
type: 'hello',
|
|
|
|
copayerId: key3.public.toString('hex') //MITM pubkey 3
|
|
|
|
};
|
|
|
|
var messagestr = JSON.stringify(message);
|
|
|
|
var messagebuf = new Buffer(messagestr);
|
|
|
|
|
|
|
|
var encoded = n._encode(key2.public, key1, messagebuf);
|
|
|
|
var encodedstr = JSON.stringify(encoded);
|
|
|
|
var encodeduint = new Buffer(encodedstr);
|
|
|
|
|
|
|
|
var isInbound = true;
|
|
|
|
var peerId = new bitcore.SIN(key1.public);
|
|
|
|
|
|
|
|
n._deletePeer = sinon.spy();
|
|
|
|
|
|
|
|
n._onData(encodeduint, isInbound, peerId);
|
|
|
|
n._deletePeer.calledOnce.should.equal(true);
|
|
|
|
n._deletePeer.getCall(0).args[0].should.equal(peerId);
|
|
|
|
n._deletePeer.getCall(0).args[1].should.equal('incorrect pubkey for peerId');
|
|
|
|
});
|
|
|
|
|
|
|
|
});
|
|
|
|
|
2014-06-11 19:39:13 -07:00
|
|
|
});
|