431e32265d63028a998c8cdd88af12ec1871e63a
   1#include "../git-compat-util.h"
   2#include "win32.h"
   3#include <conio.h>
   4#include "../strbuf.h"
   5
   6int err_win_to_posix(DWORD winerr)
   7{
   8        int error = ENOSYS;
   9        switch(winerr) {
  10        case ERROR_ACCESS_DENIED: error = EACCES; break;
  11        case ERROR_ACCOUNT_DISABLED: error = EACCES; break;
  12        case ERROR_ACCOUNT_RESTRICTION: error = EACCES; break;
  13        case ERROR_ALREADY_ASSIGNED: error = EBUSY; break;
  14        case ERROR_ALREADY_EXISTS: error = EEXIST; break;
  15        case ERROR_ARITHMETIC_OVERFLOW: error = ERANGE; break;
  16        case ERROR_BAD_COMMAND: error = EIO; break;
  17        case ERROR_BAD_DEVICE: error = ENODEV; break;
  18        case ERROR_BAD_DRIVER_LEVEL: error = ENXIO; break;
  19        case ERROR_BAD_EXE_FORMAT: error = ENOEXEC; break;
  20        case ERROR_BAD_FORMAT: error = ENOEXEC; break;
  21        case ERROR_BAD_LENGTH: error = EINVAL; break;
  22        case ERROR_BAD_PATHNAME: error = ENOENT; break;
  23        case ERROR_BAD_PIPE: error = EPIPE; break;
  24        case ERROR_BAD_UNIT: error = ENODEV; break;
  25        case ERROR_BAD_USERNAME: error = EINVAL; break;
  26        case ERROR_BROKEN_PIPE: error = EPIPE; break;
  27        case ERROR_BUFFER_OVERFLOW: error = ENAMETOOLONG; break;
  28        case ERROR_BUSY: error = EBUSY; break;
  29        case ERROR_BUSY_DRIVE: error = EBUSY; break;
  30        case ERROR_CALL_NOT_IMPLEMENTED: error = ENOSYS; break;
  31        case ERROR_CANNOT_MAKE: error = EACCES; break;
  32        case ERROR_CANTOPEN: error = EIO; break;
  33        case ERROR_CANTREAD: error = EIO; break;
  34        case ERROR_CANTWRITE: error = EIO; break;
  35        case ERROR_CRC: error = EIO; break;
  36        case ERROR_CURRENT_DIRECTORY: error = EACCES; break;
  37        case ERROR_DEVICE_IN_USE: error = EBUSY; break;
  38        case ERROR_DEV_NOT_EXIST: error = ENODEV; break;
  39        case ERROR_DIRECTORY: error = EINVAL; break;
  40        case ERROR_DIR_NOT_EMPTY: error = ENOTEMPTY; break;
  41        case ERROR_DISK_CHANGE: error = EIO; break;
  42        case ERROR_DISK_FULL: error = ENOSPC; break;
  43        case ERROR_DRIVE_LOCKED: error = EBUSY; break;
  44        case ERROR_ENVVAR_NOT_FOUND: error = EINVAL; break;
  45        case ERROR_EXE_MARKED_INVALID: error = ENOEXEC; break;
  46        case ERROR_FILENAME_EXCED_RANGE: error = ENAMETOOLONG; break;
  47        case ERROR_FILE_EXISTS: error = EEXIST; break;
  48        case ERROR_FILE_INVALID: error = ENODEV; break;
  49        case ERROR_FILE_NOT_FOUND: error = ENOENT; break;
  50        case ERROR_GEN_FAILURE: error = EIO; break;
  51        case ERROR_HANDLE_DISK_FULL: error = ENOSPC; break;
  52        case ERROR_INSUFFICIENT_BUFFER: error = ENOMEM; break;
  53        case ERROR_INVALID_ACCESS: error = EACCES; break;
  54        case ERROR_INVALID_ADDRESS: error = EFAULT; break;
  55        case ERROR_INVALID_BLOCK: error = EFAULT; break;
  56        case ERROR_INVALID_DATA: error = EINVAL; break;
  57        case ERROR_INVALID_DRIVE: error = ENODEV; break;
  58        case ERROR_INVALID_EXE_SIGNATURE: error = ENOEXEC; break;
  59        case ERROR_INVALID_FLAGS: error = EINVAL; break;
  60        case ERROR_INVALID_FUNCTION: error = ENOSYS; break;
  61        case ERROR_INVALID_HANDLE: error = EBADF; break;
  62        case ERROR_INVALID_LOGON_HOURS: error = EACCES; break;
  63        case ERROR_INVALID_NAME: error = EINVAL; break;
  64        case ERROR_INVALID_OWNER: error = EINVAL; break;
  65        case ERROR_INVALID_PARAMETER: error = EINVAL; break;
  66        case ERROR_INVALID_PASSWORD: error = EPERM; break;
  67        case ERROR_INVALID_PRIMARY_GROUP: error = EINVAL; break;
  68        case ERROR_INVALID_SIGNAL_NUMBER: error = EINVAL; break;
  69        case ERROR_INVALID_TARGET_HANDLE: error = EIO; break;
  70        case ERROR_INVALID_WORKSTATION: error = EACCES; break;
  71        case ERROR_IO_DEVICE: error = EIO; break;
  72        case ERROR_IO_INCOMPLETE: error = EINTR; break;
  73        case ERROR_LOCKED: error = EBUSY; break;
  74        case ERROR_LOCK_VIOLATION: error = EACCES; break;
  75        case ERROR_LOGON_FAILURE: error = EACCES; break;
  76        case ERROR_MAPPED_ALIGNMENT: error = EINVAL; break;
  77        case ERROR_META_EXPANSION_TOO_LONG: error = E2BIG; break;
  78        case ERROR_MORE_DATA: error = EPIPE; break;
  79        case ERROR_NEGATIVE_SEEK: error = ESPIPE; break;
  80        case ERROR_NOACCESS: error = EFAULT; break;
  81        case ERROR_NONE_MAPPED: error = EINVAL; break;
  82        case ERROR_NOT_ENOUGH_MEMORY: error = ENOMEM; break;
  83        case ERROR_NOT_READY: error = EAGAIN; break;
  84        case ERROR_NOT_SAME_DEVICE: error = EXDEV; break;
  85        case ERROR_NO_DATA: error = EPIPE; break;
  86        case ERROR_NO_MORE_SEARCH_HANDLES: error = EIO; break;
  87        case ERROR_NO_PROC_SLOTS: error = EAGAIN; break;
  88        case ERROR_NO_SUCH_PRIVILEGE: error = EACCES; break;
  89        case ERROR_OPEN_FAILED: error = EIO; break;
  90        case ERROR_OPEN_FILES: error = EBUSY; break;
  91        case ERROR_OPERATION_ABORTED: error = EINTR; break;
  92        case ERROR_OUTOFMEMORY: error = ENOMEM; break;
  93        case ERROR_PASSWORD_EXPIRED: error = EACCES; break;
  94        case ERROR_PATH_BUSY: error = EBUSY; break;
  95        case ERROR_PATH_NOT_FOUND: error = ENOENT; break;
  96        case ERROR_PIPE_BUSY: error = EBUSY; break;
  97        case ERROR_PIPE_CONNECTED: error = EPIPE; break;
  98        case ERROR_PIPE_LISTENING: error = EPIPE; break;
  99        case ERROR_PIPE_NOT_CONNECTED: error = EPIPE; break;
 100        case ERROR_PRIVILEGE_NOT_HELD: error = EACCES; break;
 101        case ERROR_READ_FAULT: error = EIO; break;
 102        case ERROR_SEEK: error = EIO; break;
 103        case ERROR_SEEK_ON_DEVICE: error = ESPIPE; break;
 104        case ERROR_SHARING_BUFFER_EXCEEDED: error = ENFILE; break;
 105        case ERROR_SHARING_VIOLATION: error = EACCES; break;
 106        case ERROR_STACK_OVERFLOW: error = ENOMEM; break;
 107        case ERROR_SWAPERROR: error = ENOENT; break;
 108        case ERROR_TOO_MANY_MODULES: error = EMFILE; break;
 109        case ERROR_TOO_MANY_OPEN_FILES: error = EMFILE; break;
 110        case ERROR_UNRECOGNIZED_MEDIA: error = ENXIO; break;
 111        case ERROR_UNRECOGNIZED_VOLUME: error = ENODEV; break;
 112        case ERROR_WAIT_NO_CHILDREN: error = ECHILD; break;
 113        case ERROR_WRITE_FAULT: error = EIO; break;
 114        case ERROR_WRITE_PROTECT: error = EROFS; break;
 115        }
 116        return error;
 117}
 118
 119#undef open
 120int mingw_open (const char *filename, int oflags, ...)
 121{
 122        va_list args;
 123        unsigned mode;
 124        int fd;
 125
 126        va_start(args, oflags);
 127        mode = va_arg(args, int);
 128        va_end(args);
 129
 130        if (!strcmp(filename, "/dev/null"))
 131                filename = "nul";
 132
 133        fd = open(filename, oflags, mode);
 134
 135        if (fd < 0 && (oflags & O_CREAT) && errno == EACCES) {
 136                DWORD attrs = GetFileAttributes(filename);
 137                if (attrs != INVALID_FILE_ATTRIBUTES && (attrs & FILE_ATTRIBUTE_DIRECTORY))
 138                        errno = EISDIR;
 139        }
 140        return fd;
 141}
 142
 143#undef write
 144ssize_t mingw_write(int fd, const void *buf, size_t count)
 145{
 146        /*
 147         * While write() calls to a file on a local disk are translated
 148         * into WriteFile() calls with a maximum size of 64KB on Windows
 149         * XP and 256KB on Vista, no such cap is placed on writes to
 150         * files over the network on Windows XP.  Unfortunately, there
 151         * seems to be a limit of 32MB-28KB on X64 and 64MB-32KB on x86;
 152         * bigger writes fail on Windows XP.
 153         * So we cap to a nice 31MB here to avoid write failures over
 154         * the net without changing the number of WriteFile() calls in
 155         * the local case.
 156         */
 157        return write(fd, buf, min(count, 31 * 1024 * 1024));
 158}
 159
 160#undef fopen
 161FILE *mingw_fopen (const char *filename, const char *otype)
 162{
 163        if (!strcmp(filename, "/dev/null"))
 164                filename = "nul";
 165        return fopen(filename, otype);
 166}
 167
 168#undef freopen
 169FILE *mingw_freopen (const char *filename, const char *otype, FILE *stream)
 170{
 171        if (filename && !strcmp(filename, "/dev/null"))
 172                filename = "nul";
 173        return freopen(filename, otype, stream);
 174}
 175
 176/*
 177 * The unit of FILETIME is 100-nanoseconds since January 1, 1601, UTC.
 178 * Returns the 100-nanoseconds ("hekto nanoseconds") since the epoch.
 179 */
 180static inline long long filetime_to_hnsec(const FILETIME *ft)
 181{
 182        long long winTime = ((long long)ft->dwHighDateTime << 32) + ft->dwLowDateTime;
 183        /* Windows to Unix Epoch conversion */
 184        return winTime - 116444736000000000LL;
 185}
 186
 187static inline time_t filetime_to_time_t(const FILETIME *ft)
 188{
 189        return (time_t)(filetime_to_hnsec(ft) / 10000000);
 190}
 191
 192/* We keep the do_lstat code in a separate function to avoid recursion.
 193 * When a path ends with a slash, the stat will fail with ENOENT. In
 194 * this case, we strip the trailing slashes and stat again.
 195 *
 196 * If follow is true then act like stat() and report on the link
 197 * target. Otherwise report on the link itself.
 198 */
 199static int do_lstat(int follow, const char *file_name, struct stat *buf)
 200{
 201        WIN32_FILE_ATTRIBUTE_DATA fdata;
 202
 203        if (!(errno = get_file_attr(file_name, &fdata))) {
 204                buf->st_ino = 0;
 205                buf->st_gid = 0;
 206                buf->st_uid = 0;
 207                buf->st_nlink = 1;
 208                buf->st_mode = file_attr_to_st_mode(fdata.dwFileAttributes);
 209                buf->st_size = fdata.nFileSizeLow |
 210                        (((off_t)fdata.nFileSizeHigh)<<32);
 211                buf->st_dev = buf->st_rdev = 0; /* not used by Git */
 212                buf->st_atime = filetime_to_time_t(&(fdata.ftLastAccessTime));
 213                buf->st_mtime = filetime_to_time_t(&(fdata.ftLastWriteTime));
 214                buf->st_ctime = filetime_to_time_t(&(fdata.ftCreationTime));
 215                if (fdata.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
 216                        WIN32_FIND_DATAA findbuf;
 217                        HANDLE handle = FindFirstFileA(file_name, &findbuf);
 218                        if (handle != INVALID_HANDLE_VALUE) {
 219                                if ((findbuf.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) &&
 220                                                (findbuf.dwReserved0 == IO_REPARSE_TAG_SYMLINK)) {
 221                                        if (follow) {
 222                                                char buffer[MAXIMUM_REPARSE_DATA_BUFFER_SIZE];
 223                                                buf->st_size = readlink(file_name, buffer, MAXIMUM_REPARSE_DATA_BUFFER_SIZE);
 224                                        } else {
 225                                                buf->st_mode = S_IFLNK;
 226                                        }
 227                                        buf->st_mode |= S_IREAD;
 228                                        if (!(findbuf.dwFileAttributes & FILE_ATTRIBUTE_READONLY))
 229                                                buf->st_mode |= S_IWRITE;
 230                                }
 231                                FindClose(handle);
 232                        }
 233                }
 234                return 0;
 235        }
 236        return -1;
 237}
 238
 239/* We provide our own lstat/fstat functions, since the provided
 240 * lstat/fstat functions are so slow. These stat functions are
 241 * tailored for Git's usage (read: fast), and are not meant to be
 242 * complete. Note that Git stat()s are redirected to mingw_lstat()
 243 * too, since Windows doesn't really handle symlinks that well.
 244 */
 245static int do_stat_internal(int follow, const char *file_name, struct stat *buf)
 246{
 247        int namelen;
 248        static char alt_name[PATH_MAX];
 249
 250        if (!do_lstat(follow, file_name, buf))
 251                return 0;
 252
 253        /* if file_name ended in a '/', Windows returned ENOENT;
 254         * try again without trailing slashes
 255         */
 256        if (errno != ENOENT)
 257                return -1;
 258
 259        namelen = strlen(file_name);
 260        if (namelen && file_name[namelen-1] != '/')
 261                return -1;
 262        while (namelen && file_name[namelen-1] == '/')
 263                --namelen;
 264        if (!namelen || namelen >= PATH_MAX)
 265                return -1;
 266
 267        memcpy(alt_name, file_name, namelen);
 268        alt_name[namelen] = 0;
 269        return do_lstat(follow, alt_name, buf);
 270}
 271
 272int mingw_lstat(const char *file_name, struct stat *buf)
 273{
 274        return do_stat_internal(0, file_name, buf);
 275}
 276int mingw_stat(const char *file_name, struct stat *buf)
 277{
 278        return do_stat_internal(1, file_name, buf);
 279}
 280
 281#undef fstat
 282int mingw_fstat(int fd, struct stat *buf)
 283{
 284        HANDLE fh = (HANDLE)_get_osfhandle(fd);
 285        BY_HANDLE_FILE_INFORMATION fdata;
 286
 287        if (fh == INVALID_HANDLE_VALUE) {
 288                errno = EBADF;
 289                return -1;
 290        }
 291        /* direct non-file handles to MS's fstat() */
 292        if (GetFileType(fh) != FILE_TYPE_DISK)
 293                return _fstati64(fd, buf);
 294
 295        if (GetFileInformationByHandle(fh, &fdata)) {
 296                buf->st_ino = 0;
 297                buf->st_gid = 0;
 298                buf->st_uid = 0;
 299                buf->st_nlink = 1;
 300                buf->st_mode = file_attr_to_st_mode(fdata.dwFileAttributes);
 301                buf->st_size = fdata.nFileSizeLow |
 302                        (((off_t)fdata.nFileSizeHigh)<<32);
 303                buf->st_dev = buf->st_rdev = 0; /* not used by Git */
 304                buf->st_atime = filetime_to_time_t(&(fdata.ftLastAccessTime));
 305                buf->st_mtime = filetime_to_time_t(&(fdata.ftLastWriteTime));
 306                buf->st_ctime = filetime_to_time_t(&(fdata.ftCreationTime));
 307                return 0;
 308        }
 309        errno = EBADF;
 310        return -1;
 311}
 312
 313static inline void time_t_to_filetime(time_t t, FILETIME *ft)
 314{
 315        long long winTime = t * 10000000LL + 116444736000000000LL;
 316        ft->dwLowDateTime = winTime;
 317        ft->dwHighDateTime = winTime >> 32;
 318}
 319
 320int mingw_utime (const char *file_name, const struct utimbuf *times)
 321{
 322        FILETIME mft, aft;
 323        int fh, rc;
 324
 325        /* must have write permission */
 326        DWORD attrs = GetFileAttributes(file_name);
 327        if (attrs != INVALID_FILE_ATTRIBUTES &&
 328            (attrs & FILE_ATTRIBUTE_READONLY)) {
 329                /* ignore errors here; open() will report them */
 330                SetFileAttributes(file_name, attrs & ~FILE_ATTRIBUTE_READONLY);
 331        }
 332
 333        if ((fh = open(file_name, O_RDWR | O_BINARY)) < 0) {
 334                rc = -1;
 335                goto revert_attrs;
 336        }
 337
 338        if (times) {
 339                time_t_to_filetime(times->modtime, &mft);
 340                time_t_to_filetime(times->actime, &aft);
 341        } else {
 342                GetSystemTimeAsFileTime(&mft);
 343                aft = mft;
 344        }
 345        if (!SetFileTime((HANDLE)_get_osfhandle(fh), NULL, &aft, &mft)) {
 346                errno = EINVAL;
 347                rc = -1;
 348        } else
 349                rc = 0;
 350        close(fh);
 351
 352revert_attrs:
 353        if (attrs != INVALID_FILE_ATTRIBUTES &&
 354            (attrs & FILE_ATTRIBUTE_READONLY)) {
 355                /* ignore errors again */
 356                SetFileAttributes(file_name, attrs);
 357        }
 358        return rc;
 359}
 360
 361unsigned int sleep (unsigned int seconds)
 362{
 363        Sleep(seconds*1000);
 364        return 0;
 365}
 366
 367int mkstemp(char *template)
 368{
 369        char *filename = mktemp(template);
 370        if (filename == NULL)
 371                return -1;
 372        return open(filename, O_RDWR | O_CREAT, 0600);
 373}
 374
 375int gettimeofday(struct timeval *tv, void *tz)
 376{
 377        FILETIME ft;
 378        long long hnsec;
 379
 380        GetSystemTimeAsFileTime(&ft);
 381        hnsec = filetime_to_hnsec(&ft);
 382        tv->tv_sec = hnsec / 10000000;
 383        tv->tv_usec = (hnsec % 10000000) / 10;
 384        return 0;
 385}
 386
 387int pipe(int filedes[2])
 388{
 389        HANDLE h[2];
 390
 391        /* this creates non-inheritable handles */
 392        if (!CreatePipe(&h[0], &h[1], NULL, 8192)) {
 393                errno = err_win_to_posix(GetLastError());
 394                return -1;
 395        }
 396        filedes[0] = _open_osfhandle((int)h[0], O_NOINHERIT);
 397        if (filedes[0] < 0) {
 398                CloseHandle(h[0]);
 399                CloseHandle(h[1]);
 400                return -1;
 401        }
 402        filedes[1] = _open_osfhandle((int)h[1], O_NOINHERIT);
 403        if (filedes[0] < 0) {
 404                close(filedes[0]);
 405                CloseHandle(h[1]);
 406                return -1;
 407        }
 408        return 0;
 409}
 410
 411int poll(struct pollfd *ufds, unsigned int nfds, int timeout)
 412{
 413        int i, pending;
 414
 415        if (timeout >= 0) {
 416                if (nfds == 0) {
 417                        Sleep(timeout);
 418                        return 0;
 419                }
 420                return errno = EINVAL, error("poll timeout not supported");
 421        }
 422
 423        /* When there is only one fd to wait for, then we pretend that
 424         * input is available and let the actual wait happen when the
 425         * caller invokes read().
 426         */
 427        if (nfds == 1) {
 428                if (!(ufds[0].events & POLLIN))
 429                        return errno = EINVAL, error("POLLIN not set");
 430                ufds[0].revents = POLLIN;
 431                return 0;
 432        }
 433
 434repeat:
 435        pending = 0;
 436        for (i = 0; i < nfds; i++) {
 437                DWORD avail = 0;
 438                HANDLE h = (HANDLE) _get_osfhandle(ufds[i].fd);
 439                if (h == INVALID_HANDLE_VALUE)
 440                        return -1;      /* errno was set */
 441
 442                if (!(ufds[i].events & POLLIN))
 443                        return errno = EINVAL, error("POLLIN not set");
 444
 445                /* this emulation works only for pipes */
 446                if (!PeekNamedPipe(h, NULL, 0, NULL, &avail, NULL)) {
 447                        int err = GetLastError();
 448                        if (err == ERROR_BROKEN_PIPE) {
 449                                ufds[i].revents = POLLHUP;
 450                                pending++;
 451                        } else {
 452                                errno = EINVAL;
 453                                return error("PeekNamedPipe failed,"
 454                                        " GetLastError: %u", err);
 455                        }
 456                } else if (avail) {
 457                        ufds[i].revents = POLLIN;
 458                        pending++;
 459                } else
 460                        ufds[i].revents = 0;
 461        }
 462        if (!pending) {
 463                /* The only times that we spin here is when the process
 464                 * that is connected through the pipes is waiting for
 465                 * its own input data to become available. But since
 466                 * the process (pack-objects) is itself CPU intensive,
 467                 * it will happily pick up the time slice that we are
 468                 * relinquishing here.
 469                 */
 470                Sleep(0);
 471                goto repeat;
 472        }
 473        return 0;
 474}
 475
 476struct tm *gmtime_r(const time_t *timep, struct tm *result)
 477{
 478        /* gmtime() in MSVCRT.DLL is thread-safe, but not reentrant */
 479        memcpy(result, gmtime(timep), sizeof(struct tm));
 480        return result;
 481}
 482
 483struct tm *localtime_r(const time_t *timep, struct tm *result)
 484{
 485        /* localtime() in MSVCRT.DLL is thread-safe, but not reentrant */
 486        memcpy(result, localtime(timep), sizeof(struct tm));
 487        return result;
 488}
 489
 490#undef getcwd
 491char *mingw_getcwd(char *pointer, int len)
 492{
 493        int i;
 494        char *ret = getcwd(pointer, len);
 495        if (!ret)
 496                return ret;
 497        for (i = 0; pointer[i]; i++)
 498                if (pointer[i] == '\\')
 499                        pointer[i] = '/';
 500        return ret;
 501}
 502
 503#undef getenv
 504char *mingw_getenv(const char *name)
 505{
 506        char *result = getenv(name);
 507        if (!result && !strcmp(name, "TMPDIR")) {
 508                /* on Windows it is TMP and TEMP */
 509                result = getenv("TMP");
 510                if (!result)
 511                        result = getenv("TEMP");
 512        }
 513        return result;
 514}
 515
 516/*
 517 * See http://msdn2.microsoft.com/en-us/library/17w5ykft(vs.71).aspx
 518 * (Parsing C++ Command-Line Arguments)
 519 */
 520static const char *quote_arg(const char *arg)
 521{
 522        /* count chars to quote */
 523        int len = 0, n = 0;
 524        int force_quotes = 0;
 525        char *q, *d;
 526        const char *p = arg;
 527        if (!*p) force_quotes = 1;
 528        while (*p) {
 529                if (isspace(*p) || *p == '*' || *p == '?' || *p == '{' || *p == '\'')
 530                        force_quotes = 1;
 531                else if (*p == '"')
 532                        n++;
 533                else if (*p == '\\') {
 534                        int count = 0;
 535                        while (*p == '\\') {
 536                                count++;
 537                                p++;
 538                                len++;
 539                        }
 540                        if (*p == '"')
 541                                n += count*2 + 1;
 542                        continue;
 543                }
 544                len++;
 545                p++;
 546        }
 547        if (!force_quotes && n == 0)
 548                return arg;
 549
 550        /* insert \ where necessary */
 551        d = q = xmalloc(len+n+3);
 552        *d++ = '"';
 553        while (*arg) {
 554                if (*arg == '"')
 555                        *d++ = '\\';
 556                else if (*arg == '\\') {
 557                        int count = 0;
 558                        while (*arg == '\\') {
 559                                count++;
 560                                *d++ = *arg++;
 561                        }
 562                        if (*arg == '"') {
 563                                while (count-- > 0)
 564                                        *d++ = '\\';
 565                                *d++ = '\\';
 566                        }
 567                }
 568                *d++ = *arg++;
 569        }
 570        *d++ = '"';
 571        *d++ = 0;
 572        return q;
 573}
 574
 575static const char *parse_interpreter(const char *cmd)
 576{
 577        static char buf[100];
 578        char *p, *opt;
 579        int n, fd;
 580
 581        /* don't even try a .exe */
 582        n = strlen(cmd);
 583        if (n >= 4 && !strcasecmp(cmd+n-4, ".exe"))
 584                return NULL;
 585
 586        fd = open(cmd, O_RDONLY);
 587        if (fd < 0)
 588                return NULL;
 589        n = read(fd, buf, sizeof(buf)-1);
 590        close(fd);
 591        if (n < 4)      /* at least '#!/x' and not error */
 592                return NULL;
 593
 594        if (buf[0] != '#' || buf[1] != '!')
 595                return NULL;
 596        buf[n] = '\0';
 597        p = buf + strcspn(buf, "\r\n");
 598        if (!*p)
 599                return NULL;
 600
 601        *p = '\0';
 602        if (!(p = strrchr(buf+2, '/')) && !(p = strrchr(buf+2, '\\')))
 603                return NULL;
 604        /* strip options */
 605        if ((opt = strchr(p+1, ' ')))
 606                *opt = '\0';
 607        return p+1;
 608}
 609
 610/*
 611 * Splits the PATH into parts.
 612 */
 613static char **get_path_split(void)
 614{
 615        char *p, **path, *envpath = getenv("PATH");
 616        int i, n = 0;
 617
 618        if (!envpath || !*envpath)
 619                return NULL;
 620
 621        envpath = xstrdup(envpath);
 622        p = envpath;
 623        while (p) {
 624                char *dir = p;
 625                p = strchr(p, ';');
 626                if (p) *p++ = '\0';
 627                if (*dir) {     /* not earlier, catches series of ; */
 628                        ++n;
 629                }
 630        }
 631        if (!n)
 632                return NULL;
 633
 634        path = xmalloc((n+1)*sizeof(char *));
 635        p = envpath;
 636        i = 0;
 637        do {
 638                if (*p)
 639                        path[i++] = xstrdup(p);
 640                p = p+strlen(p)+1;
 641        } while (i < n);
 642        path[i] = NULL;
 643
 644        free(envpath);
 645
 646        return path;
 647}
 648
 649static void free_path_split(char **path)
 650{
 651        char **p = path;
 652
 653        if (!path)
 654                return;
 655
 656        while (*p)
 657                free(*p++);
 658        free(path);
 659}
 660
 661/*
 662 * exe_only means that we only want to detect .exe files, but not scripts
 663 * (which do not have an extension)
 664 */
 665static char *lookup_prog(const char *dir, const char *cmd, int isexe, int exe_only)
 666{
 667        char path[MAX_PATH];
 668        snprintf(path, sizeof(path), "%s/%s.exe", dir, cmd);
 669
 670        if (!isexe && access(path, F_OK) == 0)
 671                return xstrdup(path);
 672        path[strlen(path)-4] = '\0';
 673        if ((!exe_only || isexe) && access(path, F_OK) == 0)
 674                if (!(GetFileAttributes(path) & FILE_ATTRIBUTE_DIRECTORY))
 675                        return xstrdup(path);
 676        return NULL;
 677}
 678
 679/*
 680 * Determines the absolute path of cmd using the split path in path.
 681 * If cmd contains a slash or backslash, no lookup is performed.
 682 */
 683static char *path_lookup(const char *cmd, char **path, int exe_only)
 684{
 685        char *prog = NULL;
 686        int len = strlen(cmd);
 687        int isexe = len >= 4 && !strcasecmp(cmd+len-4, ".exe");
 688
 689        if (strchr(cmd, '/') || strchr(cmd, '\\'))
 690                prog = xstrdup(cmd);
 691
 692        while (!prog && *path)
 693                prog = lookup_prog(*path++, cmd, isexe, exe_only);
 694
 695        return prog;
 696}
 697
 698static int env_compare(const void *a, const void *b)
 699{
 700        char *const *ea = a;
 701        char *const *eb = b;
 702        return strcasecmp(*ea, *eb);
 703}
 704
 705static pid_t mingw_spawnve_fd(const char *cmd, const char **argv, char **env,
 706                              const char *dir,
 707                              int prepend_cmd, int fhin, int fhout, int fherr)
 708{
 709        STARTUPINFO si;
 710        PROCESS_INFORMATION pi;
 711        struct strbuf envblk, args;
 712        unsigned flags;
 713        BOOL ret;
 714
 715        /* Determine whether or not we are associated to a console */
 716        HANDLE cons = CreateFile("CONOUT$", GENERIC_WRITE,
 717                        FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
 718                        FILE_ATTRIBUTE_NORMAL, NULL);
 719        if (cons == INVALID_HANDLE_VALUE) {
 720                /* There is no console associated with this process.
 721                 * Since the child is a console process, Windows
 722                 * would normally create a console window. But
 723                 * since we'll be redirecting std streams, we do
 724                 * not need the console.
 725                 * It is necessary to use DETACHED_PROCESS
 726                 * instead of CREATE_NO_WINDOW to make ssh
 727                 * recognize that it has no console.
 728                 */
 729                flags = DETACHED_PROCESS;
 730        } else {
 731                /* There is already a console. If we specified
 732                 * DETACHED_PROCESS here, too, Windows would
 733                 * disassociate the child from the console.
 734                 * The same is true for CREATE_NO_WINDOW.
 735                 * Go figure!
 736                 */
 737                flags = 0;
 738                CloseHandle(cons);
 739        }
 740        memset(&si, 0, sizeof(si));
 741        si.cb = sizeof(si);
 742        si.dwFlags = STARTF_USESTDHANDLES;
 743        si.hStdInput = (HANDLE) _get_osfhandle(fhin);
 744        si.hStdOutput = (HANDLE) _get_osfhandle(fhout);
 745        si.hStdError = (HANDLE) _get_osfhandle(fherr);
 746
 747        /* concatenate argv, quoting args as we go */
 748        strbuf_init(&args, 0);
 749        if (prepend_cmd) {
 750                char *quoted = (char *)quote_arg(cmd);
 751                strbuf_addstr(&args, quoted);
 752                if (quoted != cmd)
 753                        free(quoted);
 754        }
 755        for (; *argv; argv++) {
 756                char *quoted = (char *)quote_arg(*argv);
 757                if (*args.buf)
 758                        strbuf_addch(&args, ' ');
 759                strbuf_addstr(&args, quoted);
 760                if (quoted != *argv)
 761                        free(quoted);
 762        }
 763
 764        if (env) {
 765                int count = 0;
 766                char **e, **sorted_env;
 767
 768                for (e = env; *e; e++)
 769                        count++;
 770
 771                /* environment must be sorted */
 772                sorted_env = xmalloc(sizeof(*sorted_env) * (count + 1));
 773                memcpy(sorted_env, env, sizeof(*sorted_env) * (count + 1));
 774                qsort(sorted_env, count, sizeof(*sorted_env), env_compare);
 775
 776                strbuf_init(&envblk, 0);
 777                for (e = sorted_env; *e; e++) {
 778                        strbuf_addstr(&envblk, *e);
 779                        strbuf_addch(&envblk, '\0');
 780                }
 781                free(sorted_env);
 782        }
 783
 784        memset(&pi, 0, sizeof(pi));
 785        ret = CreateProcess(cmd, args.buf, NULL, NULL, TRUE, flags,
 786                env ? envblk.buf : NULL, dir, &si, &pi);
 787
 788        if (env)
 789                strbuf_release(&envblk);
 790        strbuf_release(&args);
 791
 792        if (!ret) {
 793                errno = ENOENT;
 794                return -1;
 795        }
 796        CloseHandle(pi.hThread);
 797        return (pid_t)pi.hProcess;
 798}
 799
 800static pid_t mingw_spawnve(const char *cmd, const char **argv, char **env,
 801                           int prepend_cmd)
 802{
 803        return mingw_spawnve_fd(cmd, argv, env, NULL, prepend_cmd, 0, 1, 2);
 804}
 805
 806pid_t mingw_spawnvpe(const char *cmd, const char **argv, char **env,
 807                     const char *dir,
 808                     int fhin, int fhout, int fherr)
 809{
 810        pid_t pid;
 811        char **path = get_path_split();
 812        char *prog = path_lookup(cmd, path, 0);
 813
 814        if (!prog) {
 815                errno = ENOENT;
 816                pid = -1;
 817        }
 818        else {
 819                const char *interpr = parse_interpreter(prog);
 820
 821                if (interpr) {
 822                        const char *argv0 = argv[0];
 823                        char *iprog = path_lookup(interpr, path, 1);
 824                        argv[0] = prog;
 825                        if (!iprog) {
 826                                errno = ENOENT;
 827                                pid = -1;
 828                        }
 829                        else {
 830                                pid = mingw_spawnve_fd(iprog, argv, env, dir, 1,
 831                                                       fhin, fhout, fherr);
 832                                free(iprog);
 833                        }
 834                        argv[0] = argv0;
 835                }
 836                else
 837                        pid = mingw_spawnve_fd(prog, argv, env, dir, 0,
 838                                               fhin, fhout, fherr);
 839                free(prog);
 840        }
 841        free_path_split(path);
 842        return pid;
 843}
 844
 845static int try_shell_exec(const char *cmd, char *const *argv, char **env)
 846{
 847        const char *interpr = parse_interpreter(cmd);
 848        char **path;
 849        char *prog;
 850        int pid = 0;
 851
 852        if (!interpr)
 853                return 0;
 854        path = get_path_split();
 855        prog = path_lookup(interpr, path, 1);
 856        if (prog) {
 857                int argc = 0;
 858                const char **argv2;
 859                while (argv[argc]) argc++;
 860                argv2 = xmalloc(sizeof(*argv) * (argc+1));
 861                argv2[0] = (char *)cmd; /* full path to the script file */
 862                memcpy(&argv2[1], &argv[1], sizeof(*argv) * argc);
 863                pid = mingw_spawnve(prog, argv2, env, 1);
 864                if (pid >= 0) {
 865                        int status;
 866                        if (waitpid(pid, &status, 0) < 0)
 867                                status = 255;
 868                        exit(status);
 869                }
 870                pid = 1;        /* indicate that we tried but failed */
 871                free(prog);
 872                free(argv2);
 873        }
 874        free_path_split(path);
 875        return pid;
 876}
 877
 878static void mingw_execve(const char *cmd, char *const *argv, char *const *env)
 879{
 880        /* check if git_command is a shell script */
 881        if (!try_shell_exec(cmd, argv, (char **)env)) {
 882                int pid, status;
 883
 884                pid = mingw_spawnve(cmd, (const char **)argv, (char **)env, 0);
 885                if (pid < 0)
 886                        return;
 887                if (waitpid(pid, &status, 0) < 0)
 888                        status = 255;
 889                exit(status);
 890        }
 891}
 892
 893void mingw_execvp(const char *cmd, char *const *argv)
 894{
 895        char **path = get_path_split();
 896        char *prog = path_lookup(cmd, path, 0);
 897
 898        if (prog) {
 899                mingw_execve(prog, argv, environ);
 900                free(prog);
 901        } else
 902                errno = ENOENT;
 903
 904        free_path_split(path);
 905}
 906
 907static char **copy_environ(void)
 908{
 909        char **env;
 910        int i = 0;
 911        while (environ[i])
 912                i++;
 913        env = xmalloc((i+1)*sizeof(*env));
 914        for (i = 0; environ[i]; i++)
 915                env[i] = xstrdup(environ[i]);
 916        env[i] = NULL;
 917        return env;
 918}
 919
 920void free_environ(char **env)
 921{
 922        int i;
 923        for (i = 0; env[i]; i++)
 924                free(env[i]);
 925        free(env);
 926}
 927
 928static int lookup_env(char **env, const char *name, size_t nmln)
 929{
 930        int i;
 931
 932        for (i = 0; env[i]; i++) {
 933                if (0 == strncmp(env[i], name, nmln)
 934                    && '=' == env[i][nmln])
 935                        /* matches */
 936                        return i;
 937        }
 938        return -1;
 939}
 940
 941/*
 942 * If name contains '=', then sets the variable, otherwise it unsets it
 943 */
 944static char **env_setenv(char **env, const char *name)
 945{
 946        char *eq = strchrnul(name, '=');
 947        int i = lookup_env(env, name, eq-name);
 948
 949        if (i < 0) {
 950                if (*eq) {
 951                        for (i = 0; env[i]; i++)
 952                                ;
 953                        env = xrealloc(env, (i+2)*sizeof(*env));
 954                        env[i] = xstrdup(name);
 955                        env[i+1] = NULL;
 956                }
 957        }
 958        else {
 959                free(env[i]);
 960                if (*eq)
 961                        env[i] = xstrdup(name);
 962                else
 963                        for (; env[i]; i++)
 964                                env[i] = env[i+1];
 965        }
 966        return env;
 967}
 968
 969/*
 970 * Copies global environ and adjusts variables as specified by vars.
 971 */
 972char **make_augmented_environ(const char *const *vars)
 973{
 974        char **env = copy_environ();
 975
 976        while (*vars)
 977                env = env_setenv(env, *vars++);
 978        return env;
 979}
 980
 981/*
 982 * Note, this isn't a complete replacement for getaddrinfo. It assumes
 983 * that service contains a numerical port, or that it it is null. It
 984 * does a simple search using gethostbyname, and returns one IPv4 host
 985 * if one was found.
 986 */
 987static int WSAAPI getaddrinfo_stub(const char *node, const char *service,
 988                                   const struct addrinfo *hints,
 989                                   struct addrinfo **res)
 990{
 991        struct hostent *h = gethostbyname(node);
 992        struct addrinfo *ai;
 993        struct sockaddr_in *sin;
 994
 995        if (!h)
 996                return WSAGetLastError();
 997
 998        ai = xmalloc(sizeof(struct addrinfo));
 999        *res = ai;
1000        ai->ai_flags = 0;
1001        ai->ai_family = AF_INET;
1002        ai->ai_socktype = hints->ai_socktype;
1003        switch (hints->ai_socktype) {
1004        case SOCK_STREAM:
1005                ai->ai_protocol = IPPROTO_TCP;
1006                break;
1007        case SOCK_DGRAM:
1008                ai->ai_protocol = IPPROTO_UDP;
1009                break;
1010        default:
1011                ai->ai_protocol = 0;
1012                break;
1013        }
1014        ai->ai_addrlen = sizeof(struct sockaddr_in);
1015        ai->ai_canonname = strdup(h->h_name);
1016
1017        sin = xmalloc(ai->ai_addrlen);
1018        memset(sin, 0, ai->ai_addrlen);
1019        sin->sin_family = AF_INET;
1020        if (service)
1021                sin->sin_port = htons(atoi(service));
1022        sin->sin_addr = *(struct in_addr *)h->h_addr;
1023        ai->ai_addr = (struct sockaddr *)sin;
1024        ai->ai_next = 0;
1025        return 0;
1026}
1027
1028static void WSAAPI freeaddrinfo_stub(struct addrinfo *res)
1029{
1030        free(res->ai_canonname);
1031        free(res->ai_addr);
1032        free(res);
1033}
1034
1035static int WSAAPI getnameinfo_stub(const struct sockaddr *sa, socklen_t salen,
1036                                   char *host, DWORD hostlen,
1037                                   char *serv, DWORD servlen, int flags)
1038{
1039        const struct sockaddr_in *sin = (const struct sockaddr_in *)sa;
1040        if (sa->sa_family != AF_INET)
1041                return EAI_FAMILY;
1042        if (!host && !serv)
1043                return EAI_NONAME;
1044
1045        if (host && hostlen > 0) {
1046                struct hostent *ent = NULL;
1047                if (!(flags & NI_NUMERICHOST))
1048                        ent = gethostbyaddr((const char *)&sin->sin_addr,
1049                                            sizeof(sin->sin_addr), AF_INET);
1050
1051                if (ent)
1052                        snprintf(host, hostlen, "%s", ent->h_name);
1053                else if (flags & NI_NAMEREQD)
1054                        return EAI_NONAME;
1055                else
1056                        snprintf(host, hostlen, "%s", inet_ntoa(sin->sin_addr));
1057        }
1058
1059        if (serv && servlen > 0) {
1060                struct servent *ent = NULL;
1061                if (!(flags & NI_NUMERICSERV))
1062                        ent = getservbyport(sin->sin_port,
1063                                            flags & NI_DGRAM ? "udp" : "tcp");
1064
1065                if (ent)
1066                        snprintf(serv, servlen, "%s", ent->s_name);
1067                else
1068                        snprintf(serv, servlen, "%d", ntohs(sin->sin_port));
1069        }
1070
1071        return 0;
1072}
1073
1074static HMODULE ipv6_dll = NULL;
1075static void (WSAAPI *ipv6_freeaddrinfo)(struct addrinfo *res);
1076static int (WSAAPI *ipv6_getaddrinfo)(const char *node, const char *service,
1077                                      const struct addrinfo *hints,
1078                                      struct addrinfo **res);
1079static int (WSAAPI *ipv6_getnameinfo)(const struct sockaddr *sa, socklen_t salen,
1080                                      char *host, DWORD hostlen,
1081                                      char *serv, DWORD servlen, int flags);
1082/*
1083 * gai_strerror is an inline function in the ws2tcpip.h header, so we
1084 * don't need to try to load that one dynamically.
1085 */
1086
1087static void socket_cleanup(void)
1088{
1089        WSACleanup();
1090        if (ipv6_dll)
1091                FreeLibrary(ipv6_dll);
1092        ipv6_dll = NULL;
1093        ipv6_freeaddrinfo = freeaddrinfo_stub;
1094        ipv6_getaddrinfo = getaddrinfo_stub;
1095        ipv6_getnameinfo = getnameinfo_stub;
1096}
1097
1098static void ensure_socket_initialization(void)
1099{
1100        WSADATA wsa;
1101        static int initialized = 0;
1102        const char *libraries[] = { "ws2_32.dll", "wship6.dll", NULL };
1103        const char **name;
1104
1105        if (initialized)
1106                return;
1107
1108        if (WSAStartup(MAKEWORD(2,2), &wsa))
1109                die("unable to initialize winsock subsystem, error %d",
1110                        WSAGetLastError());
1111
1112        for (name = libraries; *name; name++) {
1113                ipv6_dll = LoadLibrary(*name);
1114                if (!ipv6_dll)
1115                        continue;
1116
1117                ipv6_freeaddrinfo = (void (WSAAPI *)(struct addrinfo *))
1118                        GetProcAddress(ipv6_dll, "freeaddrinfo");
1119                ipv6_getaddrinfo = (int (WSAAPI *)(const char *, const char *,
1120                                                   const struct addrinfo *,
1121                                                   struct addrinfo **))
1122                        GetProcAddress(ipv6_dll, "getaddrinfo");
1123                ipv6_getnameinfo = (int (WSAAPI *)(const struct sockaddr *,
1124                                                   socklen_t, char *, DWORD,
1125                                                   char *, DWORD, int))
1126                        GetProcAddress(ipv6_dll, "getnameinfo");
1127                if (!ipv6_freeaddrinfo || !ipv6_getaddrinfo || !ipv6_getnameinfo) {
1128                        FreeLibrary(ipv6_dll);
1129                        ipv6_dll = NULL;
1130                } else
1131                        break;
1132        }
1133        if (!ipv6_freeaddrinfo || !ipv6_getaddrinfo || !ipv6_getnameinfo) {
1134                ipv6_freeaddrinfo = freeaddrinfo_stub;
1135                ipv6_getaddrinfo = getaddrinfo_stub;
1136                ipv6_getnameinfo = getnameinfo_stub;
1137        }
1138
1139        atexit(socket_cleanup);
1140        initialized = 1;
1141}
1142
1143#undef gethostbyname
1144struct hostent *mingw_gethostbyname(const char *host)
1145{
1146        ensure_socket_initialization();
1147        return gethostbyname(host);
1148}
1149
1150void mingw_freeaddrinfo(struct addrinfo *res)
1151{
1152        ipv6_freeaddrinfo(res);
1153}
1154
1155int mingw_getaddrinfo(const char *node, const char *service,
1156                      const struct addrinfo *hints, struct addrinfo **res)
1157{
1158        ensure_socket_initialization();
1159        return ipv6_getaddrinfo(node, service, hints, res);
1160}
1161
1162int mingw_getnameinfo(const struct sockaddr *sa, socklen_t salen,
1163                      char *host, DWORD hostlen, char *serv, DWORD servlen,
1164                      int flags)
1165{
1166        ensure_socket_initialization();
1167        return ipv6_getnameinfo(sa, salen, host, hostlen, serv, servlen, flags);
1168}
1169
1170int mingw_socket(int domain, int type, int protocol)
1171{
1172        int sockfd;
1173        SOCKET s = WSASocket(domain, type, protocol, NULL, 0, 0);
1174        if (s == INVALID_SOCKET) {
1175                /*
1176                 * WSAGetLastError() values are regular BSD error codes
1177                 * biased by WSABASEERR.
1178                 * However, strerror() does not know about networking
1179                 * specific errors, which are values beginning at 38 or so.
1180                 * Therefore, we choose to leave the biased error code
1181                 * in errno so that _if_ someone looks up the code somewhere,
1182                 * then it is at least the number that are usually listed.
1183                 */
1184                errno = WSAGetLastError();
1185                return -1;
1186        }
1187        /* convert into a file descriptor */
1188        if ((sockfd = _open_osfhandle(s, O_RDWR|O_BINARY)) < 0) {
1189                closesocket(s);
1190                return error("unable to make a socket file descriptor: %s",
1191                        strerror(errno));
1192        }
1193        return sockfd;
1194}
1195
1196#undef connect
1197int mingw_connect(int sockfd, struct sockaddr *sa, size_t sz)
1198{
1199        SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1200        return connect(s, sa, sz);
1201}
1202
1203#undef rename
1204int mingw_rename(const char *pold, const char *pnew)
1205{
1206        DWORD attrs, gle;
1207        int tries = 0;
1208        static const int delay[] = { 0, 1, 10, 20, 40 };
1209
1210        /*
1211         * Try native rename() first to get errno right.
1212         * It is based on MoveFile(), which cannot overwrite existing files.
1213         */
1214        if (!rename(pold, pnew))
1215                return 0;
1216        if (errno != EEXIST)
1217                return -1;
1218repeat:
1219        if (MoveFileEx(pold, pnew, MOVEFILE_REPLACE_EXISTING))
1220                return 0;
1221        /* TODO: translate more errors */
1222        gle = GetLastError();
1223        if (gle == ERROR_ACCESS_DENIED &&
1224            (attrs = GetFileAttributes(pnew)) != INVALID_FILE_ATTRIBUTES) {
1225                if (attrs & FILE_ATTRIBUTE_DIRECTORY) {
1226                        errno = EISDIR;
1227                        return -1;
1228                }
1229                if ((attrs & FILE_ATTRIBUTE_READONLY) &&
1230                    SetFileAttributes(pnew, attrs & ~FILE_ATTRIBUTE_READONLY)) {
1231                        if (MoveFileEx(pold, pnew, MOVEFILE_REPLACE_EXISTING))
1232                                return 0;
1233                        gle = GetLastError();
1234                        /* revert file attributes on failure */
1235                        SetFileAttributes(pnew, attrs);
1236                }
1237        }
1238        if (tries < ARRAY_SIZE(delay) && gle == ERROR_ACCESS_DENIED) {
1239                /*
1240                 * We assume that some other process had the source or
1241                 * destination file open at the wrong moment and retry.
1242                 * In order to give the other process a higher chance to
1243                 * complete its operation, we give up our time slice now.
1244                 * If we have to retry again, we do sleep a bit.
1245                 */
1246                Sleep(delay[tries]);
1247                tries++;
1248                goto repeat;
1249        }
1250        errno = EACCES;
1251        return -1;
1252}
1253
1254/*
1255 * Note that this doesn't return the actual pagesize, but
1256 * the allocation granularity. If future Windows specific git code
1257 * needs the real getpagesize function, we need to find another solution.
1258 */
1259int mingw_getpagesize(void)
1260{
1261        SYSTEM_INFO si;
1262        GetSystemInfo(&si);
1263        return si.dwAllocationGranularity;
1264}
1265
1266struct passwd *getpwuid(int uid)
1267{
1268        static char user_name[100];
1269        static struct passwd p;
1270
1271        DWORD len = sizeof(user_name);
1272        if (!GetUserName(user_name, &len))
1273                return NULL;
1274        p.pw_name = user_name;
1275        p.pw_gecos = "unknown";
1276        p.pw_dir = NULL;
1277        return &p;
1278}
1279
1280static HANDLE timer_event;
1281static HANDLE timer_thread;
1282static int timer_interval;
1283static int one_shot;
1284static sig_handler_t timer_fn = SIG_DFL;
1285
1286/* The timer works like this:
1287 * The thread, ticktack(), is a trivial routine that most of the time
1288 * only waits to receive the signal to terminate. The main thread tells
1289 * the thread to terminate by setting the timer_event to the signalled
1290 * state.
1291 * But ticktack() interrupts the wait state after the timer's interval
1292 * length to call the signal handler.
1293 */
1294
1295static unsigned __stdcall ticktack(void *dummy)
1296{
1297        while (WaitForSingleObject(timer_event, timer_interval) == WAIT_TIMEOUT) {
1298                if (timer_fn == SIG_DFL)
1299                        die("Alarm");
1300                if (timer_fn != SIG_IGN)
1301                        timer_fn(SIGALRM);
1302                if (one_shot)
1303                        break;
1304        }
1305        return 0;
1306}
1307
1308static int start_timer_thread(void)
1309{
1310        timer_event = CreateEvent(NULL, FALSE, FALSE, NULL);
1311        if (timer_event) {
1312                timer_thread = (HANDLE) _beginthreadex(NULL, 0, ticktack, NULL, 0, NULL);
1313                if (!timer_thread )
1314                        return errno = ENOMEM,
1315                                error("cannot start timer thread");
1316        } else
1317                return errno = ENOMEM,
1318                        error("cannot allocate resources for timer");
1319        return 0;
1320}
1321
1322static void stop_timer_thread(void)
1323{
1324        if (timer_event)
1325                SetEvent(timer_event);  /* tell thread to terminate */
1326        if (timer_thread) {
1327                int rc = WaitForSingleObject(timer_thread, 1000);
1328                if (rc == WAIT_TIMEOUT)
1329                        error("timer thread did not terminate timely");
1330                else if (rc != WAIT_OBJECT_0)
1331                        error("waiting for timer thread failed: %lu",
1332                              GetLastError());
1333                CloseHandle(timer_thread);
1334        }
1335        if (timer_event)
1336                CloseHandle(timer_event);
1337        timer_event = NULL;
1338        timer_thread = NULL;
1339}
1340
1341static inline int is_timeval_eq(const struct timeval *i1, const struct timeval *i2)
1342{
1343        return i1->tv_sec == i2->tv_sec && i1->tv_usec == i2->tv_usec;
1344}
1345
1346int setitimer(int type, struct itimerval *in, struct itimerval *out)
1347{
1348        static const struct timeval zero;
1349        static int atexit_done;
1350
1351        if (out != NULL)
1352                return errno = EINVAL,
1353                        error("setitimer param 3 != NULL not implemented");
1354        if (!is_timeval_eq(&in->it_interval, &zero) &&
1355            !is_timeval_eq(&in->it_interval, &in->it_value))
1356                return errno = EINVAL,
1357                        error("setitimer: it_interval must be zero or eq it_value");
1358
1359        if (timer_thread)
1360                stop_timer_thread();
1361
1362        if (is_timeval_eq(&in->it_value, &zero) &&
1363            is_timeval_eq(&in->it_interval, &zero))
1364                return 0;
1365
1366        timer_interval = in->it_value.tv_sec * 1000 + in->it_value.tv_usec / 1000;
1367        one_shot = is_timeval_eq(&in->it_interval, &zero);
1368        if (!atexit_done) {
1369                atexit(stop_timer_thread);
1370                atexit_done = 1;
1371        }
1372        return start_timer_thread();
1373}
1374
1375int sigaction(int sig, struct sigaction *in, struct sigaction *out)
1376{
1377        if (sig != SIGALRM)
1378                return errno = EINVAL,
1379                        error("sigaction only implemented for SIGALRM");
1380        if (out != NULL)
1381                return errno = EINVAL,
1382                        error("sigaction: param 3 != NULL not implemented");
1383
1384        timer_fn = in->sa_handler;
1385        return 0;
1386}
1387
1388#undef signal
1389sig_handler_t mingw_signal(int sig, sig_handler_t handler)
1390{
1391        sig_handler_t old = timer_fn;
1392        if (sig != SIGALRM)
1393                return signal(sig, handler);
1394        timer_fn = handler;
1395        return old;
1396}
1397
1398static const char *make_backslash_path(const char *path)
1399{
1400        static char buf[PATH_MAX + 1];
1401        char *c;
1402
1403        if (strlcpy(buf, path, PATH_MAX) >= PATH_MAX)
1404                die("Too long path: %.*s", 60, path);
1405
1406        for (c = buf; *c; c++) {
1407                if (*c == '/')
1408                        *c = '\\';
1409        }
1410        return buf;
1411}
1412
1413void mingw_open_html(const char *unixpath)
1414{
1415        const char *htmlpath = make_backslash_path(unixpath);
1416        typedef HINSTANCE (WINAPI *T)(HWND, const char *,
1417                        const char *, const char *, const char *, INT);
1418        T ShellExecute;
1419        HMODULE shell32;
1420        int r;
1421
1422        shell32 = LoadLibrary("shell32.dll");
1423        if (!shell32)
1424                die("cannot load shell32.dll");
1425        ShellExecute = (T)GetProcAddress(shell32, "ShellExecuteA");
1426        if (!ShellExecute)
1427                die("cannot run browser");
1428
1429        printf("Launching default browser to display HTML ...\n");
1430        r = (int)ShellExecute(NULL, "open", htmlpath, NULL, "\\", SW_SHOWNORMAL);
1431        FreeLibrary(shell32);
1432        /* see the MSDN documentation referring to the result codes here */
1433        if (r <= 32) {
1434                die("failed to launch browser for %.*s", MAX_PATH, unixpath);
1435        }
1436}
1437
1438int link(const char *oldpath, const char *newpath)
1439{
1440        typedef BOOL (WINAPI *T)(const char*, const char*, LPSECURITY_ATTRIBUTES);
1441        static T create_hard_link = NULL;
1442        if (!create_hard_link) {
1443                create_hard_link = (T) GetProcAddress(
1444                        GetModuleHandle("kernel32.dll"), "CreateHardLinkA");
1445                if (!create_hard_link)
1446                        create_hard_link = (T)-1;
1447        }
1448        if (create_hard_link == (T)-1) {
1449                errno = ENOSYS;
1450                return -1;
1451        }
1452        if (!create_hard_link(newpath, oldpath, NULL)) {
1453                errno = err_win_to_posix(GetLastError());
1454                return -1;
1455        }
1456        return 0;
1457}
1458
1459char *getpass(const char *prompt)
1460{
1461        struct strbuf buf = STRBUF_INIT;
1462
1463        fputs(prompt, stderr);
1464        for (;;) {
1465                char c = _getch();
1466                if (c == '\r' || c == '\n')
1467                        break;
1468                strbuf_addch(&buf, c);
1469        }
1470        fputs("\n", stderr);
1471        return strbuf_detach(&buf, NULL);
1472}
1473
1474#ifndef NO_MINGW_REPLACE_READDIR
1475/* MinGW readdir implementation to avoid extra lstats for Git */
1476struct mingw_DIR
1477{
1478        struct _finddata_t      dd_dta;         /* disk transfer area for this dir */
1479        struct mingw_dirent     dd_dir;         /* Our own implementation, including d_type */
1480        long                    dd_handle;      /* _findnext handle */
1481        int                     dd_stat;        /* 0 = next entry to read is first entry, -1 = off the end, positive = 0 based index of next entry */
1482        char                    dd_name[1];     /* given path for dir with search pattern (struct is extended) */
1483};
1484
1485struct dirent *mingw_readdir(DIR *dir)
1486{
1487        WIN32_FIND_DATAA buf;
1488        HANDLE handle;
1489        struct mingw_DIR *mdir = (struct mingw_DIR*)dir;
1490
1491        if (!dir->dd_handle) {
1492                errno = EBADF; /* No set_errno for mingw */
1493                return NULL;
1494        }
1495
1496        if (dir->dd_handle == (long)INVALID_HANDLE_VALUE && dir->dd_stat == 0)
1497        {
1498                DWORD lasterr;
1499                handle = FindFirstFileA(dir->dd_name, &buf);
1500                lasterr = GetLastError();
1501                dir->dd_handle = (long)handle;
1502                if (handle == INVALID_HANDLE_VALUE && (lasterr != ERROR_NO_MORE_FILES)) {
1503                        errno = err_win_to_posix(lasterr);
1504                        return NULL;
1505                }
1506        } else if (dir->dd_handle == (long)INVALID_HANDLE_VALUE) {
1507                return NULL;
1508        } else if (!FindNextFileA((HANDLE)dir->dd_handle, &buf)) {
1509                DWORD lasterr = GetLastError();
1510                FindClose((HANDLE)dir->dd_handle);
1511                dir->dd_handle = (long)INVALID_HANDLE_VALUE;
1512                /* POSIX says you shouldn't set errno when readdir can't
1513                   find any more files; so, if another error we leave it set. */
1514                if (lasterr != ERROR_NO_MORE_FILES)
1515                        errno = err_win_to_posix(lasterr);
1516                return NULL;
1517        }
1518
1519        /* We get here if `buf' contains valid data.  */
1520        strcpy(dir->dd_dir.d_name, buf.cFileName);
1521        ++dir->dd_stat;
1522
1523        /* Set file type, based on WIN32_FIND_DATA */
1524        mdir->dd_dir.d_type = 0;
1525        if (buf.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
1526                mdir->dd_dir.d_type |= DT_DIR;
1527        else
1528                mdir->dd_dir.d_type |= DT_REG;
1529
1530        return (struct dirent*)&dir->dd_dir;
1531}
1532#endif // !NO_MINGW_REPLACE_READDIR