compat/mingw.c

   1 #include "../git-compat-util.h"
   2 #include "win32.h"
   3 #include <conio.h>
   4 #include <wchar.h>
   5 #include "../strbuf.h"
   6 #include "../run-command.h"
   7 #include "../cache.h"
   8
   9 static const int delay[] = { 0, 1, 10, 20, 40 };
  10
  11 int err_win_to_posix(DWORD winerr)
  12 {
  13         int error = ENOSYS;
  14         switch(winerr) {
  15         case ERROR_ACCESS_DENIED: error = EACCES; break;
  16         case ERROR_ACCOUNT_DISABLED: error = EACCES; break;
  17         case ERROR_ACCOUNT_RESTRICTION: error = EACCES; break;
  18         case ERROR_ALREADY_ASSIGNED: error = EBUSY; break;
  19         case ERROR_ALREADY_EXISTS: error = EEXIST; break;
  20         case ERROR_ARITHMETIC_OVERFLOW: error = ERANGE; break;
  21         case ERROR_BAD_COMMAND: error = EIO; break;
  22         case ERROR_BAD_DEVICE: error = ENODEV; break;
  23         case ERROR_BAD_DRIVER_LEVEL: error = ENXIO; break;
  24         case ERROR_BAD_EXE_FORMAT: error = ENOEXEC; break;
  25         case ERROR_BAD_FORMAT: error = ENOEXEC; break;
  26         case ERROR_BAD_LENGTH: error = EINVAL; break;
  27         case ERROR_BAD_PATHNAME: error = ENOENT; break;
  28         case ERROR_BAD_PIPE: error = EPIPE; break;
  29         case ERROR_BAD_UNIT: error = ENODEV; break;
  30         case ERROR_BAD_USERNAME: error = EINVAL; break;
  31         case ERROR_BROKEN_PIPE: error = EPIPE; break;
  32         case ERROR_BUFFER_OVERFLOW: error = ENAMETOOLONG; break;
  33         case ERROR_BUSY: error = EBUSY; break;
  34         case ERROR_BUSY_DRIVE: error = EBUSY; break;
  35         case ERROR_CALL_NOT_IMPLEMENTED: error = ENOSYS; break;
  36         case ERROR_CANNOT_MAKE: error = EACCES; break;
  37         case ERROR_CANTOPEN: error = EIO; break;
  38         case ERROR_CANTREAD: error = EIO; break;
  39         case ERROR_CANTWRITE: error = EIO; break;
  40         case ERROR_CRC: error = EIO; break;
  41         case ERROR_CURRENT_DIRECTORY: error = EACCES; break;
  42         case ERROR_DEVICE_IN_USE: error = EBUSY; break;
  43         case ERROR_DEV_NOT_EXIST: error = ENODEV; break;
  44         case ERROR_DIRECTORY: error = EINVAL; break;
  45         case ERROR_DIR_NOT_EMPTY: error = ENOTEMPTY; break;
  46         case ERROR_DISK_CHANGE: error = EIO; break;
  47         case ERROR_DISK_FULL: error = ENOSPC; break;
  48         case ERROR_DRIVE_LOCKED: error = EBUSY; break;
  49         case ERROR_ENVVAR_NOT_FOUND: error = EINVAL; break;
  50         case ERROR_EXE_MARKED_INVALID: error = ENOEXEC; break;
  51         case ERROR_FILENAME_EXCED_RANGE: error = ENAMETOOLONG; break;
  52         case ERROR_FILE_EXISTS: error = EEXIST; break;
  53         case ERROR_FILE_INVALID: error = ENODEV; break;
  54         case ERROR_FILE_NOT_FOUND: error = ENOENT; break;
  55         case ERROR_GEN_FAILURE: error = EIO; break;
  56         case ERROR_HANDLE_DISK_FULL: error = ENOSPC; break;
  57         case ERROR_INSUFFICIENT_BUFFER: error = ENOMEM; break;
  58         case ERROR_INVALID_ACCESS: error = EACCES; break;
  59         case ERROR_INVALID_ADDRESS: error = EFAULT; break;
  60         case ERROR_INVALID_BLOCK: error = EFAULT; break;
  61         case ERROR_INVALID_DATA: error = EINVAL; break;
  62         case ERROR_INVALID_DRIVE: error = ENODEV; break;
  63         case ERROR_INVALID_EXE_SIGNATURE: error = ENOEXEC; break;
  64         case ERROR_INVALID_FLAGS: error = EINVAL; break;
  65         case ERROR_INVALID_FUNCTION: error = ENOSYS; break;
  66         case ERROR_INVALID_HANDLE: error = EBADF; break;
  67         case ERROR_INVALID_LOGON_HOURS: error = EACCES; break;
  68         case ERROR_INVALID_NAME: error = EINVAL; break;
  69         case ERROR_INVALID_OWNER: error = EINVAL; break;
  70         case ERROR_INVALID_PARAMETER: error = EINVAL; break;
  71         case ERROR_INVALID_PASSWORD: error = EPERM; break;
  72         case ERROR_INVALID_PRIMARY_GROUP: error = EINVAL; break;
  73         case ERROR_INVALID_SIGNAL_NUMBER: error = EINVAL; break;
  74         case ERROR_INVALID_TARGET_HANDLE: error = EIO; break;
  75         case ERROR_INVALID_WORKSTATION: error = EACCES; break;
  76         case ERROR_IO_DEVICE: error = EIO; break;
  77         case ERROR_IO_INCOMPLETE: error = EINTR; break;
  78         case ERROR_LOCKED: error = EBUSY; break;
  79         case ERROR_LOCK_VIOLATION: error = EACCES; break;
  80         case ERROR_LOGON_FAILURE: error = EACCES; break;
  81         case ERROR_MAPPED_ALIGNMENT: error = EINVAL; break;
  82         case ERROR_META_EXPANSION_TOO_LONG: error = E2BIG; break;
  83         case ERROR_MORE_DATA: error = EPIPE; break;
  84         case ERROR_NEGATIVE_SEEK: error = ESPIPE; break;
  85         case ERROR_NOACCESS: error = EFAULT; break;
  86         case ERROR_NONE_MAPPED: error = EINVAL; break;
  87         case ERROR_NOT_ENOUGH_MEMORY: error = ENOMEM; break;
  88         case ERROR_NOT_READY: error = EAGAIN; break;
  89         case ERROR_NOT_SAME_DEVICE: error = EXDEV; break;
  90         case ERROR_NO_DATA: error = EPIPE; break;
  91         case ERROR_NO_MORE_SEARCH_HANDLES: error = EIO; break;
  92         case ERROR_NO_PROC_SLOTS: error = EAGAIN; break;
  93         case ERROR_NO_SUCH_PRIVILEGE: error = EACCES; break;
  94         case ERROR_OPEN_FAILED: error = EIO; break;
  95         case ERROR_OPEN_FILES: error = EBUSY; break;
  96         case ERROR_OPERATION_ABORTED: error = EINTR; break;
  97         case ERROR_OUTOFMEMORY: error = ENOMEM; break;
  98         case ERROR_PASSWORD_EXPIRED: error = EACCES; break;
  99         case ERROR_PATH_BUSY: error = EBUSY; break;
 100         case ERROR_PATH_NOT_FOUND: error = ENOENT; break;
 101         case ERROR_PIPE_BUSY: error = EBUSY; break;
 102         case ERROR_PIPE_CONNECTED: error = EPIPE; break;
 103         case ERROR_PIPE_LISTENING: error = EPIPE; break;
 104         case ERROR_PIPE_NOT_CONNECTED: error = EPIPE; break;
 105         case ERROR_PRIVILEGE_NOT_HELD: error = EACCES; break;
 106         case ERROR_READ_FAULT: error = EIO; break;
 107         case ERROR_SEEK: error = EIO; break;
 108         case ERROR_SEEK_ON_DEVICE: error = ESPIPE; break;
 109         case ERROR_SHARING_BUFFER_EXCEEDED: error = ENFILE; break;
 110         case ERROR_SHARING_VIOLATION: error = EACCES; break;
 111         case ERROR_STACK_OVERFLOW: error = ENOMEM; break;
 112         case ERROR_SWAPERROR: error = ENOENT; break;
 113         case ERROR_TOO_MANY_MODULES: error = EMFILE; break;
 114         case ERROR_TOO_MANY_OPEN_FILES: error = EMFILE; break;
 115         case ERROR_UNRECOGNIZED_MEDIA: error = ENXIO; break;
 116         case ERROR_UNRECOGNIZED_VOLUME: error = ENODEV; break;
 117         case ERROR_WAIT_NO_CHILDREN: error = ECHILD; break;
 118         case ERROR_WRITE_FAULT: error = EIO; break;
 119         case ERROR_WRITE_PROTECT: error = EROFS; break;
 120         }
 121         return error;
 122 }
 123
 124 static inline int is_file_in_use_error(DWORD errcode)
 125 {
 126         switch (errcode) {
 127         case ERROR_SHARING_VIOLATION:
 128         case ERROR_ACCESS_DENIED:
 129                 return 1;
 130         }
 131
 132         return 0;
 133 }
 134
 135 static int read_yes_no_answer(void)
 136 {
 137         char answer[1024];
 138
 139         if (fgets(answer, sizeof(answer), stdin)) {
 140                 size_t answer_len = strlen(answer);
 141                 int got_full_line = 0, c;
 142
 143                 /* remove the newline */
 144                 if (answer_len >= 2 && answer[answer_len-2] == '\r') {
 145                         answer[answer_len-2] = '\0';
 146                         got_full_line = 1;
 147                 } else if (answer_len >= 1 && answer[answer_len-1] == '\n') {
 148                         answer[answer_len-1] = '\0';
 149                         got_full_line = 1;
 150                 }
 151                 /* flush the buffer in case we did not get the full line */
 152                 if (!got_full_line)
 153                         while ((c = getchar()) != EOF && c != '\n')
 154                                 ;
 155         } else
 156                 /* we could not read, return the
 157                  * default answer which is no */
 158                 return 0;
 159
 160         if (tolower(answer[0]) == 'y' && !answer[1])
 161                 return 1;
 162         if (!strncasecmp(answer, "yes", sizeof(answer)))
 163                 return 1;
 164         if (tolower(answer[0]) == 'n' && !answer[1])
 165                 return 0;
 166         if (!strncasecmp(answer, "no", sizeof(answer)))
 167                 return 0;
 168
 169         /* did not find an answer we understand */
 170         return -1;
 171 }
 172
 173 static int ask_yes_no_if_possible(const char *format, ...)
 174 {
 175         char question[4096];
 176         const char *retry_hook[] = { NULL, NULL, NULL };
 177         va_list args;
 178
 179         va_start(args, format);
 180         vsnprintf(question, sizeof(question), format, args);
 181         va_end(args);
 182
 183         if ((retry_hook[0] = mingw_getenv("GIT_ASK_YESNO"))) {
 184                 retry_hook[1] = question;
 185                 return !run_command_v_opt(retry_hook, 0);
 186         }
 187
 188         if (!isatty(_fileno(stdin)) || !isatty(_fileno(stderr)))
 189                 return 0;
 190
 191         while (1) {
 192                 int answer;
 193                 fprintf(stderr, "%s (y/n) ", question);
 194
 195                 if ((answer = read_yes_no_answer()) >= 0)
 196                         return answer;
 197
 198                 fprintf(stderr, "Sorry, I did not understand your answer. "
 199                                 "Please type 'y' or 'n'\n");
 200         }
 201 }
 202
 203 int mingw_unlink(const char *pathname)
 204 {
 205         int ret, tries = 0;
 206         wchar_t wpathname[MAX_PATH];
 207         if (xutftowcs_path(wpathname, pathname) < 0)
 208                 return -1;
 209
 210         /* read-only files cannot be removed */
 211         _wchmod(wpathname, 0666);
 212         while ((ret = _wunlink(wpathname)) == -1 && tries < ARRAY_SIZE(delay)) {
 213                 if (!is_file_in_use_error(GetLastError()))
 214                         break;
 215                 /*
 216                  * We assume that some other process had the source or
 217                  * destination file open at the wrong moment and retry.
 218                  * In order to give the other process a higher chance to
 219                  * complete its operation, we give up our time slice now.
 220                  * If we have to retry again, we do sleep a bit.
 221                  */
 222                 Sleep(delay[tries]);
 223                 tries++;
 224         }
 225         while (ret == -1 && is_file_in_use_error(GetLastError()) &&
 226                ask_yes_no_if_possible("Unlink of file '%s' failed. "
 227                         "Should I try again?", pathname))
 228                ret = _wunlink(wpathname);
 229         return ret;
 230 }
 231
 232 static int is_dir_empty(const wchar_t *wpath)
 233 {
 234         WIN32_FIND_DATAW findbuf;
 235         HANDLE handle;
 236         wchar_t wbuf[MAX_PATH + 2];
 237         wcscpy(wbuf, wpath);
 238         wcscat(wbuf, L"\\*");
 239         handle = FindFirstFileW(wbuf, &findbuf);
 240         if (handle == INVALID_HANDLE_VALUE)
 241                 return GetLastError() == ERROR_NO_MORE_FILES;
 242
 243         while (!wcscmp(findbuf.cFileName, L".") ||
 244                         !wcscmp(findbuf.cFileName, L".."))
 245                 if (!FindNextFileW(handle, &findbuf)) {
 246                         DWORD err = GetLastError();
 247                         FindClose(handle);
 248                         return err == ERROR_NO_MORE_FILES;
 249                 }
 250         FindClose(handle);
 251         return 0;
 252 }
 253
 254 int mingw_rmdir(const char *pathname)
 255 {
 256         int ret, tries = 0;
 257         wchar_t wpathname[MAX_PATH];
 258         if (xutftowcs_path(wpathname, pathname) < 0)
 259                 return -1;
 260
 261         while ((ret = _wrmdir(wpathname)) == -1 && tries < ARRAY_SIZE(delay)) {
 262                 if (!is_file_in_use_error(GetLastError()))
 263                         errno = err_win_to_posix(GetLastError());
 264                 if (errno != EACCES)
 265                         break;
 266                 if (!is_dir_empty(wpathname)) {
 267                         errno = ENOTEMPTY;
 268                         break;
 269                 }
 270                 /*
 271                  * We assume that some other process had the source or
 272                  * destination file open at the wrong moment and retry.
 273                  * In order to give the other process a higher chance to
 274                  * complete its operation, we give up our time slice now.
 275                  * If we have to retry again, we do sleep a bit.
 276                  */
 277                 Sleep(delay[tries]);
 278                 tries++;
 279         }
 280         while (ret == -1 && errno == EACCES && is_file_in_use_error(GetLastError()) &&
 281                ask_yes_no_if_possible("Deletion of directory '%s' failed. "
 282                         "Should I try again?", pathname))
 283                ret = _wrmdir(wpathname);
 284         return ret;
 285 }
 286
 287 int mingw_mkdir(const char *path, int mode)
 288 {
 289         int ret;
 290         wchar_t wpath[MAX_PATH];
 291         if (xutftowcs_path(wpath, path) < 0)
 292                 return -1;
 293         ret = _wmkdir(wpath);
 294         return ret;
 295 }
 296
 297 int mingw_open (const char *filename, int oflags, ...)
 298 {
 299         va_list args;
 300         unsigned mode;
 301         int fd;
 302         wchar_t wfilename[MAX_PATH];
 303
 304         va_start(args, oflags);
 305         mode = va_arg(args, int);
 306         va_end(args);
 307
 308         if (filename && !strcmp(filename, "/dev/null"))
 309                 filename = "nul";
 310
 311         if (xutftowcs_path(wfilename, filename) < 0)
 312                 return -1;
 313         fd = _wopen(wfilename, oflags, mode);
 314
 315         if (fd < 0 && (oflags & O_CREAT) && errno == EACCES) {
 316                 DWORD attrs = GetFileAttributesW(wfilename);
 317                 if (attrs != INVALID_FILE_ATTRIBUTES && (attrs & FILE_ATTRIBUTE_DIRECTORY))
 318                         errno = EISDIR;
 319         }
 320         return fd;
 321 }
 322
 323 static BOOL WINAPI ctrl_ignore(DWORD type)
 324 {
 325         return TRUE;
 326 }
 327
 328 #undef fgetc
 329 int mingw_fgetc(FILE *stream)
 330 {
 331         int ch;
 332         if (!isatty(_fileno(stream)))
 333                 return fgetc(stream);
 334
 335         SetConsoleCtrlHandler(ctrl_ignore, TRUE);
 336         while (1) {
 337                 ch = fgetc(stream);
 338                 if (ch != EOF || GetLastError() != ERROR_OPERATION_ABORTED)
 339                         break;
 340
 341                 /* Ctrl+C was pressed, simulate SIGINT and retry */
 342                 mingw_raise(SIGINT);
 343         }
 344         SetConsoleCtrlHandler(ctrl_ignore, FALSE);
 345         return ch;
 346 }
 347
 348 #undef fopen
 349 FILE *mingw_fopen (const char *filename, const char *otype)
 350 {
 351         FILE *file;
 352         wchar_t wfilename[MAX_PATH], wotype[4];
 353         if (filename && !strcmp(filename, "/dev/null"))
 354                 filename = "nul";
 355         if (xutftowcs_path(wfilename, filename) < 0 ||
 356                 xutftowcs(wotype, otype, ARRAY_SIZE(wotype)) < 0)
 357                 return NULL;
 358         file = _wfopen(wfilename, wotype);
 359         return file;
 360 }
 361
 362 FILE *mingw_freopen (const char *filename, const char *otype, FILE *stream)
 363 {
 364         FILE *file;
 365         wchar_t wfilename[MAX_PATH], wotype[4];
 366         if (filename && !strcmp(filename, "/dev/null"))
 367                 filename = "nul";
 368         if (xutftowcs_path(wfilename, filename) < 0 ||
 369                 xutftowcs(wotype, otype, ARRAY_SIZE(wotype)) < 0)
 370                 return NULL;
 371         file = _wfreopen(wfilename, wotype, stream);
 372         return file;
 373 }
 374
 375 #undef fflush
 376 int mingw_fflush(FILE *stream)
 377 {
 378         int ret = fflush(stream);
 379
 380         /*
 381          * write() is used behind the scenes of stdio output functions.
 382          * Since git code does not check for errors after each stdio write
 383          * operation, it can happen that write() is called by a later
 384          * stdio function even if an earlier write() call failed. In the
 385          * case of a pipe whose readable end was closed, only the first
 386          * call to write() reports EPIPE on Windows. Subsequent write()
 387          * calls report EINVAL. It is impossible to notice whether this
 388          * fflush invocation triggered such a case, therefore, we have to
 389          * catch all EINVAL errors whole-sale.
 390          */
 391         if (ret && errno == EINVAL)
 392                 errno = EPIPE;
 393
 394         return ret;
 395 }
 396
 397 int mingw_access(const char *filename, int mode)
 398 {
 399         wchar_t wfilename[MAX_PATH];
 400         if (xutftowcs_path(wfilename, filename) < 0)
 401                 return -1;
 402         /* X_OK is not supported by the MSVCRT version */
 403         return _waccess(wfilename, mode & ~X_OK);
 404 }
 405
 406 int mingw_chdir(const char *dirname)
 407 {
 408         wchar_t wdirname[MAX_PATH];
 409         if (xutftowcs_path(wdirname, dirname) < 0)
 410                 return -1;
 411         return _wchdir(wdirname);
 412 }
 413
 414 int mingw_chmod(const char *filename, int mode)
 415 {
 416         wchar_t wfilename[MAX_PATH];
 417         if (xutftowcs_path(wfilename, filename) < 0)
 418                 return -1;
 419         return _wchmod(wfilename, mode);
 420 }
 421
 422 /*
 423  * The unit of FILETIME is 100-nanoseconds since January 1, 1601, UTC.
 424  * Returns the 100-nanoseconds ("hekto nanoseconds") since the epoch.
 425  */
 426 static inline long long filetime_to_hnsec(const FILETIME *ft)
 427 {
 428         long long winTime = ((long long)ft->dwHighDateTime << 32) + ft->dwLowDateTime;
 429         /* Windows to Unix Epoch conversion */
 430         return winTime - 116444736000000000LL;
 431 }
 432
 433 static inline time_t filetime_to_time_t(const FILETIME *ft)
 434 {
 435         return (time_t)(filetime_to_hnsec(ft) / 10000000);
 436 }
 437
 438 /* We keep the do_lstat code in a separate function to avoid recursion.
 439  * When a path ends with a slash, the stat will fail with ENOENT. In
 440  * this case, we strip the trailing slashes and stat again.
 441  *
 442  * If follow is true then act like stat() and report on the link
 443  * target. Otherwise report on the link itself.
 444  */
 445 static int do_lstat(int follow, const char *file_name, struct stat *buf)
 446 {
 447         WIN32_FILE_ATTRIBUTE_DATA fdata;
 448         wchar_t wfilename[MAX_PATH];
 449         if (xutftowcs_path(wfilename, file_name) < 0)
 450                 return -1;
 451
 452         if (GetFileAttributesExW(wfilename, GetFileExInfoStandard, &fdata)) {
 453                 buf->st_ino = 0;
 454                 buf->st_gid = 0;
 455                 buf->st_uid = 0;
 456                 buf->st_nlink = 1;
 457                 buf->st_mode = file_attr_to_st_mode(fdata.dwFileAttributes);
 458                 buf->st_size = fdata.nFileSizeLow |
 459                         (((off_t)fdata.nFileSizeHigh)<<32);
 460                 buf->st_dev = buf->st_rdev = 0; /* not used by Git */
 461                 buf->st_atime = filetime_to_time_t(&(fdata.ftLastAccessTime));
 462                 buf->st_mtime = filetime_to_time_t(&(fdata.ftLastWriteTime));
 463                 buf->st_ctime = filetime_to_time_t(&(fdata.ftCreationTime));
 464                 if (fdata.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) {
 465                         WIN32_FIND_DATAW findbuf;
 466                         HANDLE handle = FindFirstFileW(wfilename, &findbuf);
 467                         if (handle != INVALID_HANDLE_VALUE) {
 468                                 if ((findbuf.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) &&
 469                                                 (findbuf.dwReserved0 == IO_REPARSE_TAG_SYMLINK)) {
 470                                         if (follow) {
 471                                                 char buffer[MAXIMUM_REPARSE_DATA_BUFFER_SIZE];
 472                                                 buf->st_size = readlink(file_name, buffer, MAXIMUM_REPARSE_DATA_BUFFER_SIZE);
 473                                         } else {
 474                                                 buf->st_mode = S_IFLNK;
 475                                         }
 476                                         buf->st_mode |= S_IREAD;
 477                                         if (!(findbuf.dwFileAttributes & FILE_ATTRIBUTE_READONLY))
 478                                                 buf->st_mode |= S_IWRITE;
 479                                 }
 480                                 FindClose(handle);
 481                         }
 482                 }
 483                 return 0;
 484         }
 485         switch (GetLastError()) {
 486         case ERROR_ACCESS_DENIED:
 487         case ERROR_SHARING_VIOLATION:
 488         case ERROR_LOCK_VIOLATION:
 489         case ERROR_SHARING_BUFFER_EXCEEDED:
 490                 errno = EACCES;
 491                 break;
 492         case ERROR_BUFFER_OVERFLOW:
 493                 errno = ENAMETOOLONG;
 494                 break;
 495         case ERROR_NOT_ENOUGH_MEMORY:
 496                 errno = ENOMEM;
 497                 break;
 498         default:
 499                 errno = ENOENT;
 500                 break;
 501         }
 502         return -1;
 503 }
 504
 505 /* We provide our own lstat/fstat functions, since the provided
 506  * lstat/fstat functions are so slow. These stat functions are
 507  * tailored for Git's usage (read: fast), and are not meant to be
 508  * complete. Note that Git stat()s are redirected to mingw_lstat()
 509  * too, since Windows doesn't really handle symlinks that well.
 510  */
 511 static int do_stat_internal(int follow, const char *file_name, struct stat *buf)
 512 {
 513         int namelen;
 514         char alt_name[PATH_MAX];
 515
 516         if (!do_lstat(follow, file_name, buf))
 517                 return 0;
 518
 519         /* if file_name ended in a '/', Windows returned ENOENT;
 520          * try again without trailing slashes
 521          */
 522         if (errno != ENOENT)
 523                 return -1;
 524
 525         namelen = strlen(file_name);
 526         if (namelen && file_name[namelen-1] != '/')
 527                 return -1;
 528         while (namelen && file_name[namelen-1] == '/')
 529                 --namelen;
 530         if (!namelen || namelen >= PATH_MAX)
 531                 return -1;
 532
 533         memcpy(alt_name, file_name, namelen);
 534         alt_name[namelen] = 0;
 535         return do_lstat(follow, alt_name, buf);
 536 }
 537
 538 int mingw_lstat(const char *file_name, struct stat *buf)
 539 {
 540         return do_stat_internal(0, file_name, buf);
 541 }
 542 int mingw_stat(const char *file_name, struct stat *buf)
 543 {
 544         return do_stat_internal(1, file_name, buf);
 545 }
 546
 547 int mingw_fstat(int fd, struct stat *buf)
 548 {
 549         HANDLE fh = (HANDLE)_get_osfhandle(fd);
 550         BY_HANDLE_FILE_INFORMATION fdata;
 551
 552         if (fh == INVALID_HANDLE_VALUE) {
 553                 errno = EBADF;
 554                 return -1;
 555         }
 556         /* direct non-file handles to MS's fstat() */
 557         if (GetFileType(fh) != FILE_TYPE_DISK)
 558                 return _fstati64(fd, buf);
 559
 560         if (GetFileInformationByHandle(fh, &fdata)) {
 561                 buf->st_ino = 0;
 562                 buf->st_gid = 0;
 563                 buf->st_uid = 0;
 564                 buf->st_nlink = 1;
 565                 buf->st_mode = file_attr_to_st_mode(fdata.dwFileAttributes);
 566                 buf->st_size = fdata.nFileSizeLow |
 567                         (((off_t)fdata.nFileSizeHigh)<<32);
 568                 buf->st_dev = buf->st_rdev = 0; /* not used by Git */
 569                 buf->st_atime = filetime_to_time_t(&(fdata.ftLastAccessTime));
 570                 buf->st_mtime = filetime_to_time_t(&(fdata.ftLastWriteTime));
 571                 buf->st_ctime = filetime_to_time_t(&(fdata.ftCreationTime));
 572                 return 0;
 573         }
 574         errno = EBADF;
 575         return -1;
 576 }
 577
 578 static inline void time_t_to_filetime(time_t t, FILETIME *ft)
 579 {
 580         long long winTime = t * 10000000LL + 116444736000000000LL;
 581         ft->dwLowDateTime = winTime;
 582         ft->dwHighDateTime = winTime >> 32;
 583 }
 584
 585 int mingw_utime (const char *file_name, const struct utimbuf *times)
 586 {
 587         FILETIME mft, aft;
 588         int fh, rc;
 589         DWORD attrs;
 590         wchar_t wfilename[MAX_PATH];
 591         if (xutftowcs_path(wfilename, file_name) < 0)
 592                 return -1;
 593
 594         /* must have write permission */
 595         attrs = GetFileAttributesW(wfilename);
 596         if (attrs != INVALID_FILE_ATTRIBUTES &&
 597             (attrs & FILE_ATTRIBUTE_READONLY)) {
 598                 /* ignore errors here; open() will report them */
 599                 SetFileAttributesW(wfilename, attrs & ~FILE_ATTRIBUTE_READONLY);
 600         }
 601
 602         if ((fh = _wopen(wfilename, O_RDWR | O_BINARY)) < 0) {
 603                 rc = -1;
 604                 goto revert_attrs;
 605         }
 606
 607         if (times) {
 608                 time_t_to_filetime(times->modtime, &mft);
 609                 time_t_to_filetime(times->actime, &aft);
 610         } else {
 611                 GetSystemTimeAsFileTime(&mft);
 612                 aft = mft;
 613         }
 614         if (!SetFileTime((HANDLE)_get_osfhandle(fh), NULL, &aft, &mft)) {
 615                 errno = EINVAL;
 616                 rc = -1;
 617         } else
 618                 rc = 0;
 619         close(fh);
 620
 621 revert_attrs:
 622         if (attrs != INVALID_FILE_ATTRIBUTES &&
 623             (attrs & FILE_ATTRIBUTE_READONLY)) {
 624                 /* ignore errors again */
 625                 SetFileAttributesW(wfilename, attrs);
 626         }
 627         return rc;
 628 }
 629
 630 unsigned int sleep (unsigned int seconds)
 631 {
 632         Sleep(seconds*1000);
 633         return 0;
 634 }
 635
 636 char *mingw_mktemp(char *template)
 637 {
 638         wchar_t wtemplate[MAX_PATH];
 639         if (xutftowcs_path(wtemplate, template) < 0)
 640                 return NULL;
 641         if (!_wmktemp(wtemplate))
 642                 return NULL;
 643         if (xwcstoutf(template, wtemplate, strlen(template) + 1) < 0)
 644                 return NULL;
 645         return template;
 646 }
 647
 648 int mkstemp(char *template)
 649 {
 650         char *filename = mktemp(template);
 651         if (filename == NULL)
 652                 return -1;
 653         return open(filename, O_RDWR | O_CREAT, 0600);
 654 }
 655
 656 int gettimeofday(struct timeval *tv, void *tz)
 657 {
 658         FILETIME ft;
 659         long long hnsec;
 660
 661         GetSystemTimeAsFileTime(&ft);
 662         hnsec = filetime_to_hnsec(&ft);
 663         tv->tv_sec = hnsec / 10000000;
 664         tv->tv_usec = (hnsec % 10000000) / 10;
 665         return 0;
 666 }
 667
 668 int pipe(int filedes[2])
 669 {
 670         HANDLE h[2];
 671
 672         /* this creates non-inheritable handles */
 673         if (!CreatePipe(&h[0], &h[1], NULL, 8192)) {
 674                 errno = err_win_to_posix(GetLastError());
 675                 return -1;
 676         }
 677         filedes[0] = _open_osfhandle((int)h[0], O_NOINHERIT);
 678         if (filedes[0] < 0) {
 679                 CloseHandle(h[0]);
 680                 CloseHandle(h[1]);
 681                 return -1;
 682         }
 683         filedes[1] = _open_osfhandle((int)h[1], O_NOINHERIT);
 684         if (filedes[0] < 0) {
 685                 close(filedes[0]);
 686                 CloseHandle(h[1]);
 687                 return -1;
 688         }
 689         return 0;
 690 }
 691
 692 struct tm *gmtime_r(const time_t *timep, struct tm *result)
 693 {
 694         /* gmtime() in MSVCRT.DLL is thread-safe, but not reentrant */
 695         memcpy(result, gmtime(timep), sizeof(struct tm));
 696         return result;
 697 }
 698
 699 struct tm *localtime_r(const time_t *timep, struct tm *result)
 700 {
 701         /* localtime() in MSVCRT.DLL is thread-safe, but not reentrant */
 702         memcpy(result, localtime(timep), sizeof(struct tm));
 703         return result;
 704 }
 705
 706 char *mingw_getcwd(char *pointer, int len)
 707 {
 708         int i;
 709         wchar_t wpointer[MAX_PATH];
 710         if (!_wgetcwd(wpointer, ARRAY_SIZE(wpointer)))
 711                 return NULL;
 712         if (xwcstoutf(pointer, wpointer, len) < 0)
 713                 return NULL;
 714         for (i = 0; pointer[i]; i++)
 715                 if (pointer[i] == '\\')
 716                         pointer[i] = '/';
 717         return pointer;
 718 }
 719
 720 /*
 721  * See http://msdn2.microsoft.com/en-us/library/17w5ykft(vs.71).aspx
 722  * (Parsing C++ Command-Line Arguments)
 723  */
 724 static const char *quote_arg(const char *arg)
 725 {
 726         /* count chars to quote */
 727         int len = 0, n = 0;
 728         int force_quotes = 0;
 729         char *q, *d;
 730         const char *p = arg;
 731         if (!*p) force_quotes = 1;
 732         while (*p) {
 733                 if (isspace(*p) || *p == '*' || *p == '?' || *p == '{' || *p == '\'')
 734                         force_quotes = 1;
 735                 else if (*p == '"')
 736                         n++;
 737                 else if (*p == '\\') {
 738                         int count = 0;
 739                         while (*p == '\\') {
 740                                 count++;
 741                                 p++;
 742                                 len++;
 743                         }
 744                         if (*p == '"')
 745                                 n += count*2 + 1;
 746                         continue;
 747                 }
 748                 len++;
 749                 p++;
 750         }
 751         if (!force_quotes && n == 0)
 752                 return arg;
 753
 754         /* insert \ where necessary */
 755         d = q = xmalloc(len+n+3);
 756         *d++ = '"';
 757         while (*arg) {
 758                 if (*arg == '"')
 759                         *d++ = '\\';
 760                 else if (*arg == '\\') {
 761                         int count = 0;
 762                         while (*arg == '\\') {
 763                                 count++;
 764                                 *d++ = *arg++;
 765                         }
 766                         if (*arg == '"') {
 767                                 while (count-- > 0)
 768                                         *d++ = '\\';
 769                                 *d++ = '\\';
 770                         }
 771                 }
 772                 *d++ = *arg++;
 773         }
 774         *d++ = '"';
 775         *d++ = 0;
 776         return q;
 777 }
 778
 779 static const char *parse_interpreter(const char *cmd)
 780 {
 781         static char buf[100];
 782         char *p, *opt;
 783         int n, fd;
 784
 785         /* don't even try a .exe */
 786         n = strlen(cmd);
 787         if (n >= 4 && !strcasecmp(cmd+n-4, ".exe"))
 788                 return NULL;
 789
 790         fd = open(cmd, O_RDONLY);
 791         if (fd < 0)
 792                 return NULL;
 793         n = read(fd, buf, sizeof(buf)-1);
 794         close(fd);
 795         if (n < 4)      /* at least '#!/x' and not error */
 796                 return NULL;
 797
 798         if (buf[0] != '#' || buf[1] != '!')
 799                 return NULL;
 800         buf[n] = '\0';
 801         p = buf + strcspn(buf, "\r\n");
 802         if (!*p)
 803                 return NULL;
 804
 805         *p = '\0';
 806         if (!(p = strrchr(buf+2, '/')) && !(p = strrchr(buf+2, '\\')))
 807                 return NULL;
 808         /* strip options */
 809         if ((opt = strchr(p+1, ' ')))
 810                 *opt = '\0';
 811         return p+1;
 812 }
 813
 814 /*
 815  * Splits the PATH into parts.
 816  */
 817 static char **get_path_split(void)
 818 {
 819         char *p, **path, *envpath = mingw_getenv("PATH");
 820         int i, n = 0;
 821
 822         if (!envpath || !*envpath)
 823                 return NULL;
 824
 825         envpath = xstrdup(envpath);
 826         p = envpath;
 827         while (p) {
 828                 char *dir = p;
 829                 p = strchr(p, ';');
 830                 if (p) *p++ = '\0';
 831                 if (*dir) {     /* not earlier, catches series of ; */
 832                         ++n;
 833                 }
 834         }
 835         if (!n)
 836                 return NULL;
 837
 838         path = xmalloc((n+1)*sizeof(char *));
 839         p = envpath;
 840         i = 0;
 841         do {
 842                 if (*p)
 843                         path[i++] = xstrdup(p);
 844                 p = p+strlen(p)+1;
 845         } while (i < n);
 846         path[i] = NULL;
 847
 848         free(envpath);
 849
 850         return path;
 851 }
 852
 853 static void free_path_split(char **path)
 854 {
 855         char **p = path;
 856
 857         if (!path)
 858                 return;
 859
 860         while (*p)
 861                 free(*p++);
 862         free(path);
 863 }
 864
 865 /*
 866  * exe_only means that we only want to detect .exe files, but not scripts
 867  * (which do not have an extension)
 868  */
 869 static char *lookup_prog(const char *dir, const char *cmd, int isexe, int exe_only)
 870 {
 871         char path[MAX_PATH];
 872         snprintf(path, sizeof(path), "%s/%s.exe", dir, cmd);
 873
 874         if (!isexe && access(path, F_OK) == 0)
 875                 return xstrdup(path);
 876         path[strlen(path)-4] = '\0';
 877         if ((!exe_only || isexe) && access(path, F_OK) == 0)
 878                 if (!(GetFileAttributes(path) & FILE_ATTRIBUTE_DIRECTORY))
 879                         return xstrdup(path);
 880         return NULL;
 881 }
 882
 883 /*
 884  * Determines the absolute path of cmd using the split path in path.
 885  * If cmd contains a slash or backslash, no lookup is performed.
 886  */
 887 static char *path_lookup(const char *cmd, char **path, int exe_only)
 888 {
 889         char *prog = NULL;
 890         int len = strlen(cmd);
 891         int isexe = len >= 4 && !strcasecmp(cmd+len-4, ".exe");
 892
 893         if (strchr(cmd, '/') || strchr(cmd, '\\'))
 894                 prog = xstrdup(cmd);
 895
 896         while (!prog && *path)
 897                 prog = lookup_prog(*path++, cmd, isexe, exe_only);
 898
 899         return prog;
 900 }
 901
 902 static char **do_putenv(char **env, const char *name, int free_old);
 903
 904 static int compareenv(const void *a, const void *b)
 905 {
 906         char *const *ea = a;
 907         char *const *eb = b;
 908         return strcasecmp(*ea, *eb);
 909 }
 910
 911 /*
 912  * Create environment block suitable for CreateProcess. Merges current
 913  * process environment and the supplied environment changes.
 914  */
 915 static wchar_t *make_environment_block(char **deltaenv)
 916 {
 917         wchar_t *wenvblk = NULL;
 918         int count = 0;
 919         char **e, **tmpenv;
 920         int size = 0, wenvsz = 0, wenvpos = 0;
 921
 922         while (environ[count])
 923                 count++;
 924
 925         /* copy the environment */
 926         tmpenv = xmalloc(sizeof(*tmpenv) * (count + 1));
 927         memcpy(tmpenv, environ, sizeof(*tmpenv) * (count + 1));
 928
 929         /* merge supplied environment changes into the temporary environment */
 930         for (e = deltaenv; e && *e; e++)
 931                 tmpenv = do_putenv(tmpenv, *e, 0);
 932
 933         /* environment must be sorted */
 934         for (count = 0; tmpenv[count]; )
 935                 count++;
 936         qsort(tmpenv, count, sizeof(*tmpenv), compareenv);
 937
 938         /* create environment block from temporary environment */
 939         for (e = tmpenv; *e; e++) {
 940                 size = 2 * strlen(*e) + 2; /* +2 for final \0 */
 941                 ALLOC_GROW(wenvblk, (wenvpos + size) * sizeof(wchar_t), wenvsz);
 942                 wenvpos += xutftowcs(&wenvblk[wenvpos], *e, size) + 1;
 943         }
 944         /* add final \0 terminator */
 945         wenvblk[wenvpos] = 0;
 946         free(tmpenv);
 947         return wenvblk;
 948 }
 949
 950 struct pinfo_t {
 951         struct pinfo_t *next;
 952         pid_t pid;
 953         HANDLE proc;
 954 };
 955 static struct pinfo_t *pinfo = NULL;
 956 CRITICAL_SECTION pinfo_cs;
 957
 958 static pid_t mingw_spawnve_fd(const char *cmd, const char **argv, char **deltaenv,
 959                               const char *dir,
 960                               int prepend_cmd, int fhin, int fhout, int fherr)
 961 {
 962         STARTUPINFOW si;
 963         PROCESS_INFORMATION pi;
 964         struct strbuf args;
 965         wchar_t wcmd[MAX_PATH], wdir[MAX_PATH], *wargs, *wenvblk = NULL;
 966         unsigned flags = CREATE_UNICODE_ENVIRONMENT;
 967         BOOL ret;
 968
 969         /* Determine whether or not we are associated to a console */
 970         HANDLE cons = CreateFile("CONOUT$", GENERIC_WRITE,
 971                         FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
 972                         FILE_ATTRIBUTE_NORMAL, NULL);
 973         if (cons == INVALID_HANDLE_VALUE) {
 974                 /* There is no console associated with this process.
 975                  * Since the child is a console process, Windows
 976                  * would normally create a console window. But
 977                  * since we'll be redirecting std streams, we do
 978                  * not need the console.
 979                  * It is necessary to use DETACHED_PROCESS
 980                  * instead of CREATE_NO_WINDOW to make ssh
 981                  * recognize that it has no console.
 982                  */
 983                 flags |= DETACHED_PROCESS;
 984         } else {
 985                 /* There is already a console. If we specified
 986                  * DETACHED_PROCESS here, too, Windows would
 987                  * disassociate the child from the console.
 988                  * The same is true for CREATE_NO_WINDOW.
 989                  * Go figure!
 990                  */
 991                 CloseHandle(cons);
 992         }
 993         memset(&si, 0, sizeof(si));
 994         si.cb = sizeof(si);
 995         si.dwFlags = STARTF_USESTDHANDLES;
 996         si.hStdInput = winansi_get_osfhandle(fhin);
 997         si.hStdOutput = winansi_get_osfhandle(fhout);
 998         si.hStdError = winansi_get_osfhandle(fherr);
 999
1000         if (xutftowcs_path(wcmd, cmd) < 0)
1001                 return -1;
1002         if (dir && xutftowcs_path(wdir, dir) < 0)
1003                 return -1;
1004
1005         /* concatenate argv, quoting args as we go */
1006         strbuf_init(&args, 0);
1007         if (prepend_cmd) {
1008                 char *quoted = (char *)quote_arg(cmd);
1009                 strbuf_addstr(&args, quoted);
1010                 if (quoted != cmd)
1011                         free(quoted);
1012         }
1013         for (; *argv; argv++) {
1014                 char *quoted = (char *)quote_arg(*argv);
1015                 if (*args.buf)
1016                         strbuf_addch(&args, ' ');
1017                 strbuf_addstr(&args, quoted);
1018                 if (quoted != *argv)
1019                         free(quoted);
1020         }
1021
1022         wargs = xmalloc((2 * args.len + 1) * sizeof(wchar_t));
1023         xutftowcs(wargs, args.buf, 2 * args.len + 1);
1024         strbuf_release(&args);
1025
1026         wenvblk = make_environment_block(deltaenv);
1027
1028         memset(&pi, 0, sizeof(pi));
1029         ret = CreateProcessW(wcmd, wargs, NULL, NULL, TRUE, flags,
1030                 wenvblk, dir ? wdir : NULL, &si, &pi);
1031
1032         free(wenvblk);
1033         free(wargs);
1034
1035         if (!ret) {
1036                 errno = ENOENT;
1037                 return -1;
1038         }
1039         CloseHandle(pi.hThread);
1040
1041         /*
1042          * The process ID is the human-readable identifier of the process
1043          * that we want to present in log and error messages. The handle
1044          * is not useful for this purpose. But we cannot close it, either,
1045          * because it is not possible to turn a process ID into a process
1046          * handle after the process terminated.
1047          * Keep the handle in a list for waitpid.
1048          */
1049         EnterCriticalSection(&pinfo_cs);
1050         {
1051                 struct pinfo_t *info = xmalloc(sizeof(struct pinfo_t));
1052                 info->pid = pi.dwProcessId;
1053                 info->proc = pi.hProcess;
1054                 info->next = pinfo;
1055                 pinfo = info;
1056         }
1057         LeaveCriticalSection(&pinfo_cs);
1058
1059         return (pid_t)pi.dwProcessId;
1060 }
1061
1062 static pid_t mingw_spawnv(const char *cmd, const char **argv, int prepend_cmd)
1063 {
1064         return mingw_spawnve_fd(cmd, argv, NULL, NULL, prepend_cmd, 0, 1, 2);
1065 }
1066
1067 pid_t mingw_spawnvpe(const char *cmd, const char **argv, char **deltaenv,
1068                      const char *dir,
1069                      int fhin, int fhout, int fherr)
1070 {
1071         pid_t pid;
1072         char **path = get_path_split();
1073         char *prog = path_lookup(cmd, path, 0);
1074
1075         if (!prog) {
1076                 errno = ENOENT;
1077                 pid = -1;
1078         }
1079         else {
1080                 const char *interpr = parse_interpreter(prog);
1081
1082                 if (interpr) {
1083                         const char *argv0 = argv[0];
1084                         char *iprog = path_lookup(interpr, path, 1);
1085                         argv[0] = prog;
1086                         if (!iprog) {
1087                                 errno = ENOENT;
1088                                 pid = -1;
1089                         }
1090                         else {
1091                                 pid = mingw_spawnve_fd(iprog, argv, deltaenv, dir, 1,
1092                                                        fhin, fhout, fherr);
1093                                 free(iprog);
1094                         }
1095                         argv[0] = argv0;
1096                 }
1097                 else
1098                         pid = mingw_spawnve_fd(prog, argv, deltaenv, dir, 0,
1099                                                fhin, fhout, fherr);
1100                 free(prog);
1101         }
1102         free_path_split(path);
1103         return pid;
1104 }
1105
1106 static int try_shell_exec(const char *cmd, char *const *argv)
1107 {
1108         const char *interpr = parse_interpreter(cmd);
1109         char **path;
1110         char *prog;
1111         int pid = 0;
1112
1113         if (!interpr)
1114                 return 0;
1115         path = get_path_split();
1116         prog = path_lookup(interpr, path, 1);
1117         if (prog) {
1118                 int argc = 0;
1119                 const char **argv2;
1120                 while (argv[argc]) argc++;
1121                 argv2 = xmalloc(sizeof(*argv) * (argc+1));
1122                 argv2[0] = (char *)cmd; /* full path to the script file */
1123                 memcpy(&argv2[1], &argv[1], sizeof(*argv) * argc);
1124                 pid = mingw_spawnv(prog, argv2, 1);
1125                 if (pid >= 0) {
1126                         int status;
1127                         if (waitpid(pid, &status, 0) < 0)
1128                                 status = 255;
1129                         exit(status);
1130                 }
1131                 pid = 1;        /* indicate that we tried but failed */
1132                 free(prog);
1133                 free(argv2);
1134         }
1135         free_path_split(path);
1136         return pid;
1137 }
1138
1139 int mingw_execv(const char *cmd, char *const *argv)
1140 {
1141         /* check if git_command is a shell script */
1142         if (!try_shell_exec(cmd, argv)) {
1143                 int pid, status;
1144
1145                 pid = mingw_spawnv(cmd, (const char **)argv, 0);
1146                 if (pid < 0)
1147                         return -1;
1148                 if (waitpid(pid, &status, 0) < 0)
1149                         status = 255;
1150                 exit(status);
1151         }
1152         return -1;
1153 }
1154
1155 int mingw_execvp(const char *cmd, char *const *argv)
1156 {
1157         char **path = get_path_split();
1158         char *prog = path_lookup(cmd, path, 0);
1159
1160         if (prog) {
1161                 mingw_execv(prog, argv);
1162                 free(prog);
1163         } else
1164                 errno = ENOENT;
1165
1166         free_path_split(path);
1167         return -1;
1168 }
1169
1170 int mingw_kill(pid_t pid, int sig)
1171 {
1172         if (pid > 0 && sig == SIGTERM) {
1173                 HANDLE h = OpenProcess(PROCESS_TERMINATE, FALSE, pid);
1174
1175                 if (TerminateProcess(h, -1)) {
1176                         CloseHandle(h);
1177                         return 0;
1178                 }
1179
1180                 errno = err_win_to_posix(GetLastError());
1181                 CloseHandle(h);
1182                 return -1;
1183         } else if (pid > 0 && sig == 0) {
1184                 HANDLE h = OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, pid);
1185                 if (h) {
1186                         CloseHandle(h);
1187                         return 0;
1188                 }
1189         }
1190
1191         errno = EINVAL;
1192         return -1;
1193 }
1194
1195 static int lookupenv(char **env, const char *name, size_t nmln)
1196 {
1197         int i;
1198
1199         for (i = 0; env[i]; i++) {
1200                 if (!strncasecmp(env[i], name, nmln) && '=' == env[i][nmln])
1201                         /* matches */
1202                         return i;
1203         }
1204         return -1;
1205 }
1206
1207 /*
1208  * If name contains '=', then sets the variable, otherwise it unsets it
1209  */
1210 static char **do_putenv(char **env, const char *name, int free_old)
1211 {
1212         char *eq = strchrnul(name, '=');
1213         int i = lookupenv(env, name, eq-name);
1214
1215         if (i < 0) {
1216                 if (*eq) {
1217                         for (i = 0; env[i]; i++)
1218                                 ;
1219                         env = xrealloc(env, (i+2)*sizeof(*env));
1220                         env[i] = (char*) name;
1221                         env[i+1] = NULL;
1222                 }
1223         }
1224         else {
1225                 if (free_old)
1226                         free(env[i]);
1227                 if (*eq)
1228                         env[i] = (char*) name;
1229                 else
1230                         for (; env[i]; i++)
1231                                 env[i] = env[i+1];
1232         }
1233         return env;
1234 }
1235
1236 #undef getenv
1237 char *mingw_getenv(const char *name)
1238 {
1239         char *result = getenv(name);
1240         if (!result && !strcmp(name, "TMPDIR")) {
1241                 /* on Windows it is TMP and TEMP */
1242                 result = getenv("TMP");
1243                 if (!result)
1244                         result = getenv("TEMP");
1245         }
1246         return result;
1247 }
1248
1249 int mingw_putenv(const char *namevalue)
1250 {
1251         environ = do_putenv(environ, namevalue, 1);
1252         return 0;
1253 }
1254
1255 /*
1256  * Note, this isn't a complete replacement for getaddrinfo. It assumes
1257  * that service contains a numerical port, or that it is null. It
1258  * does a simple search using gethostbyname, and returns one IPv4 host
1259  * if one was found.
1260  */
1261 static int WSAAPI getaddrinfo_stub(const char *node, const char *service,
1262                                    const struct addrinfo *hints,
1263                                    struct addrinfo **res)
1264 {
1265         struct hostent *h = NULL;
1266         struct addrinfo *ai;
1267         struct sockaddr_in *sin;
1268
1269         if (node) {
1270                 h = gethostbyname(node);
1271                 if (!h)
1272                         return WSAGetLastError();
1273         }
1274
1275         ai = xmalloc(sizeof(struct addrinfo));
1276         *res = ai;
1277         ai->ai_flags = 0;
1278         ai->ai_family = AF_INET;
1279         ai->ai_socktype = hints ? hints->ai_socktype : 0;
1280         switch (ai->ai_socktype) {
1281         case SOCK_STREAM:
1282                 ai->ai_protocol = IPPROTO_TCP;
1283                 break;
1284         case SOCK_DGRAM:
1285                 ai->ai_protocol = IPPROTO_UDP;
1286                 break;
1287         default:
1288                 ai->ai_protocol = 0;
1289                 break;
1290         }
1291         ai->ai_addrlen = sizeof(struct sockaddr_in);
1292         if (hints && (hints->ai_flags & AI_CANONNAME))
1293                 ai->ai_canonname = h ? xstrdup(h->h_name) : NULL;
1294         else
1295                 ai->ai_canonname = NULL;
1296
1297         sin = xmalloc(ai->ai_addrlen);
1298         memset(sin, 0, ai->ai_addrlen);
1299         sin->sin_family = AF_INET;
1300         /* Note: getaddrinfo is supposed to allow service to be a string,
1301          * which should be looked up using getservbyname. This is
1302          * currently not implemented */
1303         if (service)
1304                 sin->sin_port = htons(atoi(service));
1305         if (h)
1306                 sin->sin_addr = *(struct in_addr *)h->h_addr;
1307         else if (hints && (hints->ai_flags & AI_PASSIVE))
1308                 sin->sin_addr.s_addr = INADDR_ANY;
1309         else
1310                 sin->sin_addr.s_addr = INADDR_LOOPBACK;
1311         ai->ai_addr = (struct sockaddr *)sin;
1312         ai->ai_next = NULL;
1313         return 0;
1314 }
1315
1316 static void WSAAPI freeaddrinfo_stub(struct addrinfo *res)
1317 {
1318         free(res->ai_canonname);
1319         free(res->ai_addr);
1320         free(res);
1321 }
1322
1323 static int WSAAPI getnameinfo_stub(const struct sockaddr *sa, socklen_t salen,
1324                                    char *host, DWORD hostlen,
1325                                    char *serv, DWORD servlen, int flags)
1326 {
1327         const struct sockaddr_in *sin = (const struct sockaddr_in *)sa;
1328         if (sa->sa_family != AF_INET)
1329                 return EAI_FAMILY;
1330         if (!host && !serv)
1331                 return EAI_NONAME;
1332
1333         if (host && hostlen > 0) {
1334                 struct hostent *ent = NULL;
1335                 if (!(flags & NI_NUMERICHOST))
1336                         ent = gethostbyaddr((const char *)&sin->sin_addr,
1337                                             sizeof(sin->sin_addr), AF_INET);
1338
1339                 if (ent)
1340                         snprintf(host, hostlen, "%s", ent->h_name);
1341                 else if (flags & NI_NAMEREQD)
1342                         return EAI_NONAME;
1343                 else
1344                         snprintf(host, hostlen, "%s", inet_ntoa(sin->sin_addr));
1345         }
1346
1347         if (serv && servlen > 0) {
1348                 struct servent *ent = NULL;
1349                 if (!(flags & NI_NUMERICSERV))
1350                         ent = getservbyport(sin->sin_port,
1351                                             flags & NI_DGRAM ? "udp" : "tcp");
1352
1353                 if (ent)
1354                         snprintf(serv, servlen, "%s", ent->s_name);
1355                 else
1356                         snprintf(serv, servlen, "%d", ntohs(sin->sin_port));
1357         }
1358
1359         return 0;
1360 }
1361
1362 static HMODULE ipv6_dll = NULL;
1363 static void (WSAAPI *ipv6_freeaddrinfo)(struct addrinfo *res);
1364 static int (WSAAPI *ipv6_getaddrinfo)(const char *node, const char *service,
1365                                       const struct addrinfo *hints,
1366                                       struct addrinfo **res);
1367 static int (WSAAPI *ipv6_getnameinfo)(const struct sockaddr *sa, socklen_t salen,
1368                                       char *host, DWORD hostlen,
1369                                       char *serv, DWORD servlen, int flags);
1370 /*
1371  * gai_strerror is an inline function in the ws2tcpip.h header, so we
1372  * don't need to try to load that one dynamically.
1373  */
1374
1375 static void socket_cleanup(void)
1376 {
1377         WSACleanup();
1378         if (ipv6_dll)
1379                 FreeLibrary(ipv6_dll);
1380         ipv6_dll = NULL;
1381         ipv6_freeaddrinfo = freeaddrinfo_stub;
1382         ipv6_getaddrinfo = getaddrinfo_stub;
1383         ipv6_getnameinfo = getnameinfo_stub;
1384 }
1385
1386 static void ensure_socket_initialization(void)
1387 {
1388         WSADATA wsa;
1389         static int initialized = 0;
1390         const char *libraries[] = { "ws2_32.dll", "wship6.dll", NULL };
1391         const char **name;
1392
1393         if (initialized)
1394                 return;
1395
1396         if (WSAStartup(MAKEWORD(2,2), &wsa))
1397                 die("unable to initialize winsock subsystem, error %d",
1398                         WSAGetLastError());
1399
1400         for (name = libraries; *name; name++) {
1401                 ipv6_dll = LoadLibrary(*name);
1402                 if (!ipv6_dll)
1403                         continue;
1404
1405                 ipv6_freeaddrinfo = (void (WSAAPI *)(struct addrinfo *))
1406                         GetProcAddress(ipv6_dll, "freeaddrinfo");
1407                 ipv6_getaddrinfo = (int (WSAAPI *)(const char *, const char *,
1408                                                    const struct addrinfo *,
1409                                                    struct addrinfo **))
1410                         GetProcAddress(ipv6_dll, "getaddrinfo");
1411                 ipv6_getnameinfo = (int (WSAAPI *)(const struct sockaddr *,
1412                                                    socklen_t, char *, DWORD,
1413                                                    char *, DWORD, int))
1414                         GetProcAddress(ipv6_dll, "getnameinfo");
1415                 if (!ipv6_freeaddrinfo || !ipv6_getaddrinfo || !ipv6_getnameinfo) {
1416                         FreeLibrary(ipv6_dll);
1417                         ipv6_dll = NULL;
1418                 } else
1419                         break;
1420         }
1421         if (!ipv6_freeaddrinfo || !ipv6_getaddrinfo || !ipv6_getnameinfo) {
1422                 ipv6_freeaddrinfo = freeaddrinfo_stub;
1423                 ipv6_getaddrinfo = getaddrinfo_stub;
1424                 ipv6_getnameinfo = getnameinfo_stub;
1425         }
1426
1427         atexit(socket_cleanup);
1428         initialized = 1;
1429 }
1430
1431 #undef gethostname
1432 int mingw_gethostname(char *name, int namelen)
1433 {
1434     ensure_socket_initialization();
1435     return gethostname(name, namelen);
1436 }
1437
1438 #undef gethostbyname
1439 struct hostent *mingw_gethostbyname(const char *host)
1440 {
1441         ensure_socket_initialization();
1442         return gethostbyname(host);
1443 }
1444
1445 void mingw_freeaddrinfo(struct addrinfo *res)
1446 {
1447         ipv6_freeaddrinfo(res);
1448 }
1449
1450 int mingw_getaddrinfo(const char *node, const char *service,
1451                       const struct addrinfo *hints, struct addrinfo **res)
1452 {
1453         ensure_socket_initialization();
1454         return ipv6_getaddrinfo(node, service, hints, res);
1455 }
1456
1457 int mingw_getnameinfo(const struct sockaddr *sa, socklen_t salen,
1458                       char *host, DWORD hostlen, char *serv, DWORD servlen,
1459                       int flags)
1460 {
1461         ensure_socket_initialization();
1462         return ipv6_getnameinfo(sa, salen, host, hostlen, serv, servlen, flags);
1463 }
1464
1465 int mingw_socket(int domain, int type, int protocol)
1466 {
1467         int sockfd;
1468         SOCKET s;
1469
1470         ensure_socket_initialization();
1471         s = WSASocket(domain, type, protocol, NULL, 0, 0);
1472         if (s == INVALID_SOCKET) {
1473                 /*
1474                  * WSAGetLastError() values are regular BSD error codes
1475                  * biased by WSABASEERR.
1476                  * However, strerror() does not know about networking
1477                  * specific errors, which are values beginning at 38 or so.
1478                  * Therefore, we choose to leave the biased error code
1479                  * in errno so that _if_ someone looks up the code somewhere,
1480                  * then it is at least the number that are usually listed.
1481                  */
1482                 errno = WSAGetLastError();
1483                 return -1;
1484         }
1485         /* convert into a file descriptor */
1486         if ((sockfd = _open_osfhandle(s, O_RDWR|O_BINARY)) < 0) {
1487                 closesocket(s);
1488                 return error("unable to make a socket file descriptor: %s",
1489                         strerror(errno));
1490         }
1491         return sockfd;
1492 }
1493
1494 #undef connect
1495 int mingw_connect(int sockfd, struct sockaddr *sa, size_t sz)
1496 {
1497         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1498         return connect(s, sa, sz);
1499 }
1500
1501 #undef bind
1502 int mingw_bind(int sockfd, struct sockaddr *sa, size_t sz)
1503 {
1504         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1505         return bind(s, sa, sz);
1506 }
1507
1508 #undef setsockopt
1509 int mingw_setsockopt(int sockfd, int lvl, int optname, void *optval, int optlen)
1510 {
1511         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1512         return setsockopt(s, lvl, optname, (const char*)optval, optlen);
1513 }
1514
1515 #undef shutdown
1516 int mingw_shutdown(int sockfd, int how)
1517 {
1518         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1519         return shutdown(s, how);
1520 }
1521
1522 #undef listen
1523 int mingw_listen(int sockfd, int backlog)
1524 {
1525         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
1526         return listen(s, backlog);
1527 }
1528
1529 #undef accept
1530 int mingw_accept(int sockfd1, struct sockaddr *sa, socklen_t *sz)
1531 {
1532         int sockfd2;
1533
1534         SOCKET s1 = (SOCKET)_get_osfhandle(sockfd1);
1535         SOCKET s2 = accept(s1, sa, sz);
1536
1537         /* convert into a file descriptor */
1538         if ((sockfd2 = _open_osfhandle(s2, O_RDWR|O_BINARY)) < 0) {
1539                 int err = errno;
1540                 closesocket(s2);
1541                 return error("unable to make a socket file descriptor: %s",
1542                         strerror(err));
1543         }
1544         return sockfd2;
1545 }
1546
1547 #undef rename
1548 int mingw_rename(const char *pold, const char *pnew)
1549 {
1550         DWORD attrs, gle;
1551         int tries = 0;
1552         wchar_t wpold[MAX_PATH], wpnew[MAX_PATH];
1553         if (xutftowcs_path(wpold, pold) < 0 || xutftowcs_path(wpnew, pnew) < 0)
1554                 return -1;
1555
1556         /*
1557          * Try native rename() first to get errno right.
1558          * It is based on MoveFile(), which cannot overwrite existing files.
1559          */
1560         if (!_wrename(wpold, wpnew))
1561                 return 0;
1562         if (errno != EEXIST)
1563                 return -1;
1564 repeat:
1565         if (MoveFileExW(wpold, wpnew, MOVEFILE_REPLACE_EXISTING))
1566                 return 0;
1567         /* TODO: translate more errors */
1568         gle = GetLastError();
1569         if (gle == ERROR_ACCESS_DENIED &&
1570             (attrs = GetFileAttributesW(wpnew)) != INVALID_FILE_ATTRIBUTES) {
1571                 if (attrs & FILE_ATTRIBUTE_DIRECTORY) {
1572                         errno = EISDIR;
1573                         return -1;
1574                 }
1575                 if ((attrs & FILE_ATTRIBUTE_READONLY) &&
1576                     SetFileAttributesW(wpnew, attrs & ~FILE_ATTRIBUTE_READONLY)) {
1577                         if (MoveFileExW(wpold, wpnew, MOVEFILE_REPLACE_EXISTING))
1578                                 return 0;
1579                         gle = GetLastError();
1580                         /* revert file attributes on failure */
1581                         SetFileAttributesW(wpnew, attrs);
1582                 }
1583         }
1584         if (tries < ARRAY_SIZE(delay) && gle == ERROR_ACCESS_DENIED) {
1585                 /*
1586                  * We assume that some other process had the source or
1587                  * destination file open at the wrong moment and retry.
1588                  * In order to give the other process a higher chance to
1589                  * complete its operation, we give up our time slice now.
1590                  * If we have to retry again, we do sleep a bit.
1591                  */
1592                 Sleep(delay[tries]);
1593                 tries++;
1594                 goto repeat;
1595         }
1596         if (gle == ERROR_ACCESS_DENIED &&
1597                ask_yes_no_if_possible("Rename from '%s' to '%s' failed. "
1598                        "Should I try again?", pold, pnew))
1599                 goto repeat;
1600
1601         errno = EACCES;
1602         return -1;
1603 }
1604
1605 /*
1606  * Note that this doesn't return the actual pagesize, but
1607  * the allocation granularity. If future Windows specific git code
1608  * needs the real getpagesize function, we need to find another solution.
1609  */
1610 int mingw_getpagesize(void)
1611 {
1612         SYSTEM_INFO si;
1613         GetSystemInfo(&si);
1614         return si.dwAllocationGranularity;
1615 }
1616
1617 struct passwd *getpwuid(int uid)
1618 {
1619         static char user_name[100];
1620         static struct passwd p;
1621
1622         DWORD len = sizeof(user_name);
1623         if (!GetUserName(user_name, &len))
1624                 return NULL;
1625         p.pw_name = user_name;
1626         p.pw_gecos = "unknown";
1627         p.pw_dir = NULL;
1628         return &p;
1629 }
1630
1631 static HANDLE timer_event;
1632 static HANDLE timer_thread;
1633 static int timer_interval;
1634 static int one_shot;
1635 static sig_handler_t timer_fn = SIG_DFL, sigint_fn = SIG_DFL;
1636
1637 /* The timer works like this:
1638  * The thread, ticktack(), is a trivial routine that most of the time
1639  * only waits to receive the signal to terminate. The main thread tells
1640  * the thread to terminate by setting the timer_event to the signalled
1641  * state.
1642  * But ticktack() interrupts the wait state after the timer's interval
1643  * length to call the signal handler.
1644  */
1645
1646 static unsigned __stdcall ticktack(void *dummy)
1647 {
1648         while (WaitForSingleObject(timer_event, timer_interval) == WAIT_TIMEOUT) {
1649                 mingw_raise(SIGALRM);
1650                 if (one_shot)
1651                         break;
1652         }
1653         return 0;
1654 }
1655
1656 static int start_timer_thread(void)
1657 {
1658         timer_event = CreateEvent(NULL, FALSE, FALSE, NULL);
1659         if (timer_event) {
1660                 timer_thread = (HANDLE) _beginthreadex(NULL, 0, ticktack, NULL, 0, NULL);
1661                 if (!timer_thread )
1662                         return errno = ENOMEM,
1663                                 error("cannot start timer thread");
1664         } else
1665                 return errno = ENOMEM,
1666                         error("cannot allocate resources for timer");
1667         return 0;
1668 }
1669
1670 static void stop_timer_thread(void)
1671 {
1672         if (timer_event)
1673                 SetEvent(timer_event);  /* tell thread to terminate */
1674         if (timer_thread) {
1675                 int rc = WaitForSingleObject(timer_thread, 1000);
1676                 if (rc == WAIT_TIMEOUT)
1677                         error("timer thread did not terminate timely");
1678                 else if (rc != WAIT_OBJECT_0)
1679                         error("waiting for timer thread failed: %lu",
1680                               GetLastError());
1681                 CloseHandle(timer_thread);
1682         }
1683         if (timer_event)
1684                 CloseHandle(timer_event);
1685         timer_event = NULL;
1686         timer_thread = NULL;
1687 }
1688
1689 static inline int is_timeval_eq(const struct timeval *i1, const struct timeval *i2)
1690 {
1691         return i1->tv_sec == i2->tv_sec && i1->tv_usec == i2->tv_usec;
1692 }
1693
1694 int setitimer(int type, struct itimerval *in, struct itimerval *out)
1695 {
1696         static const struct timeval zero;
1697         static int atexit_done;
1698
1699         if (out != NULL)
1700                 return errno = EINVAL,
1701                         error("setitimer param 3 != NULL not implemented");
1702         if (!is_timeval_eq(&in->it_interval, &zero) &&
1703             !is_timeval_eq(&in->it_interval, &in->it_value))
1704                 return errno = EINVAL,
1705                         error("setitimer: it_interval must be zero or eq it_value");
1706
1707         if (timer_thread)
1708                 stop_timer_thread();
1709
1710         if (is_timeval_eq(&in->it_value, &zero) &&
1711             is_timeval_eq(&in->it_interval, &zero))
1712                 return 0;
1713
1714         timer_interval = in->it_value.tv_sec * 1000 + in->it_value.tv_usec / 1000;
1715         one_shot = is_timeval_eq(&in->it_interval, &zero);
1716         if (!atexit_done) {
1717                 atexit(stop_timer_thread);
1718                 atexit_done = 1;
1719         }
1720         return start_timer_thread();
1721 }
1722
1723 int sigaction(int sig, struct sigaction *in, struct sigaction *out)
1724 {
1725         if (sig != SIGALRM)
1726                 return errno = EINVAL,
1727                         error("sigaction only implemented for SIGALRM");
1728         if (out != NULL)
1729                 return errno = EINVAL,
1730                         error("sigaction: param 3 != NULL not implemented");
1731
1732         timer_fn = in->sa_handler;
1733         return 0;
1734 }
1735
1736 #undef signal
1737 sig_handler_t mingw_signal(int sig, sig_handler_t handler)
1738 {
1739         sig_handler_t old;
1740
1741         switch (sig) {
1742         case SIGALRM:
1743                 old = timer_fn;
1744                 timer_fn = handler;
1745                 break;
1746
1747         case SIGINT:
1748                 old = sigint_fn;
1749                 sigint_fn = handler;
1750                 break;
1751
1752         default:
1753                 return signal(sig, handler);
1754         }
1755
1756         return old;
1757 }
1758
1759 #undef raise
1760 int mingw_raise(int sig)
1761 {
1762         switch (sig) {
1763         case SIGALRM:
1764                 if (timer_fn == SIG_DFL) {
1765                         if (isatty(STDERR_FILENO))
1766                                 fputs("Alarm clock\n", stderr);
1767                         exit(128 + SIGALRM);
1768                 } else if (timer_fn != SIG_IGN)
1769                         timer_fn(SIGALRM);
1770                 return 0;
1771
1772         case SIGINT:
1773                 if (sigint_fn == SIG_DFL)
1774                         exit(128 + SIGINT);
1775                 else if (sigint_fn != SIG_IGN)
1776                         sigint_fn(SIGINT);
1777                 return 0;
1778
1779         default:
1780                 return raise(sig);
1781         }
1782 }
1783
1784
1785 static const char *make_backslash_path(const char *path)
1786 {
1787         static char buf[PATH_MAX + 1];
1788         char *c;
1789
1790         if (strlcpy(buf, path, PATH_MAX) >= PATH_MAX)
1791                 die("Too long path: %.*s", 60, path);
1792
1793         for (c = buf; *c; c++) {
1794                 if (*c == '/')
1795                         *c = '\\';
1796         }
1797         return buf;
1798 }
1799
1800 void mingw_open_html(const char *unixpath)
1801 {
1802         const char *htmlpath = make_backslash_path(unixpath);
1803         typedef HINSTANCE (WINAPI *T)(HWND, const char *,
1804                         const char *, const char *, const char *, INT);
1805         T ShellExecute;
1806         HMODULE shell32;
1807         int r;
1808
1809         shell32 = LoadLibrary("shell32.dll");
1810         if (!shell32)
1811                 die("cannot load shell32.dll");
1812         ShellExecute = (T)GetProcAddress(shell32, "ShellExecuteA");
1813         if (!ShellExecute)
1814                 die("cannot run browser");
1815
1816         printf("Launching default browser to display HTML ...\n");
1817         r = (int)ShellExecute(NULL, "open", htmlpath, NULL, "\\", SW_SHOWNORMAL);
1818         FreeLibrary(shell32);
1819         /* see the MSDN documentation referring to the result codes here */
1820         if (r <= 32) {
1821                 die("failed to launch browser for %.*s", MAX_PATH, unixpath);
1822         }
1823 }
1824
1825 int link(const char *oldpath, const char *newpath)
1826 {
1827         typedef BOOL (WINAPI *T)(LPCWSTR, LPCWSTR, LPSECURITY_ATTRIBUTES);
1828         static T create_hard_link = NULL;
1829         wchar_t woldpath[MAX_PATH], wnewpath[MAX_PATH];
1830         if (xutftowcs_path(woldpath, oldpath) < 0 ||
1831                 xutftowcs_path(wnewpath, newpath) < 0)
1832                 return -1;
1833
1834         if (!create_hard_link) {
1835                 create_hard_link = (T) GetProcAddress(
1836                         GetModuleHandle("kernel32.dll"), "CreateHardLinkW");
1837                 if (!create_hard_link)
1838                         create_hard_link = (T)-1;
1839         }
1840         if (create_hard_link == (T)-1) {
1841                 errno = ENOSYS;
1842                 return -1;
1843         }
1844         if (!create_hard_link(wnewpath, woldpath, NULL)) {
1845                 errno = err_win_to_posix(GetLastError());
1846                 return -1;
1847         }
1848         return 0;
1849 }
1850
1851 pid_t waitpid(pid_t pid, int *status, int options)
1852 {
1853         HANDLE h = OpenProcess(SYNCHRONIZE | PROCESS_QUERY_INFORMATION,
1854             FALSE, pid);
1855         if (!h) {
1856                 errno = ECHILD;
1857                 return -1;
1858         }
1859
1860         if (pid > 0 && options & WNOHANG) {
1861                 if (WAIT_OBJECT_0 != WaitForSingleObject(h, 0)) {
1862                         CloseHandle(h);
1863                         return 0;
1864                 }
1865                 options &= ~WNOHANG;
1866         }
1867
1868         if (options == 0) {
1869                 struct pinfo_t **ppinfo;
1870                 if (WaitForSingleObject(h, INFINITE) != WAIT_OBJECT_0) {
1871                         CloseHandle(h);
1872                         return 0;
1873                 }
1874
1875                 if (status)
1876                         GetExitCodeProcess(h, (LPDWORD)status);
1877
1878                 EnterCriticalSection(&pinfo_cs);
1879
1880                 ppinfo = &pinfo;
1881                 while (*ppinfo) {
1882                         struct pinfo_t *info = *ppinfo;
1883                         if (info->pid == pid) {
1884                                 CloseHandle(info->proc);
1885                                 *ppinfo = info->next;
1886                                 free(info);
1887                                 break;
1888                         }
1889                         ppinfo = &info->next;
1890                 }
1891
1892                 LeaveCriticalSection(&pinfo_cs);
1893
1894                 CloseHandle(h);
1895                 return pid;
1896         }
1897         CloseHandle(h);
1898
1899         errno = EINVAL;
1900         return -1;
1901 }
1902
1903 int mingw_offset_1st_component(const char *path)
1904 {
1905         int offset = 0;
1906         if (has_dos_drive_prefix(path))
1907                 offset = 2;
1908
1909         /* unc paths */
1910         else if (is_dir_sep(path[0]) && is_dir_sep(path[1])) {
1911
1912                 /* skip server name */
1913                 char *pos = strpbrk(path + 2, "\\/");
1914                 if (!pos)
1915                         return 0; /* Error: malformed unc path */
1916
1917                 do {
1918                         pos++;
1919                 } while (*pos && !is_dir_sep(*pos));
1920
1921                 offset = pos - path;
1922         }
1923
1924         return offset + is_dir_sep(path[offset]);
1925 }
1926
1927 int xutftowcsn(wchar_t *wcs, const char *utfs, size_t wcslen, int utflen)
1928 {
1929         int upos = 0, wpos = 0;
1930         const unsigned char *utf = (const unsigned char*) utfs;
1931         if (!utf || !wcs || wcslen < 1) {
1932                 errno = EINVAL;
1933                 return -1;
1934         }
1935         /* reserve space for \0 */
1936         wcslen--;
1937         if (utflen < 0)
1938                 utflen = INT_MAX;
1939
1940         while (upos < utflen) {
1941                 int c = utf[upos++] & 0xff;
1942                 if (utflen == INT_MAX && c == 0)
1943                         break;
1944
1945                 if (wpos >= wcslen) {
1946                         wcs[wpos] = 0;
1947                         errno = ERANGE;
1948                         return -1;
1949                 }
1950
1951                 if (c < 0x80) {
1952                         /* ASCII */
1953                         wcs[wpos++] = c;
1954                 } else if (c >= 0xc2 && c < 0xe0 && upos < utflen &&
1955                                 (utf[upos] & 0xc0) == 0x80) {
1956                         /* 2-byte utf-8 */
1957                         c = ((c & 0x1f) << 6);
1958                         c |= (utf[upos++] & 0x3f);
1959                         wcs[wpos++] = c;
1960                 } else if (c >= 0xe0 && c < 0xf0 && upos + 1 < utflen &&
1961                                 !(c == 0xe0 && utf[upos] < 0xa0) && /* over-long encoding */
1962                                 (utf[upos] & 0xc0) == 0x80 &&
1963                                 (utf[upos + 1] & 0xc0) == 0x80) {
1964                         /* 3-byte utf-8 */
1965                         c = ((c & 0x0f) << 12);
1966                         c |= ((utf[upos++] & 0x3f) << 6);
1967                         c |= (utf[upos++] & 0x3f);
1968                         wcs[wpos++] = c;
1969                 } else if (c >= 0xf0 && c < 0xf5 && upos + 2 < utflen &&
1970                                 wpos + 1 < wcslen &&
1971                                 !(c == 0xf0 && utf[upos] < 0x90) && /* over-long encoding */
1972                                 !(c == 0xf4 && utf[upos] >= 0x90) && /* > \u10ffff */
1973                                 (utf[upos] & 0xc0) == 0x80 &&
1974                                 (utf[upos + 1] & 0xc0) == 0x80 &&
1975                                 (utf[upos + 2] & 0xc0) == 0x80) {
1976                         /* 4-byte utf-8: convert to \ud8xx \udcxx surrogate pair */
1977                         c = ((c & 0x07) << 18);
1978                         c |= ((utf[upos++] & 0x3f) << 12);
1979                         c |= ((utf[upos++] & 0x3f) << 6);
1980                         c |= (utf[upos++] & 0x3f);
1981                         c -= 0x10000;
1982                         wcs[wpos++] = 0xd800 | (c >> 10);
1983                         wcs[wpos++] = 0xdc00 | (c & 0x3ff);
1984                 } else if (c >= 0xa0) {
1985                         /* invalid utf-8 byte, printable unicode char: convert 1:1 */
1986                         wcs[wpos++] = c;
1987                 } else {
1988                         /* invalid utf-8 byte, non-printable unicode: convert to hex */
1989                         static const char *hex = "0123456789abcdef";
1990                         wcs[wpos++] = hex[c >> 4];
1991                         if (wpos < wcslen)
1992                                 wcs[wpos++] = hex[c & 0x0f];
1993                 }
1994         }
1995         wcs[wpos] = 0;
1996         return wpos;
1997 }
1998
1999 int xwcstoutf(char *utf, const wchar_t *wcs, size_t utflen)
2000 {
2001         if (!wcs || !utf || utflen < 1) {
2002                 errno = EINVAL;
2003                 return -1;
2004         }
2005         utflen = WideCharToMultiByte(CP_UTF8, 0, wcs, -1, utf, utflen, NULL, NULL);
2006         if (utflen)
2007                 return utflen - 1;
2008         errno = ERANGE;
2009         return -1;
2010 }
2011
2012 /*
2013  * Disable MSVCRT command line wildcard expansion (__getmainargs called from
2014  * mingw startup code, see init.c in mingw runtime).
2015  */
2016 int _CRT_glob = 0;
2017
2018 typedef struct {
2019         int newmode;
2020 } _startupinfo;
2021
2022 extern int __wgetmainargs(int *argc, wchar_t ***argv, wchar_t ***env, int glob,
2023                 _startupinfo *si);
2024
2025 static NORETURN void die_startup()
2026 {
2027         fputs("fatal: not enough memory for initialization", stderr);
2028         exit(128);
2029 }
2030
2031 void mingw_startup()
2032 {
2033         int i, len, maxlen, argc;
2034         char *buffer;
2035         wchar_t **wenv, **wargv;
2036         _startupinfo si;
2037
2038         /* get wide char arguments and environment */
2039         si.newmode = 0;
2040         if (__wgetmainargs(&argc, &wargv, &wenv, _CRT_glob, &si) < 0)
2041                 die_startup();
2042
2043         /* determine size of argv and environ conversion buffer */
2044         maxlen = wcslen(_wpgmptr);
2045         for (i = 1; i < argc; i++)
2046                 maxlen = max(maxlen, wcslen(wargv[i]));
2047         for (i = 0; wenv[i]; i++)
2048                 maxlen = max(maxlen, wcslen(wenv[i]));
2049
2050         /* nedmalloc can't free CRT memory, allocate resizable environment list */
2051         environ = xcalloc(i + 1, sizeof(char*));
2052
2053         /* allocate buffer (wchar_t encodes to max 3 UTF-8 bytes) */
2054         maxlen = 3 * maxlen + 1;
2055         buffer = xmalloc(maxlen);
2056
2057         /* convert command line arguments and environment to UTF-8 */
2058         len = xwcstoutf(buffer, _wpgmptr, maxlen);
2059         __argv[0] = xmemdupz(buffer, len);
2060         for (i = 1; i < argc; i++) {
2061                 len = xwcstoutf(buffer, wargv[i], maxlen);
2062                 __argv[i] = xmemdupz(buffer, len);
2063         }
2064         for (i = 0; wenv[i]; i++) {
2065                 len = xwcstoutf(buffer, wenv[i], maxlen);
2066                 environ[i] = xmemdupz(buffer, len);
2067         }
2068         free(buffer);
2069
2070         /* initialize critical section for waitpid pinfo_t list */
2071         InitializeCriticalSection(&pinfo_cs);
2072
2073         /* set up default file mode and file modes for stdin/out/err */
2074         _fmode = _O_BINARY;
2075         _setmode(_fileno(stdin), _O_BINARY);
2076         _setmode(_fileno(stdout), _O_BINARY);
2077         _setmode(_fileno(stderr), _O_BINARY);
2078
2079         /* initialize Unicode console */
2080         winansi_init();
2081 }