Merge branch 'dr/ceiling'
[git] / compat / mingw.c
1 #include "../git-compat-util.h"
2 #include "../strbuf.h"
3
4 unsigned int _CRT_fmode = _O_BINARY;
5
6 #undef open
7 int mingw_open (const char *filename, int oflags, ...)
8 {
9         va_list args;
10         unsigned mode;
11         va_start(args, oflags);
12         mode = va_arg(args, int);
13         va_end(args);
14
15         if (!strcmp(filename, "/dev/null"))
16                 filename = "nul";
17         int fd = open(filename, oflags, mode);
18         if (fd < 0 && (oflags & O_CREAT) && errno == EACCES) {
19                 DWORD attrs = GetFileAttributes(filename);
20                 if (attrs != INVALID_FILE_ATTRIBUTES && (attrs & FILE_ATTRIBUTE_DIRECTORY))
21                         errno = EISDIR;
22         }
23         return fd;
24 }
25
26 static inline time_t filetime_to_time_t(const FILETIME *ft)
27 {
28         long long winTime = ((long long)ft->dwHighDateTime << 32) + ft->dwLowDateTime;
29         winTime -= 116444736000000000LL; /* Windows to Unix Epoch conversion */
30         winTime /= 10000000;             /* Nano to seconds resolution */
31         return (time_t)winTime;
32 }
33
34 static inline size_t size_to_blocks(size_t s)
35 {
36         return (s+511)/512;
37 }
38
39 extern int _getdrive( void );
40 /* We keep the do_lstat code in a separate function to avoid recursion.
41  * When a path ends with a slash, the stat will fail with ENOENT. In
42  * this case, we strip the trailing slashes and stat again.
43  */
44 static int do_lstat(const char *file_name, struct stat *buf)
45 {
46         WIN32_FILE_ATTRIBUTE_DATA fdata;
47
48         if (GetFileAttributesExA(file_name, GetFileExInfoStandard, &fdata)) {
49                 int fMode = S_IREAD;
50                 if (fdata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
51                         fMode |= S_IFDIR;
52                 else
53                         fMode |= S_IFREG;
54                 if (!(fdata.dwFileAttributes & FILE_ATTRIBUTE_READONLY))
55                         fMode |= S_IWRITE;
56
57                 buf->st_ino = 0;
58                 buf->st_gid = 0;
59                 buf->st_uid = 0;
60                 buf->st_mode = fMode;
61                 buf->st_size = fdata.nFileSizeLow; /* Can't use nFileSizeHigh, since it's not a stat64 */
62                 buf->st_blocks = size_to_blocks(buf->st_size);
63                 buf->st_dev = _getdrive() - 1;
64                 buf->st_atime = filetime_to_time_t(&(fdata.ftLastAccessTime));
65                 buf->st_mtime = filetime_to_time_t(&(fdata.ftLastWriteTime));
66                 buf->st_ctime = filetime_to_time_t(&(fdata.ftCreationTime));
67                 errno = 0;
68                 return 0;
69         }
70
71         switch (GetLastError()) {
72         case ERROR_ACCESS_DENIED:
73         case ERROR_SHARING_VIOLATION:
74         case ERROR_LOCK_VIOLATION:
75         case ERROR_SHARING_BUFFER_EXCEEDED:
76                 errno = EACCES;
77                 break;
78         case ERROR_BUFFER_OVERFLOW:
79                 errno = ENAMETOOLONG;
80                 break;
81         case ERROR_NOT_ENOUGH_MEMORY:
82                 errno = ENOMEM;
83                 break;
84         default:
85                 errno = ENOENT;
86                 break;
87         }
88         return -1;
89 }
90
91 /* We provide our own lstat/fstat functions, since the provided
92  * lstat/fstat functions are so slow. These stat functions are
93  * tailored for Git's usage (read: fast), and are not meant to be
94  * complete. Note that Git stat()s are redirected to mingw_lstat()
95  * too, since Windows doesn't really handle symlinks that well.
96  */
97 int mingw_lstat(const char *file_name, struct mingw_stat *buf)
98 {
99         int namelen;
100         static char alt_name[PATH_MAX];
101
102         if (!do_lstat(file_name, buf))
103                 return 0;
104
105         /* if file_name ended in a '/', Windows returned ENOENT;
106          * try again without trailing slashes
107          */
108         if (errno != ENOENT)
109                 return -1;
110
111         namelen = strlen(file_name);
112         if (namelen && file_name[namelen-1] != '/')
113                 return -1;
114         while (namelen && file_name[namelen-1] == '/')
115                 --namelen;
116         if (!namelen || namelen >= PATH_MAX)
117                 return -1;
118
119         memcpy(alt_name, file_name, namelen);
120         alt_name[namelen] = 0;
121         return do_lstat(alt_name, buf);
122 }
123
124 #undef fstat
125 #undef stat
126 int mingw_fstat(int fd, struct mingw_stat *buf)
127 {
128         HANDLE fh = (HANDLE)_get_osfhandle(fd);
129         BY_HANDLE_FILE_INFORMATION fdata;
130
131         if (fh == INVALID_HANDLE_VALUE) {
132                 errno = EBADF;
133                 return -1;
134         }
135         /* direct non-file handles to MS's fstat() */
136         if (GetFileType(fh) != FILE_TYPE_DISK) {
137                 struct stat st;
138                 if (fstat(fd, &st))
139                         return -1;
140                 buf->st_ino = st.st_ino;
141                 buf->st_gid = st.st_gid;
142                 buf->st_uid = st.st_uid;
143                 buf->st_mode = st.st_mode;
144                 buf->st_size = st.st_size;
145                 buf->st_blocks = size_to_blocks(buf->st_size);
146                 buf->st_dev = st.st_dev;
147                 buf->st_atime = st.st_atime;
148                 buf->st_mtime = st.st_mtime;
149                 buf->st_ctime = st.st_ctime;
150                 return 0;
151         }
152
153         if (GetFileInformationByHandle(fh, &fdata)) {
154                 int fMode = S_IREAD;
155                 if (fdata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
156                         fMode |= S_IFDIR;
157                 else
158                         fMode |= S_IFREG;
159                 if (!(fdata.dwFileAttributes & FILE_ATTRIBUTE_READONLY))
160                         fMode |= S_IWRITE;
161
162                 buf->st_ino = 0;
163                 buf->st_gid = 0;
164                 buf->st_uid = 0;
165                 buf->st_mode = fMode;
166                 buf->st_size = fdata.nFileSizeLow; /* Can't use nFileSizeHigh, since it's not a stat64 */
167                 buf->st_blocks = size_to_blocks(buf->st_size);
168                 buf->st_dev = _getdrive() - 1;
169                 buf->st_atime = filetime_to_time_t(&(fdata.ftLastAccessTime));
170                 buf->st_mtime = filetime_to_time_t(&(fdata.ftLastWriteTime));
171                 buf->st_ctime = filetime_to_time_t(&(fdata.ftCreationTime));
172                 return 0;
173         }
174         errno = EBADF;
175         return -1;
176 }
177
178 static inline void time_t_to_filetime(time_t t, FILETIME *ft)
179 {
180         long long winTime = t * 10000000LL + 116444736000000000LL;
181         ft->dwLowDateTime = winTime;
182         ft->dwHighDateTime = winTime >> 32;
183 }
184
185 int mingw_utime (const char *file_name, const struct utimbuf *times)
186 {
187         FILETIME mft, aft;
188         int fh, rc;
189
190         /* must have write permission */
191         if ((fh = open(file_name, O_RDWR | O_BINARY)) < 0)
192                 return -1;
193
194         time_t_to_filetime(times->modtime, &mft);
195         time_t_to_filetime(times->actime, &aft);
196         if (!SetFileTime((HANDLE)_get_osfhandle(fh), NULL, &aft, &mft)) {
197                 errno = EINVAL;
198                 rc = -1;
199         } else
200                 rc = 0;
201         close(fh);
202         return rc;
203 }
204
205 unsigned int sleep (unsigned int seconds)
206 {
207         Sleep(seconds*1000);
208         return 0;
209 }
210
211 int mkstemp(char *template)
212 {
213         char *filename = mktemp(template);
214         if (filename == NULL)
215                 return -1;
216         return open(filename, O_RDWR | O_CREAT, 0600);
217 }
218
219 int gettimeofday(struct timeval *tv, void *tz)
220 {
221         SYSTEMTIME st;
222         struct tm tm;
223         GetSystemTime(&st);
224         tm.tm_year = st.wYear-1900;
225         tm.tm_mon = st.wMonth-1;
226         tm.tm_mday = st.wDay;
227         tm.tm_hour = st.wHour;
228         tm.tm_min = st.wMinute;
229         tm.tm_sec = st.wSecond;
230         tv->tv_sec = tm_to_time_t(&tm);
231         if (tv->tv_sec < 0)
232                 return -1;
233         tv->tv_usec = st.wMilliseconds*1000;
234         return 0;
235 }
236
237 int pipe(int filedes[2])
238 {
239         int fd;
240         HANDLE h[2], parent;
241
242         if (_pipe(filedes, 8192, 0) < 0)
243                 return -1;
244
245         parent = GetCurrentProcess();
246
247         if (!DuplicateHandle (parent, (HANDLE)_get_osfhandle(filedes[0]),
248                         parent, &h[0], 0, FALSE, DUPLICATE_SAME_ACCESS)) {
249                 close(filedes[0]);
250                 close(filedes[1]);
251                 return -1;
252         }
253         if (!DuplicateHandle (parent, (HANDLE)_get_osfhandle(filedes[1]),
254                         parent, &h[1], 0, FALSE, DUPLICATE_SAME_ACCESS)) {
255                 close(filedes[0]);
256                 close(filedes[1]);
257                 CloseHandle(h[0]);
258                 return -1;
259         }
260         fd = _open_osfhandle((int)h[0], O_NOINHERIT);
261         if (fd < 0) {
262                 close(filedes[0]);
263                 close(filedes[1]);
264                 CloseHandle(h[0]);
265                 CloseHandle(h[1]);
266                 return -1;
267         }
268         close(filedes[0]);
269         filedes[0] = fd;
270         fd = _open_osfhandle((int)h[1], O_NOINHERIT);
271         if (fd < 0) {
272                 close(filedes[0]);
273                 close(filedes[1]);
274                 CloseHandle(h[1]);
275                 return -1;
276         }
277         close(filedes[1]);
278         filedes[1] = fd;
279         return 0;
280 }
281
282 int poll(struct pollfd *ufds, unsigned int nfds, int timeout)
283 {
284         int i, pending;
285
286         if (timeout != -1)
287                 return errno = EINVAL, error("poll timeout not supported");
288
289         /* When there is only one fd to wait for, then we pretend that
290          * input is available and let the actual wait happen when the
291          * caller invokes read().
292          */
293         if (nfds == 1) {
294                 if (!(ufds[0].events & POLLIN))
295                         return errno = EINVAL, error("POLLIN not set");
296                 ufds[0].revents = POLLIN;
297                 return 0;
298         }
299
300 repeat:
301         pending = 0;
302         for (i = 0; i < nfds; i++) {
303                 DWORD avail = 0;
304                 HANDLE h = (HANDLE) _get_osfhandle(ufds[i].fd);
305                 if (h == INVALID_HANDLE_VALUE)
306                         return -1;      /* errno was set */
307
308                 if (!(ufds[i].events & POLLIN))
309                         return errno = EINVAL, error("POLLIN not set");
310
311                 /* this emulation works only for pipes */
312                 if (!PeekNamedPipe(h, NULL, 0, NULL, &avail, NULL)) {
313                         int err = GetLastError();
314                         if (err == ERROR_BROKEN_PIPE) {
315                                 ufds[i].revents = POLLHUP;
316                                 pending++;
317                         } else {
318                                 errno = EINVAL;
319                                 return error("PeekNamedPipe failed,"
320                                         " GetLastError: %u", err);
321                         }
322                 } else if (avail) {
323                         ufds[i].revents = POLLIN;
324                         pending++;
325                 } else
326                         ufds[i].revents = 0;
327         }
328         if (!pending) {
329                 /* The only times that we spin here is when the process
330                  * that is connected through the pipes is waiting for
331                  * its own input data to become available. But since
332                  * the process (pack-objects) is itself CPU intensive,
333                  * it will happily pick up the time slice that we are
334                  * relinguishing here.
335                  */
336                 Sleep(0);
337                 goto repeat;
338         }
339         return 0;
340 }
341
342 struct tm *gmtime_r(const time_t *timep, struct tm *result)
343 {
344         /* gmtime() in MSVCRT.DLL is thread-safe, but not reentrant */
345         memcpy(result, gmtime(timep), sizeof(struct tm));
346         return result;
347 }
348
349 struct tm *localtime_r(const time_t *timep, struct tm *result)
350 {
351         /* localtime() in MSVCRT.DLL is thread-safe, but not reentrant */
352         memcpy(result, localtime(timep), sizeof(struct tm));
353         return result;
354 }
355
356 #undef getcwd
357 char *mingw_getcwd(char *pointer, int len)
358 {
359         int i;
360         char *ret = getcwd(pointer, len);
361         if (!ret)
362                 return ret;
363         for (i = 0; pointer[i]; i++)
364                 if (pointer[i] == '\\')
365                         pointer[i] = '/';
366         return ret;
367 }
368
369 #undef getenv
370 char *mingw_getenv(const char *name)
371 {
372         char *result = getenv(name);
373         if (!result && !strcmp(name, "TMPDIR")) {
374                 /* on Windows it is TMP and TEMP */
375                 result = getenv("TMP");
376                 if (!result)
377                         result = getenv("TEMP");
378         }
379         return result;
380 }
381
382 /*
383  * See http://msdn2.microsoft.com/en-us/library/17w5ykft(vs.71).aspx
384  * (Parsing C++ Command-Line Arguments)
385  */
386 static const char *quote_arg(const char *arg)
387 {
388         /* count chars to quote */
389         int len = 0, n = 0;
390         int force_quotes = 0;
391         char *q, *d;
392         const char *p = arg;
393         if (!*p) force_quotes = 1;
394         while (*p) {
395                 if (isspace(*p) || *p == '*' || *p == '?' || *p == '{')
396                         force_quotes = 1;
397                 else if (*p == '"')
398                         n++;
399                 else if (*p == '\\') {
400                         int count = 0;
401                         while (*p == '\\') {
402                                 count++;
403                                 p++;
404                                 len++;
405                         }
406                         if (*p == '"')
407                                 n += count*2 + 1;
408                         continue;
409                 }
410                 len++;
411                 p++;
412         }
413         if (!force_quotes && n == 0)
414                 return arg;
415
416         /* insert \ where necessary */
417         d = q = xmalloc(len+n+3);
418         *d++ = '"';
419         while (*arg) {
420                 if (*arg == '"')
421                         *d++ = '\\';
422                 else if (*arg == '\\') {
423                         int count = 0;
424                         while (*arg == '\\') {
425                                 count++;
426                                 *d++ = *arg++;
427                         }
428                         if (*arg == '"') {
429                                 while (count-- > 0)
430                                         *d++ = '\\';
431                                 *d++ = '\\';
432                         }
433                 }
434                 *d++ = *arg++;
435         }
436         *d++ = '"';
437         *d++ = 0;
438         return q;
439 }
440
441 static const char *parse_interpreter(const char *cmd)
442 {
443         static char buf[100];
444         char *p, *opt;
445         int n, fd;
446
447         /* don't even try a .exe */
448         n = strlen(cmd);
449         if (n >= 4 && !strcasecmp(cmd+n-4, ".exe"))
450                 return NULL;
451
452         fd = open(cmd, O_RDONLY);
453         if (fd < 0)
454                 return NULL;
455         n = read(fd, buf, sizeof(buf)-1);
456         close(fd);
457         if (n < 4)      /* at least '#!/x' and not error */
458                 return NULL;
459
460         if (buf[0] != '#' || buf[1] != '!')
461                 return NULL;
462         buf[n] = '\0';
463         p = strchr(buf, '\n');
464         if (!p)
465                 return NULL;
466
467         *p = '\0';
468         if (!(p = strrchr(buf+2, '/')) && !(p = strrchr(buf+2, '\\')))
469                 return NULL;
470         /* strip options */
471         if ((opt = strchr(p+1, ' ')))
472                 *opt = '\0';
473         return p+1;
474 }
475
476 /*
477  * Splits the PATH into parts.
478  */
479 static char **get_path_split(void)
480 {
481         char *p, **path, *envpath = getenv("PATH");
482         int i, n = 0;
483
484         if (!envpath || !*envpath)
485                 return NULL;
486
487         envpath = xstrdup(envpath);
488         p = envpath;
489         while (p) {
490                 char *dir = p;
491                 p = strchr(p, ';');
492                 if (p) *p++ = '\0';
493                 if (*dir) {     /* not earlier, catches series of ; */
494                         ++n;
495                 }
496         }
497         if (!n)
498                 return NULL;
499
500         path = xmalloc((n+1)*sizeof(char*));
501         p = envpath;
502         i = 0;
503         do {
504                 if (*p)
505                         path[i++] = xstrdup(p);
506                 p = p+strlen(p)+1;
507         } while (i < n);
508         path[i] = NULL;
509
510         free(envpath);
511
512         return path;
513 }
514
515 static void free_path_split(char **path)
516 {
517         if (!path)
518                 return;
519
520         char **p = path;
521         while (*p)
522                 free(*p++);
523         free(path);
524 }
525
526 /*
527  * exe_only means that we only want to detect .exe files, but not scripts
528  * (which do not have an extension)
529  */
530 static char *lookup_prog(const char *dir, const char *cmd, int isexe, int exe_only)
531 {
532         char path[MAX_PATH];
533         snprintf(path, sizeof(path), "%s/%s.exe", dir, cmd);
534
535         if (!isexe && access(path, F_OK) == 0)
536                 return xstrdup(path);
537         path[strlen(path)-4] = '\0';
538         if ((!exe_only || isexe) && access(path, F_OK) == 0)
539                 return xstrdup(path);
540         return NULL;
541 }
542
543 /*
544  * Determines the absolute path of cmd using the the split path in path.
545  * If cmd contains a slash or backslash, no lookup is performed.
546  */
547 static char *path_lookup(const char *cmd, char **path, int exe_only)
548 {
549         char *prog = NULL;
550         int len = strlen(cmd);
551         int isexe = len >= 4 && !strcasecmp(cmd+len-4, ".exe");
552
553         if (strchr(cmd, '/') || strchr(cmd, '\\'))
554                 prog = xstrdup(cmd);
555
556         while (!prog && *path)
557                 prog = lookup_prog(*path++, cmd, isexe, exe_only);
558
559         return prog;
560 }
561
562 static int env_compare(const void *a, const void *b)
563 {
564         char *const *ea = a;
565         char *const *eb = b;
566         return strcasecmp(*ea, *eb);
567 }
568
569 static pid_t mingw_spawnve(const char *cmd, const char **argv, char **env,
570                            int prepend_cmd)
571 {
572         STARTUPINFO si;
573         PROCESS_INFORMATION pi;
574         struct strbuf envblk, args;
575         unsigned flags;
576         BOOL ret;
577
578         /* Determine whether or not we are associated to a console */
579         HANDLE cons = CreateFile("CONOUT$", GENERIC_WRITE,
580                         FILE_SHARE_WRITE, NULL, OPEN_EXISTING,
581                         FILE_ATTRIBUTE_NORMAL, NULL);
582         if (cons == INVALID_HANDLE_VALUE) {
583                 /* There is no console associated with this process.
584                  * Since the child is a console process, Windows
585                  * would normally create a console window. But
586                  * since we'll be redirecting std streams, we do
587                  * not need the console.
588                  */
589                 flags = CREATE_NO_WINDOW;
590         } else {
591                 /* There is already a console. If we specified
592                  * CREATE_NO_WINDOW here, too, Windows would
593                  * disassociate the child from the console.
594                  * Go figure!
595                  */
596                 flags = 0;
597                 CloseHandle(cons);
598         }
599         memset(&si, 0, sizeof(si));
600         si.cb = sizeof(si);
601         si.dwFlags = STARTF_USESTDHANDLES;
602         si.hStdInput = (HANDLE) _get_osfhandle(0);
603         si.hStdOutput = (HANDLE) _get_osfhandle(1);
604         si.hStdError = (HANDLE) _get_osfhandle(2);
605
606         /* concatenate argv, quoting args as we go */
607         strbuf_init(&args, 0);
608         if (prepend_cmd) {
609                 char *quoted = (char *)quote_arg(cmd);
610                 strbuf_addstr(&args, quoted);
611                 if (quoted != cmd)
612                         free(quoted);
613         }
614         for (; *argv; argv++) {
615                 char *quoted = (char *)quote_arg(*argv);
616                 if (*args.buf)
617                         strbuf_addch(&args, ' ');
618                 strbuf_addstr(&args, quoted);
619                 if (quoted != *argv)
620                         free(quoted);
621         }
622
623         if (env) {
624                 int count = 0;
625                 char **e, **sorted_env;
626
627                 for (e = env; *e; e++)
628                         count++;
629
630                 /* environment must be sorted */
631                 sorted_env = xmalloc(sizeof(*sorted_env) * (count + 1));
632                 memcpy(sorted_env, env, sizeof(*sorted_env) * (count + 1));
633                 qsort(sorted_env, count, sizeof(*sorted_env), env_compare);
634
635                 strbuf_init(&envblk, 0);
636                 for (e = sorted_env; *e; e++) {
637                         strbuf_addstr(&envblk, *e);
638                         strbuf_addch(&envblk, '\0');
639                 }
640                 free(sorted_env);
641         }
642
643         memset(&pi, 0, sizeof(pi));
644         ret = CreateProcess(cmd, args.buf, NULL, NULL, TRUE, flags,
645                 env ? envblk.buf : NULL, NULL, &si, &pi);
646
647         if (env)
648                 strbuf_release(&envblk);
649         strbuf_release(&args);
650
651         if (!ret) {
652                 errno = ENOENT;
653                 return -1;
654         }
655         CloseHandle(pi.hThread);
656         return (pid_t)pi.hProcess;
657 }
658
659 pid_t mingw_spawnvpe(const char *cmd, const char **argv, char **env)
660 {
661         pid_t pid;
662         char **path = get_path_split();
663         char *prog = path_lookup(cmd, path, 0);
664
665         if (!prog) {
666                 errno = ENOENT;
667                 pid = -1;
668         }
669         else {
670                 const char *interpr = parse_interpreter(prog);
671
672                 if (interpr) {
673                         const char *argv0 = argv[0];
674                         char *iprog = path_lookup(interpr, path, 1);
675                         argv[0] = prog;
676                         if (!iprog) {
677                                 errno = ENOENT;
678                                 pid = -1;
679                         }
680                         else {
681                                 pid = mingw_spawnve(iprog, argv, env, 1);
682                                 free(iprog);
683                         }
684                         argv[0] = argv0;
685                 }
686                 else
687                         pid = mingw_spawnve(prog, argv, env, 0);
688                 free(prog);
689         }
690         free_path_split(path);
691         return pid;
692 }
693
694 static int try_shell_exec(const char *cmd, char *const *argv, char **env)
695 {
696         const char *interpr = parse_interpreter(cmd);
697         char **path;
698         char *prog;
699         int pid = 0;
700
701         if (!interpr)
702                 return 0;
703         path = get_path_split();
704         prog = path_lookup(interpr, path, 1);
705         if (prog) {
706                 int argc = 0;
707                 const char **argv2;
708                 while (argv[argc]) argc++;
709                 argv2 = xmalloc(sizeof(*argv) * (argc+1));
710                 argv2[0] = (char *)cmd; /* full path to the script file */
711                 memcpy(&argv2[1], &argv[1], sizeof(*argv) * argc);
712                 pid = mingw_spawnve(prog, argv2, env, 1);
713                 if (pid >= 0) {
714                         int status;
715                         if (waitpid(pid, &status, 0) < 0)
716                                 status = 255;
717                         exit(status);
718                 }
719                 pid = 1;        /* indicate that we tried but failed */
720                 free(prog);
721                 free(argv2);
722         }
723         free_path_split(path);
724         return pid;
725 }
726
727 static void mingw_execve(const char *cmd, char *const *argv, char *const *env)
728 {
729         /* check if git_command is a shell script */
730         if (!try_shell_exec(cmd, argv, (char **)env)) {
731                 int pid, status;
732
733                 pid = mingw_spawnve(cmd, (const char **)argv, (char **)env, 0);
734                 if (pid < 0)
735                         return;
736                 if (waitpid(pid, &status, 0) < 0)
737                         status = 255;
738                 exit(status);
739         }
740 }
741
742 void mingw_execvp(const char *cmd, char *const *argv)
743 {
744         char **path = get_path_split();
745         char *prog = path_lookup(cmd, path, 0);
746
747         if (prog) {
748                 mingw_execve(prog, argv, environ);
749                 free(prog);
750         } else
751                 errno = ENOENT;
752
753         free_path_split(path);
754 }
755
756 char **copy_environ()
757 {
758         char **env;
759         int i = 0;
760         while (environ[i])
761                 i++;
762         env = xmalloc((i+1)*sizeof(*env));
763         for (i = 0; environ[i]; i++)
764                 env[i] = xstrdup(environ[i]);
765         env[i] = NULL;
766         return env;
767 }
768
769 void free_environ(char **env)
770 {
771         int i;
772         for (i = 0; env[i]; i++)
773                 free(env[i]);
774         free(env);
775 }
776
777 static int lookup_env(char **env, const char *name, size_t nmln)
778 {
779         int i;
780
781         for (i = 0; env[i]; i++) {
782                 if (0 == strncmp(env[i], name, nmln)
783                     && '=' == env[i][nmln])
784                         /* matches */
785                         return i;
786         }
787         return -1;
788 }
789
790 /*
791  * If name contains '=', then sets the variable, otherwise it unsets it
792  */
793 char **env_setenv(char **env, const char *name)
794 {
795         char *eq = strchrnul(name, '=');
796         int i = lookup_env(env, name, eq-name);
797
798         if (i < 0) {
799                 if (*eq) {
800                         for (i = 0; env[i]; i++)
801                                 ;
802                         env = xrealloc(env, (i+2)*sizeof(*env));
803                         env[i] = xstrdup(name);
804                         env[i+1] = NULL;
805                 }
806         }
807         else {
808                 free(env[i]);
809                 if (*eq)
810                         env[i] = xstrdup(name);
811                 else
812                         for (; env[i]; i++)
813                                 env[i] = env[i+1];
814         }
815         return env;
816 }
817
818 /* this is the first function to call into WS_32; initialize it */
819 #undef gethostbyname
820 struct hostent *mingw_gethostbyname(const char *host)
821 {
822         WSADATA wsa;
823
824         if (WSAStartup(MAKEWORD(2,2), &wsa))
825                 die("unable to initialize winsock subsystem, error %d",
826                         WSAGetLastError());
827         atexit((void(*)(void)) WSACleanup);
828         return gethostbyname(host);
829 }
830
831 int mingw_socket(int domain, int type, int protocol)
832 {
833         int sockfd;
834         SOCKET s = WSASocket(domain, type, protocol, NULL, 0, 0);
835         if (s == INVALID_SOCKET) {
836                 /*
837                  * WSAGetLastError() values are regular BSD error codes
838                  * biased by WSABASEERR.
839                  * However, strerror() does not know about networking
840                  * specific errors, which are values beginning at 38 or so.
841                  * Therefore, we choose to leave the biased error code
842                  * in errno so that _if_ someone looks up the code somewhere,
843                  * then it is at least the number that are usually listed.
844                  */
845                 errno = WSAGetLastError();
846                 return -1;
847         }
848         /* convert into a file descriptor */
849         if ((sockfd = _open_osfhandle(s, O_RDWR|O_BINARY)) < 0) {
850                 closesocket(s);
851                 return error("unable to make a socket file descriptor: %s",
852                         strerror(errno));
853         }
854         return sockfd;
855 }
856
857 #undef connect
858 int mingw_connect(int sockfd, struct sockaddr *sa, size_t sz)
859 {
860         SOCKET s = (SOCKET)_get_osfhandle(sockfd);
861         return connect(s, sa, sz);
862 }
863
864 #undef rename
865 int mingw_rename(const char *pold, const char *pnew)
866 {
867         /*
868          * Try native rename() first to get errno right.
869          * It is based on MoveFile(), which cannot overwrite existing files.
870          */
871         if (!rename(pold, pnew))
872                 return 0;
873         if (errno != EEXIST)
874                 return -1;
875         if (MoveFileEx(pold, pnew, MOVEFILE_REPLACE_EXISTING))
876                 return 0;
877         /* TODO: translate more errors */
878         if (GetLastError() == ERROR_ACCESS_DENIED) {
879                 DWORD attrs = GetFileAttributes(pnew);
880                 if (attrs != INVALID_FILE_ATTRIBUTES && (attrs & FILE_ATTRIBUTE_DIRECTORY)) {
881                         errno = EISDIR;
882                         return -1;
883                 }
884         }
885         errno = EACCES;
886         return -1;
887 }
888
889 struct passwd *getpwuid(int uid)
890 {
891         static char user_name[100];
892         static struct passwd p;
893
894         DWORD len = sizeof(user_name);
895         if (!GetUserName(user_name, &len))
896                 return NULL;
897         p.pw_name = user_name;
898         p.pw_gecos = "unknown";
899         p.pw_dir = NULL;
900         return &p;
901 }
902
903 static HANDLE timer_event;
904 static HANDLE timer_thread;
905 static int timer_interval;
906 static int one_shot;
907 static sig_handler_t timer_fn = SIG_DFL;
908
909 /* The timer works like this:
910  * The thread, ticktack(), is a trivial routine that most of the time
911  * only waits to receive the signal to terminate. The main thread tells
912  * the thread to terminate by setting the timer_event to the signalled
913  * state.
914  * But ticktack() interrupts the wait state after the timer's interval
915  * length to call the signal handler.
916  */
917
918 static __stdcall unsigned ticktack(void *dummy)
919 {
920         while (WaitForSingleObject(timer_event, timer_interval) == WAIT_TIMEOUT) {
921                 if (timer_fn == SIG_DFL)
922                         die("Alarm");
923                 if (timer_fn != SIG_IGN)
924                         timer_fn(SIGALRM);
925                 if (one_shot)
926                         break;
927         }
928         return 0;
929 }
930
931 static int start_timer_thread(void)
932 {
933         timer_event = CreateEvent(NULL, FALSE, FALSE, NULL);
934         if (timer_event) {
935                 timer_thread = (HANDLE) _beginthreadex(NULL, 0, ticktack, NULL, 0, NULL);
936                 if (!timer_thread )
937                         return errno = ENOMEM,
938                                 error("cannot start timer thread");
939         } else
940                 return errno = ENOMEM,
941                         error("cannot allocate resources for timer");
942         return 0;
943 }
944
945 static void stop_timer_thread(void)
946 {
947         if (timer_event)
948                 SetEvent(timer_event);  /* tell thread to terminate */
949         if (timer_thread) {
950                 int rc = WaitForSingleObject(timer_thread, 1000);
951                 if (rc == WAIT_TIMEOUT)
952                         error("timer thread did not terminate timely");
953                 else if (rc != WAIT_OBJECT_0)
954                         error("waiting for timer thread failed: %lu",
955                               GetLastError());
956                 CloseHandle(timer_thread);
957         }
958         if (timer_event)
959                 CloseHandle(timer_event);
960         timer_event = NULL;
961         timer_thread = NULL;
962 }
963
964 static inline int is_timeval_eq(const struct timeval *i1, const struct timeval *i2)
965 {
966         return i1->tv_sec == i2->tv_sec && i1->tv_usec == i2->tv_usec;
967 }
968
969 int setitimer(int type, struct itimerval *in, struct itimerval *out)
970 {
971         static const struct timeval zero;
972         static int atexit_done;
973
974         if (out != NULL)
975                 return errno = EINVAL,
976                         error("setitimer param 3 != NULL not implemented");
977         if (!is_timeval_eq(&in->it_interval, &zero) &&
978             !is_timeval_eq(&in->it_interval, &in->it_value))
979                 return errno = EINVAL,
980                         error("setitimer: it_interval must be zero or eq it_value");
981
982         if (timer_thread)
983                 stop_timer_thread();
984
985         if (is_timeval_eq(&in->it_value, &zero) &&
986             is_timeval_eq(&in->it_interval, &zero))
987                 return 0;
988
989         timer_interval = in->it_value.tv_sec * 1000 + in->it_value.tv_usec / 1000;
990         one_shot = is_timeval_eq(&in->it_interval, &zero);
991         if (!atexit_done) {
992                 atexit(stop_timer_thread);
993                 atexit_done = 1;
994         }
995         return start_timer_thread();
996 }
997
998 int sigaction(int sig, struct sigaction *in, struct sigaction *out)
999 {
1000         if (sig != SIGALRM)
1001                 return errno = EINVAL,
1002                         error("sigaction only implemented for SIGALRM");
1003         if (out != NULL)
1004                 return errno = EINVAL,
1005                         error("sigaction: param 3 != NULL not implemented");
1006
1007         timer_fn = in->sa_handler;
1008         return 0;
1009 }
1010
1011 #undef signal
1012 sig_handler_t mingw_signal(int sig, sig_handler_t handler)
1013 {
1014         if (sig != SIGALRM)
1015                 return signal(sig, handler);
1016         sig_handler_t old = timer_fn;
1017         timer_fn = handler;
1018         return old;
1019 }