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 */