fcoin/lib/bcoin/utils.js
2015-12-21 15:20:01 -08:00

787 lines
15 KiB
JavaScript

/**
* utils.js - utils for bcoin
* Copyright (c) 2014-2015, Fedor Indutny (MIT License)
* https://github.com/indutny/bcoin
*/
var utils = exports;
var bn = require('bn.js');
var hash = require('hash.js');
var util = require('util');
/**
* Utils
*/
function toArray(msg, enc) {
if (Array.isArray(msg))
return msg.slice();
if (!msg)
return [];
var res = [];
var i, c, hi, lo, slice, num;
if (typeof msg === 'string') {
if (!enc) {
for (i = 0; i < msg.length; i++) {
c = msg.charCodeAt(i);
hi = c >> 8;
lo = c & 0xff;
if (hi)
res.push(hi, lo);
else
res.push(lo);
}
} else if (enc === 'hex') {
msg = msg.replace(/[^a-z0-9]+/ig, '');
if (msg.length % 2 !== 0)
msg = '0' + msg;
for (i = 0; i < msg.length; i += 8) {
slice = msg.slice(i, i + 8);
num = parseInt(slice, 16);
if (slice.length === 8)
res.push((num >>> 24) & 0xff);
if (slice.length >= 6)
res.push((num >>> 16) & 0xff);
if (slice.length >= 4)
res.push((num >>> 8) & 0xff);
res.push(num & 0xff);
}
}
} else {
for (i = 0; i < msg.length; i++)
res[i] = msg[i] | 0;
}
return res;
}
utils.toArray = toArray;
var base58 = '123456789ABCDEFGHJKLMNPQRSTUVWXYZ'
+ 'abcdefghijkmnopqrstuvwxyz';
utils.toBase58 = function toBase58(arr) {
var n = new bn(arr, 16);
// 58 ^ 4
var mod = new bn(0xacad10);
var res = '';
var r, end, i, c;
do {
r = n.mod(mod);
n = n.div(mod);
end = n.cmpn(0) === 0;
utils.assert.equal(r.length, 1);
r = r.words[0];
for (i = 0; i < 4; i++) {
c = r % 58;
r = (r - c) / 58;
if (c === 0 && r === 0 && end)
break;
res = base58[c] + res;
}
utils.assert.equal(r, 0);
} while (!end);
// Add leading "zeroes"
for (i = 0; i < arr.length; i++) {
if (arr[i] !== 0)
break;
res = '1' + res;
}
return res;
};
utils.fromBase58 = function fromBase58(str) {
var i, zeroes, q, w, res, c, z;
// Count leading "zeroes"
for (i = 0; i < str.length; i++)
if (str[i] !== '1')
break;
zeroes = i;
// Read 4-char words and add them to bignum
q = 1;
w = 0;
res = new bn(0);
for (i = zeroes; i < str.length; i++) {
c = base58.indexOf(str[i]);
if (!(c >= 0 && c < 58))
return [];
q *= 58;
w *= 58;
w += c;
if (i === str.length - 1 || q === 0xacad10) {
res = res.mul(new bn(q)).add(new bn(w));
q = 1;
w = 0;
}
}
// Add leading "zeroes"
z = [];
for (i = 0; i < zeroes; i++)
z.push(0);
return z.concat(res.toArray());
};
utils.isBase58 = function isBase58(msg) {
return typeof msg === 'string' && /^[1-9a-zA-Z]+$/.test(msg);
};
utils.ripemd160 = function ripemd160(data, enc) {
return hash.ripemd160().update(data, enc).digest();
};
utils.sha1 = function sha1(data, enc) {
return hash.sha1().update(data, enc).digest();
};
utils.ripesha = function ripesha(data, enc) {
return hash.ripemd160().update(utils.sha256(data, enc)).digest();
};
utils.checksum = function checksum(data, enc) {
return utils.dsha256(data, enc).slice(0, 4);
};
utils.sha256 = function sha256(data, enc) {
return hash.sha256().update(data, enc).digest();
};
utils.dsha256 = function dsha256(data, enc) {
return utils.sha256(utils.sha256(data, enc));
};
utils.readU16 = function readU16(arr, off) {
if (!off)
off = 0;
return arr[off] | (arr[off + 1] << 8);
};
utils.readU32 = function readU32(arr, off) {
if (!off)
off = 0;
var r = arr[off]
| (arr[off + 1] << 8)
| (arr[off + 2] << 16)
| (arr[off + 3] << 24);
if (r < 0)
r += 0x100000000;
return r;
};
utils.readU64 = function readU64(arr, off) {
if (!off)
off = 0;
return utils.readU32(arr, off) + utils.readU32(arr, off + 4) * 0x100000000;
};
utils.writeU16 = function writeU16(dst, num, off) {
if (!off)
off = 0;
dst[off] = num & 0xff;
dst[off + 1] = (num >>> 8) & 0xff;
return 2;
};
utils.writeU32 = function writeU32(dst, num, off) {
if (!off)
off = 0;
dst[off] = num & 0xff;
dst[off + 1] = (num >>> 8) & 0xff;
dst[off + 2] = (num >>> 16) & 0xff;
dst[off + 3] = (num >>> 24) & 0xff;
return 4;
};
utils.writeU64 = function writeU64(dst, num, off) {
var i = 0;
if (!off)
off = 0;
num = new bn(num).maskn(64).toArray();
while (num.length < 8)
num.unshift(0);
num.reverse().forEach(function(ch) {
dst[off++] = ch;
});
i = num.length;
while (i--)
dst[off++] = num[i];
return 8;
};
utils.writeU16BE = function writeU16BE(dst, num, off) {
if (!off)
off = 0;
dst[off] = (num >>> 8) & 0xff;
dst[off + 1] = num & 0xff;
return 2;
};
utils.writeU32BE = function writeU32BE(dst, num, off) {
if (!off)
off = 0;
dst[off] = (num >>> 24) & 0xff;
dst[off + 1] = (num >>> 16) & 0xff;
dst[off + 2] = (num >>> 8) & 0xff;
dst[off + 3] = num & 0xff;
return 4;
};
utils.writeU64BE = function writeU64BE(dst, num, off) {
var i = 0;
if (!off)
off = 0;
num = new bn(num).maskn(64).toArray();
while (num.length < 8)
num.unshift(0);
for (; i < num.length; i++)
dst[off++] = num[i];
return 8;
};
utils.readU16BE = function readU16BE(arr, off) {
if (!off)
off = 0;
return (arr[off] << 8) | arr[off + 1];
};
utils.readU32BE = function readU32BE(arr, off) {
if (!off)
off = 0;
var r = (arr[off] << 24)
| (arr[off + 1] << 16)
| (arr[off + 2] << 8)
| arr[off + 3];
if (r < 0)
r += 0x100000000;
return r;
};
utils.readU64BE = function readU64BE(arr, off) {
if (!off)
off = 0;
return utils.readU32BE(arr, off) * 0x100000000 + utils.readU32BE(arr, off + 4);
};
utils.writeAscii = function writeAscii(dst, str, off) {
var i = 0;
var c;
for (; i < str.length; i++) {
c = str.charCodeAt(i);
dst[off + i] = c & 0xff;
}
return i;
};
utils.readAscii = function readAscii(arr, off, len) {
var str = '';
var i = off;
var c;
for (i = off; i < off + len; i++) {
c = String.fromCharCode(arr[i] & 0xff);
str += c;
}
return str;
};
utils.ascii2array = function ascii2array(str) {
var dst = [];
utils.writeAscii(dst, str, 0);
return dst;
};
utils.array2ascii = function array2ascii(arr) {
return utils.readAscii(arr, 0, arr.length);
};
utils.copy = function copy(src, dst, off, force) {
var len = src.length;
var i = 0;
if (!force)
len = Math.min(dst.length - off, len);
for (; i < len; i++)
dst[i + off] = src[i];
return i;
};
utils.stringify = function stringify(arr) {
var res = '';
var i = 0;
for (; i < arr.length; i++)
res += String.fromCharCode(arr[i]);
return res;
};
function zero2(word) {
if (word.length === 1)
return '0' + word;
return word;
}
function toHex(msg) {
var res = '';
var i = 0;
if (typeof msg === 'string')
return msg;
for (; i < msg.length; i++)
res += zero2(msg[i].toString(16));
return res;
}
utils.toHex = toHex;
utils.isHex = function isHex(msg) {
return typeof msg === 'string' && /^[0-9a-f]+$/i.test(msg);
};
function binaryInsert(list, item, compare, search) {
var start = 0;
var end = list.length;
var pos, cmp;
while (start < end) {
pos = (start + end) >> 1;
cmp = compare(item, list[pos]);
if (cmp === 0) {
start = pos;
end = pos;
break;
} else if (cmp < 0) {
end = pos;
} else {
start = pos + 1;
}
}
if (!search)
list.splice(start, 0, item);
return start;
}
utils.binaryInsert = binaryInsert;
function bitsToTarget(bits) {
var len = (bits >>> 24) & 0xff;
var hi = (bits >>> 16) & 0xff;
var mid = (bits >>> 8) & 0xff;
var lo = bits & 0xff;
var res = new Array(len);
var i = 0;
for (; i < len - 3; i++)
res[i] = 0;
res[i++] = lo;
res[i++] = mid;
res[i++] = hi;
if (hi === 0)
res.pop();
if (hi === 0 && mid === 0)
res.pop();
return res;
}
utils.bitsToTarget = bitsToTarget;
function testTarget(target, hash) {
var i;
if (typeof target === 'number')
target = bitsToTarget(target);
hash = utils.toArray(hash, 'hex');
for (i = hash.length - 1; i >= target.length; i--)
if (hash[i] !== 0)
return false;
for (; i >= 0; i--) {
if (hash[i] === target[i])
continue;
if (hash[i] > target[i])
return false;
break;
}
return true;
}
utils.testTarget = testTarget;
utils.isEqual = function isEqual(a, b) {
var i = 0;
if (a.length !== b.length)
return false;
for (; i < a.length; i++)
if (a[i] !== b[i])
return false;
return true;
};
utils.nextTick = function nextTick(fn) {
if (typeof setImmediate === 'function') {
setImmediate(fn);
return;
}
if (typeof process === 'object' && process.nextTick) {
process.nextTick(fn);
return;
}
setTimeout(fn, 1);
};
function RequestCache() {
this.map = {};
this.count = 0;
}
RequestCache.prototype.add = function add(id, cb) {
if (this.map[id]) {
this.map[id].push(cb);
return false;
} else {
this.map[id] = [ cb ];
this.count++;
return true;
}
};
RequestCache.prototype.fullfill = function fullfill(id, err, data) {
var cbs = this.map[id];
if (!this.map[id])
return;
delete this.map[id];
this.count--;
cbs.forEach(function(cb) {
cb(err, data);
});
};
utils.RequestCache = RequestCache;
utils.asyncify = function asyncify(fn) {
return function _asynicifedFn(err, data1, data2) {
if (!fn)
return err || data1;
utils.nextTick(function() {
fn(err, data1, data2);
});
};
};
utils.revHex = function revHex(s) {
var r = '';
var i = 0;
for (; i < s.length; i += 2)
r = s.slice(i, i + 2) + r;
return r;
};
utils.assert = function assert(val, msg) {
if (!val)
throw new Error(msg || 'Assertion failed');
};
utils.assert.equal = function assertEqual(l, r, msg) {
if (l != r)
throw new Error(msg || ('Assertion failed: ' + l + ' != ' + r));
};
utils.toBTC = function toBTC(satoshi, strict) {
var m = new bn(10000000).mul(new bn(10));
var lo;
if (utils.isBTC(satoshi))
return utils.isBTC(satoshi);
if (!strict && utils.isFinite(satoshi))
satoshi = new bn(Math.floor(satoshi) + '', 10);
satoshi = utils.isSatoshi(satoshi);
if (!satoshi)
throw new Error('Could not calculate BTC');
lo = satoshi.mod(m);
if (lo.cmpn(0) !== 0) {
lo = lo.toString(10);
while (lo.length < 8)
lo = '0' + lo;
lo = '.' + lo;
} else {
lo = '.0';
}
lo = lo.replace(/0+$/, '');
if (lo === '.')
lo += '0';
return satoshi.div(m).toString(10) + lo;
};
utils.fromBTC = function fromBTC(btc, strict) {
var satoshi, parts, hi, lo;
if (utils.isSatoshi(btc))
return utils.isSatoshi(btc);
if (!strict && utils.isFinite(btc)) {
btc = btc + '';
if (utils.isInt(btc))
btc += '.0';
}
btc = utils.isBTC(btc);
if (!btc)
throw new Error('Could not calculate satoshis');
parts = btc.split('.');
hi = parts[0] || '0';
lo = parts[1] || '0';
while (lo.length < 8)
lo += '0';
satoshi = (hi + lo).replace(/^0+/, '');
return new bn(satoshi, 10);
};
utils.isInt = function isInt(val) {
return typeof val === 'string' && /^\d+$/.test(val);
};
utils.isFloat = function isFloat(val) {
return typeof val === 'string' && /^\d+\.\d+$/.test(val);
};
utils.isFinite = function _isFinite(val) {
return typeof val === 'number' && isFinite(val);
};
utils.isSatoshi = function isSatoshi(val) {
if (val instanceof bn)
return val;
if (utils.isInt(val))
return new bn(val, 10);
if (utils.isHex(val))
return new bn(val, 'hex');
if (Array.isArray(val))
return new bn(val);
return false;
};
utils.isBTC = function isBTC(val) {
if (utils.isFloat(val))
return val;
// For user input strings. Might cause overlap
// with isSatoshi if not used carefully.
// if (utils.isInt(val))
// return val;
return false;
};
utils.toFloat = function toFloat(val) {
if (utils.isFloat(val))
return val;
if (utils.isInt(val))
return val + '.0';
throw new Error('Could not convert ' + val + ' to float');
};
utils.isIP = function isIP(ip) {
if (typeof ip !== 'string')
return 0;
if (~ip.indexOf('.'))
return 4;
if (~ip.indexOf(':'))
return 6;
return 0;
};
utils.isArrayLike = function isArrayLike(msg) {
return msg
&& !Array.isArray(msg)
&& typeof msg === 'object'
&& typeof msg.length === 'number';
};
utils.toKeyArray = function toKeyArray(msg) {
if (Array.isArray(msg))
return msg;
if (utils.isArrayLike(msg))
return Array.prototype.slice.call(msg);
if (utils.isHex(msg))
return utils.toArray(msg, 'hex');
if (utils.isBase58(msg))
return utils.fromBase58(msg);
throw new Error('Cannot ensure array');
};
utils.inspect = function inspect(obj) {
return typeof obj !== 'string'
? util.inspect(obj, null, 20, true)
: obj;
};
utils.print = function print(msg) {
return typeof msg === 'object'
? process.stdout.write(utils.inspect(msg) + '\n')
: console.log.apply(console, arguments);
};
utils.debug = function debug() {
var args = Array.prototype.slice.call(arguments);
args[0] = '\x1b[31m' + args[0] + '\x1b[m';
return utils.print.apply(null, args);
};
utils.merge = function merge(target) {
var args = Array.prototype.slice.call(arguments, 1);
args.forEach(function(obj) {
Object.keys(obj).forEach(function(key) {
target[key] = obj[key];
});
});
return target;
};
utils.hidden = function hidden(obj, prop, value) {
Object.defineProperty(obj, prop, {
value: value,
enumerable: false,
configurable: true,
writable: true
});
return obj;
};
utils.sortKeys = function sortKeys(keys) {
return keys.sort(function(a, b) {
return new bn(a).cmp(new bn(b)) > 0;
});
};
utils.readIntv = function readIntv(p, off) {
var r, bytes;
if (!off)
off = 0;
if (p[off] < 0xfd) {
r = p[off];
bytes = 1;
} else if (p[off] === 0xfd) {
r = p[off + 1] | (p[off + 2] << 8);
bytes = 3;
} else if (p[off] === 0xfe) {
r = utils.readU32(p, off + 1);
bytes = 5;
} else {
r = 0;
bytes = 9;
}
return { off: off + bytes, r: r };
};
utils.writeIntv = function writeIntv(arr, value, off) {
if (!off)
off = 0;
if (value < 0xfd) {
arr[off] = value;
return 1;
}
if (value <= 0xffff) {
arr[off] = 0xfd;
arr[off + 1] = value & 0xff;
arr[off + 2] = value >>> 8;
return 3;
}
if (value <= 0xffffffff) {
arr[off] = 0xfe;
arr[off + 1] = value & 0xff;
arr[off + 2] = (value >>> 8) & 0xff;
arr[off + 3] = (value >>> 16) & 0xff;
arr[off + 4] = value >>> 24;
return 5;
}
arr[off] = 0xff;
utils.writeU64(arr, value, off + 1);
return 9;
};