itns-sidechain/lib/script/sigcache.js

/*!
 * sigcache.js - signature cache for bcoin
 * Copyright (c) 2014-2017, Christopher Jeffrey (MIT License).
 * https://github.com/bcoin-org/bcoin
 */

'use strict';

var assert = require('assert');
var util = require('../utils/util');
var ec = require('../crypto/ec');

/**
 * Signature cache.
 * @alias module:script.SigCache
 * @constructor
 * @param {Number} [size=50000]
 * @property {Number} size
 * @property {Hash[]} keys
 * @property {Object} valid
 */

function SigCache(size) {
  if (!(this instanceof SigCache))
    return new SigCache(size);

  if (size == null)
    size = 10000;

  assert(util.isNumber(size));
  assert(size >= 0);

  this.size = size;
  this.keys = [];
  this.valid = {};
}

/**
 * Resize the sigcache.
 * @param {Number} size
 */

SigCache.prototype.resize = function resize(size) {
  assert(util.isNumber(size));
  assert(size >= 0);

  this.size = size;
  this.keys.length = 0;
  this.valid = {};
};

/**
 * Add item to the sigcache.
 * Potentially evict a random member.
 * @param {Hash} hash - Sig hash.
 * @param {Buffer} sig
 * @param {Buffer} key
 */

SigCache.prototype.add = function add(hash, sig, key) {
  var i, k;

  if (this.size === 0)
    return;

  this.valid[hash] = new SigCacheEntry(sig, key);

  if (this.keys.length >= this.size) {
    i = Math.floor(Math.random() * this.keys.length);
    k = this.keys[i];
    delete this.valid[k];
    this.keys[i] = hash;
  } else {
    this.keys.push(hash);
  }
};

/**
 * Test whether the sig exists.
 * @param {Hash} hash - Sig hash.
 * @param {Buffer} sig
 * @param {Buffer} key
 * @returns {Boolean}
 */

SigCache.prototype.has = function has(hash, sig, key) {
  var entry = this.valid[hash];

  if (!entry)
    return false;

  return entry.equal(sig, key);
};

/**
 * Verify a signature, testing
 * it against the cache first.
 * @param {Buffer} msg
 * @param {Buffer} sig
 * @param {Buffer} key
 * @param {Boolean?} historical
 * @param {Boolean?} high
 * @returns {Boolean}
 */

SigCache.prototype.verify = function verify(msg, sig, key, historical, high) {
  var hash, result;

  if (historical || this.size === 0)
    return ec.verify(msg, sig, key, historical, high);

  hash = msg.toString('hex');

  if (this.has(hash, sig, key))
    return true;

  result = ec.verify(msg, sig, key, historical, high);

  if (!result)
    return false;

  this.add(hash, sig, key);

  return true;
};

/**
 * Signature cache entry.
 * @constructor
 * @ignore
 * @param {Buffer} sig
 * @param {Buffer} key
 * @property {Buffer} sig
 * @property {Buffer} key
 */

function SigCacheEntry(sig, key) {
  this.sig = sig;
  this.key = key;
}

/**
 * Compare an entry to a sig and key.
 * @param {Buffer} sig
 * @param {Buffer} key
 * @returns {Boolean}
 */

SigCacheEntry.prototype.equal = function equal(sig, key) {
  return util.equal(this.sig, sig) && util.equal(this.key, key);
};

/*
 * Expose
 */

module.exports = new SigCache(+process.env.BCOIN_SIGCACHE_SIZE || 0);