Changeset 2501 for trunk/src/win32k/dev32/d32init.c

Timestamp:

Jan 22, 2000, 7:21:03 PM (26 years ago)

Author:

bird

Message:

Temporary backup checkin.

File:

• trunk/src/win32k/dev32/d32init.c (modified) (15 diffs)

trunk/src/win32k/dev32/d32init.c

@@ -1 @@
-/* $Id: d32init.c,v 1.4 1999-11-10 01:45:30 bird Exp $
+/* $Id: d32init.c,v 1.5 2000-01-22 18:20:57 bird Exp $
  *
  * d32init.c - 32-bits init routines.
@@ -19 +19 @@
 #define INCL_NOPMAPI
 #define LDR_INCL_INITONLY
+
 
 /*******************************************************************************
@@ -43 +44 @@
 *******************************************************************************/
 static ULONG    readnum(const char *pszNum);
-static int      interpretFunctionProlog(char *p);
+static int      interpretFunctionProlog(char *p, BOOL fOverload);
 static int      procInit(void);
 
 
 /* externs located in 16-bit data segement */
-extern ULONG  _TKSSBase16;
-extern USHORT _R0FlatCS16;
-extern USHORT _R0FlatDS16;
+extern ULONG    _TKSSBase16;
+extern USHORT   _R0FlatCS16;
+extern USHORT   _R0FlatDS16;
 
 
 /* extern(s) located in calltab.asm */
-extern char callTab[NUMBER_OF_PROCS][MAXSIZE_PROLOG];
+extern char     callTab[NUMBER_OF_PROCS][MAXSIZE_PROLOG];
 
 
@@ -88 +89 @@
     while (pszTmp != NULL)
     {
-        char cch;
+        int cch;
         pszTmp++; //skip [-/]
         cch = strlen(pszTmp);
@@ -309 +310 @@
         if (pszNum[1] == 'x' || pszNum[1] == 'X')
         {
-            ulBase == 16;
+            ulBase = 16;
             pszNum += 2;
         }
         else
         {
-            ulBase == 8;
+            ulBase = 8;
             i = 1;
         }
@@ -364 +365 @@
         }
 
-        /* verify known function prolog. (only EPT_PROC) */
-        if (_aProcTab[i].fType == EPT_PROC)
-        {
-            if ((cb = interpretFunctionProlog((char*)_aProcTab[i].ulAddress)) <= 0 && cb + 5 >= MAXSIZE_PROLOG)
-            {
-                kprintf(("VerifyProcTab32: verify failed for procedure no.%d\n",i));
-                return STATUS_DONE | STERR | 3;
-            }
-        }
-        else
-        {
-            kprintf(("VerifyProcTab32: only EPT_PROC is implemented\n",i));
-            return STATUS_DONE | STERR | 4;
+        switch (_aProcTab[i].fType)
+        {
+            case EPT_PROC:
+            case EPT_PROCIMPORT:
+                /* verify known function prolog. */
+                if ((cb = interpretFunctionProlog((char*)_aProcTab[i].ulAddress, _aProcTab[i].fType == EPT_PROC))
+                    <= 0 && cb + 5 >= MAXSIZE_PROLOG)
+                {
+                    kprintf(("VerifyProcTab32: verify failed for procedure no.%d\n",i));
+                    return STATUS_DONE | STERR | 3;
+                }
+                break;
+
+            default:
+                kprintf(("VerifyProcTab32: only EPT_PROC is implemented\n",i));
+                return STATUS_DONE | STERR | 4;
         }
     }
