1 files changed, 175 insertions, 1 deletions

diff --git a/trust/tests/test-trust.c b/trust/tests/test-trust.c
index 6a22946..33ba19e 100644
--- a/trust/tests/test-trust.c
+++ b/trust/tests/test-trust.c
@@ -33,15 +33,24 @@
  */
 
 #include "config.h"
-#include "test.h"
 
 #include "attrs.h"
+#include "debug.h"
+#include "test.h"
+
 #include "test-trust.h"
 
+#include <sys/stat.h>
+
 #include <assert.h>
+#include <dirent.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <stdarg.h>
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
+#include <unistd.h>
 
 void
 test_check_object_msg (const char *file,
@@ -150,3 +159,168 @@ test_check_attr_msg (const char *file,
 		               attr ? p11_attr_to_string (attr, klass) : "(null)");
 	}
 }
+
+static char *
+read_file (const char *file,
+           int line,
+           const char *function,
+           const char *filename,
+           long *len)
+{
+	struct stat sb;
+	FILE *f = NULL;
+	char *data;
+
+	f = fopen (filename, "rb");
+	if (f == NULL)
+		p11_test_fail (file, line, function, "Couldn't open file: %s", filename);
+
+	/* Figure out size */
+	if (stat (filename, &sb) < 0)
+		p11_test_fail (file, line, function, "Couldn't stat file: %s", 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)
+		p11_test_fail (file, line, function, "Couldn't read file: %s", filename);
+
+	fclose (f);
+
+	return data;
+}
+
+void
+test_check_file_msg (const char *file,
+                     int line,
+                     const char *function,
+                     const char *directory,
+                     const char *name,
+                     const char *reference)
+{
+	char *refdata;
+	long reflen;
+
+	refdata = read_file (file, line, function, reference, &reflen);
+	test_check_data_msg (file, line, function, directory, name, refdata, reflen);
+	free (refdata);
+}
+
+void
+test_check_data_msg (const char *file,
+                     int line,
+                     const char *function,
+                     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)
+		assert_not_reached ();
+
+	filedata = read_file (file, line, function, filename, &filelen);
+
+	if (filelen != reflen || memcmp (filedata, refdata, reflen) != 0)
+		p11_test_fail (file, line, function, "File contents not as expected: %s", filename);
+
+	if (unlink (filename) < 0)
+		p11_test_fail (file, line, function, "Couldn't remove file: %s", filename);
+	free (filename);
+	free (filedata);
+}
+
+#ifdef OS_UNIX
+
+void
+test_check_symlink_msg (const char *file,
+                        int line,
+                        const char *function,
+                        const char *directory,
+                        const char *name,
+                        const char *destination)
+{
+	char buf[1024] = { 0, };
+	char *filename;
+
+	if (asprintf (&filename, "%s/%s", directory, name) < 0)
+		assert_not_reached ();
+
+	if (readlink (filename, buf, sizeof (buf)) < 0)
+		p11_test_fail (file, line, function, "Couldn't read symlink: %s", filename);
+
+	if (strcmp (destination, buf) != 0)
+		p11_test_fail (file, line, function, "Symlink contents wrong: %s != %s", destination, buf);
+
+	if (unlink (filename) < 0)
+		p11_test_fail (file, line, function, "Couldn't remove symlink: %s", filename);
+	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 (const char *file,
+                          int line,
+                          const char *function,
+                          const char *directory,
+                          p11_dict *files)
+{
+	p11_dictiter iter;
+	struct dirent *dp;
+	const char *name;
+	DIR *dir;
+
+	dir = opendir (directory);
+	if (dir == NULL)
+		p11_test_fail (file ,line, function, "Couldn't open directory: %s", 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))
+			p11_test_fail  (file, line, function, "Unexpected file in directory: %s", dp->d_name);
+	}
+
+	closedir (dir);
+
+#ifdef OS_UNIX
+	if (chmod (directory, S_IRWXU) < 0)
+		p11_test_fail (file, line, function, "couldn't chown directory: %s: %s", directory, strerror (errno));
+#endif
+
+	p11_dict_iterate (files, &iter);
+	while (p11_dict_next (&iter, (void **)&name, NULL))
+		p11_test_fail (file, line, function, "Couldn't find file in directory: %s", name);
+
+	p11_dict_free (files);
+}