Home | History | Annotate | Download | only in crypto 
      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 (at) 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 }
    156