Andrew's git - gitweb.git/blob

   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        wchar_t wpath[MAX_PATH];
1165        snprintf(path, sizeof(path), "%.*s\\%s.exe", dirlen, dir, cmd);
1166
1167        if (xutftowcs_path(wpath, path) < 0)
1168                return NULL;
1169
1170        if (!isexe && _waccess(wpath, F_OK) == 0)
1171                return xstrdup(path);
1172        wpath[wcslen(wpath)-4] = '\0';
1173        if ((!exe_only || isexe) && _waccess(wpath, F_OK) == 0) {
1174                if (!(GetFileAttributesW(wpath) & FILE_ATTRIBUTE_DIRECTORY)) {
1175                        path[strlen(path)-4] = '\0';
1176                        return xstrdup(path);
1177                }
1178        }
1179        return NULL;
1180}
1181
1182/*
1183 * Determines the absolute path of cmd using the split path in path.
1184 * If cmd contains a slash or backslash, no lookup is performed.
1185 */
1186static char *path_lookup(const char *cmd, int exe_only)
1187{
1188        const char *path;
1189        char *prog = NULL;
1190        int len = strlen(cmd);
1191        int isexe = len >= 4 && !strcasecmp(cmd+len-4, ".exe");
1192
1193        if (strchr(cmd, '/') || strchr(cmd, '\\'))
1194                return xstrdup(cmd);
1195
1196        path = mingw_getenv("PATH");
1197        if (!path)
1198                return NULL;
1199
1200        while (!prog) {
1201                const char *sep = strchrnul(path, ';');
1202                int dirlen = sep - path;
1203                if (dirlen)
1204                        prog = lookup_prog(path, dirlen, cmd, isexe, exe_only);
1205                if (!*sep)
1206                        break;
1207                path = sep + 1;
1208        }
1209
1210        return prog;
1211}
1212
1213static const wchar_t *wcschrnul(const wchar_t *s, wchar_t c)
1214{
1215        while (*s && *s != c)
1216                s++;
1217        return s;
1218}
1219
1220/* Compare only keys */
1221static int wenvcmp(const void *a, const void *b)
1222{
1223        wchar_t *p = *(wchar_t **)a, *q = *(wchar_t **)b;
1224        size_t p_len, q_len;
1225
1226        /* Find the keys */
1227        p_len = wcschrnul(p, L'=') - p;
1228        q_len = wcschrnul(q, L'=') - q;
1229
1230        /* If the length differs, include the shorter key's NUL */
1231        if (p_len < q_len)
1232                p_len++;
1233        else if (p_len > q_len)
1234                p_len = q_len + 1;
1235
1236        return _wcsnicmp(p, q, p_len);
1237}
1238
1239/* We need a stable sort to convert the environment between UTF-16 <-> UTF-8 */
1240#ifndef INTERNAL_QSORT
1241#include "qsort.c"
1242#endif
1243
1244/*
1245 * Build an environment block combining the inherited environment
1246 * merged with the given list of settings.
1247 *
1248 * Values of the form "KEY=VALUE" in deltaenv override inherited values.
1249 * Values of the form "KEY" in deltaenv delete inherited values.
1250 *
1251 * Multiple entries in deltaenv for the same key are explicitly allowed.
1252 *
1253 * We return a contiguous block of UNICODE strings with a final trailing
1254 * zero word.
1255 */
1256static wchar_t *make_environment_block(char **deltaenv)
1257{
1258        wchar_t *wenv = GetEnvironmentStringsW(), *wdeltaenv, *result, *p;
1259        size_t wlen, s, delta_size, size;
1260
1261        wchar_t **array = NULL;
1262        size_t alloc = 0, nr = 0, i;
1263
1264        size = 1; /* for extra NUL at the end */
1265
1266        /* If there is no deltaenv to apply, simply return a copy. */
1267        if (!deltaenv || !*deltaenv) {
1268                for (p = wenv; p && *p; ) {
1269                        size_t s = wcslen(p) + 1;
1270                        size += s;
1271                        p += s;
1272                }
1273
1274                ALLOC_ARRAY(result, size);
1275                COPY_ARRAY(result, wenv, size);
1276                FreeEnvironmentStringsW(wenv);
1277                return result;
1278        }
1279
1280        /*
1281         * If there is a deltaenv, let's accumulate all keys into `array`,
1282         * sort them using the stable git_qsort() and then copy, skipping
1283         * duplicate keys
1284         */
1285        for (p = wenv; p && *p; ) {
1286                ALLOC_GROW(array, nr + 1, alloc);
1287                s = wcslen(p) + 1;
1288                array[nr++] = p;
1289                p += s;
1290                size += s;
1291        }
1292
1293        /* (over-)assess size needed for wchar version of deltaenv */
1294        for (delta_size = 0, i = 0; deltaenv[i]; i++)
1295                delta_size += strlen(deltaenv[i]) * 2 + 1;
1296        ALLOC_ARRAY(wdeltaenv, delta_size);
1297
1298        /* convert the deltaenv, appending to array */
1299        for (i = 0, p = wdeltaenv; deltaenv[i]; i++) {
1300                ALLOC_GROW(array, nr + 1, alloc);
1301                wlen = xutftowcs(p, deltaenv[i], wdeltaenv + delta_size - p);
1302                array[nr++] = p;
1303                p += wlen + 1;
1304        }
1305
1306        git_qsort(array, nr, sizeof(*array), wenvcmp);
1307        ALLOC_ARRAY(result, size + delta_size);
1308
1309        for (p = result, i = 0; i < nr; i++) {
1310                /* Skip any duplicate keys; last one wins */
1311                while (i + 1 < nr && !wenvcmp(array + i, array + i + 1))
1312                       i++;
1313
1314                /* Skip "to delete" entry */
1315                if (!wcschr(array[i], L'='))
1316                        continue;
1317
1318                size = wcslen(array[i]) + 1;
1319                COPY_ARRAY(p, array[i], size);
1320                p += size;
1321        }
1322        *p = L'\0';
1323
1324        free(array);
1325        free(wdeltaenv);
1326        FreeEnvironmentStringsW(wenv);
1327        return result;
1328}
1329
1330static void do_unset_environment_variables(void)
1331{
1332        static int done;
1333        char *p = unset_environment_variables;
1334
1335        if (done || !p)
1336                return;
1337        done = 1;
1338
1339        for (;;) {
1340                char *comma = strchr(p, ',');
1341
1342                if (comma)
1343                        *comma = '\0';
1344                unsetenv(p);
1345                if (!comma)
1346                        break;
1347                p = comma + 1;
1348        }
1349}
1350
1351struct pinfo_t {
1352        struct pinfo_t *next;
1353        pid_t pid;
1354        HANDLE proc;
1355};
1356static struct pinfo_t *pinfo = NULL;
1357CRITICAL_SECTION pinfo_cs;
1358
1359/* Used to match and chomp off path components */
1360static inline int match_last_path_component(const char *path, size_t *len,
1361                                            const char *component)
1362{
1363        size_t component_len = strlen(component);
1364        if (*len < component_len + 1 ||
1365            !is_dir_sep(path[*len - component_len - 1]) ||
1366            fspathncmp(path + *len - component_len, component, component_len))
1367                return 0;
1368        *len -= component_len + 1;
1369        /* chomp off repeated dir separators */
1370        while (*len > 0 && is_dir_sep(path[*len - 1]))
1371                (*len)--;
1372        return 1;
1373}
1374
1375static int is_msys2_sh(const char *cmd)
1376{
1377        if (cmd && !strcmp(cmd, "sh")) {
1378                static int ret = -1;
1379                char *p;
1380
1381                if (ret >= 0)
1382                        return ret;
1383
1384                p = path_lookup(cmd, 0);
1385                if (!p)
1386                        ret = 0;
1387                else {
1388                        size_t len = strlen(p);
1389
1390                        ret = match_last_path_component(p, &len, "sh.exe") &&
1391                                match_last_path_component(p, &len, "bin") &&
1392                                match_last_path_component(p, &len, "usr");
1393                        free(p);
1394                }
1395                return ret;
1396        }
1397        return 0;
1398}
1399
1400static pid_t mingw_spawnve_fd(const char *cmd, const char **argv, char **deltaenv,
1401                              const char *dir,
1402                              int prepend_cmd, int fhin, int fhout, int fherr)
1403{
1404        STARTUPINFOW si;
1405        PROCESS_INFORMATION pi;
1406        struct strbuf args;
1407        wchar_t wcmd[MAX_PATH], wdir[MAX_PATH], *wargs, *wenvblk = NULL;
1408        unsigned flags = CREATE_UNICODE_ENVIRONMENT;
1409        BOOL ret;
1410        HANDLE cons;
1411        const char *(*quote_arg)(const char *arg) =
1412                is_msys2_sh(*argv) ? quote_arg_msys2 : quote_arg_msvc;
1413
1414        do_unset_environment_variables();
1415
1416        /* Determine whether or not we are associated to a console */
1417        cons = CreateFileW(L"CONOUT$", GENERIC_WRITE,
1418                        FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
1419                        FILE_ATTRIBUTE_NORMAL, NULL);
1420        if (cons == INVALID_HANDLE_VALUE) {
1421                /* There is no console associated with this process.
1422                 * Since the child is a console process, Windows
1423                 * would normally create a console window. But
1424                 * since we'll be redirecting std streams, we do
1425                 * not need the console.
1426                 * It is necessary to use DETACHED_PROCESS
1427                 * instead of CREATE_NO_WINDOW to make ssh
1428                 * recognize that it has no console.
1429                 */
1430                flags |= DETACHED_PROCESS;
1431        } else {
1432                /* There is already a console. If we specified
1433                 * DETACHED_PROCESS here, too, Windows would
1434                 * disassociate the child from the console.
1435                 * The same is true for CREATE_NO_WINDOW.
1436                 * Go figure!
1437                 */
1438                CloseHandle(cons);
1439        }
1440        memset(&si, 0, sizeof(si));
1441        si.cb = sizeof(si);
1442        si.dwFlags = STARTF_USESTDHANDLES;
1443        si.hStdInput = winansi_get_osfhandle(fhin);
1444        si.hStdOutput = winansi_get_osfhandle(fhout);
1445        si.hStdError = winansi_get_osfhandle(fherr);
1446
1447        if (*argv && !strcmp(cmd, *argv))
1448                wcmd[0] = L'\0';
1449        else if (xutftowcs_path(wcmd, cmd) < 0)
1450                return -1;
1451        if (dir && xutftowcs_path(wdir, dir) < 0)
1452                return -1;
1453
1454        /* concatenate argv, quoting args as we go */
1455        strbuf_init(&args, 0);
1456        if (prepend_cmd) {
1457                char *quoted = (char *)quote_arg(cmd);
1458                strbuf_addstr(&args, quoted);
1459                if (quoted != cmd)
1460                        free(quoted);
1461        }
1462        for (; *argv; argv++) {
1463                char *quoted = (char *)quote_arg(*argv);
1464                if (*args.buf)
1465                        strbuf_addch(&args, ' ');
1466                strbuf_addstr(&args, quoted);
1467                if (quoted != *argv)
1468                        free(quoted);
1469        }
1470
1471        ALLOC_ARRAY(wargs, st_add(st_mult(2, args.len), 1));
1472        xutftowcs(wargs, args.buf, 2 * args.len + 1);
1473        strbuf_release(&args);
1474
1475        wenvblk = make_environment_block(deltaenv);
1476
1477        memset(&pi, 0, sizeof(pi));
1478        ret = CreateProcessW(*wcmd ? wcmd : NULL, wargs, NULL, NULL, TRUE,
1479                flags, wenvblk, dir ? wdir : NULL, &si, &pi);
1480
1481        free(wenvblk);
1482        free(wargs);
1483
1484        if (!ret) {
1485                errno = ENOENT;
1486                return -1;
1487        }
1488        CloseHandle(pi.hThread);
1489
1490        /*
1491         * The process ID is the human-readable identifier of the process
1492         * that we want to present in log and error messages. The handle
1493         * is not useful for this purpose. But we cannot close it, either,
1494         * because it is not possible to turn a process ID into a process
1495         * handle after the process terminated.
1496         * Keep the handle in a list for waitpid.
1497         */
1498        EnterCriticalSection(&pinfo_cs);
1499        {
1500                struct pinfo_t *info = xmalloc(sizeof(struct pinfo_t));
1501                info->pid = pi.dwProcessId;
1502                info->proc = pi.hProcess;
1503                info->next = pinfo;
1504                pinfo = info;
1505        }
1506        LeaveCriticalSection(&pinfo_cs);
1507
1508        return (pid_t)pi.dwProcessId;
1509}
1510
1511static pid_t mingw_spawnv(const char *cmd, const char **argv, int prepend_cmd)
1512{
1513        return mingw_spawnve_fd(cmd, argv, NULL, NULL, prepend_cmd, 0, 1, 2);
1514}
1515
1516pid_t mingw_spawnvpe(const char *cmd, const char **argv, char **deltaenv,
1517                     const char *dir,
1518                     int fhin, int fhout, int fherr)
1519{
1520        pid_t pid;
1521        char *prog = path_lookup(cmd, 0);
1522
1523        if (!prog) {
1524                errno = ENOENT;
1525                pid = -1;
1526        }
1527        else {
1528                const char *interpr = parse_interpreter(prog);
1529
1530                if (interpr) {
1531                        const char *argv0 = argv[0];
1532                        char *iprog = path_lookup(interpr, 1);
1533                        argv[0] = prog;
1534                        if (!iprog) {
1535                                errno = ENOENT;
1536                                pid = -1;
1537                        }
1538                        else {
1539                                pid = mingw_spawnve_fd(iprog, argv, deltaenv, dir, 1,
1540                                                       fhin, fhout, fherr);
1541                                free(iprog);
1542                        }
1543                        argv[0] = argv0;
1544                }
1545                else
1546                        pid = mingw_spawnve_fd(prog, argv, deltaenv, dir, 0,
1547                                               fhin, fhout, fherr);
1548                free(prog);
1549        }
1550        return pid;
1551}
1552
1553static int try_shell_exec(const char *cmd, char *const *argv)
1554{
1555        const char *interpr = parse_interpreter(cmd);
1556        char *prog;
1557        int pid = 0;
1558
1559        if (!interpr)
1560                return 0;
1561        prog = path_lookup(interpr, 1);
1562        if (prog) {
1563                int exec_id;
1564                int argc = 0;
1565#ifndef _MSC_VER
1566                const
1567#endif
1568                char **argv2;
1569                while (argv[argc]) argc++;
1570                ALLOC_ARRAY(argv2, argc + 1);
1571                argv2[0] = (char *)cmd; /* full path to the script file */
1572                memcpy(&argv2[1], &argv[1], sizeof(*argv) * argc);
1573                exec_id = trace2_exec(prog, argv2);
1574                pid = mingw_spawnv(prog, argv2, 1);
1575                if (pid >= 0) {
1576                        int status;
1577                        if (waitpid(pid, &status, 0) < 0)
1578                                status = 255;
1579                        trace2_exec_result(exec_id, status);
1580                        exit(status);
1581                }
1582                trace2_exec_result(exec_id, -1);
1583                pid = 1;        /* indicate that we tried but failed */
1584                free(prog);
1585                free(argv2);
1586        }
1587        return pid;
1588}
1589
1590int mingw_execv(const char *cmd, char *const *argv)
1591{
1592        /* check if git_command is a shell script */
1593        if (!try_shell_exec(cmd, argv)) {
1594                int pid, status;
1595                int exec_id;
1596
1597                exec_id = trace2_exec(cmd, (const char **)argv);
1598                pid = mingw_spawnv(cmd, (const char **)argv, 0);
1599                if (pid < 0) {
1600                        trace2_exec_result(exec_id, -1);
1601                        return -1;
1602                }
1603                if (waitpid(pid, &status, 0) < 0)
1604                        status = 255;
1605                trace2_exec_result(exec_id, status);
1606                exit(status);
1607        }
1608        return -1;
1609}
1610
1611int mingw_execvp(const char *cmd, char *const *argv)
1612{
1613        char *prog = path_lookup(cmd, 0);
1614
1615        if (prog) {
1616                mingw_execv(prog, argv);
1617                free(prog);
1618        } else
1619                errno = ENOENT;
1620
1621        return -1;
1622}
1623
1624int mingw_kill(pid_t pid, int sig)
1625{
1626        if (pid > 0 && sig == SIGTERM) {
1627                HANDLE h = OpenProcess(PROCESS_TERMINATE, FALSE, pid);
1628
1629                if (TerminateProcess(h, -1)) {
1630                        CloseHandle(h);
1631                        return 0;
1632                }
1633
1634                errno = err_win_to_posix(GetLastError());
1635                CloseHandle(h);
1636                return -1;
1637        } else if (pid > 0 && sig == 0) {
1638                HANDLE h = OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, pid);
1639                if (h) {
1640                        CloseHandle(h);
1641                        return 0;
1642                }
1643        }
1644
1645        errno = EINVAL;
1646        return -1;
1647}
1648
1649/*
1650 * UTF-8 versions of getenv(), putenv() and unsetenv().
1651 * Internally, they use the CRT's stock UNICODE routines
1652 * to avoid data loss.
1653 */
1654char *mingw_getenv(const char *name)
1655{
1656#define GETENV_MAX_RETAIN 64
1657        static char *values[GETENV_MAX_RETAIN];
1658        static int value_counter;
1659        int len_key, len_value;
1660        wchar_t *w_key;
1661        char *value;
1662        wchar_t w_value[32768];
1663
1664        if (!name || !*name)
1665                return NULL;
1666
1667        len_key = strlen(name) + 1;
1668        /* We cannot use xcalloc() here because that uses getenv() itself */
1669        w_key = calloc(len_key, sizeof(wchar_t));
1670        if (!w_key)
1671                die("Out of memory, (tried to allocate %u wchar_t's)", len_key);
1672        xutftowcs(w_key, name, len_key);
1673        len_value = GetEnvironmentVariableW(w_key, w_value, ARRAY_SIZE(w_value));
1674        if (!len_value && GetLastError() == ERROR_ENVVAR_NOT_FOUND) {
1675                free(w_key);
1676                return NULL;
1677        }
1678        free(w_key);
1679
1680        len_value = len_value * 3 + 1;
1681        /* We cannot use xcalloc() here because that uses getenv() itself */
1682        value = calloc(len_value, sizeof(char));
1683        if (!value)
1684                die("Out of memory, (tried to allocate %u bytes)", len_value);
1685        xwcstoutf(value, w_value, len_value);
1686
1687        /*
1688         * We return `value` which is an allocated value and the caller is NOT
1689         * expecting to have to free it, so we keep a round-robin array,
1690         * invalidating the buffer after GETENV_MAX_RETAIN getenv() calls.
1691         */
1692        free(values[value_counter]);
1693        values[value_counter++] = value;
1694        if (value_counter >= ARRAY_SIZE(values))
1695                value_counter = 0;
1696
1697        return value;
1698}
1699
1700int mingw_putenv(const char *namevalue)
1701{
1702        int size;
1703        wchar_t *wide, *equal;
1704        BOOL result;
1705
1706        if (!namevalue || !*namevalue)
1707                return 0;
1708
1709        size = strlen(namevalue) * 2 + 1;
1710        wide = calloc(size, sizeof(wchar_t));
1711        if (!wide)
1712                die("Out of memory, (tried to allocate %u wchar_t's)", size);
1713        xutftowcs(wide, namevalue, size);
1714        equal = wcschr(wide, L'=');
1715        if (!equal)
1716                result = SetEnvironmentVariableW(wide, NULL);
1717        else {
1718                *equal = L'\0';
1719                result = SetEnvironmentVariableW(wide, equal + 1);
1720        }
1721        free(wide);
1722
1723        if (!result)
1724                errno = err_win_to_posix(GetLastError());
1725
1726        return result ? 0 : -1;
1727}
1728
1729static void ensure_socket_initialization(void)
1730{
1731        WSADATA wsa;
1732        static int initialized = 0;
1733
1734        if (initialized)
1735                return;
1736
1737        if (WSAStartup(MAKEWORD(2,2), &wsa))
1738                die("unable to initialize winsock subsystem, error %d",
1739                        WSAGetLastError());
1740
1741        atexit((void(*)(void)) WSACleanup);
1742        initialized = 1;
1743}
1744
1745#undef gethostname
1746int mingw_gethostname(char *name, int namelen)
1747{
1748    ensure_socket_initialization();
1749    return gethostname(name, namelen);
1750}
1751
1752#undef gethostbyname
1753struct hostent *mingw_gethostbyname(const char *host)
1754{
1755        ensure_socket_initialization();
1756        return gethostbyname(host);
1757}
1758
1759#undef getaddrinfo
1760int mingw_getaddrinfo(const char *node, const char *service,
1761                      const struct addrinfo *hints, struct addrinfo **res)
1762{
1763        ensure_socket_initialization();
1764        return getaddrinfo(node, service, hints, res);
1765}
1766
1767int mingw_socket(int domain, int type, int protocol)
1768{
1769        int sockfd;
1770        SOCKET s;
1771
1772        ensure_socket_initialization();
1773        s = WSASocket(domain, type, protocol, NULL, 0, 0);
1774        if (s == INVALID_SOCKET) {
1775                /*
1776                 * WSAGetLastError() values are regular BSD error codes
1777                 * biased by WSABASEERR.
1778                 * However, strerror() does not know about networking
1779                 * specific errors, which are values beginning at 38 or so.
1780                 * Therefore, we choose to leave the biased error code
1781                 * in errno so that _if_ someone looks up the code somewhere,
1782                 * then it is at least the number that are usually listed.
1783                 */
1784                errno = WSAGetLastError();
1785                return -1;
1786        }
1787        /* convert into a file descriptor */
1788        if ((sockfd = _open_osfhandle(s, O_RDWR|O_BINARY)) < 0) {
1789                closesocket(s);
1790                return error("unable to make a socket file descriptor: %s",
1791                        strerror(errno));
1792        }
1793        return sockfd;
1794}
1795
1796#undef connect
1797int mingw_connect(int sockfd, struct sockaddr *sa, size_t sz)
1798{
1799        SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1800        return connect(s, sa, sz);
1801}
1802
1803#undef bind
1804int mingw_bind(int sockfd, struct sockaddr *sa, size_t sz)
1805{
1806        SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1807        return bind(s, sa, sz);
1808}
1809
1810#undef setsockopt
1811int mingw_setsockopt(int sockfd, int lvl, int optname, void *optval, int optlen)
1812{
1813        SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1814        return setsockopt(s, lvl, optname, (const char*)optval, optlen);
1815}
1816
1817#undef shutdown
1818int mingw_shutdown(int sockfd, int how)
1819{
1820        SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1821        return shutdown(s, how);
1822}
1823
1824#undef listen
1825int mingw_listen(int sockfd, int backlog)
1826{
1827        SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1828        return listen(s, backlog);
1829}
1830
1831#undef accept
1832int mingw_accept(int sockfd1, struct sockaddr *sa, socklen_t *sz)
1833{
1834        int sockfd2;
1835
1836        SOCKET s1 = (SOCKET)_get_osfhandle(sockfd1);
1837        SOCKET s2 = accept(s1, sa, sz);
1838
1839        /* convert into a file descriptor */
1840        if ((sockfd2 = _open_osfhandle(s2, O_RDWR|O_BINARY)) < 0) {
1841                int err = errno;
1842                closesocket(s2);
1843                return error("unable to make a socket file descriptor: %s",
1844                        strerror(err));
1845        }
1846        return sockfd2;
1847}
1848
1849#undef rename
1850int mingw_rename(const char *pold, const char *pnew)
1851{
1852        DWORD attrs, gle;
1853        int tries = 0;
1854        wchar_t wpold[MAX_PATH], wpnew[MAX_PATH];
1855        if (xutftowcs_path(wpold, pold) < 0 || xutftowcs_path(wpnew, pnew) < 0)
1856                return -1;
1857
1858        /*
1859         * Try native rename() first to get errno right.
1860         * It is based on MoveFile(), which cannot overwrite existing files.
1861         */
1862        if (!_wrename(wpold, wpnew))
1863                return 0;
1864        if (errno != EEXIST)
1865                return -1;
1866repeat:
1867        if (MoveFileExW(wpold, wpnew, MOVEFILE_REPLACE_EXISTING))
1868                return 0;
1869        /* TODO: translate more errors */
1870        gle = GetLastError();
1871        if (gle == ERROR_ACCESS_DENIED &&
1872            (attrs = GetFileAttributesW(wpnew)) != INVALID_FILE_ATTRIBUTES) {
1873                if (attrs & FILE_ATTRIBUTE_DIRECTORY) {
1874                        DWORD attrsold = GetFileAttributesW(wpold);
1875                        if (attrsold == INVALID_FILE_ATTRIBUTES ||
1876                            !(attrsold & FILE_ATTRIBUTE_DIRECTORY))
1877                                errno = EISDIR;
1878                        else if (!_wrmdir(wpnew))
1879                                goto repeat;
1880                        return -1;
1881                }
1882                if ((attrs & FILE_ATTRIBUTE_READONLY) &&
1883                    SetFileAttributesW(wpnew, attrs & ~FILE_ATTRIBUTE_READONLY)) {
1884                        if (MoveFileExW(wpold, wpnew, MOVEFILE_REPLACE_EXISTING))
1885                                return 0;
1886                        gle = GetLastError();
1887                        /* revert file attributes on failure */
1888                        SetFileAttributesW(wpnew, attrs);
1889                }
1890        }
1891        if (tries < ARRAY_SIZE(delay) && gle == ERROR_ACCESS_DENIED) {
1892                /*
1893                 * We assume that some other process had the source or
1894                 * destination file open at the wrong moment and retry.
1895                 * In order to give the other process a higher chance to
1896                 * complete its operation, we give up our time slice now.
1897                 * If we have to retry again, we do sleep a bit.
1898                 */
1899                Sleep(delay[tries]);
1900                tries++;
1901                goto repeat;
1902        }
1903        if (gle == ERROR_ACCESS_DENIED &&
1904               ask_yes_no_if_possible("Rename from '%s' to '%s' failed. "
1905                       "Should I try again?", pold, pnew))
1906                goto repeat;
1907
1908        errno = EACCES;
1909        return -1;
1910}
1911
1912/*
1913 * Note that this doesn't return the actual pagesize, but
1914 * the allocation granularity. If future Windows specific git code
1915 * needs the real getpagesize function, we need to find another solution.
1916 */
1917int mingw_getpagesize(void)
1918{
1919        SYSTEM_INFO si;
1920        GetSystemInfo(&si);
1921        return si.dwAllocationGranularity;
1922}
1923
1924/* See https://msdn.microsoft.com/en-us/library/windows/desktop/ms724435.aspx */
1925enum EXTENDED_NAME_FORMAT {
1926        NameDisplay = 3,
1927        NameUserPrincipal = 8
1928};
1929
1930static char *get_extended_user_info(enum EXTENDED_NAME_FORMAT type)
1931{
1932        DECLARE_PROC_ADDR(secur32.dll, BOOL, GetUserNameExW,
1933                enum EXTENDED_NAME_FORMAT, LPCWSTR, PULONG);
1934        static wchar_t wbuffer[1024];
1935        DWORD len;
1936
1937        if (!INIT_PROC_ADDR(GetUserNameExW))
1938                return NULL;
1939
1940        len = ARRAY_SIZE(wbuffer);
1941        if (GetUserNameExW(type, wbuffer, &len)) {
1942                char *converted = xmalloc((len *= 3));
1943                if (xwcstoutf(converted, wbuffer, len) >= 0)
1944                        return converted;
1945                free(converted);
1946        }
1947
1948        return NULL;
1949}
1950
1951char *mingw_query_user_email(void)
1952{
1953        return get_extended_user_info(NameUserPrincipal);
1954}
1955
1956struct passwd *getpwuid(int uid)
1957{
1958        static unsigned initialized;
1959        static char user_name[100];
1960        static struct passwd *p;
1961        wchar_t buf[100];
1962        DWORD len;
1963
1964        if (initialized)
1965                return p;
1966
1967        len = ARRAY_SIZE(buf);
1968        if (!GetUserNameW(buf, &len)) {
1969                initialized = 1;
1970                return NULL;
1971        }
1972
1973        if (xwcstoutf(user_name, buf, sizeof(user_name)) < 0) {
1974                initialized = 1;
1975                return NULL;
1976        }
1977
1978        p = xmalloc(sizeof(*p));
1979        p->pw_name = user_name;
1980        p->pw_gecos = get_extended_user_info(NameDisplay);
1981        if (!p->pw_gecos)
1982                p->pw_gecos = "unknown";
1983        p->pw_dir = NULL;
1984
1985        initialized = 1;
1986        return p;
1987}
1988
1989static HANDLE timer_event;
1990static HANDLE timer_thread;
1991static int timer_interval;
1992static int one_shot;
1993static sig_handler_t timer_fn = SIG_DFL, sigint_fn = SIG_DFL;
1994
1995/* The timer works like this:
1996 * The thread, ticktack(), is a trivial routine that most of the time
1997 * only waits to receive the signal to terminate. The main thread tells
1998 * the thread to terminate by setting the timer_event to the signalled
1999 * state.
2000 * But ticktack() interrupts the wait state after the timer's interval
2001 * length to call the signal handler.
2002 */
2003
2004static unsigned __stdcall ticktack(void *dummy)
2005{
2006        while (WaitForSingleObject(timer_event, timer_interval) == WAIT_TIMEOUT) {
2007                mingw_raise(SIGALRM);
2008                if (one_shot)
2009                        break;
2010        }
2011        return 0;
2012}
2013
2014static int start_timer_thread(void)
2015{
2016        timer_event = CreateEvent(NULL, FALSE, FALSE, NULL);
2017        if (timer_event) {
2018                timer_thread = (HANDLE) _beginthreadex(NULL, 0, ticktack, NULL, 0, NULL);
2019                if (!timer_thread )
2020                        return errno = ENOMEM,
2021                                error("cannot start timer thread");
2022        } else
2023                return errno = ENOMEM,
2024                        error("cannot allocate resources for timer");
2025        return 0;
2026}
2027
2028static void stop_timer_thread(void)
2029{
2030        if (timer_event)
2031                SetEvent(timer_event);  /* tell thread to terminate */
2032        if (timer_thread) {
2033                int rc = WaitForSingleObject(timer_thread, 10000);
2034                if (rc == WAIT_TIMEOUT)
2035                        error("timer thread did not terminate timely");
2036                else if (rc != WAIT_OBJECT_0)
2037                        error("waiting for timer thread failed: %lu",
2038                              GetLastError());
2039                CloseHandle(timer_thread);
2040        }
2041        if (timer_event)
2042                CloseHandle(timer_event);
2043        timer_event = NULL;
2044        timer_thread = NULL;
2045}
2046
2047static inline int is_timeval_eq(const struct timeval *i1, const struct timeval *i2)
2048{
2049        return i1->tv_sec == i2->tv_sec && i1->tv_usec == i2->tv_usec;
2050}
2051
2052int setitimer(int type, struct itimerval *in, struct itimerval *out)
2053{
2054        static const struct timeval zero;
2055        static int atexit_done;
2056
2057        if (out != NULL)
2058                return errno = EINVAL,
2059                        error("setitimer param 3 != NULL not implemented");
2060        if (!is_timeval_eq(&in->it_interval, &zero) &&
2061            !is_timeval_eq(&in->it_interval, &in->it_value))
2062                return errno = EINVAL,
2063                        error("setitimer: it_interval must be zero or eq it_value");
2064
2065        if (timer_thread)
2066                stop_timer_thread();
2067
2068        if (is_timeval_eq(&in->it_value, &zero) &&
2069            is_timeval_eq(&in->it_interval, &zero))
2070                return 0;
2071
2072        timer_interval = in->it_value.tv_sec * 1000 + in->it_value.tv_usec / 1000;
2073        one_shot = is_timeval_eq(&in->it_interval, &zero);
2074        if (!atexit_done) {
2075                atexit(stop_timer_thread);
2076                atexit_done = 1;
2077        }
2078        return start_timer_thread();
2079}
2080
2081int sigaction(int sig, struct sigaction *in, struct sigaction *out)
2082{
2083        if (sig != SIGALRM)
2084                return errno = EINVAL,
2085                        error("sigaction only implemented for SIGALRM");
2086        if (out != NULL)
2087                return errno = EINVAL,
2088                        error("sigaction: param 3 != NULL not implemented");
2089
2090        timer_fn = in->sa_handler;
2091        return 0;
2092}
2093
2094#undef signal
2095sig_handler_t mingw_signal(int sig, sig_handler_t handler)
2096{
2097        sig_handler_t old;
2098
2099        switch (sig) {
2100        case SIGALRM:
2101                old = timer_fn;
2102                timer_fn = handler;
2103                break;
2104
2105        case SIGINT:
2106                old = sigint_fn;
2107                sigint_fn = handler;
2108                break;
2109
2110        default:
2111                return signal(sig, handler);
2112        }
2113
2114        return old;
2115}
2116
2117#undef raise
2118int mingw_raise(int sig)
2119{
2120        switch (sig) {
2121        case SIGALRM:
2122                if (timer_fn == SIG_DFL) {
2123                        if (isatty(STDERR_FILENO))
2124                                fputs("Alarm clock\n", stderr);
2125                        exit(128 + SIGALRM);
2126                } else if (timer_fn != SIG_IGN)
2127                        timer_fn(SIGALRM);
2128                return 0;
2129
2130        case SIGINT:
2131                if (sigint_fn == SIG_DFL)
2132                        exit(128 + SIGINT);
2133                else if (sigint_fn != SIG_IGN)
2134                        sigint_fn(SIGINT);
2135                return 0;
2136
2137#if defined(_MSC_VER)
2138        case SIGILL:
2139        case SIGFPE:
2140        case SIGSEGV:
2141        case SIGTERM:
2142        case SIGBREAK:
2143        case SIGABRT:
2144        case SIGABRT_COMPAT:
2145                /*
2146                 * The <signal.h> header in the MS C Runtime defines 8 signals
2147                 * as being supported on the platform. Anything else causes an
2148                 * "Invalid signal or error" (which in DEBUG builds causes the
2149                 * Abort/Retry/Ignore dialog). We by-pass the CRT for things we
2150                 * already know will fail.
2151                 */
2152                return raise(sig);
2153        default:
2154                errno = EINVAL;
2155                return -1;
2156
2157#else
2158
2159        default:
2160                return raise(sig);
2161
2162#endif
2163
2164        }
2165}
2166
2167int link(const char *oldpath, const char *newpath)
2168{
2169        wchar_t woldpath[MAX_PATH], wnewpath[MAX_PATH];
2170        if (xutftowcs_path(woldpath, oldpath) < 0 ||
2171                xutftowcs_path(wnewpath, newpath) < 0)
2172                return -1;
2173
2174        if (!CreateHardLinkW(wnewpath, woldpath, NULL)) {
2175                errno = err_win_to_posix(GetLastError());
2176                return -1;
2177        }
2178        return 0;
2179}
2180
2181pid_t waitpid(pid_t pid, int *status, int options)
2182{
2183        HANDLE h = OpenProcess(SYNCHRONIZE | PROCESS_QUERY_INFORMATION,
2184            FALSE, pid);
2185        if (!h) {
2186                errno = ECHILD;
2187                return -1;
2188        }
2189
2190        if (pid > 0 && options & WNOHANG) {
2191                if (WAIT_OBJECT_0 != WaitForSingleObject(h, 0)) {
2192                        CloseHandle(h);
2193                        return 0;
2194                }
2195                options &= ~WNOHANG;
2196        }
2197
2198        if (options == 0) {
2199                struct pinfo_t **ppinfo;
2200                if (WaitForSingleObject(h, INFINITE) != WAIT_OBJECT_0) {
2201                        CloseHandle(h);
2202                        return 0;
2203                }
2204
2205                if (status)
2206                        GetExitCodeProcess(h, (LPDWORD)status);
2207
2208                EnterCriticalSection(&pinfo_cs);
2209
2210                ppinfo = &pinfo;
2211                while (*ppinfo) {
2212                        struct pinfo_t *info = *ppinfo;
2213                        if (info->pid == pid) {
2214                                CloseHandle(info->proc);
2215                                *ppinfo = info->next;
2216                                free(info);
2217                                break;
2218                        }
2219                        ppinfo = &info->next;
2220                }
2221
2222                LeaveCriticalSection(&pinfo_cs);
2223
2224                CloseHandle(h);
2225                return pid;
2226        }
2227        CloseHandle(h);
2228
2229        errno = EINVAL;
2230        return -1;
2231}
2232
2233int xutftowcsn(wchar_t *wcs, const char *utfs, size_t wcslen, int utflen)
2234{
2235        int upos = 0, wpos = 0;
2236        const unsigned char *utf = (const unsigned char*) utfs;
2237        if (!utf || !wcs || wcslen < 1) {
2238                errno = EINVAL;
2239                return -1;
2240        }
2241        /* reserve space for \0 */
2242        wcslen--;
2243        if (utflen < 0)
2244                utflen = INT_MAX;
2245
2246        while (upos < utflen) {
2247                int c = utf[upos++] & 0xff;
2248                if (utflen == INT_MAX && c == 0)
2249                        break;
2250
2251                if (wpos >= wcslen) {
2252                        wcs[wpos] = 0;
2253                        errno = ERANGE;
2254                        return -1;
2255                }
2256
2257                if (c < 0x80) {
2258                        /* ASCII */
2259                        wcs[wpos++] = c;
2260                } else if (c >= 0xc2 && c < 0xe0 && upos < utflen &&
2261                                (utf[upos] & 0xc0) == 0x80) {
2262                        /* 2-byte utf-8 */
2263                        c = ((c & 0x1f) << 6);
2264                        c |= (utf[upos++] & 0x3f);
2265                        wcs[wpos++] = c;
2266                } else if (c >= 0xe0 && c < 0xf0 && upos + 1 < utflen &&
2267                                !(c == 0xe0 && utf[upos] < 0xa0) && /* over-long encoding */
2268                                (utf[upos] & 0xc0) == 0x80 &&
2269                                (utf[upos + 1] & 0xc0) == 0x80) {
2270                        /* 3-byte utf-8 */
2271                        c = ((c & 0x0f) << 12);
2272                        c |= ((utf[upos++] & 0x3f) << 6);
2273                        c |= (utf[upos++] & 0x3f);
2274                        wcs[wpos++] = c;
2275                } else if (c >= 0xf0 && c < 0xf5 && upos + 2 < utflen &&
2276                                wpos + 1 < wcslen &&
2277                                !(c == 0xf0 && utf[upos] < 0x90) && /* over-long encoding */
2278                                !(c == 0xf4 && utf[upos] >= 0x90) && /* > \u10ffff */
2279                                (utf[upos] & 0xc0) == 0x80 &&
2280                                (utf[upos + 1] & 0xc0) == 0x80 &&
2281                                (utf[upos + 2] & 0xc0) == 0x80) {
2282                        /* 4-byte utf-8: convert to \ud8xx \udcxx surrogate pair */
2283                        c = ((c & 0x07) << 18);
2284                        c |= ((utf[upos++] & 0x3f) << 12);
2285                        c |= ((utf[upos++] & 0x3f) << 6);
2286                        c |= (utf[upos++] & 0x3f);
2287                        c -= 0x10000;
2288                        wcs[wpos++] = 0xd800 | (c >> 10);
2289                        wcs[wpos++] = 0xdc00 | (c & 0x3ff);
2290                } else if (c >= 0xa0) {
2291                        /* invalid utf-8 byte, printable unicode char: convert 1:1 */
2292                        wcs[wpos++] = c;
2293                } else {
2294                        /* invalid utf-8 byte, non-printable unicode: convert to hex */
2295                        static const char *hex = "0123456789abcdef";
2296                        wcs[wpos++] = hex[c >> 4];
2297                        if (wpos < wcslen)
2298                                wcs[wpos++] = hex[c & 0x0f];
2299                }
2300        }
2301        wcs[wpos] = 0;
2302        return wpos;
2303}
2304
2305int xwcstoutf(char *utf, const wchar_t *wcs, size_t utflen)
2306{
2307        if (!wcs || !utf || utflen < 1) {
2308                errno = EINVAL;
2309                return -1;
2310        }
2311        utflen = WideCharToMultiByte(CP_UTF8, 0, wcs, -1, utf, utflen, NULL, NULL);
2312        if (utflen)
2313                return utflen - 1;
2314        errno = ERANGE;
2315        return -1;
2316}
2317
2318static void setup_windows_environment(void)
2319{
2320        char *tmp = getenv("TMPDIR");
2321
2322        /* on Windows it is TMP and TEMP */
2323        if (!tmp) {
2324                if (!(tmp = getenv("TMP")))
2325                        tmp = getenv("TEMP");
2326                if (tmp) {
2327                        setenv("TMPDIR", tmp, 1);
2328                        tmp = getenv("TMPDIR");
2329                }
2330        }
2331
2332        if (tmp) {
2333                /*
2334                 * Convert all dir separators to forward slashes,
2335                 * to help shell commands called from the Git
2336                 * executable (by not mistaking the dir separators
2337                 * for escape characters).
2338                 */
2339                convert_slashes(tmp);
2340        }
2341
2342        /* simulate TERM to enable auto-color (see color.c) */
2343        if (!getenv("TERM"))
2344                setenv("TERM", "cygwin", 1);
2345
2346        /* calculate HOME if not set */
2347        if (!getenv("HOME")) {
2348                /*
2349                 * try $HOMEDRIVE$HOMEPATH - the home share may be a network
2350                 * location, thus also check if the path exists (i.e. is not
2351                 * disconnected)
2352                 */
2353                if ((tmp = getenv("HOMEDRIVE"))) {
2354                        struct strbuf buf = STRBUF_INIT;
2355                        strbuf_addstr(&buf, tmp);
2356                        if ((tmp = getenv("HOMEPATH"))) {
2357                                strbuf_addstr(&buf, tmp);
2358                                if (is_directory(buf.buf))
2359                                        setenv("HOME", buf.buf, 1);
2360                                else
2361                                        tmp = NULL; /* use $USERPROFILE */
2362                        }
2363                        strbuf_release(&buf);
2364                }
2365                /* use $USERPROFILE if the home share is not available */
2366                if (!tmp && (tmp = getenv("USERPROFILE")))
2367                        setenv("HOME", tmp, 1);
2368        }
2369}
2370
2371#if !defined(_MSC_VER)
2372/*
2373 * Disable MSVCRT command line wildcard expansion (__getmainargs called from
2374 * mingw startup code, see init.c in mingw runtime).
2375 */
2376int _CRT_glob = 0;
2377#endif
2378
2379static NORETURN void die_startup(void)
2380{
2381        fputs("fatal: not enough memory for initialization", stderr);
2382        exit(128);
2383}
2384
2385static void *malloc_startup(size_t size)
2386{
2387        void *result = malloc(size);
2388        if (!result)
2389                die_startup();
2390        return result;
2391}
2392
2393static char *wcstoutfdup_startup(char *buffer, const wchar_t *wcs, size_t len)
2394{
2395        len = xwcstoutf(buffer, wcs, len) + 1;
2396        return memcpy(malloc_startup(len), buffer, len);
2397}
2398
2399static void maybe_redirect_std_handle(const wchar_t *key, DWORD std_id, int fd,
2400                                      DWORD desired_access, DWORD flags)
2401{
2402        DWORD create_flag = fd ? OPEN_ALWAYS : OPEN_EXISTING;
2403        wchar_t buf[MAX_PATH];
2404        DWORD max = ARRAY_SIZE(buf);
2405        HANDLE handle;
2406        DWORD ret = GetEnvironmentVariableW(key, buf, max);
2407
2408        if (!ret || ret >= max)
2409                return;
2410
2411        /* make sure this does not leak into child processes */
2412        SetEnvironmentVariableW(key, NULL);
2413        if (!wcscmp(buf, L"off")) {
2414                close(fd);
2415                handle = GetStdHandle(std_id);
2416                if (handle != INVALID_HANDLE_VALUE)
2417                        CloseHandle(handle);
2418                return;
2419        }
2420        if (std_id == STD_ERROR_HANDLE && !wcscmp(buf, L"2>&1")) {
2421                handle = GetStdHandle(STD_OUTPUT_HANDLE);
2422                if (handle == INVALID_HANDLE_VALUE) {
2423                        close(fd);
2424                        handle = GetStdHandle(std_id);
2425                        if (handle != INVALID_HANDLE_VALUE)
2426                                CloseHandle(handle);
2427                } else {
2428                        int new_fd = _open_osfhandle((intptr_t)handle, O_BINARY);
2429                        SetStdHandle(std_id, handle);
2430                        dup2(new_fd, fd);
2431                        /* do *not* close the new_fd: that would close stdout */
2432                }
2433                return;
2434        }
2435        handle = CreateFileW(buf, desired_access, 0, NULL, create_flag,
2436                             flags, NULL);
2437        if (handle != INVALID_HANDLE_VALUE) {
2438                int new_fd = _open_osfhandle((intptr_t)handle, O_BINARY);
2439                SetStdHandle(std_id, handle);
2440                dup2(new_fd, fd);
2441                close(new_fd);
2442        }
2443}
2444
2445static void maybe_redirect_std_handles(void)
2446{
2447        maybe_redirect_std_handle(L"GIT_REDIRECT_STDIN", STD_INPUT_HANDLE, 0,
2448                                  GENERIC_READ, FILE_ATTRIBUTE_NORMAL);
2449        maybe_redirect_std_handle(L"GIT_REDIRECT_STDOUT", STD_OUTPUT_HANDLE, 1,
2450                                  GENERIC_WRITE, FILE_ATTRIBUTE_NORMAL);
2451        maybe_redirect_std_handle(L"GIT_REDIRECT_STDERR", STD_ERROR_HANDLE, 2,
2452                                  GENERIC_WRITE, FILE_FLAG_NO_BUFFERING);
2453}
2454
2455#ifdef _MSC_VER
2456#ifdef _DEBUG
2457#include <crtdbg.h>
2458#endif
2459#endif
2460
2461/*
2462 * We implement wmain() and compile with -municode, which would
2463 * normally ignore main(), but we call the latter from the former
2464 * so that we can handle non-ASCII command-line parameters
2465 * appropriately.
2466 *
2467 * To be more compatible with the core git code, we convert
2468 * argv into UTF8 and pass them directly to main().
2469 */
2470int wmain(int argc, const wchar_t **wargv)
2471{
2472        int i, maxlen, exit_status;
2473        char *buffer, **save;
2474        const char **argv;
2475
2476        trace2_initialize_clock();
2477
2478#ifdef _MSC_VER
2479#ifdef _DEBUG
2480        _CrtSetReportMode(_CRT_ASSERT, _CRTDBG_MODE_DEBUG);
2481#endif
2482
2483#ifdef USE_MSVC_CRTDBG
2484        _CrtSetDbgFlag(_CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF);
2485#endif
2486#endif
2487
2488        maybe_redirect_std_handles();
2489
2490        /* determine size of argv and environ conversion buffer */
2491        maxlen = wcslen(wargv[0]);
2492        for (i = 1; i < argc; i++)
2493                maxlen = max(maxlen, wcslen(wargv[i]));
2494
2495        /* allocate buffer (wchar_t encodes to max 3 UTF-8 bytes) */
2496        maxlen = 3 * maxlen + 1;
2497        buffer = malloc_startup(maxlen);
2498
2499        /*
2500         * Create a UTF-8 version of w_argv. Also create a "save" copy
2501         * to remember all the string pointers because parse_options()
2502         * will remove claimed items from the argv that we pass down.
2503         */
2504        ALLOC_ARRAY(argv, argc + 1);
2505        ALLOC_ARRAY(save, argc + 1);
2506        for (i = 0; i < argc; i++)
2507                argv[i] = save[i] = wcstoutfdup_startup(buffer, wargv[i], maxlen);
2508        argv[i] = save[i] = NULL;
2509        free(buffer);
2510
2511        /* fix Windows specific environment settings */
2512        setup_windows_environment();
2513
2514        unset_environment_variables = xstrdup("PERL5LIB");
2515
2516        /* initialize critical section for waitpid pinfo_t list */
2517        InitializeCriticalSection(&pinfo_cs);
2518
2519        /* set up default file mode and file modes for stdin/out/err */
2520        _fmode = _O_BINARY;
2521        _setmode(_fileno(stdin), _O_BINARY);
2522        _setmode(_fileno(stdout), _O_BINARY);
2523        _setmode(_fileno(stderr), _O_BINARY);
2524
2525        /* initialize Unicode console */
2526        winansi_init();
2527
2528        /* invoke the real main() using our utf8 version of argv. */
2529        exit_status = main(argc, argv);
2530
2531        for (i = 0; i < argc; i++)
2532                free(save[i]);
2533        free(save);
2534        free(argv);
2535
2536        return exit_status;
2537}
2538
2539int uname(struct utsname *buf)
2540{
2541        unsigned v = (unsigned)GetVersion();
2542        memset(buf, 0, sizeof(*buf));
2543        xsnprintf(buf->sysname, sizeof(buf->sysname), "Windows");
2544        xsnprintf(buf->release, sizeof(buf->release),
2545                 "%u.%u", v & 0xff, (v >> 8) & 0xff);
2546        /* assuming NT variants only.. */
2547        xsnprintf(buf->version, sizeof(buf->version),
2548                  "%u", (v >> 16) & 0x7fff);
2549        return 0;
2550}