#include <stdio.h>
#include <math.h>
#include <stdint.h>
#include "../wots.h"
#include "../hash.h"
#include "../randombytes.h"

static void hexdump(unsigned char *a, size_t len)
{
  size_t i;
  for (i = 0; i < len; i++)
    printf("%02x", a[i]);
}

int main()
{
  int n = 32;
  unsigned char seed[n];
  unsigned char pub_seed[n];
  wots_params params;
  wots_set_params(&params, n, 16, XMSS_SHA2);

  int sig_len = params.len*params.n;

  unsigned char pk1[sig_len];
  unsigned char pk2[sig_len];
  unsigned char sig[sig_len];
  uint32_t addr[8] = {1,2,3,4};

  unsigned char msg[n];
  int i;

  randombytes(seed, n);
  randombytes(pub_seed, n);
  randombytes(msg, n);
  //randombytes(addr, 16);

  wots_pkgen(pk1, seed, &params, pub_seed, addr);
  wots_sign(sig, msg, seed, &params, pub_seed, addr);
  wots_pkFromSig(pk2, sig, msg, &params, pub_seed, addr);

  for (i = 0; i < sig_len; i++)
    if (pk1[i] != pk2[i]) {
      printf("pk1 != pk2 %d\n",i);
      return -1;
    }
  printf("worked great!\npk1: ");
  hexdump(pk1, sig_len);
  printf("\npk2: ");
  hexdump(pk2, sig_len);
  printf("\nsig: ");
  hexdump(sig, sig_len);
  printf("\n");

  return 0;
}