feat(hardware-trezor): whitelist trezor safe 3 device model

input-output-hk · Jun 17, 2024 · 2109d34 · 2109d34
1 parent c359086
commit 2109d34
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/packages/hardware-trezor/src/TrezorKeyAgent.ts b/packages/hardware-trezor/src/TrezorKeyAgent.ts
@@ -16,6 +16,7 @@ import {
   errors,
   util
 } from '@cardano-sdk/key-management';
+import { WhitelistedDeviceModel } from './utils';
 import { areStringsEqualInConstantTime } from '@cardano-sdk/util';
 import { txToTrezor } from './transformers/tx';
 import _TrezorConnectWeb from '@trezor/connect-web';
@@ -129,7 +130,7 @@ export class TrezorKeyAgent extends KeyAgentBase {
       if (!deviceFeatures.success) {
         throw new errors.TransportError('Failed to get device', deviceFeatures.payload);
       }
-      if (deviceFeatures.payload.model !== 'T') {
+      if (!(Object.values(WhitelistedDeviceModel) as string[]).includes(deviceFeatures.payload.model)) {
         throw new errors.TransportError(`Trezor device model "${deviceFeatures.payload.model}" is not supported.`);
       }
       return deviceFeatures.payload;

diff --git a/packages/hardware-trezor/src/utils.ts b/packages/hardware-trezor/src/utils.ts
@@ -0,0 +1,4 @@
+export enum WhitelistedDeviceModel {
+  ModelT = 'T',
+  Safe3 = 'Safe 3'
+}