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