ApplicationChecksum.hxx

Go to the documentation of this file.

 
#ifndef _OPENLCB_APPLICATIONCHECKSUM_HXX_
#define _OPENLCB_APPLICATIONCHECKSUM_HXX_
 
#ifdef ESP_PLATFORM
#include "bootloader_hal.h"
#else
#include "freertos/bootloader_hal.h"
#endif
 
extern "C" {
bool check_application_checksum()
{
    uint32_t checksum[CHECKSUM_COUNT];
    const void *flash_min;
    const void *flash_max;
    const struct app_header *app_header_ptr;
    get_flash_boundaries(&flash_min, &flash_max, &app_header_ptr);
 
    struct app_header app_header;
    memcpy(&app_header, app_header_ptr, sizeof(app_header));
 
    uint32_t pre_size = reinterpret_cast<const uint8_t *>(app_header_ptr) -
        static_cast<const uint8_t *>(flash_min);
    checksum_data(flash_min, pre_size, checksum);
    if (memcmp(app_header.checksum_pre, checksum, sizeof(checksum)))
    {
        return false;
    }
    size_t flash_size = (size_t)flash_max - (size_t)flash_min;
    if (app_header.app_size > flash_size) return false;
    uint32_t post_offset = sizeof(struct app_header) +
        (reinterpret_cast<const uint8_t *>(app_header_ptr) -
         static_cast<const uint8_t *>(flash_min));
    uint32_t post_size = (post_offset < app_header.app_size)
        ? app_header.app_size - post_offset
        : 0;
    checksum_data(app_header_ptr + 1, post_size, checksum);
    if (memcmp(app_header.checksum_post, checksum, sizeof(checksum)))
    {
        return false;
    }
    return true;
}
}
 
#endif // _OPENLCB_APPLICATIONCHECKSUM_HXX_