tests/utils_tests.h

   1 #include <time.h>
   2 #include <stdlib.h>
   3 #include <stdio.h>
   4 #include <math.h>
   5 #include <assert.h>
   6 #include "config.h"
   7
   8 #ifdef HAVE_STRING_H
   9 #include <string.h>
  10 #endif
  11
  12 #ifdef HAVE_UNISTD_H
  13 #include <unistd.h> // unlink, close
  14 #endif
  15
  16 #ifdef HAVE_LIMITS_H
  17 #include <limits.h> // PATH_MAX
  18 #endif /* HAVE_LIMITS_H */
  19 #ifndef PATH_MAX
  20 #define PATH_MAX 1024
  21 #endif
  22
  23 #if defined(HAVE_WIN_HACKS) && !defined(__GNUC__)
  24 #include <io.h> // _access
  25 #endif
  26
  27 // This macro is used to pass a string to msvc compiler: since msvc's -D flag
  28 // strips the quotes, we define the string without quotes and re-add them with
  29 // this macro.
  30
  31 #define REDEFINESTRING(x) #x
  32 #define DEFINEDSTRING(x) REDEFINESTRING(x)
  33
  34 #ifndef AUBIO_TESTS_SOURCE
  35 #error "AUBIO_TESTS_SOURCE is not defined"
  36 #endif
  37
  38 #ifdef HAVE_C99_VARARGS_MACROS
  39 #define PRINT_ERR(...)   fprintf(stderr, "AUBIO-TESTS ERROR: " __VA_ARGS__)
  40 #define PRINT_MSG(...)   fprintf(stdout, __VA_ARGS__)
  41 #define PRINT_DBG(...)   fprintf(stderr, __VA_ARGS__)
  42 #define PRINT_WRN(...)   fprintf(stderr, "AUBIO-TESTS WARNING: " __VA_ARGS__)
  43 #else
  44 #define PRINT_ERR(format, args...)   fprintf(stderr, "AUBIO-TESTS ERROR: " format , ##args)
  45 #define PRINT_MSG(format, args...)   fprintf(stdout, format , ##args)
  46 #define PRINT_DBG(format, args...)   fprintf(stderr, format , ##args)
  47 #define PRINT_WRN(format, args...)   fprintf(stderr, "AUBIO-TESTS WARNING: " format, ##args)
  48 #endif
  49
  50 #ifndef M_PI
  51 #define M_PI         (3.14159265358979323846)
  52 #endif
  53
  54 #ifndef RAND_MAX
  55 #define RAND_MAX 32767
  56 #endif
  57
  58 #if defined(HAVE_WIN_HACKS)
  59
  60 // use srand/rand on windows
  61 #define srandom srand
  62 #define random rand
  63
  64 #elif defined(__STRICT_ANSI__)
  65
  66 // workaround to build with -std=c99 (for instance with older cygwin),
  67 // assuming libbc is recent enough to supports these functions.
  68 extern void srandom(unsigned);
  69 extern int random(void);
  70 extern char mkstemp(const char *pat);
  71
  72 #endif
  73
  74 void utils_init_random (void);
  75
  76 void utils_init_random (void) {
  77   time_t now = time(0);
  78   struct tm *tm_struct = localtime(&now);
  79   size_t **tm_address = (void*)&tm_struct;
  80   int seed = tm_struct->tm_sec + (size_t)tm_address;
  81   //PRINT_WRN("current seed: %d\n", seed);
  82   srandom ((unsigned int)seed);
  83 }
  84
  85 // create_temp_sink / close_temp_sink
  86 #if defined(__GNUC__) // mkstemp
  87
  88 int check_source(char *source_path)
  89 {
  90   return access(source_path, R_OK);
  91 }
  92
  93 int create_temp_sink(char *sink_path)
  94 {
  95   return mkstemp(sink_path);
  96 }
  97
  98 int close_temp_sink(char *sink_path, int sink_fildes)
  99 {
 100   int err;
 101   if ((err = close(sink_fildes)) != 0) return err;
 102   if ((err = unlink(sink_path)) != 0) return err;
 103   return err;
 104 }
 105
 106 #elif defined(HAVE_WIN_HACKS) //&& !defined(__GNUC__)
 107 // windows workaround, where mkstemp does not exist...
 108
 109 int check_source(char *source_path)
 110 {
 111   return _access(source_path, 04);
 112 }
 113
 114 int create_temp_sink(char *templ)
 115 {
 116   int i = 0;
 117   static const char letters[] = "abcdefg0123456789";
 118   int letters_len = strlen(letters);
 119   int templ_len = strlen(templ);
 120   if (templ_len == 0) return 0;
 121   utils_init_random();
 122   for (i = 0; i < 6; i++)
 123   {
 124     templ[templ_len - i] = letters[rand() % letters_len];
 125   }
 126   return 1;
 127 }
 128
 129 int close_temp_sink(char* sink_path, int sink_fildes) {
 130   // the file should be closed when not using mkstemp, no need to open it
 131   if (sink_fildes == 0) return 1;
 132   return _unlink(sink_path);
 133 }
 134
 135 #else // windows workaround
 136 // otherwise, we don't really know what to do yet
 137 #error "mkstemp undefined, but not on windows. additional workaround required."
 138 #endif
 139
 140 // pass progname / default
 141 int run_on_default_source( int main(int, char**) )
 142 {
 143   const int argc = 2;
 144   int err = 0;
 145   char** argv = (char**)calloc(argc, sizeof(char*));
 146   argv[0] = __FILE__;
 147   argv[1] = DEFINEDSTRING(AUBIO_TESTS_SOURCE);
 148   // check if the file can be read
 149   if ( check_source(argv[1]) ) return 1;
 150   err = main(argc, argv);
 151   if (argv) free(argv);
 152   return err;
 153 }
 154
 155 int run_on_default_sink( int main(int, char**) )
 156 {
 157   const int argc = 2;
 158   int err = 0;
 159   char** argv = (char**)calloc(argc, sizeof(char*));
 160   char sink_path[PATH_MAX] = "tmp_aubio_XXXXXX";
 161   int fd = create_temp_sink(sink_path);
 162   if (!fd) return 1;
 163   argv[0] = __FILE__;
 164   argv[1] = sink_path;
 165   err = main(argc, argv);
 166   close_temp_sink(sink_path, fd);
 167   if (argv) free(argv);
 168   return err;
 169 }
 170
 171 int run_on_default_source_and_sink( int main(int, char**) )
 172 {
 173   const int argc = 3;
 174   int err = 0;
 175   char** argv = (char**)calloc(argc, sizeof(char*));
 176   char sink_path[PATH_MAX] = "tmp_aubio_XXXXXX";
 177   int fd = create_temp_sink(sink_path);
 178   if (!fd) return 1;
 179   argv[0] = __FILE__;
 180   argv[1] = DEFINEDSTRING(AUBIO_TESTS_SOURCE);
 181   argv[2] = sink_path;
 182   // check if the file can be read
 183   if ( check_source(argv[1]) ) return 1;
 184   err = main(argc, argv);
 185   close_temp_sink(sink_path, fd);
 186   if (argv) free(argv);
 187   return err;
 188 }