/*
 * This file is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License, version 2,
 * as published by the Free Software Foundation.
 *
 * In addition to the permissions in the GNU General Public License,
 * the authors give you unlimited permission to link the compiled
 * version of this file into combinations with other programs,
 * and to distribute those combinations without any restriction
 * coming from the use of this file.  (The General Public License
 * restrictions do apply in other respects; for example, they cover
 * modification of the file, and distribution when not linked into
 * a combined executable.)
 *
 * This file is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; see the file COPYING.  If not, write to
 * the Free Software Foundation, 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */
#include "test_lib.h"

#include "vector.h"
#include "fileops.h"

BEGIN_TEST(string0, "compare prefixes")
	must_be_true(git__prefixcmp("", "") == 0);
	must_be_true(git__prefixcmp("a", "") == 0);
	must_be_true(git__prefixcmp("", "a") < 0);
	must_be_true(git__prefixcmp("a", "b") < 0);
	must_be_true(git__prefixcmp("b", "a") > 0);
	must_be_true(git__prefixcmp("ab", "a") == 0);
	must_be_true(git__prefixcmp("ab", "ac") < 0);
	must_be_true(git__prefixcmp("ab", "aa") > 0);
END_TEST

BEGIN_TEST(string1, "compare suffixes")
	must_be_true(git__suffixcmp("", "") == 0);
	must_be_true(git__suffixcmp("a", "") == 0);
	must_be_true(git__suffixcmp("", "a") < 0);
	must_be_true(git__suffixcmp("a", "b") < 0);
	must_be_true(git__suffixcmp("b", "a") > 0);
	must_be_true(git__suffixcmp("ba", "a") == 0);
	must_be_true(git__suffixcmp("zaa", "ac") < 0);
	must_be_true(git__suffixcmp("zaz", "ac") > 0);
END_TEST


BEGIN_TEST(vector0, "initial size of 1 would cause writing past array bounds")
  git_vector x;
  int i;
  git_vector_init(&x, 1, NULL);
  for (i = 0; i < 10; ++i) {
    git_vector_insert(&x, (void*) 0xabc);
  }
  git_vector_free(&x);
END_TEST

BEGIN_TEST(vector1, "don't read past array bounds on remove()")
  git_vector x;
  // make initial capacity exact for our insertions.
  git_vector_init(&x, 3, NULL);
  git_vector_insert(&x, (void*) 0xabc);
  git_vector_insert(&x, (void*) 0xdef);
  git_vector_insert(&x, (void*) 0x123);

  git_vector_remove(&x, 0);  // used to read past array bounds.
  git_vector_free(&x);
END_TEST


BEGIN_TEST(path0, "get the dirname of a path")
	char dir[64], *dir2;

#define DIRNAME_TEST(A, B) { \
	must_be_true(git__dirname_r(dir, sizeof(dir), A) >= 0); \
	must_be_true(strcmp(dir, B) == 0);				\
	must_be_true((dir2 = git__dirname(A)) != NULL);	\
	must_be_true(strcmp(dir2, B) == 0);				\
	free(dir2);										\
}

	DIRNAME_TEST(NULL, ".");
	DIRNAME_TEST("", ".");
	DIRNAME_TEST("a", ".");
	DIRNAME_TEST("/", "/");
	DIRNAME_TEST("/usr", "/");
	DIRNAME_TEST("/usr/", "/");
	DIRNAME_TEST("/usr/lib", "/usr");
	DIRNAME_TEST("/usr/lib/", "/usr");
	DIRNAME_TEST("/usr/lib//", "/usr");
	DIRNAME_TEST("usr/lib", "usr");
	DIRNAME_TEST("usr/lib/", "usr");
	DIRNAME_TEST("usr/lib//", "usr");
	DIRNAME_TEST(".git/", ".");

#undef DIRNAME_TEST

END_TEST

BEGIN_TEST(path1, "get the base name of a path")
	char base[64], *base2;

#define BASENAME_TEST(A, B) { \
	must_be_true(git__basename_r(base, sizeof(base), A) >= 0); \
	must_be_true(strcmp(base, B) == 0);					\
	must_be_true((base2 = git__basename(A)) != NULL);	\
	must_be_true(strcmp(base2, B) == 0);				\
	free(base2);										\
}

	BASENAME_TEST(NULL, ".");
	BASENAME_TEST("", ".");
	BASENAME_TEST("a", "a");
	BASENAME_TEST("/", "/");
	BASENAME_TEST("/usr", "usr");
	BASENAME_TEST("/usr/", "usr");
	BASENAME_TEST("/usr/lib", "lib");
	BASENAME_TEST("/usr/lib//", "lib");
	BASENAME_TEST("usr/lib", "lib");

