/*-
 * See the file LICENSE for redistribution information.
 *
 * Copyright (c) 1997-2005
 *	Sleepycat Software.  All rights reserved.
 *
 * $Id: os_dir.c,v 12.2 2005/07/06 23:52:43 dda Exp $
 */

#include "db_config.h"

#include "db_int.h"

/*
 * __os_dirlist --
 *	Return a list of the files in a directory.
 */
int
__os_dirlist(dbenv, dir, namesp, cntp)
	DB_ENV *dbenv;
	const char *dir;
	char ***namesp;
	int *cntp;
{
	HANDLE dirhandle;
	WIN32_FIND_DATA fdata;
	int arraysz, cnt, ret;
	char **names, *onename;
	_TCHAR tfilespec[MAXPATHLEN + 1];
	_TCHAR *tdir;

	if (DB_GLOBAL(j_dirlist) != NULL)
		return (DB_GLOBAL(j_dirlist)(dir, namesp, cntp));

	TO_TSTRING(dbenv, dir, tdir, ret);
	if (ret != 0)
		return (ret);

	(void)_sntprintf(tfilespec, MAXPATHLEN,
	    _T("%s%hc*"), tdir, PATH_SEPARATOR[0]);
	if ((dirhandle = FindFirstFile(tfilespec, &fdata))
	    == INVALID_HANDLE_VALUE)
		return (__os_get_errno());

	names = NULL;
	arraysz = cnt = ret = 0;
	for (;;) {
		if (cnt >= arraysz) {
			arraysz += 100;
			if ((ret = __os_realloc(dbenv,
			    arraysz * sizeof(names[0]), &names)) != 0)
				goto err;
		}
		/*
		 * FROM_TSTRING doesn't necessarily allocate new memory, so we
		 * must do that explicitly.  Unfortunately, when compiled with
		 * UNICODE, we'll copy twice.
		 */
		FROM_TSTRING(dbenv, fdata.cFileName, onename, ret);
		if (ret != 0)
			goto err;
		ret = __os_strdup(dbenv, onename, &names[cnt]);
		FREE_STRING(dbenv, onename);
		if (ret != 0)
			goto err;
		cnt++;
		if (!FindNextFile(dirhandle, &fdata)) {
			if (GetLastError() == ERROR_NO_MORE_FILES)
				break;
			else {
				ret = __os_get_errno();
				goto err;
			}
		}
	}

err:	if (!FindClose(dirhandle) && ret == 0)
		ret = __os_get_errno();

	if (ret == 0) {
		*namesp = names;
		*cntp = cnt;
	} else if (names != NULL)
		__os_dirfree(dbenv, names, cnt);

	FREE_STRING(dbenv, tdir);

	return (ret);
}

/*
 * __os_dirfree --
 *	Free the list of files.
 */
void
__os_dirfree(dbenv, names, cnt)
	DB_ENV *dbenv;
	char **names;
	int cnt;
{
	if (DB_GLOBAL(j_dirfree) != NULL) {
		DB_GLOBAL(j_dirfree)(names, cnt);
		return;
	}

	while (cnt > 0)
		__os_free(dbenv, names[--cnt]);
	__os_free(dbenv, names);
}
