51 lines
1.4 KiB
TypeScript
51 lines
1.4 KiB
TypeScript
|
// Copyright 2021-2022 Signal Messenger, LLC
|
||
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
||
|
|
|
||
|
|
import { HsmEnclaveClient } from '@signalapp/libsignal-client';
|
||
|
|
import type { connection as WebSocket } from 'websocket';
|
||
|
|
|
||
|
|
import * as Bytes from '../../Bytes';
|
||
|
|
import { CDSHSocket } from './CDSHSocket';
|
||
|
|
import type { CDSSocketManagerBaseOptionsType } from './CDSSocketManagerBase';
|
||
|
|
import { CDSSocketManagerBase } from './CDSSocketManagerBase';
|
||
|
|
|
||
|
|
export type CDSHOptionsType = Readonly<{
|
||
|
|
publicKey: string;
|
||
|
|
codeHashes: ReadonlyArray<string>;
|
||
|
|
}> &
|
||
|
|
CDSSocketManagerBaseOptionsType;
|
||
|
|
|
||
|
|
export class CDSH extends CDSSocketManagerBase<CDSHSocket, CDSHOptionsType> {
|
||
|
|
private readonly publicKey: Buffer;
|
||
|
|
|
||
|
|
private readonly codeHashes: Array<Buffer>;
|
||
|
|
|
||
|
|
constructor(options: CDSHOptionsType) {
|
||
|
|
super(options);
|
||
|
|
|
||
|
|
this.publicKey = Buffer.from(Bytes.fromHex(options.publicKey));
|
||
|
|
this.codeHashes = options.codeHashes.map(hash =>
|
||
|
|
Buffer.from(Bytes.fromHex(hash))
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected override getSocketUrl(): string {
|
||
|
|
const { publicKey: publicKeyHex, codeHashes } = this.options;
|
||
|
|
|
||
|
|
return (
|
||
|
|
`${this.options.url}/discovery/${publicKeyHex}/` +
|
||
|
|
`${codeHashes.join(',')}`
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected override createSocket(socket: WebSocket): CDSHSocket {
|
||
|
|
const enclaveClient = HsmEnclaveClient.new(this.publicKey, this.codeHashes);
|
||
|
|
|
||
|
|
return new CDSHSocket({
|
||
|
|
logger: this.logger,
|
||
|
|
socket,
|
||
|
|
enclaveClient,
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|