diff options
Diffstat (limited to 'tools')
-rw-r--r-- | tools/tests/test.c | 266 |
1 files changed, 266 insertions, 0 deletions
diff --git a/tools/tests/test.c b/tools/tests/test.c new file mode 100644 index 0000000..2aaf2c7 --- /dev/null +++ b/tools/tests/test.c @@ -0,0 +1,266 @@ +/* + * Copyright (c) 2013, Red Hat Inc. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * * Redistributions of source code must retain the above + * copyright notice, this list of conditions and the + * following disclaimer. + * * Redistributions in binary form must reproduce the + * above copyright notice, this list of conditions and + * the following disclaimer in the documentation and/or + * other materials provided with the distribution. + * * The names of contributors to this software may not be + * used to endorse or promote products derived from this + * software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS + * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE + * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, + * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, + * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS + * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED + * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, + * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF + * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH + * DAMAGE. + * + * Author: Stef Walter <stefw@collabora.co.uk> + */ + +#include "config.h" +#include "CuTest.h" + +#include "debug.h" +#include "message.h" +#include "path.h" +#include "test.h" + +#include <sys/stat.h> + +#include <assert.h> +#include <dirent.h> +#include <errno.h> +#include <fcntl.h> +#include <stdio.h> +#include <string.h> +#include <unistd.h> + +#ifdef OS_UNIX +#include <paths.h> +#endif + +static char * +read_file (CuTest *tc, + const char *file, + int line, + const char *filename, + long *len) +{ + struct stat sb; + FILE *f = NULL; + char *data; + + f = fopen (filename, "rb"); + if (f == NULL) + CuFail_Line (tc, file, line, "Couldn't open file", filename); + + /* Figure out size */ + if (stat (filename, &sb) < 0) + CuFail_Line (tc, file, line, "Couldn't stat file", filename); + + *len = sb.st_size; + data = malloc (*len ? *len : 1); + assert (data != NULL); + + /* And read in one block */ + if (fread (data, 1, *len, f) != *len) + CuFail_Line (tc, file, line, "Couldn't read file", filename); + + fclose (f); + + return data; +} + +void +test_check_file_msg (CuTest *tc, + const char *file, + int line, + const char *directory, + const char *name, + const char *reference) +{ + char *refdata; + long reflen; + + refdata = read_file (tc, file, line, reference, &reflen); + test_check_data_msg (tc, file, line, directory, name, refdata, reflen); + free (refdata); +} + +void +test_check_data_msg (CuTest *tc, + const char *file, + int line, + const char *directory, + const char *name, + const void *refdata, + long reflen) +{ + char *filedata; + char *filename; + long filelen; + + if (asprintf (&filename, "%s/%s", directory, name) < 0) + CuFail_Line (tc, file, line, "asprintf() failed", NULL); + + filedata = read_file (tc, file, line, filename, &filelen); + + if (filelen != reflen || memcmp (filedata, refdata, reflen) != 0) + CuFail_Line (tc, file, line, "File contents not as expected", filename); + + CuAssert_Line (tc, file, line, "couldn't remove file", unlink (filename) >= 0); + free (filename); + free (filedata); +} + +#ifdef OS_UNIX + +void +test_check_symlink_msg (CuTest *tc, + const char *file, + int line, + const char *directory, + const char *name, + const char *destination) +{ + char buf[1024] = { 0, }; + char *filename; + + if (asprintf (&filename, "%s/%s", directory, name) < 0) + CuFail_Line (tc, file, line, "asprintf() failed", NULL); + + if (readlink (filename, buf, sizeof (buf)) < 0) + CuFail_Line (tc, file, line, "Couldn't read symlink", filename); + + CuAssertStrEquals_LineMsg (tc, file, line, "symlink contents wrong", destination, buf); + + CuAssert_Line (tc, file, line, "couldn't remove symlink", unlink (filename) >= 0); + free (filename); +} + +#endif /* OS_UNIX */ + +p11_dict * +test_check_directory_files (const char *file, + ...) +{ + p11_dict *files; + va_list va; + + files = p11_dict_new (p11_dict_str_hash, p11_dict_str_equal, NULL, NULL); + + va_start (va, file); + + while (file != NULL) { + if (!p11_dict_set (files, (void *)file, (void *)file)) + return_val_if_reached (NULL); + file = va_arg (va, const char *); + } + + va_end (va); + + return files; +} + +void +test_check_directory_msg (CuTest *tc, + const char *file, + int line, + const char *directory, + p11_dict *files) +{ + p11_dictiter iter; + struct dirent *dp; + const char *name; + DIR *dir; + + dir = opendir (directory); + if (dir == NULL) + CuFail_Line (tc, file ,line, "Couldn't open directory", directory); + + while ((dp = readdir (dir)) != NULL) { + if (strcmp (dp->d_name, ".") == 0 || + strcmp (dp->d_name, "..") == 0) + continue; + + if (!p11_dict_remove (files, dp->d_name)) + CuFail_Line (tc, file, line, "Unexpected file in directory", dp->d_name); + } + + closedir (dir); + +#ifdef OS_UNIX + CuAssert_Line (tc, file, line, "couldn't chown directory", chmod (directory, S_IRWXU) >= 0); +#endif + + p11_dict_iterate (files, &iter); + while (p11_dict_next (&iter, (void **)&name, NULL)) + CuFail_Line (tc, file, line, "Couldn't find file in directory", name); + + p11_dict_free (files); +} + +static char * +expand_tempdir (const char *name) +{ + const char *env; + + env = getenv ("TMPDIR"); + if (env && env[0]) { + return p11_path_build (env, name, NULL); + + } else { +#ifdef OS_UNIX +#ifdef _PATH_TMP + return p11_path_build (_PATH_TMP, name, NULL); +#else + return p11_path_build ("/tmp", name, NULL); +#endif + +#else /* OS_WIN32 */ + char directory[MAX_PATH + 1]; + + if (!GetTempPathA (MAX_PATH + 1, directory)) { + p11_message ("couldn't lookup temp directory"); + errno = ENOTDIR; + return NULL; + } + + return p11_path_build (directory, name, NULL); + +#endif /* OS_WIN32 */ + } +} + +char * +test_temp_directory (const char *templ) +{ + char *directory; + + directory = expand_tempdir (templ); + if (directory == NULL) + return NULL; + + if (!mkdtemp (directory)) { + p11_message_err (errno, "couldn't create temp directory: %s", directory); + free (directory); + assert (0 && "not reached"); + } + + return directory; +} |