mirror of https://github.com/Lurkars/esp-ena.git
119 lines
3.3 KiB
C
119 lines
3.3 KiB
C
|
|
#include <stdio.h>
|
|
#include <time.h>
|
|
|
|
#include "esp_system.h"
|
|
#include "esp_log.h"
|
|
#include "esp_bt.h"
|
|
#include "esp_bt_main.h"
|
|
#include "esp_gap_ble_api.h"
|
|
|
|
#include "nvs_flash.h"
|
|
|
|
#include "ena-crypto.h"
|
|
#include "ena-storage.h"
|
|
#include "ena-bluetooth-scan.h"
|
|
#include "ena-bluetooth-advertise.h"
|
|
|
|
#include "ena.h"
|
|
|
|
static uint32_t last_enin; // last ENIN
|
|
static uint8_t tek[ENA_KEY_LENGTH] = {0}; // current TEK
|
|
|
|
void ena_init(void)
|
|
{
|
|
// init NVS for BLE
|
|
esp_err_t ret;
|
|
ret = nvs_flash_init();
|
|
if (ret == ESP_ERR_NVS_NO_FREE_PAGES || ret == ESP_ERR_NVS_NEW_VERSION_FOUND)
|
|
{
|
|
ESP_ERROR_CHECK(nvs_flash_erase());
|
|
ESP_ERROR_CHECK(nvs_flash_init());
|
|
}
|
|
|
|
// init BLE
|
|
esp_bt_controller_config_t bt_cfg = BT_CONTROLLER_INIT_CONFIG_DEFAULT();
|
|
ESP_ERROR_CHECK(esp_bt_controller_init(&bt_cfg));
|
|
ESP_ERROR_CHECK(esp_bt_controller_enable(ESP_BT_MODE_BLE));
|
|
ESP_ERROR_CHECK(esp_bluedroid_init());
|
|
ESP_ERROR_CHECK(esp_bluedroid_enable());
|
|
|
|
ESP_ERROR_CHECK(esp_ble_tx_power_set(ESP_BLE_PWR_TYPE_DEFAULT, ESP_PWR_LVL_P9));
|
|
ESP_ERROR_CHECK(esp_ble_tx_power_set(ESP_BLE_PWR_TYPE_ADV, ESP_PWR_LVL_P9));
|
|
ESP_ERROR_CHECK(esp_ble_tx_power_set(ESP_BLE_PWR_TYPE_SCAN, ESP_PWR_LVL_P9));
|
|
ESP_ERROR_CHECK(esp_ble_gap_config_local_privacy(true));
|
|
|
|
/* nesseccary?
|
|
// new bluetooth address
|
|
uint8_t bt_address[ESP_BD_ADDR_LEN];
|
|
esp_fill_random(bt_address, ESP_BD_ADDR_LEN);
|
|
bt_address[0] |= 0xC0;
|
|
|
|
ESP_ERROR_CHECK(esp_ble_gap_set_rand_addr(bt_address));
|
|
*/
|
|
// init ENA
|
|
ena_crypto_init();
|
|
|
|
uint32_t current_enin = ena_crypto_enin((uint32_t)time(NULL));
|
|
|
|
last_enin = ena_storage_read_enin();
|
|
|
|
// read last TEK or create new
|
|
if (ena_storage_read_u8(ENA_STORAGE_TEK_COUNT_ADDRESS) > 0 && (current_enin - last_enin) < ENA_TEK_ROLLING_PERIOD)
|
|
{
|
|
ena_storage_read_tek(tek);
|
|
}
|
|
else
|
|
{
|
|
ena_crypto_tek(tek);
|
|
ena_storage_write_tek(ena_crypto_enin((uint32_t)time(NULL)), tek);
|
|
last_enin = ena_storage_read_enin();
|
|
}
|
|
|
|
// init scan
|
|
ena_bluetooth_scan_init();
|
|
|
|
// init and start advertising
|
|
ena_bluetooth_advertise_set_payload(current_enin, tek);
|
|
ena_bluetooth_advertise_start();
|
|
// initial scan on every start
|
|
ena_bluetooth_scan_start(ENA_SCANNING_TIME);
|
|
}
|
|
|
|
void ena_run(void)
|
|
{
|
|
uint32_t unix_timestamp = (uint32_t)time(NULL);
|
|
uint32_t current_enin = ena_crypto_enin(unix_timestamp);
|
|
if (current_enin - last_enin >= ENA_TEK_ROLLING_PERIOD)
|
|
{
|
|
ena_crypto_tek(tek);
|
|
ena_storage_write_tek(current_enin, tek);
|
|
last_enin = current_enin;
|
|
}
|
|
|
|
// change RPI
|
|
if (unix_timestamp % ENA_TIME_WINDOW == 0)
|
|
{
|
|
|
|
//
|
|
|
|
if (ena_bluetooth_scan_get_status() == ENA_SCAN_STATUS_SCANNING)
|
|
{
|
|
ena_bluetooth_scan_stop();
|
|
}
|
|
ena_bluetooth_advertise_stop();
|
|
ena_bluetooth_advertise_set_payload(current_enin, tek);
|
|
ena_bluetooth_advertise_start();
|
|
if (ena_bluetooth_scan_get_status() == ENA_SCAN_STATUS_WAITING)
|
|
{
|
|
ena_bluetooth_scan_start(ENA_SCANNING_TIME);
|
|
}
|
|
}
|
|
|
|
// scan
|
|
if (unix_timestamp % ENA_SCANNING_INTERVAL == 0 && ena_bluetooth_scan_get_status() == ENA_SCAN_STATUS_NOT_SCANNING)
|
|
{
|
|
ena_bluetooth_scan_start(ENA_SCANNING_TIME);
|
|
}
|
|
}
|