Momentum-Apps/example/main/serial_flash_example_main.c

/* Serial flasher Example

   This example code is in the Public Domain (or CC0 licensed, at your option.)

   Unless required by applicable law or agreed to in writing, this
   software is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
   CONDITIONS OF ANY KIND, either express or implied.
*/

#include <string.h>
#include "esp_err.h"
#include "esp_log.h"
#include "esp_spiffs.h"
#include "esp_loader.h"
#include "serial_io.h"
#include <sys/param.h>


static const char *TAG = "example";

const uint32_t HIGHER_BAUD_RATE = 921600;
const uint32_t APP_START_ADDRESS = 0x10000;
static uint8_t payload[1024];

esp_loader_error_t loader_port_change_baudrate(uint32_t baudrate);


static void flash_binary(FILE *image, size_t image_size)
{
    esp_loader_error_t err;
    int32_t packet_number = 0;
    esp_loader_connect_args_t connect_config = ESP_LOADER_CONNECT_DEFAULT();

    err = esp_loader_connect(&connect_config);
    if (err != ESP_LOADER_SUCCESS) {
        ESP_LOGE(TAG, "Cannot connect to target.");
        return;
    }
    ESP_LOGI(TAG, "Connected to target");

    err = esp_loader_change_baudrate(HIGHER_BAUD_RATE);
    if (err != ESP_LOADER_SUCCESS) {
        ESP_LOGE(TAG, "Unable to change baud rate on target.");
        return;
    }

    err = loader_port_change_baudrate(HIGHER_BAUD_RATE);
    if (err != ESP_LOADER_SUCCESS) {
        ESP_LOGE(TAG, "Unable to change baud rate.");
        return;
    }

    err = esp_loader_flash_start(APP_START_ADDRESS, image_size, sizeof(payload));
    if (err != ESP_LOADER_SUCCESS) {
        ESP_LOGE(TAG, "Flash start operation failed.");
        return;
    }
    ESP_LOGI(TAG, "Start programming");

    while (image_size > 0) {
        size_t to_read = MIN(image_size, sizeof(payload));

        size_t read = fread(payload, 1, to_read, image);
        if (read != to_read) {
            ESP_LOGE(TAG, "Error occurred while reading file.");
            return;
        }

        err = esp_loader_flash_write(payload, to_read);
        if (err != ESP_LOADER_SUCCESS) {
            ESP_LOGE(TAG, "Packet could not be written");
            return;
        }

        ESP_LOGI(TAG, "packet: %d  written: %u B", packet_number++, to_read);

        image_size -= to_read;
    };

    ESP_LOGI(TAG, "Finished programming");

    err = esp_loader_flash_verify();
    if (err != ESP_LOADER_SUCCESS) {
        ESP_LOGE(TAG, "MD5 does not match. err: %d", err);
        return;
    }
    ESP_LOGI(TAG, "Flash verified");
}


static FILE *get_image_and_its_size(size_t *image_size)
{
    ESP_LOGI(TAG, "Initializing SPIFFS");

    esp_vfs_spiffs_conf_t conf = {
        .base_path = "/spiffs",
        .partition_label = NULL,
        .max_files = 5,
        .format_if_mount_failed = false
    };

    // Use settings defined above to initialize and mount SPIFFS filesystem.
    esp_err_t ret = esp_vfs_spiffs_register(&conf);

    if (ret != ESP_OK) {
        if (ret == ESP_FAIL) {
            ESP_LOGE(TAG, "Failed to mount or format filesystem");
        } else if (ret == ESP_ERR_NOT_FOUND) {
            ESP_LOGE(TAG, "Failed to find SPIFFS partition");
        } else {
            ESP_LOGE(TAG, "Failed to initialize SPIFFS (%s)", esp_err_to_name(ret));
        }
        return NULL;
    }

    FILE *image = fopen("/spiffs/hello-world.bin", "r");
    if (image == NULL) {
        ESP_LOGE(TAG, "Failed to open file");
        esp_vfs_spiffs_unregister(NULL);
        return NULL;
    }

    fseek(image, 0L, SEEK_END);
    *image_size = ftell(image);
    rewind(image);

    ESP_LOGI(TAG, "Image size: %u", *image_size);

    return image;
}


void app_main(void)
{
    // Initialize UART
    esp_loader_error_t err = loader_port_serial_init(115200);
    if (err != ESP_LOADER_SUCCESS) {
        ESP_LOGE(TAG, "serial initialization failed.");
        return;
    }

    size_t image_size;
    FILE *image = get_image_and_its_size(&image_size);
    if (image == NULL) {
        return;
    }

    flash_binary(image, image_size);

    // All done, close file, unmount partition and disable SPIFFS
    fclose(image);
    esp_vfs_spiffs_unregister(NULL);
}