#include "ChecksumHelper.hpp"

namespace util {

namespace {
uint32_t getValue(const std::vector<uint8_t>& data, size_t index)
{
    return ((uint32_t)data[index + 3] << 24)
            + ((uint32_t)data[index + 2] << 16)
            + ((uint32_t)data[index + 1] << 8)
            + (uint32_t)data[index];
}

void setValue(std::vector<uint8_t>& data, size_t index, uint32_t value)
{
    data[index] = static_cast<uint8_t>(value);
    data[index + 1] = static_cast<uint8_t>(value >> 8);
    data[index + 2] = static_cast<uint8_t>(value >> 16);
    data[index + 3] = static_cast<uint8_t>(value >> 24);
}

}

bool ChecksumHelper::check(const std::vector<uint8_t>& data) const
{
    uint32_t buffer_index = 0x1F810;
    uint32_t max_buffer_index = 0x1FA00;
    if (data.size() == 1024 * 1024)
        max_buffer_index = 0x1FC00;
    do {
        uint32_t start_addr = getValue(data, buffer_index);

        // Get the checksum zone end address
        uint32_t end_addr = getValue(data, buffer_index + 4);

        if (start_addr >= data.size() || end_addr >= data.size())
            break;

        uint32_t checksum = 0;
        for (uint32_t addr = start_addr; addr < end_addr; addr += 2)
            checksum += ((uint32_t)data[addr + 1] << 8) + (uint32_t)data[addr];

        uint32_t curr_checksum = getValue(data, buffer_index + 8);
        uint32_t compliment_curr_checksum = getValue(data, buffer_index + 12);

        uint32_t complchecksum = ~checksum;

        if (curr_checksum != checksum || compliment_curr_checksum != complchecksum)
            return false;

        buffer_index += 0x10;
    }
    while(buffer_index < max_buffer_index);
    return true;
}

void ChecksumHelper::update(std::vector<uint8_t>& data) const
{
    uint32_t buffer_index = 0x1F810;
    uint32_t max_buffer_index = 0x1FA00;
    if (data.size() == 1024 * 1024)
        max_buffer_index = 0x1FC00;
    do {
        uint32_t start_addr = getValue(data, buffer_index);

        // Get the checksum zone end address
        uint32_t end_addr = getValue(data, buffer_index + 4);

        if (start_addr >= data.size() || end_addr >= data.size())
            break;

        uint32_t checksum = 0;
        for (uint32_t addr = start_addr; addr < end_addr; addr += 2)
            checksum += ((uint32_t)data[addr + 1] << 8) + (uint32_t)data[addr];

        uint32_t complchecksum = ~checksum;

        setValue(data, buffer_index + 8, checksum);
        setValue(data, buffer_index + 12, complchecksum);

        buffer_index += 0x10;
    }
    while(buffer_index < max_buffer_index);
}

}