Dir.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#if __DCL_WINDOWS
    #include <windows.h>
#else
    #include <errno.h>
    #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_DEBUG
#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(this, 0,
#if __DCL_WINDOWS
        sizeof(WIN32_FIND_DATAW)
#else
        sizeof(struct dirent)
#endif
        );
}
 
String Dir::Entry::name() const
        __DCL_THROWS1(CharsetConvertException*)
{
#if __DCL_WINDOWS
    return String(this->cFileName);
#else
    return __mbstostr(this->d_name);
#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("/");
 
#if __DCL_WINDOWS
    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) == -1)
        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;
 
#if __DCL_WINDOWS
    if (!FindClose(handle))
        throw new IOException(__path, GetLastError());
#else
    if (closedir(handle))
        throw new IOException(__path, errno);
#endif
}
 
void Dir::rewind() __DCL_THROWS1(IOException*)
{
#if __DCL_WINDOWS
    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& _buf) __DCL_THROWS1(IOException*)
{
#if __DCL_WINDOWS
    if (__handle == INVALID_HANDLE_VALUE) {
        HANDLE h = FindFirstFileW(__path + __T('*'), &_buf);
        if (h == INVALID_HANDLE_VALUE)
            throw new IOException(__path, GetLastError());
        __handle = h;
    }
    else {
        if (!FindNextFileW(__handle, &_buf)) {
            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* result = readdir(__handle);
    if (result != NULL) {
        memcpy((void*) &_buf, result, sizeof(struct dirent));
    }
    else if (errno != 0 ) {
        throw new IOException(__path, errno);
    }   
    return result != NULL;        // EOF
#endif
}
 
__DCL_END_NAMESPACE