65 lines
2.3 KiB
TypeScript
65 lines
2.3 KiB
TypeScript
import {spawn, Pool, Worker, Thread} from "@chainsafe/threads";
|
|
import {Implementation, PointFormat} from "../../../../src/index.js";
|
|
import {PublicKey, Signature} from "../../../../src/types.js";
|
|
import {WorkerApi} from "./worker.js";
|
|
|
|
type ThreadType = {
|
|
[K in keyof WorkerApi]: (...args: Parameters<WorkerApi[K]>) => Promise<ReturnType<WorkerApi[K]>>;
|
|
};
|
|
|
|
import path from "path";
|
|
import {fileURLToPath} from "url";
|
|
|
|
const __dirname = path.dirname(fileURLToPath(import.meta.url));
|
|
|
|
export class BlsMultiThreadNaive {
|
|
impl: Implementation;
|
|
pool: Pool<Thread & ThreadType>;
|
|
format: PointFormat;
|
|
|
|
constructor(impl: Implementation, workerCount?: number) {
|
|
this.impl = impl;
|
|
// Use compressed for herumi for now.
|
|
// THe worker is not able to deserialize from uncompressed
|
|
// `Error: err _wrapDeserialize`
|
|
this.format = impl === "blst-native" ? PointFormat.uncompressed : PointFormat.compressed;
|
|
this.pool = Pool(
|
|
() =>
|
|
(spawn(
|
|
// There is still an annoyance dealing with ESM imports here:
|
|
// threads.js attempts to require.resolve any files passed to Worker, and
|
|
// the esm module resolver requires the .js extension, even though the .js file does not actually exist.
|
|
// The solution for now:
|
|
// Pass in the script path as an absolute path and suppress threads.js default behavior when importing
|
|
new Worker(path.join(__dirname, "./worker.js"), {suppressResolveScript: true, suppressTranspileTS: true})
|
|
) as any) as Promise<Thread & ThreadType>,
|
|
workerCount
|
|
);
|
|
}
|
|
|
|
async destroy(): Promise<void> {
|
|
await this.pool.terminate(true);
|
|
}
|
|
|
|
async verify(pk: PublicKey, msg: Uint8Array, sig: Signature): Promise<boolean> {
|
|
return this.pool.queue((worker) =>
|
|
worker.verify(this.impl, pk.toBytes(PointFormat.uncompressed), msg, sig.toBytes(PointFormat.uncompressed))
|
|
);
|
|
}
|
|
|
|
async verifyMultipleAggregateSignatures(
|
|
sets: {publicKey: PublicKey; message: Uint8Array; signature: Signature}[]
|
|
): Promise<boolean> {
|
|
return this.pool.queue((worker) =>
|
|
worker.verifyMultipleAggregateSignatures(
|
|
this.impl,
|
|
sets.map((s) => ({
|
|
publicKey: s.publicKey.toBytes(PointFormat.uncompressed),
|
|
message: s.message,
|
|
signature: s.signature.toBytes(PointFormat.uncompressed),
|
|
}))
|
|
)
|
|
);
|
|
}
|
|
}
|