xref: /openssl/crypto/LPdir_win.c (revision 0e9725bc)
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)) > 0 ||
110                     GetLastError() != ERROR_NO_UNICODE_TRANSLATION)
111                     break;
112 # endif
113                 sz = MultiByteToWideChar((cp = CP_ACP), 0,
114                                          directory, len_0,
115                                          NULL, 0);
116             } while (0);
117 
118             if (sz > 0) {
119                 /*
120                  * allocate two additional characters in case we need to
121                  * concatenate asterisk, |sz| covers trailing '\0'!
122                  */
123                 wdir = _alloca((sz + 2) * sizeof(TCHAR));
124                 if (!MultiByteToWideChar(cp, 0, directory, len_0,
125                                          (WCHAR *)wdir, sz)) {
126                     free(*ctx);
127                     *ctx = NULL;
128                     errno = EINVAL;
129                     return 0;
130                 }
131             } else
132 #endif
133             {
134                 sz = len_0;
135                 /*
136                  * allocate two additional characters in case we need to
137                  * concatenate asterisk, |sz| covers trailing '\0'!
138                  */
139                 wdir = _alloca((sz + 2) * sizeof(TCHAR));
140                 for (index = 0; index < len_0; index++)
141                     wdir[index] = (TCHAR)directory[index];
142             }
143 
144             sz--; /* wdir[sz] is trailing '\0' now */
145             if (wdir[sz - 1] != TEXT('*')) {
146                 if (wdir[sz - 1] != TEXT('/') && wdir[sz - 1] != TEXT('\\'))
147                     _tcscpy(wdir + sz, TEXT("/*"));
148                 else
149                     _tcscpy(wdir + sz, TEXT("*"));
150             }
151 
152             (*ctx)->handle = FindFirstFile(wdir, &(*ctx)->ctx);
153         } else {
154             if (directory[dirlen - 1] != '*') {
155                 char *buf = _alloca(dirlen + 3);
156 
157                 strcpy(buf, directory);
158                 if (buf[dirlen - 1] != '/' && buf[dirlen - 1] != '\\')
159                     strcpy(buf + dirlen, "/*");
160                 else
161                     strcpy(buf + dirlen, "*");
162 
163                 directory = buf;
164             }
165 
166             (*ctx)->handle = FindFirstFile((TCHAR *)directory, &(*ctx)->ctx);
167         }
168 
169         if ((*ctx)->handle == INVALID_HANDLE_VALUE) {
170             free(*ctx);
171             *ctx = NULL;
172             errno = EINVAL;
173             return 0;
174         }
175     } else {
176         if (FindNextFile((*ctx)->handle, &(*ctx)->ctx) == FALSE) {
177             return 0;
178         }
179     }
180     if (sizeof(TCHAR) != sizeof(char)) {
181         TCHAR *wdir = (*ctx)->ctx.cFileName;
182         size_t index, len_0 = 0;
183 
184         while (wdir[len_0] && len_0 < (sizeof((*ctx)->entry_name) - 1))
185             len_0++;
186         len_0++;
187 
188 #ifdef LP_MULTIBYTE_AVAILABLE
189         if (!WideCharToMultiByte(CP_DEFAULT, 0, (WCHAR *)wdir, len_0,
190                                  (*ctx)->entry_name,
191                                  sizeof((*ctx)->entry_name), NULL, 0))
192 #endif
193             for (index = 0; index < len_0; index++)
194                 (*ctx)->entry_name[index] = (char)wdir[index];
195     } else
196         strncpy((*ctx)->entry_name, (const char *)(*ctx)->ctx.cFileName,
197                 sizeof((*ctx)->entry_name) - 1);
198 
199     (*ctx)->entry_name[sizeof((*ctx)->entry_name) - 1] = '\0';
200 
201     return (*ctx)->entry_name;
202 }
203 
LP_find_file_end(LP_DIR_CTX ** ctx)204 int LP_find_file_end(LP_DIR_CTX **ctx)
205 {
206     if (ctx != NULL && *ctx != NULL) {
207         FindClose((*ctx)->handle);
208         free(*ctx);
209         *ctx = NULL;
210         return 1;
211     }
212     errno = EINVAL;
213     return 0;
214 }
215