apps/win32_init.c

   1 /*
   2  * Copyright 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 #include <windows.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13 #include <malloc.h>
  14
  15 #if defined(CP_UTF8)
  16
  17 static UINT saved_cp;
  18 static int newargc;
  19 static char **newargv;
  20
  21 static void cleanup(void)
  22 {
  23     int i;
  24
  25     SetConsoleOutputCP(saved_cp);
  26
  27     for (i = 0; i < newargc; i++)
  28         free(newargv[i]);
  29
  30     free(newargv);
  31 }
  32
  33 /*
  34  * Incrementally [re]allocate newargv and keep it NULL-terminated.
  35  */
  36 static int validate_argv(int argc)
  37 {
  38     static int size = 0;
  39
  40     if (argc >= size) {
  41         char **ptr;
  42
  43         while (argc >= size)
  44             size += 64;
  45
  46         ptr = realloc(newargv, size * sizeof(newargv[0]));
  47         if (ptr == NULL)
  48             return 0;
  49
  50         (newargv = ptr)[argc] = NULL;
  51     } else {
  52         newargv[argc] = NULL;
  53     }
  54
  55     return 1;
  56 }
  57
  58 static int process_glob(WCHAR *wstr, int wlen)
  59 {
  60     int i, slash, udlen;
  61     WCHAR saved_char;
  62     WIN32_FIND_DATAW data;
  63     HANDLE h;
  64
  65     /*
  66      * Note that we support wildcard characters only in filename part
  67      * of the path, and not in directories. Windows users are used to
  68      * this, that's why recursive glob processing is not implemented.
  69      */
  70     /*
  71      * Start by looking for last slash or backslash, ...
  72      */
  73     for (slash = 0, i = 0; i < wlen; i++)
  74         if (wstr[i] == L'/' || wstr[i] == L'\\')
  75             slash = i + 1;
  76     /*
  77      * ... then look for asterisk or question mark in the file name.
  78      */
  79     for (i = slash; i < wlen; i++)
  80         if (wstr[i] == L'*' || wstr[i] == L'?')
  81             break;
  82
  83     if (i == wlen)
  84         return 0;   /* definitely not a glob */
  85
  86     saved_char = wstr[wlen];
  87     wstr[wlen] = L'\0';
  88     h = FindFirstFileW(wstr, &data);
  89     wstr[wlen] = saved_char;
  90     if (h == INVALID_HANDLE_VALUE)
  91         return 0;   /* not a valid glob, just pass... */
  92
  93     if (slash)
  94         udlen = WideCharToMultiByte(CP_UTF8, 0, wstr, slash,
  95                                     NULL, 0, NULL, NULL);
  96     else
  97         udlen = 0;
  98
  99     do {
 100         int uflen;
 101         char *arg;
 102
 103         /*
 104          * skip over . and ..
 105          */
 106         if (data.cFileName[0] == L'.') {
 107             if ((data.cFileName[1] == L'\0') ||
 108                 (data.cFileName[1] == L'.' && data.cFileName[2] == L'\0'))
 109                 continue;
 110         }
 111
 112         if (!validate_argv(newargc + 1))
 113             break;
 114
 115         /*
 116          * -1 below means "scan for trailing '\0' *and* count it",
 117          * so that |uflen| covers even trailing '\0'.
 118          */
 119         uflen = WideCharToMultiByte(CP_UTF8, 0, data.cFileName, -1,
 120                                     NULL, 0, NULL, NULL);
 121
 122         arg = malloc(udlen + uflen);
 123         if (arg == NULL)
 124             break;
 125
 126         if (udlen)
 127             WideCharToMultiByte(CP_UTF8, 0, wstr, slash,
 128                                 arg, udlen, NULL, NULL);
 129
 130         WideCharToMultiByte(CP_UTF8, 0, data.cFileName, -1,
 131                             arg + udlen, uflen, NULL, NULL);
 132
 133         newargv[newargc++] = arg;
 134     } while (FindNextFileW(h, &data));
 135
 136     CloseHandle(h);
 137
 138     return 1;
 139 }
 140
 141 void win32_utf8argv(int *argc, char **argv[])
 142 {
 143     const WCHAR *wcmdline;
 144     WCHAR *warg, *wend, *p;
 145     int wlen, ulen, valid = 1;
 146     char *arg;
 147
 148     newargc = 0;
 149     newargv = NULL;
 150     if (!validate_argv(newargc))
 151         return;
 152
 153     wcmdline = GetCommandLineW();
 154     if (wcmdline == NULL) return;
 155
 156     /*
 157      * make a copy of the command line, since we might have to modify it...
 158      */
 159     wlen = wcslen(wcmdline);
 160     p = _alloca((wlen + 1) * sizeof(WCHAR));
 161     wcscpy(p, wcmdline);
 162
 163     while (*p != L'\0') {
 164         int in_quote = 0;
 165
 166         if (*p == L' ' || *p == L'\t') {
 167             p++; /* skip over white spaces */
 168             continue;
 169         }
 170
 171         /*
 172          * Note: because we may need to fiddle with the number of backslashes,
 173          * the argument string is copied into itself.  This is safe because
 174          * the number of characters will never expand.
 175          */
 176         warg = wend = p;
 177         while (*p != L'\0'
 178                && (in_quote || (*p != L' ' && *p != L'\t'))) {
 179             switch (*p) {
 180             case L'\\':
 181                 /*
 182                  * Microsoft documentation on how backslashes are treated
 183                  * is:
 184                  *
 185                  * + Backslashes are interpreted literally, unless they
 186                  *   immediately precede a double quotation mark.
 187                  * + If an even number of backslashes is followed by a double
 188                  *   quotation mark, one backslash is placed in the argv array
 189                  *   for every pair of backslashes, and the double quotation
 190                  *   mark is interpreted as a string delimiter.
 191                  * + If an odd number of backslashes is followed by a double
 192                  *   quotation mark, one backslash is placed in the argv array
 193                  *   for every pair of backslashes, and the double quotation
 194                  *   mark is "escaped" by the remaining backslash, causing a
 195                  *   literal double quotation mark (") to be placed in argv.
 196                  *
 197                  * Ref: https://msdn.microsoft.com/en-us/library/17w5ykft.aspx
 198                  *
 199                  * Though referred page doesn't mention it, multiple qouble
 200                  * quotes are also special. Pair of double quotes in quoted
 201                  * string is counted as single double quote.
 202                  */
 203                 {
 204                     const WCHAR *q = p;
 205                     int i;
 206
 207                     while (*p == L'\\')
 208                         p++;
 209
 210                     if (*p == L'"') {
 211                         int i;
 212
 213                         for (i = (p - q) / 2; i > 0; i--)
 214                             *wend++ = L'\\';
 215
 216                         /*
 217                          * if odd amount of backslashes before the quote,
 218                          * said quote is part of the argument, not a delimiter
 219                          */
 220                         if ((p - q) % 2 == 1)
 221                             *wend++ = *p++;
 222                     } else {
 223                         for (i = p - q; i > 0; i--)
 224                             *wend++ = L'\\';
 225                     }
 226                 }
 227                 break;
 228             case L'"':
 229                 /*
 230                  * Without the preceding backslash (or when preceded with an
 231                  * even number of backslashes), the double quote is a simple
 232                  * string delimiter and just slightly change the parsing state
 233                  */
 234                 if (in_quote && p[1] == L'"')
 235                     *wend++ = *p++;
 236                 else
 237                     in_quote = !in_quote;
 238                 p++;
 239                 break;
 240             default:
 241                 /*
 242                  * Any other non-delimiter character is just taken verbatim
 243                  */
 244                 *wend++ = *p++;
 245             }
 246         }
 247
 248         wlen = wend - warg;
 249
 250         if (wlen == 0 || !process_glob(warg, wlen)) {
 251             if (!validate_argv(newargc + 1)) {
 252                 valid = 0;
 253                 break;
 254             }
 255
 256             ulen = 0;
 257             if (wlen > 0) {
 258                 ulen = WideCharToMultiByte(CP_UTF8, 0, warg, wlen,
 259                                            NULL, 0, NULL, NULL);
 260                 if (ulen <= 0)
 261                     continue;
 262             }
 263
 264             arg = malloc(ulen + 1);
 265             if (arg == NULL) {
 266                 valid = 0;
 267                 break;
 268             }
 269
 270             if (wlen > 0)
 271                 WideCharToMultiByte(CP_UTF8, 0, warg, wlen,
 272                                     arg, ulen, NULL, NULL);
 273             arg[ulen] = '\0';
 274
 275             newargv[newargc++] = arg;
 276         }
 277     }
 278
 279     if (valid) {
 280         saved_cp = GetConsoleOutputCP();
 281         SetConsoleOutputCP(CP_UTF8);
 282
 283         *argc = newargc;
 284         *argv = newargv;
 285
 286         atexit(cleanup);
 287     } else if (newargv != NULL) {
 288         int i;
 289
 290         for (i = 0; i < newargc; i++)
 291             free(newargv[i]);
 292
 293         free(newargv);
 294
 295         newargc = 0;
 296         newargv = NULL;
 297     }
 298
 299     return;
 300 }
 301 #else
 302 void win32_utf8argv(int &argc, char **argv[])
 303 {   return;   }
 304 #endif