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