1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
|
/*
This file is part of TALER
(C) 2016 GNUnet e.V.
TALER is free software; you can redistribute it and/or modify it under the
terms of the GNU General Public License as published by the Free Software
Foundation; either version 3, or (at your option) any later version.
TALER is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE. See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License along with
TALER; see the file COPYING. If not, see <http://www.gnu.org/licenses/>
*/
/**
* Imports
*/
import { CryptoWorkerFactory } from "./cryptoDispatcher.js";
import { CryptoWorker } from "./cryptoWorkerInterface.js";
import os from "os";
import { Logger } from "@gnu-taler/taler-util";
import { nativeCryptoR } from "../cryptoImplementation.js";
import { getErrorDetailFromException } from "../../errors.js";
const logger = new Logger("nodeThreadWorker.ts");
const f = __filename;
const workerCode = `
// Try loading the glue library for embedded
try {
require("akono");
} catch (e) {
try {
require("iono");
} catch (e2) {
// Probably we're not on embedded ...
}
}
const worker_threads = require('worker_threads');
const parentPort = worker_threads.parentPort;
let tw;
try {
tw = require("${f}");
} catch (e) {
console.warn("could not load from ${f}");
}
if (!tw) {
try {
tw = require("@gnu-taler/taler-wallet-embedded");
} catch (e) {
console.warn("could not load taler-wallet-embedded either");
throw e;
}
}
if (typeof tw.handleWorkerMessage !== "function") {
throw Error("module loaded for crypto worker lacks handleWorkerMessage");
}
if (typeof tw.handleWorkerError !== "function") {
throw Error("module loaded for crypto worker lacks handleWorkerError");
}
parentPort.on("message", tw.handleWorkerMessage);
parentPort.on("error", tw.handleWorkerError);
`;
/**
* This function is executed in the worker thread to handle
* a message.
*/
export function handleWorkerMessage(msg: any): void {
const handleRequest = async (): Promise<void> => {
const req = msg.req;
if (typeof req !== "object") {
logger.error("request must be an object");
return;
}
const id = msg.id;
if (typeof id !== "number") {
logger.error("RPC id must be number");
return;
}
const operation = msg.operation;
if (typeof operation !== "string") {
logger.error("RPC operation must be string");
return;
}
const impl = nativeCryptoR;
if (!(operation in impl)) {
logger.error(`crypto operation '${operation}' not found`);
return;
}
let responseMsg: any;
try {
const result = await (impl as any)[operation](impl, req);
responseMsg = { data: { type: "success", result, id } };
} catch (e: any) {
logger.error(`error during operation: ${e.stack ?? e.toString()}`);
responseMsg = {
data: {
type: "error",
error: getErrorDetailFromException(e),
id,
},
};
}
try {
// eslint-disable-next-line @typescript-eslint/no-var-requires
const _r = "require";
const worker_threads: typeof import("worker_threads") =
module[_r]("worker_threads");
// const worker_threads = require("worker_threads");
const p = worker_threads.parentPort;
if (p) {
p.postMessage(responseMsg);
} else {
logger.error("parent port not available (not running in thread?");
}
} catch (e: any) {
logger.error(
`error sending back operation result: ${e.stack ?? e.toString()}`,
);
return;
}
};
handleRequest().catch((e) => {
logger.error("error in node worker", e);
});
}
export function handleWorkerError(e: Error): void {
logger.error(`got error from worker: ${e.stack ?? e.toString()}`);
}
export class NodeThreadCryptoWorkerFactory implements CryptoWorkerFactory {
startWorker(): CryptoWorker {
if (typeof require === "undefined") {
throw Error("cannot make worker, require(...) not defined");
}
return new NodeThreadCryptoWorker();
}
getConcurrency(): number {
return Math.max(1, os.cpus().length - 1);
}
}
/**
* Worker implementation that uses node subprocesses.
*/
class NodeThreadCryptoWorker implements CryptoWorker {
/**
* Function to be called when we receive a message from the worker thread.
*/
onmessage: undefined | ((m: any) => void);
/**
* Function to be called when we receive an error from the worker thread.
*/
onerror: undefined | ((m: any) => void);
private nodeWorker: import("worker_threads").Worker;
constructor() {
// eslint-disable-next-line @typescript-eslint/no-var-requires
const _r = "require";
const worker_threads = module[_r]("worker_threads");
logger.trace("starting node crypto worker");
this.nodeWorker = new worker_threads.Worker(workerCode, { eval: true });
this.nodeWorker.on("error", (err: Error) => {
logger.error("error in node worker:", err);
if (this.onerror) {
this.onerror(err);
}
});
this.nodeWorker.on("exit", (err) => {
logger.trace(`worker exited with code ${err}`);
});
this.nodeWorker.on("message", (v: any) => {
if (this.onmessage) {
this.onmessage(v);
}
});
this.nodeWorker.unref();
}
/**
* Add an event listener for either an "error" or "message" event.
*/
addEventListener(event: "message" | "error", fn: (x: any) => void): void {
switch (event) {
case "message":
this.onmessage = fn;
break;
case "error":
this.onerror = fn;
break;
}
}
/**
* Send a message to the worker thread.
*/
postMessage(msg: any): void {
this.nodeWorker.postMessage(msg);
}
/**
* Forcibly terminate the worker thread.
*/
terminate(): void {
this.nodeWorker.terminate();
}
}
|