#undef BASENAME_TEST

END_TEST

BEGIN_TEST(path2, "get the latest component in a path")
	const char *dir;

#define TOPDIR_TEST(A, B) { \
	must_be_true((dir = git__topdir(A)) != NULL);	\
	must_be_true(strcmp(dir, B) == 0);				\
}

	TOPDIR_TEST(".git/", ".git/");
	TOPDIR_TEST("/.git/", ".git/");
	TOPDIR_TEST("usr/local/.git/", ".git/");
	TOPDIR_TEST("./.git/", ".git/");
	TOPDIR_TEST("/usr/.git/", ".git/");
	TOPDIR_TEST("/", "/");
	TOPDIR_TEST("a/", "a/");

	must_be_true(git__topdir("/usr/.git") == NULL);
	must_be_true(git__topdir(".") == NULL);
	must_be_true(git__topdir("") == NULL);
	must_be_true(git__topdir("a") == NULL);

#undef TOPDIR_TEST
END_TEST

typedef int (normalize_path)(char *, size_t, const char *);

/* Assert flags */
#define CWD_AS_PREFIX 1
#define PATH_AS_SUFFIX 2
#define ROOTED_PATH 4

static int ensure_normalized(const char *input_path, const char *expected_path, normalize_path normalizer, int assert_flags)
{
	int error = GIT_SUCCESS;
	char buffer_out[GIT_PATH_MAX];
	char current_workdir[GIT_PATH_MAX];

	error = gitfo_getcwd(current_workdir, sizeof(current_workdir));
	if (error < GIT_SUCCESS)
		return error;

	error = normalizer(buffer_out, sizeof(buffer_out), input_path);
	if (error < GIT_SUCCESS)
		return error;

	if (expected_path == NULL)
		return error;

	if ((assert_flags & PATH_AS_SUFFIX) != 0)
		if (git__suffixcmp(buffer_out, expected_path))
			return GIT_ERROR;

	if ((assert_flags & CWD_AS_PREFIX) != 0)
		if (git__prefixcmp(buffer_out, current_workdir))
			return GIT_ERROR;

	if ((assert_flags & ROOTED_PATH) != 0) {
		error = strcmp(expected_path, buffer_out);
	}

	return error;
}

static int ensure_dir_path_normalized(const char *input_path, const char *expected_path, int assert_flags)
{
	return ensure_normalized(input_path, expected_path, gitfo_prettify_dir_path, assert_flags);
}

static int ensure_file_path_normalized(const char *input_path, const char *expected_path, int assert_flags)
{
	return ensure_normalized(input_path, expected_path, gitfo_prettify_file_path, assert_flags);
}

