#include <config.h>
#include <string.h>
#include <direct.h>
#include <process.h>
#include <io.h>
#include <sys/stat.h>
#include <isc/dir.h>
#include <isc/magic.h>
#include <isc/assertions.h>
#include <isc/util.h>
#include "errno2result.h"
#define ISC_DIR_MAGIC ISC_MAGIC('D', 'I', 'R', '*')
#define VALID_DIR(dir) ISC_MAGIC_VALID(dir, ISC_DIR_MAGIC)
static isc_result_t
start_directory(isc_dir_t *p);
void
isc_dir_init(isc_dir_t *dir) {
REQUIRE(dir != NULL);
dir->dirname[0] = '\0';
dir->entry.name[0] = '\0';
dir->entry.length = 0;
memset(&(dir->entry.find_data), 0, sizeof(dir->entry.find_data));
dir->entry_filled = ISC_FALSE;
dir->search_handle = INVALID_HANDLE_VALUE;
dir->magic = ISC_DIR_MAGIC;
}
isc_result_t
isc_dir_open(isc_dir_t *dir, const char *dirname) {
char *p;
isc_result_t result;
REQUIRE(dirname != NULL);
REQUIRE(VALID_DIR(dir) && dir->search_handle == INVALID_HANDLE_VALUE);
if (strlen(dirname) + 3 > sizeof(dir->dirname))
return (ISC_R_NOSPACE);
strcpy(dir->dirname, dirname);
p = dir->dirname + strlen(dir->dirname);
if (dir->dirname < p && *(p - 1) != '\\' && *(p - 1) != ':')
*p++ = '\\';
*p++ = '*';
*p++ = '\0';
result = start_directory(dir);
return (result);
}
isc_result_t
isc_dir_read(isc_dir_t *dir) {
REQUIRE(VALID_DIR(dir) && dir->search_handle != INVALID_HANDLE_VALUE);
if (dir->entry_filled)
dir->entry_filled = ISC_FALSE;
else {
if (FindNextFile(dir->search_handle,
&dir->entry.find_data) == FALSE)
if (GetLastError() == ERROR_NO_MORE_FILES)
return (ISC_R_NOMORE);
else
return (ISC_R_UNEXPECTED);
}
strcpy(dir->entry.name, dir->entry.find_data.cFileName);
dir->entry.length = strlen(dir->entry.name);
return (ISC_R_SUCCESS);
}
void
isc_dir_close(isc_dir_t *dir) {
REQUIRE(VALID_DIR(dir) && dir->search_handle != INVALID_HANDLE_VALUE);
FindClose(dir->search_handle);
dir->search_handle = INVALID_HANDLE_VALUE;
}
isc_result_t
isc_dir_reset(isc_dir_t *dir) {
isc_result_t result;
REQUIRE(VALID_DIR(dir) && dir->search_handle != INVALID_HANDLE_VALUE);
REQUIRE(dir->dirname != NULL);
isc_dir_close(dir);
result = start_directory(dir);
return (result);
}
static isc_result_t
start_directory(isc_dir_t *dir)
{
REQUIRE(VALID_DIR(dir));
REQUIRE(dir->search_handle == INVALID_HANDLE_VALUE);
dir->entry_filled = ISC_FALSE;
dir->search_handle = FindFirstFile(dir->dirname,
&dir->entry.find_data);
if (dir->search_handle == INVALID_HANDLE_VALUE) {
return (ISC_R_FAILURE);
}
INSIST(sizeof(dir->entry.name) >
strlen(dir->entry.find_data.cFileName));
strcpy(dir->entry.name, dir->entry.find_data.cFileName);
dir->entry.length = strlen(dir->entry.name);
dir->entry_filled = ISC_TRUE;
return (ISC_R_SUCCESS);
}
isc_result_t
isc_dir_chdir(const char *dirname) {
REQUIRE(dirname != NULL);
if (chdir(dirname) < 0)
return (isc__errno2result(errno));
return (ISC_R_SUCCESS);
}
isc_result_t
isc_dir_chroot(const char *dirname) {
return (ISC_R_NOTIMPLEMENTED);
}
isc_result_t
isc_dir_createunique(char *templet) {
isc_result_t result;
char *x;
char *p;
int i;
int pid;
REQUIRE(templet != NULL);
pid = getpid();
for (x = templet + strlen(templet) - 1; *x == 'X' && x >= templet;
x--, pid /= 10)
*x = pid % 10 + '0';
x++;
do {
i = mkdir(templet);
i = chmod(templet, 0700);
if (i == 0 || errno != EEXIST)
break;
p = x;
while (*p != '\0') {
if (isdigit(*p & 0xff))
*p = 'a';
else if (*p != 'z')
++*p;
else {
*p++ = 'a';
continue;
}
break;
}
if (*p == '\0') {
errno = EEXIST;
break;
}
} while (1);
if (i == -1)
result = isc__errno2result(errno);
else
result = ISC_R_SUCCESS;
return (result);
}