arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-25 17:09:44 +00:00
Code Issues Releases Wiki Activity

WIP - feat(core/prodtest): add pmic control commands

Browse Source
This commit is contained in:
cepetr 2024-11-20 16:24:20 +01:00
parent 58533d2286
commit 7a6d28a2d0
2 changed files with 98 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
core/SConscript.prodtest
View File

@ -217,7 +217,7 @@ program_elf = env.Command(
target='prodtest.elf',
source=obj_program,
action=
'$LINK -o $TARGET $CCFLAGS $CFLAGS $LINKFLAGS $SOURCES -lc_nano -lgcc',
'$LINK -o $TARGET $CCFLAGS $CFLAGS $LINKFLAGS $SOURCES -lc_nano -lgcc -lm',
)
env.Depends(program_elf, linkerscript_gen)

97
core/embed/projects/prodtest/main.c
View File

@ -77,6 +77,10 @@
#include "memzero.h"
#ifdef USE_POWERCTL
#include "../../sys/powerctl/npm1300/npm1300.h"
#endif
#ifdef USE_STORAGE_HWKEY
#include <sec/secure_aes.h>
#endif
@ -847,6 +851,94 @@ void cpuid_read(void) {
vcp_println_hex((uint8_t *)cpuid, sizeof(cpuid));
}
void test_pmic(const char *args) {
if (strcmp(args, "INIT") == 0) {
npm1300_deinit();
bool ok = npm1300_init();
if (ok) {
vcp_println("OK");
} else {
vcp_println("ERROR # I/O error");
}
} else if (strcmp(args, "CHGSTART") == 0) {
bool ok = npm1300_set_charging(true);
if (ok) {
vcp_println("OK # Charging started with %dmA current limit",
npm1300_get_charging_limit());
} else {
vcp_println("ERROR");
}
} else if (strcmp(args, "CHGSTOP") == 0) {
bool ok = npm1300_set_charging(false);
if (ok) {
vcp_println("OK # Charging stopped");
} else {
vcp_println("ERROR # I/O error");
}
} else if (strncmp(args, "CHGLIMIT", 8) == 0) {
int i_charge = atoi(&args[8]);
if (i_charge < NPM1300_CHARGING_LIMIT_MIN ||
i_charge > NPM1300_CHARGING_LIMIT_MAX) {
vcp_println("ERROR # Out of range");
return;
} else {
bool ok = npm1300_set_charging_limit(i_charge);
if (ok) {
vcp_println("OK # %dmA current limit", npm1300_get_charging_limit());
} else {
vcp_println("ERROR # I/O error");
}
}
} else if (strncmp(args, "MEASURE", 7) == 0) {
int seconds = atoi(&args[7]);
uint32_t ticks = hal_ticks_ms();
vcp_println(
"time; vbat; ibat; ntc_temp; vsys; die_temp; iba_meas_status; mode");
do {
npm1300_report_t report;
bool ok = npm1300_measure_sync(&report);
if (!ok) {
vcp_println("ERROR # I/O error");
break;
}
vcp_print("%09d; ", ticks);
vcp_print("%d.%03d; ", (int)report.vbat,
(int)(report.vbat * 1000) % 1000);
vcp_print("%d.%03d; ", (int)report.ibat,
(int)abs(report.ibat * 1000) % 1000);
vcp_print("%d.%03d; ", (int)report.ntc_temp,
(int)abs(report.ntc_temp * 1000) % 1000);
vcp_print("%d.%03d; ", (int)report.vsys,
(int)(report.vsys * 1000) % 1000);
vcp_print("%d.%03d; ", (int)report.die_temp,
(int)abs(report.die_temp * 1000) % 1000);
vcp_print("%02X; ", report.ibat_meas_status);
bool ibat_discharging = ((report.ibat_meas_status >> 2) & 0x03) == 1;
bool ibat_charging = ((report.ibat_meas_status >> 2) & 0x03) == 3;
if (ibat_discharging) {
vcp_print("DISCHARGING");
} else if (ibat_charging) {
vcp_print("CHARGING");
} else {
vcp_print("IDLE");
}
vcp_println("");
while (!ticks_expired(ticks + 1000)) {
};
ticks += 1000;
} while (seconds-- > 0);
vcp_println("OK # Measurement finished");
}
}
#define BACKLIGHT_NORMAL 150
int main(void) {
@ -866,6 +958,7 @@ int main(void) {
#ifdef USE_BUTTON
button_init();
#endif
npm1300_init();
#ifdef USE_TOUCH
touch_init();
#endif
@ -1009,6 +1102,10 @@ int main(void) {
test_wipe();
} else if (startswith(line, "REBOOT")) {
test_reboot();
} else if (startswith(line, "PMIC ")) {
test_pmic(line + 5);
} else if (startswith(line, "PMIC")) {
test_pmic(line + 4);
} else {
vcp_println("UNKNOWN");
}