Daniel Cousens
10 years ago
3 changed files with 279 additions and 0 deletions
@ -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 |
@ -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) |
||||
|
}) |
||||
|
}) |
||||
|
}) |
||||
|
}) |
File diff suppressed because one or more lines are too long
Loading…
Reference in new issue