RanchiMall/flocore-p2p
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added preconditions to filteradd message.

Browse Source
This commit is contained in:
Braydon Fuller 2015-04-01 11:59:17 -04:00
parent 28b05e3cc1
commit 81047d73bd
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lib/messages/commands/filteradd.js
View File

@ -8,6 +8,7 @@ var BufferUtil = bitcore.util.buffer;
var BufferWriter = bitcore.encoding.BufferWriter;
var BufferReader = bitcore.encoding.BufferReader;
var $ = bitcore.util.preconditions;
var _ = bitcore.deps._;
/**
* Request peer to add data to a bloom filter already set by 'filterload'
@ -21,6 +22,10 @@ function FilteraddMessage(arg, options) {
Message.call(this, arg, options);
this.magicNumber = options.magicNumber;
this.command = 'filteradd';
$.checkArgument(
_.isUndefined(arg) || BufferUtil.isBuffer(arg),
'First argument is expected to be a Buffer or undefined'
);
this.data = arg || BufferUtil.EMPTY_BUFFER;
}
inherits(FilteraddMessage, Message);

10
test/messages/commands/index.js
View File

@ -102,6 +102,16 @@ describe('Command Messages', function() {
});
describe('FilterAdd', function() {
it('should error if arg is not a buffer', function() {
(function() {
var message = messages.FilterAdd('not a buffer');
}).should.throw('First argument is expected to be a Buffer or undefined');
});
});
describe('FilterLoad', function() {
it('should return a null payload', function() {