crypto/LPdir_win.c

   1 /* $LP: LPlib/source/LPdir_win.c,v 1.10 2004/08/26 13:36:05 _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 COPYRIGHT HOLDERS AND CONTRIBUTORS
  16  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  17  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  18  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  19  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  20  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  21  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  25  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27 #include <windows.h>
  28 #include <tchar.h>
  29 #ifndef LPDIR_H
  30 #include "LPdir.h"
  31 #endif
  32
  33 /* We're most likely overcautious here, but let's reserve for
  34     broken WinCE headers and explicitly opt for UNICODE call.
  35     Keep in mind that our WinCE builds are compiled with -DUNICODE
  36     [as well as -D_UNICODE]. */
  37 #if defined(LP_SYS_WINCE) && !defined(FindFirstFile)
  38 # define FindFirstFile FindFirstFileW
  39 #endif
  40 #if defined(LP_SYS_WINCE) && !defined(FindFirstFile)
  41 # define FindNextFile FindNextFileW
  42 #endif
  43
  44 #ifndef NAME_MAX
  45 #define NAME_MAX 255
  46 #endif
  47
  48 struct LP_dir_context_st
  49 {
  50   WIN32_FIND_DATA ctx;
  51   HANDLE handle;
  52   char entry_name[NAME_MAX+1];
  53 };
  54
  55 const char *LP_find_file(LP_DIR_CTX **ctx, const char *directory)
  56 {
  57   struct dirent *direntry = NULL;
  58
  59   if (ctx == NULL || directory == NULL)
  60     {
  61       errno = EINVAL;
  62       return 0;
  63     }
  64
  65   errno = 0;
  66   if (*ctx == NULL)
  67     {
  68       *ctx = (LP_DIR_CTX *)malloc(sizeof(LP_DIR_CTX));
  69       if (*ctx == NULL)
  70         {
  71           errno = ENOMEM;
  72           return 0;
  73         }
  74       memset(*ctx, '\0', sizeof(LP_DIR_CTX));
  75
  76       if (sizeof(TCHAR) != sizeof(char))
  77         {
  78           TCHAR *wdir = NULL;
  79           /* len_0 denotes string length *with* trailing 0 */
  80           size_t index = 0,len_0 = strlen(directory) + 1;
  81
  82           wdir = (TCHAR *)malloc(len_0 * sizeof(TCHAR));
  83           if (wdir == NULL)
  84             {
  85               free(*ctx);
  86               *ctx = NULL;
  87               errno = ENOMEM;
  88               return 0;
  89             }
  90
  91 #ifdef LP_MULTIBYTE_AVAILABLE
  92           if (!MultiByteToWideChar(CP_ACP, 0, directory, len_0, (WCHAR *)wdir, len_0))
  93 #endif
  94             for (index = 0; index < len_0; index++)
  95               wdir[index] = (TCHAR)directory[index];
  96
  97           (*ctx)->handle = FindFirstFile(wdir, &(*ctx)->ctx);
  98
  99           free(wdir);
 100         }
 101       else
 102         (*ctx)->handle = FindFirstFile((TCHAR *)directory, &(*ctx)->ctx);
 103
 104       if ((*ctx)->handle == INVALID_HANDLE_VALUE)
 105         {
 106           free(*ctx);
 107           *ctx = NULL;
 108           errno = EINVAL;
 109           return 0;
 110         }
 111     }
 112   else
 113     {
 114       if (FindNextFile((*ctx)->handle, &(*ctx)->ctx) == FALSE)
 115         {
 116           return 0;
 117         }
 118     }
 119
 120   if (sizeof(TCHAR) != sizeof(char))
 121     {
 122       TCHAR *wdir = (*ctx)->ctx.cFileName;
 123       size_t index, len_0 = 0;
 124
 125       while (wdir[len_0] && len_0 < (sizeof((*ctx)->entry_name) - 1)) len_0++;
 126       len_0++;
 127
 128 #ifdef LP_MULTIBYTE_AVAILABLE
 129       if (!WideCharToMultiByte(CP_ACP, 0, (WCHAR *)wdir, len_0, (*ctx)->entry_name,
 130                                sizeof((*ctx)->entry_name), NULL, 0))
 131 #endif
 132         for (index = 0; index < len_0; index++)
 133           (*ctx)->entry_name[index] = (char)wdir[index];
 134     }
 135   else
 136     strncpy((*ctx)->entry_name, (const char *)(*ctx)->ctx.cFileName,
 137             sizeof((*ctx)->entry_name)-1);
 138
 139   (*ctx)->entry_name[sizeof((*ctx)->entry_name)-1] = '\0';
 140
 141   return (*ctx)->entry_name;
 142 }
 143
 144 int LP_find_file_end(LP_DIR_CTX **ctx)
 145 {
 146   if (ctx != NULL && *ctx != NULL)
 147     {
 148       FindClose((*ctx)->handle);
 149       free(*ctx);
 150       *ctx = NULL;
 151       return 1;
 152     }
 153   errno = EINVAL;
 154   return 0;
 155 }