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