Changeset 2860 for trunk/kLdr/kLdrModPE.c

Timestamp:

Nov 8, 2006, 5:10:14 AM (19 years ago)

Author:

bird

Message:

Working on the mapping stuff (windows is gonna be a headache).

File:

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

trunk/kLdr/kLdrModPE.c

@@ -551 +551 @@
 static int kldrModPEDoMap(PKLDRMODPE pModPE, unsigned fForReal)
 {
-    return -1;
+    PKLDRMOD    pMod = pModPE->pMod;
+    unsigned    fFixed;
+    void       *pvBase;
+    size_t      cb;
+    int         rc;
+    uint32_t    i;
+
+    /*
+     * Prepare it.
+     */
+    /* size */
+    cb = (size_t)pMod->pOps->pfnSize(pModPE->pMod);
+    if (cb != pMod->pOps->pfnSize(pModPE->pMod))
+        return KLDR_ERR_ADDRESS_OVERFLOW;
+
+    /* fixed image? */
+    fFixed = fForReal
+          && (   pMod->enmType == KLDRTYPE_EXECUTABLE_FIXED
+              || pMod->enmType == KLDRTYPE_SHARED_LIBRARY_FIXED);
+    if (!fFixed)
+        pvBase = NULL;
+    else
+    {
+        pvBase = (void *)(uintptr_t)pMod->aSegments[0].LinkAddress;
+        if ((uintptr_t)pvBase != pMod->aSegments[0].LinkAddress)
+            return KLDR_ERR_ADDRESS_OVERFLOW;
+    }
+
+    /* try do the prepare */
+    rc = kLdrRdrPrepare(pMod->pRdr, &pvBase, cb, fFixed);
+    if (rc)
+        return rc;
+
+    /*
+     * Map the segments (sub sections in PE terms).
+     */
+    for (i = 0; i < pMod->cSegments; i++)
+    {
+        void *pvSeg;
+
+        if (!pMod->aSegments[i].Alignment)
+            continue;
+
+        pvSeg = (uint8_t *)pvBase + pMod->aSegments[i].RVA;
+        rc = kLdrRdrMap(pMod->pRdr, pvSeg, pMod->aSegments[i].cbMapped, pMod->aSegments[i].enmProt,
+                        pMod->aSegments[i].offFile, pMod->aSegments[i].cbFile);
+        if (rc)
+        {
+            /* bailout */
+            while (i-- > 0)
+            {
+                if (!pMod->aSegments[i].Alignment)
+                    continue;
+
+                kLdrRdrUnmap(pMod->pRdr, (void *)pMod->aSegments[i].MapAddress, pMod->aSegments[i].cbMapped);
+                pMod->aSegments[i].MapAddress = 0;
+            }
+            break;
+        }
+        pMod->aSegments[i].MapAddress = (uintptr_t)pvSeg;
+    }
+
+    if (!rc)
+    {
+        if (fForReal)
+            pModPE->pvMapping = pvBase;
+        else
+            pModPE->pvBits = pvBase;
+    }
+    else
+        kLdrRdrUnprepare(pMod->pRdr, pvBase, cb);
+    return rc;
 }
 
@@ -567 +638 @@
 static int kldrModPEDoUnmap(PKLDRMODPE pModPE, const void *pvMapping)
 {
-    return -1;
+    PKLDRMOD    pMod = pModPE->pMod;
+    size_t      cb = (size_t)pMod->pOps->pfnSize(pModPE->pMod);
+    int         rc2;
+    int         rc = 0;
+    uint32_t    i;
+
+    /*
+     * Unmap the segments (sub sections in PE terms).
+     */
+    for (i = 0; i < pMod->cSegments; i++)
+    {
+        if (!pMod->aSegments[i].MapAddress)
+            continue;
+
+        rc2 = kLdrRdrUnmap(pMod->pRdr, (void *)pMod->aSegments[i].MapAddress, pMod->aSegments[i].cbMapped);
+        if (!rc2)
+            pMod->aSegments[i].MapAddress = 0;
+        else if (!rc)
+            rc = rc2;
+    }
+
+    /*
+     * 'Unprepare' the mapping region.
+     */
+    if (!rc)
+    {
+        rc = kLdrRdrUnprepare(pMod->pRdr, (void *)pvMapping, cb);
+        if (!rc)
+        {
+            if (pModPE->pvBits == pvMapping)
+                pModPE->pvBits = NULL;
+            if (pModPE->pvMapping == pvMapping)
+                pModPE->pvMapping = NULL;
+        }
+    }
+
+    return rc;
 }