crypto/LPdir_win.c

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