libtom.libtomcrypt/tests/common.c

/* LibTomCrypt, modular cryptographic library -- Tom St Denis */
/* SPDX-License-Identifier: Unlicense */

#include "common.h"

/**
  @file common.c

  Steffen Jaeckel
*/

void run_cmd(int res, int line, const char *file, const char *cmd, const char *algorithm)
{
   if (res != CRYPT_OK) {
      fprintf(stderr, "%s (%d)%s%s\n%s:%d:%s\n",
              error_to_string(res), res,
              (algorithm ? " - " : ""), (algorithm ? algorithm : ""),
              file, line, cmd);
      if (res != CRYPT_NOP) {
         exit(EXIT_FAILURE);
      }
   }
}

void print_hex(const char* what, const void* v, const unsigned long l)
{
  const unsigned char* p = v;
  unsigned long x, y = 0, z;
  fprintf(stderr, "%s contents: \n", what);
  for (x = 0; x < l; ) {
      fprintf(stderr, "%02X ", p[x]);
      if (!(++x % 16) || x == l) {
         if((x % 16) != 0) {
            z = 16 - (x % 16);
            if(z >= 8)
               fprintf(stderr, " ");
            for (; z != 0; --z) {
               fprintf(stderr, "   ");
            }
         }
         fprintf(stderr, " | ");
         for(; y < x; y++) {
            if((y % 8) == 0)
               fprintf(stderr, " ");
            if(isgraph(p[y]))
               fprintf(stderr, "%c", p[y]);
            else
               fprintf(stderr, ".");
         }
         fprintf(stderr, "\n");
      }
      else if((x % 8) == 0) {
         fprintf(stderr, " ");
      }
  }
}

#ifdef LTC_TEST_READDIR

#include <sys/stat.h>
#include <sys/types.h>
#include <dirent.h>

static off_t fsize(const char *filename)
{
   struct stat st;

   if (stat(filename, &st) == 0) {
      /* filename is no regular file */
      if (!S_ISREG(st.st_mode))
         return 0;
      return st.st_size;
   }

   return -1;
}
static DIR *s_opendir(const char *path, char *mypath, unsigned long l)
{
#ifdef CMAKE_SOURCE_DIR
#define SOURCE_PREFIX CMAKE_SOURCE_DIR "/"
#else
#define SOURCE_PREFIX ""
#endif
   DIR *d = NULL;
   int r = snprintf(mypath, l, "%s%s", SOURCE_PREFIX, path);
   if (r > 0 && (unsigned int)r < l) {
      d = opendir(mypath);
   }

   return d;
}

static int s_read_and_process(FILE *f, unsigned long sz, void *ctx, dir_iter_cb process)
{
   int err = CRYPT_OK;
   void* buf;
   if (f == NULL)
      return CRYPT_FILE_NOTFOUND;
   buf = XMALLOC(sz + 1);
   if (buf == NULL)
      return CRYPT_MEM;
   if (fread(buf, 1, sz, f) != sz) {
      err = CRYPT_ERROR;
      goto out;
   }
   ((unsigned char *)buf)[sz] = 0x0;
   err = process(buf, sz, ctx);
out:
   XFREE(buf);
   return err;
}

int test_process_dir(const char *path, void *ctx, dir_iter_cb iter, dir_fiter_cb fiter, dir_cleanup_cb cleanup, const char *test)
{
   char mypath[PATH_MAX];
   DIR *d = s_opendir(path, mypath, sizeof(mypath));
   struct dirent *de;
   char fname[PATH_MAX];
   FILE *f = NULL;
   off_t fsz;
   int err = CRYPT_FILE_NOTFOUND;
   if (d == NULL)
      return CRYPT_FILE_NOTFOUND;
   while((de = readdir(d)) != NULL) {
      fname[0] = '\0';
      if (strcmp(de->d_name, ".") == 0 || strcmp(de->d_name, "..") == 0 || strcmp(de->d_name, "README.txt") == 0)
         continue;
      strcat(fname, mypath);
      strcat(fname, "/");
      strcat(fname, de->d_name);
      fsz = fsize(fname);
      if (fsz == 0)
         continue;
      if (fsz == -1) {
         err = CRYPT_FILE_NOTFOUND;
         break;
      }
#if defined(LTC_TEST_DBG) && LTC_TEST_DBG > 1
      fprintf(stderr, "%s: Try to process %s\n", test, fname);
#endif
      f = fopen(fname, "rb");

      if (iter) {
         err = s_read_and_process(f, fsz, ctx, iter);
      } else if (fiter) {
         err = fiter(f, ctx);
      } else {
         err = CRYPT_NOP;
#if defined(LTC_TEST_DBG) && LTC_TEST_DBG > 1
         fprintf(stderr, "%s: No call-back set for %s\n", test, fname);
#endif
      }

      if (err == CRYPT_NOP) {
#if defined(LTC_TEST_DBG) && LTC_TEST_DBG > 1
         fprintf(stderr, "%s: Skip: %s\n", test, fname);
#endif
         err = CRYPT_OK;
         goto continue_loop;
      } else if (err != CRYPT_OK) {
#if defined(LTC_TEST_DBG)
         fprintf(stderr, "%s: Test %s failed (cause: %s).\n\n", test, fname, error_to_string(err));
#else
         LTC_UNUSED_PARAM(test);
#endif
         break;
      }
      if (cleanup != NULL) {
         cleanup(ctx);
      }

continue_loop:
      if (f != NULL) fclose(f);
      f = NULL;
   }
   if (f != NULL) fclose(f);
   closedir(d);
   return err;
}
#endif


prng_state yarrow_prng;