1 /*
2 * Copyright 2004-2016 The OpenSSL Project Authors. All Rights Reserved.
3 *
4 * Licensed under the Apache License 2.0 (the "License"). You may not use
5 * this file except in compliance with the License. You can obtain a copy
6 * in the file LICENSE in the source distribution or at
7 * https://www.openssl.org/source/license.html
8 */
9
10 /*
11 * This file is dual-licensed and is also available under the following
12 * terms:
13 *
14 * Copyright (c) 2004, Richard Levitte <richard@levitte.org>
15 * All rights reserved.
16 *
17 * Redistribution and use in source and binary forms, with or without
18 * modification, are permitted provided that the following conditions
19 * are met:
20 * 1. Redistributions of source code must retain the above copyright
21 * notice, this list of conditions and the following disclaimer.
22 * 2. Redistributions in binary form must reproduce the above copyright
23 * notice, this list of conditions and the following disclaimer in the
24 * documentation and/or other materials provided with the distribution.
25 *
26 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
27 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
28 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
29 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
30 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
31 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
32 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
33 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
34 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
35 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
36 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37 */
38
39 #include <windows.h>
40 #include <tchar.h>
41 #include "internal/numbers.h"
42 #ifndef LPDIR_H
43 #include "LPdir.h"
44 #endif
45
46 /*
47 * We're most likely overcautious here, but let's reserve for broken WinCE
48 * headers and explicitly opt for UNICODE call. Keep in mind that our WinCE
49 * builds are compiled with -DUNICODE [as well as -D_UNICODE].
50 */
51 #if defined(LP_SYS_WINCE) && !defined(FindFirstFile)
52 #define FindFirstFile FindFirstFileW
53 #endif
54 #if defined(LP_SYS_WINCE) && !defined(FindNextFile)
55 #define FindNextFile FindNextFileW
56 #endif
57
58 #ifndef NAME_MAX
59 #define NAME_MAX 255
60 #endif
61
62 #ifdef CP_UTF8
63 #define CP_DEFAULT CP_UTF8
64 #else
65 #define CP_DEFAULT CP_ACP
66 #endif
67
68 struct LP_dir_context_st {
69 WIN32_FIND_DATA ctx;
70 HANDLE handle;
71 char entry_name[NAME_MAX + 1];
72 };
73
LP_find_file(LP_DIR_CTX ** ctx,const char * directory)74 const char *LP_find_file(LP_DIR_CTX **ctx, const char *directory)
75 {
76 if (ctx == NULL || directory == NULL) {
77 errno = EINVAL;
78 return 0;
79 }
80
81 errno = 0;
82 if (*ctx == NULL) {
83 size_t dirlen = strlen(directory);
84
85 if (dirlen == 0 || dirlen > INT_MAX - 3) {
86 errno = ENOENT;
87 return 0;
88 }
89
90 *ctx = malloc(sizeof(**ctx));
91 if (*ctx == NULL) {
92 errno = ENOMEM;
93 return 0;
94 }
95 memset(*ctx, 0, sizeof(**ctx));
96
97 if (sizeof(TCHAR) != sizeof(char)) {
98 TCHAR *wdir = NULL;
99 /* len_0 denotes string length *with* trailing 0 */
100 size_t index = 0, len_0 = dirlen + 1;
101 #ifdef LP_MULTIBYTE_AVAILABLE
102 int sz = 0;
103 UINT cp;
104
105 do {
106 #ifdef CP_UTF8
107 if ((sz = MultiByteToWideChar((cp = CP_UTF8), 0,
108 directory, len_0,
109 NULL, 0))
110 > 0
111 || GetLastError() != ERROR_NO_UNICODE_TRANSLATION)
112 break;
113 #endif
114 sz = MultiByteToWideChar((cp = CP_ACP), 0,
115 directory, len_0,
116 NULL, 0);
117 } while (0);
118
119 if (sz > 0) {
120 /*
121 * allocate two additional characters in case we need to
122 * concatenate asterisk, |sz| covers trailing '\0'!
123 */
124 wdir = _alloca((sz + 2) * sizeof(TCHAR));
125 if (!MultiByteToWideChar(cp, 0, directory, len_0,
126 (WCHAR *)wdir, sz)) {
127 free(*ctx);
128 *ctx = NULL;
129 errno = EINVAL;
130 return 0;
131 }
132 } else
133 #endif
134 {
135 sz = len_0;
136 /*
137 * allocate two additional characters in case we need to
138 * concatenate asterisk, |sz| covers trailing '\0'!
139 */
140 wdir = _alloca((sz + 2) * sizeof(TCHAR));
141 for (index = 0; index < len_0; index++)
142 wdir[index] = (TCHAR)directory[index];
143 }
144
145 sz--; /* wdir[sz] is trailing '\0' now */
146 if (wdir[sz - 1] != TEXT('*')) {
147 if (wdir[sz - 1] != TEXT('/') && wdir[sz - 1] != TEXT('\\'))
148 _tcscpy(wdir + sz, TEXT("/*"));
149 else
150 _tcscpy(wdir + sz, TEXT("*"));
151 }
152
153 (*ctx)->handle = FindFirstFile(wdir, &(*ctx)->ctx);
154 } else {
155 if (directory[dirlen - 1] != '*') {
156 char *buf = _alloca(dirlen + 3);
157
158 strcpy(buf, directory);
159 if (buf[dirlen - 1] != '/' && buf[dirlen - 1] != '\\')
160 strcpy(buf + dirlen, "/*");
161 else
162 strcpy(buf + dirlen, "*");
163
164 directory = buf;
165 }
166
167 (*ctx)->handle = FindFirstFile((TCHAR *)directory, &(*ctx)->ctx);
168 }
169
170 if ((*ctx)->handle == INVALID_HANDLE_VALUE) {
171 free(*ctx);
172 *ctx = NULL;
173 errno = EINVAL;
174 return 0;
175 }
176 } else {
177 if (FindNextFile((*ctx)->handle, &(*ctx)->ctx) == FALSE) {
178 return 0;
179 }
180 }
181 if (sizeof(TCHAR) != sizeof(char)) {
182 TCHAR *wdir = (*ctx)->ctx.cFileName;
183 size_t index, len_0 = 0;
184
185 while (wdir[len_0] && len_0 < (sizeof((*ctx)->entry_name) - 1))
186 len_0++;
187 len_0++;
188
189 #ifdef LP_MULTIBYTE_AVAILABLE
190 if (!WideCharToMultiByte(CP_DEFAULT, 0, (WCHAR *)wdir, len_0,
191 (*ctx)->entry_name,
192 sizeof((*ctx)->entry_name), NULL, 0))
193 #endif
194 for (index = 0; index < len_0; index++)
195 (*ctx)->entry_name[index] = (char)wdir[index];
196 } else
197 strncpy((*ctx)->entry_name, (const char *)(*ctx)->ctx.cFileName,
198 sizeof((*ctx)->entry_name) - 1);
199
200 (*ctx)->entry_name[sizeof((*ctx)->entry_name) - 1] = '\0';
201
202 return (*ctx)->entry_name;
203 }
204
LP_find_file_end(LP_DIR_CTX ** ctx)205 int LP_find_file_end(LP_DIR_CTX **ctx)
206 {
207 if (ctx != NULL && *ctx != NULL) {
208 FindClose((*ctx)->handle);
209 free(*ctx);
210 *ctx = NULL;
211 return 1;
212 }
213 errno = EINVAL;
214 return 0;
215 }
216