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