Use lcall opcodes, so all gas-es understand it...
[wine] / msdos / dpmi.c
1 /*
2  * DPMI 0.9 emulation
3  *
4  * Copyright 1995 Alexandre Julliard
5  */
6
7 #include <unistd.h>
8 #include <string.h>
9 #include "windef.h"
10 #include "wine/winbase16.h"
11 #include "ldt.h"
12 #include "global.h"
13 #include "module.h"
14 #include "miscemu.h"
15 #include "msdos.h"
16 #include "task.h"
17 #include "thread.h"     /* for !MZ_SUPPORTED */
18 #include "stackframe.h" /* for !MZ_SUPPORTED */
19 #include "toolhelp.h"
20 #include "selectors.h"
21 #include "process.h"
22 #include "callback.h"
23 #include "debugtools.h"
24
25 DEFAULT_DEBUG_CHANNEL(int31);
26
27 #define DOS_GET_DRIVE(reg) ((reg) ? (reg) - 1 : DRIVE_GetCurrentDrive())
28
29 void CreateBPB(int drive, BYTE *data, BOOL16 limited);  /* defined in int21.c */
30
31 static void* lastvalloced = NULL;
32
33 /* Structure for real-mode callbacks */
34 typedef struct
35 {
36     DWORD edi;
37     DWORD esi;
38     DWORD ebp;
39     DWORD reserved;
40     DWORD ebx;
41     DWORD edx;
42     DWORD ecx;
43     DWORD eax;
44     WORD  fl;
45     WORD  es;
46     WORD  ds;
47     WORD  fs;
48     WORD  gs;
49     WORD  ip;
50     WORD  cs;
51     WORD  sp;
52     WORD  ss;
53 } REALMODECALL;
54
55
56
57 typedef struct tagRMCB {
58     DWORD address;
59     DWORD proc_ofs,proc_sel;
60     DWORD regs_ofs,regs_sel;
61     struct tagRMCB *next;
62 } RMCB;
63
64 static RMCB *FirstRMCB = NULL;
65
66 UINT16 DPMI_wrap_seg;
67
68 /**********************************************************************
69  *          DPMI_xalloc
70  * special virtualalloc, allocates lineary monoton growing memory.
71  * (the usual VirtualAlloc does not satisfy that restriction)
72  */
73 static LPVOID
74 DPMI_xalloc(int len) {
75         LPVOID  ret;
76         LPVOID  oldlastv = lastvalloced;
77
78         if (lastvalloced) {
79                 int     xflag = 0;
80                 ret = NULL;
81                 while (!ret) {
82                         ret=VirtualAlloc(lastvalloced,len,MEM_COMMIT|MEM_RESERVE,PAGE_EXECUTE_READWRITE);
83                         if (!ret)
84                                 lastvalloced = (char *) lastvalloced + 0x10000;
85                         /* we failed to allocate one in the first round. 
86                          * try non-linear
87                          */
88                         if (!xflag && (lastvalloced<oldlastv)) { /* wrapped */
89                                 FIXME("failed to allocate lineary growing memory (%d bytes), using non-linear growing...\n",len);
90                                 xflag++;
91                         }
92                         /* if we even fail to allocate something in the next 
93                          * round, return NULL
94                          */
95                         if ((xflag==1) && (lastvalloced >= oldlastv))
96                                 xflag++;
97                         if ((xflag==2) && (lastvalloced < oldlastv)) {
98                                 FIXME("failed to allocate any memory of %d bytes!\n",len);
99                                 return NULL;
100                         }
101                 }
102         } else
103                  ret=VirtualAlloc(NULL,len,MEM_COMMIT|MEM_RESERVE,PAGE_EXECUTE_READWRITE);
104         lastvalloced = (LPVOID)(((DWORD)ret+len+0xffff)&~0xffff);
105         return ret;
106 }
107
108 static void
109 DPMI_xfree(LPVOID ptr) {
110         VirtualFree(ptr,0,MEM_RELEASE);
111 }
112
113 /* FIXME: perhaps we could grow this mapped area... */
114 static LPVOID
115 DPMI_xrealloc(LPVOID ptr,int newsize) {
116         MEMORY_BASIC_INFORMATION        mbi;
117         LPVOID                          newptr;
118
119         newptr = DPMI_xalloc(newsize);
120         if (ptr) {
121                 if (!VirtualQuery(ptr,&mbi,sizeof(mbi))) {
122                         FIXME("realloc of DPMI_xallocd region %p?\n",ptr);
123                         return NULL;
124                 }
125                 if (mbi.State == MEM_FREE) {
126                         FIXME("realloc of DPMI_xallocd region %p?\n",ptr);
127                         return NULL;
128                 }
129                 /* We do not shrink allocated memory. most reallocs
130                  * only do grows anyway
131                  */
132                 if (newsize<=mbi.RegionSize)
133                         return ptr;
134                 memcpy(newptr,ptr,mbi.RegionSize);
135                 DPMI_xfree(ptr);
136         }
137         return newptr;
138 }
139 /**********************************************************************
140  *          INT_GetRealModeContext
141  */
142 static void INT_GetRealModeContext( REALMODECALL *call, CONTEXT86 *context )
143 {
144     EAX_reg(context) = call->eax;
145     EBX_reg(context) = call->ebx;
146     ECX_reg(context) = call->ecx;
147     EDX_reg(context) = call->edx;
148     ESI_reg(context) = call->esi;
149     EDI_reg(context) = call->edi;
150     EBP_reg(context) = call->ebp;
151     EFL_reg(context) = call->fl | V86_FLAG;
152     EIP_reg(context) = call->ip;
153     ESP_reg(context) = call->sp;
154     CS_reg(context)  = call->cs;
155     DS_reg(context)  = call->ds;
156     ES_reg(context)  = call->es;
157     FS_reg(context)  = call->fs;
158     GS_reg(context)  = call->gs;
159     SS_reg(context)  = call->ss;
160 }
161
162
163 /**********************************************************************
164  *          INT_SetRealModeContext
165  */
166 static void INT_SetRealModeContext( REALMODECALL *call, CONTEXT86 *context )
167 {
168     call->eax = EAX_reg(context);
169     call->ebx = EBX_reg(context);
170     call->ecx = ECX_reg(context);
171     call->edx = EDX_reg(context);
172     call->esi = ESI_reg(context);
173     call->edi = EDI_reg(context);
174     call->ebp = EBP_reg(context);
175     call->fl  = LOWORD(EFL_reg(context));
176     call->ip  = LOWORD(EIP_reg(context));
177     call->sp  = LOWORD(ESP_reg(context));
178     call->cs  = CS_reg(context);
179     call->ds  = DS_reg(context);
180     call->es  = ES_reg(context);
181     call->fs  = FS_reg(context);
182     call->gs  = GS_reg(context);
183     call->ss  = SS_reg(context);
184 }
185
186
187 /**********************************************************************
188  *          DPMI_CallRMCBProc
189  *
190  * This routine does the hard work of calling a callback procedure.
191  */
192 static void DPMI_CallRMCBProc( CONTEXT86 *context, RMCB *rmcb, WORD flag )
193 {
194     if (IS_SELECTOR_SYSTEM( rmcb->proc_sel )) {
195         /* Wine-internal RMCB, call directly */
196         ((RMCBPROC)rmcb->proc_ofs)(context);
197     } else {
198 #ifdef __i386__
199         UINT16 ss,es;
200         DWORD edi;
201
202         INT_SetRealModeContext((REALMODECALL *)PTR_SEG_OFF_TO_LIN( rmcb->regs_sel, rmcb->regs_ofs ), context);
203         ss = SELECTOR_AllocBlock( DOSMEM_MemoryBase(0) + (DWORD)(SS_reg(context)<<4), 0x10000, SEGMENT_DATA, FALSE, FALSE );
204
205         FIXME("untested!\n");
206
207         /* The called proc ends with an IRET, and takes these parameters:
208          * DS:ESI = pointer to real-mode SS:SP
209          * ES:EDI = pointer to real-mode call structure
210          * It returns:
211          * ES:EDI = pointer to real-mode call structure (may be a copy)
212          * It is the proc's responsibility to change the return CS:IP in the
213          * real-mode call structure. */
214         if (flag & 1) {
215             /* 32-bit DPMI client */
216             int _clobber;
217             __asm__ __volatile__(
218                  "pushl %%ebp\n"
219                  "pushl %%es\n"
220                  "pushl %%ds\n"
221                  "pushfl\n"
222                  "mov %7,%%es\n"
223                  "mov %5,%%ds\n"
224                  ".byte 0x36, 0xff, 0x18\n" /* lcall *%ss:(%eax) */
225                  "popl %%ds\n"
226                  "mov %%es,%0\n"
227                  "popl %%es\n"
228                  "popl %%ebp\n"
229              : "=d" (es), "=D" (edi), "=S" (_clobber), "=a" (_clobber), "=c" (_clobber)
230              : "0" (ss), "2" (ESP_reg(context)),
231                "4" (rmcb->regs_sel), "1" (rmcb->regs_ofs),
232                "3" (&rmcb->proc_ofs)
233              : "ebx" );
234         } else {
235             /* 16-bit DPMI client */
236             CONTEXT86 ctx = *context;
237             CS_reg(&ctx) = rmcb->proc_sel;
238             EIP_reg(&ctx) = rmcb->proc_ofs;
239             DS_reg(&ctx) = ss;
240             ESI_reg(&ctx) = ESP_reg(context);
241             ES_reg(&ctx) = rmcb->regs_sel;
242             EDI_reg(&ctx) = rmcb->regs_ofs;
243             /* FIXME: I'm pretty sure this isn't right */
244             Callbacks->CallRegisterShortProc(&ctx, 2);
245             es = ES_reg(&ctx);
246             edi = EDI_reg(&ctx);
247         }
248         SELECTOR_FreeBlock(ss, 1);
249         INT_GetRealModeContext((REALMODECALL*)PTR_SEG_OFF_TO_LIN( es, edi ), context);
250 #else
251         ERR("RMCBs only implemented for i386\n");
252 #endif
253     }
254 }
255
256
257 /**********************************************************************
258  *          DPMI_CallRMProc
259  *
260  * This routine does the hard work of calling a real mode procedure.
261  */
262 int DPMI_CallRMProc( CONTEXT86 *context, LPWORD stack, int args, int iret )
263 {
264     LPWORD stack16;
265     LPVOID addr = NULL; /* avoid gcc warning */
266     TDB *pTask = (TDB *)GlobalLock16( GetCurrentTask() );
267     NE_MODULE *pModule = pTask ? NE_GetPtr( pTask->hModule ) : NULL;
268     RMCB *CurrRMCB;
269     int alloc = 0, already = 0;
270     BYTE *code;
271
272     GlobalUnlock16( GetCurrentTask() );
273
274     TRACE("EAX=%08lx EBX=%08lx ECX=%08lx EDX=%08lx\n",
275                  EAX_reg(context), EBX_reg(context), ECX_reg(context), EDX_reg(context) );
276     TRACE("ESI=%08lx EDI=%08lx ES=%04lx DS=%04lx CS:IP=%04lx:%04x, %d WORD arguments, %s\n",
277                  ESI_reg(context), EDI_reg(context), ES_reg(context), DS_reg(context),
278                  CS_reg(context), LOWORD(EIP_reg(context)), args, iret?"IRET":"FAR" );
279
280 callrmproc_again:
281
282 /* there might be some code that just jumps to RMCBs or the like,
283    in which case following the jumps here might get us to a shortcut */
284     code = CTX_SEG_OFF_TO_LIN(context, CS_reg(context), EIP_reg(context));
285     switch (*code) {
286     case 0xe9: /* JMP NEAR */
287       EIP_reg(context) += 3 + *(WORD *)(code+1);
288       /* yeah, I know these gotos don't look good... */
289       goto callrmproc_again;
290     case 0xea: /* JMP FAR */
291       EIP_reg(context) = *(WORD *)(code+1);
292       CS_reg(context) = *(WORD *)(code+3);
293       /* ...but since the label is there anyway... */
294       goto callrmproc_again;
295     case 0xeb: /* JMP SHORT */
296       EIP_reg(context) += 2 + *(signed char *)(code+1);
297       /* ...because of other gotos below, so... */
298       goto callrmproc_again;
299     }
300
301 /* shortcut for chaining to internal interrupt handlers */
302     if ((CS_reg(context) == 0xF000) && iret) {
303         return INT_RealModeInterrupt( LOWORD(EIP_reg(context))/4, context);
304     }
305
306 /* shortcut for RMCBs */
307     CurrRMCB = FirstRMCB;
308
309     while (CurrRMCB && (HIWORD(CurrRMCB->address) != CS_reg(context)))
310         CurrRMCB = CurrRMCB->next;
311
312     if (!(CurrRMCB || pModule->lpDosTask)) {
313 #ifdef MZ_SUPPORTED
314         FIXME("DPMI real-mode call using DOS VM task system, not fully tested!\n");
315         TRACE("creating VM86 task\n");
316         if (!MZ_InitTask( MZ_AllocDPMITask( pModule->self ) )) {
317             ERR("could not setup VM86 task\n");
318             return 1;
319         }
320 #else
321         ERR("Actual real-mode calls not supported on this architecture!\n");
322         return 1;
323 #endif
324     }
325     if (!already) {
326         if (!SS_reg(context)) {
327             alloc = 1; /* allocate default stack */
328             stack16 = addr = DOSMEM_GetBlock( pModule->self, 64, (UINT16 *)&(SS_reg(context)) );
329             ESP_reg(context) = 64-2;
330             stack16 += 32-1;
331             if (!addr) {
332                 ERR("could not allocate default stack\n");
333                 return 1;
334             }
335         } else {
336             stack16 = CTX_SEG_OFF_TO_LIN(context, SS_reg(context), ESP_reg(context));
337         }
338         ESP_reg(context) -= (args + (iret?1:0)) * sizeof(WORD);
339         stack16 -= args;
340         if (args) memcpy(stack16, stack, args*sizeof(WORD) );
341         /* push flags if iret */
342         if (iret) {
343             stack16--; args++;
344             *stack16 = LOWORD(EFL_reg(context));
345         }
346         /* push return address (return to interrupt wrapper) */
347         *(--stack16) = DPMI_wrap_seg;
348         *(--stack16) = 0;
349         /* adjust stack */
350         ESP_reg(context) -= 2*sizeof(WORD);
351         already = 1;
352     }
353
354     if (CurrRMCB) {
355         /* RMCB call, invoke protected-mode handler directly */
356         DPMI_CallRMCBProc(context, CurrRMCB, pModule->lpDosTask?pModule->lpDosTask->dpmi_flag:0);
357         /* check if we returned to where we thought we would */
358         if ((CS_reg(context) != DPMI_wrap_seg) ||
359             (LOWORD(EIP_reg(context)) != 0)) {
360             /* we need to continue at different address in real-mode space,
361                so we need to set it all up for real mode again */
362             goto callrmproc_again;
363         }
364     } else {
365 #ifdef MZ_SUPPORTED
366         TRACE("entering real mode...\n");
367         DOSVM_Enter( context );
368         TRACE("returned from real-mode call\n");
369 #else
370         /* we should never get here, but... */
371         ERR("cannot perform real-mode call\n");
372 #endif
373     }
374     if (alloc) DOSMEM_FreeBlock( pModule->self, addr );
375     return 0;
376 }
377
378
379 /**********************************************************************
380  *          CallRMInt
381  */
382 static void CallRMInt( CONTEXT86 *context )
383 {
384     CONTEXT86 realmode_ctx;
385     FARPROC16 rm_int = INT_GetRMHandler( BL_reg(context) );
386     REALMODECALL *call = (REALMODECALL *)PTR_SEG_OFF_TO_LIN( ES_reg(context),
387                                                           DI_reg(context) );
388     INT_GetRealModeContext( call, &realmode_ctx );
389
390     /* we need to check if a real-mode program has hooked the interrupt */
391     if (HIWORD(rm_int)!=0xF000) {
392         /* yup, which means we need to switch to real mode... */
393         CS_reg(&realmode_ctx) = HIWORD(rm_int);
394         EIP_reg(&realmode_ctx) = LOWORD(rm_int);
395         if (DPMI_CallRMProc( &realmode_ctx, NULL, 0, TRUE))
396           SET_CFLAG(context);
397     } else {
398         RESET_CFLAG(context);
399         /* use the IP we have instead of BL_reg, in case some apps
400            decide to move interrupts around for whatever reason... */
401         if (INT_RealModeInterrupt( LOWORD(rm_int)/4, &realmode_ctx ))
402           SET_CFLAG(context);
403         if (EFL_reg(context)&1) {
404           FIXME("%02x: EAX=%08lx EBX=%08lx ECX=%08lx EDX=%08lx\n",
405                 BL_reg(context), EAX_reg(&realmode_ctx), EBX_reg(&realmode_ctx), 
406                 ECX_reg(&realmode_ctx), EDX_reg(&realmode_ctx));
407           FIXME("      ESI=%08lx EDI=%08lx DS=%04lx ES=%04lx\n",
408                 ESI_reg(&realmode_ctx), EDI_reg(&realmode_ctx), 
409                 DS_reg(&realmode_ctx), ES_reg(&realmode_ctx) );
410         }
411     }
412     INT_SetRealModeContext( call, &realmode_ctx );
413 }
414
415
416 static void CallRMProc( CONTEXT86 *context, int iret )
417 {
418     REALMODECALL *p = (REALMODECALL *)PTR_SEG_OFF_TO_LIN( ES_reg(context), DI_reg(context) );
419     CONTEXT86 context16;
420
421     TRACE("RealModeCall: EAX=%08lx EBX=%08lx ECX=%08lx EDX=%08lx\n",
422         p->eax, p->ebx, p->ecx, p->edx);
423     TRACE("              ESI=%08lx EDI=%08lx ES=%04x DS=%04x CS:IP=%04x:%04x, %d WORD arguments, %s\n",
424         p->esi, p->edi, p->es, p->ds, p->cs, p->ip, CX_reg(context), iret?"IRET":"FAR" );
425
426     if (!(p->cs) && !(p->ip)) { /* remove this check
427                                    if Int21/6501 case map function
428                                    has been implemented */
429         SET_CFLAG(context);
430         return;
431     }
432     INT_GetRealModeContext(p, &context16);
433     DPMI_CallRMProc( &context16, ((LPWORD)PTR_SEG_OFF_TO_LIN(SS_reg(context), LOWORD(ESP_reg(context))))+3,
434                      CX_reg(context), iret );
435     INT_SetRealModeContext(p, &context16);
436 }
437
438
439 static RMCB *DPMI_AllocRMCB( void )
440 {
441     RMCB *NewRMCB = HeapAlloc(GetProcessHeap(), 0, sizeof(RMCB));
442     UINT16 uParagraph;
443
444     if (NewRMCB)
445     {
446         LPVOID RMCBmem = DOSMEM_GetBlock(0, 4, &uParagraph);
447         LPBYTE p = RMCBmem;
448
449         *p++ = 0xcd; /* RMCB: */
450         *p++ = 0x31; /* int $0x31 */
451 /* it is the called procedure's task to change the return CS:EIP
452    the DPMI 0.9 spec states that if it doesn't, it will be called again */
453         *p++ = 0xeb;
454         *p++ = 0xfc; /* jmp RMCB */
455         NewRMCB->address = MAKELONG(0, uParagraph);
456         NewRMCB->next = FirstRMCB;
457         FirstRMCB = NewRMCB;
458     }
459     return NewRMCB;
460 }
461
462
463 static void AllocRMCB( CONTEXT86 *context )
464 {
465     RMCB *NewRMCB = DPMI_AllocRMCB();
466
467     TRACE("Function to call: %04x:%04x\n", (WORD)DS_reg(context), SI_reg(context) );
468
469     if (NewRMCB)
470     {
471         /* FIXME: if 32-bit DPMI client, use ESI and EDI */
472         NewRMCB->proc_ofs = SI_reg(context);
473         NewRMCB->proc_sel = DS_reg(context);
474         NewRMCB->regs_ofs = DI_reg(context);
475         NewRMCB->regs_sel = ES_reg(context);
476         SET_LOWORD( ECX_reg(context), HIWORD(NewRMCB->address) );
477         SET_LOWORD( EDX_reg(context), LOWORD(NewRMCB->address) );
478     }
479     else
480     {
481         SET_LOWORD( EAX_reg(context), 0x8015 ); /* callback unavailable */
482         SET_CFLAG(context);
483     }
484 }
485
486
487 FARPROC16 WINAPI DPMI_AllocInternalRMCB( RMCBPROC proc )
488 {
489     RMCB *NewRMCB = DPMI_AllocRMCB();
490
491     if (NewRMCB) {
492         NewRMCB->proc_ofs = (DWORD)proc;
493         NewRMCB->proc_sel = 0;
494         NewRMCB->regs_ofs = 0;
495         NewRMCB->regs_sel = 0;
496         return (FARPROC16)(NewRMCB->address);
497     }
498     return NULL;
499 }
500
501
502 static int DPMI_FreeRMCB( DWORD address )
503 {
504     RMCB *CurrRMCB = FirstRMCB;
505     RMCB *PrevRMCB = NULL;
506
507     while (CurrRMCB && (CurrRMCB->address != address))
508     {
509         PrevRMCB = CurrRMCB;
510         CurrRMCB = CurrRMCB->next;
511     }
512     if (CurrRMCB)
513     {
514         if (PrevRMCB)
515         PrevRMCB->next = CurrRMCB->next;
516             else
517         FirstRMCB = CurrRMCB->next;
518         DOSMEM_FreeBlock(0, DOSMEM_MapRealToLinear(CurrRMCB->address));
519         HeapFree(GetProcessHeap(), 0, CurrRMCB);
520         return 0;
521     }
522     return 1;
523 }
524
525
526 static void FreeRMCB( CONTEXT86 *context )
527 {
528     FIXME("callback address: %04x:%04x\n",
529           CX_reg(context), DX_reg(context));
530
531     if (DPMI_FreeRMCB(MAKELONG(DX_reg(context), CX_reg(context)))) {
532         SET_LOWORD( EAX_reg(context), 0x8024 ); /* invalid callback address */
533         SET_CFLAG(context);
534     }
535 }
536
537
538 void WINAPI DPMI_FreeInternalRMCB( FARPROC16 proc )
539 {
540     DPMI_FreeRMCB( (DWORD)proc );
541 }
542
543
544 #ifdef MZ_SUPPORTED
545 /* (see loader/dos/module.c, function MZ_InitDPMI) */
546
547 static void StartPM( CONTEXT86 *context, LPDOSTASK lpDosTask )
548 {
549     char *base = DOSMEM_MemoryBase(0);
550     UINT16 cs, ss, ds, es;
551     CONTEXT86 pm_ctx;
552     DWORD psp_ofs = (DWORD)(lpDosTask->psp_seg<<4);
553     PDB16 *psp = (PDB16 *)(base + psp_ofs);
554     HANDLE16 env_seg = psp->environment;
555     int is32;
556
557     RESET_CFLAG(context);
558     lpDosTask->dpmi_flag = AX_reg(context);
559     is32 = lpDosTask->dpmi_flag & 1;
560 /* our mode switch wrapper have placed the desired CS into DX */
561     cs = SELECTOR_AllocBlock( base + (DWORD)(DX_reg(context)<<4), 0x10000, SEGMENT_CODE, FALSE, FALSE );
562 /* due to a flaw in some CPUs (at least mine), it is best to mark stack segments as 32-bit if they
563    can be used in 32-bit code. Otherwise, these CPUs may not set the high word of esp during a
564    ring transition (from kernel code) to the 16-bit stack, and this causes trouble if executing
565    32-bit code using this stack. */
566     ss = SELECTOR_AllocBlock( base + (DWORD)(SS_reg(context)<<4), 0x10000, SEGMENT_DATA, is32, FALSE );
567 /* do the same for the data segments, just in case */
568     if (DS_reg(context) == SS_reg(context)) ds = ss;
569     else ds = SELECTOR_AllocBlock( base + (DWORD)(DS_reg(context)<<4), 0x10000, SEGMENT_DATA, is32, FALSE );
570     es = SELECTOR_AllocBlock( base + psp_ofs, 0x100, SEGMENT_DATA, is32, FALSE );
571 /* convert environment pointer, as the spec says, but we're a bit lazy about the size here... */
572     psp->environment = SELECTOR_AllocBlock( base + (DWORD)(env_seg<<4),
573                                             0x10000, SEGMENT_DATA, FALSE, FALSE );
574
575     pm_ctx = *context;
576     CS_reg(&pm_ctx) = lpDosTask->dpmi_sel;
577 /* our mode switch wrapper expects the new CS in DX, and the new SS in AX */
578     EAX_reg(&pm_ctx) = ss;
579     EDX_reg(&pm_ctx) = cs;
580     DS_reg(&pm_ctx) = ds;
581     ES_reg(&pm_ctx) = es;
582     FS_reg(&pm_ctx) = 0;
583     GS_reg(&pm_ctx) = 0;
584
585     TRACE("DOS program is now entering protected mode\n");
586     Callbacks->CallRegisterShortProc(&pm_ctx, 0);
587
588     /* in the current state of affairs, we won't ever actually return here... */
589     /* we should have int21/ah=4c do it someday, though... */
590
591     SELECTOR_FreeBlock(psp->environment, 1);
592     psp->environment = env_seg;
593     SELECTOR_FreeBlock(es, 1);
594     if (ds != ss) SELECTOR_FreeBlock(ds, 1);
595     SELECTOR_FreeBlock(ss, 1);
596     SELECTOR_FreeBlock(cs, 1);
597 }
598
599 /* DPMI Raw Mode Switch handler */
600
601 #if 0
602 void WINAPI DPMI_RawModeSwitch( SIGCONTEXT *context )
603 {
604   LPDOSTASK lpDosTask = MZ_Current();
605   CONTEXT86 rm_ctx;
606   int ret;
607
608   if (!lpDosTask) {
609     /* we could probably start a DPMI-only dosmod task here, but I doubt
610        anything other than real DOS apps want to call raw mode switch */
611     ERR("attempting raw mode switch without DOS task!\n");
612     ExitProcess(1);
613   }
614   /* initialize real-mode context as per spec */
615   memset(&rm_ctx, 0, sizeof(rm_ctx));
616   DS_reg(&rm_ctx) = AX_sig(context);
617   ES_reg(&rm_ctx) = CX_sig(context);
618   SS_reg(&rm_ctx) = DX_sig(context);
619   ESP_reg(&rm_ctx) = EBX_sig(context);
620   CS_reg(&rm_ctx) = SI_sig(context);
621   EIP_reg(&rm_ctx) = EDI_sig(context);
622   EBP_reg(&rm_ctx) = EBP_sig(context);
623   FS_reg(&rm_ctx) = 0;
624   GS_reg(&rm_ctx) = 0;
625   EFL_reg(&rm_ctx) = EFL_sig(context); /* at least we need the IF flag */
626
627   /* enter real mode again */
628   TRACE("re-entering real mode at %04lx:%04lx\n",
629         CS_reg(&rm_ctx),EIP_reg(&rm_ctx));
630   ret = DOSVM_Enter( &rm_ctx );
631   /* when the real-mode stuff call its mode switch address,
632      DOSVM_Enter will return and we will continue here */
633
634   if (ret<0) {
635     /* if the sync was lost, there's no way to recover */
636     ExitProcess(1);
637   }
638
639   /* alter protected-mode context as per spec */
640   DS_sig(context) = AX_reg(&rm_ctx);
641   ES_sig(context) = CX_reg(&rm_ctx);
642   SS_sig(context) = DX_reg(&rm_ctx);
643   ESP_sig(context) = EBX_reg(&rm_ctx);
644   CS_sig(context) = SI_reg(&rm_ctx);
645   EIP_sig(context) = EDI_reg(&rm_ctx);
646   EBP_sig(context) = EBP_reg(&rm_ctx);
647   FS_sig(context) = 0;
648   GS_sig(context) = 0;
649
650   /* Return to new address and hope that we didn't mess up */
651   TRACE("re-entering protected mode at %04x:%08lx\n",
652         CS_sig(context), EIP_sig(context));
653 }
654 #endif
655
656 #else
657 #if 0
658 void WINAPI DPMI_RawModeSwitch( SIGCONTEXT *context )
659 {
660   ERR("don't even think about DPMI raw mode switch without DOS support!\n");
661   ExitProcess(1);
662 }
663 #endif
664 #endif
665
666 #define DOS_APP_ISDOS(addr,base) ((addr) < 0x110000)
667 #define DOS_WINE_ISDOS(addr,base) (((addr) >= (base)) && ((addr) < (base) + 0x110000))
668 #define DOS_UC_APPTOWINE(addr,base) ((addr) + (base))
669 #define DOS_UC_WINETOAPP(addr,base) ((addr) - (base))
670 #define DOS_APPTOWINE(addr,base) (DOS_APP_ISDOS(addr,base) ? DOS_UC_APPTOWINE(addr,base) : (addr))
671 #define DOS_WINETOAPP(addr,base) (DOS_WINE_ISDOS(addr,base) ? DOS_UC_WINETOAPP(addr,base) : (addr))
672 #define DOS_BADLIMIT(addr,base,limit) \
673   ((limit == 0xffffffff) || /* disallow "fat DS" for now */ \
674    (DOS_WINE_ISDOS(addr,base) && \
675     ((addr) + (limit) > (base) + 0x110000)))
676
677 /**********************************************************************
678  *          INT_Int31Handler
679  *
680  * Handler for int 31h (DPMI).
681  */
682
683 void WINAPI INT_Int31Handler( CONTEXT86 *context )
684 {
685     /*
686      * Note: For Win32s processes, the whole linear address space is
687      *       shifted by 0x10000 relative to the OS linear address space.
688      *       See the comment in msdos/vxd.c.
689      */
690     DWORD offset = W32S_APPLICATION() ? W32S_OFFSET : 0;
691
692     DWORD dw;
693     BYTE *ptr;
694
695     LPDOSTASK lpDosTask = MZ_Current();
696
697 #ifdef MZ_SUPPORTED
698     if (ISV86(context) && lpDosTask) {
699         /* Called from real mode, check if it's our wrapper */
700         TRACE("called from real mode\n");
701         if (CS_reg(context)==lpDosTask->dpmi_seg) {
702             /* This is the protected mode switch */
703             StartPM(context,lpDosTask);
704             return;
705         } else
706         if (CS_reg(context)==lpDosTask->xms_seg) {
707             /* This is the XMS driver entry point */
708             XMS_Handler(context);
709             return;
710         } else
711         {
712             /* Check for RMCB */
713             RMCB *CurrRMCB = FirstRMCB;
714             
715             while (CurrRMCB && (HIWORD(CurrRMCB->address) != CS_reg(context)))
716                 CurrRMCB = CurrRMCB->next;
717             
718             if (CurrRMCB) {
719                 /* RMCB call, propagate to protected-mode handler */
720                 DPMI_CallRMCBProc(context, CurrRMCB, lpDosTask->dpmi_flag);
721                 return;
722             }
723         }
724     }
725 #endif
726
727     RESET_CFLAG(context);
728     switch(AX_reg(context))
729     {
730     case 0x0000:  /* Allocate LDT descriptors */
731         TRACE("allocate LDT descriptors (%d)\n",CX_reg(context));
732         if (!(EAX_reg(context) = AllocSelectorArray16( CX_reg(context) )))
733         {
734             TRACE("failed\n");
735             EAX_reg(context) = 0x8011;  /* descriptor unavailable */
736             SET_CFLAG(context);
737         }
738         TRACE("success, array starts at 0x%04x\n",AX_reg(context));
739         break;
740
741     case 0x0001:  /* Free LDT descriptor */
742         TRACE("free LDT descriptor (0x%04x)\n",BX_reg(context));
743         if (FreeSelector16( BX_reg(context) ))
744         {
745             EAX_reg(context) = 0x8022;  /* invalid selector */
746             SET_CFLAG(context);
747         }
748         else
749         {
750             /* If a segment register contains the selector being freed, */
751             /* set it to zero. */
752             if (!((DS_reg(context)^BX_reg(context)) & ~3)) DS_reg(context) = 0;
753             if (!((ES_reg(context)^BX_reg(context)) & ~3)) ES_reg(context) = 0;
754             if (!((FS_reg(context)^BX_reg(context)) & ~3)) FS_reg(context) = 0;
755             if (!((GS_reg(context)^BX_reg(context)) & ~3)) GS_reg(context) = 0;
756         }
757         break;
758
759     case 0x0002:  /* Real mode segment to descriptor */
760         TRACE("real mode segment to descriptor (0x%04x)\n",BX_reg(context));
761         {
762             WORD entryPoint = 0;  /* KERNEL entry point for descriptor */
763             switch(BX_reg(context))
764             {
765             case 0x0000: entryPoint = 183; break;  /* __0000H */
766             case 0x0040: entryPoint = 193; break;  /* __0040H */
767             case 0xa000: entryPoint = 174; break;  /* __A000H */
768             case 0xb000: entryPoint = 181; break;  /* __B000H */
769             case 0xb800: entryPoint = 182; break;  /* __B800H */
770             case 0xc000: entryPoint = 195; break;  /* __C000H */
771             case 0xd000: entryPoint = 179; break;  /* __D000H */
772             case 0xe000: entryPoint = 190; break;  /* __E000H */
773             case 0xf000: entryPoint = 194; break;  /* __F000H */
774             default:
775                 EAX_reg(context) = DOSMEM_AllocSelector(BX_reg(context));
776                 break;
777             }
778             if (entryPoint) 
779                 EAX_reg(context) = LOWORD(NE_GetEntryPoint( GetModuleHandle16( "KERNEL" ),
780                                                             entryPoint ));
781         }
782         break;
783
784     case 0x0003:  /* Get next selector increment */
785         TRACE("get selector increment (__AHINCR)\n");
786         EAX_reg(context) = __AHINCR;
787         break;
788
789     case 0x0004:  /* Lock selector (not supported) */
790         FIXME("lock selector not supported\n");
791         EAX_reg(context) = 0;  /* FIXME: is this a correct return value? */
792         break;
793
794     case 0x0005:  /* Unlock selector (not supported) */
795         FIXME("unlock selector not supported\n");
796         EAX_reg(context) = 0;  /* FIXME: is this a correct return value? */
797         break;
798
799     case 0x0006:  /* Get selector base address */
800         TRACE("get selector base address (0x%04x)\n",BX_reg(context));
801         if (!(dw = GetSelectorBase( BX_reg(context) )))
802         {
803             EAX_reg(context) = 0x8022;  /* invalid selector */
804             SET_CFLAG(context);
805         }
806         else
807         {
808 #ifdef MZ_SUPPORTED
809             if (lpDosTask) {
810                 DWORD base = (DWORD)DOSMEM_MemoryBase(lpDosTask->hModule);
811                 dw = DOS_WINETOAPP(dw, base);
812             }
813 #endif
814             CX_reg(context) = HIWORD(W32S_WINE2APP(dw, offset));
815             DX_reg(context) = LOWORD(W32S_WINE2APP(dw, offset));
816         }
817         break;
818
819     case 0x0007:  /* Set selector base address */
820         TRACE("set selector base address (0x%04x,0x%08lx)\n",
821                      BX_reg(context),
822                      W32S_APP2WINE(MAKELONG(DX_reg(context),CX_reg(context)), offset));
823         dw = W32S_APP2WINE(MAKELONG(DX_reg(context), CX_reg(context)), offset);
824 #ifdef MZ_SUPPORTED
825         if (lpDosTask) {
826             DWORD base = (DWORD)DOSMEM_MemoryBase(lpDosTask->hModule);
827             dw = DOS_APPTOWINE(dw, base);
828         }
829 #endif
830         SetSelectorBase(BX_reg(context), dw);
831         break;
832
833     case 0x0008:  /* Set selector limit */
834         TRACE("set selector limit (0x%04x,0x%08lx)\n",BX_reg(context),MAKELONG(DX_reg(context),CX_reg(context)));
835         dw = MAKELONG( DX_reg(context), CX_reg(context) );
836 #ifdef MZ_SUPPORTED
837         if (lpDosTask) {
838             DWORD base = (DWORD)DOSMEM_MemoryBase(lpDosTask->hModule);
839             DWORD sbase = GetSelectorBase( BX_reg(context) );
840             if (!sbase) {
841                 /* the app has set the limit without setting the base,
842                  * it must be relying on that the default should be DOS space;
843                  * so set the base address now */
844                 SetSelectorBase( BX_reg(context), sbase = base );
845                 if (dw == 0xffffffff) {
846                     /* djgpp does this without checking (in _dos_ds setup, crt1.c),
847                      * so we have to override the limit here */
848                     dw = 0x110000;
849                 }
850             }
851             if (DOS_BADLIMIT(sbase, base, dw)) {
852                 AX_reg(context) = 0x8021;  /* invalid value */
853                 SET_CFLAG(context);
854                 break;
855             }
856         }
857 #endif
858         SetSelectorLimit16( BX_reg(context), dw );
859         break;
860
861     case 0x0009:  /* Set selector access rights */
862         TRACE("set selector access rights(0x%04x,0x%04x)\n",BX_reg(context),CX_reg(context));
863         SelectorAccessRights16( BX_reg(context), 1, CX_reg(context) );
864         break;
865
866     case 0x000a:  /* Allocate selector alias */
867         TRACE("allocate selector alias (0x%04x)\n",BX_reg(context));
868         if (!(AX_reg(context) = AllocCStoDSAlias16( BX_reg(context) )))
869         {
870             AX_reg(context) = 0x8011;  /* descriptor unavailable */
871             SET_CFLAG(context);
872         }
873         break;
874
875     case 0x000b:  /* Get descriptor */
876         TRACE("get descriptor (0x%04x)\n",BX_reg(context));
877         {
878             ldt_entry entry;
879             LDT_GetEntry( SELECTOR_TO_ENTRY( BX_reg(context) ), &entry );
880 #ifdef MZ_SUPPORTED
881             if (lpDosTask) {
882                 DWORD base = (DWORD)DOSMEM_MemoryBase(lpDosTask->hModule);
883                 entry.base = DOS_WINETOAPP(entry.base, base);
884             }
885 #endif
886             entry.base = W32S_WINE2APP(entry.base, offset);
887
888             /* FIXME: should use ES:EDI for 32-bit clients */
889             LDT_EntryToBytes( PTR_SEG_OFF_TO_LIN( ES_reg(context),
890                                                   DI_reg(context) ), &entry );
891         }
892         break;
893
894     case 0x000c:  /* Set descriptor */
895         TRACE("set descriptor (0x%04x)\n",BX_reg(context));
896         {
897             ldt_entry entry;
898             LDT_BytesToEntry( PTR_SEG_OFF_TO_LIN( ES_reg(context),
899                                                   DI_reg(context) ), &entry );
900             entry.base = W32S_APP2WINE(entry.base, offset);
901 #ifdef MZ_SUPPORTED
902             if (lpDosTask) {
903                 DWORD base = (DWORD)DOSMEM_MemoryBase(lpDosTask->hModule);
904                 entry.base = DOS_APPTOWINE(entry.base, base);
905                 if (DOS_BADLIMIT(entry.base, base, entry.limit)) {
906                     AX_reg(context) = 0x8021;  /* invalid value */
907                     SET_CFLAG(context);
908                     break;
909                 }
910             }
911 #endif
912
913             LDT_SetEntry( SELECTOR_TO_ENTRY( BX_reg(context) ), &entry );
914         }
915         break;
916
917     case 0x000d:  /* Allocate specific LDT descriptor */
918         FIXME("allocate descriptor (0x%04x), stub!\n",BX_reg(context));
919         AX_reg(context) = 0x8011; /* descriptor unavailable */
920         SET_CFLAG(context);
921         break;
922     case 0x0100:  /* Allocate DOS memory block */
923         TRACE("allocate DOS memory block (0x%x paragraphs)\n",BX_reg(context));
924         dw = GlobalDOSAlloc16((DWORD)BX_reg(context)<<4);
925         if (dw) {
926             AX_reg(context) = HIWORD(dw);
927             DX_reg(context) = LOWORD(dw);
928         } else {
929             AX_reg(context) = 0x0008; /* insufficient memory */
930             BX_reg(context) = DOSMEM_Available(0)>>4;
931             SET_CFLAG(context);
932         }
933         break;
934     case 0x0101:  /* Free DOS memory block */
935         TRACE("free DOS memory block (0x%04x)\n",DX_reg(context));
936         dw = GlobalDOSFree16(DX_reg(context));
937         if (!dw) {
938             AX_reg(context) = 0x0009; /* memory block address invalid */
939             SET_CFLAG(context);
940         }
941         break;
942     case 0x0200: /* get real mode interrupt vector */
943         FIXME("get realmode interupt vector(0x%02x) unimplemented.\n",
944               BL_reg(context));
945         SET_CFLAG(context);
946         break;
947     case 0x0201: /* set real mode interrupt vector */
948         FIXME("set realmode interupt vector(0x%02x,0x%04x:0x%04x) unimplemented\n", BL_reg(context),CX_reg(context),DX_reg(context));
949         SET_CFLAG(context);
950         break;
951     case 0x0204:  /* Get protected mode interrupt vector */
952         TRACE("get protected mode interrupt handler (0x%02x), stub!\n",BL_reg(context));
953         dw = (DWORD)INT_GetPMHandler( BL_reg(context) );
954         CX_reg(context) = HIWORD(dw);
955         DX_reg(context) = LOWORD(dw);
956         break;
957
958     case 0x0205:  /* Set protected mode interrupt vector */
959         TRACE("set protected mode interrupt handler (0x%02x,%p), stub!\n",
960             BL_reg(context),PTR_SEG_OFF_TO_LIN(CX_reg(context),DX_reg(context)));
961         INT_SetPMHandler( BL_reg(context),
962                           (FARPROC16)PTR_SEG_OFF_TO_SEGPTR( CX_reg(context),
963                                                             DX_reg(context) ));
964         break;
965
966     case 0x0300:  /* Simulate real mode interrupt */
967         CallRMInt( context );
968         break;
969
970     case 0x0301:  /* Call real mode procedure with far return */
971         CallRMProc( context, FALSE );
972         break;
973
974     case 0x0302:  /* Call real mode procedure with interrupt return */
975         CallRMProc( context, TRUE );
976         break;
977
978     case 0x0303:  /* Allocate Real Mode Callback Address */
979         AllocRMCB( context );
980         break;
981
982     case 0x0304:  /* Free Real Mode Callback Address */
983         FreeRMCB( context );
984         break;
985
986     case 0x0305:  /* Get State Save/Restore Addresses */
987         TRACE("get state save/restore addresses\n");
988         /* we probably won't need this kind of state saving */
989         AX_reg(context) = 0;
990         /* real mode: just point to the lret */
991         BX_reg(context) = DPMI_wrap_seg;
992         ECX_reg(context) = 2;
993         /* protected mode: don't have any handler yet... */
994         FIXME("no protected-mode dummy state save/restore handler yet\n");
995         SI_reg(context) = 0;
996         EDI_reg(context) = 0;
997         break;
998
999     case 0x0306:  /* Get Raw Mode Switch Addresses */
1000         TRACE("get raw mode switch addresses\n");
1001         if (lpDosTask) {
1002           /* real mode, point to standard DPMI return wrapper */
1003           BX_reg(context) = DPMI_wrap_seg;
1004           ECX_reg(context) = 0;
1005           /* protected mode, point to DPMI call wrapper */
1006           SI_reg(context) = lpDosTask->dpmi_sel;
1007           EDI_reg(context) = 8; /* offset of the INT 0x31 call */
1008         } else {
1009           ERR("win app attempting to get raw mode switch!\n");
1010           AX_reg(context) = 0x8001; /* unsupported function */
1011           SET_CFLAG(context);
1012         }
1013         break;
1014     case 0x0400:  /* Get DPMI version */
1015         TRACE("get DPMI version\n");
1016         {
1017             SYSTEM_INFO si;
1018
1019             GetSystemInfo(&si);
1020             AX_reg(context) = 0x005a;  /* DPMI version 0.90 */
1021             BX_reg(context) = 0x0005;  /* Flags: 32-bit, virtual memory */
1022             CL_reg(context) = si.wProcessorLevel;
1023             DX_reg(context) = 0x0102;  /* Master/slave interrupt controller base*/
1024             break;
1025         }
1026     case 0x0500:  /* Get free memory information */
1027         TRACE("get free memory information\n");
1028         {
1029             MEMMANINFO mmi;
1030
1031             mmi.dwSize = sizeof(mmi);
1032             MemManInfo16(&mmi);
1033             ptr = (BYTE *)PTR_SEG_OFF_TO_LIN(ES_reg(context),DI_reg(context));
1034             /* the layout is just the same as MEMMANINFO, but without
1035              * the dwSize entry.
1036              */
1037             memcpy(ptr,((char*)&mmi)+4,sizeof(mmi)-4);
1038             break;
1039         }
1040     case 0x0501:  /* Allocate memory block */
1041         TRACE("allocate memory block (%ld)\n",MAKELONG(CX_reg(context),BX_reg(context)));
1042         if (!(ptr = (BYTE *)DPMI_xalloc(MAKELONG(CX_reg(context), BX_reg(context)))))
1043         {
1044             AX_reg(context) = 0x8012;  /* linear memory not available */
1045             SET_CFLAG(context);
1046         } else {
1047             BX_reg(context) = SI_reg(context) = HIWORD(W32S_WINE2APP(ptr, offset));
1048             CX_reg(context) = DI_reg(context) = LOWORD(W32S_WINE2APP(ptr, offset));
1049         }
1050         break;
1051
1052     case 0x0502:  /* Free memory block */
1053         TRACE("free memory block (0x%08lx)\n",
1054                      W32S_APP2WINE(MAKELONG(DI_reg(context),SI_reg(context)), offset));
1055         DPMI_xfree( (void *)W32S_APP2WINE(MAKELONG(DI_reg(context), 
1056                                                SI_reg(context)), offset) );
1057         break;
1058
1059     case 0x0503:  /* Resize memory block */
1060         TRACE("resize memory block (0x%08lx,%ld)\n",
1061                      W32S_APP2WINE(MAKELONG(DI_reg(context),SI_reg(context)), offset),
1062                      MAKELONG(CX_reg(context),BX_reg(context)));
1063         if (!(ptr = (BYTE *)DPMI_xrealloc(
1064                 (void *)W32S_APP2WINE(MAKELONG(DI_reg(context),SI_reg(context)), offset),
1065                 MAKELONG(CX_reg(context),BX_reg(context)))))
1066         {
1067             AX_reg(context) = 0x8012;  /* linear memory not available */
1068             SET_CFLAG(context);
1069         } else {
1070             BX_reg(context) = SI_reg(context) = HIWORD(W32S_WINE2APP(ptr, offset));
1071             CX_reg(context) = DI_reg(context) = LOWORD(W32S_WINE2APP(ptr, offset));
1072         }
1073         break;
1074
1075     case 0x0507:  /* Modify page attributes */
1076         FIXME("modify page attributes unimplemented\n");
1077         break;  /* Just ignore it */
1078
1079     case 0x0600:  /* Lock linear region */
1080         FIXME("lock linear region unimplemented\n");
1081         break;  /* Just ignore it */
1082
1083     case 0x0601:  /* Unlock linear region */
1084         FIXME("unlock linear region unimplemented\n");
1085         break;  /* Just ignore it */
1086
1087     case 0x0602:  /* Unlock real-mode region */
1088         FIXME("unlock realmode region unimplemented\n");
1089         break;  /* Just ignore it */
1090
1091     case 0x0603:  /* Lock real-mode region */
1092         FIXME("lock realmode region unimplemented\n");
1093         break;  /* Just ignore it */
1094
1095     case 0x0604:  /* Get page size */
1096         TRACE("get pagesize\n");
1097         BX_reg(context) = 0;
1098         CX_reg(context) = VIRTUAL_GetPageSize();
1099         break;
1100
1101     case 0x0702:  /* Mark page as demand-paging candidate */
1102         FIXME("mark page as demand-paging candidate\n");
1103         break;  /* Just ignore it */
1104
1105     case 0x0703:  /* Discard page contents */
1106         FIXME("discard page contents\n");
1107         break;  /* Just ignore it */
1108
1109      case 0x0800:  /* Physical address mapping */
1110         FIXME("map real to linear (0x%08lx)\n",MAKELONG(CX_reg(context),BX_reg(context)));
1111          if(!(ptr=DOSMEM_MapRealToLinear(MAKELONG(CX_reg(context),BX_reg(context)))))
1112          {
1113              AX_reg(context) = 0x8021; 
1114              SET_CFLAG(context);
1115          }
1116          else
1117          {
1118              BX_reg(context) = HIWORD(W32S_WINE2APP(ptr, offset));
1119              CX_reg(context) = LOWORD(W32S_WINE2APP(ptr, offset));
1120              RESET_CFLAG(context);
1121          }
1122          break;
1123
1124     default:
1125         INT_BARF( context, 0x31 );
1126         AX_reg(context) = 0x8001;  /* unsupported function */
1127         SET_CFLAG(context);
1128         break;
1129     }
1130
1131 }