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 static char **copy_environ(void)
 940 {
 941         char **env;
 942         int i = 0;
 943         while (environ[i])
 944                 i++;
 945         env = xmalloc((i+1)*sizeof(*env));
 946         for (i = 0; environ[i]; i++)
 947                 env[i] = xstrdup(environ[i]);
 948         env[i] = NULL;
 949         return env;
 950 }
 951
 952 void free_environ(char **env)
 953 {
 954         int i;
 955         for (i = 0; env[i]; i++)
 956                 free(env[i]);
 957         free(env);
 958 }
 959
 960 static int lookup_env(char **env, const char *name, size_t nmln)
 961 {
 962         int i;
 963
 964         for (i = 0; env[i]; i++) {
 965                 if (0 == strncmp(env[i], name, nmln)
 966                     && '=' == env[i][nmln])
 967                         /* matches */
 968                         return i;
 969         }
 970         return -1;
 971 }
 972
 973 /*
 974  * If name contains '=', then sets the variable, otherwise it unsets it
 975  */
 976 static char **env_setenv(char **env, const char *name)
 977 {
 978         char *eq = strchrnul(name, '=');
 979         int i = lookup_env(env, name, eq-name);
 980
 981         if (i < 0) {
 982                 if (*eq) {
 983                         for (i = 0; env[i]; i++)
 984                                 ;
 985                         env = xrealloc(env, (i+2)*sizeof(*env));
 986                         env[i] = xstrdup(name);
 987                         env[i+1] = NULL;
 988                 }
 989         }
 990         else {
 991                 free(env[i]);
 992                 if (*eq)
 993                         env[i] = xstrdup(name);
 994                 else
 995                         for (; env[i]; i++)
 996                                 env[i] = env[i+1];
 997         }
 998         return env;
 999 }
