Moved a bunch of definitions out of module.h into a new
[wine] / dlls / kernel / task.c
1 /*
2  * Task functions
3  *
4  * Copyright 1995 Alexandre Julliard
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
21 #include "config.h"
22 #include "wine/port.h"
23
24 #include <stdarg.h>
25 #include <stdlib.h>
26 #include <string.h>
27 #include <assert.h>
28 #ifdef HAVE_UNISTD_H
29 # include <unistd.h>
30 #endif
31
32 #include "windef.h"
33 #include "winbase.h"
34 #include "wingdi.h"
35 #include "winnt.h"
36 #include "wownt32.h"
37 #include "winuser.h"
38
39 #include "wine/winbase16.h"
40 #include "thread.h"
41 #include "winternl.h"
42 #include "wine/server.h"
43 #include "toolhelp.h"
44 #include "kernel_private.h"
45 #include "kernel16_private.h"
46
47 #include "wine/debug.h"
48
49 WINE_DEFAULT_DEBUG_CHANNEL(task);
50 WINE_DECLARE_DEBUG_CHANNEL(relay);
51 WINE_DECLARE_DEBUG_CHANNEL(toolhelp);
52
53 #include "pshpack1.h"
54
55 /* Segment containing MakeProcInstance() thunks */
56 typedef struct
57 {
58     WORD  next;       /* Selector of next segment */
59     WORD  magic;      /* Thunks signature */
60     WORD  unused;
61     WORD  free;       /* Head of the free list */
62     WORD  thunks[4];  /* Each thunk is 4 words long */
63 } THUNKS;
64
65 #include "poppack.h"
66
67 #define THUNK_MAGIC  ('P' | ('T' << 8))
68
69   /* Min. number of thunks allocated when creating a new segment */
70 #define MIN_THUNKS  32
71
72 #define TDB_MAGIC    ('T' | ('D' << 8))
73
74 static THHOOK DefaultThhook;
75 THHOOK *pThhook = &DefaultThhook;
76
77 #define hFirstTask   (pThhook->HeadTDB)
78 #define hLockedTask  (pThhook->LockTDB)
79
80 static UINT16 nTaskCount = 0;
81
82 static HTASK16 initial_task;
83
84 /***********************************************************************
85  *           TASK_InstallTHHook
86  */
87 void TASK_InstallTHHook( THHOOK *pNewThhook )
88 {
89      THHOOK *pOldThhook = pThhook;
90
91      pThhook = pNewThhook? pNewThhook : &DefaultThhook;
92
93      *pThhook = *pOldThhook;
94 }
95
96 /***********************************************************************
97  *           TASK_GetPtr
98  */
99 static TDB *TASK_GetPtr( HTASK16 hTask )
100 {
101     return GlobalLock16( hTask );
102 }
103
104
105 /***********************************************************************
106  *           TASK_GetCurrent
107  */
108 TDB *TASK_GetCurrent(void)
109 {
110     return TASK_GetPtr( GetCurrentTask() );
111 }
112
113
114 /***********************************************************************
115  *           TASK_LinkTask
116  */
117 static void TASK_LinkTask( HTASK16 hTask )
118 {
119     HTASK16 *prevTask;
120     TDB *pTask;
121
122     if (!(pTask = TASK_GetPtr( hTask ))) return;
123     prevTask = &hFirstTask;
124     while (*prevTask)
125     {
126         TDB *prevTaskPtr = TASK_GetPtr( *prevTask );
127         if (prevTaskPtr->priority >= pTask->priority) break;
128         prevTask = &prevTaskPtr->hNext;
129     }
130     pTask->hNext = *prevTask;
131     *prevTask = hTask;
132     nTaskCount++;
133 }
134
135
136 /***********************************************************************
137  *           TASK_UnlinkTask
138  */
139 static void TASK_UnlinkTask( HTASK16 hTask )
140 {
141     HTASK16 *prevTask;
142     TDB *pTask;
143
144     prevTask = &hFirstTask;
145     while (*prevTask && (*prevTask != hTask))
146     {
147         pTask = TASK_GetPtr( *prevTask );
148         prevTask = &pTask->hNext;
149     }
150     if (*prevTask)
151     {
152         pTask = TASK_GetPtr( *prevTask );
153         *prevTask = pTask->hNext;
154         pTask->hNext = 0;
155         nTaskCount--;
156     }
157 }
158
159
160 /***********************************************************************
161  *           TASK_CreateThunks
162  *
163  * Create a thunk free-list in segment 'handle', starting from offset 'offset'
164  * and containing 'count' entries.
165  */
166 static void TASK_CreateThunks( HGLOBAL16 handle, WORD offset, WORD count )
167 {
168     int i;
169     WORD free;
170     THUNKS *pThunk;
171
172     pThunk = (THUNKS *)((BYTE *)GlobalLock16( handle ) + offset);
173     pThunk->next = 0;
174     pThunk->magic = THUNK_MAGIC;
175     pThunk->free = (int)&pThunk->thunks - (int)pThunk;
176     free = pThunk->free;
177     for (i = 0; i < count-1; i++)
178     {
179         free += 8;  /* Offset of next thunk */
180         pThunk->thunks[4*i] = free;
181     }
182     pThunk->thunks[4*i] = 0;  /* Last thunk */
183 }
184
185
186 /***********************************************************************
187  *           TASK_AllocThunk
188  *
189  * Allocate a thunk for MakeProcInstance().
190  */
191 static SEGPTR TASK_AllocThunk(void)
192 {
193     TDB *pTask;
194     THUNKS *pThunk;
195     WORD sel, base;
196
197     if (!(pTask = TASK_GetCurrent())) return 0;
198     sel = pTask->hCSAlias;
199     pThunk = (THUNKS *)&pTask->thunks;
200     base = (char *)pThunk - (char *)pTask;
201     while (!pThunk->free)
202     {
203         sel = pThunk->next;
204         if (!sel)  /* Allocate a new segment */
205         {
206             sel = GLOBAL_Alloc( GMEM_FIXED, sizeof(THUNKS) + (MIN_THUNKS-1)*8,
207                                 pTask->hPDB, WINE_LDT_FLAGS_CODE );
208             if (!sel) return (SEGPTR)0;
209             TASK_CreateThunks( sel, 0, MIN_THUNKS );
210             pThunk->next = sel;
211         }
212         pThunk = (THUNKS *)GlobalLock16( sel );
213         base = 0;
214     }
215     base += pThunk->free;
216     pThunk->free = *(WORD *)((BYTE *)pThunk + pThunk->free);
217     return MAKESEGPTR( sel, base );
218 }
219
220
221 /***********************************************************************
222  *           TASK_FreeThunk
223  *
224  * Free a MakeProcInstance() thunk.
225  */
226 static BOOL TASK_FreeThunk( SEGPTR thunk )
227 {
228     TDB *pTask;
229     THUNKS *pThunk;
230     WORD sel, base;
231
232     if (!(pTask = TASK_GetCurrent())) return 0;
233     sel = pTask->hCSAlias;
234     pThunk = (THUNKS *)&pTask->thunks;
235     base = (char *)pThunk - (char *)pTask;
236     while (sel && (sel != HIWORD(thunk)))
237     {
238         sel = pThunk->next;
239         pThunk = (THUNKS *)GlobalLock16( sel );
240         base = 0;
241     }
242     if (!sel) return FALSE;
243     *(WORD *)((BYTE *)pThunk + LOWORD(thunk) - base) = pThunk->free;
244     pThunk->free = LOWORD(thunk) - base;
245     return TRUE;
246 }
247
248
249 /***********************************************************************
250  *           TASK_Create
251  *
252  * NOTE: This routine might be called by a Win32 thread. Thus, we need
253  *       to be careful to protect global data structures. We do this
254  *       by entering the Win16Lock while linking the task into the
255  *       global task list.
256  */
257 static TDB *TASK_Create( NE_MODULE *pModule, UINT16 cmdShow, LPCSTR cmdline, BYTE len )
258 {
259     HTASK16 hTask;
260     TDB *pTask;
261     FARPROC16 proc;
262     char curdir[MAX_PATH];
263     HMODULE16 hModule = pModule ? pModule->self : 0;
264
265       /* Allocate the task structure */
266
267     hTask = GlobalAlloc16( GMEM_FIXED | GMEM_ZEROINIT, sizeof(TDB) );
268     if (!hTask) return NULL;
269     pTask = TASK_GetPtr( hTask );
270     FarSetOwner16( hTask, hModule );
271
272     /* Fill the task structure */
273
274     pTask->hSelf = hTask;
275
276     pTask->version       = pModule ? pModule->expected_version : 0x0400;
277     pTask->hModule       = hModule;
278     pTask->hParent       = GetCurrentTask();
279     pTask->magic         = TDB_MAGIC;
280     pTask->nCmdShow      = cmdShow;
281
282     GetCurrentDirectoryA( sizeof(curdir), curdir );
283     GetShortPathNameA( curdir, curdir, sizeof(curdir) );
284     pTask->curdrive = (curdir[0] - 'A') | 0x80;
285     lstrcpynA( pTask->curdir, curdir + 2, sizeof(pTask->curdir) );
286
287       /* Create the thunks block */
288
289     TASK_CreateThunks( hTask, (char *)&pTask->thunks - (char *)pTask, 7 );
290
291       /* Copy the module name */
292
293     if (hModule)
294     {
295         char name[sizeof(pTask->module_name)+1];
296         size_t len;
297         GetModuleName16( hModule, name, sizeof(name) );
298         len = strlen(name) + 1;
299         memcpy(pTask->module_name, name, min(len,sizeof(pTask->module_name)));
300         pTask->compat_flags = GetProfileIntA( "Compatibility", name, 0 );
301     }
302
303       /* Allocate a selector for the PDB */
304
305     pTask->hPDB = GLOBAL_CreateBlock( GMEM_FIXED, &pTask->pdb, sizeof(PDB16),
306                                       hModule, WINE_LDT_FLAGS_DATA );
307
308       /* Fill the PDB */
309
310     pTask->pdb.int20 = 0x20cd;
311     pTask->pdb.dispatcher[0] = 0x9a;  /* ljmp */
312     proc = GetProcAddress16( GetModuleHandle16("KERNEL"), "DOS3Call" );
313     memcpy( &pTask->pdb.dispatcher[1], &proc, sizeof(proc) );
314     pTask->pdb.savedint22 = 0;
315     pTask->pdb.savedint23 = 0;
316     pTask->pdb.savedint24 = 0;
317     pTask->pdb.fileHandlesPtr =
318         MAKESEGPTR( GlobalHandleToSel16(pTask->hPDB), (int)&((PDB16 *)0)->fileHandles );
319     pTask->pdb.hFileHandles = 0;
320     memset( pTask->pdb.fileHandles, 0xff, sizeof(pTask->pdb.fileHandles) );
321     /* FIXME: should we make a copy of the environment? */
322     pTask->pdb.environment    = SELECTOROF(GetDOSEnvironment16());
323     pTask->pdb.nbFiles        = 20;
324
325     /* Fill the command line */
326
327     if (!cmdline)
328     {
329         cmdline = GetCommandLineA();
330         /* remove the first word (program name) */
331         if (*cmdline == '"')
332             if (!(cmdline = strchr( cmdline+1, '"' ))) cmdline = GetCommandLineA();
333         while (*cmdline && (*cmdline != ' ') && (*cmdline != '\t')) cmdline++;
334         while ((*cmdline == ' ') || (*cmdline == '\t')) cmdline++;
335         len = strlen(cmdline);
336     }
337     if (len >= sizeof(pTask->pdb.cmdLine)) len = sizeof(pTask->pdb.cmdLine)-1;
338     pTask->pdb.cmdLine[0] = len;
339     memcpy( pTask->pdb.cmdLine + 1, cmdline, len );
340     /* pTask->pdb.cmdLine[len+1] = 0; */
341
342     TRACE("cmdline='%.*s' task=%04x\n", len, cmdline, hTask );
343
344       /* Allocate a code segment alias for the TDB */
345
346     pTask->hCSAlias = GLOBAL_CreateBlock( GMEM_FIXED, (void *)pTask,
347                                           sizeof(TDB), pTask->hPDB, WINE_LDT_FLAGS_CODE );
348
349       /* Default DTA overwrites command line */
350
351     pTask->dta = MAKESEGPTR( pTask->hPDB, (int)&pTask->pdb.cmdLine - (int)&pTask->pdb );
352
353     /* Create scheduler event for 16-bit tasks */
354
355     if ( !(pTask->flags & TDBF_WIN32) )
356         NtCreateEvent( &pTask->hEvent, EVENT_ALL_ACCESS, NULL, TRUE, FALSE );
357
358     if (!initial_task) initial_task = hTask;
359
360     return pTask;
361 }
362
363
364 /***********************************************************************
365  *           TASK_DeleteTask
366  */
367 static void TASK_DeleteTask( HTASK16 hTask )
368 {
369     TDB *pTask;
370     HGLOBAL16 hPDB;
371
372     if (!(pTask = TASK_GetPtr( hTask ))) return;
373     hPDB = pTask->hPDB;
374
375     pTask->magic = 0xdead; /* invalidate signature */
376
377     /* Free the selector aliases */
378
379     GLOBAL_FreeBlock( pTask->hCSAlias );
380     GLOBAL_FreeBlock( pTask->hPDB );
381
382     /* Free the task module */
383
384     FreeModule16( pTask->hModule );
385
386     /* Free the task structure itself */
387
388     GlobalFree16( hTask );
389
390     /* Free all memory used by this task (including the 32-bit stack, */
391     /* the environment block and the thunk segments). */
392
393     GlobalFreeAll16( hPDB );
394 }
395
396
397 /***********************************************************************
398  *           TASK_CreateMainTask
399  *
400  * Create a task for the main (32-bit) process.
401  */
402 void TASK_CreateMainTask(void)
403 {
404     TDB *pTask;
405     STARTUPINFOA startup_info;
406     UINT cmdShow = 1; /* SW_SHOWNORMAL but we don't want to include winuser.h here */
407
408     GetStartupInfoA( &startup_info );
409     if (startup_info.dwFlags & STARTF_USESHOWWINDOW) cmdShow = startup_info.wShowWindow;
410     pTask = TASK_Create( NULL, cmdShow, NULL, 0 );
411     if (!pTask)
412     {
413         ERR("could not create task for main process\n");
414         ExitProcess(1);
415     }
416
417     pTask->flags        |= TDBF_WIN32;
418     pTask->hInstance     = 0;
419     pTask->hPrevInstance = 0;
420     pTask->teb           = NtCurrentTeb();
421
422     NtCurrentTeb()->htask16 = pTask->hSelf;
423
424     /* Add the task to the linked list */
425     /* (no need to get the win16 lock, we are the only thread at this point) */
426     TASK_LinkTask( pTask->hSelf );
427 }
428
429
430 /* allocate the win16 TIB for a new 16-bit task */
431 static WIN16_SUBSYSTEM_TIB *allocate_win16_tib( TDB *pTask )
432 {
433     WCHAR path[MAX_PATH];
434     WIN16_SUBSYSTEM_TIB *tib;
435     UNICODE_STRING *curdir;
436     NE_MODULE *pModule = NE_GetPtr( pTask->hModule );
437
438     if (!(tib = HeapAlloc( GetProcessHeap(), 0, sizeof(*tib) ))) return NULL;
439     MultiByteToWideChar( CP_ACP, 0, NE_MODULE_NAME(pModule), -1, path, MAX_PATH );
440     GetLongPathNameW( path, path, MAX_PATH );
441     if (RtlCreateUnicodeString( &tib->exe_str, path )) tib->exe_name = &tib->exe_str;
442     else tib->exe_name = NULL;
443
444     RtlAcquirePebLock();
445     curdir = &NtCurrentTeb()->Peb->ProcessParameters->CurrentDirectory.DosPath;
446     tib->curdir.DosPath.MaximumLength = sizeof(tib->curdir_buffer);
447     tib->curdir.DosPath.Length = min( curdir->Length, tib->curdir.DosPath.MaximumLength-sizeof(WCHAR) );
448     tib->curdir.DosPath.Buffer = tib->curdir_buffer;
449     tib->curdir.Handle = 0;
450     memcpy( tib->curdir_buffer, curdir->Buffer, tib->curdir.DosPath.Length );
451     tib->curdir_buffer[tib->curdir.DosPath.Length/sizeof(WCHAR)] = 0;
452     RtlReleasePebLock();
453     return tib;
454 }
455
456 /* startup routine for a new 16-bit thread */
457 static DWORD CALLBACK task_start( LPVOID p )
458 {
459     TDB *pTask = (TDB *)p;
460     DWORD ret;
461
462     NtCurrentTeb()->htask16 = pTask->hSelf;
463     NtCurrentTeb()->Tib.SubSystemTib = allocate_win16_tib( pTask );
464
465     _EnterWin16Lock();
466     TASK_LinkTask( pTask->hSelf );
467     pTask->teb = NtCurrentTeb();
468     ret = NE_StartTask();
469     _LeaveWin16Lock();
470     return ret;
471 }
472
473
474 /***********************************************************************
475  *           TASK_SpawnTask
476  *
477  * Spawn a new 16-bit task.
478  */
479 HTASK16 TASK_SpawnTask( NE_MODULE *pModule, WORD cmdShow,
480                         LPCSTR cmdline, BYTE len, HANDLE *hThread )
481 {
482     TDB *pTask;
483
484     if (!(pTask = TASK_Create( pModule, cmdShow, cmdline, len ))) return 0;
485     if (!(*hThread = CreateThread( NULL, 0, task_start, pTask, 0, NULL )))
486     {
487         TASK_DeleteTask( pTask->hSelf );
488         return 0;
489     }
490     return pTask->hSelf;
491 }
492
493
494 /***********************************************************************
495  *           TASK_GetTaskFromThread
496  */
497 HTASK16 TASK_GetTaskFromThread( DWORD thread )
498 {
499     TDB *p = TASK_GetPtr( hFirstTask );
500     while (p)
501     {
502         if (p->teb->ClientId.UniqueThread == (HANDLE)thread) return p->hSelf;
503         p = TASK_GetPtr( p->hNext );
504     }
505     return 0;
506 }
507
508
509 /***********************************************************************
510  *           TASK_CallTaskSignalProc
511  */
512 static void TASK_CallTaskSignalProc( UINT16 uCode, HANDLE16 hTaskOrModule )
513 {
514     WORD args[5];
515     TDB *pTask = TASK_GetCurrent();
516
517     if ( !pTask || !pTask->userhandler ) return;
518
519     args[4] = hTaskOrModule;
520     args[3] = uCode;
521     args[2] = 0;
522     args[1] = pTask->hInstance;
523     args[0] = pTask->hQueue;
524     WOWCallback16Ex( (DWORD)pTask->userhandler, WCB16_PASCAL, sizeof(args), args, NULL );
525 }
526
527
528 /***********************************************************************
529  *           TASK_ExitTask
530  */
531 void TASK_ExitTask(void)
532 {
533     WIN16_SUBSYSTEM_TIB *tib;
534     TDB *pTask;
535     DWORD lockCount;
536
537     /* Enter the Win16Lock to protect global data structures */
538     _EnterWin16Lock();
539
540     pTask = TASK_GetCurrent();
541     if ( !pTask )
542     {
543         _LeaveWin16Lock();
544         return;
545     }
546
547     TRACE("Killing task %04x\n", pTask->hSelf );
548
549     /* Perform USER cleanup */
550
551     TASK_CallTaskSignalProc( USIG16_TERMINATION, pTask->hSelf );
552
553     /* Remove the task from the list to be sure we never switch back to it */
554     TASK_UnlinkTask( pTask->hSelf );
555
556     if (!nTaskCount || (nTaskCount == 1 && hFirstTask == initial_task))
557     {
558         TRACE("this is the last task, exiting\n" );
559         ExitKernel16();
560     }
561
562     pTask->nEvents = 0;
563
564     if ( hLockedTask == pTask->hSelf )
565         hLockedTask = 0;
566
567     TASK_DeleteTask( pTask->hSelf );
568
569     if ((tib = NtCurrentTeb()->Tib.SubSystemTib))
570     {
571         if (tib->exe_name) RtlFreeUnicodeString( tib->exe_name );
572         HeapFree( GetProcessHeap(), 0, tib );
573         NtCurrentTeb()->Tib.SubSystemTib = NULL;
574     }
575
576     /* ... and completely release the Win16Lock, just in case. */
577     ReleaseThunkLock( &lockCount );
578 }
579
580
581 /***********************************************************************
582  *           ExitKernel (KERNEL.2)
583  *
584  * Clean-up everything and exit the Wine process.
585  */
586 void WINAPI ExitKernel16(void)
587 {
588     WriteOutProfiles16();
589     TerminateProcess( GetCurrentProcess(), 0 );
590 }
591
592
593 /***********************************************************************
594  *           InitTask  (KERNEL.91)
595  *
596  * Called by the application startup code.
597  */
598 void WINAPI InitTask16( CONTEXT86 *context )
599 {
600     TDB *pTask;
601     INSTANCEDATA *pinstance;
602     SEGPTR ptr;
603
604     context->Eax = 0;
605     if (!(pTask = TASK_GetCurrent())) return;
606
607     /* Note: we need to trust that BX/CX contain the stack/heap sizes,
608        as some apps, notably Visual Basic apps, *modify* the heap/stack
609        size of the instance data segment before calling InitTask() */
610
611     /* Initialize the INSTANCEDATA structure */
612     pinstance = MapSL( MAKESEGPTR(CURRENT_DS, 0) );
613     pinstance->stackmin    = OFFSETOF(NtCurrentTeb()->WOW32Reserved) + sizeof( STACK16FRAME );
614     pinstance->stackbottom = pinstance->stackmin; /* yup, that's right. Confused me too. */
615     pinstance->stacktop    = ( pinstance->stackmin > LOWORD(context->Ebx) ?
616                                pinstance->stackmin - LOWORD(context->Ebx) : 0 ) + 150;
617
618     /* Initialize the local heap */
619     if (LOWORD(context->Ecx))
620         LocalInit16( GlobalHandleToSel16(pTask->hInstance), 0, LOWORD(context->Ecx) );
621
622     /* Initialize implicitly loaded DLLs */
623     NE_InitializeDLLs( pTask->hModule );
624     NE_DllProcessAttach( pTask->hModule );
625
626     /* Registers on return are:
627      * ax     1 if OK, 0 on error
628      * cx     stack limit in bytes
629      * dx     cmdShow parameter
630      * si     instance handle of the previous instance
631      * di     instance handle of the new task
632      * es:bx  pointer to command line inside PSP
633      *
634      * 0 (=%bp) is pushed on the stack
635      */
636     ptr = stack16_push( sizeof(WORD) );
637     *(WORD *)MapSL(ptr) = 0;
638     context->Esp -= 2;
639
640     context->Eax = 1;
641
642     if (!pTask->pdb.cmdLine[0]) context->Ebx = 0x80;
643     else
644     {
645         LPBYTE p = &pTask->pdb.cmdLine[1];
646         while ((*p == ' ') || (*p == '\t')) p++;
647         context->Ebx = 0x80 + (p - pTask->pdb.cmdLine);
648     }
649     context->Ecx   = pinstance->stacktop;
650     context->Edx   = pTask->nCmdShow;
651     context->Esi   = (DWORD)pTask->hPrevInstance;
652     context->Edi   = (DWORD)pTask->hInstance;
653     context->SegEs = (WORD)pTask->hPDB;
654 }
655
656
657 /***********************************************************************
658  *           WaitEvent  (KERNEL.30)
659  */
660 BOOL16 WINAPI WaitEvent16( HTASK16 hTask )
661 {
662     TDB *pTask;
663
664     if (!hTask) hTask = GetCurrentTask();
665     pTask = TASK_GetPtr( hTask );
666
667     if (pTask->flags & TDBF_WIN32)
668     {
669         FIXME("called for Win32 thread (%04lx)!\n", GetCurrentThreadId());
670         return TRUE;
671     }
672
673     if (pTask->nEvents > 0)
674     {
675         pTask->nEvents--;
676         return FALSE;
677     }
678
679     if (pTask->teb == NtCurrentTeb())
680     {
681         DWORD lockCount;
682
683         NtResetEvent( pTask->hEvent, NULL );
684         ReleaseThunkLock( &lockCount );
685         SYSLEVEL_CheckNotLevel( 1 );
686         WaitForSingleObject( pTask->hEvent, INFINITE );
687         RestoreThunkLock( lockCount );
688         if (pTask->nEvents > 0) pTask->nEvents--;
689     }
690     else FIXME("for other task %04x cur=%04x\n",pTask->hSelf,GetCurrentTask());
691
692     return TRUE;
693 }
694
695
696 /***********************************************************************
697  *           PostEvent  (KERNEL.31)
698  */
699 void WINAPI PostEvent16( HTASK16 hTask )
700 {
701     TDB *pTask;
702
703     if (!hTask) hTask = GetCurrentTask();
704     if (!(pTask = TASK_GetPtr( hTask ))) return;
705
706     if (pTask->flags & TDBF_WIN32)
707     {
708         FIXME("called for Win32 thread (%04lx)!\n", (DWORD)pTask->teb->ClientId.UniqueThread );
709         return;
710     }
711
712     pTask->nEvents++;
713
714     if (pTask->nEvents == 1) NtSetEvent( pTask->hEvent, NULL );
715 }
716
717
718 /***********************************************************************
719  *           SetPriority  (KERNEL.32)
720  */
721 void WINAPI SetPriority16( HTASK16 hTask, INT16 delta )
722 {
723     TDB *pTask;
724     INT16 newpriority;
725
726     if (!hTask) hTask = GetCurrentTask();
727     if (!(pTask = TASK_GetPtr( hTask ))) return;
728     newpriority = pTask->priority + delta;
729     if (newpriority < -32) newpriority = -32;
730     else if (newpriority > 15) newpriority = 15;
731
732     pTask->priority = newpriority + 1;
733     TASK_UnlinkTask( pTask->hSelf );
734     TASK_LinkTask( pTask->hSelf );
735     pTask->priority--;
736 }
737
738
739 /***********************************************************************
740  *           LockCurrentTask  (KERNEL.33)
741  */
742 HTASK16 WINAPI LockCurrentTask16( BOOL16 bLock )
743 {
744     if (bLock) hLockedTask = GetCurrentTask();
745     else hLockedTask = 0;
746     return hLockedTask;
747 }
748
749
750 /***********************************************************************
751  *           IsTaskLocked  (KERNEL.122)
752  */
753 HTASK16 WINAPI IsTaskLocked16(void)
754 {
755     return hLockedTask;
756 }
757
758
759 /***********************************************************************
760  *           OldYield  (KERNEL.117)
761  */
762 void WINAPI OldYield16(void)
763 {
764    DWORD count;
765
766    ReleaseThunkLock(&count);
767    RestoreThunkLock(count);
768 }
769
770 /***********************************************************************
771  *           WIN32_OldYield  (KERNEL.447)
772  */
773 void WINAPI WIN32_OldYield16(void)
774 {
775    DWORD count;
776
777    ReleaseThunkLock(&count);
778    RestoreThunkLock(count);
779 }
780
781 /***********************************************************************
782  *           DirectedYield  (KERNEL.150)
783  */
784 void WINAPI DirectedYield16( HTASK16 hTask )
785 {
786     OldYield16();
787 }
788
789 /***********************************************************************
790  *           Yield  (KERNEL.29)
791  */
792 void WINAPI Yield16(void)
793 {
794     TDB *pCurTask = TASK_GetCurrent();
795
796     if (pCurTask && pCurTask->hQueue)
797     {
798         HMODULE mod = GetModuleHandleA( "user32.dll" );
799         if (mod)
800         {
801             FARPROC proc = GetProcAddress( mod, "UserYield16" );
802             if (proc)
803             {
804                 proc();
805                 return;
806             }
807         }
808     }
809     OldYield16();
810 }
811
812 /***********************************************************************
813  *           KERNEL_490  (KERNEL.490)
814  */
815 HTASK16 WINAPI KERNEL_490( HTASK16 someTask )
816 {
817     if ( !someTask ) return 0;
818
819     FIXME("(%04x): stub\n", someTask );
820     return 0;
821 }
822
823 /***********************************************************************
824  *           MakeProcInstance  (KERNEL.51)
825  */
826 FARPROC16 WINAPI MakeProcInstance16( FARPROC16 func, HANDLE16 hInstance )
827 {
828     BYTE *thunk,*lfunc;
829     SEGPTR thunkaddr;
830     WORD hInstanceSelector;
831
832     hInstanceSelector = GlobalHandleToSel16(hInstance);
833
834     TRACE("(%08lx, %04x);\n", (DWORD)func, hInstance);
835
836     if (!HIWORD(func)) {
837       /* Win95 actually protects via SEH, but this is better for debugging */
838       WARN("Ouch ! Called with invalid func 0x%08lx !\n", (DWORD)func);
839       return (FARPROC16)0;
840     }
841
842     if ( (GlobalHandleToSel16(CURRENT_DS) != hInstanceSelector)
843       && (hInstance != 0)
844       && (hInstance != 0xffff) )
845     {
846         /* calling MPI with a foreign DSEG is invalid ! */
847         WARN("Problem with hInstance? Got %04x, using %04x instead\n",
848                    hInstance,CURRENT_DS);
849     }
850
851     /* Always use the DSEG that MPI was entered with.
852      * We used to set hInstance to GetTaskDS16(), but this should be wrong
853      * as CURRENT_DS provides the DSEG value we need.
854      * ("calling" DS, *not* "task" DS !) */
855     hInstanceSelector = CURRENT_DS;
856     hInstance = GlobalHandle16(hInstanceSelector);
857
858     /* no thunking for DLLs */
859     if (NE_GetPtr(FarGetOwner16(hInstance))->flags & NE_FFLAGS_LIBMODULE)
860         return func;
861
862     thunkaddr = TASK_AllocThunk();
863     if (!thunkaddr) return (FARPROC16)0;
864     thunk = MapSL( thunkaddr );
865     lfunc = MapSL( (SEGPTR)func );
866
867     TRACE("(%08lx,%04x): got thunk %08lx\n",
868           (DWORD)func, hInstance, (DWORD)thunkaddr );
869     if (((lfunc[0]==0x8c) && (lfunc[1]==0xd8)) || /* movw %ds, %ax */
870         ((lfunc[0]==0x1e) && (lfunc[1]==0x58))    /* pushw %ds, popw %ax */
871     ) {
872         WARN("This was the (in)famous \"thunk useless\" warning. We thought we have to overwrite with nop;nop;, but this isn't true.\n");
873     }
874
875     *thunk++ = 0xb8;    /* movw instance, %ax */
876     *thunk++ = (BYTE)(hInstanceSelector & 0xff);
877     *thunk++ = (BYTE)(hInstanceSelector >> 8);
878     *thunk++ = 0xea;    /* ljmp func */
879     *(DWORD *)thunk = (DWORD)func;
880     return (FARPROC16)thunkaddr;
881     /* CX reg indicates if thunkaddr != NULL, implement if needed */
882 }
883
884
885 /***********************************************************************
886  *           FreeProcInstance  (KERNEL.52)
887  */
888 void WINAPI FreeProcInstance16( FARPROC16 func )
889 {
890     TRACE("(%08lx)\n", (DWORD)func );
891     TASK_FreeThunk( (SEGPTR)func );
892 }
893
894 /**********************************************************************
895  *          TASK_GetCodeSegment
896  *
897  * Helper function for GetCodeHandle/GetCodeInfo: Retrieve the module
898  * and logical segment number of a given code segment.
899  *
900  * 'proc' either *is* already a pair of module handle and segment number,
901  * in which case there's nothing to do.  Otherwise, it is a pointer to
902  * a function, and we need to retrieve the code segment.  If the pointer
903  * happens to point to a thunk, we'll retrieve info about the code segment
904  * where the function pointed to by the thunk resides, not the thunk itself.
905  *
906  * FIXME: if 'proc' is a SNOOP16 return stub, we should retrieve info about
907  *        the function the snoop code will return to ...
908  *
909  */
910 static BOOL TASK_GetCodeSegment( FARPROC16 proc, NE_MODULE **ppModule,
911                                  SEGTABLEENTRY **ppSeg, int *pSegNr )
912 {
913     NE_MODULE *pModule = NULL;
914     SEGTABLEENTRY *pSeg = NULL;
915     int segNr=0;
916
917     /* Try pair of module handle / segment number */
918     pModule = (NE_MODULE *) GlobalLock16( HIWORD( proc ) );
919     if ( pModule && pModule->magic == IMAGE_OS2_SIGNATURE )
920     {
921         segNr = LOWORD( proc );
922         if ( segNr && segNr <= pModule->seg_count )
923             pSeg = NE_SEG_TABLE( pModule ) + segNr-1;
924     }
925
926     /* Try thunk or function */
927     else
928     {
929         BYTE *thunk = MapSL( (SEGPTR)proc );
930         WORD selector;
931
932         if ((thunk[0] == 0xb8) && (thunk[3] == 0xea))
933             selector = thunk[6] + (thunk[7] << 8);
934         else
935             selector = HIWORD( proc );
936
937         pModule = NE_GetPtr( GlobalHandle16( selector ) );
938         pSeg = pModule? NE_SEG_TABLE( pModule ) : NULL;
939
940         if ( pModule )
941             for ( segNr = 1; segNr <= pModule->seg_count; segNr++, pSeg++ )
942                 if ( GlobalHandleToSel16(pSeg->hSeg) == selector )
943                     break;
944
945         if ( pModule && segNr > pModule->seg_count )
946             pSeg = NULL;
947     }
948
949     /* Abort if segment not found */
950
951     if ( !pModule || !pSeg )
952         return FALSE;
953
954     /* Return segment data */
955
956     if ( ppModule ) *ppModule = pModule;
957     if ( ppSeg    ) *ppSeg    = pSeg;
958     if ( pSegNr   ) *pSegNr   = segNr;
959
960     return TRUE;
961 }
962
963 /**********************************************************************
964  *          GetCodeHandle    (KERNEL.93)
965  */
966 HANDLE16 WINAPI GetCodeHandle16( FARPROC16 proc )
967 {
968     SEGTABLEENTRY *pSeg;
969
970     if ( !TASK_GetCodeSegment( proc, NULL, &pSeg, NULL ) )
971         return (HANDLE16)0;
972
973     return pSeg->hSeg;
974 }
975
976 /**********************************************************************
977  *          GetCodeInfo    (KERNEL.104)
978  */
979 BOOL16 WINAPI GetCodeInfo16( FARPROC16 proc, SEGINFO *segInfo )
980 {
981     NE_MODULE *pModule;
982     SEGTABLEENTRY *pSeg;
983     int segNr;
984
985     if ( !TASK_GetCodeSegment( proc, &pModule, &pSeg, &segNr ) )
986         return FALSE;
987
988     /* Fill in segment information */
989
990     segInfo->offSegment = pSeg->filepos;
991     segInfo->cbSegment  = pSeg->size;
992     segInfo->flags      = pSeg->flags;
993     segInfo->cbAlloc    = pSeg->minsize;
994     segInfo->h          = pSeg->hSeg;
995     segInfo->alignShift = pModule->alignment;
996
997     if ( segNr == pModule->dgroup )
998         segInfo->cbAlloc += pModule->heap_size + pModule->stack_size;
999
1000     /* Return module handle in %es */
1001
1002     CURRENT_STACK16->es = GlobalHandleToSel16( pModule->self );
1003
1004     return TRUE;
1005 }
1006
1007
1008 /**********************************************************************
1009  *          DefineHandleTable    (KERNEL.94)
1010  */
1011 BOOL16 WINAPI DefineHandleTable16( WORD wOffset )
1012 {
1013     FIXME("(%04x): stub ?\n", wOffset);
1014     return TRUE;
1015 }
1016
1017
1018 /***********************************************************************
1019  *           SetTaskQueue  (KERNEL.34)
1020  */
1021 HQUEUE16 WINAPI SetTaskQueue16( HTASK16 hTask, HQUEUE16 hQueue )
1022 {
1023     FIXME( "stub, should not get called\n" );
1024     return 0xbeef;
1025 }
1026
1027
1028 /***********************************************************************
1029  *           GetTaskQueue  (KERNEL.35)
1030  */
1031 HQUEUE16 WINAPI GetTaskQueue16( HTASK16 hTask )
1032 {
1033     FIXME( "stub, should not get called\n" );
1034     return 0xbeef;
1035 }
1036
1037 /***********************************************************************
1038  *           SetThreadQueue  (KERNEL.463)
1039  */
1040 HQUEUE16 WINAPI SetThreadQueue16( DWORD thread, HQUEUE16 hQueue )
1041 {
1042     FIXME( "stub, should not get called\n" );
1043     return 0xbeef;
1044 }
1045
1046 /***********************************************************************
1047  *           GetThreadQueue  (KERNEL.464)
1048  */
1049 HQUEUE16 WINAPI GetThreadQueue16( DWORD thread )
1050 {
1051     FIXME( "stub, should not get called\n" );
1052     return 0xbeef;
1053 }
1054
1055 /***********************************************************************
1056  *           SetFastQueue  (KERNEL.624)
1057  */
1058 VOID WINAPI SetFastQueue16( DWORD thread, HQUEUE16 hQueue )
1059 {
1060     FIXME( "stub, should not get called\n" );
1061 }
1062
1063 /***********************************************************************
1064  *           GetFastQueue  (KERNEL.625)
1065  */
1066 HQUEUE16 WINAPI GetFastQueue16( void )
1067 {
1068     FIXME( "stub, should not get called\n" );
1069     return 0xbeef;
1070 }
1071
1072 /***********************************************************************
1073  *           SwitchStackTo   (KERNEL.108)
1074  */
1075 void WINAPI SwitchStackTo16( WORD seg, WORD ptr, WORD top )
1076 {
1077     STACK16FRAME *oldFrame, *newFrame;
1078     INSTANCEDATA *pData;
1079     UINT16 copySize;
1080
1081     if (!(pData = (INSTANCEDATA *)GlobalLock16( seg ))) return;
1082     TRACE("old=%04x:%04x new=%04x:%04x\n",
1083           SELECTOROF( NtCurrentTeb()->WOW32Reserved ),
1084           OFFSETOF( NtCurrentTeb()->WOW32Reserved ), seg, ptr );
1085
1086     /* Save the old stack */
1087
1088     oldFrame = CURRENT_STACK16;
1089     /* pop frame + args and push bp */
1090     pData->old_ss_sp   = (SEGPTR)NtCurrentTeb()->WOW32Reserved + sizeof(STACK16FRAME)
1091                            + 2 * sizeof(WORD);
1092     *(WORD *)MapSL(pData->old_ss_sp) = oldFrame->bp;
1093     pData->stacktop    = top;
1094     pData->stackmin    = ptr;
1095     pData->stackbottom = ptr;
1096
1097     /* Switch to the new stack */
1098
1099     /* Note: we need to take the 3 arguments into account; otherwise,
1100      * the stack will underflow upon return from this function.
1101      */
1102     copySize = oldFrame->bp - OFFSETOF(pData->old_ss_sp);
1103     copySize += 3 * sizeof(WORD) + sizeof(STACK16FRAME);
1104     NtCurrentTeb()->WOW32Reserved = (void *)MAKESEGPTR( seg, ptr - copySize );
1105     newFrame = CURRENT_STACK16;
1106
1107     /* Copy the stack frame and the local variables to the new stack */
1108
1109     memmove( newFrame, oldFrame, copySize );
1110     newFrame->bp = ptr;
1111     *(WORD *)MapSL( MAKESEGPTR( seg, ptr ) ) = 0;  /* clear previous bp */
1112 }
1113
1114
1115 /***********************************************************************
1116  *           SwitchStackBack   (KERNEL.109)
1117  */
1118 void WINAPI SwitchStackBack16( CONTEXT86 *context )
1119 {
1120     STACK16FRAME *oldFrame, *newFrame;
1121     INSTANCEDATA *pData;
1122
1123     if (!(pData = (INSTANCEDATA *)GlobalLock16(SELECTOROF(NtCurrentTeb()->WOW32Reserved))))
1124         return;
1125     if (!pData->old_ss_sp)
1126     {
1127         WARN("No previous SwitchStackTo\n" );
1128         return;
1129     }
1130     TRACE("restoring stack %04x:%04x\n",
1131           SELECTOROF(pData->old_ss_sp), OFFSETOF(pData->old_ss_sp) );
1132
1133     oldFrame = CURRENT_STACK16;
1134
1135     /* Pop bp from the previous stack */
1136
1137     context->Ebp = (context->Ebp & ~0xffff) | *(WORD *)MapSL(pData->old_ss_sp);
1138     pData->old_ss_sp += sizeof(WORD);
1139
1140     /* Switch back to the old stack */
1141
1142     NtCurrentTeb()->WOW32Reserved = (void *)(pData->old_ss_sp - sizeof(STACK16FRAME));
1143     context->SegSs = SELECTOROF(pData->old_ss_sp);
1144     context->Esp   = OFFSETOF(pData->old_ss_sp) - sizeof(DWORD); /*ret addr*/
1145     pData->old_ss_sp = 0;
1146
1147     /* Build a stack frame for the return */
1148
1149     newFrame = CURRENT_STACK16;
1150     newFrame->frame32     = oldFrame->frame32;
1151     newFrame->module_cs   = oldFrame->module_cs;
1152     newFrame->callfrom_ip = oldFrame->callfrom_ip;
1153     newFrame->entry_ip    = oldFrame->entry_ip;
1154 }
1155
1156
1157 /***********************************************************************
1158  *           GetTaskQueueDS  (KERNEL.118)
1159  */
1160 void WINAPI GetTaskQueueDS16(void)
1161 {
1162     CURRENT_STACK16->ds = GlobalHandleToSel16( GetTaskQueue16(0) );
1163 }
1164
1165
1166 /***********************************************************************
1167  *           GetTaskQueueES  (KERNEL.119)
1168  */
1169 void WINAPI GetTaskQueueES16(void)
1170 {
1171     CURRENT_STACK16->es = GlobalHandleToSel16( GetTaskQueue16(0) );
1172 }
1173
1174
1175 /***********************************************************************
1176  *           GetCurrentTask   (KERNEL32.@)
1177  */
1178 HTASK16 WINAPI GetCurrentTask(void)
1179 {
1180     return NtCurrentTeb()->htask16;
1181 }
1182
1183 /***********************************************************************
1184  *           GetCurrentTask   (KERNEL.36)
1185  */
1186 DWORD WINAPI WIN16_GetCurrentTask(void)
1187 {
1188     /* This is the version used by relay code; the first task is */
1189     /* returned in the high word of the result */
1190     return MAKELONG( GetCurrentTask(), hFirstTask );
1191 }
1192
1193
1194 /***********************************************************************
1195  *           GetCurrentPDB   (KERNEL.37)
1196  *
1197  * UNDOC: returns PSP of KERNEL in high word
1198  */
1199 DWORD WINAPI GetCurrentPDB16(void)
1200 {
1201     TDB *pTask;
1202
1203     if (!(pTask = TASK_GetCurrent())) return 0;
1204     return MAKELONG(pTask->hPDB, 0); /* FIXME */
1205 }
1206
1207
1208 /***********************************************************************
1209  *           GetCurPID   (KERNEL.157)
1210  */
1211 DWORD WINAPI GetCurPID16( DWORD unused )
1212 {
1213     return 0;
1214 }
1215
1216
1217 /***********************************************************************
1218  *           GetInstanceData   (KERNEL.54)
1219  */
1220 INT16 WINAPI GetInstanceData16( HINSTANCE16 instance, WORD buffer, INT16 len )
1221 {
1222     char *ptr = (char *)GlobalLock16( instance );
1223     if (!ptr || !len) return 0;
1224     if ((int)buffer + len >= 0x10000) len = 0x10000 - buffer;
1225     memcpy( (char *)GlobalLock16(CURRENT_DS) + buffer, ptr + buffer, len );
1226     return len;
1227 }
1228
1229
1230 /***********************************************************************
1231  *           GetExeVersion   (KERNEL.105)
1232  */
1233 WORD WINAPI GetExeVersion16(void)
1234 {
1235     TDB *pTask;
1236
1237     if (!(pTask = TASK_GetCurrent())) return 0;
1238     return pTask->version;
1239 }
1240
1241
1242 /***********************************************************************
1243  *           SetErrorMode   (KERNEL.107)
1244  */
1245 UINT16 WINAPI SetErrorMode16( UINT16 mode )
1246 {
1247     TDB *pTask;
1248     if (!(pTask = TASK_GetCurrent())) return 0;
1249     pTask->error_mode = mode;
1250     return SetErrorMode( mode );
1251 }
1252
1253
1254 /***********************************************************************
1255  *           GetNumTasks   (KERNEL.152)
1256  */
1257 UINT16 WINAPI GetNumTasks16(void)
1258 {
1259     return nTaskCount;
1260 }
1261
1262
1263 /***********************************************************************
1264  *           GetTaskDS   (KERNEL.155)
1265  *
1266  * Note: this function apparently returns a DWORD with LOWORD == HIWORD.
1267  * I don't think we need to bother with this.
1268  */
1269 HINSTANCE16 WINAPI GetTaskDS16(void)
1270 {
1271     TDB *pTask;
1272
1273     if (!(pTask = TASK_GetCurrent())) return 0;
1274     return GlobalHandleToSel16(pTask->hInstance);
1275 }
1276
1277 /***********************************************************************
1278  *           GetDummyModuleHandleDS   (KERNEL.602)
1279  */
1280 WORD WINAPI GetDummyModuleHandleDS16(void)
1281 {
1282     TDB *pTask;
1283     WORD selector;
1284
1285     if (!(pTask = TASK_GetCurrent())) return 0;
1286     if (!(pTask->flags & TDBF_WIN32)) return 0;
1287     selector = GlobalHandleToSel16( pTask->hModule );
1288     CURRENT_DS = selector;
1289     return selector;
1290 }
1291
1292 /***********************************************************************
1293  *           IsTask   (KERNEL.320)
1294  */
1295 BOOL16 WINAPI IsTask16( HTASK16 hTask )
1296 {
1297     TDB *pTask;
1298
1299     if (!(pTask = TASK_GetPtr( hTask ))) return FALSE;
1300     if (GlobalSize16( hTask ) < sizeof(TDB)) return FALSE;
1301     return (pTask->magic == TDB_MAGIC);
1302 }
1303
1304
1305 /***********************************************************************
1306  *           IsWinOldApTask   (KERNEL.158)
1307  */
1308 BOOL16 WINAPI IsWinOldApTask16( HTASK16 hTask )
1309 {
1310     /* should return bit 0 of byte 0x48 in PSP */
1311     return FALSE;
1312 }
1313
1314 /***********************************************************************
1315  *           SetTaskSignalProc   (KERNEL.38)
1316  */
1317 FARPROC16 WINAPI SetTaskSignalProc( HTASK16 hTask, FARPROC16 proc )
1318 {
1319     TDB *pTask;
1320     FARPROC16 oldProc;
1321
1322     if (!hTask) hTask = GetCurrentTask();
1323     if (!(pTask = TASK_GetPtr( hTask ))) return NULL;
1324     oldProc = pTask->userhandler;
1325     pTask->userhandler = proc;
1326     return oldProc;
1327 }
1328
1329 /***********************************************************************
1330  *           SetSigHandler   (KERNEL.140)
1331  */
1332 WORD WINAPI SetSigHandler16( FARPROC16 newhandler, FARPROC16* oldhandler,
1333                            UINT16 *oldmode, UINT16 newmode, UINT16 flag )
1334 {
1335     FIXME("(%p,%p,%p,%d,%d), unimplemented.\n",
1336           newhandler,oldhandler,oldmode,newmode,flag );
1337
1338     if (flag != 1) return 0;
1339     if (!newmode) newhandler = NULL;  /* Default handler */
1340     if (newmode != 4)
1341     {
1342         TDB *pTask;
1343
1344         if (!(pTask = TASK_GetCurrent())) return 0;
1345         if (oldmode) *oldmode = pTask->signal_flags;
1346         pTask->signal_flags = newmode;
1347         if (oldhandler) *oldhandler = pTask->sighandler;
1348         pTask->sighandler = newhandler;
1349     }
1350     return 0;
1351 }
1352
1353
1354 /***********************************************************************
1355  *           GlobalNotify   (KERNEL.154)
1356  *
1357  * Note that GlobalNotify does _not_ return the old NotifyProc
1358  * -- contrary to LocalNotify !!
1359  */
1360 VOID WINAPI GlobalNotify16( FARPROC16 proc )
1361 {
1362     TDB *pTask;
1363
1364     if (!(pTask = TASK_GetCurrent())) return;
1365     pTask->discardhandler = proc;
1366 }
1367
1368
1369 /***********************************************************************
1370  *           GetExePtrHelper
1371  */
1372 static inline HMODULE16 GetExePtrHelper( HANDLE16 handle, HTASK16 *hTask )
1373 {
1374     char *ptr;
1375     HANDLE16 owner;
1376
1377       /* Check for module handle */
1378
1379     if (!(ptr = GlobalLock16( handle ))) return 0;
1380     if (((NE_MODULE *)ptr)->magic == IMAGE_OS2_SIGNATURE) return handle;
1381
1382       /* Search for this handle inside all tasks */
1383
1384     *hTask = hFirstTask;
1385     while (*hTask)
1386     {
1387         TDB *pTask = TASK_GetPtr( *hTask );
1388         if ((*hTask == handle) ||
1389             (pTask->hInstance == handle) ||
1390             (pTask->hQueue == handle) ||
1391             (pTask->hPDB == handle)) return pTask->hModule;
1392         *hTask = pTask->hNext;
1393     }
1394
1395       /* Check the owner for module handle */
1396
1397     owner = FarGetOwner16( handle );
1398     if (!(ptr = GlobalLock16( owner ))) return 0;
1399     if (((NE_MODULE *)ptr)->magic == IMAGE_OS2_SIGNATURE) return owner;
1400
1401       /* Search for the owner inside all tasks */
1402
1403     *hTask = hFirstTask;
1404     while (*hTask)
1405     {
1406         TDB *pTask = TASK_GetPtr( *hTask );
1407         if ((*hTask == owner) ||
1408             (pTask->hInstance == owner) ||
1409             (pTask->hQueue == owner) ||
1410             (pTask->hPDB == owner)) return pTask->hModule;
1411         *hTask = pTask->hNext;
1412     }
1413
1414     return 0;
1415 }
1416
1417 /***********************************************************************
1418  *           GetExePtr   (KERNEL.133)
1419  */
1420 HMODULE16 WINAPI WIN16_GetExePtr( HANDLE16 handle )
1421 {
1422     HTASK16 hTask = 0;
1423     HMODULE16 hModule = GetExePtrHelper( handle, &hTask );
1424     STACK16FRAME *frame = CURRENT_STACK16;
1425     frame->ecx = hModule;
1426     if (hTask) frame->es = hTask;
1427     return hModule;
1428 }
1429
1430
1431 /***********************************************************************
1432  *           K228   (KERNEL.228)
1433  */
1434 HMODULE16 WINAPI GetExePtr( HANDLE16 handle )
1435 {
1436     HTASK16 hTask = 0;
1437     return GetExePtrHelper( handle, &hTask );
1438 }
1439
1440
1441 /***********************************************************************
1442  *           TaskFirst   (TOOLHELP.63)
1443  */
1444 BOOL16 WINAPI TaskFirst16( TASKENTRY *lpte )
1445 {
1446     lpte->hNext = hFirstTask;
1447     return TaskNext16( lpte );
1448 }
1449
1450
1451 /***********************************************************************
1452  *           TaskNext   (TOOLHELP.64)
1453  */
1454 BOOL16 WINAPI TaskNext16( TASKENTRY *lpte )
1455 {
1456     TDB *pTask;
1457     INSTANCEDATA *pInstData;
1458
1459     TRACE_(toolhelp)("(%p): task=%04x\n", lpte, lpte->hNext );
1460     if (!lpte->hNext) return FALSE;
1461
1462     /* make sure that task and hInstance are valid (skip initial Wine task !) */
1463     while (1) {
1464         pTask = TASK_GetPtr( lpte->hNext );
1465         if (!pTask || pTask->magic != TDB_MAGIC) return FALSE;
1466         if (pTask->hInstance)
1467             break;
1468         lpte->hNext = pTask->hNext;
1469     }
1470     pInstData = MapSL( MAKESEGPTR( GlobalHandleToSel16(pTask->hInstance), 0 ) );
1471     lpte->hTask         = lpte->hNext;
1472     lpte->hTaskParent   = pTask->hParent;
1473     lpte->hInst         = pTask->hInstance;
1474     lpte->hModule       = pTask->hModule;
1475     lpte->wSS           = SELECTOROF( pTask->teb->WOW32Reserved );
1476     lpte->wSP           = OFFSETOF( pTask->teb->WOW32Reserved );
1477     lpte->wStackTop     = pInstData->stacktop;
1478     lpte->wStackMinimum = pInstData->stackmin;
1479     lpte->wStackBottom  = pInstData->stackbottom;
1480     lpte->wcEvents      = pTask->nEvents;
1481     lpte->hQueue        = pTask->hQueue;
1482     lstrcpynA( lpte->szModule, pTask->module_name, sizeof(lpte->szModule) );
1483     lpte->wPSPOffset    = 0x100;  /*??*/
1484     lpte->hNext         = pTask->hNext;
1485     return TRUE;
1486 }
1487
1488
1489 /***********************************************************************
1490  *           TaskFindHandle   (TOOLHELP.65)
1491  */
1492 BOOL16 WINAPI TaskFindHandle16( TASKENTRY *lpte, HTASK16 hTask )
1493 {
1494     lpte->hNext = hTask;
1495     return TaskNext16( lpte );
1496 }
1497
1498
1499 typedef INT (WINAPI *MessageBoxA_funcptr)(HWND hWnd, LPCSTR text, LPCSTR title, UINT type);
1500
1501 /**************************************************************************
1502  *           FatalAppExit   (KERNEL.137)
1503  */
1504 void WINAPI FatalAppExit16( UINT16 action, LPCSTR str )
1505 {
1506     TDB *pTask = TASK_GetCurrent();
1507
1508     if (!pTask || !(pTask->error_mode & SEM_NOGPFAULTERRORBOX))
1509     {
1510         HMODULE mod = GetModuleHandleA( "user32.dll" );
1511         if (mod)
1512         {
1513             MessageBoxA_funcptr pMessageBoxA = (MessageBoxA_funcptr)GetProcAddress( mod, "MessageBoxA" );
1514             if (pMessageBoxA)
1515             {
1516                 pMessageBoxA( 0, str, NULL, MB_SYSTEMMODAL | MB_OK );
1517                 goto done;
1518             }
1519         }
1520         ERR( "%s\n", debugstr_a(str) );
1521     }
1522  done:
1523     ExitThread(0xff);
1524 }
1525
1526
1527 /***********************************************************************
1528  *           TerminateApp   (TOOLHELP.77)
1529  *
1530  * See "Undocumented Windows".
1531  */
1532 void WINAPI TerminateApp16(HTASK16 hTask, WORD wFlags)
1533 {
1534     if (hTask && hTask != GetCurrentTask())
1535     {
1536         FIXME("cannot terminate task %x\n", hTask);
1537         return;
1538     }
1539
1540     if (wFlags & NO_UAE_BOX)
1541     {
1542         UINT16 old_mode;
1543         old_mode = SetErrorMode16(0);
1544         SetErrorMode16(old_mode|SEM_NOGPFAULTERRORBOX);
1545     }
1546     FatalAppExit16( 0, NULL );
1547
1548     /* hmm, we're still alive ?? */
1549
1550     /* check undocumented flag */
1551     if (!(wFlags & 0x8000))
1552         TASK_CallTaskSignalProc( USIG16_TERMINATION, hTask );
1553
1554     /* UndocWin says to call int 0x21/0x4c exit=0xff here,
1555        but let's just call ExitThread */
1556     ExitThread(0xff);
1557 }
1558
1559
1560 /***********************************************************************
1561  *           GetAppCompatFlags   (KERNEL.354)
1562  */
1563 DWORD WINAPI GetAppCompatFlags16( HTASK16 hTask )
1564 {
1565     TDB *pTask;
1566
1567     if (!hTask) hTask = GetCurrentTask();
1568     if (!(pTask=TASK_GetPtr( hTask ))) return 0;
1569     if (GlobalSize16(hTask) < sizeof(TDB)) return 0;
1570     return pTask->compat_flags;
1571 }
1572
1573
1574 /***********************************************************************
1575  *           GetDOSEnvironment     (KERNEL.131)
1576  *
1577  * Note: the environment is allocated once, it doesn't track changes
1578  * made using the Win32 API. This shouldn't matter.
1579  *
1580  * Format of a 16-bit environment block:
1581  * ASCIIZ   string 1 (xx=yy format)
1582  * ...
1583  * ASCIIZ   string n
1584  * BYTE     0
1585  * WORD     1
1586  * ASCIIZ   program name (e.g. C:\WINDOWS\SYSTEM\KRNL386.EXE)
1587  */
1588 SEGPTR WINAPI GetDOSEnvironment16(void)
1589 {
1590     static const char ENV_program_name[] = "C:\\WINDOWS\\SYSTEM\\KRNL386.EXE";
1591     static HGLOBAL16 handle;  /* handle to the 16 bit environment */
1592
1593     if (!handle)
1594     {
1595         DWORD size;
1596         LPSTR p, env;
1597
1598         p = env = GetEnvironmentStringsA();
1599         while (*p) p += strlen(p) + 1;
1600         p++;  /* skip last null */
1601         size = (p - env) + sizeof(WORD) + sizeof(ENV_program_name);
1602         handle = GlobalAlloc16( GMEM_FIXED, size );
1603         if (handle)
1604         {
1605             WORD one = 1;
1606             LPSTR env16 = GlobalLock16( handle );
1607             memcpy( env16, env, p - env );
1608             memcpy( env16 + (p - env), &one, sizeof(one));
1609             memcpy( env16 + (p - env) + sizeof(WORD), ENV_program_name, sizeof(ENV_program_name));
1610             GlobalUnlock16( handle );
1611         }
1612         FreeEnvironmentStringsA( env );
1613     }
1614     return K32WOWGlobalLock16( handle );
1615 }