scripts: add a fuzzer.
This commit is contained in:
Christopher Jeffrey
parent
8b035fd9a8
commit
60897be916
450
scripts/fuzz.js
Normal file
450
scripts/fuzz.js
Normal file
@ -0,0 +1,450 @@
|
||||
var assert = require('assert');
|
||||
var bcoin = require('../');
|
||||
var util = bcoin.util;
|
||||
var Script = bcoin.script;
|
||||
var Stack = bcoin.stack;
|
||||
var Witness = bcoin.witness;
|
||||
var Input = bcoin.input;
|
||||
var Output = bcoin.output;
|
||||
var Outpoint = bcoin.outpoint;
|
||||
var TX = bcoin.tx;
|
||||
var crypto = bcoin.crypto;
|
||||
var consensus = bcoin.consensus;
|
||||
|
||||
var MANDATORY = Script.flags.MANDATORY_VERIFY_FLAGS | Script.flags.VERIFY_WITNESS;
|
||||
var STANDARD = Script.flags.STANDARD_VERIFY_FLAGS;
|
||||
|
||||
function randomOutpoint() {
|
||||
var hash = crypto.randomBytes(32).toString('hex');
|
||||
return new Outpoint(hash, util.random(0, 0xffffffff));
|
||||
}
|
||||
|
||||
function randomInput() {
|
||||
var input = Input.fromOutpoint(randomOutpoint());
|
||||
|
||||
if (util.random(0, 5) === 0)
|
||||
input.sequence = util.random(0, 0xffffffff);
|
||||
|
||||
return input;
|
||||
}
|
||||
|
||||
function randomOutput() {
|
||||
return Output.fromScript(randomScript(), util.random(0, 1e8));
|
||||
}
|
||||
|
||||
function randomTX() {
|
||||
var tx = new TX();
|
||||
var inputs = util.random(1, 5);
|
||||
var outputs = util.random(0, 5);
|
||||
var i;
|
||||
|
||||
tx.version = util.random(0, 0xffffffff);
|
||||
|
||||
for (i = 0; i < inputs; i++)
|
||||
tx.inputs.push(randomInput());
|
||||
|
||||
for (i = 0; i < outputs; i++)
|
||||
tx.inputs.push(randomOutput());
|
||||
|
||||
if (util.random(0, 5) === 0)
|
||||
tx.locktime = util.random(0, 0xffffffff);
|
||||
|
||||
tx.refresh();
|
||||
|
||||
return tx;
|
||||
}
|
||||
|
||||
function randomWitness(redeem) {
|
||||
var size = util.random(1, 100);
|
||||
var witness = new Witness();
|
||||
var i, len;
|
||||
|
||||
for (i = 0; i < size; i++) {
|
||||
len = util.random(0, 100);
|
||||
witness.push(crypto.randomBytes(len));
|
||||
}
|
||||
|
||||
if (redeem)
|
||||
witness.push(redeem);
|
||||
|
||||
witness.compile();
|
||||
|
||||
return witness;
|
||||
}
|
||||
|
||||
function randomInputScript(redeem) {
|
||||
var size = util.random(1, 100);
|
||||
var script = new Script();
|
||||
var i, len;
|
||||
|
||||
for (i = 0; i < size; i++) {
|
||||
len = util.random(0, 100);
|
||||
script.push(crypto.randomBytes(len));
|
||||
}
|
||||
|
||||
if (redeem)
|
||||
script.push(redeem);
|
||||
|
||||
script.compile();
|
||||
|
||||
return script;
|
||||
}
|
||||
|
||||
function randomOutputScript() {
|
||||
var size = util.random(1, 10000);
|
||||
return Script.fromRaw(crypto.randomBytes(size));
|
||||
}
|
||||
|
||||
function isPushOnly(script) {
|
||||
var i, op;
|
||||
|
||||
if (script.isPushOnly())
|
||||
return true;
|
||||
|
||||
for (i = 0; i < script.code.length; i++) {
|
||||
op = script.code[i];
|
||||
|
||||
if (op.value === Script.opcodes.NOP)
|
||||
continue;
|
||||
|
||||
if (op.value === Script.opcodes.NOP_1)
|
||||
continue;
|
||||
|
||||
if (op.value > Script.opcodes.NOP_3)
|
||||
continue;
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
function randomPubkey() {
|
||||
var len = util.random(0, 2) === 0 ? 33 : 65;
|
||||
return Script.fromPubkey(crypto.randomBytes(len));
|
||||
}
|
||||
|
||||
function randomPubkeyhash() {
|
||||
return Script.fromPubkeyhash(crypto.randomBytes(20));
|
||||
}
|
||||
|
||||
function randomMultisig() {
|
||||
var n = util.random(1, 16);
|
||||
var m = util.random(1, n);
|
||||
var keys = [];
|
||||
var i, len;
|
||||
|
||||
for (i = 0; i < n; i++) {
|
||||
len = util.random(0, 2) === 0 ? 33 : 65;
|
||||
keys.push(crypto.randomBytes(len));
|
||||
}
|
||||
|
||||
return Script.fromMultisig(m, n, keys);
|
||||
}
|
||||
|
||||
function randomScripthash() {
|
||||
return Script.fromScripthash(crypto.randomBytes(20));
|
||||
}
|
||||
|
||||
function randomWitnessPubkeyhash() {
|
||||
return Script.fromProgram(0, crypto.randomBytes(20));
|
||||
}
|
||||
|
||||
function randomWitnessScripthash() {
|
||||
return Script.fromProgram(0, crypto.randomBytes(32));
|
||||
}
|
||||
|
||||
function randomProgram() {
|
||||
var version = util.random(0, 16);
|
||||
var size = util.random(2, 41);
|
||||
return Script.fromProgram(version, crypto.randomBytes(size));
|
||||
}
|
||||
|
||||
function randomRedeem() {
|
||||
switch (util.random(0, 5)) {
|
||||
case 0:
|
||||
return randomPubkey();
|
||||
case 1:
|
||||
return randomPubkeyhash();
|
||||
case 2:
|
||||
return randomMultisig();
|
||||
case 3:
|
||||
return randomWitnessPubkeyhash();
|
||||
case 4:
|
||||
return randomProgram();
|
||||
}
|
||||
assert(false);
|
||||
}
|
||||
|
||||
function randomScript() {
|
||||
switch (util.random(0, 7)) {
|
||||
case 0:
|
||||
return randomPubkey();
|
||||
case 1:
|
||||
return randomPubkeyhash();
|
||||
case 2:
|
||||
return randomMultisig();
|
||||
case 3:
|
||||
return randomScripthash();
|
||||
case 4:
|
||||
return randomWitnessPubkeyhash();
|
||||
case 5:
|
||||
return randomWitnessScripthash();
|
||||
case 6:
|
||||
return randomProgram();
|
||||
}
|
||||
assert(false);
|
||||
}
|
||||
|
||||
function randomPubkeyContext() {
|
||||
return {
|
||||
input: randomInputScript(),
|
||||
witness: new Witness(),
|
||||
output: randomPubkey(),
|
||||
redeem: null
|
||||
};
|
||||
}
|
||||
|
||||
function randomPubkeyhashContext() {
|
||||
return {
|
||||
input: randomInputScript(),
|
||||
witness: new Witness(),
|
||||
output: randomPubkeyhash(),
|
||||
redeem: null
|
||||
};
|
||||
}
|
||||
|
||||
function randomScripthashContext() {
|
||||
var redeem = randomRedeem();
|
||||
return {
|
||||
input: randomInputScript(redeem.toRaw()),
|
||||
witness: new Witness(),
|
||||
output: Script.fromScripthash(redeem.hash160()),
|
||||
redeem: redeem
|
||||
};
|
||||
}
|
||||
|
||||
function randomWitnessPubkeyhashContext() {
|
||||
return {
|
||||
input: new Script(),
|
||||
witness: randomWitness(),
|
||||
output: randomWitnessPubkeyhash(),
|
||||
redeem: null
|
||||
};
|
||||
}
|
||||
|
||||
function randomWitnessScripthashContext() {
|
||||
var redeem = randomRedeem();
|
||||
return {
|
||||
input: new Script(),
|
||||
witness: randomWitness(redeem.toRaw()),
|
||||
output: Script.fromProgram(0, redeem.sha256()),
|
||||
redeem: redeem
|
||||
};
|
||||
}
|
||||
|
||||
function randomWitnessNestedContext() {
|
||||
var redeem = randomRedeem();
|
||||
var program = Script.fromProgram(0, redeem.sha256());
|
||||
return {
|
||||
input: new Script([program.toRaw()]),
|
||||
witness: randomWitness(redeem.toRaw()),
|
||||
output: Script.fromScripthash(program.hash160()),
|
||||
redeem: redeem
|
||||
};
|
||||
}
|
||||
|
||||
function randomContext() {
|
||||
switch (util.random(0, 6)) {
|
||||
case 0:
|
||||
return randomPubkeyContext();
|
||||
case 1:
|
||||
return randomPubkeyhashContext();
|
||||
case 2:
|
||||
return randomScripthashContext();
|
||||
case 3:
|
||||
return randomWitnessPubkeyhashContext();
|
||||
case 4:
|
||||
return randomWitnessScripthashContext();
|
||||
case 5:
|
||||
return randomWitnessNestedContext();
|
||||
}
|
||||
assert(false);
|
||||
}
|
||||
|
||||
function fuzzSimple(flags) {
|
||||
var tx = randomTX();
|
||||
var total = -1;
|
||||
var stack, input, output;
|
||||
|
||||
for (;;) {
|
||||
if (++total % 1000 === 0)
|
||||
util.log('Fuzzed %d scripts.', total);
|
||||
|
||||
if (total % 500 === 0)
|
||||
tx = randomTX();
|
||||
|
||||
stack = new Stack();
|
||||
input = randomInputScript();
|
||||
|
||||
try {
|
||||
input.execute(stack, flags, tx, 0, 0, 0);
|
||||
} catch (e) {
|
||||
if (e.type === 'ScriptError')
|
||||
continue;
|
||||
throw e;
|
||||
}
|
||||
|
||||
output = randomOutputScript();
|
||||
|
||||
try {
|
||||
output.execute(stack, flags, tx, 0, 0, 0);
|
||||
} catch (e) {
|
||||
if (e.type === 'ScriptError')
|
||||
continue;
|
||||
throw e;
|
||||
}
|
||||
|
||||
if (stack.length === 0)
|
||||
continue;
|
||||
|
||||
if (!Script.bool(stack.top(-1)))
|
||||
continue;
|
||||
|
||||
if (isPushOnly(output))
|
||||
continue;
|
||||
|
||||
util.log('Produced valid scripts:');
|
||||
|
||||
util.log('Input:');
|
||||
util.log(input);
|
||||
|
||||
util.log('Output:');
|
||||
util.log(output);
|
||||
|
||||
util.log('Stack:');
|
||||
util.log(stack);
|
||||
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
function fuzzVerify(flags) {
|
||||
var tx = randomTX();
|
||||
var total = -1;
|
||||
var input, output, witness;
|
||||
|
||||
for (;;) {
|
||||
if (++total % 1000 === 0)
|
||||
util.log('Fuzzed %d scripts.', total);
|
||||
|
||||
if (total % 500 === 0)
|
||||
tx = randomTX();
|
||||
|
||||
input = randomInputScript();
|
||||
witness = randomWitness();
|
||||
output = randomOutputScript();
|
||||
|
||||
try {
|
||||
Script.verify(
|
||||
input,
|
||||
witness,
|
||||
output,
|
||||
tx,
|
||||
0,
|
||||
0,
|
||||
flags
|
||||
);
|
||||
} catch (e) {
|
||||
if (e.type === 'ScriptError')
|
||||
continue;
|
||||
throw e;
|
||||
}
|
||||
|
||||
if (isPushOnly(output))
|
||||
continue;
|
||||
|
||||
util.log('Produced valid scripts:');
|
||||
|
||||
util.log('Input:');
|
||||
util.log(input);
|
||||
|
||||
util.log('Witness:');
|
||||
util.log(witness);
|
||||
|
||||
util.log('Output:');
|
||||
util.log(output);
|
||||
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
function fuzzLess(flags) {
|
||||
var tx = randomTX();
|
||||
var total = -1;
|
||||
var ctx;
|
||||
|
||||
for (;;) {
|
||||
if (++total % 1000 === 0)
|
||||
util.log('Fuzzed %d scripts.', total);
|
||||
|
||||
if (total % 500 === 0)
|
||||
tx = randomTX();
|
||||
|
||||
ctx = randomContext();
|
||||
|
||||
try {
|
||||
Script.verify(
|
||||
ctx.input,
|
||||
ctx.witness,
|
||||
ctx.output,
|
||||
tx,
|
||||
0,
|
||||
0,
|
||||
flags
|
||||
);
|
||||
} catch (e) {
|
||||
if (e.type === 'ScriptError')
|
||||
continue;
|
||||
throw e;
|
||||
}
|
||||
|
||||
util.log('Produced valid scripts:');
|
||||
|
||||
util.log('Input:');
|
||||
util.log(ctx.input);
|
||||
|
||||
util.log('Witness:');
|
||||
util.log(ctx.witness);
|
||||
|
||||
util.log('Output:');
|
||||
util.log(ctx.output);
|
||||
|
||||
if (ctx.redeem) {
|
||||
util.log('Redeem:');
|
||||
util.log(ctx.redeem);
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
function main() {
|
||||
var flags = process.argv.indexOf('--standard') !== -1 ? STANDARD : MANDATORY;
|
||||
|
||||
switch (process.argv[2]) {
|
||||
case 'simple':
|
||||
return fuzzSimple(flags);
|
||||
case 'verify':
|
||||
return fuzzVerify(flags);
|
||||
case 'less':
|
||||
return fuzzLess(flags);
|
||||
default:
|
||||
util.log('Please select a mode:');
|
||||
util.log('simple, verify, less');
|
||||
util.log('Optional `--standard` flag.');
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
main();
|
||||
Loading…
Reference in New Issue
Block a user