/*
 *  Copyright (C) 2013-2020 Cisco Systems, Inc. and/or its affiliates. All rights reserved.
 *  Copyright (C) 2009-2013 Sourcefire, Inc.
 *
 *  Authors: aCaB <acab@clamav.net>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License version 2 as
 *  published by the Free Software Foundation.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 *  MA 02110-1301, USA.
 */

#include <errno.h>

#include "clamav.h"
#include "others.h"
#include "dirent.h"
#include "w32_stat.h"
#include "shared/misc.h"

DIR *opendir(const char *name)
{
    DIR *d;
    DWORD attrs;
    int len;
    struct stat sb;
    wchar_t *wpath;

    if (stat(name, &sb) < 0)
        return NULL;

    if (!S_ISDIR(sb.st_mode)) {
        errno = ENOTDIR;
        return NULL;
    }
    if (!(d = cli_malloc(sizeof(*d)))) {
        errno = ENOMEM;
        return NULL;
    }
    wpath = uncpath(name);
    if (!wpath)
        return NULL;
    wcsncpy(d->entry, wpath, sizeof(d->entry) / sizeof(d->entry[0]));
    free(wpath);
    d->entry[sizeof(d->entry) / sizeof(d->entry[0])] = L'\0';
    len                                              = wcslen(d->entry);

    if (len >= sizeof(d->entry) / sizeof(d->entry[0]) - 4) {
        free(d);
        errno = ENAMETOOLONG;
        return NULL;
    }
    while (len--) {
        if (d->entry[len] == L'\\')
            d->entry[len] = L'\0';
        else
            break;
    }

    wcsncat(d->entry, L"\\*.*", 4);
    d->dh = INVALID_HANDLE_VALUE;
    return d;
}

struct dirent *readdir(DIR *dirp)
{
    while (1) {
        if (dirp->dh == INVALID_HANDLE_VALUE) {
            if ((dirp->dh = FindFirstFileW(dirp->entry, &dirp->wfd)) == INVALID_HANDLE_VALUE) {
                errno = ENOENT;
                return NULL;
            }
        } else {
            if (!(FindNextFileW(dirp->dh, &dirp->wfd))) {
                errno = (GetLastError() == ERROR_NO_MORE_FILES) ? 0 : ENOENT;
                return NULL;
            }
        }
        if (!WideCharToMultiByte(CP_UTF8, 0, dirp->wfd.cFileName, -1, dirp->ent.d_name, sizeof(dirp->ent.d_name), NULL, NULL))
            continue; /* FIXME: WARN HERE ! */
        dirp->ent.d_ino = dirp->wfd.ftCreationTime.dwLowDateTime ^ dirp->wfd.nFileSizeLow;
        if (!dirp->ent.d_ino) dirp->ent.d_ino = 0x1337;
        dirp->ent.d_type = (dirp->wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ? DT_DIR : DT_REG;
        break;
    }
    return &dirp->ent;
}

void rewinddir(DIR *dirp)
{
    if (dirp->dh != INVALID_HANDLE_VALUE)
        FindClose(dirp->dh);
    dirp->dh = INVALID_HANDLE_VALUE;
}

int closedir(DIR *dirp)
{
    rewinddir(dirp);
    free(dirp);
    return 0;
}
