b-parasite-esp32/code/nrf-connect/samples/zigbee/src/double_reset_detector.c
2023-06-07 08:26:05 +02:00

66 lines
No EOL
1.8 KiB
C

#include "double_reset_detector.h"
#include <prstlib/macros.h>
#include <string.h>
#include <zephyr/fs/fs.h>
#include <zephyr/fs/littlefs.h>
#include <zephyr/kernel.h>
#include <zephyr/logging/log.h>
#include "debug_counters.h"
LOG_MODULE_REGISTER(double_reset_detector, CONFIG_LOG_DEFAULT_LEVEL);
static const char *flag_filename = "/lfs/reset_flag";
static const char flag_prefix[] = "prst-rst-count";
static struct fs_file_t flag_file;
static int erase_flag() {
return fs_unlink(flag_filename);
}
void erase_flag_callback(struct k_work *work) {
LOG_INF("Erasing double reset flag.");
if (erase_flag() != 0) {
LOG_ERR("Error deleting flag");
}
}
K_WORK_DELAYABLE_DEFINE(erase_flag_work, erase_flag_callback);
int prst_detect_double_reset(prst_double_reset_callback_t on_double_reset) {
fs_file_t_init(&flag_file);
RET_IF_ERR(fs_open(&flag_file, flag_filename, FS_O_CREATE | FS_O_RDWR));
char buff[sizeof(flag_prefix)];
RET_CHECK(fs_read(&flag_file, buff, sizeof(buff)) >= 0, "Unable to read file");
// Consider it a double reset if the flag is present in the FS.
if (strcmp(buff, flag_prefix) == 0) {
RET_IF_ERR(fs_close(&flag_file));
RET_IF_ERR(erase_flag());
prst_debug_counters_increment("double_reset");
return on_double_reset();
}
// Rewind file.
RET_IF_ERR(fs_seek(&flag_file, 0, SEEK_SET));
// Write the flag and erase it after some time.
ssize_t written = fs_write(&flag_file, flag_prefix, sizeof(flag_prefix));
if (written != sizeof(flag_prefix)) {
LOG_ERR("fs_write returned %d, expected %d", written, sizeof(flag_prefix));
return -1;
}
RET_IF_ERR(fs_close(&flag_file));
// Schedule the erasure of the flag after some time.
RET_CHECK(k_work_schedule(&erase_flag_work, K_SECONDS(5)) == 1,
"Work not scheduled");
return 0;
}