feat(core): Add power manager prodtest

[no changelog]

core/SConscript.prodtest

@@ -152,6 +152,7 @@ SOURCE_PRODTEST = [
     'embed/projects/prodtest/cmd/prodtest_ping.c',
     'embed/projects/prodtest/cmd/prodtest_pmic.c',
     'embed/projects/prodtest/cmd/prodtest_powerctl.c',
+    'embed/projects/prodtest/cmd/prodtest_power_manager.c',
     'embed/projects/prodtest/cmd/prodtest_reboot.c',
     'embed/projects/prodtest/cmd/prodtest_rgbled.c',
     'embed/projects/prodtest/cmd/prodtest_sdcard.c',

core/embed/projects/prodtest/cmd/prodtest_power_manager.c

@@ -0,0 +1,191 @@
+/*
+ * This file is part of the Trezor project, https://trezor.io/
+ *
+ * Copyright (c) SatoshiLabs
+ *
+ * This program 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 of the License, or
+ * (at your option) any later version.
+ *
+ * This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifdef USE_POWERCTL
+
+#include <rtl/cli.h>
+#include <rtl/unit_test.h>
+#include <sys/power_manager.h>
+#include <sys/systick.h>
+#include <trezor_rtl.h>
+
+void prodtest_power_manager_hibernate(cli_t* cli) {
+  if (cli_arg_count(cli) > 0) {
+    cli_error_arg_count(cli);
+    return;
+  }
+
+  cli_trace(cli, "Hibernating the device...");
+
+  if (!power_manager_hibernate()) {
+    cli_error(cli, CLI_ERROR, "Failed to hibernate.");
+    return;
+  }
+
+  cli_trace(cli, "Device is powered externally, hibernation is not possible.");
+  cli_ok(cli, "");
+}
+
+void prodtest_power_manager_suspend(cli_t* cli) {
+  if (cli_arg_count(cli) > 0) {
+    cli_error_arg_count(cli);
+    return;
+  }
+
+  cli_trace(cli, "Suspending the device to low-power mode...");
+  cli_trace(cli, "Press the POWER button to resume.");
+  systick_delay_ms(1000);
+
+  power_manager_suspend();
+
+  systick_delay_ms(1500);
+  cli_trace(cli, "Resumed to active mode.");
+
+  cli_ok(cli, "");
+}
+
+void prodtest_power_manager_report(cli_t* cli) {
+  if (cli_arg_count(cli) > 0) {
+    cli_error_arg_count(cli);
+    return;
+  }
+
+  power_manager_report_t report;
+  power_manager_status_t status = power_manager_get_report(&report);
+  if (status != POWER_MANAGER_OK) {
+    cli_error(cli, CLI_ERROR, "Failed to get power manager report");
+    return;
+  }
+  cli_trace(cli, "Power manager report:");
+  cli_trace(cli, "  USB %s",
+            report.usb_connected ? "connected" : "disconnected");
+  cli_trace(cli, "  WLC %s",
+            report.wireless_charger_connected ? "connected" : "disconnected");
+  cli_trace(cli, "  Battery voltage: %d.%03d V", (int)report.battery_voltage_v,
+            (int)(report.battery_voltage_v * 1000) % 1000);
+  cli_trace(cli, "  Battery current: %d.%03d mA",
+            (int)report.battery_current_ma,
+            (int)(report.battery_current_ma * 1000) % 1000);
+  cli_trace(cli, "  Battery temperature: %d.%03d C", (int)report.battery_temp_c,
+            (int)(report.battery_temp_c * 1000) % 1000);
+  cli_trace(cli, "  PMIC die temperature: %d.%03d C", (int)report.pmic_temp_c,
+            (int)(report.pmic_temp_c * 1000) % 1000);
+  cli_trace(cli, "  WLC voltage: %d.%03d V",
+            (int)report.wireless_output_voltage_v,
+            (int)(report.wireless_output_voltage_v * 1000) % 1000);
+  cli_trace(cli, "  WLC current: %d.%03d mA", (int)report.wireless_current_ma,
+            (int)(report.wireless_current_ma * 1000) % 1000);
+  cli_trace(cli, "  WLC die temperature: %d.%03d C",
+            (int)report.wireless_temp_c,
+            (int)(report.wireless_temp_c * 1000) % 1000);
+  cli_trace(cli, "  System voltage: %d.%03d V", (int)report.system_voltage_v,
+            (int)(report.system_voltage_v * 1000) % 1000);
+
+  cli_ok(cli, "");
+}
+
+void prodtest_power_manager_monitor(cli_t* cli) {
+  if (cli_arg_count(cli) > 0) {
+    cli_error_arg_count(cli);
+    return;
+  }
+
+  power_manager_status_t status;
+  power_manager_event_t event_flag;
+  power_manager_state_t state;
+
+  status = power_manager_get_state(&state);
+
+  // Clear leftover events
+  power_manager_get_events(&event_flag);
+
+  cli_trace(cli, "Start power manager monitor, current state: {%s}",
+            power_manager_get_state_name(state));
+
+  while (true) {
+    if (cli_aborted(cli)) {
+      cli_trace(cli, "power manager test aborted");
+      break;
+    }
+
+    status = power_manager_get_events(&event_flag);
+    if (status != POWER_MANAGER_OK) {
+      cli_error(cli, CLI_ERROR, "Failed to get power manager events");
+    }
+
+    if (event_flag & POWER_MANAGER_EVENT_STATE_CHANGED) {
+      status = power_manager_get_state(&state);
+
+      cli_trace(cli, "Power manager state changed to {%s}",
+                power_manager_get_state_name(state));
+    }
+
+    if (event_flag & POWER_MANAGER_EVENT_USB_CONNECTED) {
+      cli_trace(cli, "USB connected");
+    }
+
+    if (event_flag & POWER_MANAGER_EVENT_USB_DISCONNECTED) {
+      cli_trace(cli, "USB disconnected");
+    }
+
+    if (event_flag & POWER_MANAGER_EVENT_WIRELESS_CONNECTED) {
+      cli_trace(cli, "WLC connected");
+    }
+
+    if (event_flag & POWER_MANAGER_EVENT_WIRELESS_DISCONNECTED) {
+      cli_trace(cli, "WLC disconnected");
+    }
+
+    systick_delay_ms(50);
+  }
+
+  cli_ok(cli, "");
+}
+
+// clang-format off
+
+PRODTEST_CLI_CMD(
+    .name = "power-manager-monitor",
+    .func = prodtest_power_manager_monitor,
+    .info = "Run power manager monitor",
+    .args = ""
+);
+
+PRODTEST_CLI_CMD(
+    .name = "power-manager-report",
+    .func = prodtest_power_manager_report,
+    .info = "Get power manager report",
+    .args = ""
+);
+
+PRODTEST_CLI_CMD(
+    .name = "power-manager-suspend",
+    .func = prodtest_power_manager_suspend,
+    .info = "Suspend the device to low-power mode",
+    .args = ""
+);
+
+PRODTEST_CLI_CMD(
+    .name = "power-manager-hibernate",
+    .func = prodtest_power_manager_hibernate,
+    .info = "Hibernate the device into a near power-off state",
+    .args = ""
+);
+
+#endif /* USE POWERCTL */

core/embed/projects/prodtest/main.c

@@ -79,7 +79,7 @@
 #endif
 
 #ifdef USE_POWERCTL
-#include <sys/powerctl.h>
+#include <sys/power_manager.h>
 #endif
 
 #ifdef USE_STORAGE_HWKEY
@@ -194,7 +194,8 @@ void prodtest_disable_rgbled_control(void) { g_rgbled_control_disabled = true; }
 
 static void drivers_init(void) {
 #ifdef USE_POWERCTL
-  powerctl_init();
+  power_manager_init();
+  // powerctl_init();
 #endif
 
   display_init(DISPLAY_RESET_CONTENT);