#include "../../lib/md2_impls/md2_2.h"

#include "../../lib/md2_impls/md2_common.h"

void process_block_hash(uint8_t block[16], uint8_t messageDigestBuf[48]) {
  for (int j = 0; j < 16; j++) {
    messageDigestBuf[16 + j] = block[j];
    messageDigestBuf[32 + j] = (messageDigestBuf[16 + j] ^ messageDigestBuf[j]);
  }

  u_int8_t t = 0;

  for (int j = 0; j < 18; j++) {
    for (int k = 0; k < 48; k++) {
      t = messageDigestBuf[k] = messageDigestBuf[k] ^ MD2_PI_SUBST[t];
    }
    t = (t + j) % 256;
  }
}

void process_block_checksum(uint8_t block[16], uint8_t checksum[16],
                            uint8_t* l) {
  for (int j = 0; j < 16; j++) {
    u_int8_t c = block[j];
    // reference is wrong. It says: Set C[j] to S[c xor L]. But it should be:
    (*l) = checksum[j] ^= MD2_PI_SUBST[c ^ (*l)];
  }
}

// unused!
void md2_checksum_2(size_t, uint8_t*) {}

void md2_hash_2(size_t len, const uint8_t buf[len], uint8_t out[16]) {
  // === step 1 ===
  int paddingNeeded = 16 - (len % 16);
  uint8_t originalPadding = paddingNeeded;
  len += paddingNeeded;

  // printf("len: %d\n", len);

  // +16 for the checksum
  uint8_t* newBuf = calloc(len, sizeof(uint8_t));
  // TODO: null check
  memcpy(newBuf, buf, len - paddingNeeded);

  // printBuf(len + 16, newBuf);

  while (paddingNeeded > 0) {
    newBuf[len - paddingNeeded] = originalPadding;
    paddingNeeded--;
  }
  // printf("buf with padding: ");
  // printBuf(len + 16, newBuf);

  // === step 3 ===
  uint8_t* messageDigestBuf = calloc(48, sizeof(uint8_t));

  // === step 4 ===
  uint8_t l = 0;
  uint8_t* checksum = calloc(16, sizeof(uint8_t));

  for (size_t i = 0; i < len / 16; i++) {
    process_block_checksum(newBuf + (i * 16), checksum, &l);
    process_block_hash(newBuf + (i * 16), messageDigestBuf);
  }

  process_block_hash(checksum, messageDigestBuf);

  memcpy(out, messageDigestBuf, 16);

  free(newBuf);
  free(messageDigestBuf);
  free(checksum);
}