RbootUpgrader.h

Go to the documentation of this file.

/****
 * Sming Framework Project - Open Source framework for high efficiency native ESP8266 development.
 * Created 2015 by Skurydin Alexey
 * http://github.com/SmingHub/Sming
 * All files of the Sming Core are provided under the LGPL v3 license.
 *
 * Ota.h
 *
 * This header includes all unified Over-The-Air functions.
 *
*/
 
#pragma once
#include <Ota/UpgraderBase.h>
#include <rboot-api.h>
 
namespace Ota
{
class RbootUpgrader : public UpgraderBase
{
public:
    bool begin(Partition partition, size_t size = 0) override;
    size_t write(const uint8_t* buffer, size_t size) override;
 
    bool end() override
    {
        return rboot_write_end(&status);
    }
 
    bool setBootPartition(Partition partition, bool save = true) override;
 
    Partition getBootPartition() override
    {
        return getPartitionForSlot(rboot_get_current_rom());
    }
 
    Partition getRunningPartition() override;
 
    Partition getNextBootPartition(Partition startFrom = {}) override
    {
        uint8_t currentSlot = rboot_get_current_rom();
        if(startFrom) {
            currentSlot = getSlotForPartition(startFrom);
        }
        return getPartitionForSlot(currentSlot ? 0 : 1);
    }
 
    static uint8_t getSlotForPartition(Partition partition)
    {
        return (partition.subType() == uint8_t(Partition::SubType::App::ota1)) ? 1 : 0;
    }
 
    static Partition getPartitionForSlot(uint8_t slot)
    {
        return Storage::spiFlash->partitions().findOta(slot);
    }
 
private:
    rboot_write_status status{};
    size_t maxSize{0};
    size_t writtenSoFar{0};
};
 
} // namespace Ota