#include <config.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <ctype.h>
#include <errno.h>
#include <unistd.h>
#include <isc/dir.h>
#include <isc/magic.h>
#include <isc/string.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)
void
isc_dir_init(isc_dir_t *dir) {
REQUIRE(dir != NULL);
dir->entry.name[0] = '\0';
dir->entry.length = 0;
dir->handle = NULL;
dir->magic = ISC_DIR_MAGIC;
}
isc_result_t
isc_dir_open(isc_dir_t *dir, const char *dirname) {
char *p;
isc_result_t result = ISC_R_SUCCESS;
REQUIRE(VALID_DIR(dir));
REQUIRE(dirname != NULL);
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++ = '/';
*p++ = '*';
*p++ = '\0';
dir->handle = opendir(dirname);
if (dir->handle == NULL)
return isc__errno2result(errno);
return (result);
}
isc_result_t
isc_dir_read(isc_dir_t *dir) {
struct dirent *entry;
REQUIRE(VALID_DIR(dir) && dir->handle != NULL);
entry = readdir(dir->handle);
if (entry == NULL)
return (ISC_R_NOMORE);
if (sizeof(dir->entry.name) <= strlen(entry->d_name))
return (ISC_R_UNEXPECTED);
strcpy(dir->entry.name, entry->d_name);
dir->entry.length = strlen(entry->d_name);
return (ISC_R_SUCCESS);
}
void
isc_dir_close(isc_dir_t *dir) {
REQUIRE(VALID_DIR(dir) && dir->handle != NULL);
(void)closedir(dir->handle);
dir->handle = NULL;
}
isc_result_t
isc_dir_reset(isc_dir_t *dir) {
REQUIRE(VALID_DIR(dir) && dir->handle != NULL);
rewinddir(dir->handle);
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) {
REQUIRE(dirname != NULL);
#ifdef HAVE_CHROOT
if (chroot(dirname) < 0 || chdir("/") < 0)
return (isc__errno2result(errno));
return (ISC_R_SUCCESS);
#else
return (ISC_R_NOTIMPLEMENTED);
#endif
}
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, 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);
}