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