BEGIN_TEST(path3, "prettify and validate a path to a file")
	must_pass(ensure_file_path_normalized("a", "a", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_file_path_normalized("./testrepo.git", "testrepo.git", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_file_path_normalized("./.git", ".git", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_file_path_normalized("./git.", "git.", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_fail(ensure_file_path_normalized("git./", NULL, 0));
	must_fail(ensure_file_path_normalized("", NULL, 0));
	must_fail(ensure_file_path_normalized(".", NULL, 0));
	must_fail(ensure_file_path_normalized("./", NULL, 0));
	must_fail(ensure_file_path_normalized("./.", NULL, 0));
	must_fail(ensure_file_path_normalized("./..", NULL, 0));
	must_fail(ensure_file_path_normalized("../.", NULL, 0));
	must_fail(ensure_file_path_normalized("./.././/", NULL, 0));
	must_fail(ensure_file_path_normalized("dir/..", NULL, 0));
	must_fail(ensure_file_path_normalized("dir/sub/../..", NULL, 0));
	must_fail(ensure_file_path_normalized("dir/sub/..///..", NULL, 0));
	must_fail(ensure_file_path_normalized("dir/sub///../..", NULL, 0));
	must_fail(ensure_file_path_normalized("dir/sub///..///..", NULL, 0));
	must_fail(ensure_file_path_normalized("dir/sub/../../..", NULL, 0));
	must_pass(ensure_file_path_normalized("dir", "dir", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_fail(ensure_file_path_normalized("dir//", NULL, 0));
	must_pass(ensure_file_path_normalized("./dir", "dir", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_fail(ensure_file_path_normalized("dir/.", NULL, 0));
	must_fail(ensure_file_path_normalized("dir///./", NULL, 0));
	must_fail(ensure_file_path_normalized("dir/sub/..", NULL, 0));
	must_fail(ensure_file_path_normalized("dir//sub/..",NULL, 0));
	must_fail(ensure_file_path_normalized("dir//sub/../", NULL, 0));
	must_fail(ensure_file_path_normalized("dir/sub/../", NULL, 0));
	must_fail(ensure_file_path_normalized("dir/sub/../.", NULL, 0));
	must_fail(ensure_file_path_normalized("dir/s1/../s2/", NULL, 0));
	must_fail(ensure_file_path_normalized("d1/s1///s2/..//../s3/", NULL, 0));
	must_pass(ensure_file_path_normalized("d1/s1//../s2/../../d2", "d2", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_fail(ensure_file_path_normalized("dir/sub/../", NULL, 0));
	must_pass(ensure_file_path_normalized("../a/../b/c/d/../../e", "b/e", PATH_AS_SUFFIX));
	must_fail(ensure_file_path_normalized("....", NULL, 0));
	must_fail(ensure_file_path_normalized("...", NULL, 0));
	must_fail(ensure_file_path_normalized("./...", NULL, 0));
	must_fail(ensure_file_path_normalized("d1/...", NULL, 0));
	must_fail(ensure_file_path_normalized("d1/.../", NULL, 0));
	must_fail(ensure_file_path_normalized("d1/.../d2", NULL, 0));
	
	must_pass(ensure_file_path_normalized("/a", "/a", ROOTED_PATH));
	must_pass(ensure_file_path_normalized("/./testrepo.git", "/testrepo.git", ROOTED_PATH));
	must_pass(ensure_file_path_normalized("/./.git", "/.git", ROOTED_PATH));
	must_pass(ensure_file_path_normalized("/./git.", "/git.", ROOTED_PATH));
	must_fail(ensure_file_path_normalized("/git./", NULL, 0));
	must_fail(ensure_file_path_normalized("/", NULL, 0));
	must_fail(ensure_file_path_normalized("/.", NULL, 0));
	must_fail(ensure_file_path_normalized("/./", NULL, 0));
	must_fail(ensure_file_path_normalized("/./.", NULL, 0));
	must_fail(ensure_file_path_normalized("/./..", NULL, 0));
	must_fail(ensure_file_path_normalized("/../.", NULL, 0));
	must_fail(ensure_file_path_normalized("/./.././/", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir/..", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir/sub/../..", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir/sub/..///..", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir/sub///../..", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir/sub///..///..", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir/sub/../../..", NULL, 0));
	must_pass(ensure_file_path_normalized("/dir", "/dir", 0));
	must_fail(ensure_file_path_normalized("/dir//", NULL, 0));
	must_pass(ensure_file_path_normalized("/./dir", "/dir", 0));
	must_fail(ensure_file_path_normalized("/dir/.", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir///./", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir/sub/..", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir//sub/..",NULL, 0));
	must_fail(ensure_file_path_normalized("/dir//sub/../", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir/sub/../", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir/sub/../.", NULL, 0));
	must_fail(ensure_file_path_normalized("/dir/s1/../s2/", NULL, 0));
	must_fail(ensure_file_path_normalized("/d1/s1///s2/..//../s3/", NULL, 0));
	must_pass(ensure_file_path_normalized("/d1/s1//../s2/../../d2", "/d2", 0));
	must_fail(ensure_file_path_normalized("/dir/sub/../", NULL, 0));
	must_fail(ensure_file_path_normalized("/....", NULL, 0));
	must_fail(ensure_file_path_normalized("/...", NULL, 0));
	must_fail(ensure_file_path_normalized("/./...", NULL, 0));
	must_fail(ensure_file_path_normalized("/d1/...", NULL, 0));
	must_fail(ensure_file_path_normalized("/d1/.../", NULL, 0));
	must_fail(ensure_file_path_normalized("/d1/.../d2", NULL, 0));
END_TEST

BEGIN_TEST(path4, "validate and prettify a path to a folder")
	must_pass(ensure_dir_path_normalized("./testrepo.git", "testrepo.git/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("./.git", ".git/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("./git.", "git./", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("git./", "git./", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("", "", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized(".", "", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("./", "", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("./.", "", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir/..", "", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir/sub/../..", "", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir/sub/..///..", "", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir/sub///../..", "", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir/sub///..///..", "", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir", "dir/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir//", "dir/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("./dir", "dir/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir/.", "dir/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir///./", "dir/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir/sub/..", "dir/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir//sub/..", "dir/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir//sub/../", "dir/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir/sub/../", "dir/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir/sub/../.", "dir/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir/s1/../s2/", "dir/s2/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("d1/s1///s2/..//../s3/", "d1/s3/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("d1/s1//../s2/../../d2", "d2/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("dir/sub/../", "dir/", CWD_AS_PREFIX | PATH_AS_SUFFIX));
	must_pass(ensure_dir_path_normalized("../a/../b/c/d/../../e", "b/e/", PATH_AS_SUFFIX));
	must_fail(ensure_dir_path_normalized("....", NULL, 0));
	must_fail(ensure_dir_path_normalized("...", NULL, 0));
	must_fail(ensure_dir_path_normalized("./...", NULL, 0));
	must_fail(ensure_dir_path_normalized("d1/...", NULL, 0));
	must_fail(ensure_dir_path_normalized("d1/.../", NULL, 0));
	must_fail(ensure_dir_path_normalized("d1/.../d2", NULL, 0));

	must_pass(ensure_dir_path_normalized("/./testrepo.git", "/testrepo.git/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/./.git", "/.git/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/./git.", "/git./", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/git./", "/git./", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/", "/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("//", "/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("///", "/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/.", "/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/./", "/", ROOTED_PATH));
	must_fail(ensure_dir_path_normalized("/./..", NULL, 0));
	must_fail(ensure_dir_path_normalized("/../.", NULL, 0));
	must_fail(ensure_dir_path_normalized("/./.././/", NULL, 0));
	must_pass(ensure_dir_path_normalized("/dir/..", "/", 0));
	must_pass(ensure_dir_path_normalized("/dir/sub/../..", "/", 0));
	must_fail(ensure_dir_path_normalized("/dir/sub/../../..", NULL, 0));
	must_pass(ensure_dir_path_normalized("/dir", "/dir/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/dir//", "/dir/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/./dir", "/dir/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/dir/.", "/dir/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/dir///./", "/dir/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/dir//sub/..", "/dir/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/dir/sub/../", "/dir/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("//dir/sub/../.", "/dir/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/dir/s1/../s2/", "/dir/s2/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/d1/s1///s2/..//../s3/", "/d1/s3/", ROOTED_PATH));
	must_pass(ensure_dir_path_normalized("/d1/s1//../s2/../../d2", "/d2/", ROOTED_PATH));
	must_fail(ensure_dir_path_normalized("/....", NULL, 0));
	must_fail(ensure_dir_path_normalized("/...", NULL, 0));
	must_fail(ensure_dir_path_normalized("/./...", NULL, 0));
	must_fail(ensure_dir_path_normalized("/d1/...", NULL, 0));
	must_fail(ensure_dir_path_normalized("/d1/.../", NULL, 0));
	must_fail(ensure_dir_path_normalized("/d1/.../d2", NULL, 0));
END_TEST

static int ensure_joinpath(const char *path_a, const char *path_b, const char *expected_path)
{
	char joined_path[GIT_PATH_MAX];
	git__joinpath(joined_path, path_a, path_b);
	return strcmp(joined_path, expected_path) == 0 ? GIT_SUCCESS : GIT_ERROR;
}

BEGIN_TEST(path5, "properly join path components")
	must_pass(ensure_joinpath("", "", ""));
	must_pass(ensure_joinpath("", "a", "a"));
	must_pass(ensure_joinpath("", "/a", "/a"));
	must_pass(ensure_joinpath("a", "", "a/"));
	must_pass(ensure_joinpath("a", "/", "a/"));
	must_pass(ensure_joinpath("a", "b", "a/b"));
	must_pass(ensure_joinpath("/", "a", "/a"));
	must_pass(ensure_joinpath("/", "", "/"));
	must_pass(ensure_joinpath("/a", "/b", "/a/b"));
	must_pass(ensure_joinpath("/a", "/b/", "/a/b/"));
	must_pass(ensure_joinpath("/a/", "b/", "/a/b/"));
	must_pass(ensure_joinpath("/a/", "/b/", "/a/b/"));
END_TEST

static int ensure_joinpath_n(const char *path_a, const char *path_b, const char *path_c, const char *path_d, const char *expected_path)
{
	char joined_path[GIT_PATH_MAX];
	git__joinpath_n(joined_path, 4, path_a, path_b, path_c, path_d);
	return strcmp(joined_path, expected_path) == 0 ? GIT_SUCCESS : GIT_ERROR;
}

BEGIN_TEST(path6, "properly join path components for more than one path")
	must_pass(ensure_joinpath_n("", "", "", "", ""));
	must_pass(ensure_joinpath_n("", "a", "", "", "a/"));
	must_pass(ensure_joinpath_n("a", "", "", "", "a/"));
	must_pass(ensure_joinpath_n("", "", "", "a", "a"));
	must_pass(ensure_joinpath_n("a", "b", "", "/c/d/", "a/b/c/d/"));
	must_pass(ensure_joinpath_n("a", "b", "", "/c/d", "a/b/c/d"));
END_TEST

static int count_number_of_path_segments(const char *path)
{
	int number = 0;
	char *current = (char *)path;

	while (*current)
	{
		if (*current++ == '/')
			number++;
	}

	assert (number > 0);

	return --number;
}

BEGIN_TEST(path7, "prevent a path which escapes the root directory from being prettified")
	char current_workdir[GIT_PATH_MAX];
	char prettified[GIT_PATH_MAX];
	int i = 0, number_to_escape;

	must_pass(gitfo_getcwd(current_workdir, sizeof(current_workdir)));

	number_to_escape = count_number_of_path_segments(current_workdir);

	for (i = 0; i < number_to_escape + 1; i++)
		git__joinpath(current_workdir, current_workdir, "../");

	must_fail(gitfo_prettify_dir_path(prettified, sizeof(prettified), current_workdir));
END_TEST

typedef struct name_data {
	int  count;  /* return count */
	char *name;  /* filename     */
} name_data;

typedef struct walk_data {
	char *sub;        /* sub-directory name */
	name_data *names; /* name state data    */
} walk_data;


static char path_buffer[GIT_PATH_MAX];
static char *top_dir = "dir-walk";
static walk_data *state_loc;

static int error(const char *fmt, ...)
{
	va_list ap;

	va_start(ap, fmt);
	vfprintf(stderr, fmt, ap);
	va_end(ap);
	fprintf(stderr, "\n");
	return -1;
}

static int setup(walk_data *d)
{
	name_data *n;

	if (gitfo_mkdir(top_dir, 0755) < 0)
		return error("can't mkdir(\"%s\")", top_dir);

	if (gitfo_chdir(top_dir) < 0)
		return error("can't chdir(\"%s\")", top_dir);

	if (strcmp(d->sub, ".") != 0)
		if (gitfo_mkdir(d->sub, 0755) < 0)
			return error("can't mkdir(\"%s\")", d->sub);

	strcpy(path_buffer, d->sub);
	state_loc = d;

	for (n = d->names; n->name; n++) {
		git_file fd = gitfo_creat(n->name, 0600);
		if (fd < 0)
			return GIT_ERROR;
		gitfo_close(fd);
		n->count = 0;
	}

	return 0;
}

static int knockdown(walk_data *d)
{
	name_data *n;

	for (n = d->names; n->name; n++) {
		if (gitfo_unlink(n->name) < 0)
			return error("can't unlink(\"%s\")", n->name);
	}

	if (strcmp(d->sub, ".") != 0)
		if (gitfo_rmdir(d->sub) < 0)
			return error("can't rmdir(\"%s\")", d->sub);

	if (gitfo_chdir("..") < 0)
		return error("can't chdir(\"..\")");

	if (gitfo_rmdir(top_dir) < 0)
		return error("can't rmdir(\"%s\")", top_dir);

	return 0;
}

static int check_counts(walk_data *d)
{
	int ret = 0;
	name_data *n;

	for (n = d->names; n->name; n++) {
		if (n->count != 1)
			ret = error("count (%d, %s)", n->count, n->name);
	}
	return ret;
}

static int one_entry(void *state, char *path)
{
	walk_data *d = (walk_data *) state;
	name_data *n;

	if (state != state_loc)
		return GIT_ERROR;

	if (path != path_buffer)
		return GIT_ERROR;

	for (n = d->names; n->name; n++) {
		if (!strcmp(n->name, path)) {
			n->count++;
			return 0;
		}
	}

	return GIT_ERROR;
}


static name_data dot_names[] = {
	{ 0, "./a" },
	{ 0, "./asdf" },
	{ 0, "./pack-foo.pack" },
	{ 0, NULL }
};
static walk_data dot = {
	".",
	dot_names
};

BEGIN_TEST(dirent0, "make sure that the '.' folder is not traversed")

	must_pass(setup(&dot));

	must_pass(gitfo_dirent(path_buffer,
			       sizeof(path_buffer),
			       one_entry,
			       &dot));

	must_pass(check_counts(&dot));

	must_pass(knockdown(&dot));
END_TEST

static name_data sub_names[] = {
	{ 0, "sub/a" },
	{ 0, "sub/asdf" },
	{ 0, "sub/pack-foo.pack" },
	{ 0, NULL }
};
static walk_data sub = {
	"sub",
	sub_names
};

BEGIN_TEST(dirent1, "traverse a subfolder")

	must_pass(setup(&sub));

	must_pass(gitfo_dirent(path_buffer,
			       sizeof(path_buffer),
			       one_entry,
			       &sub));

	must_pass(check_counts(&sub));

	must_pass(knockdown(&sub));
END_TEST

static walk_data sub_slash = {
	"sub/",
	sub_names
};

BEGIN_TEST(dirent2, "traverse a slash-terminated subfolder")

	must_pass(setup(&sub_slash));

	must_pass(gitfo_dirent(path_buffer,
			       sizeof(path_buffer),
			       one_entry,
			       &sub_slash));

	must_pass(check_counts(&sub_slash));

	must_pass(knockdown(&sub_slash));
END_TEST

static name_data empty_names[] = {
	{ 0, NULL }
};
static walk_data empty = {
	"empty",
	empty_names
};

static int dont_call_me(void *GIT_UNUSED(state), char *GIT_UNUSED(path))
{
	GIT_UNUSED_ARG(state)
	GIT_UNUSED_ARG(path)
	return GIT_ERROR;
}

BEGIN_TEST(dirent3, "make sure that empty folders are not traversed")

	must_pass(setup(&empty));

	must_pass(gitfo_dirent(path_buffer,
			       sizeof(path_buffer),
			       one_entry,
			       &empty));

	must_pass(check_counts(&empty));

	/* make sure callback not called */
	must_pass(gitfo_dirent(path_buffer,
			       sizeof(path_buffer),
			       dont_call_me,
			       &empty));

	must_pass(knockdown(&empty));
END_TEST

static name_data odd_names[] = {
	{ 0, "odd/.a" },
	{ 0, "odd/..c" },
	/* the following don't work on cygwin/win32 */
	/* { 0, "odd/.b." }, */
	/* { 0, "odd/..d.." },  */
	{ 0, NULL }
};
static walk_data odd = {
	"odd",
	odd_names
};

BEGIN_TEST(dirent4, "make sure that strange looking filenames ('..c') are traversed")

	must_pass(setup(&odd));

	must_pass(gitfo_dirent(path_buffer,
			       sizeof(path_buffer),
			       one_entry,
			       &odd));

	must_pass(check_counts(&odd));

	must_pass(knockdown(&odd));
END_TEST


BEGIN_SUITE(core)
	ADD_TEST(string0);
	ADD_TEST(string1);

	ADD_TEST(vector0);
	ADD_TEST(vector1);

	ADD_TEST(path0);
	ADD_TEST(path1);
	ADD_TEST(path2);
	ADD_TEST(path3);
	ADD_TEST(path4);
	ADD_TEST(path5);
	ADD_TEST(path6);
	ADD_TEST(path7);

	ADD_TEST(dirent0);
	ADD_TEST(dirent1);
	ADD_TEST(dirent2);
	ADD_TEST(dirent3);
	ADD_TEST(dirent4);
END_SUITE
