loader/task.c

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