Changeset 2855 for trunk/kLdr/kLdrModPE.c

Timestamp:

Nov 4, 2006, 3:30:19 AM (19 years ago)

Author:

bird

Message:

More code.

File:

• trunk/kLdr/kLdrModPE.c (modified) (5 diffs)

trunk/kLdr/kLdrModPE.c

@@ -99 +99 @@
 static int kldrModPEQueryForwarder(PKLDRMODPE pModPE, const void *pvBits, const char *pszForwarder,
                                    PFNKLDRMODGETIMPORT pfnGetImport, void *pvUser, PKLDRADDR puValue, uint32_t *pfKind);
+static int32_t kldrModPENumberOfImports(PKLDRMOD pMod, const void *pvBits);
 
 
@@ -643 +644 @@
 {
     const IMAGE_IMPORT_DESCRIPTOR *paImpDir;
-    uint32_t        cImpModules;
     uint32_t        iImpModule;
     uint32_t        cchImpModule;
@@ -718 +718 @@
      * entry for the module... not sure if this is right though.
      */
-    cImpModules = pModPE->Hdrs.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].Size
-                / sizeof(IMAGE_IMPORT_DESCRIPTOR);
+    if (    !pModPE->Hdrs.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].Size
+        ||  !pModPE->Hdrs.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress)
+        return KLDR_ERR_PE_FORWARDER_IMPORT_NOT_FOUND;
     paImpDir = KLDRMODPE_RVA2TYPE(pvBits,
                                   pModPE->Hdrs.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress,
                                   const IMAGE_IMPORT_DESCRIPTOR *);
-    for (iImpModule = 0; iImpModule < cImpModules; cImpModules++)
-    {
-        const char     *pszName = KLDRMODPE_RVA2TYPE(pvBits, paImpDir[iImpModule].Name, const char *);
-        const size_t    cchName = kLdrHlpStrLen(pszName);
+
+    kldrModPENumberOfImports(pModPE->pMod, pvBits);
+    for (iImpModule = 0; iImpModule < pModPE->cImportModules; iImpModule++)
+    {
+        const char *pszName = KLDRMODPE_RVA2TYPE(pvBits, paImpDir[iImpModule].Name, const char *);
+        size_t      cchName = kLdrHlpStrLen(pszName);
         if (    (   cchName == cchImpModule
                  || (   cchName > cchImpModule
@@ -736 +739 @@
             &&  kLdrHlpMemIComp(pszName, pszForwarder, cchImpModule)
            )
-            break;
-    }
-    if (iImpModule >= cImpModules)
-        return KLDR_ERR_PE_FORWARDER_IMPORT_NOT_FOUND;
-
-    /*
-     * Now the rest is up to the callback (almost).
-     */
-    rc = pfnGetForwarder(pModPE->pMod, iImpModule, uSymbol, pszSymbol, puValue, pfKind, pvUser);
-    if (!rc && pfKind)
-        *pfKind |= KLDRSYMKIND_FORWARDER;
-    return rc;
+        {
+            /*
+             * Now the rest is up to the callback (almost).
+             */
+            rc = pfnGetForwarder(pModPE->pMod, iImpModule, uSymbol, pszSymbol, puValue, pfKind, pvUser);
+            if (!rc && pfKind)
+                *pfKind |= KLDRSYMKIND_FORWARDER;
+            return rc;
+        }
+    }
+    return KLDR_ERR_PE_FORWARDER_IMPORT_NOT_FOUND;
 }
 
@@ -831 +833 @@
 
 /** @copydoc kLdrModGetImport */
-int (* pfnGetImport)(PKLDRMOD pMod, void *pvBits, uint32_t iImport, const char *pszName, size_t cchName);
+static int kldrModPEGetImport(PKLDRMOD pMod, void *pvBits, uint32_t iImport, char *pszName, size_t cchName)
+{
+    PKLDRMODPE                      pModPE = (PKLDRMODPE)pMod->pvData;
+    const IMAGE_IMPORT_DESCRIPTOR  *pImpDesc;
+    const char                     *pszImportName;
+    size_t                          cchImportName;
+    int                             rc;
+
+    /*
+     * Make sure we've got mapped bits and resolve any base address aliases.
+     */
+    rc = kldrModPEBitsAndBaseAddress(pModPE, &pvBits, NULL);
+    if (rc)
+        return rc;
+
+    /*
+     * Simple bounds check.
+     */
+    if (iImport >= (uint32_t)kldrModPENumberOfImports(pMod, pvBits))
+        return KLDR_ERR_IMPORT_ORDINAL_OUT_OF_BOUNDS;
+
+    /*
+     * Get the name.
+     */
+    pImpDesc = KLDRMODPE_RVA2TYPE(pvBits,
+                                  pModPE->Hdrs.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress
+                                  + sizeof(IMAGE_IMPORT_DESCRIPTOR) * iImport,
+                                  const IMAGE_IMPORT_DESCRIPTOR *);
+    pszImportName = KLDRMODPE_RVA2TYPE(pvBits, pImpDesc->Name, const char *);
+    cchImportName = strlen(pszImportName);
+    if (cchImportName < cchName)
+    {
+        kLdrHlpMemCopy(pszName, pszImportName, cchImportName + 1);
+        rc = 0;
+    }
+    else
+    {
+        kLdrHlpMemCopy(pszName, pszImportName, cchName);
+        if (cchName)
+            pszName[cchName - 1] = '\0';
+        rc = KLDR_ERR_BUFFER_OVERFLOW;
+    }
+
+    return rc;
+}
+
+
 /** @copydoc kLdrModNumberOfImports */
-int32_t (* pfnNumberOfImports)(PKLDRMOD pMod, void *pvBits);
-/** @copydoc kLdrModCanExecuteOn */
-int (* pfnCanExecuteOn)(PKLDRMOD pMod, void *pvBits, KLDRARCH enmArch, KLDRCPU enmCpu);
+static int32_t kldrModPENumberOfImports(PKLDRMOD pMod, const void *pvBits)
+{
+    PKLDRMODPE pModPE = (PKLDRMODPE)pMod->pvData;
+    if (pModPE->cImportModules == ~(uint32_t)0)
+    {
+        /*
+         * We'll have to walk the import descriptors to figure out their number.
+         * First, make sure we've got mapped bits and resolve any base address aliases.
+         */
+        if (kldrModPEBitsAndBaseAddress(pModPE, &pvBits, NULL))
+            return -1;
+        pModPE->cImportModules = 0;
+        if (    pModPE->Hdrs.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].Size
+            &&  pModPE->Hdrs.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress)
+        {
+            const IMAGE_IMPORT_DESCRIPTOR  *pImpDesc;
+
+            pImpDesc = KLDRMODPE_RVA2TYPE(pvBits,
+                                          pModPE->Hdrs.OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress,
+                                          const IMAGE_IMPORT_DESCRIPTOR *);
+            while (pImpDesc->Name && pImpDesc->FirstThunk)
+            {
+                pModPE->cImportModules = 0;
+                pImpDesc++;
+            }
+        }
+    }
+    return pModPE->cImportModules;
+}
+
+
 /** @copydoc kLdrModGetStackInfo */
