/*

  *  Copyright (C) 2013-2019 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;

 }