crypto/LPdir_win.c

   1 /* $LP: LPlib/source/LPdir_win.c,v 1.6 2004/07/21 21:16:23 _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 /* 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 struct LP_dir_context_st
  45 {
  46   WIN32_FIND_DATA ctx;
  47   HANDLE handle;
  48   char entry_name[NAME_MAX+1];
  49 };
  50
  51 const char *LP_find_file(LP_DIR_CTX **ctx, const char *directory)
  52 {
  53   struct dirent *direntry = NULL;
  54
  55   if (ctx == NULL || directory == NULL)
  56     {
  57       errno = EINVAL;
  58       return 0;
  59     }
  60
  61   errno = 0;
  62   if (*ctx == NULL)
  63     {
  64       *ctx = (LP_DIR_CTX *)malloc(sizeof(LP_DIR_CTX));
  65       if (*ctx == NULL)
  66         {
  67           errno = ENOMEM;
  68           return 0;
  69         }
  70       memset(*ctx, '\0', sizeof(LP_DIR_CTX));
  71
  72       if (sizeof(TCHAR) != sizeof(char))
  73         {
  74           TCHAR *wdir = NULL;
  75           /* len_0 denotes string length *with* trailing 0 */
  76           size_t index = 0,len_0 = strlen(direcory) + 1;
  77
  78           wdir = (TCHAR *)malloc(len_0 * sizeof(TCHAR));
  79           if (wdir == NULL)
  80             {
  81               free(*ctx);
  82               *ctx = NULL;
  83               errno = ENOMEM;
  84               return 0;
  85             }
  86
  87 #ifdef LP_MULTIBYTE_AVAILABLE
  88           if (!MultiByteToWideChar(CP_ACP, 0, directory, len_0, wdir, len_0))
  89 #endif
  90             for (index = 0; index < len_0; index++)
  91               wdir[index] = (TCHAR)directory[index];
  92
  93           (*ctx)->handle = FindFirstFile(wdir, &(*ctx)->ctx);
  94
  95           free(wdir);
  96         }
  97       else
  98         (*ctx)->handle = FindFirstFile(directory, &(*ctx)->ctx);
  99
 100       if ((*ctx)->handle == INVALID_HANDLE_VALUE)
 101         {
 102           free(*ctx);
 103           *ctx = NULL;
 104           errno = EINVAL;
 105           return 0;
 106         }
 107     }
 108   else
 109     {
 110       if (FindNextFile((*ctx)->handle, (*ctx)->ctx) == FALSE)
 111         {
 112           return 0;
 113         }
 114     }
 115
 116   if (sizeof(TCHAR) != sizeof(char))
 117     {
 118       TCHAR *wdir = (*ctx)->ctx.cFileName;
 119       size_t index, len_0 = 0;
 120
 121       while (wdir[len] && len < (sizeof((*ctx)->entry_name) - 1)) len_0++;
 122       len_0++;
 123
 124 #ifdef LP_MULTIBYTE_AVAILABLE
 125       if (!WideCharToMultiByte(CP_ACP, 0, wdir, len_0, (*ctx)->entry_name,
 126                                sizeof((*ctx)->entry_name), NULL, 0))
 127 #endif
 128         for (index = 0; index < len_0; index++)
 129           (*ctx)->entry_name[index] = (char)wdir[index];
 130     }
 131   else
 132     strncpy((*ctx)->entry_name, (*ctx)->ctx.cFileName,
 133             sizeof((*ctx)->entry_name)-1);
 134
 135   (*ctx)->entry_name[sizeof((*ctx)->entry_name)-1] = '\0';
 136
 137   return (*ctx)->entry_name;
 138 }
 139
 140 int LP_find_file_end(LP_DIR_CTX **ctx)
 141 {
 142   if (ctx != NULL && *ctx != NULL)
 143     {
 144       FindClose((*ctx)->handle);
 145       free(*ctx);
 146       *ctx = NULL;
 147       return 1;
 148     }
 149   errno = EINVAL;
 150   return 0;
 151 }