@@ -408 +412 @@
         if (pSMTE != NULL)
         {
-            pOTEBuf->cObjects = pSMTE->smte_objcnt;
+            pOTEBuf->cObjects = (unsigned char)pSMTE->smte_objcnt;
             if (pSMTE->smte_objcnt <= MAXKRNLOBJECTS)
             {
@@ -433 +437 @@
         kprintf(("GetOTEs32: failed. usRc = %d\n", usRc));
 
-    return usRc | (usRc != 0 ? STATUS_DONE | STERR : STATUS_DONE);
+    return (USHORT)(usRc | (usRc != NO_ERROR ? STATUS_DONE | STERR : STATUS_DONE));
 }
 
@@ -442 +446 @@
  *            where the jmp instr should be placed.
  *            On error it returns 0.
- * @param     p  Pointer to prolog.
+ * @param     pach       Pointer to prolog.
+ * @param     fOverload  TRUE:  Function is to be overloaded.
+ *                       FALSE: Function is to be imported.
  */
-static int interpretFunctionProlog(char *p)
+static int interpretFunctionProlog(char *pach, BOOL fOverload)
 {
-    int length;
     int rc;
 
@@ -459 +464 @@
      */
 
-    if (p[0] == 0x55 && p[1] == 0x8b)
-    {
-        if (p[2] == 0xec)
+    if (pach[0] == 0x55 && pach[1] == 0x8b)
+    {
+        if (pach[2] == 0xec)
             rc = 3;
         else
@@ -472 +477 @@
              * There will never be any doubt when something goes wrong!
              */
-            switch(p[rc])
+            switch(pach[rc])
             {
                 case 0x33: /* xor (ldrClose, ldrOpen) */
@@ -478 +483 @@
                     break;
                 case 0x8b:
-                    if (p[rc+1] == 0x0d)
+                    if (pach[rc+1] == 0x0d)
                         rc += 6;
                     else
@@ -490 +495 @@
                     break;
                 default:
-                    kprintf(("interpretFunctionProlog: unknown instruction 0x%x \n", p[rc]));
+                    kprintf(("interpretFunctionProlog: unknown instruction 0x%x\n", pach[rc]));
                     return 0;
             }
@@ -496 +501 @@
     }
     else
-        rc = 0;
+    {
+        /* special case for IOSftReadAt and IOSftWriteAt */
+        if (fOverload == FALSE && pach[0] == 0xB8 && (pach[5] == 0xEB || pach[5] == 0x55))
+            rc = 5;
+        else
+            rc = 0;
+    }
 
     return rc;
@@ -512 +523 @@
     int cb;
 
-    /* verify */
+    /*
+     * verify proctable
+     */
     for (i = 0; i < NUMBER_OF_PROCS; i++)
     {
-        cb = interpretFunctionProlog((char*)_aProcTab[i].ulAddress);
+        if (_aProcTab[i].fType != EPT_PROC && _aProcTab[i].fType != EPT_PROCIMPORT)
+        {
+            kprintf(("procInit: EPT_VAR is not supported. (procedure no.%d, cb=%d)\n", i, cb));
+            return 1;
+        }
+        cb = interpretFunctionProlog((char*)_aProcTab[i].ulAddress, _aProcTab[i].fType == EPT_PROC);
         if (cb <= 0 || cb + 5 >= MAXSIZE_PROLOG)
         {
-            kprintf(("rehookFunctions: verify failed for procedure no.%d, cb=%d\n", i, cb));
+            kprintf(("procInit: verify failed for procedure no.%d, cb=%d\n", i, cb));
             return 1;
         }
     }
 
-    /* rehook */
+    /*
+     * rehook / import
+     */
     for (i = 0; i < NUMBER_OF_PROCS; i++)
     {
-        cb = interpretFunctionProlog((char*)_aProcTab[i].ulAddress);
-        if (cb > 0 && cb + 5 < MAXSIZE_PROLOG)
-        {
-            char *pMy;
-            switch (i)
+        switch (_aProcTab[i].fType)
+        {
+            case EPT_PROC:
             {
-                case iLDRREAD:          pMy = (char*)myldrRead; break;
-                case iLDROPEN:          pMy = (char*)myldrOpen; break;
-                case iLDRCLOSE:         pMy = (char*)myldrClose; break;
-                case iLDRQAPPTYPE:      pMy = (char*)myLDRQAppType; break;
-
-                default:
-                    kprintf(("rehookFunctions: Someone has added function without updating the switch! i=%d\n", i));
-                    Int3();
-                    return 2;
+                cb = interpretFunctionProlog((char*)_aProcTab[i].ulAddress, TRUE);
+                if (cb > 0 && cb + 5 < MAXSIZE_PROLOG)
+                {
+                    static unsigned auFuncs[NUMBER_OF_PROCS] = /* This table must be updated with the overloading functions. */
+                    {
+                        (unsigned)myldrRead,
+                        (unsigned)myldrOpen,
+                        (unsigned)myldrClose,
+                        (unsigned)myLDRQAppType,
+                        (unsigned)myldrEnum32bitRelRecs,
+                        0,
+                        0,
+                        0,
+                        0,
+                        0,
+                    };
+
+                    /* copy function prolog */
+                    memcpy(callTab[i], (void*)_aProcTab[i].ulAddress, (size_t)cb);
+
+                    /* jump from calltab to original function */
+                    callTab[i][cb] = 0xE9; /* jmp */
+                    *(unsigned*)(void*)&callTab[i][cb+1] = _aProcTab[i].ulAddress + cb - (unsigned)&callTab[i][cb+5];
+
+
+                    /* jump from original function to my function - an cli(?) could be needed here */
+                    *(char*)_aProcTab[i].ulAddress = 0xE9; /* jmp */
+                    *(unsigned*)(_aProcTab[i].ulAddress + 1) = auFuncs[i] - (_aProcTab[i].ulAddress + 5);
+                }
+                else
+                {   /* !fatal! - this could never happen really... */
+                    kprintf(("procInit: FATAL verify failed for procedure no.%d when rehooking it!\n",i));
+                    Int3(); /* ipe - later! */
+                    return 1;
+                }
+                break;
             }
 
-            /* copy function prolog */
-            memcpy(callTab[i], (void*)_aProcTab[i].ulAddress, (size_t)cb);
-
-            /* jump from calltab to original function */
-            callTab[i][cb] = 0xE9; /* jmp */
-            *(unsigned*)&callTab[i][cb+1] = _aProcTab[i].ulAddress + cb - (unsigned)&callTab[i][cb+5];
-
-            /* jump from original function to my function - an cli could be needed here */
-            *(char*)_aProcTab[i].ulAddress = 0xE9; /* jmp */
-            *(unsigned*)(_aProcTab[i].ulAddress + 1) = (unsigned)pMy - (_aProcTab[i].ulAddress + 5);
-        }
-        else
-        {
-            /* !fatal! - this could never happen really... */
-            kprintf(("rehookFunctions: FATAL verify failed for procedure no.%d when rehooking it!\n",i));
-            Int3(); /* ipe - later! */
-            return 1;
-        }
-    }
+            case EPT_PROCIMPORT:
+            {
+                cb = interpretFunctionProlog((char*)_aProcTab[i].ulAddress, FALSE);
+                if (cb > 0 && cb + 5 < MAXSIZE_PROLOG)
+                {
+                    /* jump from calltab to original function */
+                    callTab[i][0] = 0xE9; /* jmp */
+                    *(unsigned*)(void*)&callTab[i][1] = _aProcTab[i].ulAddress - (unsigned)&callTab[i][cb+5];
+                }
+                else
+                {   /* !fatal! - this could never happen really... */
+                    kprintf(("procInit: FATAL verify failed for procedure no.%d when importing it!\n",i));
+                    Int3(); /* ipe - later! */
+                    return 1;
+                }
+                break;
+            }
+
+            default:
+                kprintf(("procInit: EPT_VAR is not supported. (procedure no.%d, cb=%d)\n", i, cb));
+                Int3(); /* ipe - later! */
+                return 1;
+        } /* switch - type */
+    }   /* for */
 
     return NO_ERROR;