#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include "se050_hmac_blake2s.h"
#include "se050_hkdf_blake2s.h"
#include "se050_tai64n.h"

static void print_hex(const char *label, const uint8_t *buf, size_t len)
{
    printf("%s: ", label);
    for (size_t i = 0; i < len; i++) printf("%02x", buf[i]);
    printf("\n");
}

int main(void)
{
    uint8_t mac[32];
    uint8_t prk[32];
    uint8_t okm[64];
    uint8_t tai64n[12];
    uint8_t current[12];
    uint64_t sec;
    uint32_t nsec;
    int passed = 0;

    printf("HMAC-BLAKE2s + HKDF + TAI64N Test Suite\n");
    printf("========================================\n\n");

    printf("Test 1: HMAC-BLAKE2s\n");
    const uint8_t key[] = "key";
    const uint8_t data[] = "The quick brown fox jumps over the lazy dog";
    se050_hmac_blake2s(mac, key, sizeof(key)-1, data, sizeof(data)-1);
    print_hex("HMAC-BLAKE2s", mac, 32);
    printf("[INFO] HMAC computed\n\n");
    passed++;

    printf("Test 2: HKDF-Extract\n");
    const uint8_t salt[] = "salt";
    const uint8_t ikm[] = "input key material";
    se050_hkdf_extract(prk, salt, sizeof(salt)-1, ikm, sizeof(ikm)-1);
    print_hex("PRK", prk, 32);
    printf("[INFO] Extract done\n\n");
    passed++;

    printf("Test 3: HKDF-Expand\n");
    const uint8_t info[] = "application info";
    se050_hkdf_expand(okm, 64, prk, info, sizeof(info)-1);
    print_hex("OKM", okm, 64);
    printf("[INFO] Expand done\n\n");
    passed++;

    printf("Test 4: HKDF Combined\n");
    se050_hkdf(okm, 32, salt, sizeof(salt)-1, ikm, sizeof(ikm)-1, info, sizeof(info)-1);
    print_hex("HKDF Output", okm, 32);
    printf("[INFO] HKDF done\n\n");
    passed++;

    printf("Test 5: TAI64N Encode/Decode\n");
    se050_tai64n_encode(tai64n, 1609459200, 123456789);
    print_hex("TAI64N", tai64n, 12);
    se050_tai64n_decode(tai64n, &sec, &nsec);
    printf("Decoded: sec=%lu, nsec=%u\n", (unsigned long)sec, nsec);
    if (sec == 1609459200 && nsec == 123456789) {
        printf("[PASS] Round-trip OK\n\n"); passed++;
    } else {
        printf("[FAIL] Round-trip failed\n\n");
    }

    printf("Test 6: TAI64N Now\n");
    se050_tai64n_now(tai64n);
    print_hex("Current TAI64N", tai64n, 12);
    se050_tai64n_decode(tai64n, &sec, &nsec);
    printf("Decoded: sec=%lu, nsec=%u\n", (unsigned long)sec, nsec);
    printf("[INFO] Current time obtained\n\n");
    passed++;

    printf("Test 7: TAI64N Window Check\n");
    se050_tai64n_now(tai64n);
    se050_tai64n_now(current);
    int result = se050_tai64n_check_window(tai64n, current, 60);
    printf("Window check (60s): %s\n", 
           result == 1 ? "PASS (within window)" : "FAIL");
    if (result == 1) {
        printf("[PASS]\n\n"); passed++;
    } else {
        printf("[FAIL]\n\n");
    }

    printf("========================================\n");
    printf("Passed: %d/7\n", passed);
    printf("========================================\n");

    return (passed == 7) ? 0 : 1;
}