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
utils_init_random(void)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
check_source(char * source_path)88 int check_source(char *source_path)
89 {
90 return access(source_path, R_OK);
91 }
92
create_temp_sink(char * sink_path)93 int create_temp_sink(char *sink_path)
94 {
95 return mkstemp(sink_path);
96 }
97
close_temp_sink(char * sink_path,int sink_fildes)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
check_source(char * source_path)109 int check_source(char *source_path)
110 {
111 return _access(source_path, 04);
112 }
113
create_temp_sink(char * templ)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
close_temp_sink(char * sink_path,int sink_fildes)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
run_on_default_source(int main (int,char **))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
run_on_default_sink(int main (int,char **))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
run_on_default_source_and_sink(int main (int,char **))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 }
189