lukechilds
/
bitcoinjs-lib
mirror of https://github.com/lukechilds/bitcoinjs-lib.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

Refactor: pass only cache to certain functions

psbt
junderw 6 years ago
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
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 38
      src/psbt.js
  2. 36
      ts_src/psbt.ts

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,

Write Preview
Loading…
Cancel
Save