#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <assert.h>
#include <sys/types.h>
#include "../src/extlib.h"

int
test_malloc_free_secure (size_t size)
{
  char *mall_test = malloc_secure (size);
  if (!mall_test)
    return 1;
  if (memvcmp (mall_test, 0, size) != 0)
    return 2;
  free_secure ((void **)&mall_test, size);
  if (mall_test)
    return 3;
  return 0;
}

int
test_fcopy()
{
  char *buf_a;
  char *buf_b;
  size_t len_a;
  size_t len_b;
  FILE *file_test_a = open_memstream (&buf_a, &len_a);
  FILE *file_test_b = open_memstream (&buf_b, &len_b);

  fprintf (file_test_a, "hello");
  fflush (file_test_a);
  fclose (file_test_a);
  file_test_a = fmemopen(buf_a, len_a, "r");

  fcopy (file_test_a, file_test_b);
  fclose (file_test_b);

  if (memcmp(buf_a, buf_b, len_a) != 0) {
    free_secure ((void**)&buf_a, len_a);
    free_secure ((void **)&buf_b, len_b);
    return 1;
  }
  free_secure ((void**)&buf_a, len_a);
  free_secure ((void **)&buf_b, len_b);
  return 0;
}

int
test_strlwr_strupr ()
{
  char *test_string = strdup ("hello");
  char *uppr_string = strupr (test_string);
  char *lower_string = strlwr (uppr_string);

  if (strcmp (uppr_string, "HELLO") != 0) {
    free_secure ((void **)&test_string, strlen (test_string));
    return 1;
  }
  if (strcmp (lower_string, "hello") != 0) {
    free_secure ((void **)&test_string, strlen (test_string));
    return 2;
  }
  free_secure ((void **)&test_string, strlen (test_string));
  return 0;
}

int
test_trim()
{
  char *test_string = strdup ("\t\thi\t\t");
  char *trimmed_string = trim (test_string);
  if (strstr (trimmed_string, "\t") != NULL) {
    free_secure ((void**)&test_string, strlen (test_string));
    return 1;
  }
  free_secure ((void**)&test_string, strlen (test_string));
  return 0;
}

int
test_replace_str ()
{
  char *test_string = strdup ("replace world!");
  char *replaced_string = replace_str (test_string, "replace", "hello");
  if (strcmp (replaced_string, "hello world!") != 0) {
    free_secure ((void**)&test_string, strlen (test_string));
    free_secure ((void**)&replaced_string, strlen (replaced_string));
    return 1;
  }
  free_secure ((void**)&test_string, strlen (test_string));
  free_secure ((void**)&replaced_string, strlen (replaced_string));
  return 0;
}


int
main (void)
{
  printf ("=== extlib unit tests ===\n");
  printf ("\n");

  printf ("Test Case 1: test_malloc_free_secure -- malloc_secure free_secure memvcmp\n");
  int test_1_result = test_malloc_free_secure (12);
  if (test_1_result == 0)
    printf ("Test Case 1: test_malloc_free_secure -- SUCCESS\n\n");
  else
    printf ("Test Case 1: test_malloc_free_secure -- FAILED %d\n\n", test_1_result);

  printf ("Test Case 2: test_fcopy -- fcopy\n");
  int test_2_result = test_fcopy ();
  if (test_2_result == 0)
    printf ("Test Case 2: test_fcopy -- SUCCESS\n\n");
  else
    printf ("Test Case 2: test_fcopy -- FAILED %d\n\n", test_2_result);

  printf ("Test Case 3: test_strlwr_strupr -- strlwr strupr\n");
  int test_3_result = test_strlwr_strupr ();
  if (test_2_result == 0)
    printf ("Test Case 3: test_strlwr_strupr -- SUCCESS\n\n");
  else
    printf ("Test Case 3: test_strlwr_strupr -- FAILED %d\n\n", test_3_result);

  printf ("Test Case 4: test_trim -- trim\n");
  int test_4_result = test_trim ();
  if (test_4_result == 0)
    printf ("Test Case 4: test_trim -- SUCCESS\n\n");
  else
    printf ("Test Case 4: test_trim -- FAILED %d\n\n", test_4_result);

  printf ("Test Case 5: test_replace_str -- replace_str\n");
  int test_5_result = test_replace_str();
  if (test_5_result == 0)
    printf ("Test Case 5: test_replace_str -- SUCCESS\n\n");
  else
    printf ("Test Case 5: test_replace_str -- FAILED %d\n\n", test_5_result);
  return 0;
}