114 lines
2.9 KiB
C
114 lines
2.9 KiB
C
#include "constants.h"
|
|
#include "storage.h"
|
|
#include "nvs_flash.h"
|
|
#include "nvs.h"
|
|
#include "esp_log.h"
|
|
#include <string.h>
|
|
|
|
#define STORAGE_NAMESPACE "storage"
|
|
|
|
#define DEV_NAME_KEY "dev_name"
|
|
#define PRIVATE_KEY "prikey"
|
|
#define SEC_ORIG_KEY "secureOrig"
|
|
#define SEC_ROOT_CA_KEY "rootCA"
|
|
#define DEV_CERT_KEY "certKey"
|
|
|
|
static const char *TAG = "storage";
|
|
|
|
bool storage_init()
|
|
{
|
|
esp_err_t err = nvs_flash_init();
|
|
if (err == ESP_ERR_NVS_NO_FREE_PAGES || err == ESP_ERR_NVS_NEW_VERSION_FOUND)
|
|
{
|
|
ESP_LOGI(TAG, "Need to reset storage");
|
|
|
|
// NVS partition was truncated and needs to be erased
|
|
// Retry nvs_flash_init
|
|
ESP_ERROR_CHECK(nvs_flash_erase());
|
|
err = nvs_flash_init();
|
|
}
|
|
|
|
return err == ESP_OK;
|
|
}
|
|
|
|
static void storage_set_str(const char *key, const char *value)
|
|
{
|
|
nvs_handle_t my_handle;
|
|
|
|
ESP_ERROR_CHECK(nvs_open(STORAGE_NAMESPACE, NVS_READWRITE, &my_handle));
|
|
|
|
ESP_ERROR_CHECK(nvs_set_blob(my_handle, key, value, strlen(value) + 1));
|
|
|
|
nvs_close(my_handle);
|
|
}
|
|
|
|
static size_t storage_get_str(const char *key, size_t dest_len, char *dest)
|
|
{
|
|
nvs_handle_t my_handle;
|
|
|
|
ESP_ERROR_CHECK(nvs_open(STORAGE_NAMESPACE, NVS_READWRITE, &my_handle));
|
|
|
|
size_t len = (dest == NULL ? 0 : dest_len);
|
|
esp_err_t res = nvs_get_blob(my_handle, key, dest, &len);
|
|
|
|
nvs_close(my_handle);
|
|
|
|
if (res == ESP_ERR_NVS_NOT_FOUND || len == 0)
|
|
return 0;
|
|
|
|
ESP_ERROR_CHECK(res);
|
|
|
|
return len;
|
|
}
|
|
|
|
void storage_set_dev_name(const char *name)
|
|
{
|
|
storage_set_str(DEV_NAME_KEY, name);
|
|
}
|
|
|
|
size_t storage_get_dev_name(char *dest)
|
|
{
|
|
return storage_get_str(DEV_NAME_KEY, DEV_NAME_LEN, dest);
|
|
}
|
|
|
|
void storage_set_priv_key(unsigned char *key, size_t len)
|
|
{
|
|
nvs_handle_t my_handle;
|
|
|
|
ESP_ERROR_CHECK(nvs_open(STORAGE_NAMESPACE, NVS_READWRITE, &my_handle));
|
|
|
|
ESP_ERROR_CHECK(nvs_set_blob(my_handle, PRIVATE_KEY, key, len));
|
|
|
|
nvs_close(my_handle);
|
|
}
|
|
|
|
size_t storage_get_priv_key(unsigned char *key)
|
|
{
|
|
nvs_handle_t my_handle;
|
|
|
|
ESP_ERROR_CHECK(nvs_open(STORAGE_NAMESPACE, NVS_READWRITE, &my_handle));
|
|
|
|
size_t len = (key == NULL ? 0 : PRV_KEY_DER_MAX_BYTES);
|
|
esp_err_t res = nvs_get_blob(my_handle, PRIVATE_KEY, key, &len);
|
|
|
|
nvs_close(my_handle);
|
|
|
|
if (res == ESP_ERR_NVS_NOT_FOUND || len == 0)
|
|
return 0;
|
|
|
|
ESP_ERROR_CHECK(res);
|
|
|
|
return len;
|
|
}
|
|
|
|
void storage_set_secure_origin(const char *orig) { storage_set_str(SEC_ORIG_KEY, orig); }
|
|
|
|
size_t storage_get_secure_origin(char *dest) { return storage_get_str(SEC_ORIG_KEY, SEC_ORIG_LEN, dest); }
|
|
|
|
void storage_set_root_ca(const char *ca) { storage_set_str(SEC_ROOT_CA_KEY, ca); }
|
|
|
|
size_t storage_get_root_ca(char *dest) { return storage_get_str(SEC_ROOT_CA_KEY, ROOT_CA_MAX_BYTES, dest); }
|
|
|
|
void storage_set_dev_cert(const char *cert) { storage_set_str(DEV_CERT_KEY, cert); }
|
|
|
|
size_t storage_get_dev_cert(char *dest) { return storage_get_str(DEV_CERT_KEY, DEV_CERT_MAX_BYTES, dest); } |