Changeset 4065 for branches/GRACE/src/win32k/pe2lx/pe2lx.cpp

Timestamp:

Aug 22, 2000, 12:59:40 AM (25 years ago)

Author:

bird

Message:

Temporary checkin (backup).

File:

• branches/GRACE/src/win32k/pe2lx/pe2lx.cpp (modified) (7 diffs)

branches/GRACE/src/win32k/pe2lx/pe2lx.cpp

@@ -1 @@
-/* $Id: pe2lx.cpp,v 1.18.4.4 2000-08-20 08:08:47 bird Exp $
+/* $Id: pe2lx.cpp,v 1.18.4.5 2000-08-21 22:59:40 bird Exp $
  *
  * Pe2Lx class implementation. Ring 0 and Ring 3
@@ -100 +100 @@
 #include <versionos2.h>                 /* Pe2Lx version. */
 #include "yield.h"                      /* Yield CPU. */
+#include "options.h"                    /* Win32k options. */
 
 
@@ -205 +206 @@
     LXHdr.e32_pagesize  = PAGESIZE;
     LXHdr.e32_objtab    = sizeof(LXHdr);
+    cLoadedModules++;
 }
 
@@ -277 +279 @@
     _res_heapmin();
     _swp_heapmin();
+
+    /*
+     * If no Pe2Lx objects left, then invalidate the Odin32Path.
+     *  We'll have to do this since we may (theoretically) not receive
+     *  a close on a kernel32 handle if loading failes before kernel32
+     *  is loaded and the odin32path is determined using ldrOpenPath.
+     *  (Ie. no sfnKernel32.)
+     */
+    if (--cLoadedModules <= 0)
+        invalidateOdin32Path();
+#ifdef DEBUG
+    if (cLoadedModules < 0)
+        printIPE(("Pe2Lx::cLoadedModules is %d which is less that zero!\n", cLoadedModules));
+#endif
 }
 
