compat/mingw.c

   1 #include "../git-compat-util.h"
   2 #include "win32.h"
   3 #include <conio.h>
   4 #include "../strbuf.h"
   5
   6 int 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
 120 int 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 (filename && !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
 144 ssize_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
 161 FILE *mingw_fopen (const char *filename, const char *otype)
 162 {
 163         if (filename && !strcmp(filename, "/dev/null"))
 164                 filename = "nul";
 165         return fopen(filename, otype);
 166 }
 167
 168 #undef freopen
 169 FILE *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  */
 180 static 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
 187 static 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  */
 199 static 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  */
 245 static 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
 272 int mingw_lstat(const char *file_name, struct stat *buf)
 273 {
 274         return do_stat_internal(0, file_name, buf);
 275 }
 276 int mingw_stat(const char *file_name, struct stat *buf)
 277 {
 278         return do_stat_internal(1, file_name, buf);
 279 }
 280
 281 #undef fstat
 282 int 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
 313 static 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
 320 int 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
 352 revert_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
 361 unsigned int sleep (unsigned int seconds)
 362 {
 363         Sleep(seconds*1000);
 364         return 0;
 365 }
 366
 367 int 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
 375 int 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
 387 int 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
 411 int 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
 434 repeat:
 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
 476 struct 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
 483 struct 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
 491 char *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
 504 char *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  */
 520 static 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
 575 static 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  */
 613 static 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
 649 static 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  */
 665 static 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  */
 683 static 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
 698 static 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
 705 struct pinfo_t {
 706         struct pinfo_t *next;
 707         pid_t pid;
 708         HANDLE proc;
 709 } pinfo_t;
 710 struct pinfo_t *pinfo = NULL;
 711 CRITICAL_SECTION pinfo_cs;
 712
 713 static pid_t mingw_spawnve_fd(const char *cmd, const char **argv, char **env,
 714                               const char *dir,
 715                               int prepend_cmd, int fhin, int fhout, int fherr)
 716 {
 717         STARTUPINFO si;
 718         PROCESS_INFORMATION pi;
 719         struct strbuf envblk, args;
 720         unsigned flags;
 721         BOOL ret;
 722
 723         /* Determine whether or not we are associated to a console */
 724         HANDLE cons = CreateFile("CONOUT$", GENERIC_WRITE,
 725                         FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
 726                         FILE_ATTRIBUTE_NORMAL, NULL);
 727         if (cons == INVALID_HANDLE_VALUE) {
 728                 /* There is no console associated with this process.
 729                  * Since the child is a console process, Windows
 730                  * would normally create a console window. But
 731                  * since we'll be redirecting std streams, we do
 732                  * not need the console.
 733                  * It is necessary to use DETACHED_PROCESS
 734                  * instead of CREATE_NO_WINDOW to make ssh
 735                  * recognize that it has no console.
 736                  */
 737                 flags = DETACHED_PROCESS;
 738         } else {
 739                 /* There is already a console. If we specified
 740                  * DETACHED_PROCESS here, too, Windows would
 741                  * disassociate the child from the console.
 742                  * The same is true for CREATE_NO_WINDOW.
 743                  * Go figure!
 744                  */
 745                 flags = 0;
 746                 CloseHandle(cons);
 747         }
 748         memset(&si, 0, sizeof(si));
 749         si.cb = sizeof(si);
 750         si.dwFlags = STARTF_USESTDHANDLES;
 751         si.hStdInput = (HANDLE) _get_osfhandle(fhin);
 752         si.hStdOutput = (HANDLE) _get_osfhandle(fhout);
 753         si.hStdError = (HANDLE) _get_osfhandle(fherr);
 754
 755         /* concatenate argv, quoting args as we go */
 756         strbuf_init(&args, 0);
 757         if (prepend_cmd) {
 758                 char *quoted = (char *)quote_arg(cmd);
 759                 strbuf_addstr(&args, quoted);
 760                 if (quoted != cmd)
 761                         free(quoted);
 762         }
 763         for (; *argv; argv++) {
 764                 char *quoted = (char *)quote_arg(*argv);
 765                 if (*args.buf)
 766                         strbuf_addch(&args, ' ');
 767                 strbuf_addstr(&args, quoted);
 768                 if (quoted != *argv)
 769                         free(quoted);
 770         }
 771
 772         if (env) {
 773                 int count = 0;
 774                 char **e, **sorted_env;
 775
 776                 for (e = env; *e; e++)
 777                         count++;
 778
 779                 /* environment must be sorted */
 780                 sorted_env = xmalloc(sizeof(*sorted_env) * (count + 1));
 781                 memcpy(sorted_env, env, sizeof(*sorted_env) * (count + 1));
 782                 qsort(sorted_env, count, sizeof(*sorted_env), env_compare);
 783
 784                 strbuf_init(&envblk, 0);
 785                 for (e = sorted_env; *e; e++) {
 786                         strbuf_addstr(&envblk, *e);
 787                         strbuf_addch(&envblk, '\0');
 788                 }
 789                 free(sorted_env);
 790         }
 791
 792         memset(&pi, 0, sizeof(pi));
 793         ret = CreateProcess(cmd, args.buf, NULL, NULL, TRUE, flags,
 794                 env ? envblk.buf : NULL, dir, &si, &pi);
 795
 796         if (env)
 797                 strbuf_release(&envblk);
 798         strbuf_release(&args);
 799
 800         if (!ret) {
 801                 errno = ENOENT;
 802                 return -1;
 803         }
 804         CloseHandle(pi.hThread);
 805
 806         /*
 807          * The process ID is the human-readable identifier of the process
 808          * that we want to present in log and error messages. The handle
 809          * is not useful for this purpose. But we cannot close it, either,
 810          * because it is not possible to turn a process ID into a process
 811          * handle after the process terminated.
 812          * Keep the handle in a list for waitpid.
 813          */
 814         EnterCriticalSection(&pinfo_cs);
 815         {
 816                 struct pinfo_t *info = xmalloc(sizeof(struct pinfo_t));
 817                 info->pid = pi.dwProcessId;
 818                 info->proc = pi.hProcess;
 819                 info->next = pinfo;
 820                 pinfo = info;
 821         }
 822         LeaveCriticalSection(&pinfo_cs);
 823
 824         return (pid_t)pi.dwProcessId;
 825 }
 826
 827 static pid_t mingw_spawnve(const char *cmd, const char **argv, char **env,
 828                            int prepend_cmd)
 829 {
 830         return mingw_spawnve_fd(cmd, argv, env, NULL, prepend_cmd, 0, 1, 2);
 831 }
 832
 833 pid_t mingw_spawnvpe(const char *cmd, const char **argv, char **env,
 834                      const char *dir,
 835                      int fhin, int fhout, int fherr)
 836 {
 837         pid_t pid;
 838         char **path = get_path_split();
 839         char *prog = path_lookup(cmd, path, 0);
 840
 841         if (!prog) {
 842                 errno = ENOENT;
 843                 pid = -1;
 844         }
 845         else {
 846                 const char *interpr = parse_interpreter(prog);
 847
 848                 if (interpr) {
 849                         const char *argv0 = argv[0];
 850                         char *iprog = path_lookup(interpr, path, 1);
 851                         argv[0] = prog;
 852                         if (!iprog) {
 853                                 errno = ENOENT;
 854                                 pid = -1;
 855                         }
 856                         else {
 857                                 pid = mingw_spawnve_fd(iprog, argv, env, dir, 1,
 858                                                        fhin, fhout, fherr);
 859                                 free(iprog);
 860                         }
 861                         argv[0] = argv0;
 862                 }
 863                 else
 864                         pid = mingw_spawnve_fd(prog, argv, env, dir, 0,
 865                                                fhin, fhout, fherr);
 866                 free(prog);
 867         }
 868         free_path_split(path);
 869         return pid;
 870 }
 871
 872 static int try_shell_exec(const char *cmd, char *const *argv, char **env)
 873 {
 874         const char *interpr = parse_interpreter(cmd);
 875         char **path;
 876         char *prog;
 877         int pid = 0;
 878
 879         if (!interpr)
 880                 return 0;
 881         path = get_path_split();
 882         prog = path_lookup(interpr, path, 1);
 883         if (prog) {
 884                 int argc = 0;
 885                 const char **argv2;
 886                 while (argv[argc]) argc++;
 887                 argv2 = xmalloc(sizeof(*argv) * (argc+1));
 888                 argv2[0] = (char *)cmd; /* full path to the script file */
 889                 memcpy(&argv2[1], &argv[1], sizeof(*argv) * argc);
 890                 pid = mingw_spawnve(prog, argv2, env, 1);
 891                 if (pid >= 0) {
 892                         int status;
 893                         if (waitpid(pid, &status, 0) < 0)
 894                                 status = 255;
 895                         exit(status);
 896                 }
 897                 pid = 1;        /* indicate that we tried but failed */
 898                 free(prog);
 899                 free(argv2);
 900         }
 901         free_path_split(path);
 902         return pid;
 903 }
 904
 905 static void mingw_execve(const char *cmd, char *const *argv, char *const *env)
 906 {
 907         /* check if git_command is a shell script */
 908         if (!try_shell_exec(cmd, argv, (char **)env)) {
 909                 int pid, status;
 910
 911                 pid = mingw_spawnve(cmd, (const char **)argv, (char **)env, 0);
 912                 if (pid < 0)
 913                         return;
 914                 if (waitpid(pid, &status, 0) < 0)
 915                         status = 255;
 916                 exit(status);
 917         }
 918 }
 919
 920 void mingw_execvp(const char *cmd, char *const *argv)
 921 {
 922         char **path = get_path_split();
 923         char *prog = path_lookup(cmd, path, 0);
 924
 925         if (prog) {
 926                 mingw_execve(prog, argv, environ);
 927                 free(prog);
 928         } else
 929                 errno = ENOENT;
 930
 931         free_path_split(path);
 932 }
 933
 934 void mingw_execv(const char *cmd, char *const *argv)
 935 {
 936         mingw_execve(cmd, argv, environ);
 937 }
 938
 939 int mingw_kill(pid_t pid, int sig)
 940 {
 941         if (pid > 0 && sig == SIGTERM) {
 942                 HANDLE h = OpenProcess(PROCESS_TERMINATE, FALSE, pid);
 943
 944                 if (TerminateProcess(h, -1)) {
 945                         CloseHandle(h);
 946                         return 0;
 947                 }
 948
 949                 errno = err_win_to_posix(GetLastError());
 950                 CloseHandle(h);
 951                 return -1;
 952         }
 953
 954         errno = EINVAL;
 955         return -1;
 956 }
 957
 958 static char **copy_environ(void)
 959 {
 960         char **env;
 961         int i = 0;
 962         while (environ[i])
 963                 i++;
 964         env = xmalloc((i+1)*sizeof(*env));
 965         for (i = 0; environ[i]; i++)
 966                 env[i] = xstrdup(environ[i]);
 967         env[i] = NULL;
 968         return env;
 969 }
 970
 971 void free_environ(char **env)
 972 {
 973         int i;
 974         for (i = 0; env[i]; i++)
 975                 free(env[i]);
 976         free(env);
 977 }
 978
 979 static int lookup_env(char **env, const char *name, size_t nmln)
 980 {
 981         int i;
 982
 983         for (i = 0; env[i]; i++) {
 984                 if (0 == strncmp(env[i], name, nmln)
 985                     && '=' == env[i][nmln])
 986                         /* matches */
 987                         return i;
 988         }
 989         return -1;
 990 }
 991
 992 /*
 993  * If name contains '=', then sets the variable, otherwise it unsets it
 994  */
 995 static char **env_setenv(char **env, const char *name)
 996 {
 997         char *eq = strchrnul(name, '=');
 998         int i = lookup_env(env, name, eq-name);
 999
1000         if (i < 0) {
1001                 if (*eq) {
1002                         for (i = 0; env[i]; i++)
1003                                 ;
1004                         env = xrealloc(env, (i+2)*sizeof(*env));
1005                         env[i] = xstrdup(name);
1006                         env[i+1] = NULL;
1007                 }
1008         }
1009         else {
1010                 free(env[i]);
1011                 if (*eq)
1012                         env[i] = xstrdup(name);
1013                 else
1014                         for (; env[i]; i++)
1015                                 env[i] = env[i+1];
1016         }
1017         return env;
1018 }
1019
1020 /*
1021  * Copies global environ and adjusts variables as specified by vars.
1022  */
1023 char **make_augmented_environ(const char *const *vars)
1024 {
1025         char **env = copy_environ();
1026
1027         while (*vars)
1028                 env = env_setenv(env, *vars++);
1029         return env;
1030 }
1031
1032 /*
1033  * Note, this isn't a complete replacement for getaddrinfo. It assumes
1034  * that service contains a numerical port, or that it it is null. It
1035  * does a simple search using gethostbyname, and returns one IPv4 host
1036  * if one was found.
1037  */
1038 static int WSAAPI getaddrinfo_stub(const char *node, const char *service,
1039                                    const struct addrinfo *hints,
1040                                    struct addrinfo **res)
1041 {
1042         struct hostent *h = gethostbyname(node);
1043         struct addrinfo *ai;
1044         struct sockaddr_in *sin;
1045
1046         if (!h)
1047                 return WSAGetLastError();
1048
1049         ai = xmalloc(sizeof(struct addrinfo));
1050         *res = ai;
1051         ai->ai_flags = 0;
1052         ai->ai_family = AF_INET;
1053         ai->ai_socktype = hints->ai_socktype;
1054         switch (hints->ai_socktype) {
1055         case SOCK_STREAM:
1056                 ai->ai_protocol = IPPROTO_TCP;
1057                 break;
1058         case SOCK_DGRAM:
1059                 ai->ai_protocol = IPPROTO_UDP;
1060                 break;
1061         default:
1062                 ai->ai_protocol = 0;
1063                 break;
1064         }
1065         ai->ai_addrlen = sizeof(struct sockaddr_in);
1066         ai->ai_canonname = strdup(h->h_name);
1067
1068         sin = xmalloc(ai->ai_addrlen);
1069         memset(sin, 0, ai->ai_addrlen);
1070         sin->sin_family = AF_INET;
1071         if (service)
1072                 sin->sin_port = htons(atoi(service));
1073         sin->sin_addr = *(struct in_addr *)h->h_addr;
1074         ai->ai_addr = (struct sockaddr *)sin;
1075         ai->ai_next = 0;
1076         return 0;
1077 }
1078
1079 static void WSAAPI freeaddrinfo_stub(struct addrinfo *res)
1080 {
1081         free(res->ai_canonname);
1082         free(res->ai_addr);
1083         free(res);
1084 }
1085
1086 static int WSAAPI getnameinfo_stub(const struct sockaddr *sa, socklen_t salen,
1087                                    char *host, DWORD hostlen,
1088                                    char *serv, DWORD servlen, int flags)
1089 {
1090         const struct sockaddr_in *sin = (const struct sockaddr_in *)sa;
1091         if (sa->sa_family != AF_INET)
1092                 return EAI_FAMILY;
1093         if (!host && !serv)
1094                 return EAI_NONAME;
1095
1096         if (host && hostlen > 0) {
1097                 struct hostent *ent = NULL;
1098                 if (!(flags & NI_NUMERICHOST))
1099                         ent = gethostbyaddr((const char *)&sin->sin_addr,
1100                                             sizeof(sin->sin_addr), AF_INET);
1101
1102                 if (ent)
1103                         snprintf(host, hostlen, "%s", ent->h_name);
1104                 else if (flags & NI_NAMEREQD)
1105                         return EAI_NONAME;
1106                 else
1107                         snprintf(host, hostlen, "%s", inet_ntoa(sin->sin_addr));
1108         }
1109
1110         if (serv && servlen > 0) {
1111                 struct servent *ent = NULL;
1112                 if (!(flags & NI_NUMERICSERV))
1113                         ent = getservbyport(sin->sin_port,
1114                                             flags & NI_DGRAM ? "udp" : "tcp");
1115
1116                 if (ent)
1117                         snprintf(serv, servlen, "%s", ent->s_name);
1118                 else
1119                         snprintf(serv, servlen, "%d", ntohs(sin->sin_port));
1120         }
1121
1122         return 0;
1123 }
1124
1125 static HMODULE ipv6_dll = NULL;
1126 static void (WSAAPI *ipv6_freeaddrinfo)(struct addrinfo *res);
1127 static int (WSAAPI *ipv6_getaddrinfo)(const char *node, const char *service,
1128                                       const struct addrinfo *hints,
1129                                       struct addrinfo **res);
1130 static int (WSAAPI *ipv6_getnameinfo)(const struct sockaddr *sa, socklen_t salen,
1131                                       char *host, DWORD hostlen,
1132                                       char *serv, DWORD servlen, int flags);
1133 /*
1134  * gai_strerror is an inline function in the ws2tcpip.h header, so we
1135  * don't need to try to load that one dynamically.
1136  */
1137
1138 static void socket_cleanup(void)
1139 {
1140         WSACleanup();
1141         if (ipv6_dll)
1142                 FreeLibrary(ipv6_dll);
1143         ipv6_dll = NULL;
1144         ipv6_freeaddrinfo = freeaddrinfo_stub;
1145         ipv6_getaddrinfo = getaddrinfo_stub;
1146         ipv6_getnameinfo = getnameinfo_stub;
1147 }
1148
1149 static void ensure_socket_initialization(void)
1150 {
1151         WSADATA wsa;
1152         static int initialized = 0;
1153         const char *libraries[] = { "ws2_32.dll", "wship6.dll", NULL };
1154         const char **name;
1155
1156         if (initialized)
1157                 return;
1158
1159         if (WSAStartup(MAKEWORD(2,2), &wsa))
1160                 die("unable to initialize winsock subsystem, error %d",
1161                         WSAGetLastError());
1162
1163         for (name = libraries; *name; name++) {
1164                 ipv6_dll = LoadLibrary(*name);
1165                 if (!ipv6_dll)
1166                         continue;
1167
1168                 ipv6_freeaddrinfo = (void (WSAAPI *)(struct addrinfo *))
1169                         GetProcAddress(ipv6_dll, "freeaddrinfo");
1170                 ipv6_getaddrinfo = (int (WSAAPI *)(const char *, const char *,
1171                                                    const struct addrinfo *,
1172                                                    struct addrinfo **))
1173                         GetProcAddress(ipv6_dll, "getaddrinfo");
1174                 ipv6_getnameinfo = (int (WSAAPI *)(const struct sockaddr *,
1175                                                    socklen_t, char *, DWORD,
1176                                                    char *, DWORD, int))
1177                         GetProcAddress(ipv6_dll, "getnameinfo");
1178                 if (!ipv6_freeaddrinfo || !ipv6_getaddrinfo || !ipv6_getnameinfo) {
1179                         FreeLibrary(ipv6_dll);
1180                         ipv6_dll = NULL;
1181                 } else
1182                         break;
1183         }
1184         if (!ipv6_freeaddrinfo || !ipv6_getaddrinfo || !ipv6_getnameinfo) {
1185                 ipv6_freeaddrinfo = freeaddrinfo_stub;
1186                 ipv6_getaddrinfo = getaddrinfo_stub;
1187                 ipv6_getnameinfo = getnameinfo_stub;
1188         }
1189
1190         atexit(socket_cleanup);
1191         initialized = 1;
1192 }
1193
1194 #undef gethostbyname
1195 struct hostent *mingw_gethostbyname(const char *host)
1196 {
1197         ensure_socket_initialization();
1198         return gethostbyname(host);
1199 }
1200
1201 void mingw_freeaddrinfo(struct addrinfo *res)
1202 {
1203         ipv6_freeaddrinfo(res);
1204 }
1205
1206 int mingw_getaddrinfo(const char *node, const char *service,
1207                       const struct addrinfo *hints, struct addrinfo **res)
1208 {
1209         ensure_socket_initialization();
1210         return ipv6_getaddrinfo(node, service, hints, res);
1211 }
1212
1213 int mingw_getnameinfo(const struct sockaddr *sa, socklen_t salen,
1214                       char *host, DWORD hostlen, char *serv, DWORD servlen,
1215                       int flags)
1216 {
1217         ensure_socket_initialization();
1218         return ipv6_getnameinfo(sa, salen, host, hostlen, serv, servlen, flags);
1219 }
1220
1221 int mingw_socket(int domain, int type, int protocol)
1222 {
1223         int sockfd;
1224         SOCKET s;
1225
1226         ensure_socket_initialization();
1227         s = WSASocket(domain, type, protocol, NULL, 0, 0);
1228         if (s == INVALID_SOCKET) {
1229                 /*
1230                  * WSAGetLastError() values are regular BSD error codes
1231                  * biased by WSABASEERR.
1232                  * However, strerror() does not know about networking
1233                  * specific errors, which are values beginning at 38 or so.
1234                  * Therefore, we choose to leave the biased error code
1235                  * in errno so that _if_ someone looks up the code somewhere,
1236                  * then it is at least the number that are usually listed.
1237                  */
1238                 errno = WSAGetLastError();
1239                 return -1;
1240         }
1241         /* convert into a file descriptor */
1242         if ((sockfd = _open_osfhandle(s, O_RDWR|O_BINARY)) < 0) {
1243                 closesocket(s);
1244                 return error("unable to make a socket file descriptor: %s",
1245                         strerror(errno));
1246         }
1247         return sockfd;
1248 }
1249
1250 #undef connect
1251 int mingw_connect(int sockfd, struct sockaddr *sa, size_t sz)
1252 {
1253         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1254         return connect(s, sa, sz);
1255 }
1256
1257 #undef bind
1258 int mingw_bind(int sockfd, struct sockaddr *sa, size_t sz)
1259 {
1260         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1261         return bind(s, sa, sz);
1262 }
1263
1264 #undef setsockopt
1265 int mingw_setsockopt(int sockfd, int lvl, int optname, void *optval, int optlen)
1266 {
1267         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1268         return setsockopt(s, lvl, optname, (const char*)optval, optlen);
1269 }
1270
1271 #undef listen
1272 int mingw_listen(int sockfd, int backlog)
1273 {
1274         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1275         return listen(s, backlog);
1276 }
1277
1278 #undef accept
1279 int mingw_accept(int sockfd1, struct sockaddr *sa, socklen_t *sz)
1280 {
1281         int sockfd2;
1282
1283         SOCKET s1 = (SOCKET)_get_osfhandle(sockfd1);
1284         SOCKET s2 = accept(s1, sa, sz);
1285
1286         /* convert into a file descriptor */
1287         if ((sockfd2 = _open_osfhandle(s2, O_RDWR|O_BINARY)) < 0) {
1288                 int err = errno;
1289                 closesocket(s2);
1290                 return error("unable to make a socket file descriptor: %s",
1291                         strerror(err));
1292         }
1293         return sockfd2;
1294 }
1295
1296 #undef rename
1297 int mingw_rename(const char *pold, const char *pnew)
1298 {
1299         DWORD attrs, gle;
1300         int tries = 0;
1301         static const int delay[] = { 0, 1, 10, 20, 40 };
1302
1303         /*
1304          * Try native rename() first to get errno right.
1305          * It is based on MoveFile(), which cannot overwrite existing files.
1306          */
1307         if (!rename(pold, pnew))
1308                 return 0;
1309         if (errno != EEXIST)
1310                 return -1;
1311 repeat:
1312         if (MoveFileEx(pold, pnew, MOVEFILE_REPLACE_EXISTING))
1313                 return 0;
1314         /* TODO: translate more errors */
1315         gle = GetLastError();
1316         if (gle == ERROR_ACCESS_DENIED &&
1317             (attrs = GetFileAttributes(pnew)) != INVALID_FILE_ATTRIBUTES) {
1318                 if (attrs & FILE_ATTRIBUTE_DIRECTORY) {
1319                         errno = EISDIR;
1320                         return -1;
1321                 }
1322                 if ((attrs & FILE_ATTRIBUTE_READONLY) &&
1323                     SetFileAttributes(pnew, attrs & ~FILE_ATTRIBUTE_READONLY)) {
1324                         if (MoveFileEx(pold, pnew, MOVEFILE_REPLACE_EXISTING))
1325                                 return 0;
1326                         gle = GetLastError();
1327                         /* revert file attributes on failure */
1328                         SetFileAttributes(pnew, attrs);
1329                 }
1330         }
1331         if (tries < ARRAY_SIZE(delay) && gle == ERROR_ACCESS_DENIED) {
1332                 /*
1333                  * We assume that some other process had the source or
1334                  * destination file open at the wrong moment and retry.
1335                  * In order to give the other process a higher chance to
1336                  * complete its operation, we give up our time slice now.
1337                  * If we have to retry again, we do sleep a bit.
1338                  */
1339                 Sleep(delay[tries]);
1340                 tries++;
1341                 goto repeat;
1342         }
1343         errno = EACCES;
1344         return -1;
1345 }
1346
1347 /*
1348  * Note that this doesn't return the actual pagesize, but
1349  * the allocation granularity. If future Windows specific git code
1350  * needs the real getpagesize function, we need to find another solution.
1351  */
1352 int mingw_getpagesize(void)
1353 {
1354         SYSTEM_INFO si;
1355         GetSystemInfo(&si);
1356         return si.dwAllocationGranularity;
1357 }
1358
1359 struct passwd *getpwuid(int uid)
1360 {
1361         static char user_name[100];
1362         static struct passwd p;
1363
1364         DWORD len = sizeof(user_name);
1365         if (!GetUserName(user_name, &len))
1366                 return NULL;
1367         p.pw_name = user_name;
1368         p.pw_gecos = "unknown";
1369         p.pw_dir = NULL;
1370         return &p;
1371 }
1372
1373 static HANDLE timer_event;
1374 static HANDLE timer_thread;
1375 static int timer_interval;
1376 static int one_shot;
1377 static sig_handler_t timer_fn = SIG_DFL;
1378
1379 /* The timer works like this:
1380  * The thread, ticktack(), is a trivial routine that most of the time
1381  * only waits to receive the signal to terminate. The main thread tells
1382  * the thread to terminate by setting the timer_event to the signalled
1383  * state.
1384  * But ticktack() interrupts the wait state after the timer's interval
1385  * length to call the signal handler.
1386  */
1387
1388 static unsigned __stdcall ticktack(void *dummy)
1389 {
1390         while (WaitForSingleObject(timer_event, timer_interval) == WAIT_TIMEOUT) {
1391                 if (timer_fn == SIG_DFL)
1392                         die("Alarm");
1393                 if (timer_fn != SIG_IGN)
1394                         timer_fn(SIGALRM);
1395                 if (one_shot)
1396                         break;
1397         }
1398         return 0;
1399 }
1400
1401 static int start_timer_thread(void)
1402 {
1403         timer_event = CreateEvent(NULL, FALSE, FALSE, NULL);
1404         if (timer_event) {
1405                 timer_thread = (HANDLE) _beginthreadex(NULL, 0, ticktack, NULL, 0, NULL);
1406                 if (!timer_thread )
1407                         return errno = ENOMEM,
1408                                 error("cannot start timer thread");
1409         } else
1410                 return errno = ENOMEM,
1411                         error("cannot allocate resources for timer");
1412         return 0;
1413 }
1414
1415 static void stop_timer_thread(void)
1416 {
1417         if (timer_event)
1418                 SetEvent(timer_event);  /* tell thread to terminate */
1419         if (timer_thread) {
1420                 int rc = WaitForSingleObject(timer_thread, 1000);
1421                 if (rc == WAIT_TIMEOUT)
1422                         error("timer thread did not terminate timely");
1423                 else if (rc != WAIT_OBJECT_0)
1424                         error("waiting for timer thread failed: %lu",
1425                               GetLastError());
1426                 CloseHandle(timer_thread);
1427         }
1428         if (timer_event)
1429                 CloseHandle(timer_event);
1430         timer_event = NULL;
1431         timer_thread = NULL;
1432 }
1433
1434 static inline int is_timeval_eq(const struct timeval *i1, const struct timeval *i2)
1435 {
1436         return i1->tv_sec == i2->tv_sec && i1->tv_usec == i2->tv_usec;
1437 }
1438
1439 int setitimer(int type, struct itimerval *in, struct itimerval *out)
1440 {
1441         static const struct timeval zero;
1442         static int atexit_done;
1443
1444         if (out != NULL)
1445                 return errno = EINVAL,
1446                         error("setitimer param 3 != NULL not implemented");
1447         if (!is_timeval_eq(&in->it_interval, &zero) &&
1448             !is_timeval_eq(&in->it_interval, &in->it_value))
1449                 return errno = EINVAL,
1450                         error("setitimer: it_interval must be zero or eq it_value");
1451
1452         if (timer_thread)
1453                 stop_timer_thread();
1454
1455         if (is_timeval_eq(&in->it_value, &zero) &&
1456             is_timeval_eq(&in->it_interval, &zero))
1457                 return 0;
1458
1459         timer_interval = in->it_value.tv_sec * 1000 + in->it_value.tv_usec / 1000;
1460         one_shot = is_timeval_eq(&in->it_interval, &zero);
1461         if (!atexit_done) {
1462                 atexit(stop_timer_thread);
1463                 atexit_done = 1;
1464         }
1465         return start_timer_thread();
1466 }
1467
1468 int sigaction(int sig, struct sigaction *in, struct sigaction *out)
1469 {
1470         if (sig != SIGALRM)
1471                 return errno = EINVAL,
1472                         error("sigaction only implemented for SIGALRM");
1473         if (out != NULL)
1474                 return errno = EINVAL,
1475                         error("sigaction: param 3 != NULL not implemented");
1476
1477         timer_fn = in->sa_handler;
1478         return 0;
1479 }
1480
1481 #undef signal
1482 sig_handler_t mingw_signal(int sig, sig_handler_t handler)
1483 {
1484         sig_handler_t old = timer_fn;
1485         if (sig != SIGALRM)
1486                 return signal(sig, handler);
1487         timer_fn = handler;
1488         return old;
1489 }
1490
1491 static const char *make_backslash_path(const char *path)
1492 {
1493         static char buf[PATH_MAX + 1];
1494         char *c;
1495
1496         if (strlcpy(buf, path, PATH_MAX) >= PATH_MAX)
1497                 die("Too long path: %.*s", 60, path);
1498
1499         for (c = buf; *c; c++) {
1500                 if (*c == '/')
1501                         *c = '\\';
1502         }
1503         return buf;
1504 }
1505
1506 void mingw_open_html(const char *unixpath)
1507 {
1508         const char *htmlpath = make_backslash_path(unixpath);
1509         typedef HINSTANCE (WINAPI *T)(HWND, const char *,
1510                         const char *, const char *, const char *, INT);
1511         T ShellExecute;
1512         HMODULE shell32;
1513         int r;
1514
1515         shell32 = LoadLibrary("shell32.dll");
1516         if (!shell32)
1517                 die("cannot load shell32.dll");
1518         ShellExecute = (T)GetProcAddress(shell32, "ShellExecuteA");
1519         if (!ShellExecute)
1520                 die("cannot run browser");
1521
1522         printf("Launching default browser to display HTML ...\n");
1523         r = (int)ShellExecute(NULL, "open", htmlpath, NULL, "\\", SW_SHOWNORMAL);
1524         FreeLibrary(shell32);
1525         /* see the MSDN documentation referring to the result codes here */
1526         if (r <= 32) {
1527                 die("failed to launch browser for %.*s", MAX_PATH, unixpath);
1528         }
1529 }
1530
1531 int link(const char *oldpath, const char *newpath)
1532 {
1533         typedef BOOL (WINAPI *T)(const char*, const char*, LPSECURITY_ATTRIBUTES);
1534         static T create_hard_link = NULL;
1535         if (!create_hard_link) {
1536                 create_hard_link = (T) GetProcAddress(
1537                         GetModuleHandle("kernel32.dll"), "CreateHardLinkA");
1538                 if (!create_hard_link)
1539                         create_hard_link = (T)-1;
1540         }
1541         if (create_hard_link == (T)-1) {
1542                 errno = ENOSYS;
1543                 return -1;
1544         }
1545         if (!create_hard_link(newpath, oldpath, NULL)) {
1546                 errno = err_win_to_posix(GetLastError());
1547                 return -1;
1548         }
1549         return 0;
1550 }
1551
1552 char *getpass(const char *prompt)
1553 {
1554         struct strbuf buf = STRBUF_INIT;
1555
1556         fputs(prompt, stderr);
1557         for (;;) {
1558                 char c = _getch();
1559                 if (c == '\r' || c == '\n')
1560                         break;
1561                 strbuf_addch(&buf, c);
1562         }
1563         fputs("\n", stderr);
1564         return strbuf_detach(&buf, NULL);
1565 }
1566
1567 pid_t waitpid(pid_t pid, int *status, unsigned options)
1568 {
1569         HANDLE h = OpenProcess(SYNCHRONIZE | PROCESS_QUERY_INFORMATION,
1570             FALSE, pid);
1571         if (!h) {
1572                 errno = ECHILD;
1573                 return -1;
1574         }
1575
1576         if (pid > 0 && options & WNOHANG) {
1577                 if (WAIT_OBJECT_0 != WaitForSingleObject(h, 0)) {
1578                         CloseHandle(h);
1579                         return 0;
1580                 }
1581                 options &= ~WNOHANG;
1582         }
1583
1584         if (options == 0) {
1585                 struct pinfo_t **ppinfo;
1586                 if (WaitForSingleObject(h, INFINITE) != WAIT_OBJECT_0) {
1587                         CloseHandle(h);
1588                         return 0;
1589                 }
1590
1591                 if (status)
1592                         GetExitCodeProcess(h, (LPDWORD)status);
1593
1594                 EnterCriticalSection(&pinfo_cs);
1595
1596                 ppinfo = &pinfo;
1597                 while (*ppinfo) {
1598                         struct pinfo_t *info = *ppinfo;
1599                         if (info->pid == pid) {
1600                                 CloseHandle(info->proc);
1601                                 *ppinfo = info->next;
1602                                 free(info);
1603                                 break;
1604                         }
1605                         ppinfo = &info->next;
1606                 }
1607
1608                 LeaveCriticalSection(&pinfo_cs);
1609
1610                 CloseHandle(h);
1611                 return pid;
1612         }
1613         CloseHandle(h);
1614
1615         errno = EINVAL;
1616         return -1;
1617 }
1618
1619 #ifndef NO_MINGW_REPLACE_READDIR
1620 /* MinGW readdir implementation to avoid extra lstats for Git */
1621 struct mingw_DIR
1622 {
1623         struct _finddata_t      dd_dta;         /* disk transfer area for this dir */
1624         struct mingw_dirent     dd_dir;         /* Our own implementation, including d_type */
1625         long                    dd_handle;      /* _findnext handle */
1626         int                     dd_stat;        /* 0 = next entry to read is first entry, -1 = off the end, positive = 0 based index of next entry */
1627         char                    dd_name[1];     /* given path for dir with search pattern (struct is extended) */
1628 };
1629
1630 struct dirent *mingw_readdir(DIR *dir)
1631 {
1632         WIN32_FIND_DATAA buf;
1633         HANDLE handle;
1634         struct mingw_DIR *mdir = (struct mingw_DIR*)dir;
1635
1636         if (!dir->dd_handle) {
1637                 errno = EBADF; /* No set_errno for mingw */
1638                 return NULL;
1639         }
1640
1641         if (dir->dd_handle == (long)INVALID_HANDLE_VALUE && dir->dd_stat == 0)
1642         {
1643                 DWORD lasterr;
1644                 handle = FindFirstFileA(dir->dd_name, &buf);
1645                 lasterr = GetLastError();
1646                 dir->dd_handle = (long)handle;
1647                 if (handle == INVALID_HANDLE_VALUE && (lasterr != ERROR_NO_MORE_FILES)) {
1648                         errno = err_win_to_posix(lasterr);
1649                         return NULL;
1650                 }
1651         } else if (dir->dd_handle == (long)INVALID_HANDLE_VALUE) {
1652                 return NULL;
1653         } else if (!FindNextFileA((HANDLE)dir->dd_handle, &buf)) {
1654                 DWORD lasterr = GetLastError();
1655                 FindClose((HANDLE)dir->dd_handle);
1656                 dir->dd_handle = (long)INVALID_HANDLE_VALUE;
1657                 /* POSIX says you shouldn't set errno when readdir can't
1658                    find any more files; so, if another error we leave it set. */
1659                 if (lasterr != ERROR_NO_MORE_FILES)
1660                         errno = err_win_to_posix(lasterr);
1661                 return NULL;
1662         }
1663
1664         /* We get here if `buf' contains valid data.  */
1665         strcpy(dir->dd_dir.d_name, buf.cFileName);
1666         ++dir->dd_stat;
1667
1668         /* Set file type, based on WIN32_FIND_DATA */
1669         mdir->dd_dir.d_type = 0;
1670         if (buf.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
1671                 mdir->dd_dir.d_type |= DT_DIR;
1672         else
1673                 mdir->dd_dir.d_type |= DT_REG;
1674
1675         return (struct dirent*)&dir->dd_dir;
1676 }
1677 #endif // !NO_MINGW_REPLACE_READDIR