git.oblomov.eu Git - git/blob - compat/mingw.c

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