add block.js and tests
This commit is contained in:
parent
936f7913db
commit
c0c47f076a
3 changed files with 279 additions and 0 deletions
149
src/block.js
Normal file
149
src/block.js
Normal file
|
@ -0,0 +1,149 @@
|
|||
var assert = require('assert')
|
||||
var bufferutils = require('./bufferutils')
|
||||
var crypto = require('./crypto')
|
||||
|
||||
var Transaction = require('./transaction')
|
||||
var Script = require('./script')
|
||||
|
||||
function Block() {
|
||||
this.version = 1
|
||||
this.prevHash = null
|
||||
this.merkleRoot = null
|
||||
this.timestamp = 0
|
||||
this.bits = 0
|
||||
this.nonce = 0
|
||||
}
|
||||
|
||||
Block.fromBuffer = function(buffer) {
|
||||
assert(buffer.length >= 80, 'Buffer too small (< 80 bytes)')
|
||||
|
||||
var offset = 0
|
||||
function readSlice(n) {
|
||||
offset += n
|
||||
return buffer.slice(offset - n, offset)
|
||||
}
|
||||
|
||||
function readUInt32() {
|
||||
var i = buffer.readUInt32LE(offset)
|
||||
offset += 4
|
||||
return i
|
||||
}
|
||||
|
||||
var block = new Block()
|
||||
block.version = readUInt32()
|
||||
block.prevHash = readSlice(32)
|
||||
block.merkleRoot = readSlice(32)
|
||||
block.timestamp = readUInt32()
|
||||
block.bits = readUInt32()
|
||||
block.nonce = readUInt32()
|
||||
|
||||
if (buffer.length === 80) return block
|
||||
|
||||
function readUInt64() {
|
||||
var i = bufferutils.readUInt64LE(buffer, offset)
|
||||
offset += 8
|
||||
return i
|
||||
}
|
||||
|
||||
function readVarInt() {
|
||||
var vi = bufferutils.readVarInt(buffer, offset)
|
||||
offset += vi.size
|
||||
return vi.number
|
||||
}
|
||||
|
||||
function readScript() {
|
||||
return Script.fromBuffer(readSlice(readVarInt()))
|
||||
}
|
||||
|
||||
function readTransaction() {
|
||||
var tx = new Transaction()
|
||||
tx.version = readUInt32()
|
||||
|
||||
var vinLen = readVarInt()
|
||||
for (var i = 0; i < vinLen; ++i) {
|
||||
tx.ins.push({
|
||||
hash: readSlice(32),
|
||||
index: readUInt32(),
|
||||
script: readScript(),
|
||||
sequence: readUInt32()
|
||||
})
|
||||
}
|
||||
|
||||
var voutLen = readVarInt()
|
||||
for (i = 0; i < voutLen; ++i) {
|
||||
tx.outs.push({
|
||||
value: readUInt64(),
|
||||
script: readScript(),
|
||||
})
|
||||
}
|
||||
|
||||
tx.locktime = readUInt32()
|
||||
|
||||
return tx
|
||||
}
|
||||
|
||||
var nTransactions = readVarInt()
|
||||
block.transactions = []
|
||||
|
||||
for (var i = 0; i < nTransactions; ++i) {
|
||||
var tx = readTransaction()
|
||||
block.transactions.push(tx)
|
||||
}
|
||||
|
||||
return block
|
||||
}
|
||||
|
||||
Block.fromHex = function(hex) {
|
||||
return Block.fromBuffer(new Buffer(hex, 'hex'))
|
||||
}
|
||||
|
||||
Block.prototype.toBuffer = function(headersOnly) {
|
||||
var buffer = new Buffer(80)
|
||||
|
||||
var offset = 0
|
||||
function writeSlice(slice) {
|
||||
slice.copy(buffer, offset)
|
||||
offset += slice.length
|
||||
}
|
||||
|
||||
function writeUInt32(i) {
|
||||
buffer.writeUInt32LE(i, offset)
|
||||
offset += 4
|
||||
}
|
||||
|
||||
writeUInt32(this.version)
|
||||
writeSlice(this.prevHash)
|
||||
writeSlice(this.merkleRoot)
|
||||
writeUInt32(this.timestamp)
|
||||
writeUInt32(this.bits)
|
||||
writeUInt32(this.nonce)
|
||||
|
||||
if (headersOnly || !this.transactions) return buffer
|
||||
|
||||
function varIntBuffer(i) {
|
||||
var ib = new Buffer(bufferutils.varIntSize(i))
|
||||
bufferutils.writeVarInt(ib, i, 0)
|
||||
return ib
|
||||
}
|
||||
|
||||
var txLenBuffer = varIntBuffer(this.transactions.length)
|
||||
var txBuffers = this.transactions.map(function(tx) {
|
||||
return tx.toBuffer()
|
||||
})
|
||||
|
||||
return Buffer.concat([buffer, txLenBuffer].concat(txBuffers))
|
||||
}
|
||||
|
||||
Block.prototype.toHex = function(headersOnly) {
|
||||
return this.toBuffer(headersOnly).toString('hex')
|
||||
}
|
||||
|
||||
Block.prototype.getId = function() {
|
||||
return bufferutils.reverse(this.getHash()).toString('hex')
|
||||
}
|
||||
|
||||
Block.prototype.getHash = function() {
|
||||
return crypto.hash256(this.toBuffer(true))
|
||||
}
|
||||
|
||||
module.exports = Block
|
72
test/block.js
Normal file
72
test/block.js
Normal file
|
@ -0,0 +1,72 @@
|
|||
var assert = require('assert')
|
||||
|
||||
var Block = require('../src/block')
|
||||
|
||||
var fixtures = require('./fixtures/block')
|
||||
|
||||
describe('Block', function() {
|
||||
describe('fromBuffer/fromHex', function() {
|
||||
fixtures.valid.forEach(function(f) {
|
||||
it('imports the block: ' + f.description + ' correctly', function() {
|
||||
var block = Block.fromHex(f.hex)
|
||||
|
||||
assert.equal(block.version, f.version)
|
||||
assert.equal(block.prevHash.toString('hex'), f.prevHash)
|
||||
assert.equal(block.merkleRoot.toString('hex'), f.merkleRoot)
|
||||
assert.equal(block.timestamp, f.timestamp)
|
||||
assert.equal(block.bits, f.bits)
|
||||
assert.equal(block.nonce, f.nonce)
|
||||
})
|
||||
})
|
||||
|
||||
fixtures.invalid.forEach(function(f) {
|
||||
it('throws on ' + f.exception, function() {
|
||||
assert.throws(function() {
|
||||
Block.fromHex(f.hex)
|
||||
}, new RegExp(f.exception))
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('toBuffer/toHex', function() {
|
||||
fixtures.valid.forEach(function(f) {
|
||||
var block
|
||||
|
||||
beforeEach(function() {
|
||||
block = Block.fromHex(f.hex)
|
||||
})
|
||||
|
||||
it('exports the block: ' + f.description + ' correctly', function() {
|
||||
assert.equal(block.toHex(), f.hex)
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('getHash', function() {
|
||||
fixtures.valid.forEach(function(f) {
|
||||
var block
|
||||
|
||||
beforeEach(function() {
|
||||
block = Block.fromHex(f.hex)
|
||||
})
|
||||
|
||||
it('calculates ' + f.hash + ' for the block: ' + f.description, function() {
|
||||
assert.equal(block.getHash().toString('hex'), f.hash)
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
describe('getId', function() {
|
||||
fixtures.valid.forEach(function(f) {
|
||||
var block
|
||||
|
||||
beforeEach(function() {
|
||||
block = Block.fromHex(f.hex)
|
||||
})
|
||||
|
||||
it('calculates ' + f.id + ' for the block: ' + f.description, function() {
|
||||
assert.equal(block.getId(), f.id)
|
||||
})
|
||||
})
|
||||
})
|
||||
})
|
58
test/fixtures/block.json
vendored
Normal file
58
test/fixtures/block.json
vendored
Normal file
File diff suppressed because one or more lines are too long
Loading…
Reference in a new issue