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