crypto/LPdir_win.c

   1 /* $LP: LPlib/source/LPdir_win.c,v 1.4 2004/07/20 21:15:55 _cvs_levitte Exp $ */
   2 /*
   3  * Copyright (c) 2004, Richard Levitte <richard@levitte.org>
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  16  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  19  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  21  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  23  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  24  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  25  * SUCH DAMAGE.
  26  */
  27 #include <windows.h>
  28 #include <tchar.h>
  29 #ifndef LPDIR_H
  30 #include "LPdir.h"
  31 #endif
  32
  33 struct LP_dir_context_st
  34 {
  35   WIN32_FIND_DATA ctx;
  36   HANDLE handle;
  37   char entry_name[NAME_MAX+1];
  38 };
  39
  40 const char *LP_find_file(LP_DIR_CTX **ctx, const char *directory)
  41 {
  42   struct dirent *direntry = NULL;
  43
  44   if (ctx == NULL || directory == NULL)
  45     {
  46       errno = EINVAL;
  47       return 0;
  48     }
  49
  50   errno = 0;
  51   if (*ctx == NULL)
  52     {
  53       *ctx = (LP_DIR_CTX *)malloc(sizeof(LP_DIR_CTX));
  54       if (*ctx == NULL)
  55         {
  56           errno = ENOMEM;
  57           return 0;
  58         }
  59       memset(*ctx, '\0', sizeof(LP_DIR_CTX));
  60
  61       if (sizeof(TCHAR) != sizeof(char))
  62         {
  63           TCHAR *wdir = NULL;
  64           size_t index = 0,len=strlen(direcory);
  65
  66           wdir = (TCHAR *)malloc((len + 1) * sizeof(TCHAR));
  67           if (wdir == NULL)
  68             {
  69               errno = ENOMEM;
  70               free(*ctx);
  71               return 0;
  72             }
  73
  74 #ifdef LP_MULTIBYTE_AVAILABLE
  75           if (!MultiByteToWideChar (CP_THREAD_ACP,0,directory,len,wdir,len+1))
  76 #endif
  77             for (index = 0; index < strlen(directory) + 1; index++)
  78               wdir[index] = (TCHAR)directory[index];
  79
  80           (*ctx)->handle = FindFirstFile(wdir, &(*ctx)->ctx);
  81
  82           free(wdir);
  83         }
  84       else
  85         (*ctx)->handle = FindFirstFile(directory, &(*ctx)->ctx);
  86
  87       if ((*ctx)->handle == INVALID_HANDLE_VALUE)
  88         {
  89           free(*ctx);
  90           *ctx = NULL;
  91           errno = EINVAL;
  92           return 0;
  93         }
  94     }
  95   else
  96     {
  97       if (FindNextFile((*ctx)->handle, (*ctx)->ctx) == FALSE)
  98         {
  99           return 0;
 100         }
 101     }
 102
 103   if (sizeof(TCHAR) != sizeof(char))
 104     {
 105       TCHAR *wdir=(*ctx)->ctx.cFileName;
 106       size_t i,len;
 107
 108       for (len=0;wdir[len] && len<(sizeof((*ctx)->entry_name)-1);)
 109         len++;
 110
 111 #ifdef LP_MULTIBYTE_AVAILABLE
 112       if (!WideCharToMultiByte (CP_THREAD_ACP,0,wdir,len,
 113                                 (*ctx)->entry_name,
 114                                 sizeof((*ctx)->entry_name)-1,NULL,0))
 115 #endif
 116         for (i=0;i<len;i++) (*ctx)->entry_name[i] = (char)wdir[i];
 117     }
 118   else
 119     strncpy((*ctx)->entry_name, (*ctx)->ctx.cFileName,
 120             sizeof((*ctx)->entry_name)-1);
 121
 122   (*ctx)->entry_name[sizeof((*ctx)->entry_name)-1] = '\0';
 123
 124   return (*ctx)->entry_name;
 125 }
 126
 127 int LP_find_file_end(LP_DIR_CTX **ctx)
 128 {
 129   if (ctx != NULL && *ctx != NULL)
 130     {
 131       FindClose((*ctx)->handle);
 132       free(*ctx);
 133       return 1;
 134     }
 135   errno = EINVAL;
 136   return 0;
 137 }