From 887ed6228afa0823e4a968ce1dd6ef950067f0cc Mon Sep 17 00:00:00 2001
From: Jukka Heinonen <jhei@iki.fi>
Date: Sun, 5 May 2002 20:22:39 +0000
Subject: [PATCH] Terminating DOS program in real mode now calls ExitThread
 which is exactly what terminating in protected mode does.

---
 dlls/winedos/dosexe.h |  2 +-
 dlls/winedos/dosvm.c  | 17 ++++++++++++++---
 dlls/winedos/module.c | 17 ++++++-----------
 3 files changed, 21 insertions(+), 15 deletions(-)

diff --git a/dlls/winedos/dosexe.h b/dlls/winedos/dosexe.h
index 45c77ef3359..e7fb71f4bcb 100644
--- a/dlls/winedos/dosexe.h
+++ b/dlls/winedos/dosexe.h
@@ -53,7 +53,7 @@ extern void WINAPI MZ_AllocDPMITask( void );
 extern void WINAPI MZ_RunInThread( PAPCFUNC proc, ULONG_PTR arg );
 extern INT WINAPI DOSVM_Enter( CONTEXT86 *context );
 extern void WINAPI DOSVM_Wait( INT read_pipe, HANDLE hObject );
-extern DWORD WINAPI DOSVM_Loop( LPVOID lpExtra );
+extern DWORD WINAPI DOSVM_Loop( HANDLE hThread );
 extern void WINAPI DOSVM_QueueEvent( INT irq, INT priority, DOSRELAY relay, LPVOID data );
 extern void WINAPI DOSVM_PIC_ioport_out( WORD port, BYTE val );
 extern void WINAPI DOSVM_SetTimer( UINT ticks );
diff --git a/dlls/winedos/dosvm.c b/dlls/winedos/dosvm.c
index be5b902411a..35f89d1a34d 100644
--- a/dlls/winedos/dosvm.c
+++ b/dlls/winedos/dosvm.c
@@ -370,19 +370,30 @@ chk_console_input:
   } while (TRUE);
 }
 
-DWORD WINAPI DOSVM_Loop( LPVOID lpExtra )
+DWORD WINAPI DOSVM_Loop( HANDLE hThread )
 {
-  HANDLE obj = GetStdHandle(STD_INPUT_HANDLE);
+  HANDLE objs[2];
   MSG msg;
   DWORD waitret;
 
+  objs[0] = GetStdHandle(STD_INPUT_HANDLE);
+  objs[1] = hThread;
+
   for(;;) {
       TRACE_(int)("waiting for action\n");
-      waitret = MsgWaitForMultipleObjects(1, &obj, FALSE, INFINITE, QS_ALLINPUT);
+      waitret = MsgWaitForMultipleObjects(2, objs, FALSE, INFINITE, QS_ALLINPUT);
       if (waitret == WAIT_OBJECT_0) {
           DOSVM_ProcessConsole();
       }
       else if (waitret == WAIT_OBJECT_0 + 1) {
+         DWORD rv;
+         if(!GetExitCodeThread(hThread, &rv)) {
+             ERR("Failed to get thread exit code!\n");
+             rv = 0;
+         }
+         return rv;
+      }
+      else if (waitret == WAIT_OBJECT_0 + 2) {
           while (PeekMessageA(&msg,0,0,0,PM_REMOVE)) {
               if (msg.hwnd) {
                   /* it's a window message */
diff --git a/dlls/winedos/module.c b/dlls/winedos/module.c
index 1a87fae852f..91cc0baa7fc 100644
--- a/dlls/winedos/module.c
+++ b/dlls/winedos/module.c
@@ -94,7 +94,6 @@ static DWORD dosvm_tid, loop_tid;
 
 static void MZ_Launch(void);
 static BOOL MZ_InitTask(void);
-static void MZ_KillTask(void);
 
 static void MZ_CreatePSP( LPVOID lpPSP, WORD env, WORD par )
 {
@@ -474,6 +473,7 @@ static void MZ_Launch(void)
   TDB *pTask = TASK_GetCurrent();
   BYTE *psp_start = PTR_REAL_TO_LIN( DOSVM_psp, 0 );
   LPSTR cmdline = GetCommandLineA();
+  DWORD rv;
 
   MZ_FillPSP(psp_start, cmdline, cmdline ? strlen(cmdline) : 0);
   pTask->flags |= TDBF_WINOLDAP;
@@ -481,20 +481,15 @@ static void MZ_Launch(void)
   _LeaveWin16Lock();
   
   ResumeThread(dosvm_thread);
-  DOSVM_Loop(NULL);
-  ExitThread(0);
-}
+  rv = DOSVM_Loop(dosvm_thread);
 
-static void MZ_KillTask(void)
-{
-  TRACE("killing DOS task\n");
-  VGA_Clean();
-  PostThreadMessageA(loop_tid, WM_QUIT, 0, 0);
-  WaitForSingleObject(loop_thread, INFINITE); /* ? */
   CloseHandle(dosvm_thread);
   dosvm_thread = 0; dosvm_tid = 0;
   CloseHandle(loop_thread);
   loop_thread = 0; loop_tid = 0;
+
+  VGA_Clean();
+  ExitThread(rv);
 }
 
 /***********************************************************************
@@ -531,7 +526,7 @@ void WINAPI MZ_Exit( CONTEXT86 *context, BOOL cs_psp, WORD retval )
       context->Esp   = OFFSETOF(psp->saveStack);
       return;
     } else
-      MZ_KillTask();
+      TRACE("killing DOS task\n");
   }
   ExitThread( retval );
 }