1000
1001 /*
1002  * Copies global environ and adjusts variables as specified by vars.
1003  */
1004 char **make_augmented_environ(const char *const *vars)
1005 {
1006         char **env = copy_environ();
1007
1008         while (*vars)
1009                 env = env_setenv(env, *vars++);
1010         return env;
1011 }
1012
1013 /*
1014  * Note, this isn't a complete replacement for getaddrinfo. It assumes
1015  * that service contains a numerical port, or that it it is null. It
1016  * does a simple search using gethostbyname, and returns one IPv4 host
1017  * if one was found.
1018  */
1019 static int WSAAPI getaddrinfo_stub(const char *node, const char *service,
1020                                    const struct addrinfo *hints,
1021                                    struct addrinfo **res)
1022 {
1023         struct hostent *h = gethostbyname(node);
1024         struct addrinfo *ai;
1025         struct sockaddr_in *sin;
1026
1027         if (!h)
1028                 return WSAGetLastError();
1029
1030         ai = xmalloc(sizeof(struct addrinfo));
1031         *res = ai;
1032         ai->ai_flags = 0;
1033         ai->ai_family = AF_INET;
1034         ai->ai_socktype = hints->ai_socktype;
1035         switch (hints->ai_socktype) {
1036         case SOCK_STREAM:
1037                 ai->ai_protocol = IPPROTO_TCP;
1038                 break;
1039         case SOCK_DGRAM:
1040                 ai->ai_protocol = IPPROTO_UDP;
1041                 break;
1042         default:
1043                 ai->ai_protocol = 0;
1044                 break;
1045         }
1046         ai->ai_addrlen = sizeof(struct sockaddr_in);
1047         ai->ai_canonname = strdup(h->h_name);
1048
1049         sin = xmalloc(ai->ai_addrlen);
1050         memset(sin, 0, ai->ai_addrlen);
1051         sin->sin_family = AF_INET;
1052         if (service)
1053                 sin->sin_port = htons(atoi(service));
1054         sin->sin_addr = *(struct in_addr *)h->h_addr;
1055         ai->ai_addr = (struct sockaddr *)sin;
1056         ai->ai_next = 0;
1057         return 0;
1058 }
1059
1060 static void WSAAPI freeaddrinfo_stub(struct addrinfo *res)
1061 {
1062         free(res->ai_canonname);
1063         free(res->ai_addr);
1064         free(res);
1065 }
1066
1067 static int WSAAPI getnameinfo_stub(const struct sockaddr *sa, socklen_t salen,
1068                                    char *host, DWORD hostlen,
1069                                    char *serv, DWORD servlen, int flags)
1070 {
1071         const struct sockaddr_in *sin = (const struct sockaddr_in *)sa;
1072         if (sa->sa_family != AF_INET)
1073                 return EAI_FAMILY;
1074         if (!host && !serv)
1075                 return EAI_NONAME;
1076
1077         if (host && hostlen > 0) {
1078                 struct hostent *ent = NULL;
1079                 if (!(flags & NI_NUMERICHOST))
1080                         ent = gethostbyaddr((const char *)&sin->sin_addr,
1081                                             sizeof(sin->sin_addr), AF_INET);
1082
1083                 if (ent)
1084                         snprintf(host, hostlen, "%s", ent->h_name);
1085                 else if (flags & NI_NAMEREQD)
1086                         return EAI_NONAME;
1087                 else
1088                         snprintf(host, hostlen, "%s", inet_ntoa(sin->sin_addr));
1089         }
1090
1091         if (serv && servlen > 0) {
1092                 struct servent *ent = NULL;
1093                 if (!(flags & NI_NUMERICSERV))
1094                         ent = getservbyport(sin->sin_port,
1095                                             flags & NI_DGRAM ? "udp" : "tcp");
1096
1097                 if (ent)
1098                         snprintf(serv, servlen, "%s", ent->s_name);
1099                 else
1100                         snprintf(serv, servlen, "%d", ntohs(sin->sin_port));
1101         }
1102
1103         return 0;
1104 }
1105
1106 static HMODULE ipv6_dll = NULL;
1107 static void (WSAAPI *ipv6_freeaddrinfo)(struct addrinfo *res);
1108 static int (WSAAPI *ipv6_getaddrinfo)(const char *node, const char *service,
1109                                       const struct addrinfo *hints,
1110                                       struct addrinfo **res);
1111 static int (WSAAPI *ipv6_getnameinfo)(const struct sockaddr *sa, socklen_t salen,
1112                                       char *host, DWORD hostlen,
1113                                       char *serv, DWORD servlen, int flags);
1114 /*
1115  * gai_strerror is an inline function in the ws2tcpip.h header, so we
1116  * don't need to try to load that one dynamically.
1117  */
1118
1119 static void socket_cleanup(void)
1120 {
1121         WSACleanup();
1122         if (ipv6_dll)
1123                 FreeLibrary(ipv6_dll);
1124         ipv6_dll = NULL;
1125         ipv6_freeaddrinfo = freeaddrinfo_stub;
1126         ipv6_getaddrinfo = getaddrinfo_stub;
1127         ipv6_getnameinfo = getnameinfo_stub;
1128 }
1129
1130 static void ensure_socket_initialization(void)
1131 {
1132         WSADATA wsa;
1133         static int initialized = 0;
1134         const char *libraries[] = { "ws2_32.dll", "wship6.dll", NULL };
1135         const char **name;
1136
1137         if (initialized)
1138                 return;
1139
1140         if (WSAStartup(MAKEWORD(2,2), &wsa))
1141                 die("unable to initialize winsock subsystem, error %d",
1142                         WSAGetLastError());
1143
1144         for (name = libraries; *name; name++) {
1145                 ipv6_dll = LoadLibrary(*name);
1146                 if (!ipv6_dll)
1147                         continue;
1148
1149                 ipv6_freeaddrinfo = (void (WSAAPI *)(struct addrinfo *))
1150                         GetProcAddress(ipv6_dll, "freeaddrinfo");
1151                 ipv6_getaddrinfo = (int (WSAAPI *)(const char *, const char *,
1152                                                    const struct addrinfo *,
1153                                                    struct addrinfo **))
1154                         GetProcAddress(ipv6_dll, "getaddrinfo");
1155                 ipv6_getnameinfo = (int (WSAAPI *)(const struct sockaddr *,
1156                                                    socklen_t, char *, DWORD,
1157                                                    char *, DWORD, int))
1158                         GetProcAddress(ipv6_dll, "getnameinfo");
1159                 if (!ipv6_freeaddrinfo || !ipv6_getaddrinfo || !ipv6_getnameinfo) {
1160                         FreeLibrary(ipv6_dll);
1161                         ipv6_dll = NULL;
1162                 } else
1163                         break;
1164         }
1165         if (!ipv6_freeaddrinfo || !ipv6_getaddrinfo || !ipv6_getnameinfo) {
1166                 ipv6_freeaddrinfo = freeaddrinfo_stub;
1167                 ipv6_getaddrinfo = getaddrinfo_stub;
1168                 ipv6_getnameinfo = getnameinfo_stub;
1169         }
1170
1171         atexit(socket_cleanup);
1172         initialized = 1;
1173 }
1174
1175 #undef gethostbyname
1176 struct hostent *mingw_gethostbyname(const char *host)
1177 {
1178         ensure_socket_initialization();
1179         return gethostbyname(host);
1180 }
1181
1182 void mingw_freeaddrinfo(struct addrinfo *res)
1183 {
1184         ipv6_freeaddrinfo(res);
1185 }
1186
1187 int mingw_getaddrinfo(const char *node, const char *service,
1188                       const struct addrinfo *hints, struct addrinfo **res)
1189 {
1190         ensure_socket_initialization();
1191         return ipv6_getaddrinfo(node, service, hints, res);
1192 }
1193
1194 int mingw_getnameinfo(const struct sockaddr *sa, socklen_t salen,
1195                       char *host, DWORD hostlen, char *serv, DWORD servlen,
1196                       int flags)
1197 {
1198         ensure_socket_initialization();
1199         return ipv6_getnameinfo(sa, salen, host, hostlen, serv, servlen, flags);
1200 }
1201
1202 int mingw_socket(int domain, int type, int protocol)
1203 {
1204         int sockfd;
1205         SOCKET s;
1206
1207         ensure_socket_initialization();
1208         s = WSASocket(domain, type, protocol, NULL, 0, 0);
1209         if (s == INVALID_SOCKET) {
1210                 /*
1211                  * WSAGetLastError() values are regular BSD error codes
1212                  * biased by WSABASEERR.
1213                  * However, strerror() does not know about networking
1214                  * specific errors, which are values beginning at 38 or so.
1215                  * Therefore, we choose to leave the biased error code
1216                  * in errno so that _if_ someone looks up the code somewhere,
1217                  * then it is at least the number that are usually listed.
1218                  */
1219                 errno = WSAGetLastError();
1220                 return -1;
1221         }
1222         /* convert into a file descriptor */
1223         if ((sockfd = _open_osfhandle(s, O_RDWR|O_BINARY)) < 0) {
1224                 closesocket(s);
1225                 return error("unable to make a socket file descriptor: %s",
1226                         strerror(errno));
1227         }
1228         return sockfd;
1229 }
1230
1231 #undef connect
1232 int mingw_connect(int sockfd, struct sockaddr *sa, size_t sz)
1233 {
1234         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1235         return connect(s, sa, sz);
1236 }
1237
1238 #undef bind
1239 int mingw_bind(int sockfd, struct sockaddr *sa, size_t sz)
1240 {
1241         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1242         return bind(s, sa, sz);
1243 }
1244
1245 #undef setsockopt
1246 int mingw_setsockopt(int sockfd, int lvl, int optname, void *optval, int optlen)
1247 {
1248         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1249         return setsockopt(s, lvl, optname, (const char*)optval, optlen);
1250 }
1251
1252 #undef listen
1253 int mingw_listen(int sockfd, int backlog)
1254 {
1255         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1256         return listen(s, backlog);
1257 }
1258
1259 #undef accept
1260 int mingw_accept(int sockfd1, struct sockaddr *sa, socklen_t *sz)
1261 {
1262         int sockfd2;
1263
1264         SOCKET s1 = (SOCKET)_get_osfhandle(sockfd1);
1265         SOCKET s2 = accept(s1, sa, sz);
1266
1267         /* convert into a file descriptor */
1268         if ((sockfd2 = _open_osfhandle(s2, O_RDWR|O_BINARY)) < 0) {
1269                 int err = errno;
1270                 closesocket(s2);
1271                 return error("unable to make a socket file descriptor: %s",
1272                         strerror(err));
1273         }
1274         return sockfd2;
1275 }
1276
1277 #undef rename
1278 int mingw_rename(const char *pold, const char *pnew)
1279 {
1280         DWORD attrs, gle;
1281         int tries = 0;
1282         static const int delay[] = { 0, 1, 10, 20, 40 };
1283
1284         /*
1285          * Try native rename() first to get errno right.
1286          * It is based on MoveFile(), which cannot overwrite existing files.
1287          */
1288         if (!rename(pold, pnew))
1289                 return 0;
1290         if (errno != EEXIST)
1291                 return -1;
1292 repeat:
1293         if (MoveFileEx(pold, pnew, MOVEFILE_REPLACE_EXISTING))
1294                 return 0;
1295         /* TODO: translate more errors */
1296         gle = GetLastError();
1297         if (gle == ERROR_ACCESS_DENIED &&
1298             (attrs = GetFileAttributes(pnew)) != INVALID_FILE_ATTRIBUTES) {
1299                 if (attrs & FILE_ATTRIBUTE_DIRECTORY) {
1300                         errno = EISDIR;
1301                         return -1;
1302                 }
1303                 if ((attrs & FILE_ATTRIBUTE_READONLY) &&
1304                     SetFileAttributes(pnew, attrs & ~FILE_ATTRIBUTE_READONLY)) {
1305                         if (MoveFileEx(pold, pnew, MOVEFILE_REPLACE_EXISTING))
1306                                 return 0;
1307                         gle = GetLastError();
1308                         /* revert file attributes on failure */
1309                         SetFileAttributes(pnew, attrs);
1310                 }
1311         }
1312         if (tries < ARRAY_SIZE(delay) && gle == ERROR_ACCESS_DENIED) {
1313                 /*
1314                  * We assume that some other process had the source or
1315                  * destination file open at the wrong moment and retry.
1316                  * In order to give the other process a higher chance to
1317                  * complete its operation, we give up our time slice now.
1318                  * If we have to retry again, we do sleep a bit.
1319                  */
1320                 Sleep(delay[tries]);
1321                 tries++;
1322                 goto repeat;
1323         }
1324         errno = EACCES;
1325         return -1;
1326 }
1327
1328 /*
1329  * Note that this doesn't return the actual pagesize, but
1330  * the allocation granularity. If future Windows specific git code
1331  * needs the real getpagesize function, we need to find another solution.
1332  */
1333 int mingw_getpagesize(void)
1334 {
1335         SYSTEM_INFO si;
1336         GetSystemInfo(&si);
1337         return si.dwAllocationGranularity;
1338 }
1339
1340 struct passwd *getpwuid(int uid)
1341 {
1342         static char user_name[100];
1343         static struct passwd p;
1344
1345         DWORD len = sizeof(user_name);
1346         if (!GetUserName(user_name, &len))
1347                 return NULL;
1348         p.pw_name = user_name;
1349         p.pw_gecos = "unknown";
1350         p.pw_dir = NULL;
1351         return &p;
1352 }
1353
1354 static HANDLE timer_event;
1355 static HANDLE timer_thread;
1356 static int timer_interval;
1357 static int one_shot;
1358 static sig_handler_t timer_fn = SIG_DFL;
1359
1360 /* The timer works like this:
1361  * The thread, ticktack(), is a trivial routine that most of the time
1362  * only waits to receive the signal to terminate. The main thread tells
1363  * the thread to terminate by setting the timer_event to the signalled
1364  * state.
1365  * But ticktack() interrupts the wait state after the timer's interval
1366  * length to call the signal handler.
1367  */
1368
1369 static unsigned __stdcall ticktack(void *dummy)
1370 {
1371         while (WaitForSingleObject(timer_event, timer_interval) == WAIT_TIMEOUT) {
1372                 if (timer_fn == SIG_DFL)
1373                         die("Alarm");
1374                 if (timer_fn != SIG_IGN)
1375                         timer_fn(SIGALRM);
1376                 if (one_shot)
1377                         break;
1378         }
1379         return 0;
1380 }
1381
1382 static int start_timer_thread(void)
1383 {
1384         timer_event = CreateEvent(NULL, FALSE, FALSE, NULL);
1385         if (timer_event) {
1386                 timer_thread = (HANDLE) _beginthreadex(NULL, 0, ticktack, NULL, 0, NULL);
1387                 if (!timer_thread )
1388                         return errno = ENOMEM,
1389                                 error("cannot start timer thread");
1390         } else
1391                 return errno = ENOMEM,
1392                         error("cannot allocate resources for timer");
1393         return 0;
1394 }
1395
1396 static void stop_timer_thread(void)
1397 {
1398         if (timer_event)
1399                 SetEvent(timer_event);  /* tell thread to terminate */
1400         if (timer_thread) {
1401                 int rc = WaitForSingleObject(timer_thread, 1000);
1402                 if (rc == WAIT_TIMEOUT)
1403                         error("timer thread did not terminate timely");
1404                 else if (rc != WAIT_OBJECT_0)
1405                         error("waiting for timer thread failed: %lu",
1406                               GetLastError());
1407                 CloseHandle(timer_thread);
1408         }
1409         if (timer_event)
1410                 CloseHandle(timer_event);
1411         timer_event = NULL;
1412         timer_thread = NULL;
1413 }
1414
1415 static inline int is_timeval_eq(const struct timeval *i1, const struct timeval *i2)
1416 {
1417         return i1->tv_sec == i2->tv_sec && i1->tv_usec == i2->tv_usec;
1418 }
1419
1420 int setitimer(int type, struct itimerval *in, struct itimerval *out)
1421 {
1422         static const struct timeval zero;
1423         static int atexit_done;
1424
1425         if (out != NULL)
1426                 return errno = EINVAL,
1427                         error("setitimer param 3 != NULL not implemented");
1428         if (!is_timeval_eq(&in->it_interval, &zero) &&
1429             !is_timeval_eq(&in->it_interval, &in->it_value))
1430                 return errno = EINVAL,
1431                         error("setitimer: it_interval must be zero or eq it_value");
1432
1433         if (timer_thread)
1434                 stop_timer_thread();
1435
1436         if (is_timeval_eq(&in->it_value, &zero) &&
1437             is_timeval_eq(&in->it_interval, &zero))
1438                 return 0;
1439
1440         timer_interval = in->it_value.tv_sec * 1000 + in->it_value.tv_usec / 1000;
1441         one_shot = is_timeval_eq(&in->it_interval, &zero);
1442         if (!atexit_done) {
1443                 atexit(stop_timer_thread);
1444                 atexit_done = 1;
1445         }
1446         return start_timer_thread();
1447 }
1448
1449 int sigaction(int sig, struct sigaction *in, struct sigaction *out)
1450 {
1451         if (sig != SIGALRM)
1452                 return errno = EINVAL,
1453                         error("sigaction only implemented for SIGALRM");
1454         if (out != NULL)
1455                 return errno = EINVAL,
1456                         error("sigaction: param 3 != NULL not implemented");
1457
1458         timer_fn = in->sa_handler;
1459         return 0;
1460 }
1461
1462 #undef signal
1463 sig_handler_t mingw_signal(int sig, sig_handler_t handler)
1464 {
1465         sig_handler_t old = timer_fn;
1466         if (sig != SIGALRM)
1467                 return signal(sig, handler);
1468         timer_fn = handler;
1469         return old;
1470 }
1471
1472 static const char *make_backslash_path(const char *path)
1473 {
1474         static char buf[PATH_MAX + 1];
1475         char *c;
1476
1477         if (strlcpy(buf, path, PATH_MAX) >= PATH_MAX)
1478                 die("Too long path: %.*s", 60, path);
1479
1480         for (c = buf; *c; c++) {
1481                 if (*c == '/')
1482                         *c = '\\';
1483         }
1484         return buf;
1485 }
1486
1487 void mingw_open_html(const char *unixpath)
1488 {
1489         const char *htmlpath = make_backslash_path(unixpath);
1490         typedef HINSTANCE (WINAPI *T)(HWND, const char *,
1491                         const char *, const char *, const char *, INT);
1492         T ShellExecute;
1493         HMODULE shell32;
1494         int r;
1495
1496         shell32 = LoadLibrary("shell32.dll");
1497         if (!shell32)
1498                 die("cannot load shell32.dll");
1499         ShellExecute = (T)GetProcAddress(shell32, "ShellExecuteA");
1500         if (!ShellExecute)
1501                 die("cannot run browser");
1502
1503         printf("Launching default browser to display HTML ...\n");
1504         r = (int)ShellExecute(NULL, "open", htmlpath, NULL, "\\", SW_SHOWNORMAL);
1505         FreeLibrary(shell32);
1506         /* see the MSDN documentation referring to the result codes here */
1507         if (r <= 32) {
1508                 die("failed to launch browser for %.*s", MAX_PATH, unixpath);
1509         }
1510 }
1511
1512 int link(const char *oldpath, const char *newpath)
1513 {
1514         typedef BOOL (WINAPI *T)(const char*, const char*, LPSECURITY_ATTRIBUTES);
1515         static T create_hard_link = NULL;
1516         if (!create_hard_link) {
1517                 create_hard_link = (T) GetProcAddress(
1518                         GetModuleHandle("kernel32.dll"), "CreateHardLinkA");
1519                 if (!create_hard_link)
1520                         create_hard_link = (T)-1;
1521         }
1522         if (create_hard_link == (T)-1) {
1523                 errno = ENOSYS;
1524                 return -1;
1525         }
1526         if (!create_hard_link(newpath, oldpath, NULL)) {
1527                 errno = err_win_to_posix(GetLastError());
1528                 return -1;
1529         }
1530         return 0;
1531 }
1532
1533 char *getpass(const char *prompt)
1534 {
1535         struct strbuf buf = STRBUF_INIT;
1536
1537         fputs(prompt, stderr);
1538         for (;;) {
1539                 char c = _getch();
1540                 if (c == '\r' || c == '\n')
1541                         break;
1542                 strbuf_addch(&buf, c);
1543         }
1544         fputs("\n", stderr);
1545         return strbuf_detach(&buf, NULL);
1546 }
1547
1548 pid_t waitpid(pid_t pid, int *status, unsigned options)
1549 {
1550         HANDLE h = OpenProcess(SYNCHRONIZE | PROCESS_QUERY_INFORMATION,
1551             FALSE, pid);
1552         if (!h) {
1553                 errno = ECHILD;
1554                 return -1;
1555         }
1556
1557         if (pid > 0 && options & WNOHANG) {
1558                 if (WAIT_OBJECT_0 != WaitForSingleObject(h, 0)) {
1559                         CloseHandle(h);
1560                         return 0;
1561                 }
1562                 options &= ~WNOHANG;
1563         }
1564
1565         if (options == 0) {
1566                 struct pinfo_t **ppinfo;
1567                 if (WaitForSingleObject(h, INFINITE) != WAIT_OBJECT_0) {
1568                         CloseHandle(h);
1569                         return 0;
1570                 }
1571
1572                 if (status)
1573                         GetExitCodeProcess(h, (LPDWORD)status);
1574
1575                 EnterCriticalSection(&pinfo_cs);
1576
1577                 ppinfo = &pinfo;
1578                 while (*ppinfo) {
1579                         struct pinfo_t *info = *ppinfo;
1580                         if (info->pid == pid) {
1581                                 CloseHandle(info->proc);
1582                                 *ppinfo = info->next;
1583                                 free(info);
1584                                 break;
1585                         }
1586                         ppinfo = &info->next;
1587                 }
1588
1589                 LeaveCriticalSection(&pinfo_cs);
1590
1591                 CloseHandle(h);
1592                 return pid;
1593         }
1594         CloseHandle(h);
1595
1596         errno = EINVAL;
1597         return -1;
1598 }
1599
1600 #ifndef NO_MINGW_REPLACE_READDIR
1601 /* MinGW readdir implementation to avoid extra lstats for Git */
1602 struct mingw_DIR
1603 {
1604         struct _finddata_t      dd_dta;         /* disk transfer area for this dir */
1605         struct mingw_dirent     dd_dir;         /* Our own implementation, including d_type */
1606         long                    dd_handle;      /* _findnext handle */
1607         int                     dd_stat;        /* 0 = next entry to read is first entry, -1 = off the end, positive = 0 based index of next entry */
1608         char                    dd_name[1];     /* given path for dir with search pattern (struct is extended) */
1609 };
1610
1611 struct dirent *mingw_readdir(DIR *dir)
1612 {
1613         WIN32_FIND_DATAA buf;
1614         HANDLE handle;
1615         struct mingw_DIR *mdir = (struct mingw_DIR*)dir;
1616
1617         if (!dir->dd_handle) {
1618                 errno = EBADF; /* No set_errno for mingw */
1619                 return NULL;
1620         }
1621
1622         if (dir->dd_handle == (long)INVALID_HANDLE_VALUE && dir->dd_stat == 0)
1623         {
1624                 DWORD lasterr;
1625                 handle = FindFirstFileA(dir->dd_name, &buf);
1626                 lasterr = GetLastError();
1627                 dir->dd_handle = (long)handle;
1628                 if (handle == INVALID_HANDLE_VALUE && (lasterr != ERROR_NO_MORE_FILES)) {
1629                         errno = err_win_to_posix(lasterr);
1630                         return NULL;
1631                 }
1632         } else if (dir->dd_handle == (long)INVALID_HANDLE_VALUE) {
1633                 return NULL;
1634         } else if (!FindNextFileA((HANDLE)dir->dd_handle, &buf)) {
1635                 DWORD lasterr = GetLastError();
1636                 FindClose((HANDLE)dir->dd_handle);
1637                 dir->dd_handle = (long)INVALID_HANDLE_VALUE;
1638                 /* POSIX says you shouldn't set errno when readdir can't
1639                    find any more files; so, if another error we leave it set. */
1640                 if (lasterr != ERROR_NO_MORE_FILES)
1641                         errno = err_win_to_posix(lasterr);
1642                 return NULL;
1643         }
1644
1645         /* We get here if `buf' contains valid data.  */
1646         strcpy(dir->dd_dir.d_name, buf.cFileName);
1647         ++dir->dd_stat;
1648
1649         /* Set file type, based on WIN32_FIND_DATA */
1650         mdir->dd_dir.d_type = 0;
1651         if (buf.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
1652                 mdir->dd_dir.d_type |= DT_DIR;
1653         else
1654                 mdir->dd_dir.d_type |= DT_REG;
1655
1656         return (struct dirent*)&dir->dd_dir;
1657 }
1658 #endif // !NO_MINGW_REPLACE_READDIR