]>
Commit | Line | Data |
---|---|---|
a2400fca RL |
1 | /* |
2 | * Copyright (c) 2004, Richard Levitte <richard@levitte.org> | |
3 | * All rights reserved. | |
4 | * | |
5 | * Redistribution and use in source and binary forms, with or without | |
6 | * modification, are permitted provided that the following conditions | |
7 | * are met: | |
8 | * 1. Redistributions of source code must retain the above copyright | |
9 | * notice, this list of conditions and the following disclaimer. | |
10 | * 2. Redistributions in binary form must reproduce the above copyright | |
11 | * notice, this list of conditions and the following disclaimer in the | |
12 | * documentation and/or other materials provided with the distribution. | |
ae5c8664 | 13 | * |
bb09fd2b RL |
14 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
15 | * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | |
16 | * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | |
17 | * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | |
18 | * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
19 | * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | |
20 | * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | |
21 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | |
22 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
23 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
24 | * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
a2400fca RL |
25 | */ |
26 | #include <windows.h> | |
64ba6cf2 | 27 | #include <tchar.h> |
a2400fca | 28 | #ifndef LPDIR_H |
ae5c8664 | 29 | # include "LPdir.h" |
a2400fca RL |
30 | #endif |
31 | ||
ae5c8664 MC |
32 | /* |
33 | * We're most likely overcautious here, but let's reserve for broken WinCE | |
34 | * headers and explicitly opt for UNICODE call. Keep in mind that our WinCE | |
35 | * builds are compiled with -DUNICODE [as well as -D_UNICODE]. | |
36 | */ | |
765e231a RL |
37 | #if defined(LP_SYS_WINCE) && !defined(FindFirstFile) |
38 | # define FindFirstFile FindFirstFileW | |
39 | #endif | |
6a7a4a4e | 40 | #if defined(LP_SYS_WINCE) && !defined(FindNextFile) |
75f134c0 RL |
41 | # define FindNextFile FindNextFileW |
42 | #endif | |
765e231a | 43 | |
da2ee71d | 44 | #ifndef NAME_MAX |
ae5c8664 | 45 | # define NAME_MAX 255 |
da2ee71d AP |
46 | #endif |
47 | ||
ae5c8664 MC |
48 | struct LP_dir_context_st { |
49 | WIN32_FIND_DATA ctx; | |
50 | HANDLE handle; | |
51 | char entry_name[NAME_MAX + 1]; | |
a2400fca RL |
52 | }; |
53 | ||
54 | const char *LP_find_file(LP_DIR_CTX **ctx, const char *directory) | |
55 | { | |
ae5c8664 MC |
56 | if (ctx == NULL || directory == NULL) { |
57 | errno = EINVAL; | |
58 | return 0; | |
a2400fca RL |
59 | } |
60 | ||
ae5c8664 MC |
61 | errno = 0; |
62 | if (*ctx == NULL) { | |
63 | const char *extdir = directory; | |
64 | char *extdirbuf = NULL; | |
65 | size_t dirlen = strlen(directory); | |
66 | ||
67 | if (dirlen == 0) { | |
68 | errno = ENOENT; | |
69 | return 0; | |
70 | } | |
71 | ||
72 | *ctx = (LP_DIR_CTX *)malloc(sizeof(LP_DIR_CTX)); | |
73 | if (*ctx == NULL) { | |
74 | errno = ENOMEM; | |
75 | return 0; | |
76 | } | |
77 | memset(*ctx, '\0', sizeof(LP_DIR_CTX)); | |
78 | ||
79 | if (directory[dirlen - 1] != '*') { | |
80 | extdirbuf = (char *)malloc(dirlen + 3); | |
81 | if (extdirbuf == NULL) { | |
82 | free(*ctx); | |
83 | *ctx = NULL; | |
84 | errno = ENOMEM; | |
85 | return 0; | |
86 | } | |
87 | if (directory[dirlen - 1] != '/' && directory[dirlen - 1] != '\\') | |
88 | extdir = strcat(strcpy(extdirbuf, directory), "/*"); | |
89 | else | |
90 | extdir = strcat(strcpy(extdirbuf, directory), "*"); | |
91 | } | |
92 | ||
93 | if (sizeof(TCHAR) != sizeof(char)) { | |
94 | TCHAR *wdir = NULL; | |
95 | /* len_0 denotes string length *with* trailing 0 */ | |
96 | size_t index = 0, len_0 = strlen(extdir) + 1; | |
21753432 | 97 | size_t amount; |
ae5c8664 | 98 | |
21753432 RL |
99 | /* |
100 | * Size check | |
101 | * The reasoning is that absolutely worst case, each byte in | |
102 | * extdir will take up one TCHAR each, so the maximum size in | |
103 | * bytes that we can tolerate is MAX_PATH TCHARs... not counting | |
104 | * the ending NUL. | |
105 | */ | |
106 | if ((len_0 - 1) > MAX_PATH * sizeof(TCHAR)) { | |
107 | free(*ctx); | |
108 | *ctx = NULL; | |
109 | errno = EINVAL; | |
110 | return 0; | |
111 | } | |
112 | amount = len_0 * sizeof(TCHAR); | |
113 | wdir = (TCHAR *)malloc(amount); | |
ae5c8664 MC |
114 | if (wdir == NULL) { |
115 | if (extdirbuf != NULL) { | |
116 | free(extdirbuf); | |
117 | } | |
118 | free(*ctx); | |
119 | *ctx = NULL; | |
120 | errno = ENOMEM; | |
121 | return 0; | |
122 | } | |
64ba6cf2 | 123 | #ifdef LP_MULTIBYTE_AVAILABLE |
ae5c8664 MC |
124 | if (!MultiByteToWideChar |
125 | (CP_ACP, 0, extdir, len_0, (WCHAR *)wdir, len_0)) | |
64ba6cf2 | 126 | #endif |
ae5c8664 MC |
127 | for (index = 0; index < len_0; index++) |
128 | wdir[index] = (TCHAR)extdir[index]; | |
129 | ||
130 | (*ctx)->handle = FindFirstFile(wdir, &(*ctx)->ctx); | |
131 | ||
132 | free(wdir); | |
133 | } else { | |
134 | (*ctx)->handle = FindFirstFile((TCHAR *)extdir, &(*ctx)->ctx); | |
135 | } | |
136 | if (extdirbuf != NULL) { | |
137 | free(extdirbuf); | |
138 | } | |
139 | ||
140 | if ((*ctx)->handle == INVALID_HANDLE_VALUE) { | |
141 | free(*ctx); | |
142 | *ctx = NULL; | |
143 | errno = EINVAL; | |
144 | return 0; | |
145 | } | |
146 | } else { | |
147 | if (FindNextFile((*ctx)->handle, &(*ctx)->ctx) == FALSE) { | |
148 | return 0; | |
149 | } | |
a2400fca | 150 | } |
ae5c8664 MC |
151 | if (sizeof(TCHAR) != sizeof(char)) { |
152 | TCHAR *wdir = (*ctx)->ctx.cFileName; | |
153 | size_t index, len_0 = 0; | |
64ba6cf2 | 154 | |
ae5c8664 MC |
155 | while (wdir[len_0] && len_0 < (sizeof((*ctx)->entry_name) - 1)) |
156 | len_0++; | |
157 | len_0++; | |
64ba6cf2 RL |
158 | |
159 | #ifdef LP_MULTIBYTE_AVAILABLE | |
ae5c8664 MC |
160 | if (!WideCharToMultiByte |
161 | (CP_ACP, 0, (WCHAR *)wdir, len_0, (*ctx)->entry_name, | |
162 | sizeof((*ctx)->entry_name), NULL, 0)) | |
64ba6cf2 | 163 | #endif |
ae5c8664 MC |
164 | for (index = 0; index < len_0; index++) |
165 | (*ctx)->entry_name[index] = (char)wdir[index]; | |
166 | } else | |
167 | strncpy((*ctx)->entry_name, (const char *)(*ctx)->ctx.cFileName, | |
168 | sizeof((*ctx)->entry_name) - 1); | |
64ba6cf2 | 169 | |
ae5c8664 | 170 | (*ctx)->entry_name[sizeof((*ctx)->entry_name) - 1] = '\0'; |
64ba6cf2 | 171 | |
ae5c8664 | 172 | return (*ctx)->entry_name; |
a2400fca RL |
173 | } |
174 | ||
175 | int LP_find_file_end(LP_DIR_CTX **ctx) | |
176 | { | |
ae5c8664 MC |
177 | if (ctx != NULL && *ctx != NULL) { |
178 | FindClose((*ctx)->handle); | |
179 | free(*ctx); | |
180 | *ctx = NULL; | |
181 | return 1; | |
a2400fca | 182 | } |
ae5c8664 MC |
183 | errno = EINVAL; |
184 | return 0; | |
a2400fca | 185 | } |