mirror of
https://github.com/Lurkars/esp-ena.git
synced 2024-11-01 19:14:37 +01:00
341 lines
9.7 KiB
C
341 lines
9.7 KiB
C
// Copyright 2020 Lukas Haubaum
|
|
//
|
|
// Licensed under the GNU Affero General Public License, Version 3;
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
|
|
// https://www.gnu.org/licenses/agpl-3.0.html
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
#include <string.h>
|
|
#include <time.h>
|
|
#include <limits.h>
|
|
|
|
#include "esp_err.h"
|
|
#include "esp_log.h"
|
|
|
|
#include "ena-crypto.h"
|
|
#include "ena-storage.h"
|
|
#include "ena-beacons.h"
|
|
|
|
#include "pb_decode.h"
|
|
#include "TemporaryExposureKeyExport.pb.h"
|
|
|
|
#include "ena-exposure.h"
|
|
|
|
static ena_exposure_config_t DEFAULT_ENA_EXPOSURE_CONFIG = {
|
|
// transmission_risk_values
|
|
{
|
|
MINIMAL, // UNKNOWN
|
|
LOW, // CONFIRMED_TEST_LOW
|
|
MEDIUM, // CONFIRMED_TEST_STANDARD
|
|
HIGH, // CONFIRMED_TEST_HIGH
|
|
VERY_HIGH, // CONFIRMED_CLINICAL_DIAGNOSIS
|
|
VERY_LOW, // SELF_REPORT
|
|
ZERO, // NEGATIVE
|
|
MINIMAL // RECURSIVE
|
|
},
|
|
// duration_risk_values
|
|
{
|
|
MINIMAL, // D = 0 min
|
|
MINIMAL, // D <= 5 min
|
|
MEDIUM, // D <= 10 min
|
|
VERY_HIGH, // D <= 15 min
|
|
VERY_HIGH, // D <= 20 min
|
|
MAXIMUM, // D <= 25 min
|
|
MAXIMUM, // D <= 30 min
|
|
MAXIMUM // D > 30 min
|
|
},
|
|
// days_risk_values
|
|
{
|
|
MINIMAL, // >= 14 days
|
|
VERY_LOW, // 12-13 days
|
|
VERY_LOW, // 10-11 days
|
|
MEDIUM, // 8-9 days
|
|
HIGH, // 6-7 days
|
|
MAXIMUM, // 4-5 days
|
|
MAXIMUM, // 2-3 days
|
|
MAXIMUM // 0-1 days
|
|
},
|
|
// attenuation_risk_values
|
|
{
|
|
MINIMAL, // A > 73 dB
|
|
MINIMAL, // 73 >= A > 63
|
|
MINIMAL, // 63 >= A > 61
|
|
MAXIMUM, // 51 >= A > 33
|
|
MAXIMUM, // 33 >= A > 27
|
|
MAXIMUM, // 27 >= A > 15
|
|
MAXIMUM, // 15 >= A > 10
|
|
MAXIMUM // A <= 10
|
|
},
|
|
};
|
|
|
|
static const char kFileHeader[] = "EK Export v1 ";
|
|
static size_t kFileHeaderSize = sizeof(kFileHeader) - 1;
|
|
|
|
extern uint8_t export_bin_start[] asm("_binary_export_bin_start"); // test data from Google or https://svc90.main.px.t-online.de/version/v1/diagnosis-keys/country/DE/date/2020-07-22
|
|
extern uint8_t export_bin_end[] asm("_binary_export_bin_end");
|
|
|
|
int ena_exposure_risk_score(ena_exposure_config_t *config, ena_exposure_parameter_t params)
|
|
{
|
|
int score = 1;
|
|
score *= config->transmission_risk_values[params.report_type];
|
|
|
|
// calc duration level
|
|
int duration_level = MINUTES_0;
|
|
if (params.duration > 0)
|
|
{
|
|
if (params.duration <= 5)
|
|
{
|
|
duration_level = MINUTES_5;
|
|
}
|
|
else if (params.duration <= 10)
|
|
{
|
|
duration_level = MINUTES_10;
|
|
}
|
|
if (params.duration <= 15)
|
|
{
|
|
duration_level = MINUTES_15;
|
|
}
|
|
if (params.duration <= 20)
|
|
{
|
|
duration_level = MINUTES_20;
|
|
}
|
|
if (params.duration <= 25)
|
|
{
|
|
duration_level = MINUTES_25;
|
|
}
|
|
if (params.duration <= 30)
|
|
{
|
|
duration_level = MINUTES_30;
|
|
}
|
|
else
|
|
{
|
|
duration_level = MINUTES_LONGER;
|
|
}
|
|
}
|
|
score *= config->duration_risk_values[duration_level];
|
|
|
|
// calc days level
|
|
int days_level = DAYS_14;
|
|
|
|
if (params.days < 2)
|
|
{
|
|
days_level = DAYS_0;
|
|
}
|
|
else if (params.days < 4)
|
|
{
|
|
days_level = DAYS_3;
|
|
}
|
|
else if (params.days < 6)
|
|
{
|
|
days_level = DAYS_5;
|
|
}
|
|
else if (params.days < 8)
|
|
{
|
|
days_level = DAYS_7;
|
|
}
|
|
else if (params.days < 10)
|
|
{
|
|
days_level = DAYS_9;
|
|
}
|
|
else if (params.days < 12)
|
|
{
|
|
days_level = DAYS_11;
|
|
}
|
|
else if (params.days < 14)
|
|
{
|
|
days_level = DAYS_13;
|
|
}
|
|
|
|
score *= config->days_risk_values[days_level];
|
|
|
|
// calc attenuation level
|
|
int attenuation_level = ATTENUATION_73;
|
|
|
|
if (params.attenuation <= 10)
|
|
{
|
|
attenuation_level = ATTENUATION_LOWER;
|
|
}
|
|
else if (params.attenuation <= 15)
|
|
{
|
|
attenuation_level = ATTENUATION_10;
|
|
}
|
|
else if (params.attenuation <= 27)
|
|
{
|
|
attenuation_level = ATTENUATION_15;
|
|
}
|
|
else if (params.attenuation <= 33)
|
|
{
|
|
attenuation_level = ATTENUATION_27;
|
|
}
|
|
else if (params.attenuation <= 51)
|
|
{
|
|
attenuation_level = ATTENUATION_33;
|
|
}
|
|
else if (params.attenuation <= 63)
|
|
{
|
|
attenuation_level = ATTENUATION_51;
|
|
}
|
|
else if (params.attenuation <= 73)
|
|
{
|
|
attenuation_level = ATTENUATION_63;
|
|
}
|
|
|
|
score *= config->attenuation_risk_values[attenuation_level];
|
|
|
|
if (score > 255)
|
|
{
|
|
score = 255;
|
|
}
|
|
|
|
return score;
|
|
}
|
|
|
|
void ena_exposure_summary(ena_exposure_config_t *config, ena_exposure_summary_t *summary)
|
|
{
|
|
// XXX TEST key export (should be called on other location though)
|
|
ESP_ERROR_CHECK_WITHOUT_ABORT(ena_exposure_check_export(export_bin_start, (export_bin_end - export_bin_start)));
|
|
|
|
uint32_t count = ena_storage_exposure_information_count();
|
|
uint32_t current_time = (uint32_t)time(NULL);
|
|
|
|
summary->days_since_last_exposure = INT_MAX;
|
|
summary->max_risk_score = 0;
|
|
summary->risk_score_sum = 0;
|
|
summary->num_exposures = count;
|
|
|
|
if (count == 0)
|
|
{
|
|
summary->days_since_last_exposure = -1;
|
|
}
|
|
|
|
ena_exposure_information_t exposure_info;
|
|
ena_exposure_parameter_t params;
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
ena_storage_get_exposure_information(i, &exposure_info);
|
|
params.days = (current_time - exposure_info.day) / (60 * 60 * 24); // difference in days
|
|
if (params.days < summary->days_since_last_exposure)
|
|
{
|
|
summary->days_since_last_exposure = params.days;
|
|
}
|
|
params.duration = exposure_info.duration_minutes;
|
|
params.attenuation = exposure_info.typical_attenuation;
|
|
int score = ena_exposure_risk_score(config, params);
|
|
if (score > summary->max_risk_score)
|
|
{
|
|
summary->max_risk_score = score;
|
|
}
|
|
summary->risk_score_sum += score;
|
|
}
|
|
}
|
|
|
|
ena_exposure_config_t *ena_exposure_default_config(void)
|
|
{
|
|
return &DEFAULT_ENA_EXPOSURE_CONFIG;
|
|
}
|
|
|
|
bool ena_exposure_decode_key_data(pb_istream_t *stream, const pb_field_t *field, void **arg)
|
|
{
|
|
uint8_t *key_data = (uint8_t *)*arg;
|
|
|
|
if (!pb_read(stream, key_data, stream->bytes_left))
|
|
{
|
|
ESP_LOGW(ENA_EXPOSURE_LOG, "Decoding failed: %s\n", PB_GET_ERROR(stream));
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ena_exposure_decode_key(pb_istream_t *stream, const pb_field_t *field, void **arg)
|
|
{
|
|
uint8_t key_data[ENA_KEY_LENGTH] = {0};
|
|
TemporaryExposureKey tek = TemporaryExposureKey_init_zero;
|
|
|
|
tek.key_data = (pb_callback_t){
|
|
.funcs.decode = ena_exposure_decode_key_data,
|
|
.arg = &key_data,
|
|
};
|
|
|
|
if (!pb_decode(stream, TemporaryExposureKey_fields, &tek))
|
|
{
|
|
ESP_LOGW(ENA_EXPOSURE_LOG, "Decoding failed: %s\n", PB_GET_ERROR(stream));
|
|
return false;
|
|
}
|
|
|
|
ESP_LOGD(ENA_EXPOSURE_LOG,
|
|
"check reported tek: rolling_start_interval_number %d, rolling_period %d, days_since_last_exposure %d, report_type %d, transmission_risk_values %d",
|
|
tek.rolling_start_interval_number, tek.rolling_period, tek.days_since_onset_of_symptoms, tek.report_type, tek.transmission_risk_level);
|
|
|
|
ESP_LOG_BUFFER_HEXDUMP(ENA_EXPOSURE_LOG, &key_data, ENA_KEY_LENGTH, ESP_LOG_DEBUG);
|
|
|
|
bool match = false;
|
|
ena_beacon_t beacon;
|
|
ena_exposure_information_t exposure_info;
|
|
exposure_info.duration_minutes = 0;
|
|
exposure_info.min_attenuation = INT_MAX;
|
|
exposure_info.typical_attenuation = 0;
|
|
exposure_info.report_type = tek.report_type;
|
|
uint8_t rpi[ENA_KEY_LENGTH];
|
|
uint8_t rpik[ENA_KEY_LENGTH];
|
|
ena_crypto_rpik(rpik, key_data);
|
|
uint32_t beacons_count = ena_storage_beacons_count();
|
|
for (int i = 0; i < tek.rolling_period; i++)
|
|
{
|
|
ena_crypto_rpi(rpi, rpik, tek.rolling_start_interval_number + i);
|
|
for (int y = 0; y < beacons_count; y++)
|
|
{
|
|
ena_storage_get_beacon(y, &beacon);
|
|
if (memcmp(beacon.rpi, rpi, sizeof(ENA_KEY_LENGTH)) == 0)
|
|
{
|
|
match = true;
|
|
exposure_info.day = tek.rolling_start_interval_number * ENA_TIME_WINDOW;
|
|
exposure_info.duration_minutes += (ENA_BEACON_TRESHOLD / 60);
|
|
exposure_info.typical_attenuation = (exposure_info.typical_attenuation + beacon.rssi) / 2;
|
|
if (beacon.rssi < exposure_info.min_attenuation)
|
|
{
|
|
exposure_info.min_attenuation = beacon.rssi;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (match)
|
|
{
|
|
ena_storage_add_exposure_information(&exposure_info);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
esp_err_t ena_exposure_check_export(uint8_t *buf, size_t size)
|
|
{
|
|
// validate header
|
|
if (memcmp(kFileHeader, buf, kFileHeaderSize) != 0)
|
|
{
|
|
ESP_LOGW(ENA_EXPOSURE_LOG, "Wrong or missing header!");
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
TemporaryExposureKeyExport tek_export = TemporaryExposureKeyExport_init_zero;
|
|
|
|
tek_export.keys = (pb_callback_t){
|
|
.funcs.decode = ena_exposure_decode_key,
|
|
};
|
|
|
|
pb_istream_t stream = pb_istream_from_buffer(&buf[kFileHeaderSize], (size - kFileHeaderSize));
|
|
|
|
if (!pb_decode(&stream, TemporaryExposureKeyExport_fields, &tek_export))
|
|
{
|
|
ESP_LOGW(ENA_EXPOSURE_LOG, "Decoding failed: %s\n", PB_GET_ERROR(&stream));
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
return ESP_OK;
|
|
} |