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