LBRYCommunity/bitcoinjs-lib
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Refactor: pass only cache to certain functions

Browse source
This commit is contained in:
junderw 2019-07-09 12:30:51 +09:00
parent 9749a216b8
commit 2fd4b9dc54
No known key found for this signature in database
GPG key ID: B256185D3A971908
2 changed files with 28 additions and 46 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
src/psbt.js
View file

@ -26,25 +26,24 @@ class Psbt extends bip174_1.Psbt {
};
// set defaults
this.opts = Object.assign({}, DEFAULT_OPTS, opts);
this.__CACHE.__TX = transaction_1.Transaction.fromBuffer(
this.globalMap.unsignedTx,
);
const c = this.__CACHE;
c.__TX = transaction_1.Transaction.fromBuffer(this.globalMap.unsignedTx);
this.setVersion(2);
// set cache
const self = this;
delete this.globalMap.unsignedTx;
Object.defineProperty(this.globalMap, 'unsignedTx', {
enumerable: true,
get() {
if (self.__CACHE.__TX_BUF_CACHE !== undefined) {
return self.__CACHE.__TX_BUF_CACHE;
const buf = c.__TX_BUF_CACHE;
if (buf !== undefined) {
return buf;
} else {
self.__CACHE.__TX_BUF_CACHE = self.__CACHE.__TX.toBuffer();
return self.__CACHE.__TX_BUF_CACHE;
c.__TX_BUF_CACHE = c.__TX.toBuffer();
return c.__TX_BUF_CACHE;
}
},
set(data) {
self.__CACHE.__TX_BUF_CACHE = data;
c.__TX_BUF_CACHE = data;
},
});
// Make data hidden when enumerating
@ -53,8 +52,6 @@ class Psbt extends bip174_1.Psbt {
enumerable,
writable,
});
dpew(this, '__TX', false, true);
dpew(this, '__EXTRACTED_TX', false, true);
dpew(this, '__CACHE', false, true);
dpew(this, 'opts', false, true);
}
@ -216,7 +213,6 @@ class Psbt extends bip174_1.Psbt {
const { script, isP2SH, isP2WSH, isSegwit } = getScriptFromInput(
inputIndex,
input,
this.__CACHE.__TX,
this.__CACHE,
);
if (!script) return false;
@ -258,7 +254,6 @@ class Psbt extends bip174_1.Psbt {
? getHashForSig(
inputIndex,
Object.assign({}, input, { sighashType: sig.hashType }),
this.__CACHE.__TX,
this.__CACHE,
)
: { hash: hashCache, script: scriptCache };
@ -327,7 +322,6 @@ class Psbt extends bip174_1.Psbt {
this.inputs,
inputIndex,
keyPair.publicKey,
this.__CACHE.__TX,
this.__CACHE,
);
const partialSig = {
@ -344,7 +338,6 @@ class Psbt extends bip174_1.Psbt {
this.inputs,
inputIndex,
keyPair.publicKey,
this.__CACHE.__TX,
this.__CACHE,
);
Promise.resolve(keyPair.sign(hash)).then(signature => {
@ -400,14 +393,9 @@ function checkTxInputCache(cache, input) {
function isFinalized(input) {
return !!input.finalScriptSig || !!input.finalScriptWitness;
}
function getHashAndSighashType(inputs, inputIndex, pubkey, unsignedTx, cache) {
function getHashAndSighashType(inputs, inputIndex, pubkey, cache) {
const input = utils_1.checkForInput(inputs, inputIndex);
const { hash, sighashType, script } = getHashForSig(
inputIndex,
input,
unsignedTx,
cache,
);
const { hash, sighashType, script } = getHashForSig(inputIndex, input, cache);
checkScriptForPubkey(pubkey, script, 'sign');
return {
hash,
@ -525,7 +513,8 @@ function checkScriptForPubkey(pubkey, script, action) {
);
}
}
function getHashForSig(inputIndex, input, unsignedTx, cache) {
function getHashForSig(inputIndex, input, cache) {
const unsignedTx = cache.__TX;
const sighashType =
input.sighashType || transaction_1.Transaction.SIGHASH_ALL;
let hash;
@ -646,7 +635,8 @@ function classifyScript(script) {
if (isP2PK(script)) return 'pubkey';
return 'nonstandard';
}
function getScriptFromInput(inputIndex, input, unsignedTx, cache) {
function getScriptFromInput(inputIndex, input, cache) {
const unsignedTx = cache.__TX;
const res = {
script: null,
isSegwit: false,

36
ts_src/psbt.ts
View file

@ -52,6 +52,7 @@ export class Psbt extends PsbtBase {
}
return psbt as InstanceType<T>;
}
static fromBuffer<T extends typeof PsbtBase>(
this: T,
buffer: Buffer,
@ -75,6 +76,7 @@ export class Psbt extends PsbtBase {
checkTxForDupeIns(tx!, psbt.__CACHE);
return psbt as InstanceType<T>;
}
private __CACHE: PsbtCache = {
__NON_WITNESS_UTXO_TX_CACHE: [],
__NON_WITNESS_UTXO_BUF_CACHE: [],
@ -82,28 +84,30 @@ export class Psbt extends PsbtBase {
__TX: new Transaction(),
};
private opts: PsbtOpts;
constructor(opts: PsbtOptsOptional = {}) {
super();
// set defaults
this.opts = Object.assign({}, DEFAULT_OPTS, opts);
this.__CACHE.__TX = Transaction.fromBuffer(this.globalMap.unsignedTx!);
const c = this.__CACHE;
c.__TX = Transaction.fromBuffer(this.globalMap.unsignedTx!);
this.setVersion(2);
// set cache
const self = this;
delete this.globalMap.unsignedTx;
Object.defineProperty(this.globalMap, 'unsignedTx', {
enumerable: true,
get(): Buffer {
if (self.__CACHE.__TX_BUF_CACHE !== undefined) {
return self.__CACHE.__TX_BUF_CACHE;
const buf = c.__TX_BUF_CACHE;
if (buf !== undefined) {
return buf;
} else {
self.__CACHE.__TX_BUF_CACHE = self.__CACHE.__TX.toBuffer();
return self.__CACHE.__TX_BUF_CACHE;
c.__TX_BUF_CACHE = c.__TX.toBuffer();
return c.__TX_BUF_CACHE;
}
},
set(data: Buffer): void {
self.__CACHE.__TX_BUF_CACHE = data;
c.__TX_BUF_CACHE = data;
},
});
@ -118,8 +122,6 @@ export class Psbt extends PsbtBase {
enumerable,
writable,
});
dpew(this, '__TX', false, true);
dpew(this, '__EXTRACTED_TX', false, true);
dpew(this, '__CACHE', false, true);
dpew(this, 'opts', false, true);
}
@ -265,7 +267,6 @@ export class Psbt extends PsbtBase {
const { script, isP2SH, isP2WSH, isSegwit } = getScriptFromInput(
inputIndex,
input,
this.__CACHE.__TX,
this.__CACHE,
);
if (!script) return false;
@ -312,7 +313,6 @@ export class Psbt extends PsbtBase {
? getHashForSig(
inputIndex,
Object.assign({}, input, { sighashType: sig.hashType }),
this.__CACHE.__TX,
this.__CACHE,
)
: { hash: hashCache!, script: scriptCache! };
@ -388,7 +388,6 @@ export class Psbt extends PsbtBase {
this.inputs,
inputIndex,
keyPair.publicKey,
this.__CACHE.__TX,
this.__CACHE,
);
@ -409,7 +408,6 @@ export class Psbt extends PsbtBase {
this.inputs,
inputIndex,
keyPair.publicKey,
this.__CACHE.__TX,
this.__CACHE,
);
@ -513,19 +511,13 @@ function getHashAndSighashType(
inputs: PsbtInput[],
inputIndex: number,
pubkey: Buffer,
unsignedTx: Transaction,
cache: PsbtCache,
): {
hash: Buffer;
sighashType: number;
} {
const input = checkForInput(inputs, inputIndex);
const { hash, sighashType, script } = getHashForSig(
inputIndex,
input,
unsignedTx,
cache,
);
const { hash, sighashType, script } = getHashForSig(inputIndex, input, cache);
checkScriptForPubkey(pubkey, script, 'sign');
return {
hash,
@ -678,9 +670,9 @@ interface HashForSigData {
function getHashForSig(
inputIndex: number,
input: PsbtInput,
unsignedTx: Transaction,
cache: PsbtCache,
): HashForSigData {
const unsignedTx = cache.__TX;
const sighashType = input.sighashType || Transaction.SIGHASH_ALL;
let hash: Buffer;
let script: Buffer;
@ -831,9 +823,9 @@ interface GetScriptReturn {
function getScriptFromInput(
inputIndex: number,
input: PsbtInput,
unsignedTx: Transaction,
cache: PsbtCache,
): GetScriptReturn {
const unsignedTx = cache.__TX;
const res: GetScriptReturn = {
script: null,
isSegwit: false,