refactor work queue

This commit is contained in:
Florian Dold 2016-10-13 20:02:42 +02:00
parent d3b49c0a2f
commit 0b198e0888
7 changed files with 222 additions and 214 deletions

View File

@ -28,11 +28,6 @@ import {CoinWithDenom} from "./wallet";
import {PayCoinInfo} from "./types"; import {PayCoinInfo} from "./types";
import {RefreshSession} from "./types"; import {RefreshSession} from "./types";
interface RegistryEntry {
resolve: any;
reject: any;
workerIndex: number;
}
interface WorkerState { interface WorkerState {
/** /**
@ -41,19 +36,14 @@ interface WorkerState {
w: Worker|null; w: Worker|null;
/** /**
* Are we currently running a task on this worker? * Work we're currently executing or null if not busy.
*/ */
busy: boolean; currentWorkItem: WorkItem|null;
/** /**
* Timer to terminate the worker if it's not busy enough. * Timer to terminate the worker if it's not busy enough.
*/ */
terminationTimerHandle: number|null; terminationTimerHandle: number|null;
/**
* Index of this worker in the list of workers.
*/
workerIndex: number;
} }
interface WorkItem { interface WorkItem {
@ -61,6 +51,11 @@ interface WorkItem {
args: any[]; args: any[];
resolve: any; resolve: any;
reject: any; reject: any;
/**
* Serial id to identify a matching response.
*/
rpcId: number;
} }
@ -72,27 +67,21 @@ const NUM_PRIO = 5;
export class CryptoApi { export class CryptoApi {
private nextRpcId: number = 1; private nextRpcId: number = 1;
private rpcRegistry: {[n: number]: RegistryEntry} = {};
private workers: WorkerState[]; private workers: WorkerState[];
private workQueues: WorkItem[][]; private workQueues: WorkItem[][];
/** /**
* Number of busy workers. * Number of busy workers.
*/ */
private numBusy: number = 0; private numBusy: number = 0;
/**
* Number if pending work items.
*/
private numWaiting: number = 0;
/** /**
* Start a worker (if not started) and set as busy. * Start a worker (if not started) and set as busy.
*/ */
wake(ws: WorkerState) { wake<T>(ws: WorkerState, work: WorkItem): void {
if (ws.busy) { if (ws.currentWorkItem != null) {
throw Error("assertion failed"); throw Error("assertion failed");
} }
ws.busy = true; ws.currentWorkItem = work;
this.numBusy++; this.numBusy++;
if (!ws.w) { if (!ws.w) {
let w = new Worker("/lib/wallet/cryptoWorker.js"); let w = new Worker("/lib/wallet/cryptoWorker.js");
@ -100,27 +89,45 @@ export class CryptoApi {
w.onerror = (e: ErrorEvent) => this.handleWorkerError(ws, e); w.onerror = (e: ErrorEvent) => this.handleWorkerError(ws, e);
ws.w = w; ws.w = w;
} }
let msg: any = {
operation: work.operation, args: work.args,
id: work.rpcId
};
this.resetWorkerTimeout(ws);
ws.w!.postMessage(msg);
}
resetWorkerTimeout(ws: WorkerState) {
if (ws.terminationTimerHandle != null) { if (ws.terminationTimerHandle != null) {
clearTimeout(ws.terminationTimerHandle); clearTimeout(ws.terminationTimerHandle);
} }
let destroy = () => { let destroy = () => {
if (ws.w && !ws.busy) { // terminate worker if it's idle
if (ws.w && ws.currentWorkItem == null) {
ws.w!.terminate(); ws.w!.terminate();
ws.w = null; ws.w = null;
} }
} };
ws.terminationTimerHandle = setTimeout(destroy, 20 * 1000); ws.terminationTimerHandle = setTimeout(destroy, 20 * 1000);
} }
handleWorkerError(ws: WorkerState, e: ErrorEvent) { handleWorkerError(ws: WorkerState, e: ErrorEvent) {
if (ws.currentWorkItem) {
console.error(`error in worker during ${ws.currentWorkItem!.operation}`, e);
} else {
console.error("error in worker", e); console.error("error in worker", e);
}
console.error(e.message);
try { try {
ws.w!.terminate(); ws.w!.terminate();
ws.w = null;
} catch (e) { } catch (e) {
console.error(e); console.error(e);
} }
if (ws.busy) { if (ws.currentWorkItem != null) {
ws.busy = false; ws.currentWorkItem.reject(e);
ws.currentWorkItem = null;
this.numBusy--; this.numBusy--;
} }
this.findWork(ws); this.findWork(ws);
@ -132,14 +139,7 @@ export class CryptoApi {
let q = this.workQueues[NUM_PRIO - i - 1]; let q = this.workQueues[NUM_PRIO - i - 1];
if (q.length != 0) { if (q.length != 0) {
let work: WorkItem = q.shift()!; let work: WorkItem = q.shift()!;
let msg: any = { this.wake(ws, work);
operation: work.operation,
args: work.args,
id: this.registerRpcId(work.resolve, work.reject, ws.workerIndex),
};
this.wake(ws);
ws.w!.postMessage(msg);
return; return;
} }
} }
@ -151,22 +151,19 @@ export class CryptoApi {
console.error("rpc id must be number"); console.error("rpc id must be number");
return; return;
} }
if (!this.rpcRegistry[id]) { let currentWorkItem = ws.currentWorkItem;
ws.currentWorkItem = null;
this.numBusy--;
this.findWork(ws);
if (!currentWorkItem) {
console.error("unsolicited response from worker");
return;
}
if (id != currentWorkItem.rpcId) {
console.error(`RPC with id ${id} has no registry entry`); console.error(`RPC with id ${id} has no registry entry`);
return; return;
} }
let {resolve, workerIndex} = this.rpcRegistry[id]; currentWorkItem.resolve(msg.data.result);
delete this.rpcRegistry[id];
if (workerIndex != ws.workerIndex) {
throw Error("assertion failed");
}
if (!ws.busy) {
throw Error("assertion failed");
}
ws.busy = false;
this.numBusy--;
resolve(msg.data.result);
this.findWork(ws);
} }
constructor() { constructor() {
@ -175,9 +172,8 @@ export class CryptoApi {
for (let i = 0; i < this.workers.length; i++) { for (let i = 0; i < this.workers.length; i++) {
this.workers[i] = { this.workers[i] = {
w: null, w: null,
busy: false,
terminationTimerHandle: null, terminationTimerHandle: null,
workerIndex: i, currentWorkItem: null,
}; };
} }
this.workQueues = []; this.workQueues = [];
@ -186,45 +182,34 @@ export class CryptoApi {
} }
} }
private registerRpcId(resolve: any, reject: any,
workerIndex: number): number {
let id = this.nextRpcId++;
this.rpcRegistry[id] = {resolve, reject, workerIndex};
return id;
}
private doRpc<T>(operation: string, priority: number, private doRpc<T>(operation: string, priority: number,
...args: any[]): Promise<T> { ...args: any[]): Promise<T> {
return new Promise((resolve, reject) => {
let rpcId = this.nextRpcId++;
let workItem: WorkItem = {operation, args, resolve, reject, rpcId};
if (this.numBusy == this.workers.length) { if (this.numBusy == this.workers.length) {
let q = this.workQueues[priority]; let q = this.workQueues[priority];
if (!q) { if (!q) {
throw Error("assertion failed"); throw Error("assertion failed");
} }
return new Promise<T>((resolve, reject) => { this.workQueues[priority].push(workItem);
this.workQueues[priority].push({operation, args, resolve, reject}); return;
});
} }
for (let i = 0; i < this.workers.length; i++) { for (let i = 0; i < this.workers.length; i++) {
let ws = this.workers[i]; let ws = this.workers[i];
if (ws.busy) { if (ws.currentWorkItem != null) {
continue; continue;
} }
this.wake(ws); this.wake<T>(ws, workItem);
return;
return new Promise<T>((resolve, reject) => {
let msg: any = {
operation, args,
id: this.registerRpcId(resolve, reject, i),
};
ws.w!.postMessage(msg);
});
} }
throw Error("assertion failed"); throw Error("assertion failed");
});
} }

View File

@ -79,10 +79,14 @@ namespace RpcFunctions {
let coinPub = coinPriv.getPublicKey(); let coinPub = coinPriv.getPublicKey();
let blindingFactor = native.RsaBlindingKeySecret.create(); let blindingFactor = native.RsaBlindingKeySecret.create();
let pubHash: native.HashCode = coinPub.hash(); let pubHash: native.HashCode = coinPub.hash();
let ev: native.ByteArray = native.rsaBlind(pubHash, let ev = native.rsaBlind(pubHash,
blindingFactor, blindingFactor,
denomPub); denomPub);
if (!ev) {
throw Error("couldn't blind (malicious exchange key?)");
}
if (!denom.fee_withdraw) { if (!denom.fee_withdraw) {
throw Error("Field fee_withdraw missing"); throw Error("Field fee_withdraw missing");
} }
@ -234,7 +238,7 @@ namespace RpcFunctions {
} }
function createWithdrawSession(kappa: number, meltCoin: Coin, export function createWithdrawSession(kappa: number, meltCoin: Coin,
newCoinDenoms: Denomination[], newCoinDenoms: Denomination[],
meltAmount: AmountJson, meltAmount: AmountJson,
meltFee: AmountJson): RefreshSession { meltFee: AmountJson): RefreshSession {
@ -268,9 +272,12 @@ namespace RpcFunctions {
let blindingFactor = native.RsaBlindingKeySecret.create(); let blindingFactor = native.RsaBlindingKeySecret.create();
let pubHash: native.HashCode = coinPub.hash(); let pubHash: native.HashCode = coinPub.hash();
let denomPub = native.RsaPublicKey.fromCrock(newCoinDenoms[i].denom_pub); let denomPub = native.RsaPublicKey.fromCrock(newCoinDenoms[i].denom_pub);
let ev: native.ByteArray = native.rsaBlind(pubHash, let ev = native.rsaBlind(pubHash,
blindingFactor, blindingFactor,
denomPub); denomPub);
if (!ev) {
throw Error("couldn't blind (malicious exchange key?)");
}
let preCoin: RefreshPreCoin = { let preCoin: RefreshPreCoin = {
blindingKey: blindingFactor.toCrock(), blindingKey: blindingFactor.toCrock(),
coinEv: ev.toCrock(), coinEv: ev.toCrock(),

View File

@ -22,7 +22,7 @@
"use strict"; "use strict";
importScripts("../emscripten/libwrapper.js", importScripts("../emscripten/taler-emscripten-lib.js",
"../vendor/system-csp-production.src.js"); "../vendor/system-csp-production.src.js");
@ -46,7 +46,7 @@ if ("object" !== typeof Module) {
{ {
let mod = System.newModule({Module: Module}); let mod = System.newModule({Module: Module});
let modName = System.normalizeSync("../emscripten/emsc"); let modName = System.normalizeSync("../emscripten/taler-emscripten-lib");
console.log("registering", modName); console.log("registering", modName);
System.set(modName, mod); System.set(modName, mod);
} }

View File

@ -14,8 +14,8 @@
TALER; see the file COPYING. If not, see <http://www.gnu.org/licenses/> TALER; see the file COPYING. If not, see <http://www.gnu.org/licenses/>
*/ */
import { AmountJson } from "./types"; import {AmountJson} from "./types";
import * as EmscWrapper from "../emscripten/emsc"; import * as EmscWrapper from "../emscripten/taler-emscripten-lib";
/** /**
* High-level interface to emscripten-compiled modules used * High-level interface to emscripten-compiled modules used
@ -137,7 +137,7 @@ var emscAlloc = {
['number', 'number', 'number']), ['number', 'number', 'number']),
rsa_blind: getEmsc('GNUNET_CRYPTO_rsa_blind', rsa_blind: getEmsc('GNUNET_CRYPTO_rsa_blind',
'number', 'number',
['number', 'number', 'number', 'number']), ['number', 'number', 'number', 'number', 'number']),
rsa_blinding_key_create: getEmsc('GNUNET_CRYPTO_rsa_blinding_key_create', rsa_blinding_key_create: getEmsc('GNUNET_CRYPTO_rsa_blinding_key_create',
'number', 'number',
['number']), ['number']),
@ -519,7 +519,7 @@ abstract class PackedArenaObject extends MallocArenaObject {
this.size()); this.size());
buf.destroy(); buf.destroy();
if (res < 1) { if (res < 1) {
throw { error: "wrong encoding" }; throw {error: "wrong encoding"};
} }
} }
@ -618,7 +618,6 @@ export class EcdsaPrivateKey extends PackedArenaObject {
mixinStatic(EcdsaPrivateKey, fromCrock); mixinStatic(EcdsaPrivateKey, fromCrock);
function fromCrock(s: string) { function fromCrock(s: string) {
let x = new this(); let x = new this();
x.alloc(); x.alloc();
@ -1123,13 +1122,23 @@ mixin(RsaSignature, makeEncode(emscAlloc.rsa_signature_encode));
export function rsaBlind(hashCode: HashCode, export function rsaBlind(hashCode: HashCode,
blindingKey: RsaBlindingKeySecret, blindingKey: RsaBlindingKeySecret,
pkey: RsaPublicKey, pkey: RsaPublicKey,
arena?: Arena): ByteArray { arena?: Arena): ByteArray|null {
let ptr = emscAlloc.malloc(PTR_SIZE); let buf_ptr_out = emscAlloc.malloc(PTR_SIZE);
let s = emscAlloc.rsa_blind(hashCode.nativePtr, let buf_size_out = emscAlloc.malloc(PTR_SIZE);
let res = emscAlloc.rsa_blind(hashCode.nativePtr,
blindingKey.nativePtr, blindingKey.nativePtr,
pkey.nativePtr, pkey.nativePtr,
ptr); buf_ptr_out,
return new ByteArray(s, Module.getValue(ptr, '*'), arena); buf_size_out);
let buf_ptr = Module.getValue(buf_ptr_out, '*');
let buf_size = Module.getValue(buf_size_out, '*');
emsc.free(buf_ptr_out);
emsc.free(buf_size_out);
if (res != GNUNET_OK) {
// malicious key
return null;
}
return new ByteArray(buf_size, buf_ptr, arena);
} }
@ -1208,16 +1217,19 @@ export interface FreshCoin {
blindingKey: RsaBlindingKeySecret; blindingKey: RsaBlindingKeySecret;
} }
export function setupFreshCoin(secretSeed: TransferSecretP, coinIndex: number): FreshCoin { export function setupFreshCoin(secretSeed: TransferSecretP,
coinIndex: number): FreshCoin {
let priv = new EddsaPrivateKey(); let priv = new EddsaPrivateKey();
priv.isWeak = true; priv.isWeak = true;
let blindingKey = new RsaBlindingKeySecret(); let blindingKey = new RsaBlindingKeySecret();
blindingKey.isWeak = true; blindingKey.isWeak = true;
let buf = kdf(priv.size() + blindingKey.size(), UInt32.fromNumber(coinIndex), ByteArray.fromString("taler-coin-derivation")); let buf = kdf(priv.size() + blindingKey.size(),
UInt32.fromNumber(coinIndex),
ByteArray.fromString("taler-coin-derivation"));
priv.nativePtr = buf.nativePtr; priv.nativePtr = buf.nativePtr;
blindingKey.nativePtr = buf.nativePtr + priv.size(); blindingKey.nativePtr = buf.nativePtr + priv.size();
return { priv, blindingKey }; return {priv, blindingKey};
} }

View File

@ -797,8 +797,6 @@ export class Wallet {
} }
async storeCoin(coin: Coin): Promise<void> { async storeCoin(coin: Coin): Promise<void> {
console.log("storing coin", new Date());
let historyEntry: HistoryRecord = { let historyEntry: HistoryRecord = {
type: "withdraw", type: "withdraw",
timestamp: (new Date).getTime(), timestamp: (new Date).getTime(),

View File

@ -286,7 +286,6 @@ class ChromeNotifier implements Notifier {
} }
notify() { notify() {
console.log("notifying all ports");
for (let p of this.ports) { for (let p of this.ports) {
p.postMessage({ notify: true }); p.postMessage({ notify: true });
} }

View File

@ -172,7 +172,7 @@ class ExchangeSelection extends ImplicitStateComponent<ExchangeSelectionProps> {
renderAdvanced(): JSX.Element { renderAdvanced(): JSX.Element {
if (this.detailCollapsed() && this.url() !== null) { if (this.detailCollapsed() && this.url() !== null && !this.statusString()) {
return ( return (
<button className="linky" <button className="linky"
onClick={() => this.detailCollapsed(false)}> onClick={() => this.detailCollapsed(false)}>
@ -224,6 +224,13 @@ class ExchangeSelection extends ImplicitStateComponent<ExchangeSelectionProps> {
<em>{shortName}</em> <em>{shortName}</em>
</p>; </p>;
} }
if (this.statusString()) {
return (
<p>
<strong style="color: red;">A problem occured, see below.</strong>
</p>
);
}
return ( return (
<p> <p>
Information about fees will be available when an exchange provider is selected. Information about fees will be available when an exchange provider is selected.