#define SLOWCRYPT_CHACHA20_IMPL
#include "../chacha20.h"

static char key[] = {
    0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a,
    0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15,
    0x16, 0x17, 0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f,
};

static char nonce[] = {
    0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x4a, 0x00, 0x00, 0x00, 0x00,
};

static char text[] =
    "Ladies and Gentlemen of the class of '99: If I could offer you only one "
    "tip for the future, sunscreen would be it.";

static char expected[] = {
    0x6E, 0x2E, 0x35, 0x9A, 0x25, 0x68, 0xF9, 0x80, 0x41, 0xBA, 0x07, 0x28,
    0xDD, 0x0D, 0x69, 0x81, 0xE9, 0x7E, 0x7A, 0xEC, 0x1D, 0x43, 0x60, 0xC2,
    0x0A, 0x27, 0xAF, 0xCC, 0xFD, 0x9F, 0xAE, 0x0B, 0xF9, 0x1B, 0x65, 0xC5,
    0x52, 0x47, 0x33, 0xAB, 0x8F, 0x59, 0x3D, 0xAB, 0xCD, 0x62, 0xB3, 0x57,
    0x16, 0x39, 0xD6, 0x24, 0xE6, 0x51, 0x52, 0xAB, 0x8F, 0x53, 0x0C, 0x35,
    0x9F, 0x08, 0x61, 0xD8, 0x07, 0xCA, 0x0D, 0xBF, 0x50, 0x0D, 0x6A, 0x61,
    0x56, 0xA3, 0x8E, 0x08, 0x8A, 0x22, 0xB6, 0x5E, 0x52, 0xBC, 0x51, 0x4D,
    0x16, 0xCC, 0xF8, 0x06, 0x81, 0x8C, 0xE9, 0x1A, 0xB7, 0x79, 0x37, 0x36,
    0x5A, 0xF9, 0x0B, 0xBF, 0x74, 0xA3, 0x5B, 0xE6, 0xB4, 0x0B, 0x8E, 0xED,
    0xF2, 0x78, 0x5E, 0x42, 0x87, 0x4D,
};

int main(int argc, char **argv) {
  slowcrypt_chacha20 state[2];
  char buf[64];
  int i, j, blksz;
  uint32_t ctr = 1;

  (void)argc;
  (void)argv;

  for (i = 0; i < (int)(sizeof(text) - 1); i += 64) {
    blksz = (sizeof(text) - 1) - i;
    if (blksz > 64)
      blksz = 64;

    for (j = 0; j < blksz; j++) {
      buf[j] = text[i + j];
    }

    slowcrypt_chacha20_block(state, key, ctr, nonce, buf);
    ctr++;

    for (j = 0; j < blksz; j++)
      if (buf[j] != expected[i + j])
        return 1;
  }

  slowcrypt_chacha20_deinit(&state[0]);
  slowcrypt_chacha20_deinit(&state[1]);
  for (i = 0; i < 64; i++)
    buf[i] = 0;

  return 0;
}