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