Dir.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#ifdef __WINNT__
#include <windows.h>        // dcl/Dir.h
#else
#include <errno.h>
#if defined(_AIX) || defined(__sun__)
#include <alloca.h>
#endif
#include <stdlib.h>         // alloca
#include <string.h>         // memset
#include <dcl/_string.h>
#include <dcl/_stat.h>
#endif
 
#include <dcl/Charset.h>
#include <dcl/Dir.h>
 
#include "__strumbs.h"
 
#if __DCL_HAVE_ALLOC_DEBUG
#undef __DCL_ALLOC_LEVEL
#define __DCL_ALLOC_LEVEL   __DCL_ALLOC_INTERNAL
#endif
 
#if __DCL_HAVE_THIS_FILE__
#undef __THIS_FILE__
static const char_t __THIS_FILE__[] = __T("dcl/Dir.cpp");
#endif
 
__DCL_BEGIN_NAMESPACE
 
#ifndef INVALID_HANDLE_VALUE
#define INVALID_HANDLE_VALUE        ((DIR*)-1)
#endif
 
IMPLEMENT_CLASSINFO(Dir, Object)
 
Dir::Entry::Entry()
{
    memset(&__dirent, 0, sizeof(__dirent));
}
 
String Dir::Entry::name() const
__DCL_THROWS1(CharsetConvertException*)
{
#ifdef __WINNT__
    return String(__dirent.cFileName);
#elif defined(__sun__)
    return __name;
#else
    return __mbstostr(__dirent.d_name);
#endif
}
 
#ifdef __WINNT__
bool Dir::Entry::isDir() const
{
    return (__dirent.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0;
}
 
bool Dir::Entry::isLink() const
{
    return (__dirent.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) != 0;
}
#else
bool Dir::Entry::isDir() const
{
#if defined(_AIX) || defined(__sun__)
    String path = __path + name();
    struct stat st;
    if (__lstat(path, &st) != 0) {
        throw new IOException(path, errno);
    }
    return S_ISDIR(st.st_mode);
#else
    return __dirent.d_type == DT_DIR;
#endif
}
 
bool Dir::Entry::isLink() const
{
#if defined(_AIX) || defined(__sun__)
    String path = __path + name();
    struct stat st;
    if (__lstat(path, &st) != 0) {
        throw new IOException(path, errno);
    }
    return S_ISLNK(st.st_mode);
#else
    return __dirent.d_type == DT_LNK;
#endif
}
#endif
 
String Dir::Entry::toString() const
__DCL_THROWS1(CharsetConvertException*)
{
    StringBuilder r = name();
    if (isDir() || isLink()) {
        r += __T(" [");
        if (isDir())
            r += __T("directory");
        if (isLink()) {
            if (isDir())
                r += __T(", ");
            r += __T("symbolic link");
        }
        r += __T(']');
    }
    return r;
}
 
Dir::~Dir()
{
    if (__handle != INVALID_HANDLE_VALUE) {
        try {
            close();
        }
        catch (Exception* e) {
            __DCL_TRACE1(__T("Dir::close Error! %ls\n"), e->toString().data());
            e->destroy();
        }
    }
}
 
Dir::Dir(const String& _path) __DCL_THROWS1(IOException*)
{
    __DCL_ASSERT(!_path.isEmpty());
 
    // 모든 OS에서 경로 구분자는 '/'를 사용한다.
    String path = _path.replace(__T('\\'), __T('/'));
    if (!path.endsWith(__T('/')))
        path = path + __T("/");
 
#ifdef __WINNT__
    WIN32_FIND_DATAW findData;
    HANDLE h = FindFirstFileW(path + __T('*'), &findData);
    if (h == INVALID_HANDLE_VALUE)
        throw new IOException(_path, GetLastError());
    FindClose(h);
    if (!(findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
        throw new IOException(_path, ERROR_DIRECTORY);
#else
    struct stat st;
    if (__stat(path, &st) != 0) {
        throw new IOException(path, errno);
    }
 
    if (!S_ISDIR(st.st_mode)) {
        throw new IOException(path, ENOTDIR);
    }
#endif
 
    __handle = INVALID_HANDLE_VALUE;
    __path = path;
    rewind();
}
 
void Dir::close() __DCL_THROWS1(IOException*)
{
    __DCL_ASSERT(__handle != INVALID_HANDLE_VALUE);
    HandleType handle = __handle;
    __handle = INVALID_HANDLE_VALUE;
 
#ifdef __WINNT__
    if (!FindClose(handle))
        throw new IOException(__path, GetLastError());
#else
    if (closedir(handle))
        throw new IOException(__path, errno);
#endif
}
 
void Dir::rewind() __DCL_THROWS1(IOException*)
{
#ifdef __WINNT__
    if (__handle != INVALID_HANDLE_VALUE)
        close();
#else
    if (__handle != INVALID_HANDLE_VALUE) {
        rewinddir(__handle);
        return;
    }
 
    STRTOMBS(__path, path)
    else {
        throw new IOException(__path, EILSEQ);
    }
 
    DIR* handle = opendir(path);
    if (handle == NULL) {
        throw new IOException(__path, errno);
    }
    __handle = handle;
#endif
}
 
bool Dir::read(Dir::Entry& _entry) __DCL_THROWS1(IOException*)
{
#ifdef __WINNT__
    if (__handle == INVALID_HANDLE_VALUE) {
        HANDLE h = FindFirstFileW(__path + __T('*'), &_entry.__dirent);
        if (h == INVALID_HANDLE_VALUE)
            throw new IOException(__path, GetLastError());
        __handle = h;
    }
    else {
        if (!FindNextFileW(__handle, &_entry.__dirent)) {
            if (GetLastError() == ERROR_NO_MORE_FILES)
                return false;
            else
                throw new IOException(__path, GetLastError());
        }
    }
    return true;
#else
    __DCL_ASSERT(__handle != INVALID_HANDLE_VALUE);
    errno = 0;
    struct dirent* ent = readdir(__handle);
    if (ent != NULL) {
        memcpy((void*)&_entry.__dirent, ent, sizeof(_entry.__dirent));
#if defined(_AIX)
        _entry.__path = __path;
#elif defined(__sun__)
        _entry.__path = __path;
        _entry.__name = __mbstostr(ent->d_name);
        _entry.__dirent.d_name[0] = '\0';
#endif
    }
    else if (errno != 0) {
        throw new IOException(__path, errno);
    }
    return ent != NULL;     // EOF
#endif
}
 
__DCL_END_NAMESPACE