/* SPDX-FileCopyrightText: © 2023-2024 Decompollaborate */
/* SPDX-License-Identifier: MIT */

#include "ipl3checksum.h"

#include <assert.h>
#include <stdio.h>

#include "utils.h"

void print_usage(int argc, char *argv[]) {
    (void)argc;

    fprintf(stderr, "Usage: %s bin_file\n", argv[0]);
    fprintf(stderr, "\n");
    fprintf(stderr, "This programs computes the ipl3 checksum of a big endian binary file by guessing its cic kind\n");
}

int main(int argc, char *argv[]) {
    int ret = 0;

    if (argc < 2) {
        print_usage(argc, argv);
        return -1;
    }

    fprintf(stderr, "Running ipl3checksum version %s\n", ipl3checksum_version_str);

    const char *bin_path = argv[1];

    size_t bin_size = 0;
    uint8_t *bin = read_binary_file(bin_path, &bin_size);
    assert(bin_size > 0);
    assert(bin != NULL);

    uint32_t expected_checksum0 = read_be_word(bin, 0x10);
    uint32_t expected_checksum1 = read_be_word(bin, 0x14);

    fprintf(stderr, "Expected checksum: %08X %08X\n", expected_checksum0, expected_checksum1);

    {
        uint32_t checksum0;
        uint32_t checksum1;

        Ipl3Checksum_Error err = ipl3checksum_calculate_checksum_autodetect(&checksum0, &checksum1, bin_size, bin);

        if (err.tag == Ipl3Checksum_Error_Okay) {
            fprintf(stderr, "Computed checksum: %08X %08X\n", checksum0, checksum1);
            if ((checksum0 == expected_checksum0) && (checksum1 == expected_checksum1)) {
                fprintf(stderr, "Checksum matches\n");
            } else {
                fprintf(stderr, "Checksum doesn't match\n");
                ret++;
            }
        } else {
            fprintf(stderr, "Error trying to compute the checksum: %s\n", get_ipl3checksum_error_str(err));
            ret++;
        }
    }

    free(bin);

    return ret;
}