Make RtlCreateUserThread fail if process handle is not for the current
[wine] / dlls / winedos / module.c
1 /*
2  * DOS (MZ) loader
3  *
4  * Copyright 1998 Ove Kåven
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  *
20  * Note: This code hasn't been completely cleaned up yet.
21  */
22
23 #include "config.h"
24 #include "wine/port.h"
25
26 #include <stdarg.h>
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <fcntl.h>
31 #include <signal.h>
32 #ifdef HAVE_UNISTD_H
33 # include <unistd.h>
34 #endif
35 #include <sys/types.h>
36 #include <sys/stat.h>
37 #ifdef HAVE_SYS_TIME_H
38 # include <sys/time.h>
39 #endif
40 #include "windef.h"
41 #include "winbase.h"
42 #include "wine/winbase16.h"
43 #include "wingdi.h"
44 #include "winuser.h"
45 #include "winerror.h"
46 #include "wine/debug.h"
47 #include "dosexe.h"
48 #include "dosvm.h"
49 #include "vga.h"
50
51 WINE_DEFAULT_DEBUG_CHANNEL(module);
52
53 static BOOL DOSVM_isdosexe;
54
55 /**********************************************************************
56  *          DOSVM_IsWin16
57  * 
58  * Return TRUE if we are in Windows process.
59  */
60 BOOL DOSVM_IsWin16(void)
61 {
62   return DOSVM_isdosexe ? FALSE : TRUE;
63 }
64
65 #ifdef MZ_SUPPORTED
66
67 #ifdef HAVE_SYS_MMAN_H
68 # include <sys/mman.h>
69 #endif
70
71 /* define this to try mapping through /proc/pid/mem instead of a temp file,
72    but Linus doesn't like mmapping /proc/pid/mem, so it doesn't work for me */
73 #undef MZ_MAPSELF
74
75 #define BIOS_DATA_SEGMENT 0x40
76 #define PSP_SIZE 0x10
77
78 #define SEG16(ptr,seg) ((LPVOID)((BYTE*)ptr+((DWORD)(seg)<<4)))
79 #define SEGPTR16(ptr,segptr) ((LPVOID)((BYTE*)ptr+((DWORD)SELECTOROF(segptr)<<4)+OFFSETOF(segptr)))
80
81 /* structures for EXEC */
82
83 #include "pshpack1.h"
84
85 typedef struct {
86   WORD env_seg;
87   DWORD cmdline;
88   DWORD fcb1;
89   DWORD fcb2;
90   WORD init_sp;
91   WORD init_ss;
92   WORD init_ip;
93   WORD init_cs;
94 } ExecBlock;
95
96 typedef struct {
97   WORD load_seg;
98   WORD rel_seg;
99 } OverlayBlock;
100
101 #include "poppack.h"
102
103 /* global variables */
104
105 pid_t dosvm_pid;
106
107 static WORD init_cs,init_ip,init_ss,init_sp;
108 static HANDLE dosvm_thread, loop_thread;
109 static DWORD dosvm_tid, loop_tid;
110
111 static void MZ_Launch( LPCSTR cmdtail, int length );
112 static BOOL MZ_InitTask(void);
113
114 static void MZ_CreatePSP( LPVOID lpPSP, WORD env, WORD par )
115 {
116   PDB16*psp=lpPSP;
117
118   psp->int20=0x20CD; /* int 20 */
119   /* some programs use this to calculate how much memory they need */
120   psp->nextParagraph=0x9FFF; /* FIXME: use a real value */
121   /* FIXME: dispatcher */
122   psp->savedint22 = DOSVM_GetRMHandler(0x22);
123   psp->savedint23 = DOSVM_GetRMHandler(0x23);
124   psp->savedint24 = DOSVM_GetRMHandler(0x24);
125   psp->parentPSP=par;
126   psp->environment=env;
127   /* FIXME: more PSP stuff */
128 }
129
130 static void MZ_FillPSP( LPVOID lpPSP, LPCSTR cmdtail, int length )
131 {
132     PDB16 *psp = lpPSP;
133
134     if(length > 127) 
135     {
136         WARN( "Command tail truncated! (length %d)\n", length );
137         length = 126;
138     }
139
140     psp->cmdLine[0] = length;
141
142     /*
143      * Length of exactly 127 bytes means that full command line is 
144      * stored in environment variable CMDLINE and PSP contains 
145      * command tail truncated to 126 bytes.
146      */
147     if(length == 127)
148         length = 126;
149
150     if(length > 0)
151         memmove(psp->cmdLine+1, cmdtail, length);
152
153     psp->cmdLine[length+1] = '\r';
154
155     /* FIXME: more PSP stuff */
156 }
157
158 static WORD MZ_InitEnvironment( LPCSTR env, LPCSTR name )
159 {
160  unsigned sz=0;
161  unsigned i=0;
162  WORD seg;
163  LPSTR envblk;
164
165  if (env) {
166   /* get size of environment block */
167   while (env[sz++]) sz+=strlen(env+sz)+1;
168  } else sz++;
169  /* allocate it */
170  envblk=DOSMEM_GetBlock(sz+sizeof(WORD)+strlen(name)+1,&seg);
171  /* fill it */
172  if (env) {
173   memcpy(envblk,env,sz);
174  } else envblk[0]=0;
175  /* DOS environment variables are uppercase */
176  while (envblk[i]){
177   while (envblk[i] != '='){
178    if (envblk[i]>='a' && envblk[i] <= 'z'){
179     envblk[i] -= 32;
180    }
181    i++;
182   }
183   i += strlen(envblk+i) + 1;
184  }
185  /* DOS 3.x: the block contains 1 additional string */
186  *(WORD*)(envblk+sz)=1;
187  /* being the program name itself */
188  strcpy(envblk+sz+sizeof(WORD),name);
189  return seg;
190 }
191
192 static BOOL MZ_InitMemory(void)
193 {
194     /* initialize the memory */
195     TRACE("Initializing DOS memory structures\n");
196     DOSMEM_Init(TRUE);
197     DOSDEV_InstallDOSDevices();
198
199     return TRUE;
200 }
201
202 static BOOL MZ_DoLoadImage( HANDLE hFile, LPCSTR filename, OverlayBlock *oblk )
203 {
204   IMAGE_DOS_HEADER mz_header;
205   DWORD image_start,image_size,min_size,max_size,avail;
206   BYTE*psp_start,*load_start,*oldenv;
207   int x, old_com=0, alloc;
208   SEGPTR reloc;
209   WORD env_seg, load_seg, rel_seg, oldpsp_seg;
210   DWORD len;
211
212   if (DOSVM_psp) {
213     /* DOS process already running, inherit from it */
214     PDB16* par_psp = (PDB16*)((DWORD)DOSVM_psp << 4);
215     alloc=0;
216     oldenv = (LPBYTE)((DWORD)par_psp->environment << 4);
217     oldpsp_seg = DOSVM_psp;
218   } else {
219     /* allocate new DOS process, inheriting from Wine environment */
220     alloc=1;
221     oldenv = GetEnvironmentStringsA();
222     oldpsp_seg = 0;
223   }
224
225  SetFilePointer(hFile,0,NULL,FILE_BEGIN);
226  if (   !ReadFile(hFile,&mz_header,sizeof(mz_header),&len,NULL)
227      || len != sizeof(mz_header)
228      || mz_header.e_magic != IMAGE_DOS_SIGNATURE) {
229   char *p = strrchr( filename, '.' );
230   if (!p || strcasecmp( p, ".com" ))  /* check for .COM extension */
231   {
232       SetLastError(ERROR_BAD_FORMAT);
233       goto load_error;
234   }
235   old_com=1; /* assume .COM file */
236   image_start=0;
237   image_size=GetFileSize(hFile,NULL);
238   min_size=0x10000; max_size=0x100000;
239   mz_header.e_crlc=0;
240   mz_header.e_ss=0; mz_header.e_sp=0xFFFE;
241   mz_header.e_cs=0; mz_header.e_ip=0x100;
242  } else {
243   /* calculate load size */
244   image_start=mz_header.e_cparhdr<<4;
245   image_size=mz_header.e_cp<<9; /* pages are 512 bytes */
246   /* From Ralf Brown Interrupt List: If the word at offset 02h is 4, it should
247    * be treated as 00h, since pre-1.10 versions of the MS linker set it that
248    * way. */
249   if ((mz_header.e_cblp!=0)&&(mz_header.e_cblp!=4)) image_size-=512-mz_header.e_cblp;
250   image_size-=image_start;
251   min_size=image_size+((DWORD)mz_header.e_minalloc<<4)+(PSP_SIZE<<4);
252   max_size=image_size+((DWORD)mz_header.e_maxalloc<<4)+(PSP_SIZE<<4);
253  }
254
255   if (alloc) MZ_InitMemory();
256
257   if (oblk) {
258     /* load overlay into preallocated memory */
259     load_seg=oblk->load_seg;
260     rel_seg=oblk->rel_seg;
261     load_start=(LPBYTE)((DWORD)load_seg<<4);
262   } else {
263     /* allocate environment block */
264     env_seg=MZ_InitEnvironment(oldenv, filename);
265
266     /* allocate memory for the executable */
267     TRACE("Allocating DOS memory (min=%ld, max=%ld)\n",min_size,max_size);
268     avail=DOSMEM_Available();
269     if (avail<min_size) {
270       ERR("insufficient DOS memory\n");
271       SetLastError(ERROR_NOT_ENOUGH_MEMORY);
272       goto load_error;
273     }
274     if (avail>max_size) avail=max_size;
275     psp_start=DOSMEM_GetBlock(avail,&DOSVM_psp);
276     if (!psp_start) {
277       ERR("error allocating DOS memory\n");
278       SetLastError(ERROR_NOT_ENOUGH_MEMORY);
279       goto load_error;
280     }
281     load_seg=DOSVM_psp+(old_com?0:PSP_SIZE);
282     rel_seg=load_seg;
283     load_start=psp_start+(PSP_SIZE<<4);
284     MZ_CreatePSP(psp_start, env_seg, oldpsp_seg);
285   }
286
287  /* load executable image */
288  TRACE("loading DOS %s image, %08lx bytes\n",old_com?"COM":"EXE",image_size);
289  SetFilePointer(hFile,image_start,NULL,FILE_BEGIN);
290  if (!ReadFile(hFile,load_start,image_size,&len,NULL) || len != image_size) {
291   /* check if this is due to the workaround for the pre-1.10 MS linker and we
292      realy had only 4 bytes on the last page */
293   if (mz_header.e_cblp != 4 || image_size - len != 512 - 4) {
294     SetLastError(ERROR_BAD_FORMAT);
295     goto load_error;
296   }
297  }
298
299  if (mz_header.e_crlc) {
300   /* load relocation table */
301   TRACE("loading DOS EXE relocation table, %d entries\n",mz_header.e_crlc);
302   /* FIXME: is this too slow without read buffering? */
303   SetFilePointer(hFile,mz_header.e_lfarlc,NULL,FILE_BEGIN);
304   for (x=0; x<mz_header.e_crlc; x++) {
305    if (!ReadFile(hFile,&reloc,sizeof(reloc),&len,NULL) || len != sizeof(reloc)) {
306     SetLastError(ERROR_BAD_FORMAT);
307     goto load_error;
308    }
309    *(WORD*)SEGPTR16(load_start,reloc)+=rel_seg;
310   }
311  }
312
313   if (!oblk) {
314     init_cs = load_seg+mz_header.e_cs;
315     init_ip = mz_header.e_ip;
316     init_ss = load_seg+mz_header.e_ss;
317     init_sp = mz_header.e_sp;
318
319     TRACE("entry point: %04x:%04x\n",init_cs,init_ip);
320   }
321
322   if (alloc && !MZ_InitTask()) {
323     SetLastError(ERROR_GEN_FAILURE);
324     return FALSE;
325   }
326
327   return TRUE;
328
329 load_error:
330   DOSVM_psp = oldpsp_seg;
331
332   return FALSE;
333 }
334
335 /***********************************************************************
336  *              wine_load_dos_exe (WINEDOS.@)
337  *
338  * Called from WineVDM when a new real-mode DOS process is started.
339  * Loads DOS program into memory and executes the program.
340  */
341 void WINAPI wine_load_dos_exe( LPCSTR filename, LPCSTR cmdline )
342 {
343     char dos_cmdtail[126];
344     int  dos_length = 0;
345
346     HANDLE hFile = CreateFileA( filename, GENERIC_READ, FILE_SHARE_READ, 
347                                 NULL, OPEN_EXISTING, 0, 0 );
348     if (hFile == INVALID_HANDLE_VALUE) return;
349     DOSVM_isdosexe = TRUE;
350
351     if(cmdline && *cmdline)
352     {
353         dos_length = strlen(cmdline);
354         memmove( dos_cmdtail + 1, cmdline, 
355                  (dos_length < 125) ? dos_length : 125 );
356
357         /* Non-empty command tail always starts with at least one space. */
358         dos_cmdtail[0] = ' ';
359         dos_length++;
360
361         /*
362          * If command tail is longer than 126 characters,
363          * set tail length to 127 and fill CMDLINE environment variable 
364          * with full command line (this includes filename).
365          */
366         if (dos_length > 126)
367         {
368             char *cmd = HeapAlloc( GetProcessHeap(), 0, 
369                                    dos_length + strlen(filename) + 4 );
370             char *ptr = cmd;
371
372             if (!cmd)
373                 return;
374
375             /*
376              * Append filename. If path includes spaces, quote the path.
377              */
378             if (strchr(filename, ' '))
379             {
380                 *ptr++ = '\"';
381                 strcpy( ptr, filename );
382                 ptr += strlen(filename);                   
383                 *ptr++ = '\"';
384             }
385             else
386             {
387                 strcpy( ptr, filename );
388                 ptr += strlen(filename);  
389             }
390
391             /*
392              * Append command tail.
393              */
394             if (cmdline[0] != ' ')
395                 *ptr++ = ' ';
396             strcpy( ptr, cmdline );
397
398             /*
399              * Set environment variable. This will be passed to
400              * new DOS process.
401              */
402             if (!SetEnvironmentVariableA( "CMDLINE", cmd ))
403             {
404                 HeapFree(GetProcessHeap(), 0, cmd );
405                 return;
406             }
407
408             HeapFree(GetProcessHeap(), 0, cmd );
409             dos_length = 127;
410         }
411     }
412
413     if (MZ_DoLoadImage( hFile, filename, NULL )) 
414         MZ_Launch( dos_cmdtail, dos_length );
415 }
416
417 /***********************************************************************
418  *              MZ_Exec
419  *
420  * this may only be called from existing DOS processes
421  */
422 BOOL WINAPI MZ_Exec( CONTEXT86 *context, LPCSTR filename, BYTE func, LPVOID paramblk )
423 {
424   DWORD binType;
425   STARTUPINFOA st;
426   PROCESS_INFORMATION pe;
427   HANDLE hFile;
428
429   BOOL ret = FALSE;
430
431   if(!GetBinaryTypeA(filename, &binType))   /* determine what kind of binary this is */
432   {
433     return FALSE; /* binary is not an executable */
434   }
435
436   /* handle non-dos executables */
437   if(binType != SCS_DOS_BINARY)
438   {
439     if(func == 0) /* load and execute */
440     {
441       LPBYTE fullCmdLine;
442       WORD fullCmdLength;
443       LPBYTE psp_start = (LPBYTE)((DWORD)DOSVM_psp << 4);
444       PDB16 *psp = (PDB16 *)psp_start;
445       ExecBlock *blk = (ExecBlock *)paramblk;
446       LPBYTE cmdline = PTR_REAL_TO_LIN(SELECTOROF(blk->cmdline),OFFSETOF(blk->cmdline));
447       LPBYTE envblock = PTR_REAL_TO_LIN(psp->environment, 0);
448       int    cmdLength = cmdline[0];
449
450       /*
451        * If cmdLength is 127, command tail is truncated and environment 
452        * variable CMDLINE should contain full command line 
453        * (this includes filename).
454        */
455       if (cmdLength == 127)
456       {
457           FIXME( "CMDLINE argument passing is unimplemented.\n" );
458           cmdLength = 126; /* FIXME */
459       }
460
461       fullCmdLength = (strlen(filename) + 1) + cmdLength + 1; /* filename + space + cmdline + terminating null character */
462
463       fullCmdLine = HeapAlloc(GetProcessHeap(), 0, fullCmdLength);
464       if(!fullCmdLine) return FALSE; /* return false on memory alloc failure */
465
466       /* build the full command line from the executable file and the command line being passed in */
467       snprintf(fullCmdLine, fullCmdLength, "%s ", filename); /* start off with the executable filename and a space */
468       memcpy(fullCmdLine + strlen(fullCmdLine), cmdline + 1, cmdLength); /* append cmdline onto the end */
469       fullCmdLine[fullCmdLength - 1] = 0; /* null terminate string */
470
471       ZeroMemory (&st, sizeof(STARTUPINFOA));
472       st.cb = sizeof(STARTUPINFOA);
473       ret = CreateProcessA (NULL, fullCmdLine, NULL, NULL, TRUE, 0, envblock, NULL, &st, &pe);
474
475       /* wait for the app to finish and clean up PROCESS_INFORMATION handles */
476       if(ret)
477       {
478         WaitForSingleObject(pe.hProcess, INFINITE);  /* wait here until the child process is complete */
479         CloseHandle(pe.hProcess);
480         CloseHandle(pe.hThread);
481       }
482
483       HeapFree(GetProcessHeap(), 0, fullCmdLine);  /* free the memory we allocated */
484     }
485     else
486     {
487       FIXME("EXEC type of %d not implemented for non-dos executables\n", func);
488       ret = FALSE;
489     }
490
491     return ret;
492   } /* if(binType != SCS_DOS_BINARY) */
493
494
495   /* handle dos executables */
496
497   hFile = CreateFileA( filename, GENERIC_READ, FILE_SHARE_READ,
498                              NULL, OPEN_EXISTING, 0, 0);
499   if (hFile == INVALID_HANDLE_VALUE) return FALSE;
500
501   switch (func) {
502   case 0: /* load and execute */
503   case 1: /* load but don't execute */
504     {
505       /* save current process's return SS:SP now */
506       LPBYTE psp_start = (LPBYTE)((DWORD)DOSVM_psp << 4);
507       PDB16 *psp = (PDB16 *)psp_start;
508       psp->saveStack = (DWORD)MAKESEGPTR(context->SegSs, LOWORD(context->Esp));
509     }
510     ret = MZ_DoLoadImage( hFile, filename, NULL );
511     if (ret) {
512       /* MZ_LoadImage created a new PSP and loaded new values into it,
513        * let's work on the new values now */
514       LPBYTE psp_start = (LPBYTE)((DWORD)DOSVM_psp << 4);
515       ExecBlock *blk = (ExecBlock *)paramblk;
516       LPBYTE cmdline = PTR_REAL_TO_LIN(SELECTOROF(blk->cmdline),OFFSETOF(blk->cmdline));
517
518       /* First character contains the length of the command line. */
519       MZ_FillPSP(psp_start, cmdline + 1, cmdline[0]);
520
521       /* the lame MS-DOS engineers decided that the return address should be in int22 */
522       DOSVM_SetRMHandler(0x22, (FARPROC16)MAKESEGPTR(context->SegCs, LOWORD(context->Eip)));
523       if (func) {
524         /* don't execute, just return startup state */
525         /*
526          * From Ralph Brown:
527          *  For function 01h, the AX value to be passed to the child program 
528          *  is put on top of the child's stack
529          */
530         LPBYTE stack;
531         init_sp -= 2;
532         stack = (LPBYTE) CTX_SEG_OFF_TO_LIN(context, init_ss, init_sp);
533         /* FIXME: push AX correctly */
534         stack[0] = 0x00;    /* push AL */
535         stack[1] = 0x00;    /* push AH */
536         
537         blk->init_cs = init_cs;
538         blk->init_ip = init_ip;
539         blk->init_ss = init_ss;
540         blk->init_sp = init_sp;
541       } else {
542         /* execute by making us return to new process */
543         context->SegCs = init_cs;
544         context->Eip   = init_ip;
545         context->SegSs = init_ss;
546         context->Esp   = init_sp;
547         context->SegDs = DOSVM_psp;
548         context->SegEs = DOSVM_psp;
549         context->Eax   = 0;
550       }
551     }
552     break;
553   case 3: /* load overlay */
554     {
555       OverlayBlock *blk = (OverlayBlock *)paramblk;
556       ret = MZ_DoLoadImage( hFile, filename, blk );
557     }
558     break;
559   default:
560     FIXME("EXEC load type %d not implemented\n", func);
561     SetLastError(ERROR_INVALID_FUNCTION);
562     break;
563   }
564   CloseHandle(hFile);
565   return ret;
566 }
567
568 /***********************************************************************
569  *              MZ_AllocDPMITask
570  */
571 void WINAPI MZ_AllocDPMITask( void )
572 {
573   MZ_InitMemory();
574   MZ_InitTask();
575 }
576
577 /***********************************************************************
578  *              MZ_RunInThread
579  */
580 void WINAPI MZ_RunInThread( PAPCFUNC proc, ULONG_PTR arg )
581 {
582   if (loop_thread) {
583     DOS_SPC spc;
584     HANDLE event;
585
586     spc.proc = proc;
587     spc.arg = arg;
588     event = CreateEventA(NULL, TRUE, FALSE, NULL);
589     PostThreadMessageA(loop_tid, WM_USER, (WPARAM)event, (LPARAM)&spc);
590     WaitForSingleObject(event, INFINITE);
591     CloseHandle(event);
592   } else
593     proc(arg);
594 }
595
596 static DWORD WINAPI MZ_DOSVM( LPVOID lpExtra )
597 {
598   CONTEXT context;
599   DWORD ret;
600
601   dosvm_pid = getpid();
602
603   memset( &context, 0, sizeof(context) );
604   context.SegCs  = init_cs;
605   context.Eip    = init_ip;
606   context.SegSs  = init_ss;
607   context.Esp    = init_sp;
608   context.SegDs  = DOSVM_psp;
609   context.SegEs  = DOSVM_psp;
610   context.EFlags = V86_FLAG | VIF_MASK;
611   DOSVM_SetTimer(0x10000);
612   ret = DOSVM_Enter( &context );
613
614   dosvm_pid = 0;
615   return ret;
616 }
617
618 static BOOL MZ_InitTask(void)
619 {
620   if (!DuplicateHandle(GetCurrentProcess(), GetCurrentThread(),
621                        GetCurrentProcess(), &loop_thread,
622                        0, FALSE, DUPLICATE_SAME_ACCESS))
623     return FALSE;
624   dosvm_thread = CreateThread(NULL, 0, MZ_DOSVM, NULL, CREATE_SUSPENDED, &dosvm_tid);
625   if (!dosvm_thread) {
626     CloseHandle(loop_thread);
627     loop_thread = 0;
628     return FALSE;
629   }
630   loop_tid = GetCurrentThreadId();
631   return TRUE;
632 }
633
634 static void MZ_Launch( LPCSTR cmdtail, int length )
635 {
636   TDB *pTask = GlobalLock16( GetCurrentTask() );
637   BYTE *psp_start = PTR_REAL_TO_LIN( DOSVM_psp, 0 );
638   DWORD rv;
639   SYSLEVEL *lock;
640
641   MZ_FillPSP(psp_start, cmdtail, length);
642   pTask->flags |= TDBF_WINOLDAP;
643
644   /* DTA is set to PSP:0080h when a program is started. */
645   pTask->dta = MAKESEGPTR( DOSVM_psp, 0x80 );
646
647   GetpWin16Lock( &lock );
648   _LeaveSysLevel( lock );
649
650   ResumeThread(dosvm_thread);
651   rv = DOSVM_Loop(dosvm_thread);
652
653   CloseHandle(dosvm_thread);
654   dosvm_thread = 0; dosvm_tid = 0;
655   CloseHandle(loop_thread);
656   loop_thread = 0; loop_tid = 0;
657
658   VGA_Clean();
659   ExitProcess(rv);
660 }
661
662 /***********************************************************************
663  *              MZ_Exit
664  */
665 void WINAPI MZ_Exit( CONTEXT86 *context, BOOL cs_psp, WORD retval )
666 {
667   if (DOSVM_psp) {
668     WORD psp_seg = cs_psp ? context->SegCs : DOSVM_psp;
669     LPBYTE psp_start = (LPBYTE)((DWORD)psp_seg << 4);
670     PDB16 *psp = (PDB16 *)psp_start;
671     WORD parpsp = psp->parentPSP; /* check for parent DOS process */
672     if (parpsp) {
673       /* retrieve parent's return address */
674       FARPROC16 retaddr = DOSVM_GetRMHandler(0x22);
675       /* restore interrupts */
676       DOSVM_SetRMHandler(0x22, psp->savedint22);
677       DOSVM_SetRMHandler(0x23, psp->savedint23);
678       DOSVM_SetRMHandler(0x24, psp->savedint24);
679       /* FIXME: deallocate file handles etc */
680       /* free process's associated memory
681        * FIXME: walk memory and deallocate all blocks owned by process */
682       DOSMEM_FreeBlock( PTR_REAL_TO_LIN(psp->environment,0) );
683       DOSMEM_FreeBlock( PTR_REAL_TO_LIN(DOSVM_psp,0) );
684       /* switch to parent's PSP */
685       DOSVM_psp = parpsp;
686       psp_start = (LPBYTE)((DWORD)parpsp << 4);
687       psp = (PDB16 *)psp_start;
688       /* now return to parent */
689       DOSVM_retval = retval;
690       context->SegCs = SELECTOROF(retaddr);
691       context->Eip   = OFFSETOF(retaddr);
692       context->SegSs = SELECTOROF(psp->saveStack);
693       context->Esp   = OFFSETOF(psp->saveStack);
694       return;
695     } else
696       TRACE("killing DOS task\n");
697   }
698   ExitThread( retval );
699 }
700
701
702 /***********************************************************************
703  *              MZ_Current
704  */
705 BOOL WINAPI MZ_Current( void )
706 {
707   return (dosvm_pid != 0); /* FIXME: do a better check */
708 }
709
710 #else /* !MZ_SUPPORTED */
711
712 /***********************************************************************
713  *              wine_load_dos_exe (WINEDOS.@)
714  */
715 void WINAPI wine_load_dos_exe( LPCSTR filename, LPCSTR cmdline )
716 {
717   WARN("DOS executables not supported on this platform\n");
718   SetLastError(ERROR_BAD_FORMAT);
719 }
720
721 /***********************************************************************
722  *              MZ_Exec
723  */
724 BOOL WINAPI MZ_Exec( CONTEXT86 *context, LPCSTR filename, BYTE func, LPVOID paramblk )
725 {
726   /* can't happen */
727   SetLastError(ERROR_BAD_FORMAT);
728   return FALSE;
729 }
730
731 /***********************************************************************
732  *              MZ_AllocDPMITask
733  */
734 void WINAPI MZ_AllocDPMITask( void )
735 {
736     ERR("Actual real-mode calls not supported on this platform!\n");
737 }
738
739 /***********************************************************************
740  *              MZ_RunInThread
741  */
742 void WINAPI MZ_RunInThread( PAPCFUNC proc, ULONG_PTR arg )
743 {
744     proc(arg);
745 }
746
747 /***********************************************************************
748  *              MZ_Exit
749  */
750 void WINAPI MZ_Exit( CONTEXT86 *context, BOOL cs_psp, WORD retval )
751 {
752   ExitThread( retval );
753 }
754
755 /***********************************************************************
756  *              MZ_Current
757  */
758 BOOL WINAPI MZ_Current( void )
759 {
760     return FALSE;
761 }
762
763 #endif /* !MZ_SUPPORTED */