5 * Permission is hereby granted, free of charge, to any person obtaining a copy of
6 * this software and associated documentation files (the "Software"), to deal in
7 * the Software without restriction, including without limitation the rights to
8 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
9 * of the Software, and to permit persons to whom the Software is furnished to do
10 * so, subject to the following conditions:
12 * The above copyright notice and this permission notice shall be included in all
13 * copies or substantial portions of the Software.
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
24 #include <sys/types.h>
28 bool opts_memchk = false;
29 bool opts_debug = false;
36 * TODO: Windows version
37 * this implements a unique bi-directional popen-like function that
38 * allows reading data from both stdout and stderr. And writing to
42 * FILE *handles[3] = task_popen("ls", "-l", "r");
43 * if (!handles) { perror("failed to open stdin/stdout/stderr to ls");
44 * // handles[0] = stdin
45 * // handles[1] = stdout
46 * // handles[2] = stderr
48 * task_pclose(handles); // to close
51 #include <sys/types.h>
64 FILE ** task_popen(const char *command, const char *mode) {
70 popen_t *data = mem_a(sizeof(popen_t));
73 * Parse the command now into a list for execv, this is a pain
76 char *line = (char*)command;
80 while (*line != '\0') {
81 while (*line == ' ' || *line == '\t' || *line == '\n')
85 while (*line != '\0' && *line != ' ' &&
86 *line != '\t' && *line != '\n') line++;
92 if ((trypipe = pipe(inhandle)) < 0) goto task_popen_error_0;
93 if ((trypipe = pipe(outhandle)) < 0) goto task_popen_error_1;
94 if ((trypipe = pipe(errhandle)) < 0) goto task_popen_error_2;
96 if ((data->pid = fork()) > 0) {
100 close(errhandle [1]);
102 data->pipes [0] = inhandle [1];
103 data->pipes [1] = outhandle[0];
104 data->pipes [2] = errhandle[0];
105 data->handles[0] = fdopen(inhandle [1], "w");
106 data->handles[1] = fdopen(outhandle[0], mode);
107 data->handles[2] = fdopen(errhandle[0], mode);
112 return data->handles;
113 } else if (data->pid == 0) {
119 /* see piping documentation for this sillyness :P */
120 close(0), dup(inhandle [0]);
121 close(1), dup(outhandle[1]);
122 close(2), dup(errhandle[1]);
128 goto task_popen_error_3;
133 return data->handles;
135 task_popen_error_3: close(errhandle[0]), close(errhandle[1]);
136 task_popen_error_2: close(outhandle[0]), close(outhandle[1]);
137 task_popen_error_1: close(inhandle [0]), close(inhandle [1]);
145 int task_pclose(FILE **handles) {
146 popen_t *data = (popen_t*)handles;
149 close(data->pipes[0]); /* stdin */
150 close(data->pipes[1]); /* stdout */
151 close(data->pipes[2]); /* stderr */
153 waitpid(data->pid, &status, 0);
163 #define TASK_COMPILE 0
164 #define TASK_EXECUTE 1
167 * Task template system:
168 * templates are rules for a specific test, used to create a "task" that
169 * is executed with those set of rules (arguments, and what not). Tests
170 * that don't have a template with them cannot become tasks, since without
171 * the information for that test there is no way to properly "test" them.
172 * Rules for these templates are described in a template file, using a
173 * task template language.
175 * The language is a basic finite statemachine, top-down single-line
176 * description language.
178 * The languge is composed entierly of "tags" which describe a string of
179 * text for a task. Think of it much like a configuration file. Except
180 * it's been designed to allow flexibility and future support for prodecual
183 * The following "tags" are suported by the language
186 * Used to set a description of the current test, this must be
187 * provided, this tag is NOT optional.
190 * Used to set a failure message, this message will be displayed
191 * if the test fails, this tag is optional
194 * Used to set a success message, this message will be displayed
195 * if the test succeeds, this tag is optional.
198 * Used to set the procedure for the given task, there are two
201 * This simply performs compilation only
203 * This will perform compilation and execution
205 * This must be provided, this tag is NOT optional.
208 * Used to set the compilation flags for the given task, this
209 * must be provided, this tag is NOT optional.
212 * Used to set the execution flags for the given task. This tag
213 * must be provided if T == -execute, otherwise it's erroneous
214 * as compilation only takes place.
217 * Used to describe a string of text that should be matched from
218 * the output of executing the task. If this doesn't match the
219 * task fails. This tag must be provided if T == -execute, otherwise
220 * it's erroneous as compilation only takes place.
223 * Used to specify the INPUT source file to operate on, this must be
224 * provided, this tag is NOT optional
228 * These tags have one-time use, using them more than once will result
229 * in template compilation errors.
231 * Lines beginning with # or // in the template file are comments and
232 * are ignored by the template parser.
234 * Whitespace is optional, with exception to the colon ':' between the
235 * tag and it's assignment value/
237 * The template compiler will detect erronrous tags (optional tags
238 * that need not be set), as well as missing tags, and error accordingly
239 * this will result in the task failing.
243 char *failuremessage;
244 char *successmessage;
254 * This is very much like a compiler code generator :-). This generates
255 * a value from some data observed from the compiler.
257 bool task_template_generate(task_template_t *template, char tag, const char *file, size_t line, const char *value) {
258 char **destval = NULL;
264 case 'D': destval = &template->description; break;
265 case 'F': destval = &template->failuremessage; break;
266 case 'S': destval = &template->successmessage; break;
267 case 'T': destval = &template->proceduretype; break;
268 case 'C': destval = &template->compileflags; break;
269 case 'E': destval = &template->executeflags; break;
270 case 'I': destval = &template->sourcefile; break;
272 con_printmsg(LVL_ERROR, __FILE__, __LINE__, "internal error",
273 "invalid tag `%c:` during code generation\n",
280 * Ensure if for the given tag, there already exists a
284 con_printmsg(LVL_ERROR, file, line, "compile error",
285 "tag `%c:` already assigned value: %s\n",
292 * Strip any whitespace that might exist in the value for assignments
295 if (value && *value && (*value == ' ' || *value == '\t'))
299 * Value will contain a newline character at the end, we need to strip
300 * this otherwise kaboom, seriously, kaboom :P
302 *strrchr(value, '\n')='\0';
305 * Now allocate and set the actual value for the specific tag. Which
306 * was properly selected and can be accessed with *destval.
308 *destval = util_strdup(value);
313 bool task_template_parse(const char *file, task_template_t *template, FILE *fp) {
322 /* top down parsing */
323 while (util_getline(&back, &size, fp) != EOF) {
324 /* skip whitespace */
326 if (*data && (*data == ' ' || *data == '\t'))
331 * Handle comments inside task template files. We're strict
332 * about the language for fun :-)
335 if (data[1] != '/') {
336 con_printmsg(LVL_ERROR, file, line, "template parse error",
337 "invalid character `/`, perhaps you meant `//` ?");
346 * Empty newlines are acceptable as well, so we handle that here
347 * despite being just odd since there should't be that many
348 * empty lines to begin with.
356 * Now begin the actual "tag" stuff. This works as you expect
366 if (data[1] != ':') {
367 con_printmsg(LVL_ERROR, file, line, "template parse error",
368 "expected `:` after `%c`",
373 if (!task_template_generate(template, *data, file, line, &data[3])) {
374 con_printmsg(LVL_ERROR, file, line, "template compile error",
375 "failed to generate for given task\n"
382 * Match requires it's own system since we allow multiple M's
383 * for multi-line matching.
387 char *value = &data[3];
388 if (data[1] != ':') {
389 con_printmsg(LVL_ERROR, file, line, "template parse error",
390 "expected `:` after `%c`",
396 if (value && *value && (*value == ' ' || *value == '\t'))
400 * Value will contain a newline character at the end, we need to strip
401 * this otherwise kaboom, seriously, kaboom :P
403 *strrchr(value, '\n')='\0';
405 vec_push(template->comparematch, util_strdup(value));
411 con_printmsg(LVL_ERROR, file, line, "template parse error",
412 "invalid tag `%c`", *data
415 /* no break required */
418 /* update line and free old sata */
434 * Nullifies the template data: used during initialization of a new
437 void task_template_nullify(task_template_t *template) {
441 template->description = NULL;
442 template->failuremessage = NULL;
443 template->successmessage = NULL;
444 template->proceduretype = NULL;
445 template->compileflags = NULL;
446 template->executeflags = NULL;
447 template->comparematch = NULL;
448 template->sourcefile = NULL;
449 template->tempfilename = NULL;
452 task_template_t *task_template_compile(const char *file, const char *dir) {
453 /* a page should be enough */
455 FILE *tempfile = NULL;
456 task_template_t *template = NULL;
458 memset (fullfile, 0, sizeof(fullfile));
459 snprintf(fullfile, sizeof(fullfile), "%s/%s", dir, file);
461 tempfile = fopen(fullfile, "r");
462 template = mem_a(sizeof(task_template_t));
463 task_template_nullify(template);
466 * Esnure the file even exists for the task, this is pretty useless
470 con_err("template file: %s does not exist or invalid permissions\n",
476 if (!task_template_parse(file, template, tempfile)) {
477 con_err("template parse error: error during parsing\n");
482 * Regardless procedure type, the following tags must exist:
488 if (!template->description) {
489 con_err("template compile error: %s missing `D:` tag\n", file);
492 if (!template->proceduretype) {
493 con_err("template compile error: %s missing `T:` tag\n", file);
496 if (!template->compileflags) {
497 con_err("template compile error: %s missing `C:` tag\n", file);
500 if (!template->sourcefile) {
501 con_err("template compile error: %s missing `I:` tag\n", file);
506 * Now lets compile the template, compilation is really just
507 * the process of validating the input.
509 if (!strcmp(template->proceduretype, "-compile")) {
510 if (template->executeflags)
511 con_err("template compile warning: %s erroneous tag `E:` when only compiling\n", file);
512 if (template->comparematch)
513 con_err("template compile warning: %s erroneous tag `M:` when only compiling\n", file);
515 } else if (!strcmp(template->proceduretype, "-execute")) {
516 if (!template->executeflags) {
517 /* default to $null */
518 template->executeflags = util_strdup("$null");
520 if (!template->comparematch) {
521 con_err("template compile error: %s missing `M:` tag (use `$null` for exclude)\n", file);
525 con_err("template compile error: %s invalid procedure type: %s\n", file, template->proceduretype);
535 * The file might not exist and we jump here when that doesn't happen
536 * so the check to see if it's not null here is required.
545 void task_template_destroy(task_template_t **template) {
549 if ((*template)->description) mem_d((*template)->description);
550 if ((*template)->failuremessage) mem_d((*template)->failuremessage);
551 if ((*template)->successmessage) mem_d((*template)->successmessage);
552 if ((*template)->proceduretype) mem_d((*template)->proceduretype);
553 if ((*template)->compileflags) mem_d((*template)->compileflags);
554 if ((*template)->executeflags) mem_d((*template)->executeflags);
555 if ((*template)->sourcefile) mem_d((*template)->sourcefile);
558 * Delete all allocated string for task template then destroy the
563 for (; i < vec_size((*template)->comparematch); i++)
564 mem_d((*template)->comparematch[i]);
566 vec_free((*template)->comparematch);
570 * Nullify all the template members otherwise NULL comparision
571 * checks will fail if template pointer is reused.
577 * Now comes the task manager, this system allows adding tasks in and out
578 * of a task list. This is the executor of the tasks essentially as well.
581 task_template_t *template;
590 task_t *task_tasks = NULL;
593 * Read a directory and searches for all template files in it
594 * which is later used to run all tests.
596 bool task_propagate(const char *curdir) {
599 struct dirent *files;
600 struct stat directory;
604 dir = opendir(curdir);
606 while ((files = readdir(dir))) {
607 memset (buffer, 0,sizeof(buffer));
608 snprintf(buffer, sizeof(buffer), "%s/%s", curdir, files->d_name);
610 if (stat(buffer, &directory) == -1) {
611 con_err("internal error: stat failed, aborting\n");
615 /* skip directories */
616 if (S_ISDIR(directory.st_mode))
620 * We made it here, which concludes the file/directory is not
621 * actually a directory, so it must be a file :)
623 if (strcmp(files->d_name + strlen(files->d_name) - 5, ".tmpl") == 0) {
624 task_template_t *template = task_template_compile(files->d_name, curdir);
625 char buf[4096]; /* one page should be enough */
628 util_debug("TEST", "compiling task template: %s/%s\n", curdir, files->d_name);
631 con_err("error compiling task template: %s\n", files->d_name);
636 * Generate a temportary file name for the output binary
637 * so we don't trample over an existing one.
639 template->tempfilename = tempnam(curdir, "TMPDAT");
642 * Generate the command required to open a pipe to a process
643 * which will be refered to with a handle in the task for
644 * reading the data from the pipe.
646 memset (buf,0,sizeof(buf));
647 snprintf(buf, sizeof(buf), "%s %s/%s %s -o %s",
648 task_bins[TASK_COMPILE],
650 template->sourcefile,
651 template->compileflags,
652 template->tempfilename
656 * The task template was compiled, now lets create a task from
657 * the template data which has now been propagated.
659 task.template = template;
660 if (!(task.runhandles = task_popen(buf, "r"))) {
661 con_err("error opening pipe to process for test: %s\n", template->description);
666 util_debug("TEST", "executing test: `%s` [%s]\n", template->description, buf);
669 * Open up some file desciptors for logging the stdout/stderr
672 memset (buf,0,sizeof(buf));
673 snprintf(buf, sizeof(buf), "%s.stdout", template->tempfilename);
674 task.stdoutlogfile = util_strdup(buf);
675 if (!(task.stdoutlog = fopen(buf, "w"))) {
676 con_err("error opening %s for stdout\n", buf);
680 memset (buf,0,sizeof(buf));
681 snprintf(buf, sizeof(buf), "%s.stderr", template->tempfilename);
682 task.stderrlogfile = util_strdup(buf);
683 if (!(task.stderrlog = fopen(buf, "w"))) {
684 con_err("error opening %s for stderr\n", buf);
688 vec_push(task_tasks, task);
692 util_debug("TEST", "compiled %d task template files out of %d\n",
693 vec_size(task_tasks),
702 * Removes all temporary 'progs.dat' files created during compilation
705 void task_cleanup(const char *curdir) {
707 struct dirent *files;
710 dir = opendir(curdir);
712 while ((files = readdir(dir))) {
713 memset(buffer, 0, sizeof(buffer));
714 if (strstr(files->d_name, "TMP")) {
715 snprintf(buffer, sizeof(buffer), "%s/%s", curdir, files->d_name);
717 con_err("error removing temporary file: %s\n", buffer);
719 util_debug("TEST", "removed temporary file: %s\n", buffer);
727 * Task precleanup removes any existing temporary files or log files
728 * left behind from a previous invoke of the test-suite.
730 void task_precleanup(const char *curdir) {
732 struct dirent *files;
735 dir = opendir(curdir);
737 while ((files = readdir(dir))) {
738 memset(buffer, 0, sizeof(buffer));
739 if (strstr(files->d_name, "TMP") ||
740 strstr(files->d_name, ".stdout") ||
741 strstr(files->d_name, ".stderr"))
743 snprintf(buffer, sizeof(buffer), "%s/%s", curdir, files->d_name);
745 con_err("error removing temporary file: %s\n", buffer);
747 util_debug("TEST", "removed temporary file: %s\n", buffer);
754 void task_destroy(const char *curdir) {
756 * Free all the data in the task list and finally the list itself
757 * then proceed to cleanup anything else outside the program like
761 for (i = 0; i < vec_size(task_tasks); i++) {
763 * Close any open handles to files or processes here. It's mighty
764 * annoying to have to do all this cleanup work.
766 if (task_tasks[i].runhandles) task_pclose(task_tasks[i].runhandles);
767 if (task_tasks[i].stdoutlog) fclose (task_tasks[i].stdoutlog);
768 if (task_tasks[i].stderrlog) fclose (task_tasks[i].stderrlog);
771 * Only remove the log files if the test actually compiled otherwise
774 if (task_tasks[i].compiled) {
775 if (remove(task_tasks[i].stdoutlogfile))
776 con_err("error removing stdout log file: %s\n", task_tasks[i].stdoutlogfile);
778 util_debug("TEST", "removed stdout log file: %s\n", task_tasks[i].stdoutlogfile);
780 if (remove(task_tasks[i].stderrlogfile))
781 con_err("error removing stderr log file: %s\n", task_tasks[i].stderrlogfile);
783 util_debug("TEST", "removed stderr log file: %s\n", task_tasks[i].stderrlogfile);
786 /* free util_strdup data for log files */
787 mem_d(task_tasks[i].stdoutlogfile);
788 mem_d(task_tasks[i].stderrlogfile);
790 task_template_destroy(&task_tasks[i].template);
792 vec_free(task_tasks);
795 * Cleanup outside stuff like temporary files.
797 task_cleanup(curdir);
801 * This executes the QCVM task for a specificly compiled progs.dat
802 * using the template passed into it for call-flags and user defined
805 bool task_execute(task_template_t *template, char ***line) {
806 bool success = false;
809 memset (buffer,0,sizeof(buffer));
812 * Drop the execution flags for the QCVM if none where
813 * actually specified.
815 if (!strcmp(template->executeflags, "$null")) {
816 snprintf(buffer, sizeof(buffer), "%s %s",
817 task_bins[TASK_EXECUTE],
818 template->tempfilename
821 snprintf(buffer, sizeof(buffer), "%s %s %s",
822 task_bins[TASK_EXECUTE],
823 template->executeflags,
824 template->tempfilename
828 util_debug("TEST", "executing qcvm: `%s` [%s]\n",
829 template->description,
833 execute = popen(buffer, "r");
838 * Now lets read the lines and compare them to the matches we expect
839 * and handle accordingly.
845 while (util_getline(&data, &size, execute) != EOF) {
846 if (!strcmp(data, "No main function found\n")) {
847 con_err("test failure: `%s` [%s] (No main function found)\n",
848 template->description,
849 (template->failuremessage) ?
850 template->failuremessage : "unknown"
857 * Trim newlines from data since they will just break our
858 * ability to properly validate matches.
860 if (strrchr(data, '\n'))
861 *strrchr(data, '\n') = '\0';
865 * We only care about the last line from the output for now
866 * implementing multi-line match is TODO.
868 success = !!!(strcmp(data, template->comparematch[compare++]));
871 * Copy to output vector for diagnostics if execution match
874 vec_push(*line, data);
884 * This schedualizes all tasks and actually runs them individually
885 * this is generally easy for just -compile variants. For compile and
886 * execution this takes more work since a task needs to be generated
887 * from thin air and executed INLINE.
889 void task_schedualize() {
890 bool execute = false;
896 util_debug("TEST", "found %d tasks, preparing to execute\n", vec_size(task_tasks));
898 for (i = 0; i < vec_size(task_tasks); i++) {
899 util_debug("TEST", "executing task: %d: %s\n", i, task_tasks[i].template->description);
901 * Generate a task from thin air if it requires execution in
904 if (!strcmp(task_tasks[i].template->proceduretype, "-execute"))
908 * We assume it compiled before we actually compiled :). On error
909 * we change the value
911 task_tasks[i].compiled = true;
914 * Read data from stdout first and pipe that stuff into a log file
915 * then we do the same for stderr.
917 while (util_getline(&data, &size, task_tasks[i].runhandles[1]) != EOF) {
918 fputs(data, task_tasks[i].stdoutlog);
920 if (strstr(data, "failed to open file")) {
921 task_tasks[i].compiled = false;
925 fflush(task_tasks[i].stdoutlog);
927 while (util_getline(&data, &size, task_tasks[i].runhandles[2]) != EOF) {
929 * If a string contains an error we just dissalow execution
932 * TODO: make this more percise, e.g if we print a warning
933 * that refers to a variable named error, or something like
934 * that .. then this will blowup :P
936 if (strstr(data, "error")) {
938 task_tasks[i].compiled = false;
941 fputs(data, task_tasks[i].stderrlog);
942 fflush(task_tasks[i].stdoutlog);
946 con_err("test failure: `%s` [%s] (failed to compile) see %s.stdout and %s.stderr\n",
947 task_tasks[i].template->description,
948 (task_tasks[i].template->failuremessage) ?
949 task_tasks[i].template->failuremessage : "unknown",
950 task_tasks[i].template->tempfilename,
951 task_tasks[i].template->tempfilename
956 * If we made it here that concludes the task is to be executed
957 * in the virtual machine.
959 if (!task_execute(task_tasks[i].template, &match)) {
962 con_err("test failure: `%s` [%s] (invalid results from execution)\n",
963 task_tasks[i].template->description,
964 (task_tasks[i].template->failuremessage) ?
965 task_tasks[i].template->failuremessage : "unknown"
969 * Print nicely formatted expected match lists to console error
970 * handler for the all the given matches in the template file and
971 * what was actually returned from executing.
973 con_err(" Expected From %u Matches:\n", vec_size(task_tasks[i].template->comparematch));
974 for (; d < vec_size(task_tasks[i].template->comparematch); d++) {
975 char *select = task_tasks[i].template->comparematch[d];
976 size_t length = 40 - strlen(select);
978 con_err(" Expected: \"%s\"", select);
981 con_err("| Got: \"%s\"\n", (d >= vec_size(match)) ? "<<nothing else to compare>>" : match[d]);
987 con_out("test succeeded: `%s` [%s]\n",
988 task_tasks[i].template->description,
989 (task_tasks[i].template->successmessage) ?
990 task_tasks[i].template->successmessage : "unknown"
997 * This is the heart of the whole test-suite process. This cleans up
998 * any existing temporary files left behind as well as log files left
999 * behind. Then it propagates a list of tests from `curdir` by scaning
1000 * it for template files and compiling them into tasks, in which it
1001 * schedualizes them (executes them) and actually reports errors and
1002 * what not. It then proceeds to destroy the tasks and return memory
1003 * it's the engine :)
1005 * It returns true of tests could be propagated, otherwise it returns
1008 * It expects con_init() was called before hand.
1010 bool test_perform(const char *curdir) {
1011 task_precleanup(curdir);
1012 if (!task_propagate(curdir)) {
1013 con_err("error: failed to propagate tasks\n");
1014 task_destroy(curdir);
1018 * If we made it here all tasks where propagated from their resultant
1019 * template file. So we can start the FILO scheduler, this has been
1020 * designed in the most thread-safe way possible for future threading
1021 * it's designed to prevent lock contention, and possible syncronization
1025 task_destroy(curdir);
1031 * Fancy GCC-like LONG parsing allows things like --opt=param with
1032 * assignment operator. This is used for redirecting stdout/stderr
1033 * console to specific files of your choice.
1035 static bool parsecmd(const char *optname, int *argc_, char ***argv_, char **out, int ds, bool split) {
1037 char **argv = *argv_;
1039 size_t len = strlen(optname);
1041 if (strncmp(argv[0]+ds, optname, len))
1044 /* it's --optname, check how the parameter is supplied */
1045 if (argv[0][ds+len] == '=') {
1046 *out = argv[0]+ds+len+1;
1050 if (!split || argc < ds) /* no parameter was provided, or only single-arg form accepted */
1053 /* using --opt param */
1060 int main(int argc, char **argv) {
1061 char *redirout = (char*)stdout;
1062 char *redirerr = (char*)stderr;
1067 * Command line option parsing commences now We only need to support
1068 * a few things in the test suite.
1074 if (argv[0][0] == '-') {
1075 if (parsecmd("redirout", &argc, &argv, &redirout, 1, false))
1077 if (parsecmd("redirerr", &argc, &argv, &redirerr, 1, false))
1080 con_change(redirout, redirerr);
1082 if (!strcmp(argv[0]+1, "debug")) {
1086 if (!strcmp(argv[0]+1, "memchk")) {
1090 if (!strcmp(argv[0]+1, "nocolor")) {
1095 con_err("invalid argument %s\n", argv[0]+1);
1099 con_change(redirout, redirerr);
1100 test_perform("tests");