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   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 }
    154