-int (* pfnGetStackInfo)(PKLDRMOD pMod, void *pvBits, KLDRADDR BaseAddress, PKLDRSTACKINFO pStackInfo);
+static int kldrModPEGetStackInfo(PKLDRMOD pMod, const void *pvBits, KLDRADDR BaseAddress, PKLDRSTACKINFO pStackInfo)
+{
+    PKLDRMODPE pModPE = (PKLDRMODPE)pMod->pvData;
+
+    pStackInfo->Address = NIL_KLDRADDR;
+    pStackInfo->LinkAddress = NIL_KLDRADDR;
+    pStackInfo->cbStack = pStackInfo->cbStackThread = pModPE->Hdrs.OptionalHeader.SizeOfStackReserve;
+
+    return 0;
+}
+
+
 /** @copydoc kLdrModQueryMainEntrypoint */
-int (* pfnQueryMainEntrypoint)(PKLDRMOD pMod, void *pvBits, KLDRADDR BaseAddress, PKLDRADDR pMainEPAddress);
+static int kldrModPEQueryMainEntrypoint(PKLDRMOD pMod, const void *pvBits, KLDRADDR BaseAddress, PKLDRADDR pMainEPAddress)
+{
+    PKLDRMODPE pModPE = (PKLDRMODPE)pMod->pvData;
+    int rc;
+
+    rc = kldrModPEBitsAndBaseAddress(pModPE, NULL, &BaseAddress);
+    if (rc)
+        return rc;
+
+    *pMainEPAddress = pModPE->Hdrs.OptionalHeader.AddressOfEntryPoint
+        ? BaseAddress + pModPE->Hdrs.OptionalHeader.AddressOfEntryPoint
+        : NIL_KLDRADDR;
+    return 0;
+}
+
+
 /** @copydoc kLdrModEnumDbgInfo */
-int (* pfnEnumDbgInfo)(PKLDRMOD pMod, void *pvBits, PFNKLDRENUMDBG pfnCallback, void *pvUser);
+int (* pfnEnumDbgInfo)(PKLDRMOD pMod, const void *pvBits, PFNKLDRENUMDBG pfnCallback, void *pvUser);
+
 /** @copydoc kLdrModHasDbgInfo */
-int (* pfnHasDbgInfo)(PKLDRMOD pMod, void *pvBits);
+int (* pfnHasDbgInfo)(PKLDRMOD pMod, const void *pvBits);
 /** @copydoc kLdrModMap */
 int (* pfnMap)(PKLDRMOD pMod);