@@ -1729 +1745 @@
 
 /**
+ * Invalidates the odin32path.
+ * Called by ldrClose when the kernel32 handle is closed.
+ * @sketch      Free path
+ *              nullify path pointer and kernel32 handle.
+ * @author      knut st. osmundsen (knut.stange.osmundsen@mynd.no)
+ */
+VOID Pe2Lx::invalidateOdin32Path()
+{
+    if (pszOdin32Path != NULL)
+    {
+        rfree((void*)pszOdin32Path);
+        pszOdin32Path = NULL;
+    }
+    sfnKernel32 = NULLHANDLE;
+}
+
+
+/**
  * Gets the size of the virtual LX-file.
  * @returns   Size of the virtual LX-file in bytes.
@@ -4647 +4681 @@
 
 
+
+/**
+ * Initiates the odin32path.
+ * @returns     Success indicator.
+ * @sketch
+ * @status
+ * @author      knut st. osmundsen (knut.stange.osmundsen@mynd.no)
+ * @remark
+ */
 BOOL Pe2Lx::initOdin32Path()
 {
@@ -4661 +4704 @@
     pMTE = NULL;
     rc = ldrFindModule("KERNEL32", 8, CLASS_GLOBAL, &pMTE);
-    if (rc == NO_ERROR && pMTE != NULL)
-    {
-        char *psz = pMTE->mte_swapmte->smte_path;
-        if (psz != NULL)
-        {
-            char * pszEnd = psz + strlen(psz) - 1;
-            while (pszEnd > psz && *pszEnd != '\\' && *pszEnd != '/' && *pszEnd != ':')
-                pszEnd--;
-            pszEnd--;
-            while (pszEnd > psz && *pszEnd != '\\' && *pszEnd != '/' && *pszEnd != ':')
-                pszEnd--;
-            if (pszEnd > psz)
-            {
-                char * pszPath = (char*)malloc(pszEnd - psz);
-                memcpy(pszPath, psz, pszEnd - psz);
-                pszPath[pszEnd - psz] = '\0';
-
-                if (pszOdin32Path)
-                    free((void*)pszOdin32Path);
-                pszOdin32Path = pszPath;
-                sfnKernel32 = pMTE->mte_sfn;
-
-                return TRUE;
-            }
-        }
-    }
+    if (rc == NO_ERROR && pMTE != NULL && pMTE->mte_swapmte != NULL)
+    {
+        /*
+         * We now take the smte_path. Start at the end and skip the filename,
+         * and one directory up. We assume a fully qualified path is found in
+         * smte_path.
+         */
+        if (pMTE->mte_swapmte->smte_path != NULL)//paranoia
+        {
+            sfnKernel32 = pMTE->mte_sfn;
+            return setOdin32Path(pMTE->mte_swapmte->smte_path);
+        }
+    }
+
+
+    /*
+     * KERNEL32 isn't loaded. We'll only search the paths if
+     */
+    if (pszOdin32Path != NULL)
+        return TRUE;
+
 
     /*
      * Try find it searching the LIBPATHs.
-     * (Don't thing we should use ldrOpenPath since it
-     *  end's up calling ldrOpen - which is overloaded.)
+     *
+     * For the time being:
+     *  We'll use ldrOpenPath to do this, but we'll have to
+     *  disable the win32k.sys overloading temporarily.
      */
+    ldrlv_t lv = {0};
+    ULONG   ful = 0;
+    ULONG   ul;
+
+    ul = options.fNoLoader;
+    options.fNoLoader = TRUE;
+    rc = ldrOpenPath("KERNEL32", 8, &lv, &ful);
+    options.fNoLoader = ul;
+    if (rc == NO_ERROR)
+    {
+        /*
+         * Set the odin32path according to the kernel32 path we've found.
+         * (ldrOpen sets ldrpFileNameBuf to the fully qualified path of
+         *  the last opended filed, which in this case is kernel32.dll.)
+         * We'll close the file handle first of course.
+         */
+        ldrClose(lv.lv_sfn);
+        return setOdin32Path(ldrpFileNameBuf);
+    }
+
+    return rc == NO_ERROR;
+}
+
+
+
+/**
+ * Sets the Odin32Path to the given fully qualified filename of kernel32.
+ * @returns     Success indicator.
+ * @param       psz     Fully qualified filename of kernel32 with path.
+ * @sketch
+ * @status
+ * @author      knut st. osmundsen (knut.stange.osmundsen@mynd.no)
+ * @remark
+ */
+BOOL Pe2Lx::setOdin32Path(const char *psz)
+{
+    const char * pszEnd;
+    char * pszPath;
+
+    /*
+     * We now take the psz. Start at the end and skip the filename,
+     * and one directory up. We assume a fully qualified path.
+     */
+    pszEnd = psz + strlen(psz) - 1;
+    while (pszEnd > psz && *pszEnd != '\\' && *pszEnd != '/' && *pszEnd != ':')
+        pszEnd--;
+    pszEnd--;
+    while (pszEnd > psz && *pszEnd != '\\' && *pszEnd != '/' && *pszEnd != ':')
+        pszEnd--;
+    if (pszEnd > psz)
+    {
+        /*
+         * Free old path (if any) and allocate space for a new path.
+         * Copy the path including the slash.
+         * Remember the kernel32 filehandle (to be able to invalidate the path).
+         */
+        if (pszOdin32Path)
+            rfree((void*)pszOdin32Path);
+        if (*pszEnd == ':') //in case someone installed odin in a root directory.
+            pszEnd++;
+        pszEnd++;       //include the slash
+        pszPath = (char*)rmalloc(pszEnd - psz);
+        if (pszPath == NULL) return FALSE;
+        memcpy(pszPath, psz, pszEnd - psz);
+        pszPath[pszEnd - psz] = '\0';
+        pszOdin32Path = pszPath;
+
+        return TRUE;
+    }
 
     return FALSE;
 }
+
+