git.oblomov.eu Git - git/blob - compat/mingw.c

   1 #include "../git-compat-util.h"
   2 #include "win32.h"
   3 #include <conio.h>
   4 #include <wchar.h>
   5 #include "../strbuf.h"
   6 #include "../run-command.h"
   7 #include "../cache.h"
   8 #include "win32/lazyload.h"
   9 #include "../config.h"
  10 #include "dir.h"
  11
  12 #define HCAST(type, handle) ((type)(intptr_t)handle)
  13
  14 static const int delay[] = { 0, 1, 10, 20, 40 };
  15
  16 int err_win_to_posix(DWORD winerr)
  17 {
  18         int error = ENOSYS;
  19         switch(winerr) {
  20         case ERROR_ACCESS_DENIED: error = EACCES; break;
  21         case ERROR_ACCOUNT_DISABLED: error = EACCES; break;
  22         case ERROR_ACCOUNT_RESTRICTION: error = EACCES; break;
  23         case ERROR_ALREADY_ASSIGNED: error = EBUSY; break;
  24         case ERROR_ALREADY_EXISTS: error = EEXIST; break;
  25         case ERROR_ARITHMETIC_OVERFLOW: error = ERANGE; break;
  26         case ERROR_BAD_COMMAND: error = EIO; break;
  27         case ERROR_BAD_DEVICE: error = ENODEV; break;
  28         case ERROR_BAD_DRIVER_LEVEL: error = ENXIO; break;
  29         case ERROR_BAD_EXE_FORMAT: error = ENOEXEC; break;
  30         case ERROR_BAD_FORMAT: error = ENOEXEC; break;
  31         case ERROR_BAD_LENGTH: error = EINVAL; break;
  32         case ERROR_BAD_PATHNAME: error = ENOENT; break;
  33         case ERROR_BAD_PIPE: error = EPIPE; break;
  34         case ERROR_BAD_UNIT: error = ENODEV; break;
  35         case ERROR_BAD_USERNAME: error = EINVAL; break;
  36         case ERROR_BROKEN_PIPE: error = EPIPE; break;
  37         case ERROR_BUFFER_OVERFLOW: error = ENAMETOOLONG; break;
  38         case ERROR_BUSY: error = EBUSY; break;
  39         case ERROR_BUSY_DRIVE: error = EBUSY; break;
  40         case ERROR_CALL_NOT_IMPLEMENTED: error = ENOSYS; break;
  41         case ERROR_CANNOT_MAKE: error = EACCES; break;
  42         case ERROR_CANTOPEN: error = EIO; break;
  43         case ERROR_CANTREAD: error = EIO; break;
  44         case ERROR_CANTWRITE: error = EIO; break;
  45         case ERROR_CRC: error = EIO; break;
  46         case ERROR_CURRENT_DIRECTORY: error = EACCES; break;
  47         case ERROR_DEVICE_IN_USE: error = EBUSY; break;
  48         case ERROR_DEV_NOT_EXIST: error = ENODEV; break;
  49         case ERROR_DIRECTORY: error = EINVAL; break;
  50         case ERROR_DIR_NOT_EMPTY: error = ENOTEMPTY; break;
  51         case ERROR_DISK_CHANGE: error = EIO; break;
  52         case ERROR_DISK_FULL: error = ENOSPC; break;
  53         case ERROR_DRIVE_LOCKED: error = EBUSY; break;
  54         case ERROR_ENVVAR_NOT_FOUND: error = EINVAL; break;
  55         case ERROR_EXE_MARKED_INVALID: error = ENOEXEC; break;
  56         case ERROR_FILENAME_EXCED_RANGE: error = ENAMETOOLONG; break;
  57         case ERROR_FILE_EXISTS: error = EEXIST; break;
  58         case ERROR_FILE_INVALID: error = ENODEV; break;
  59         case ERROR_FILE_NOT_FOUND: error = ENOENT; break;
  60         case ERROR_GEN_FAILURE: error = EIO; break;
  61         case ERROR_HANDLE_DISK_FULL: error = ENOSPC; break;
  62         case ERROR_INSUFFICIENT_BUFFER: error = ENOMEM; break;
  63         case ERROR_INVALID_ACCESS: error = EACCES; break;
  64         case ERROR_INVALID_ADDRESS: error = EFAULT; break;
  65         case ERROR_INVALID_BLOCK: error = EFAULT; break;
  66         case ERROR_INVALID_DATA: error = EINVAL; break;
  67         case ERROR_INVALID_DRIVE: error = ENODEV; break;
  68         case ERROR_INVALID_EXE_SIGNATURE: error = ENOEXEC; break;
  69         case ERROR_INVALID_FLAGS: error = EINVAL; break;
  70         case ERROR_INVALID_FUNCTION: error = ENOSYS; break;
  71         case ERROR_INVALID_HANDLE: error = EBADF; break;
  72         case ERROR_INVALID_LOGON_HOURS: error = EACCES; break;
  73         case ERROR_INVALID_NAME: error = EINVAL; break;
  74         case ERROR_INVALID_OWNER: error = EINVAL; break;
  75         case ERROR_INVALID_PARAMETER: error = EINVAL; break;
  76         case ERROR_INVALID_PASSWORD: error = EPERM; break;
  77         case ERROR_INVALID_PRIMARY_GROUP: error = EINVAL; break;
  78         case ERROR_INVALID_SIGNAL_NUMBER: error = EINVAL; break;
  79         case ERROR_INVALID_TARGET_HANDLE: error = EIO; break;
  80         case ERROR_INVALID_WORKSTATION: error = EACCES; break;
  81         case ERROR_IO_DEVICE: error = EIO; break;
  82         case ERROR_IO_INCOMPLETE: error = EINTR; break;
  83         case ERROR_LOCKED: error = EBUSY; break;
  84         case ERROR_LOCK_VIOLATION: error = EACCES; break;
  85         case ERROR_LOGON_FAILURE: error = EACCES; break;
  86         case ERROR_MAPPED_ALIGNMENT: error = EINVAL; break;
  87         case ERROR_META_EXPANSION_TOO_LONG: error = E2BIG; break;
  88         case ERROR_MORE_DATA: error = EPIPE; break;
  89         case ERROR_NEGATIVE_SEEK: error = ESPIPE; break;
  90         case ERROR_NOACCESS: error = EFAULT; break;
  91         case ERROR_NONE_MAPPED: error = EINVAL; break;
  92         case ERROR_NOT_ENOUGH_MEMORY: error = ENOMEM; break;
  93         case ERROR_NOT_READY: error = EAGAIN; break;
  94         case ERROR_NOT_SAME_DEVICE: error = EXDEV; break;
  95         case ERROR_NO_DATA: error = EPIPE; break;
  96         case ERROR_NO_MORE_SEARCH_HANDLES: error = EIO; break;
  97         case ERROR_NO_PROC_SLOTS: error = EAGAIN; break;
  98         case ERROR_NO_SUCH_PRIVILEGE: error = EACCES; break;
  99         case ERROR_OPEN_FAILED: error = EIO; break;
 100         case ERROR_OPEN_FILES: error = EBUSY; break;
 101         case ERROR_OPERATION_ABORTED: error = EINTR; break;
 102         case ERROR_OUTOFMEMORY: error = ENOMEM; break;
 103         case ERROR_PASSWORD_EXPIRED: error = EACCES; break;
 104         case ERROR_PATH_BUSY: error = EBUSY; break;
 105         case ERROR_PATH_NOT_FOUND: error = ENOENT; break;
 106         case ERROR_PIPE_BUSY: error = EBUSY; break;
 107         case ERROR_PIPE_CONNECTED: error = EPIPE; break;
 108         case ERROR_PIPE_LISTENING: error = EPIPE; break;
 109         case ERROR_PIPE_NOT_CONNECTED: error = EPIPE; break;
 110         case ERROR_PRIVILEGE_NOT_HELD: error = EACCES; break;
 111         case ERROR_READ_FAULT: error = EIO; break;
 112         case ERROR_SEEK: error = EIO; break;
 113         case ERROR_SEEK_ON_DEVICE: error = ESPIPE; break;
 114         case ERROR_SHARING_BUFFER_EXCEEDED: error = ENFILE; break;
 115         case ERROR_SHARING_VIOLATION: error = EACCES; break;
 116         case ERROR_STACK_OVERFLOW: error = ENOMEM; break;
 117         case ERROR_SWAPERROR: error = ENOENT; break;
 118         case ERROR_TOO_MANY_MODULES: error = EMFILE; break;
 119         case ERROR_TOO_MANY_OPEN_FILES: error = EMFILE; break;
 120         case ERROR_UNRECOGNIZED_MEDIA: error = ENXIO; break;
 121         case ERROR_UNRECOGNIZED_VOLUME: error = ENODEV; break;
 122         case ERROR_WAIT_NO_CHILDREN: error = ECHILD; break;
 123         case ERROR_WRITE_FAULT: error = EIO; break;
 124         case ERROR_WRITE_PROTECT: error = EROFS; break;
 125         }
 126         return error;
 127 }
 128
 129 static inline int is_file_in_use_error(DWORD errcode)
 130 {
 131         switch (errcode) {
 132         case ERROR_SHARING_VIOLATION:
 133         case ERROR_ACCESS_DENIED:
 134                 return 1;
 135         }
 136
 137         return 0;
 138 }
 139
 140 static int read_yes_no_answer(void)
 141 {
 142         char answer[1024];
 143
 144         if (fgets(answer, sizeof(answer), stdin)) {
 145                 size_t answer_len = strlen(answer);
 146                 int got_full_line = 0, c;
 147
 148                 /* remove the newline */
 149                 if (answer_len >= 2 && answer[answer_len-2] == '\r') {
 150                         answer[answer_len-2] = '\0';
 151                         got_full_line = 1;
 152                 } else if (answer_len >= 1 && answer[answer_len-1] == '\n') {
 153                         answer[answer_len-1] = '\0';
 154                         got_full_line = 1;
 155                 }
 156                 /* flush the buffer in case we did not get the full line */
 157                 if (!got_full_line)
 158                         while ((c = getchar()) != EOF && c != '\n')
 159                                 ;
 160         } else
 161                 /* we could not read, return the
 162                  * default answer which is no */
 163                 return 0;
 164
 165         if (tolower(answer[0]) == 'y' && !answer[1])
 166                 return 1;
 167         if (!strncasecmp(answer, "yes", sizeof(answer)))
 168                 return 1;
 169         if (tolower(answer[0]) == 'n' && !answer[1])
 170                 return 0;
 171         if (!strncasecmp(answer, "no", sizeof(answer)))
 172                 return 0;
 173
 174         /* did not find an answer we understand */
 175         return -1;
 176 }
 177
 178 static int ask_yes_no_if_possible(const char *format, ...)
 179 {
 180         char question[4096];
 181         const char *retry_hook[] = { NULL, NULL, NULL };
 182         va_list args;
 183
 184         va_start(args, format);
 185         vsnprintf(question, sizeof(question), format, args);
 186         va_end(args);
 187
 188         if ((retry_hook[0] = mingw_getenv("GIT_ASK_YESNO"))) {
 189                 retry_hook[1] = question;
 190                 return !run_command_v_opt(retry_hook, 0);
 191         }
 192
 193         if (!isatty(_fileno(stdin)) || !isatty(_fileno(stderr)))
 194                 return 0;
 195
 196         while (1) {
 197                 int answer;
 198                 fprintf(stderr, "%s (y/n) ", question);
 199
 200                 if ((answer = read_yes_no_answer()) >= 0)
 201                         return answer;
 202
 203                 fprintf(stderr, "Sorry, I did not understand your answer. "
 204                                 "Please type 'y' or 'n'\n");
 205         }
 206 }
 207
 208 /* Windows only */
 209 enum hide_dotfiles_type {
 210         HIDE_DOTFILES_FALSE = 0,
 211         HIDE_DOTFILES_TRUE,
 212         HIDE_DOTFILES_DOTGITONLY
 213 };
 214
 215 static enum hide_dotfiles_type hide_dotfiles = HIDE_DOTFILES_DOTGITONLY;
 216 static char *unset_environment_variables;
 217
 218 int mingw_core_config(const char *var, const char *value, void *cb)
 219 {
 220         if (!strcmp(var, "core.hidedotfiles")) {
 221                 if (value && !strcasecmp(value, "dotgitonly"))
 222                         hide_dotfiles = HIDE_DOTFILES_DOTGITONLY;
 223                 else
 224                         hide_dotfiles = git_config_bool(var, value);
 225                 return 0;
 226         }
 227
 228         if (!strcmp(var, "core.unsetenvvars")) {
 229                 free(unset_environment_variables);
 230                 unset_environment_variables = xstrdup(value);
 231                 return 0;
 232         }
 233
 234         return 0;
 235 }
 236
 237 /* Normalizes NT paths as returned by some low-level APIs. */
 238 static wchar_t *normalize_ntpath(wchar_t *wbuf)
 239 {
 240         int i;
 241         /* fix absolute path prefixes */
 242         if (wbuf[0] == '\\') {
 243                 /* strip NT namespace prefixes */
 244                 if (!wcsncmp(wbuf, L"\\??\\", 4) ||
 245                     !wcsncmp(wbuf, L"\\\\?\\", 4))
 246                         wbuf += 4;
 247                 else if (!wcsnicmp(wbuf, L"\\DosDevices\\", 12))
 248                         wbuf += 12;
 249                 /* replace remaining '...UNC\' with '\\' */
 250                 if (!wcsnicmp(wbuf, L"UNC\\", 4)) {
 251                         wbuf += 2;
 252                         *wbuf = '\\';
 253                 }
 254         }
 255         /* convert backslashes to slashes */
 256         for (i = 0; wbuf[i]; i++)
 257                 if (wbuf[i] == '\\')
 258                         wbuf[i] = '/';
 259         return wbuf;
 260 }
 261
 262 int mingw_unlink(const char *pathname)
 263 {
 264         int ret, tries = 0;
 265         wchar_t wpathname[MAX_PATH];
 266         if (xutftowcs_path(wpathname, pathname) < 0)
 267                 return -1;
 268
 269         /* read-only files cannot be removed */
 270         _wchmod(wpathname, 0666);
 271         while ((ret = _wunlink(wpathname)) == -1 && tries < ARRAY_SIZE(delay)) {
 272                 if (!is_file_in_use_error(GetLastError()))
 273                         break;
 274                 /*
 275                  * We assume that some other process had the source or
 276                  * destination file open at the wrong moment and retry.
 277                  * In order to give the other process a higher chance to
 278                  * complete its operation, we give up our time slice now.
 279                  * If we have to retry again, we do sleep a bit.
 280                  */
 281                 Sleep(delay[tries]);
 282                 tries++;
 283         }
 284         while (ret == -1 && is_file_in_use_error(GetLastError()) &&
 285                ask_yes_no_if_possible("Unlink of file '%s' failed. "
 286                         "Should I try again?", pathname))
 287                ret = _wunlink(wpathname);
 288         return ret;
 289 }
 290
 291 static int is_dir_empty(const wchar_t *wpath)
 292 {
 293         WIN32_FIND_DATAW findbuf;
 294         HANDLE handle;
 295         wchar_t wbuf[MAX_PATH + 2];
 296         wcscpy(wbuf, wpath);
 297         wcscat(wbuf, L"\\*");
 298         handle = FindFirstFileW(wbuf, &findbuf);
 299         if (handle == INVALID_HANDLE_VALUE)
 300                 return GetLastError() == ERROR_NO_MORE_FILES;
 301
 302         while (!wcscmp(findbuf.cFileName, L".") ||
 303                         !wcscmp(findbuf.cFileName, L".."))
 304                 if (!FindNextFileW(handle, &findbuf)) {
 305                         DWORD err = GetLastError();
 306                         FindClose(handle);
 307                         return err == ERROR_NO_MORE_FILES;
 308                 }
 309         FindClose(handle);
 310         return 0;
 311 }
 312
 313 int mingw_rmdir(const char *pathname)
 314 {
 315         int ret, tries = 0;
 316         wchar_t wpathname[MAX_PATH];
 317         if (xutftowcs_path(wpathname, pathname) < 0)
 318                 return -1;
 319
 320         while ((ret = _wrmdir(wpathname)) == -1 && tries < ARRAY_SIZE(delay)) {
 321                 if (!is_file_in_use_error(GetLastError()))
 322                         errno = err_win_to_posix(GetLastError());
 323                 if (errno != EACCES)
 324                         break;
 325                 if (!is_dir_empty(wpathname)) {
 326                         errno = ENOTEMPTY;
 327                         break;
 328                 }
 329                 /*
 330                  * We assume that some other process had the source or
 331                  * destination file open at the wrong moment and retry.
 332                  * In order to give the other process a higher chance to
 333                  * complete its operation, we give up our time slice now.
 334                  * If we have to retry again, we do sleep a bit.
 335                  */
 336                 Sleep(delay[tries]);
 337                 tries++;
 338         }
 339         while (ret == -1 && errno == EACCES && is_file_in_use_error(GetLastError()) &&
 340                ask_yes_no_if_possible("Deletion of directory '%s' failed. "
 341                         "Should I try again?", pathname))
 342                ret = _wrmdir(wpathname);
 343         return ret;
 344 }
 345
 346 static inline int needs_hiding(const char *path)
 347 {
 348         const char *basename;
 349
 350         if (hide_dotfiles == HIDE_DOTFILES_FALSE)
 351                 return 0;
 352
 353         /* We cannot use basename(), as it would remove trailing slashes */
 354         win32_skip_dos_drive_prefix((char **)&path);
 355         if (!*path)
 356                 return 0;
 357
 358         for (basename = path; *path; path++)
 359                 if (is_dir_sep(*path)) {
 360                         do {
 361                                 path++;
 362                         } while (is_dir_sep(*path));
 363                         /* ignore trailing slashes */
 364                         if (*path)
 365                                 basename = path;
 366                 }
 367
 368         if (hide_dotfiles == HIDE_DOTFILES_TRUE)
 369                 return *basename == '.';
 370
 371         assert(hide_dotfiles == HIDE_DOTFILES_DOTGITONLY);
 372         return !strncasecmp(".git", basename, 4) &&
 373                 (!basename[4] || is_dir_sep(basename[4]));
 374 }
 375
 376 static int set_hidden_flag(const wchar_t *path, int set)
 377 {
 378         DWORD original = GetFileAttributesW(path), modified;
 379         if (set)
 380                 modified = original | FILE_ATTRIBUTE_HIDDEN;
 381         else
 382                 modified = original & ~FILE_ATTRIBUTE_HIDDEN;
 383         if (original == modified || SetFileAttributesW(path, modified))
 384                 return 0;
 385         errno = err_win_to_posix(GetLastError());
 386         return -1;
 387 }
 388
 389 int mingw_mkdir(const char *path, int mode)
 390 {
 391         int ret;
 392         wchar_t wpath[MAX_PATH];
 393
 394         if (!is_valid_win32_path(path)) {
 395                 errno = EINVAL;
 396                 return -1;
 397         }
 398
 399         if (xutftowcs_path(wpath, path) < 0)
 400                 return -1;
 401         ret = _wmkdir(wpath);
 402         if (!ret && needs_hiding(path))
 403                 return set_hidden_flag(wpath, 1);
 404         return ret;
 405 }
 406
 407 /*
 408  * Calling CreateFile() using FILE_APPEND_DATA and without FILE_WRITE_DATA
 409  * is documented in [1] as opening a writable file handle in append mode.
 410  * (It is believed that) this is atomic since it is maintained by the
 411  * kernel unlike the O_APPEND flag which is racily maintained by the CRT.
 412  *
 413  * [1] https://docs.microsoft.com/en-us/windows/desktop/fileio/file-access-rights-constants
 414  *
 415  * This trick does not appear to work for named pipes.  Instead it creates
 416  * a named pipe client handle that cannot be written to.  Callers should
 417  * just use the regular _wopen() for them.  (And since client handle gets
 418  * bound to a unique server handle, it isn't really an issue.)
 419  */
 420 static int mingw_open_append(wchar_t const *wfilename, int oflags, ...)
 421 {
 422         HANDLE handle;
 423         int fd;
 424         DWORD create = (oflags & O_CREAT) ? OPEN_ALWAYS : OPEN_EXISTING;
 425
 426         /* only these flags are supported */
 427         if ((oflags & ~O_CREAT) != (O_WRONLY | O_APPEND))
 428                 return errno = ENOSYS, -1;
 429
 430         /*
 431          * FILE_SHARE_WRITE is required to permit child processes
 432          * to append to the file.
 433          */
 434         handle = CreateFileW(wfilename, FILE_APPEND_DATA,
 435                         FILE_SHARE_WRITE | FILE_SHARE_READ,
 436                         NULL, create, FILE_ATTRIBUTE_NORMAL, NULL);
 437         if (handle == INVALID_HANDLE_VALUE)
 438                 return errno = err_win_to_posix(GetLastError()), -1;
 439
 440         /*
 441          * No O_APPEND here, because the CRT uses it only to reset the
 442          * file pointer to EOF before each write(); but that is not
 443          * necessary (and may lead to races) for a file created with
 444          * FILE_APPEND_DATA.
 445          */
 446         fd = _open_osfhandle((intptr_t)handle, O_BINARY);
 447         if (fd < 0)
 448                 CloseHandle(handle);
 449         return fd;
 450 }
 451
 452 /*
 453  * Does the pathname map to the local named pipe filesystem?
 454  * That is, does it have a "//./pipe/" prefix?
 455  */
 456 static int is_local_named_pipe_path(const char *filename)
 457 {
 458         return (is_dir_sep(filename[0]) &&
 459                 is_dir_sep(filename[1]) &&
 460                 filename[2] == '.'  &&
 461                 is_dir_sep(filename[3]) &&
 462                 !strncasecmp(filename+4, "pipe", 4) &&
 463                 is_dir_sep(filename[8]) &&
 464                 filename[9]);
 465 }
 466
 467 int mingw_open (const char *filename, int oflags, ...)
 468 {
 469         typedef int (*open_fn_t)(wchar_t const *wfilename, int oflags, ...);
 470         va_list args;
 471         unsigned mode;
 472         int fd, create = (oflags & (O_CREAT | O_EXCL)) == (O_CREAT | O_EXCL);
 473         wchar_t wfilename[MAX_PATH];
 474         open_fn_t open_fn;
 475
 476         va_start(args, oflags);
 477         mode = va_arg(args, int);
 478         va_end(args);
 479
 480         if (!is_valid_win32_path(filename)) {
 481                 errno = create ? EINVAL : ENOENT;
 482                 return -1;
 483         }
 484
 485         if (filename && !strcmp(filename, "/dev/null"))
 486                 filename = "nul";
 487
 488         if ((oflags & O_APPEND) && !is_local_named_pipe_path(filename))
 489                 open_fn = mingw_open_append;
 490         else
 491                 open_fn = _wopen;
 492
 493         if (xutftowcs_path(wfilename, filename) < 0)
 494                 return -1;
 495         fd = open_fn(wfilename, oflags, mode);
 496
 497         if (fd < 0 && (oflags & O_ACCMODE) != O_RDONLY && errno == EACCES) {
 498                 DWORD attrs = GetFileAttributesW(wfilename);
 499                 if (attrs != INVALID_FILE_ATTRIBUTES && (attrs & FILE_ATTRIBUTE_DIRECTORY))
 500                         errno = EISDIR;
 501         }
 502         if ((oflags & O_CREAT) && needs_hiding(filename)) {
 503                 /*
 504                  * Internally, _wopen() uses the CreateFile() API which errors
 505                  * out with an ERROR_ACCESS_DENIED if CREATE_ALWAYS was
 506                  * specified and an already existing file's attributes do not
 507                  * match *exactly*. As there is no mode or flag we can set that
 508                  * would correspond to FILE_ATTRIBUTE_HIDDEN, let's just try
 509                  * again *without* the O_CREAT flag (that corresponds to the
 510                  * CREATE_ALWAYS flag of CreateFile()).
 511                  */
 512                 if (fd < 0 && errno == EACCES)
 513                         fd = open_fn(wfilename, oflags & ~O_CREAT, mode);
 514                 if (fd >= 0 && set_hidden_flag(wfilename, 1))
 515                         warning("could not mark '%s' as hidden.", filename);
 516         }
 517         return fd;
 518 }
 519
 520 static BOOL WINAPI ctrl_ignore(DWORD type)
 521 {
 522         return TRUE;
 523 }
 524
 525 #undef fgetc
 526 int mingw_fgetc(FILE *stream)
 527 {
 528         int ch;
 529         if (!isatty(_fileno(stream)))
 530                 return fgetc(stream);
 531
 532         SetConsoleCtrlHandler(ctrl_ignore, TRUE);
 533         while (1) {
 534                 ch = fgetc(stream);
 535                 if (ch != EOF || GetLastError() != ERROR_OPERATION_ABORTED)
 536                         break;
 537
 538                 /* Ctrl+C was pressed, simulate SIGINT and retry */
 539                 mingw_raise(SIGINT);
 540         }
 541         SetConsoleCtrlHandler(ctrl_ignore, FALSE);
 542         return ch;
 543 }
 544
 545 #undef fopen
 546 FILE *mingw_fopen (const char *filename, const char *otype)
 547 {
 548         int hide = needs_hiding(filename);
 549         FILE *file;
 550         wchar_t wfilename[MAX_PATH], wotype[4];
 551         if (!is_valid_win32_path(filename)) {
 552                 int create = otype && strchr(otype, 'w');
 553                 errno = create ? EINVAL : ENOENT;
 554                 return NULL;
 555         }
 556         if (filename && !strcmp(filename, "/dev/null"))
 557                 filename = "nul";
 558         if (xutftowcs_path(wfilename, filename) < 0 ||
 559                 xutftowcs(wotype, otype, ARRAY_SIZE(wotype)) < 0)
 560                 return NULL;
 561         if (hide && !access(filename, F_OK) && set_hidden_flag(wfilename, 0)) {
 562                 error("could not unhide %s", filename);
 563                 return NULL;
 564         }
 565         file = _wfopen(wfilename, wotype);
 566         if (!file && GetLastError() == ERROR_INVALID_NAME)
 567                 errno = ENOENT;
 568         if (file && hide && set_hidden_flag(wfilename, 1))
 569                 warning("could not mark '%s' as hidden.", filename);
 570         return file;
 571 }
 572
 573 FILE *mingw_freopen (const char *filename, const char *otype, FILE *stream)
 574 {
 575         int hide = needs_hiding(filename);
 576         FILE *file;
 577         wchar_t wfilename[MAX_PATH], wotype[4];
 578         if (!is_valid_win32_path(filename)) {
 579                 int create = otype && strchr(otype, 'w');
 580                 errno = create ? EINVAL : ENOENT;
 581                 return NULL;
 582         }
 583         if (filename && !strcmp(filename, "/dev/null"))
 584                 filename = "nul";
 585         if (xutftowcs_path(wfilename, filename) < 0 ||
 586                 xutftowcs(wotype, otype, ARRAY_SIZE(wotype)) < 0)
 587                 return NULL;
 588         if (hide && !access(filename, F_OK) && set_hidden_flag(wfilename, 0)) {
 589                 error("could not unhide %s", filename);
 590                 return NULL;
 591         }
 592         file = _wfreopen(wfilename, wotype, stream);
 593         if (file && hide && set_hidden_flag(wfilename, 1))
 594                 warning("could not mark '%s' as hidden.", filename);
 595         return file;
 596 }
 597
 598 #undef fflush
 599 int mingw_fflush(FILE *stream)
 600 {
 601         int ret = fflush(stream);
 602
 603         /*
 604          * write() is used behind the scenes of stdio output functions.
 605          * Since git code does not check for errors after each stdio write
 606          * operation, it can happen that write() is called by a later
 607          * stdio function even if an earlier write() call failed. In the
 608          * case of a pipe whose readable end was closed, only the first
 609          * call to write() reports EPIPE on Windows. Subsequent write()
 610          * calls report EINVAL. It is impossible to notice whether this
 611          * fflush invocation triggered such a case, therefore, we have to
 612          * catch all EINVAL errors whole-sale.
 613          */
 614         if (ret && errno == EINVAL)
 615                 errno = EPIPE;
 616
 617         return ret;
 618 }
 619
 620 #undef write
 621 ssize_t mingw_write(int fd, const void *buf, size_t len)
 622 {
 623         ssize_t result = write(fd, buf, len);
 624
 625         if (result < 0 && errno == EINVAL && buf) {
 626                 /* check if fd is a pipe */
 627                 HANDLE h = (HANDLE) _get_osfhandle(fd);
 628                 if (GetFileType(h) == FILE_TYPE_PIPE)
 629                         errno = EPIPE;
 630                 else
 631                         errno = EINVAL;
 632         }
 633
 634         return result;
 635 }
 636
 637 int mingw_access(const char *filename, int mode)
 638 {
 639         wchar_t wfilename[MAX_PATH];
 640         if (xutftowcs_path(wfilename, filename) < 0)
 641                 return -1;
 642         /* X_OK is not supported by the MSVCRT version */
 643         return _waccess(wfilename, mode & ~X_OK);
 644 }
 645
 646 int mingw_chdir(const char *dirname)
 647 {
 648         wchar_t wdirname[MAX_PATH];
 649         if (xutftowcs_path(wdirname, dirname) < 0)
 650                 return -1;
 651         return _wchdir(wdirname);
 652 }
 653
 654 int mingw_chmod(const char *filename, int mode)
 655 {
 656         wchar_t wfilename[MAX_PATH];
 657         if (xutftowcs_path(wfilename, filename) < 0)
 658                 return -1;
 659         return _wchmod(wfilename, mode);
 660 }
 661
 662 /*
 663  * The unit of FILETIME is 100-nanoseconds since January 1, 1601, UTC.
 664  * Returns the 100-nanoseconds ("hekto nanoseconds") since the epoch.
 665  */
 666 static inline long long filetime_to_hnsec(const FILETIME *ft)
 667 {
 668         long long winTime = ((long long)ft->dwHighDateTime << 32) + ft->dwLowDateTime;
 669         /* Windows to Unix Epoch conversion */
 670         return winTime - 116444736000000000LL;
 671 }
 672
 673 static inline void filetime_to_timespec(const FILETIME *ft, struct timespec *ts)
 674 {
 675         long long hnsec = filetime_to_hnsec(ft);
 676         ts->tv_sec = (time_t)(hnsec / 10000000);
 677         ts->tv_nsec = (hnsec % 10000000) * 100;
 678 }
 679
 680 /**
 681  * Verifies that safe_create_leading_directories() would succeed.
 682  */
 683 static int has_valid_directory_prefix(wchar_t *wfilename)
 684 {
 685         int n = wcslen(wfilename);
 686
 687         while (n > 0) {
 688                 wchar_t c = wfilename[--n];
 689                 DWORD attributes;
 690
 691                 if (!is_dir_sep(c))
 692                         continue;
 693
 694                 wfilename[n] = L'\0';
 695                 attributes = GetFileAttributesW(wfilename);
 696                 wfilename[n] = c;
 697                 if (attributes == FILE_ATTRIBUTE_DIRECTORY ||
 698                                 attributes == FILE_ATTRIBUTE_DEVICE)
 699                         return 1;
 700                 if (attributes == INVALID_FILE_ATTRIBUTES)
 701                         switch (GetLastError()) {
 702                         case ERROR_PATH_NOT_FOUND:
 703                                 continue;
 704                         case ERROR_FILE_NOT_FOUND:
 705                                 /* This implies parent directory exists. */
 706                                 return 1;
 707                         }
 708                 return 0;
 709         }
 710         return 1;
 711 }
 712
 713 /* We keep the do_lstat code in a separate function to avoid recursion.
 714  * When a path ends with a slash, the stat will fail with ENOENT. In
 715  * this case, we strip the trailing slashes and stat again.
 716  *
 717  * If follow is true then act like stat() and report on the link
 718  * target. Otherwise report on the link itself.
 719  */
 720 static int do_lstat(int follow, const char *file_name, struct stat *buf)
 721 {
 722         WIN32_FILE_ATTRIBUTE_DATA fdata;
 723         wchar_t wfilename[MAX_PATH];
 724         if (xutftowcs_path(wfilename, file_name) < 0)
 725                 return -1;
 726
 727         if (GetFileAttributesExW(wfilename, GetFileExInfoStandard, &fdata)) {
 728                 buf->st_ino = 0;
 729                 buf->st_gid = 0;
 730                 buf->st_uid = 0;
 731                 buf->st_nlink = 1;
 732                 buf->st_mode = file_attr_to_st_mode(fdata.dwFileAttributes);
 733                 buf->st_size = fdata.nFileSizeLow |
 734                         (((off_t)fdata.nFileSizeHigh)<<32);
 735                 buf->st_dev = buf->st_rdev = 0; /* not used by Git */
 736                 filetime_to_timespec(&(fdata.ftLastAccessTime), &(buf->st_atim));
 737                 filetime_to_timespec(&(fdata.ftLastWriteTime), &(buf->st_mtim));
 738                 filetime_to_timespec(&(fdata.ftCreationTime), &(buf->st_ctim));
 739                 if (fdata.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
 740                         WIN32_FIND_DATAW findbuf;
 741                         HANDLE handle = FindFirstFileW(wfilename, &findbuf);
 742                         if (handle != INVALID_HANDLE_VALUE) {
 743                                 if ((findbuf.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) &&
 744                                                 (findbuf.dwReserved0 == IO_REPARSE_TAG_SYMLINK)) {
 745                                         if (follow) {
 746                                                 char buffer[MAXIMUM_REPARSE_DATA_BUFFER_SIZE];
 747                                                 buf->st_size = readlink(file_name, buffer, MAXIMUM_REPARSE_DATA_BUFFER_SIZE);
 748                                         } else {
 749                                                 buf->st_mode = S_IFLNK;
 750                                         }
 751                                         buf->st_mode |= S_IREAD;
 752                                         if (!(findbuf.dwFileAttributes & FILE_ATTRIBUTE_READONLY))
 753                                                 buf->st_mode |= S_IWRITE;
 754                                 }
 755                                 FindClose(handle);
 756                         }
 757                 }
 758                 return 0;
 759         }
 760         switch (GetLastError()) {
 761         case ERROR_ACCESS_DENIED:
 762         case ERROR_SHARING_VIOLATION:
 763         case ERROR_LOCK_VIOLATION:
 764         case ERROR_SHARING_BUFFER_EXCEEDED:
 765                 errno = EACCES;
 766                 break;
 767         case ERROR_BUFFER_OVERFLOW:
 768                 errno = ENAMETOOLONG;
 769                 break;
 770         case ERROR_NOT_ENOUGH_MEMORY:
 771                 errno = ENOMEM;
 772                 break;
 773         case ERROR_PATH_NOT_FOUND:
 774                 if (!has_valid_directory_prefix(wfilename)) {
 775                         errno = ENOTDIR;
 776                         break;
 777                 }
 778                 /* fallthru */
 779         default:
 780                 errno = ENOENT;
 781                 break;
 782         }
 783         return -1;
 784 }
 785
 786 /* We provide our own lstat/fstat functions, since the provided
 787  * lstat/fstat functions are so slow. These stat functions are
 788  * tailored for Git's usage (read: fast), and are not meant to be
 789  * complete. Note that Git stat()s are redirected to mingw_lstat()
 790  * too, since Windows doesn't really handle symlinks that well.
 791  */
 792 static int do_stat_internal(int follow, const char *file_name, struct stat *buf)
 793 {
 794         int namelen;
 795         char alt_name[PATH_MAX];
 796
 797         if (!do_lstat(follow, file_name, buf))
 798                 return 0;
 799
 800         /* if file_name ended in a '/', Windows returned ENOENT;
 801          * try again without trailing slashes
 802          */
 803         if (errno != ENOENT)
 804                 return -1;
 805
 806         namelen = strlen(file_name);
 807         if (namelen && file_name[namelen-1] != '/')
 808                 return -1;
 809         while (namelen && file_name[namelen-1] == '/')
 810                 --namelen;
 811         if (!namelen || namelen >= PATH_MAX)
 812                 return -1;
 813
 814         memcpy(alt_name, file_name, namelen);
 815         alt_name[namelen] = 0;
 816         return do_lstat(follow, alt_name, buf);
 817 }
 818
 819 static int get_file_info_by_handle(HANDLE hnd, struct stat *buf)
 820 {
 821         BY_HANDLE_FILE_INFORMATION fdata;
 822
 823         if (!GetFileInformationByHandle(hnd, &fdata)) {
 824                 errno = err_win_to_posix(GetLastError());
 825                 return -1;
 826         }
 827
 828         buf->st_ino = 0;
 829         buf->st_gid = 0;
 830         buf->st_uid = 0;
 831         buf->st_nlink = 1;
 832         buf->st_mode = file_attr_to_st_mode(fdata.dwFileAttributes);
 833         buf->st_size = fdata.nFileSizeLow |
 834                 (((off_t)fdata.nFileSizeHigh)<<32);
 835         buf->st_dev = buf->st_rdev = 0; /* not used by Git */
 836         filetime_to_timespec(&(fdata.ftLastAccessTime), &(buf->st_atim));
 837         filetime_to_timespec(&(fdata.ftLastWriteTime), &(buf->st_mtim));
 838         filetime_to_timespec(&(fdata.ftCreationTime), &(buf->st_ctim));
 839         return 0;
 840 }
 841
 842 int mingw_lstat(const char *file_name, struct stat *buf)
 843 {
 844         return do_stat_internal(0, file_name, buf);
 845 }
 846 int mingw_stat(const char *file_name, struct stat *buf)
 847 {
 848         return do_stat_internal(1, file_name, buf);
 849 }
 850
 851 int mingw_fstat(int fd, struct stat *buf)
 852 {
 853         HANDLE fh = (HANDLE)_get_osfhandle(fd);
 854         DWORD avail, type = GetFileType(fh) & ~FILE_TYPE_REMOTE;
 855
 856         switch (type) {
 857         case FILE_TYPE_DISK:
 858                 return get_file_info_by_handle(fh, buf);
 859
 860         case FILE_TYPE_CHAR:
 861         case FILE_TYPE_PIPE:
 862                 /* initialize stat fields */
 863                 memset(buf, 0, sizeof(*buf));
 864                 buf->st_nlink = 1;
 865
 866                 if (type == FILE_TYPE_CHAR) {
 867                         buf->st_mode = _S_IFCHR;
 868                 } else {
 869                         buf->st_mode = _S_IFIFO;
 870                         if (PeekNamedPipe(fh, NULL, 0, NULL, &avail, NULL))
 871                                 buf->st_size = avail;
 872                 }
 873                 return 0;
 874
 875         default:
 876                 errno = EBADF;
 877                 return -1;
 878         }
 879 }
 880
 881 static inline void time_t_to_filetime(time_t t, FILETIME *ft)
 882 {
 883         long long winTime = t * 10000000LL + 116444736000000000LL;
 884         ft->dwLowDateTime = winTime;
 885         ft->dwHighDateTime = winTime >> 32;
 886 }
 887
 888 int mingw_utime (const char *file_name, const struct utimbuf *times)
 889 {
 890         FILETIME mft, aft;
 891         int fh, rc;
 892         DWORD attrs;
 893         wchar_t wfilename[MAX_PATH];
 894         if (xutftowcs_path(wfilename, file_name) < 0)
 895                 return -1;
 896
 897         /* must have write permission */
 898         attrs = GetFileAttributesW(wfilename);
 899         if (attrs != INVALID_FILE_ATTRIBUTES &&
 900             (attrs & FILE_ATTRIBUTE_READONLY)) {
 901                 /* ignore errors here; open() will report them */
 902                 SetFileAttributesW(wfilename, attrs & ~FILE_ATTRIBUTE_READONLY);
 903         }
 904
 905         if ((fh = _wopen(wfilename, O_RDWR | O_BINARY)) < 0) {
 906                 rc = -1;
 907                 goto revert_attrs;
 908         }
 909
 910         if (times) {
 911                 time_t_to_filetime(times->modtime, &mft);
 912                 time_t_to_filetime(times->actime, &aft);
 913         } else {
 914                 GetSystemTimeAsFileTime(&mft);
 915                 aft = mft;
 916         }
 917         if (!SetFileTime((HANDLE)_get_osfhandle(fh), NULL, &aft, &mft)) {
 918                 errno = EINVAL;
 919                 rc = -1;
 920         } else
 921                 rc = 0;
 922         close(fh);
 923
 924 revert_attrs:
 925         if (attrs != INVALID_FILE_ATTRIBUTES &&
 926             (attrs & FILE_ATTRIBUTE_READONLY)) {
 927                 /* ignore errors again */
 928                 SetFileAttributesW(wfilename, attrs);
 929         }
 930         return rc;
 931 }
 932
 933 #undef strftime
 934 size_t mingw_strftime(char *s, size_t max,
 935                       const char *format, const struct tm *tm)
 936 {
 937         size_t ret = strftime(s, max, format, tm);
 938
 939         if (!ret && errno == EINVAL)
 940                 die("invalid strftime format: '%s'", format);
 941         return ret;
 942 }
 943
 944 unsigned int sleep (unsigned int seconds)
 945 {
 946         Sleep(seconds*1000);
 947         return 0;
 948 }
 949
 950 char *mingw_mktemp(char *template)
 951 {
 952         wchar_t wtemplate[MAX_PATH];
 953         if (xutftowcs_path(wtemplate, template) < 0)
 954                 return NULL;
 955         if (!_wmktemp(wtemplate))
 956                 return NULL;
 957         if (xwcstoutf(template, wtemplate, strlen(template) + 1) < 0)
 958                 return NULL;
 959         return template;
 960 }
 961
 962 int mkstemp(char *template)
 963 {
 964         char *filename = mktemp(template);
 965         if (filename == NULL)
 966                 return -1;
 967         return open(filename, O_RDWR | O_CREAT, 0600);
 968 }
 969
 970 int gettimeofday(struct timeval *tv, void *tz)
 971 {
 972         FILETIME ft;
 973         long long hnsec;
 974
 975         GetSystemTimeAsFileTime(&ft);
 976         hnsec = filetime_to_hnsec(&ft);
 977         tv->tv_sec = hnsec / 10000000;
 978         tv->tv_usec = (hnsec % 10000000) / 10;
 979         return 0;
 980 }
 981
 982 int pipe(int filedes[2])
 983 {
 984         HANDLE h[2];
 985
 986         /* this creates non-inheritable handles */
 987         if (!CreatePipe(&h[0], &h[1], NULL, 8192)) {
 988                 errno = err_win_to_posix(GetLastError());
 989                 return -1;
 990         }
 991         filedes[0] = _open_osfhandle(HCAST(int, h[0]), O_NOINHERIT);
 992         if (filedes[0] < 0) {
 993                 CloseHandle(h[0]);
 994                 CloseHandle(h[1]);
 995                 return -1;
 996         }
 997         filedes[1] = _open_osfhandle(HCAST(int, h[1]), O_NOINHERIT);
 998         if (filedes[1] < 0) {
 999                 close(filedes[0]);
1000                 CloseHandle(h[1]);
1001                 return -1;
1002         }
1003         return 0;
1004 }
1005
1006 struct tm *gmtime_r(const time_t *timep, struct tm *result)
1007 {
1008         /* gmtime() in MSVCRT.DLL is thread-safe, but not reentrant */
1009         memcpy(result, gmtime(timep), sizeof(struct tm));
1010         return result;
1011 }
1012
1013 struct tm *localtime_r(const time_t *timep, struct tm *result)
1014 {
1015         /* localtime() in MSVCRT.DLL is thread-safe, but not reentrant */
1016         memcpy(result, localtime(timep), sizeof(struct tm));
1017         return result;
1018 }
1019
1020 char *mingw_getcwd(char *pointer, int len)
1021 {
1022         wchar_t cwd[MAX_PATH], wpointer[MAX_PATH];
1023         DWORD ret = GetCurrentDirectoryW(ARRAY_SIZE(cwd), cwd);
1024
1025         if (!ret || ret >= ARRAY_SIZE(cwd)) {
1026                 errno = ret ? ENAMETOOLONG : err_win_to_posix(GetLastError());
1027                 return NULL;
1028         }
1029         ret = GetLongPathNameW(cwd, wpointer, ARRAY_SIZE(wpointer));
1030         if (!ret && GetLastError() == ERROR_ACCESS_DENIED) {
1031                 HANDLE hnd = CreateFileW(cwd, 0,
1032                         FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL,
1033                         OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
1034                 if (hnd == INVALID_HANDLE_VALUE)
1035                         return NULL;
1036                 ret = GetFinalPathNameByHandleW(hnd, wpointer, ARRAY_SIZE(wpointer), 0);
1037                 CloseHandle(hnd);
1038                 if (!ret || ret >= ARRAY_SIZE(wpointer))
1039                         return NULL;
1040                 if (xwcstoutf(pointer, normalize_ntpath(wpointer), len) < 0)
1041                         return NULL;
1042                 return pointer;
1043         }
1044         if (!ret || ret >= ARRAY_SIZE(wpointer))
1045                 return NULL;
1046         if (xwcstoutf(pointer, wpointer, len) < 0)
1047                 return NULL;
1048         convert_slashes(pointer);
1049         return pointer;
1050 }
1051
1052 /*
1053  * See "Parsing C++ Command-Line Arguments" at Microsoft's Docs:
1054  * https://docs.microsoft.com/en-us/cpp/cpp/parsing-cpp-command-line-arguments
1055  */
1056 static const char *quote_arg_msvc(const char *arg)
1057 {
1058         /* count chars to quote */
1059         int len = 0, n = 0;
1060         int force_quotes = 0;
1061         char *q, *d;
1062         const char *p = arg;
1063         if (!*p) force_quotes = 1;
1064         while (*p) {
1065                 if (isspace(*p) || *p == '*' || *p == '?' || *p == '{' || *p == '\'')
1066                         force_quotes = 1;
1067                 else if (*p == '"')
1068                         n++;
1069                 else if (*p == '\\') {
1070                         int count = 0;
1071                         while (*p == '\\') {
1072                                 count++;
1073                                 p++;
1074                                 len++;
1075                         }
1076                         if (*p == '"' || !*p)
1077                                 n += count*2 + 1;
1078                         continue;
1079                 }
1080                 len++;
1081                 p++;
1082         }
1083         if (!force_quotes && n == 0)
1084                 return arg;
1085
1086         /* insert \ where necessary */
1087         d = q = xmalloc(st_add3(len, n, 3));
1088         *d++ = '"';
1089         while (*arg) {
1090                 if (*arg == '"')
1091                         *d++ = '\\';
1092                 else if (*arg == '\\') {
1093                         int count = 0;
1094                         while (*arg == '\\') {
1095                                 count++;
1096                                 *d++ = *arg++;
1097                         }
1098                         if (*arg == '"' || !*arg) {
1099                                 while (count-- > 0)
1100                                         *d++ = '\\';
1101                                 /* don't escape the surrounding end quote */
1102                                 if (!*arg)
1103                                         break;
1104                                 *d++ = '\\';
1105                         }
1106                 }
1107                 *d++ = *arg++;
1108         }
1109         *d++ = '"';
1110         *d++ = '\0';
1111         return q;
1112 }
1113
1114 #include "quote.h"
1115
1116 static const char *quote_arg_msys2(const char *arg)
1117 {
1118         struct strbuf buf = STRBUF_INIT;
1119         const char *p2 = arg, *p;
1120
1121         for (p = arg; *p; p++) {
1122                 int ws = isspace(*p);
1123                 if (!ws && *p != '\\' && *p != '"' && *p != '{' && *p != '\'' &&
1124                     *p != '?' && *p != '*' && *p != '~')
1125                         continue;
1126                 if (!buf.len)
1127                         strbuf_addch(&buf, '"');
1128                 if (p != p2)
1129                         strbuf_add(&buf, p2, p - p2);
1130                 if (*p == '\\' || *p == '"')
1131                         strbuf_addch(&buf, '\\');
1132                 p2 = p;
1133         }
1134
1135         if (p == arg)
1136                 strbuf_addch(&buf, '"');
1137         else if (!buf.len)
1138                 return arg;
1139         else
1140                 strbuf_add(&buf, p2, p - p2);
1141
1142         strbuf_addch(&buf, '"');
1143         return strbuf_detach(&buf, 0);
1144 }
1145
1146 static const char *parse_interpreter(const char *cmd)
1147 {
1148         static char buf[100];
1149         char *p, *opt;
1150         int n, fd;
1151
1152         /* don't even try a .exe */
1153         n = strlen(cmd);
1154         if (n >= 4 && !strcasecmp(cmd+n-4, ".exe"))
1155                 return NULL;
1156
1157         fd = open(cmd, O_RDONLY);
1158         if (fd < 0)
1159                 return NULL;
1160         n = read(fd, buf, sizeof(buf)-1);
1161         close(fd);
1162         if (n < 4)      /* at least '#!/x' and not error */
1163                 return NULL;
1164
1165         if (buf[0] != '#' || buf[1] != '!')
1166                 return NULL;
1167         buf[n] = '\0';
1168         p = buf + strcspn(buf, "\r\n");
1169         if (!*p)
1170                 return NULL;
1171
1172         *p = '\0';
1173         if (!(p = strrchr(buf+2, '/')) && !(p = strrchr(buf+2, '\\')))
1174                 return NULL;
1175         /* strip options */
1176         if ((opt = strchr(p+1, ' ')))
1177                 *opt = '\0';
1178         return p+1;
1179 }
1180
1181 /*
1182  * exe_only means that we only want to detect .exe files, but not scripts
1183  * (which do not have an extension)
1184  */
1185 static char *lookup_prog(const char *dir, int dirlen, const char *cmd,
1186                          int isexe, int exe_only)
1187 {
1188         char path[MAX_PATH];
1189         snprintf(path, sizeof(path), "%.*s\\%s.exe", dirlen, dir, cmd);
1190
1191         if (!isexe && access(path, F_OK) == 0)
1192                 return xstrdup(path);
1193         path[strlen(path)-4] = '\0';
1194         if ((!exe_only || isexe) && access(path, F_OK) == 0)
1195                 if (!(GetFileAttributes(path) & FILE_ATTRIBUTE_DIRECTORY))
1196                         return xstrdup(path);
1197         return NULL;
1198 }
1199
1200 /*
1201  * Determines the absolute path of cmd using the split path in path.
1202  * If cmd contains a slash or backslash, no lookup is performed.
1203  */
1204 static char *path_lookup(const char *cmd, int exe_only)
1205 {
1206         const char *path;
1207         char *prog = NULL;
1208         int len = strlen(cmd);
1209         int isexe = len >= 4 && !strcasecmp(cmd+len-4, ".exe");
1210
1211         if (strchr(cmd, '/') || strchr(cmd, '\\'))
1212                 return xstrdup(cmd);
1213
1214         path = mingw_getenv("PATH");
1215         if (!path)
1216                 return NULL;
1217
1218         while (!prog) {
1219                 const char *sep = strchrnul(path, ';');
1220                 int dirlen = sep - path;
1221                 if (dirlen)
1222                         prog = lookup_prog(path, dirlen, cmd, isexe, exe_only);
1223                 if (!*sep)
1224                         break;
1225                 path = sep + 1;
1226         }
1227
1228         return prog;
1229 }
1230
1231 static const wchar_t *wcschrnul(const wchar_t *s, wchar_t c)
1232 {
1233         while (*s && *s != c)
1234                 s++;
1235         return s;
1236 }
1237
1238 /* Compare only keys */
1239 static int wenvcmp(const void *a, const void *b)
1240 {
1241         wchar_t *p = *(wchar_t **)a, *q = *(wchar_t **)b;
1242         size_t p_len, q_len;
1243
1244         /* Find the keys */
1245         p_len = wcschrnul(p, L'=') - p;
1246         q_len = wcschrnul(q, L'=') - q;
1247
1248         /* If the length differs, include the shorter key's NUL */
1249         if (p_len < q_len)
1250                 p_len++;
1251         else if (p_len > q_len)
1252                 p_len = q_len + 1;
1253
1254         return _wcsnicmp(p, q, p_len);
1255 }
1256
1257 /* We need a stable sort to convert the environment between UTF-16 <-> UTF-8 */
1258 #ifndef INTERNAL_QSORT
1259 #include "qsort.c"
1260 #endif
1261
1262 /*
1263  * Build an environment block combining the inherited environment
1264  * merged with the given list of settings.
1265  *
1266  * Values of the form "KEY=VALUE" in deltaenv override inherited values.
1267  * Values of the form "KEY" in deltaenv delete inherited values.
1268  *
1269  * Multiple entries in deltaenv for the same key are explicitly allowed.
1270  *
1271  * We return a contiguous block of UNICODE strings with a final trailing
1272  * zero word.
1273  */
1274 static wchar_t *make_environment_block(char **deltaenv)
1275 {
1276         wchar_t *wenv = GetEnvironmentStringsW(), *wdeltaenv, *result, *p;
1277         size_t wlen, s, delta_size, size;
1278
1279         wchar_t **array = NULL;
1280         size_t alloc = 0, nr = 0, i;
1281
1282         size = 1; /* for extra NUL at the end */
1283
1284         /* If there is no deltaenv to apply, simply return a copy. */
1285         if (!deltaenv || !*deltaenv) {
1286                 for (p = wenv; p && *p; ) {
1287                         size_t s = wcslen(p) + 1;
1288                         size += s;
1289                         p += s;
1290                 }
1291
1292                 ALLOC_ARRAY(result, size);
1293                 memcpy(result, wenv, size * sizeof(*wenv));
1294                 FreeEnvironmentStringsW(wenv);
1295                 return result;
1296         }
1297
1298         /*
1299          * If there is a deltaenv, let's accumulate all keys into `array`,
1300          * sort them using the stable git_qsort() and then copy, skipping
1301          * duplicate keys
1302          */
1303         for (p = wenv; p && *p; ) {
1304                 ALLOC_GROW(array, nr + 1, alloc);
1305                 s = wcslen(p) + 1;
1306                 array[nr++] = p;
1307                 p += s;
1308                 size += s;
1309         }
1310
1311         /* (over-)assess size needed for wchar version of deltaenv */
1312         for (delta_size = 0, i = 0; deltaenv[i]; i++)
1313                 delta_size += strlen(deltaenv[i]) * 2 + 1;
1314         ALLOC_ARRAY(wdeltaenv, delta_size);
1315
1316         /* convert the deltaenv, appending to array */
1317         for (i = 0, p = wdeltaenv; deltaenv[i]; i++) {
1318                 ALLOC_GROW(array, nr + 1, alloc);
1319                 wlen = xutftowcs(p, deltaenv[i], wdeltaenv + delta_size - p);
1320                 array[nr++] = p;
1321                 p += wlen + 1;
1322         }
1323
1324         git_qsort(array, nr, sizeof(*array), wenvcmp);
1325         ALLOC_ARRAY(result, size + delta_size);
1326
1327         for (p = result, i = 0; i < nr; i++) {
1328                 /* Skip any duplicate keys; last one wins */
1329                 while (i + 1 < nr && !wenvcmp(array + i, array + i + 1))
1330                        i++;
1331
1332                 /* Skip "to delete" entry */
1333                 if (!wcschr(array[i], L'='))
1334                         continue;
1335
1336                 size = wcslen(array[i]) + 1;
1337                 memcpy(p, array[i], size * sizeof(*p));
1338                 p += size;
1339         }
1340         *p = L'\0';
1341
1342         free(array);
1343         free(wdeltaenv);
1344         FreeEnvironmentStringsW(wenv);
1345         return result;
1346 }
1347
1348 static void do_unset_environment_variables(void)
1349 {
1350         static int done;
1351         char *p = unset_environment_variables;
1352
1353         if (done || !p)
1354                 return;
1355         done = 1;
1356
1357         for (;;) {
1358                 char *comma = strchr(p, ',');
1359
1360                 if (comma)
1361                         *comma = '\0';
1362                 unsetenv(p);
1363                 if (!comma)
1364                         break;
1365                 p = comma + 1;
1366         }
1367 }
1368
1369 struct pinfo_t {
1370         struct pinfo_t *next;
1371         pid_t pid;
1372         HANDLE proc;
1373 };
1374 static struct pinfo_t *pinfo = NULL;
1375 CRITICAL_SECTION pinfo_cs;
1376
1377 /* Used to match and chomp off path components */
1378 static inline int match_last_path_component(const char *path, size_t *len,
1379                                             const char *component)
1380 {
1381         size_t component_len = strlen(component);
1382         if (*len < component_len + 1 ||
1383             !is_dir_sep(path[*len - component_len - 1]) ||
1384             fspathncmp(path + *len - component_len, component, component_len))
1385                 return 0;
1386         *len -= component_len + 1;
1387         /* chomp off repeated dir separators */
1388         while (*len > 0 && is_dir_sep(path[*len - 1]))
1389                 (*len)--;
1390         return 1;
1391 }
1392
1393 static int is_msys2_sh(const char *cmd)
1394 {
1395         if (!cmd)
1396                 return 0;
1397
1398         if (!strcmp(cmd, "sh")) {
1399                 static int ret = -1;
1400                 char *p;
1401
1402                 if (ret >= 0)
1403                         return ret;
1404
1405                 p = path_lookup(cmd, 0);
1406                 if (!p)
1407                         ret = 0;
1408                 else {
1409                         size_t len = strlen(p);
1410
1411                         ret = match_last_path_component(p, &len, "sh.exe") &&
1412                                 match_last_path_component(p, &len, "bin") &&
1413                                 match_last_path_component(p, &len, "usr");
1414                         free(p);
1415                 }
1416                 return ret;
1417         }
1418
1419         if (ends_with(cmd, "\\sh.exe")) {
1420                 static char *sh;
1421
1422                 if (!sh)
1423                         sh = path_lookup("sh", 0);
1424
1425                 return !fspathcmp(cmd, sh);
1426         }
1427
1428         return 0;
1429 }
1430
1431 static pid_t mingw_spawnve_fd(const char *cmd, const char **argv, char **deltaenv,
1432                               const char *dir,
1433                               int prepend_cmd, int fhin, int fhout, int fherr)
1434 {
1435         STARTUPINFOW si;
1436         PROCESS_INFORMATION pi;
1437         struct strbuf args;
1438         wchar_t wcmd[MAX_PATH], wdir[MAX_PATH], *wargs, *wenvblk = NULL;
1439         unsigned flags = CREATE_UNICODE_ENVIRONMENT;
1440         BOOL ret;
1441         HANDLE cons;
1442         const char *(*quote_arg)(const char *arg) =
1443                 is_msys2_sh(cmd ? cmd : *argv) ?
1444                 quote_arg_msys2 : quote_arg_msvc;
1445
1446         do_unset_environment_variables();
1447
1448         /* Determine whether or not we are associated to a console */
1449         cons = CreateFile("CONOUT$", GENERIC_WRITE,
1450                         FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
1451                         FILE_ATTRIBUTE_NORMAL, NULL);
1452         if (cons == INVALID_HANDLE_VALUE) {
1453                 /* There is no console associated with this process.
1454                  * Since the child is a console process, Windows
1455                  * would normally create a console window. But
1456                  * since we'll be redirecting std streams, we do
1457                  * not need the console.
1458                  * It is necessary to use DETACHED_PROCESS
1459                  * instead of CREATE_NO_WINDOW to make ssh
1460                  * recognize that it has no console.
1461                  */
1462                 flags |= DETACHED_PROCESS;
1463         } else {
1464                 /* There is already a console. If we specified
1465                  * DETACHED_PROCESS here, too, Windows would
1466                  * disassociate the child from the console.
1467                  * The same is true for CREATE_NO_WINDOW.
1468                  * Go figure!
1469                  */
1470                 CloseHandle(cons);
1471         }
1472         memset(&si, 0, sizeof(si));
1473         si.cb = sizeof(si);
1474         si.dwFlags = STARTF_USESTDHANDLES;
1475         si.hStdInput = winansi_get_osfhandle(fhin);
1476         si.hStdOutput = winansi_get_osfhandle(fhout);
1477         si.hStdError = winansi_get_osfhandle(fherr);
1478
1479         if (xutftowcs_path(wcmd, cmd) < 0)
1480                 return -1;
1481         if (dir && xutftowcs_path(wdir, dir) < 0)
1482                 return -1;
1483
1484         /* concatenate argv, quoting args as we go */
1485         strbuf_init(&args, 0);
1486         if (prepend_cmd) {
1487                 char *quoted = (char *)quote_arg(cmd);
1488                 strbuf_addstr(&args, quoted);
1489                 if (quoted != cmd)
1490                         free(quoted);
1491         }
1492         for (; *argv; argv++) {
1493                 char *quoted = (char *)quote_arg(*argv);
1494                 if (*args.buf)
1495                         strbuf_addch(&args, ' ');
1496                 strbuf_addstr(&args, quoted);
1497                 if (quoted != *argv)
1498                         free(quoted);
1499         }
1500
1501         ALLOC_ARRAY(wargs, st_add(st_mult(2, args.len), 1));
1502         xutftowcs(wargs, args.buf, 2 * args.len + 1);
1503         strbuf_release(&args);
1504
1505         wenvblk = make_environment_block(deltaenv);
1506
1507         memset(&pi, 0, sizeof(pi));
1508         ret = CreateProcessW(wcmd, wargs, NULL, NULL, TRUE, flags,
1509                 wenvblk, dir ? wdir : NULL, &si, &pi);
1510
1511         free(wenvblk);
1512         free(wargs);
1513
1514         if (!ret) {
1515                 errno = ENOENT;
1516                 return -1;
1517         }
1518         CloseHandle(pi.hThread);
1519
1520         /*
1521          * The process ID is the human-readable identifier of the process
1522          * that we want to present in log and error messages. The handle
1523          * is not useful for this purpose. But we cannot close it, either,
1524          * because it is not possible to turn a process ID into a process
1525          * handle after the process terminated.
1526          * Keep the handle in a list for waitpid.
1527          */
1528         EnterCriticalSection(&pinfo_cs);
1529         {
1530                 struct pinfo_t *info = xmalloc(sizeof(struct pinfo_t));
1531                 info->pid = pi.dwProcessId;
1532                 info->proc = pi.hProcess;
1533                 info->next = pinfo;
1534                 pinfo = info;
1535         }
1536         LeaveCriticalSection(&pinfo_cs);
1537
1538         return (pid_t)pi.dwProcessId;
1539 }
1540
1541 static pid_t mingw_spawnv(const char *cmd, const char **argv, int prepend_cmd)
1542 {
1543         return mingw_spawnve_fd(cmd, argv, NULL, NULL, prepend_cmd, 0, 1, 2);
1544 }
1545
1546 pid_t mingw_spawnvpe(const char *cmd, const char **argv, char **deltaenv,
1547                      const char *dir,
1548                      int fhin, int fhout, int fherr)
1549 {
1550         pid_t pid;
1551         char *prog = path_lookup(cmd, 0);
1552
1553         if (!prog) {
1554                 errno = ENOENT;
1555                 pid = -1;
1556         }
1557         else {
1558                 const char *interpr = parse_interpreter(prog);
1559
1560                 if (interpr) {
1561                         const char *argv0 = argv[0];
1562                         char *iprog = path_lookup(interpr, 1);
1563                         argv[0] = prog;
1564                         if (!iprog) {
1565                                 errno = ENOENT;
1566                                 pid = -1;
1567                         }
1568                         else {
1569                                 pid = mingw_spawnve_fd(iprog, argv, deltaenv, dir, 1,
1570                                                        fhin, fhout, fherr);
1571                                 free(iprog);
1572                         }
1573                         argv[0] = argv0;
1574                 }
1575                 else
1576                         pid = mingw_spawnve_fd(prog, argv, deltaenv, dir, 0,
1577                                                fhin, fhout, fherr);
1578                 free(prog);
1579         }
1580         return pid;
1581 }
1582
1583 static int try_shell_exec(const char *cmd, char *const *argv)
1584 {
1585         const char *interpr = parse_interpreter(cmd);
1586         char *prog;
1587         int pid = 0;
1588
1589         if (!interpr)
1590                 return 0;
1591         prog = path_lookup(interpr, 1);
1592         if (prog) {
1593                 int argc = 0;
1594                 const char **argv2;
1595                 while (argv[argc]) argc++;
1596                 ALLOC_ARRAY(argv2, argc + 1);
1597                 argv2[0] = (char *)cmd; /* full path to the script file */
1598                 memcpy(&argv2[1], &argv[1], sizeof(*argv) * argc);
1599                 pid = mingw_spawnv(prog, argv2, 1);
1600                 if (pid >= 0) {
1601                         int status;
1602                         if (waitpid(pid, &status, 0) < 0)
1603                                 status = 255;
1604                         exit(status);
1605                 }
1606                 pid = 1;        /* indicate that we tried but failed */
1607                 free(prog);
1608                 free(argv2);
1609         }
1610         return pid;
1611 }
1612
1613 int mingw_execv(const char *cmd, char *const *argv)
1614 {
1615         /* check if git_command is a shell script */
1616         if (!try_shell_exec(cmd, argv)) {
1617                 int pid, status;
1618
1619                 pid = mingw_spawnv(cmd, (const char **)argv, 0);
1620                 if (pid < 0)
1621                         return -1;
1622                 if (waitpid(pid, &status, 0) < 0)
1623                         status = 255;
1624                 exit(status);
1625         }
1626         return -1;
1627 }
1628
1629 int mingw_execvp(const char *cmd, char *const *argv)
1630 {
1631         char *prog = path_lookup(cmd, 0);
1632
1633         if (prog) {
1634                 mingw_execv(prog, argv);
1635                 free(prog);
1636         } else
1637                 errno = ENOENT;
1638
1639         return -1;
1640 }
1641
1642 int mingw_kill(pid_t pid, int sig)
1643 {
1644         if (pid > 0 && sig == SIGTERM) {
1645                 HANDLE h = OpenProcess(PROCESS_TERMINATE, FALSE, pid);
1646
1647                 if (TerminateProcess(h, -1)) {
1648                         CloseHandle(h);
1649                         return 0;
1650                 }
1651
1652                 errno = err_win_to_posix(GetLastError());
1653                 CloseHandle(h);
1654                 return -1;
1655         } else if (pid > 0 && sig == 0) {
1656                 HANDLE h = OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, pid);
1657                 if (h) {
1658                         CloseHandle(h);
1659                         return 0;
1660                 }
1661         }
1662
1663         errno = EINVAL;
1664         return -1;
1665 }
1666
1667 /*
1668  * UTF-8 versions of getenv(), putenv() and unsetenv().
1669  * Internally, they use the CRT's stock UNICODE routines
1670  * to avoid data loss.
1671  */
1672 char *mingw_getenv(const char *name)
1673 {
1674 #define GETENV_MAX_RETAIN 64
1675         static char *values[GETENV_MAX_RETAIN];
1676         static int value_counter;
1677         int len_key, len_value;
1678         wchar_t *w_key;
1679         char *value;
1680         wchar_t w_value[32768];
1681
1682         if (!name || !*name)
1683                 return NULL;
1684
1685         len_key = strlen(name) + 1;
1686         /* We cannot use xcalloc() here because that uses getenv() itself */
1687         w_key = calloc(len_key, sizeof(wchar_t));
1688         if (!w_key)
1689                 die("Out of memory, (tried to allocate %u wchar_t's)", len_key);
1690         xutftowcs(w_key, name, len_key);
1691         len_value = GetEnvironmentVariableW(w_key, w_value, ARRAY_SIZE(w_value));
1692         if (!len_value && GetLastError() == ERROR_ENVVAR_NOT_FOUND) {
1693                 free(w_key);
1694                 return NULL;
1695         }
1696         free(w_key);
1697
1698         len_value = len_value * 3 + 1;
1699         /* We cannot use xcalloc() here because that uses getenv() itself */
1700         value = calloc(len_value, sizeof(char));
1701         if (!value)
1702                 die("Out of memory, (tried to allocate %u bytes)", len_value);
1703         xwcstoutf(value, w_value, len_value);
1704
1705         /*
1706          * We return `value` which is an allocated value and the caller is NOT
1707          * expecting to have to free it, so we keep a round-robin array,
1708          * invalidating the buffer after GETENV_MAX_RETAIN getenv() calls.
1709          */
1710         free(values[value_counter]);
1711         values[value_counter++] = value;
1712         if (value_counter >= ARRAY_SIZE(values))
1713                 value_counter = 0;
1714
1715         return value;
1716 }
1717
1718 int mingw_putenv(const char *namevalue)
1719 {
1720         int size;
1721         wchar_t *wide, *equal;
1722         BOOL result;
1723
1724         if (!namevalue || !*namevalue)
1725                 return 0;
1726
1727         size = strlen(namevalue) * 2 + 1;
1728         wide = calloc(size, sizeof(wchar_t));
1729         if (!wide)
1730                 die("Out of memory, (tried to allocate %u wchar_t's)", size);
1731         xutftowcs(wide, namevalue, size);
1732         equal = wcschr(wide, L'=');
1733         if (!equal)
1734                 result = SetEnvironmentVariableW(wide, NULL);
1735         else {
1736                 *equal = L'\0';
1737                 result = SetEnvironmentVariableW(wide, equal + 1);
1738         }
1739         free(wide);
1740
1741         if (!result)
1742                 errno = err_win_to_posix(GetLastError());
1743
1744         return result ? 0 : -1;
1745 }
1746
1747 /*
1748  * Note, this isn't a complete replacement for getaddrinfo. It assumes
1749  * that service contains a numerical port, or that it is null. It
1750  * does a simple search using gethostbyname, and returns one IPv4 host
1751  * if one was found.
1752  */
1753 static int WSAAPI getaddrinfo_stub(const char *node, const char *service,
1754                                    const struct addrinfo *hints,
1755                                    struct addrinfo **res)
1756 {
1757         struct hostent *h = NULL;
1758         struct addrinfo *ai;
1759         struct sockaddr_in *sin;
1760
1761         if (node) {
1762                 h = gethostbyname(node);
1763                 if (!h)
1764                         return WSAGetLastError();
1765         }
1766
1767         ai = xmalloc(sizeof(struct addrinfo));
1768         *res = ai;
1769         ai->ai_flags = 0;
1770         ai->ai_family = AF_INET;
1771         ai->ai_socktype = hints ? hints->ai_socktype : 0;
1772         switch (ai->ai_socktype) {
1773         case SOCK_STREAM:
1774                 ai->ai_protocol = IPPROTO_TCP;
1775                 break;
1776         case SOCK_DGRAM:
1777                 ai->ai_protocol = IPPROTO_UDP;
1778                 break;
1779         default:
1780                 ai->ai_protocol = 0;
1781                 break;
1782         }
1783         ai->ai_addrlen = sizeof(struct sockaddr_in);
1784         if (hints && (hints->ai_flags & AI_CANONNAME))
1785                 ai->ai_canonname = h ? xstrdup(h->h_name) : NULL;
1786         else
1787                 ai->ai_canonname = NULL;
1788
1789         sin = xcalloc(1, ai->ai_addrlen);
1790         sin->sin_family = AF_INET;
1791         /* Note: getaddrinfo is supposed to allow service to be a string,
1792          * which should be looked up using getservbyname. This is
1793          * currently not implemented */
1794         if (service)
1795                 sin->sin_port = htons(atoi(service));
1796         if (h)
1797                 sin->sin_addr = *(struct in_addr *)h->h_addr;
1798         else if (hints && (hints->ai_flags & AI_PASSIVE))
1799                 sin->sin_addr.s_addr = INADDR_ANY;
1800         else
1801                 sin->sin_addr.s_addr = INADDR_LOOPBACK;
1802         ai->ai_addr = (struct sockaddr *)sin;
1803         ai->ai_next = NULL;
1804         return 0;
1805 }
1806
1807 static void WSAAPI freeaddrinfo_stub(struct addrinfo *res)
1808 {
1809         free(res->ai_canonname);
1810         free(res->ai_addr);
1811         free(res);
1812 }
1813
1814 static int WSAAPI getnameinfo_stub(const struct sockaddr *sa, socklen_t salen,
1815                                    char *host, DWORD hostlen,
1816                                    char *serv, DWORD servlen, int flags)
1817 {
1818         const struct sockaddr_in *sin = (const struct sockaddr_in *)sa;
1819         if (sa->sa_family != AF_INET)
1820                 return EAI_FAMILY;
1821         if (!host && !serv)
1822                 return EAI_NONAME;
1823
1824         if (host && hostlen > 0) {
1825                 struct hostent *ent = NULL;
1826                 if (!(flags & NI_NUMERICHOST))
1827                         ent = gethostbyaddr((const char *)&sin->sin_addr,
1828                                             sizeof(sin->sin_addr), AF_INET);
1829
1830                 if (ent)
1831                         snprintf(host, hostlen, "%s", ent->h_name);
1832                 else if (flags & NI_NAMEREQD)
1833                         return EAI_NONAME;
1834                 else
1835                         snprintf(host, hostlen, "%s", inet_ntoa(sin->sin_addr));
1836         }
1837
1838         if (serv && servlen > 0) {
1839                 struct servent *ent = NULL;
1840                 if (!(flags & NI_NUMERICSERV))
1841                         ent = getservbyport(sin->sin_port,
1842                                             flags & NI_DGRAM ? "udp" : "tcp");
1843
1844                 if (ent)
1845                         snprintf(serv, servlen, "%s", ent->s_name);
1846                 else
1847                         snprintf(serv, servlen, "%d", ntohs(sin->sin_port));
1848         }
1849
1850         return 0;
1851 }
1852
1853 static HMODULE ipv6_dll = NULL;
1854 static void (WSAAPI *ipv6_freeaddrinfo)(struct addrinfo *res);
1855 static int (WSAAPI *ipv6_getaddrinfo)(const char *node, const char *service,
1856                                       const struct addrinfo *hints,
1857                                       struct addrinfo **res);
1858 static int (WSAAPI *ipv6_getnameinfo)(const struct sockaddr *sa, socklen_t salen,
1859                                       char *host, DWORD hostlen,
1860                                       char *serv, DWORD servlen, int flags);
1861 /*
1862  * gai_strerror is an inline function in the ws2tcpip.h header, so we
1863  * don't need to try to load that one dynamically.
1864  */
1865
1866 static void socket_cleanup(void)
1867 {
1868         WSACleanup();
1869         if (ipv6_dll)
1870                 FreeLibrary(ipv6_dll);
1871         ipv6_dll = NULL;
1872         ipv6_freeaddrinfo = freeaddrinfo_stub;
1873         ipv6_getaddrinfo = getaddrinfo_stub;
1874         ipv6_getnameinfo = getnameinfo_stub;
1875 }
1876
1877 static void ensure_socket_initialization(void)
1878 {
1879         WSADATA wsa;
1880         static int initialized = 0;
1881         const char *libraries[] = { "ws2_32.dll", "wship6.dll", NULL };
1882         const char **name;
1883
1884         if (initialized)
1885                 return;
1886
1887         if (WSAStartup(MAKEWORD(2,2), &wsa))
1888                 die("unable to initialize winsock subsystem, error %d",
1889                         WSAGetLastError());
1890
1891         for (name = libraries; *name; name++) {
1892                 ipv6_dll = LoadLibraryExA(*name, NULL,
1893                                           LOAD_LIBRARY_SEARCH_SYSTEM32);
1894                 if (!ipv6_dll)
1895                         continue;
1896
1897                 ipv6_freeaddrinfo = (void (WSAAPI *)(struct addrinfo *))
1898                         GetProcAddress(ipv6_dll, "freeaddrinfo");
1899                 ipv6_getaddrinfo = (int (WSAAPI *)(const char *, const char *,
1900                                                    const struct addrinfo *,
1901                                                    struct addrinfo **))
1902                         GetProcAddress(ipv6_dll, "getaddrinfo");
1903                 ipv6_getnameinfo = (int (WSAAPI *)(const struct sockaddr *,
1904                                                    socklen_t, char *, DWORD,
1905                                                    char *, DWORD, int))
1906                         GetProcAddress(ipv6_dll, "getnameinfo");
1907                 if (!ipv6_freeaddrinfo || !ipv6_getaddrinfo || !ipv6_getnameinfo) {
1908                         FreeLibrary(ipv6_dll);
1909                         ipv6_dll = NULL;
1910                 } else
1911                         break;
1912         }
1913         if (!ipv6_freeaddrinfo || !ipv6_getaddrinfo || !ipv6_getnameinfo) {
1914                 ipv6_freeaddrinfo = freeaddrinfo_stub;
1915                 ipv6_getaddrinfo = getaddrinfo_stub;
1916                 ipv6_getnameinfo = getnameinfo_stub;
1917         }
1918
1919         atexit(socket_cleanup);
1920         initialized = 1;
1921 }
1922
1923 #undef gethostname
1924 int mingw_gethostname(char *name, int namelen)
1925 {
1926     ensure_socket_initialization();
1927     return gethostname(name, namelen);
1928 }
1929
1930 #undef gethostbyname
1931 struct hostent *mingw_gethostbyname(const char *host)
1932 {
1933         ensure_socket_initialization();
1934         return gethostbyname(host);
1935 }
1936
1937 void mingw_freeaddrinfo(struct addrinfo *res)
1938 {
1939         ipv6_freeaddrinfo(res);
1940 }
1941
1942 int mingw_getaddrinfo(const char *node, const char *service,
1943                       const struct addrinfo *hints, struct addrinfo **res)
1944 {
1945         ensure_socket_initialization();
1946         return ipv6_getaddrinfo(node, service, hints, res);
1947 }
1948
1949 int mingw_getnameinfo(const struct sockaddr *sa, socklen_t salen,
1950                       char *host, DWORD hostlen, char *serv, DWORD servlen,
1951                       int flags)
1952 {
1953         ensure_socket_initialization();
1954         return ipv6_getnameinfo(sa, salen, host, hostlen, serv, servlen, flags);
1955 }
1956
1957 int mingw_socket(int domain, int type, int protocol)
1958 {
1959         int sockfd;
1960         SOCKET s;
1961
1962         ensure_socket_initialization();
1963         s = WSASocket(domain, type, protocol, NULL, 0, 0);
1964         if (s == INVALID_SOCKET) {
1965                 /*
1966                  * WSAGetLastError() values are regular BSD error codes
1967                  * biased by WSABASEERR.
1968                  * However, strerror() does not know about networking
1969                  * specific errors, which are values beginning at 38 or so.
1970                  * Therefore, we choose to leave the biased error code
1971                  * in errno so that _if_ someone looks up the code somewhere,
1972                  * then it is at least the number that are usually listed.
1973                  */
1974                 errno = WSAGetLastError();
1975                 return -1;
1976         }
1977         /* convert into a file descriptor */
1978         if ((sockfd = _open_osfhandle(s, O_RDWR|O_BINARY)) < 0) {
1979                 closesocket(s);
1980                 return error("unable to make a socket file descriptor: %s",
1981                         strerror(errno));
1982         }
1983         return sockfd;
1984 }
1985
1986 #undef connect
1987 int mingw_connect(int sockfd, struct sockaddr *sa, size_t sz)
1988 {
1989         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1990         return connect(s, sa, sz);
1991 }
1992
1993 #undef bind
1994 int mingw_bind(int sockfd, struct sockaddr *sa, size_t sz)
1995 {
1996         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1997         return bind(s, sa, sz);
1998 }
1999
2000 #undef setsockopt
2001 int mingw_setsockopt(int sockfd, int lvl, int optname, void *optval, int optlen)
2002 {
2003         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
2004         return setsockopt(s, lvl, optname, (const char*)optval, optlen);
2005 }
2006
2007 #undef shutdown
2008 int mingw_shutdown(int sockfd, int how)
2009 {
2010         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
2011         return shutdown(s, how);
2012 }
2013
2014 #undef listen
2015 int mingw_listen(int sockfd, int backlog)
2016 {
2017         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
2018         return listen(s, backlog);
2019 }
2020
2021 #undef accept
2022 int mingw_accept(int sockfd1, struct sockaddr *sa, socklen_t *sz)
2023 {
2024         int sockfd2;
2025
2026         SOCKET s1 = (SOCKET)_get_osfhandle(sockfd1);
2027         SOCKET s2 = accept(s1, sa, sz);
2028
2029         /* convert into a file descriptor */
2030         if ((sockfd2 = _open_osfhandle(s2, O_RDWR|O_BINARY)) < 0) {
2031                 int err = errno;
2032                 closesocket(s2);
2033                 return error("unable to make a socket file descriptor: %s",
2034                         strerror(err));
2035         }
2036         return sockfd2;
2037 }
2038
2039 #undef rename
2040 int mingw_rename(const char *pold, const char *pnew)
2041 {
2042         DWORD attrs, gle;
2043         int tries = 0;
2044         wchar_t wpold[MAX_PATH], wpnew[MAX_PATH];
2045         if (xutftowcs_path(wpold, pold) < 0 || xutftowcs_path(wpnew, pnew) < 0)
2046                 return -1;
2047
2048         /*
2049          * Try native rename() first to get errno right.
2050          * It is based on MoveFile(), which cannot overwrite existing files.
2051          */
2052         if (!_wrename(wpold, wpnew))
2053                 return 0;
2054         if (errno != EEXIST)
2055                 return -1;
2056 repeat:
2057         if (MoveFileExW(wpold, wpnew, MOVEFILE_REPLACE_EXISTING))
2058                 return 0;
2059         /* TODO: translate more errors */
2060         gle = GetLastError();
2061         if (gle == ERROR_ACCESS_DENIED &&
2062             (attrs = GetFileAttributesW(wpnew)) != INVALID_FILE_ATTRIBUTES) {
2063                 if (attrs & FILE_ATTRIBUTE_DIRECTORY) {
2064                         DWORD attrsold = GetFileAttributesW(wpold);
2065                         if (attrsold == INVALID_FILE_ATTRIBUTES ||
2066                             !(attrsold & FILE_ATTRIBUTE_DIRECTORY))
2067                                 errno = EISDIR;
2068                         else if (!_wrmdir(wpnew))
2069                                 goto repeat;
2070                         return -1;
2071                 }
2072                 if ((attrs & FILE_ATTRIBUTE_READONLY) &&
2073                     SetFileAttributesW(wpnew, attrs & ~FILE_ATTRIBUTE_READONLY)) {
2074                         if (MoveFileExW(wpold, wpnew, MOVEFILE_REPLACE_EXISTING))
2075                                 return 0;
2076                         gle = GetLastError();
2077                         /* revert file attributes on failure */
2078                         SetFileAttributesW(wpnew, attrs);
2079                 }
2080         }
2081         if (tries < ARRAY_SIZE(delay) && gle == ERROR_ACCESS_DENIED) {
2082                 /*
2083                  * We assume that some other process had the source or
2084                  * destination file open at the wrong moment and retry.
2085                  * In order to give the other process a higher chance to
2086                  * complete its operation, we give up our time slice now.
2087                  * If we have to retry again, we do sleep a bit.
2088                  */
2089                 Sleep(delay[tries]);
2090                 tries++;
2091                 goto repeat;
2092         }
2093         if (gle == ERROR_ACCESS_DENIED &&
2094                ask_yes_no_if_possible("Rename from '%s' to '%s' failed. "
2095                        "Should I try again?", pold, pnew))
2096                 goto repeat;
2097
2098         errno = EACCES;
2099         return -1;
2100 }
2101
2102 /*
2103  * Note that this doesn't return the actual pagesize, but
2104  * the allocation granularity. If future Windows specific git code
2105  * needs the real getpagesize function, we need to find another solution.
2106  */
2107 int mingw_getpagesize(void)
2108 {
2109         SYSTEM_INFO si;
2110         GetSystemInfo(&si);
2111         return si.dwAllocationGranularity;
2112 }
2113
2114 /* See https://msdn.microsoft.com/en-us/library/windows/desktop/ms724435.aspx */
2115 enum EXTENDED_NAME_FORMAT {
2116         NameDisplay = 3,
2117         NameUserPrincipal = 8
2118 };
2119
2120 static char *get_extended_user_info(enum EXTENDED_NAME_FORMAT type)
2121 {
2122         DECLARE_PROC_ADDR(secur32.dll, BOOL, GetUserNameExW,
2123                 enum EXTENDED_NAME_FORMAT, LPCWSTR, PULONG);
2124         static wchar_t wbuffer[1024];
2125         DWORD len;
2126
2127         if (!INIT_PROC_ADDR(GetUserNameExW))
2128                 return NULL;
2129
2130         len = ARRAY_SIZE(wbuffer);
2131         if (GetUserNameExW(type, wbuffer, &len)) {
2132                 char *converted = xmalloc((len *= 3));
2133                 if (xwcstoutf(converted, wbuffer, len) >= 0)
2134                         return converted;
2135                 free(converted);
2136         }
2137
2138         return NULL;
2139 }
2140
2141 char *mingw_query_user_email(void)
2142 {
2143         return get_extended_user_info(NameUserPrincipal);
2144 }
2145
2146 struct passwd *getpwuid(int uid)
2147 {
2148         static unsigned initialized;
2149         static char user_name[100];
2150         static struct passwd *p;
2151         DWORD len;
2152
2153         if (initialized)
2154                 return p;
2155
2156         len = sizeof(user_name);
2157         if (!GetUserName(user_name, &len)) {
2158                 initialized = 1;
2159                 return NULL;
2160         }
2161
2162         p = xmalloc(sizeof(*p));
2163         p->pw_name = user_name;
2164         p->pw_gecos = get_extended_user_info(NameDisplay);
2165         if (!p->pw_gecos)
2166                 p->pw_gecos = "unknown";
2167         p->pw_dir = NULL;
2168
2169         initialized = 1;
2170         return p;
2171 }
2172
2173 static HANDLE timer_event;
2174 static HANDLE timer_thread;
2175 static int timer_interval;
2176 static int one_shot;
2177 static sig_handler_t timer_fn = SIG_DFL, sigint_fn = SIG_DFL;
2178
2179 /* The timer works like this:
2180  * The thread, ticktack(), is a trivial routine that most of the time
2181  * only waits to receive the signal to terminate. The main thread tells
2182  * the thread to terminate by setting the timer_event to the signalled
2183  * state.
2184  * But ticktack() interrupts the wait state after the timer's interval
2185  * length to call the signal handler.
2186  */
2187
2188 static unsigned __stdcall ticktack(void *dummy)
2189 {
2190         while (WaitForSingleObject(timer_event, timer_interval) == WAIT_TIMEOUT) {
2191                 mingw_raise(SIGALRM);
2192                 if (one_shot)
2193                         break;
2194         }
2195         return 0;
2196 }
2197
2198 static int start_timer_thread(void)
2199 {
2200         timer_event = CreateEvent(NULL, FALSE, FALSE, NULL);
2201         if (timer_event) {
2202                 timer_thread = (HANDLE) _beginthreadex(NULL, 0, ticktack, NULL, 0, NULL);
2203                 if (!timer_thread )
2204                         return errno = ENOMEM,
2205                                 error("cannot start timer thread");
2206         } else
2207                 return errno = ENOMEM,
2208                         error("cannot allocate resources for timer");
2209         return 0;
2210 }
2211
2212 static void stop_timer_thread(void)
2213 {
2214         if (timer_event)
2215                 SetEvent(timer_event);  /* tell thread to terminate */
2216         if (timer_thread) {
2217                 int rc = WaitForSingleObject(timer_thread, 10000);
2218                 if (rc == WAIT_TIMEOUT)
2219                         error("timer thread did not terminate timely");
2220                 else if (rc != WAIT_OBJECT_0)
2221                         error("waiting for timer thread failed: %lu",
2222                               GetLastError());
2223                 CloseHandle(timer_thread);
2224         }
2225         if (timer_event)
2226                 CloseHandle(timer_event);
2227         timer_event = NULL;
2228         timer_thread = NULL;
2229 }
2230
2231 static inline int is_timeval_eq(const struct timeval *i1, const struct timeval *i2)
2232 {
2233         return i1->tv_sec == i2->tv_sec && i1->tv_usec == i2->tv_usec;
2234 }
2235
2236 int setitimer(int type, struct itimerval *in, struct itimerval *out)
2237 {
2238         static const struct timeval zero;
2239         static int atexit_done;
2240
2241         if (out != NULL)
2242                 return errno = EINVAL,
2243                         error("setitimer param 3 != NULL not implemented");
2244         if (!is_timeval_eq(&in->it_interval, &zero) &&
2245             !is_timeval_eq(&in->it_interval, &in->it_value))
2246                 return errno = EINVAL,
2247                         error("setitimer: it_interval must be zero or eq it_value");
2248
2249         if (timer_thread)
2250                 stop_timer_thread();
2251
2252         if (is_timeval_eq(&in->it_value, &zero) &&
2253             is_timeval_eq(&in->it_interval, &zero))
2254                 return 0;
2255
2256         timer_interval = in->it_value.tv_sec * 1000 + in->it_value.tv_usec / 1000;
2257         one_shot = is_timeval_eq(&in->it_interval, &zero);
2258         if (!atexit_done) {
2259                 atexit(stop_timer_thread);
2260                 atexit_done = 1;
2261         }
2262         return start_timer_thread();
2263 }
2264
2265 int sigaction(int sig, struct sigaction *in, struct sigaction *out)
2266 {
2267         if (sig != SIGALRM)
2268                 return errno = EINVAL,
2269                         error("sigaction only implemented for SIGALRM");
2270         if (out != NULL)
2271                 return errno = EINVAL,
2272                         error("sigaction: param 3 != NULL not implemented");
2273
2274         timer_fn = in->sa_handler;
2275         return 0;
2276 }
2277
2278 #undef signal
2279 sig_handler_t mingw_signal(int sig, sig_handler_t handler)
2280 {
2281         sig_handler_t old;
2282
2283         switch (sig) {
2284         case SIGALRM:
2285                 old = timer_fn;
2286                 timer_fn = handler;
2287                 break;
2288
2289         case SIGINT:
2290                 old = sigint_fn;
2291                 sigint_fn = handler;
2292                 break;
2293
2294         default:
2295                 return signal(sig, handler);
2296         }
2297
2298         return old;
2299 }
2300
2301 #undef raise
2302 int mingw_raise(int sig)
2303 {
2304         switch (sig) {
2305         case SIGALRM:
2306                 if (timer_fn == SIG_DFL) {
2307                         if (isatty(STDERR_FILENO))
2308                                 fputs("Alarm clock\n", stderr);
2309                         exit(128 + SIGALRM);
2310                 } else if (timer_fn != SIG_IGN)
2311                         timer_fn(SIGALRM);
2312                 return 0;
2313
2314         case SIGINT:
2315                 if (sigint_fn == SIG_DFL)
2316                         exit(128 + SIGINT);
2317                 else if (sigint_fn != SIG_IGN)
2318                         sigint_fn(SIGINT);
2319                 return 0;
2320
2321         default:
2322                 return raise(sig);
2323         }
2324 }
2325
2326 int link(const char *oldpath, const char *newpath)
2327 {
2328         wchar_t woldpath[MAX_PATH], wnewpath[MAX_PATH];
2329         if (xutftowcs_path(woldpath, oldpath) < 0 ||
2330                 xutftowcs_path(wnewpath, newpath) < 0)
2331                 return -1;
2332
2333         if (!CreateHardLinkW(wnewpath, woldpath, NULL)) {
2334                 errno = err_win_to_posix(GetLastError());
2335                 return -1;
2336         }
2337         return 0;
2338 }
2339
2340 pid_t waitpid(pid_t pid, int *status, int options)
2341 {
2342         HANDLE h = OpenProcess(SYNCHRONIZE | PROCESS_QUERY_INFORMATION,
2343             FALSE, pid);
2344         if (!h) {
2345                 errno = ECHILD;
2346                 return -1;
2347         }
2348
2349         if (pid > 0 && options & WNOHANG) {
2350                 if (WAIT_OBJECT_0 != WaitForSingleObject(h, 0)) {
2351                         CloseHandle(h);
2352                         return 0;
2353                 }
2354                 options &= ~WNOHANG;
2355         }
2356
2357         if (options == 0) {
2358                 struct pinfo_t **ppinfo;
2359                 if (WaitForSingleObject(h, INFINITE) != WAIT_OBJECT_0) {
2360                         CloseHandle(h);
2361                         return 0;
2362                 }
2363
2364                 if (status)
2365                         GetExitCodeProcess(h, (LPDWORD)status);
2366
2367                 EnterCriticalSection(&pinfo_cs);
2368
2369                 ppinfo = &pinfo;
2370                 while (*ppinfo) {
2371                         struct pinfo_t *info = *ppinfo;
2372                         if (info->pid == pid) {
2373                                 CloseHandle(info->proc);
2374                                 *ppinfo = info->next;
2375                                 free(info);
2376                                 break;
2377                         }
2378                         ppinfo = &info->next;
2379                 }
2380
2381                 LeaveCriticalSection(&pinfo_cs);
2382
2383                 CloseHandle(h);
2384                 return pid;
2385         }
2386         CloseHandle(h);
2387
2388         errno = EINVAL;
2389         return -1;
2390 }
2391
2392 int xutftowcsn(wchar_t *wcs, const char *utfs, size_t wcslen, int utflen)
2393 {
2394         int upos = 0, wpos = 0;
2395         const unsigned char *utf = (const unsigned char*) utfs;
2396         if (!utf || !wcs || wcslen < 1) {
2397                 errno = EINVAL;
2398                 return -1;
2399         }
2400         /* reserve space for \0 */
2401         wcslen--;
2402         if (utflen < 0)
2403                 utflen = INT_MAX;
2404
2405         while (upos < utflen) {
2406                 int c = utf[upos++] & 0xff;
2407                 if (utflen == INT_MAX && c == 0)
2408                         break;
2409
2410                 if (wpos >= wcslen) {
2411                         wcs[wpos] = 0;
2412                         errno = ERANGE;
2413                         return -1;
2414                 }
2415
2416                 if (c < 0x80) {
2417                         /* ASCII */
2418                         wcs[wpos++] = c;
2419                 } else if (c >= 0xc2 && c < 0xe0 && upos < utflen &&
2420                                 (utf[upos] & 0xc0) == 0x80) {
2421                         /* 2-byte utf-8 */
2422                         c = ((c & 0x1f) << 6);
2423                         c |= (utf[upos++] & 0x3f);
2424                         wcs[wpos++] = c;
2425                 } else if (c >= 0xe0 && c < 0xf0 && upos + 1 < utflen &&
2426                                 !(c == 0xe0 && utf[upos] < 0xa0) && /* over-long encoding */
2427                                 (utf[upos] & 0xc0) == 0x80 &&
2428                                 (utf[upos + 1] & 0xc0) == 0x80) {
2429                         /* 3-byte utf-8 */
2430                         c = ((c & 0x0f) << 12);
2431                         c |= ((utf[upos++] & 0x3f) << 6);
2432                         c |= (utf[upos++] & 0x3f);
2433                         wcs[wpos++] = c;
2434                 } else if (c >= 0xf0 && c < 0xf5 && upos + 2 < utflen &&
2435                                 wpos + 1 < wcslen &&
2436                                 !(c == 0xf0 && utf[upos] < 0x90) && /* over-long encoding */
2437                                 !(c == 0xf4 && utf[upos] >= 0x90) && /* > \u10ffff */
2438                                 (utf[upos] & 0xc0) == 0x80 &&
2439                                 (utf[upos + 1] & 0xc0) == 0x80 &&
2440                                 (utf[upos + 2] & 0xc0) == 0x80) {
2441                         /* 4-byte utf-8: convert to \ud8xx \udcxx surrogate pair */
2442                         c = ((c & 0x07) << 18);
2443                         c |= ((utf[upos++] & 0x3f) << 12);
2444                         c |= ((utf[upos++] & 0x3f) << 6);
2445                         c |= (utf[upos++] & 0x3f);
2446                         c -= 0x10000;
2447                         wcs[wpos++] = 0xd800 | (c >> 10);
2448                         wcs[wpos++] = 0xdc00 | (c & 0x3ff);
2449                 } else if (c >= 0xa0) {
2450                         /* invalid utf-8 byte, printable unicode char: convert 1:1 */
2451                         wcs[wpos++] = c;
2452                 } else {
2453                         /* invalid utf-8 byte, non-printable unicode: convert to hex */
2454                         static const char *hex = "0123456789abcdef";
2455                         wcs[wpos++] = hex[c >> 4];
2456                         if (wpos < wcslen)
2457                                 wcs[wpos++] = hex[c & 0x0f];
2458                 }
2459         }
2460         wcs[wpos] = 0;
2461         return wpos;
2462 }
2463
2464 int xwcstoutf(char *utf, const wchar_t *wcs, size_t utflen)
2465 {
2466         if (!wcs || !utf || utflen < 1) {
2467                 errno = EINVAL;
2468                 return -1;
2469         }
2470         utflen = WideCharToMultiByte(CP_UTF8, 0, wcs, -1, utf, utflen, NULL, NULL);
2471         if (utflen)
2472                 return utflen - 1;
2473         errno = ERANGE;
2474         return -1;
2475 }
2476
2477 static void setup_windows_environment(void)
2478 {
2479         char *tmp = getenv("TMPDIR");
2480
2481         /* on Windows it is TMP and TEMP */
2482         if (!tmp) {
2483                 if (!(tmp = getenv("TMP")))
2484                         tmp = getenv("TEMP");
2485                 if (tmp) {
2486                         setenv("TMPDIR", tmp, 1);
2487                         tmp = getenv("TMPDIR");
2488                 }
2489         }
2490
2491         if (tmp) {
2492                 /*
2493                  * Convert all dir separators to forward slashes,
2494                  * to help shell commands called from the Git
2495                  * executable (by not mistaking the dir separators
2496                  * for escape characters).
2497                  */
2498                 convert_slashes(tmp);
2499         }
2500
2501         /* simulate TERM to enable auto-color (see color.c) */
2502         if (!getenv("TERM"))
2503                 setenv("TERM", "cygwin", 1);
2504 }
2505
2506 int is_valid_win32_path(const char *path)
2507 {
2508         int preceding_space_or_period = 0, i = 0, periods = 0;
2509
2510         if (!protect_ntfs)
2511                 return 1;
2512
2513         skip_dos_drive_prefix((char **)&path);
2514
2515         for (;;) {
2516                 char c = *(path++);
2517                 switch (c) {
2518                 case '\0':
2519                 case '/': case '\\':
2520                         /* cannot end in ` ` or `.`, except for `.` and `..` */
2521                         if (preceding_space_or_period &&
2522                             (i != periods || periods > 2))
2523                                 return 0;
2524                         if (!c)
2525                                 return 1;
2526
2527                         i = periods = preceding_space_or_period = 0;
2528                         continue;
2529                 case '.':
2530                         periods++;
2531                         /* fallthru */
2532                 case ' ':
2533                         preceding_space_or_period = 1;
2534                         i++;
2535                         continue;
2536                 case ':': /* DOS drive prefix was already skipped */
2537                 case '<': case '>': case '"': case '|': case '?': case '*':
2538                         /* illegal character */
2539                         return 0;
2540                 default:
2541                         if (c > '\0' && c < '\x20')
2542                                 /* illegal character */
2543                                 return 0;
2544                 }
2545                 preceding_space_or_period = 0;
2546                 i++;
2547         }
2548 }
2549
2550 /*
2551  * Disable MSVCRT command line wildcard expansion (__getmainargs called from
2552  * mingw startup code, see init.c in mingw runtime).
2553  */
2554 int _CRT_glob = 0;
2555
2556 typedef struct {
2557         int newmode;
2558 } _startupinfo;
2559
2560 extern int __wgetmainargs(int *argc, wchar_t ***argv, wchar_t ***env, int glob,
2561                 _startupinfo *si);
2562
2563 static NORETURN void die_startup(void)
2564 {
2565         fputs("fatal: not enough memory for initialization", stderr);
2566         exit(128);
2567 }
2568
2569 static void *malloc_startup(size_t size)
2570 {
2571         void *result = malloc(size);
2572         if (!result)
2573                 die_startup();
2574         return result;
2575 }
2576
2577 static char *wcstoutfdup_startup(char *buffer, const wchar_t *wcs, size_t len)
2578 {
2579         len = xwcstoutf(buffer, wcs, len) + 1;
2580         return memcpy(malloc_startup(len), buffer, len);
2581 }
2582
2583 static void maybe_redirect_std_handle(const wchar_t *key, DWORD std_id, int fd,
2584                                       DWORD desired_access, DWORD flags)
2585 {
2586         DWORD create_flag = fd ? OPEN_ALWAYS : OPEN_EXISTING;
2587         wchar_t buf[MAX_PATH];
2588         DWORD max = ARRAY_SIZE(buf);
2589         HANDLE handle;
2590         DWORD ret = GetEnvironmentVariableW(key, buf, max);
2591
2592         if (!ret || ret >= max)
2593                 return;
2594
2595         /* make sure this does not leak into child processes */
2596         SetEnvironmentVariableW(key, NULL);
2597         if (!wcscmp(buf, L"off")) {
2598                 close(fd);
2599                 handle = GetStdHandle(std_id);
2600                 if (handle != INVALID_HANDLE_VALUE)
2601                         CloseHandle(handle);
2602                 return;
2603         }
2604         if (std_id == STD_ERROR_HANDLE && !wcscmp(buf, L"2>&1")) {
2605                 handle = GetStdHandle(STD_OUTPUT_HANDLE);
2606                 if (handle == INVALID_HANDLE_VALUE) {
2607                         close(fd);
2608                         handle = GetStdHandle(std_id);
2609                         if (handle != INVALID_HANDLE_VALUE)
2610                                 CloseHandle(handle);
2611                 } else {
2612                         int new_fd = _open_osfhandle((intptr_t)handle, O_BINARY);
2613                         SetStdHandle(std_id, handle);
2614                         dup2(new_fd, fd);
2615                         /* do *not* close the new_fd: that would close stdout */
2616                 }
2617                 return;
2618         }
2619         handle = CreateFileW(buf, desired_access, 0, NULL, create_flag,
2620                              flags, NULL);
2621         if (handle != INVALID_HANDLE_VALUE) {
2622                 int new_fd = _open_osfhandle((intptr_t)handle, O_BINARY);
2623                 SetStdHandle(std_id, handle);
2624                 dup2(new_fd, fd);
2625                 close(new_fd);
2626         }
2627 }
2628
2629 static void maybe_redirect_std_handles(void)
2630 {
2631         maybe_redirect_std_handle(L"GIT_REDIRECT_STDIN", STD_INPUT_HANDLE, 0,
2632                                   GENERIC_READ, FILE_ATTRIBUTE_NORMAL);
2633         maybe_redirect_std_handle(L"GIT_REDIRECT_STDOUT", STD_OUTPUT_HANDLE, 1,
2634                                   GENERIC_WRITE, FILE_ATTRIBUTE_NORMAL);
2635         maybe_redirect_std_handle(L"GIT_REDIRECT_STDERR", STD_ERROR_HANDLE, 2,
2636                                   GENERIC_WRITE, FILE_FLAG_NO_BUFFERING);
2637 }
2638
2639 void mingw_startup(void)
2640 {
2641         int i, maxlen, argc;
2642         char *buffer;
2643         wchar_t **wenv, **wargv;
2644         _startupinfo si;
2645
2646         maybe_redirect_std_handles();
2647
2648         /* get wide char arguments and environment */
2649         si.newmode = 0;
2650         if (__wgetmainargs(&argc, &wargv, &wenv, _CRT_glob, &si) < 0)
2651                 die_startup();
2652
2653         /* determine size of argv and environ conversion buffer */
2654         maxlen = wcslen(wargv[0]);
2655         for (i = 1; i < argc; i++)
2656                 maxlen = max(maxlen, wcslen(wargv[i]));
2657
2658         /* allocate buffer (wchar_t encodes to max 3 UTF-8 bytes) */
2659         maxlen = 3 * maxlen + 1;
2660         buffer = malloc_startup(maxlen);
2661
2662         /* convert command line arguments and environment to UTF-8 */
2663         for (i = 0; i < argc; i++)
2664                 __argv[i] = wcstoutfdup_startup(buffer, wargv[i], maxlen);
2665         free(buffer);
2666
2667         /* fix Windows specific environment settings */
2668         setup_windows_environment();
2669
2670         unset_environment_variables = xstrdup("PERL5LIB");
2671
2672         /* initialize critical section for waitpid pinfo_t list */
2673         InitializeCriticalSection(&pinfo_cs);
2674
2675         /* set up default file mode and file modes for stdin/out/err */
2676         _fmode = _O_BINARY;
2677         _setmode(_fileno(stdin), _O_BINARY);
2678         _setmode(_fileno(stdout), _O_BINARY);
2679         _setmode(_fileno(stderr), _O_BINARY);
2680
2681         /* initialize Unicode console */
2682         winansi_init();
2683 }
2684
2685 int uname(struct utsname *buf)
2686 {
2687         unsigned v = (unsigned)GetVersion();
2688         memset(buf, 0, sizeof(*buf));
2689         xsnprintf(buf->sysname, sizeof(buf->sysname), "Windows");
2690         xsnprintf(buf->release, sizeof(buf->release),
2691                  "%u.%u", v & 0xff, (v >> 8) & 0xff);
2692         /* assuming NT variants only.. */
2693         xsnprintf(buf->version, sizeof(buf->version),
2694                   "%u", (v >> 16) & 0x7fff);
2695         return 0;
2696 }