feat(suite): useRetryFwAuthenticityChecks in Desktop

Author: Lemonexe

packages/suite/src/components/suite/Preloader/Preloader.tsx

@@ -1,11 +1,17 @@
 import { type FC, type PropsWithChildren, useEffect } from 'react';
 
-import { selectShouldDisplayDeviceCompromisedOnRoute } from '@suite/authenticity-checks';
+import {
+    selectFirmwareRevisionCheckErrorIfEnabled,
+    selectShouldDisplayDeviceCompromisedOnRoute,
+} from '@suite/authenticity-checks';
 import { selectDesktopUpdateAllowPrerelease } from '@suite/desktop-update';
-import { useDevice } from '@suite/device';
+import { rerunFwAuthenticityChecksThunk, useDevice } from '@suite/device';
 import { KillswitchMessageScreen } from '@suite/message-system';
 import { selectIsAnalyticsConfirmed } from '@suite-common/analytics-redux';
-import { useReportDeviceCompromised } from '@suite-common/firmware-authenticity';
+import {
+    useReportDeviceCompromised,
+    useRetryFwAuthenticityChecks,
+} from '@suite-common/firmware-authenticity';
 import { selectActiveKillswitchMessage } from '@suite-common/message-system';
 import { Card } from '@trezor/components';
 
@@ -67,6 +73,10 @@ export const Preloader = ({ children }: PropsWithChildren) => {
 
     const dispatch = useDispatch();
 
+    const firmwareRevisionCheckError = useSelector(selectFirmwareRevisionCheckErrorIfEnabled);
+    const rerunFwAuthenticityChecksCall = () => dispatch(rerunFwAuthenticityChecksThunk());
+    useRetryFwAuthenticityChecks({ firmwareRevisionCheckError, rerunFwAuthenticityChecksCall });
+
     useEffect(() => {
         // Analytics needs to be resolved before we show anything to the user. Until this is solved,
         // we do not init anything. Especially nothing related to the devices/connect. With THP,

suite/device/src/deviceThunks.ts

@@ -1,9 +1,11 @@
+import { selectIsDeviceLocked } from '@suite/locks';
 import { selectRouterApp } from '@suite/router';
 import { selectSelectedDevice } from '@suite-common/device';
 import { createThunk } from '@suite-common/redux-utils';
 import type { TrezorDevice } from '@suite-common/suite-types';
 import { handleDeviceDisconnect, selectDeviceThunk } from '@suite-common/wallet-core';
 import type { Device } from '@trezor/connect';
+import TrezorConnect from '@trezor/connect';
 
 const DEVICE_MODULE_PREFIX = '@suite';
 
@@ -29,3 +31,14 @@ export const disconnectDeviceThunk = createThunk(
         dispatch(handleDeviceDisconnect(device));
     },
 );
+
+/**
+ * Connect call to rerun FW authenticity checks (getFeatures used as the most basic no-op device call).
+ */
+export const rerunFwAuthenticityChecksThunk = createThunk(
+    `${DEVICE_MODULE_PREFIX}/rerunFwAuthenticityChecksThunk`,
+    (_, { getState }) => {
+        if (selectIsDeviceLocked(getState())) return;
+        TrezorConnect.getFeatures();
+    },
+);