msvcrt: fread: Fill buffer on small reads.
[wine] / dlls / msvcrt / process.c
1 /*
2  * msvcrt.dll spawn/exec functions
3  *
4  * Copyright 1996,1998 Marcus Meissner
5  * Copyright 1996 Jukka Iivonen
6  * Copyright 1997,2000 Uwe Bonnes
7  * Copyright 2000 Jon Griffiths
8  *
9  * This library is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * This library is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with this library; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
22  *
23  * FIXME:
24  * -File handles need some special handling. Sometimes children get
25  *  open file handles, sometimes not. The docs are confusing
26  * -No check for maximum path/argument/environment size is done
27  */
28 #include "config.h"
29
30 #include <stdarg.h>
31
32 #include "msvcrt.h"
33 #include "wine/debug.h"
34
35 WINE_DEFAULT_DEBUG_CHANNEL(msvcrt);
36
37 /* INTERNAL: Spawn a child process */
38 static MSVCRT_intptr_t msvcrt_spawn(int flags, const char* exe, char* cmdline, char* env)
39 {
40   STARTUPINFOA si;
41   PROCESS_INFORMATION pi;
42
43   if ((unsigned)flags > MSVCRT__P_DETACH)
44   {
45     *MSVCRT__errno() = MSVCRT_EINVAL;
46     return -1;
47   }
48
49   memset(&si, 0, sizeof(si));
50   si.cb = sizeof(si);
51   msvcrt_create_io_inherit_block(&si);
52   if (!CreateProcessA(exe, cmdline, NULL, NULL, TRUE,
53                      flags == MSVCRT__P_DETACH ? DETACHED_PROCESS : 0,
54                      env, NULL, &si, &pi))
55   {
56     msvcrt_set_errno(GetLastError());
57     MSVCRT_free(si.lpReserved2);
58     return -1;
59   }
60
61   MSVCRT_free(si.lpReserved2);
62   switch(flags)
63   {
64   case MSVCRT__P_WAIT:
65     WaitForSingleObject(pi.hProcess, INFINITE);
66     GetExitCodeProcess(pi.hProcess,&pi.dwProcessId);
67     CloseHandle(pi.hProcess);
68     CloseHandle(pi.hThread);
69     return pi.dwProcessId;
70   case MSVCRT__P_DETACH:
71     CloseHandle(pi.hProcess);
72     pi.hProcess = 0;
73     /* fall through */
74   case MSVCRT__P_NOWAIT:
75   case MSVCRT__P_NOWAITO:
76     CloseHandle(pi.hThread);
77     return (MSVCRT_intptr_t)pi.hProcess;
78   case  MSVCRT__P_OVERLAY:
79     MSVCRT__exit(0);
80   }
81   return -1; /* can't reach here */
82 }
83
84 /* INTERNAL: Convert argv list to a single 'delim'-separated string, with an
85  * extra '\0' to terminate it
86  */
87 static char* msvcrt_argvtos(const char* const* arg, char delim)
88 {
89   const char* const* a;
90   long size;
91   char* p;
92   char* ret;
93
94   if (!arg && !delim)
95   {
96       /* Return NULL for an empty environment list */
97       return NULL;
98   }
99
100   /* get length */
101   a = arg;
102   size = 0;
103   while (*a)
104   {
105     size += strlen(*a) + 1;
106     a++;
107   }
108
109   ret = (char*)MSVCRT_malloc(size + 1);
110   if (!ret)
111     return NULL;
112
113   /* fill string */
114   a = arg;
115   p = ret;
116   while (*a)
117   {
118     int len = strlen(*a);
119     memcpy(p,*a,len);
120     p += len;
121     *p++ = delim;
122     a++;
123   }
124   if (delim && p > ret) p[-1] = 0;
125   else *p = 0;
126   return ret;
127 }
128
129 /* INTERNAL: Convert va_list to a single 'delim'-separated string, with an
130  * extra '\0' to terminate it
131  */
132 static char* msvcrt_valisttos(const char* arg0, va_list alist, char delim)
133 {
134   va_list alist2;
135   long size;
136   const char *arg;
137   char* p;
138   char *ret;
139
140 #ifdef HAVE_VA_COPY
141   va_copy(alist2,alist);
142 #else
143 # ifdef HAVE___VA_COPY
144   __va_copy(alist2,alist);
145 # else
146   alist2 = alist;
147 # endif
148 #endif
149
150   if (!arg0 && !delim)
151   {
152       /* Return NULL for an empty environment list */
153       return NULL;
154   }
155
156   /* get length */
157   arg = arg0;
158   size = 0;
159   do {
160       size += strlen(arg) + 1;
161       arg = va_arg(alist, char*);
162   } while (arg != NULL);
163
164   ret = (char*)MSVCRT_malloc(size + 1);
165   if (!ret)
166     return NULL;
167
168   /* fill string */
169   arg = arg0;
170   p = ret;
171   do {
172       int len = strlen(arg);
173       memcpy(p,arg,len);
174       p += len;
175       *p++ = delim;
176       arg = va_arg(alist2, char*);
177   } while (arg != NULL);
178   if (delim && p > ret) p[-1] = 0;
179   else *p = 0;
180   return ret;
181 }
182
183 /*********************************************************************
184  *              _cwait (MSVCRT.@)
185  */
186 MSVCRT_intptr_t CDECL _cwait(int *status, MSVCRT_intptr_t pid, int action)
187 {
188   HANDLE hPid = (HANDLE)pid;
189   int doserrno;
190
191   action = action; /* Remove warning */
192
193   if (!WaitForSingleObject(hPid, INFINITE))
194   {
195     if (status)
196     {
197       DWORD stat;
198       GetExitCodeProcess(hPid, &stat);
199       *status = (int)stat;
200     }
201     return pid;
202   }
203   doserrno = GetLastError();
204
205   if (doserrno == ERROR_INVALID_HANDLE)
206   {
207     *MSVCRT__errno() =  MSVCRT_ECHILD;
208     *MSVCRT___doserrno() = doserrno;
209   }
210   else
211     msvcrt_set_errno(doserrno);
212
213   return status ? *status = -1 : -1;
214 }
215
216 /*********************************************************************
217  *              _execl (MSVCRT.@)
218  *
219  * Like on Windows, this function does not handle arguments with spaces
220  * or double-quotes.
221  */
222 MSVCRT_intptr_t CDECL _execl(const char* name, const char* arg0, ...)
223 {
224   va_list ap;
225   char * args;
226   MSVCRT_intptr_t ret;
227
228   va_start(ap, arg0);
229   args = msvcrt_valisttos(arg0, ap, ' ');
230   va_end(ap);
231
232   ret = msvcrt_spawn(MSVCRT__P_OVERLAY, name, args, NULL);
233   MSVCRT_free(args);
234
235   return ret;
236 }
237
238 /*********************************************************************
239  *              _execle (MSVCRT.@)
240  */
241 MSVCRT_intptr_t CDECL _execle(const char* name, const char* arg0, ...)
242 {
243     FIXME("stub\n");
244     return -1;
245 }
246
247 /*********************************************************************
248  *              _execlp (MSVCRT.@)
249  *
250  * Like on Windows, this function does not handle arguments with spaces
251  * or double-quotes.
252  */
253 MSVCRT_intptr_t CDECL _execlp(const char* name, const char* arg0, ...)
254 {
255   va_list ap;
256   char * args;
257   MSVCRT_intptr_t ret;
258   char fullname[MAX_PATH];
259
260   _searchenv(name, "PATH", fullname);
261
262   va_start(ap, arg0);
263   args = msvcrt_valisttos(arg0, ap, ' ');
264   va_end(ap);
265
266   ret = msvcrt_spawn(MSVCRT__P_OVERLAY, fullname[0] ? fullname : name, args, NULL);
267   MSVCRT_free(args);
268
269   return ret;
270 }
271
272 /*********************************************************************
273  *              _execlpe (MSVCRT.@)
274  */
275 MSVCRT_intptr_t CDECL _execlpe(const char* name, const char* arg0, ...)
276 {
277     FIXME("stub\n");
278     return -1;
279 }
280
281 /*********************************************************************
282  *              _execv (MSVCRT.@)
283  *
284  * Like on Windows, this function does not handle arguments with spaces
285  * or double-quotes.
286  */
287 MSVCRT_intptr_t CDECL _execv(const char* name, char* const* argv)
288 {
289   return _spawnve(MSVCRT__P_OVERLAY, name, (const char* const*) argv, NULL);
290 }
291
292 /*********************************************************************
293  *              _execve (MSVCRT.@)
294  *
295  * Like on Windows, this function does not handle arguments with spaces
296  * or double-quotes.
297  */
298 MSVCRT_intptr_t CDECL MSVCRT__execve(const char* name, char* const* argv, const char* const* envv)
299 {
300   return _spawnve(MSVCRT__P_OVERLAY, name, (const char* const*) argv, envv);
301 }
302
303 /*********************************************************************
304  *              _execvpe (MSVCRT.@)
305  *
306  * Like on Windows, this function does not handle arguments with spaces
307  * or double-quotes.
308  */
309 MSVCRT_intptr_t CDECL _execvpe(const char* name, char* const* argv, const char* const* envv)
310 {
311   char fullname[MAX_PATH];
312
313   _searchenv(name, "PATH", fullname);
314   return _spawnve(MSVCRT__P_OVERLAY, fullname[0] ? fullname : name,
315                   (const char* const*) argv, envv);
316 }
317
318 /*********************************************************************
319  *              _execvp (MSVCRT.@)
320  *
321  * Like on Windows, this function does not handle arguments with spaces
322  * or double-quotes.
323  */
324 MSVCRT_intptr_t CDECL _execvp(const char* name, char* const* argv)
325 {
326   return _execvpe(name, argv, NULL);
327 }
328
329 /*********************************************************************
330  *              _spawnl (MSVCRT.@)
331  *
332  * Like on Windows, this function does not handle arguments with spaces
333  * or double-quotes.
334  */
335 MSVCRT_intptr_t CDECL _spawnl(int flags, const char* name, const char* arg0, ...)
336 {
337   va_list ap;
338   char * args;
339   MSVCRT_intptr_t ret;
340
341   va_start(ap, arg0);
342   args = msvcrt_valisttos(arg0, ap, ' ');
343   va_end(ap);
344
345   ret = msvcrt_spawn(flags, name, args, NULL);
346   MSVCRT_free(args);
347
348   return ret;
349 }
350
351 /*********************************************************************
352  *              _spawnle (MSVCRT.@)
353  */
354 MSVCRT_intptr_t CDECL _spawnle(int flags, const char* name, const char* arg0, ...)
355 {
356     va_list ap;
357     char *args, *envs = NULL;
358     const char * const *envp;
359     MSVCRT_intptr_t ret;
360
361     va_start(ap, arg0);
362     args = msvcrt_valisttos(arg0, ap, ' ');
363     va_end(ap);
364
365     va_start(ap, arg0);
366     while (va_arg( ap, char * ) != NULL) /*nothing*/;
367     envp = va_arg( ap, const char * const * );
368     if (envp) envs = msvcrt_argvtos(envp, 0);
369     va_end(ap);
370
371     ret = msvcrt_spawn(flags, name, args, envs);
372
373     MSVCRT_free(args);
374     MSVCRT_free(envs);
375     return ret;
376 }
377
378
379 /*********************************************************************
380  *              _spawnlp (MSVCRT.@)
381  *
382  * Like on Windows, this function does not handle arguments with spaces
383  * or double-quotes.
384  */
385 MSVCRT_intptr_t CDECL _spawnlp(int flags, const char* name, const char* arg0, ...)
386 {
387   va_list ap;
388   char * args;
389   MSVCRT_intptr_t ret;
390   char fullname[MAX_PATH];
391
392   _searchenv(name, "PATH", fullname);
393
394   va_start(ap, arg0);
395   args = msvcrt_valisttos(arg0, ap, ' ');
396   va_end(ap);
397
398   ret = msvcrt_spawn(flags, fullname[0] ? fullname : name, args, NULL);
399   MSVCRT_free(args);
400
401   return ret;
402 }
403
404 /*********************************************************************
405  *              _spawnlpe (MSVCRT.@)
406  */
407 MSVCRT_intptr_t CDECL _spawnlpe(int flags, const char* name, const char* arg0, ...)
408 {
409     va_list ap;
410     char *args, *envs = NULL;
411     const char * const *envp;
412     MSVCRT_intptr_t ret;
413     char fullname[MAX_PATH];
414
415     _searchenv(name, "PATH", fullname);
416
417     va_start(ap, arg0);
418     args = msvcrt_valisttos(arg0, ap, ' ');
419     va_end(ap);
420
421     va_start(ap, arg0);
422     while (va_arg( ap, char * ) != NULL) /*nothing*/;
423     envp = va_arg( ap, const char * const * );
424     if (envp) envs = msvcrt_argvtos(envp, 0);
425     va_end(ap);
426
427     ret = msvcrt_spawn(flags, fullname[0] ? fullname : name, args, envs);
428
429     MSVCRT_free(args);
430     MSVCRT_free(envs);
431     return ret;
432 }
433
434 /*********************************************************************
435  *              _spawnve (MSVCRT.@)
436  *
437  * Like on Windows, this function does not handle arguments with spaces
438  * or double-quotes.
439  */
440 MSVCRT_intptr_t CDECL _spawnve(int flags, const char* name, const char* const* argv,
441                                const char* const* envv)
442 {
443   char * args = msvcrt_argvtos(argv,' ');
444   char * envs = msvcrt_argvtos(envv,0);
445   char fullname[MAX_PATH];
446   const char *p;
447   int len;
448   MSVCRT_intptr_t ret = -1;
449
450   TRACE(":call (%s), params (%s), env (%s)\n",debugstr_a(name),debugstr_a(args),
451    envs?"Custom":"Null");
452
453   /* no check for NULL name.
454      native doesn't do it */
455
456   p = memchr(name, '\0', MAX_PATH);
457   if( !p )
458     p = name + MAX_PATH - 1;
459   len = p - name;
460
461   /* extra-long names are silently truncated. */
462   memcpy(fullname, name, len);
463
464   for( p--; p >= name; p-- )
465   {
466     if( *p == '\\' || *p == '/' || *p == ':' || *p == '.' )
467       break;
468   }
469
470   /* if no extension is given, assume .exe */
471   if( (p < name || *p != '.') && len <= MAX_PATH - 5 )
472   {
473     FIXME("only trying .exe when no extension given\n");
474     memcpy(fullname+len, ".exe", 4);
475     len += 4;
476   }
477
478   fullname[len] = '\0';
479
480   if (args)
481   {
482     ret = msvcrt_spawn(flags, fullname, args, envs);
483     MSVCRT_free(args);
484   }
485   MSVCRT_free(envs);
486
487   return ret;
488 }
489
490 /*********************************************************************
491  *              _spawnv (MSVCRT.@)
492  *
493  * Like on Windows, this function does not handle arguments with spaces
494  * or double-quotes.
495  */
496 MSVCRT_intptr_t CDECL _spawnv(int flags, const char* name, const char* const* argv)
497 {
498   return _spawnve(flags, name, argv, NULL);
499 }
500
501 /*********************************************************************
502  *              _spawnvpe (MSVCRT.@)
503  *
504  * Like on Windows, this function does not handle arguments with spaces
505  * or double-quotes.
506  */
507 MSVCRT_intptr_t CDECL _spawnvpe(int flags, const char* name, const char* const* argv,
508                                 const char* const* envv)
509 {
510   char fullname[MAX_PATH];
511   _searchenv(name, "PATH", fullname);
512   return _spawnve(flags, fullname[0] ? fullname : name, argv, envv);
513 }
514
515 /*********************************************************************
516  *              _spawnvp (MSVCRT.@)
517  *
518  * Like on Windows, this function does not handle arguments with spaces
519  * or double-quotes.
520  */
521 MSVCRT_intptr_t CDECL _spawnvp(int flags, const char* name, const char* const* argv)
522 {
523   return _spawnvpe(flags, name, argv, NULL);
524 }
525
526 /*********************************************************************
527  *              _popen (MSVCRT.@)
528  * FIXME: convert to _wpopen and call that from here instead?  But it
529  * would have to convert the command back to ANSI to call msvcrt_spawn,
530  * less than ideal.
531  */
532 MSVCRT_FILE* CDECL MSVCRT__popen(const char* command, const char* mode)
533 {
534   static const char wcmd[] = "cmd", cmdFlag[] = " /C ", comSpec[] = "COMSPEC";
535   MSVCRT_FILE *ret;
536   BOOL readPipe = TRUE;
537   int textmode, fds[2], fdToDup, fdToOpen, fdStdHandle = -1, fdStdErr = -1;
538   const char *p;
539   char *cmdcopy;
540   DWORD comSpecLen;
541
542   TRACE("(command=%s, mode=%s)\n", debugstr_a(command), debugstr_a(mode));
543
544   if (!command || !mode)
545     return NULL;
546
547   textmode = *__p__fmode() & (MSVCRT__O_BINARY | MSVCRT__O_TEXT);
548   for (p = mode; *p; p++)
549   {
550     switch (*p)
551     {
552       case 'W':
553       case 'w':
554         readPipe = FALSE;
555         break;
556       case 'B':
557       case 'b':
558         textmode |= MSVCRT__O_BINARY;
559         textmode &= ~MSVCRT__O_TEXT;
560         break;
561       case 'T':
562       case 't':
563         textmode |= MSVCRT__O_TEXT;
564         textmode &= ~MSVCRT__O_BINARY;
565         break;
566     }
567   }
568   if (_pipe(fds, 0, textmode) == -1)
569     return NULL;
570
571   fdToDup = readPipe ? 1 : 0;
572   fdToOpen = readPipe ? 0 : 1;
573
574   if ((fdStdHandle = _dup(fdToDup)) == -1)
575     goto error;
576   if (_dup2(fds[fdToDup], fdToDup) != 0)
577     goto error;
578   if (readPipe)
579   {
580     if ((fdStdErr = _dup(MSVCRT_STDERR_FILENO)) == -1)
581       goto error;
582     if (_dup2(fds[fdToDup], MSVCRT_STDERR_FILENO) != 0)
583       goto error;
584   }
585
586   _close(fds[fdToDup]);
587
588   comSpecLen = GetEnvironmentVariableA(comSpec, NULL, 0);
589   if (!comSpecLen)
590     comSpecLen = strlen(wcmd) + 1;
591   cmdcopy = HeapAlloc(GetProcessHeap(), 0, comSpecLen + strlen(cmdFlag)
592    + strlen(command));
593   if (!GetEnvironmentVariableA(comSpec, cmdcopy, comSpecLen))
594     strcpy(cmdcopy, wcmd);
595   strcat(cmdcopy, cmdFlag);
596   strcat(cmdcopy, command);
597   if (msvcrt_spawn(MSVCRT__P_NOWAIT, NULL, cmdcopy, NULL) == -1)
598   {
599     _close(fds[fdToOpen]);
600     ret = NULL;
601   }
602   else
603   {
604     ret = MSVCRT__fdopen(fds[fdToOpen], mode);
605     if (!ret)
606       _close(fds[fdToOpen]);
607   }
608   HeapFree(GetProcessHeap(), 0, cmdcopy);
609   _dup2(fdStdHandle, fdToDup);
610   _close(fdStdHandle);
611   if (readPipe)
612   {
613     _dup2(fdStdErr, MSVCRT_STDERR_FILENO);
614     _close(fdStdErr);
615   }
616   return ret;
617
618 error:
619   if (fdStdHandle != -1) _close(fdStdHandle);
620   if (fdStdErr != -1)    _close(fdStdErr);
621   _close(fds[0]);
622   _close(fds[1]);
623   return NULL;
624 }
625
626 /*********************************************************************
627  *              _wpopen (MSVCRT.@)
628  */
629 MSVCRT_FILE* CDECL MSVCRT__wpopen(const MSVCRT_wchar_t* command, const MSVCRT_wchar_t* mode)
630 {
631   FIXME("(command=%s, mode=%s): stub\n", debugstr_w(command), debugstr_w(mode));
632   return NULL;
633 }
634
635 /*********************************************************************
636  *              _pclose (MSVCRT.@)
637  */
638 int CDECL MSVCRT__pclose(MSVCRT_FILE* file)
639 {
640   return MSVCRT_fclose(file);
641 }
642
643 /*********************************************************************
644  *              system (MSVCRT.@)
645  */
646 int CDECL MSVCRT_system(const char* cmd)
647 {
648     char* cmdcopy;
649     int res;
650
651     /* Make a writable copy for CreateProcess */
652     cmdcopy=_strdup(cmd);
653     /* FIXME: should probably launch cmd interpreter in COMSPEC */
654     res=msvcrt_spawn(MSVCRT__P_WAIT, NULL, cmdcopy, NULL);
655     MSVCRT_free(cmdcopy);
656     return res;
657 }
658
659 /*********************************************************************
660  *              _loaddll (MSVCRT.@)
661  */
662 MSVCRT_intptr_t CDECL _loaddll(const char* dllname)
663 {
664   return (MSVCRT_intptr_t)LoadLibraryA(dllname);
665 }
666
667 /*********************************************************************
668  *              _unloaddll (MSVCRT.@)
669  */
670 int CDECL _unloaddll(MSVCRT_intptr_t dll)
671 {
672   if (FreeLibrary((HMODULE)dll))
673     return 0;
674   else
675   {
676     int err = GetLastError();
677     msvcrt_set_errno(err);
678     return err;
679   }
680 }
681
682 /*********************************************************************
683  *              _getdllprocaddr (MSVCRT.@)
684  */
685 void * CDECL _getdllprocaddr(MSVCRT_intptr_t dll, const char *name, int ordinal)
686 {
687     if (name)
688     {
689         if (ordinal != -1) return NULL;
690         return GetProcAddress( (HMODULE)dll, name );
691     }
692     if (HIWORD(ordinal)) return NULL;
693     return GetProcAddress( (HMODULE)dll, (LPCSTR)(ULONG_PTR)ordinal );
694 }