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