Changeset 2860 for trunk/kLdr/kLdrRdrFile.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/kLdrRdrFile.c (modified) (8 diffs)

trunk/kLdr/kLdrRdrFile.c

@@ -52 +52 @@
 *******************************************************************************/
 /**
+ * Prepared stuff.
+ */
+typedef struct KLDRRDRFILEPREP
+{
+    /** The address of the prepared region. */
+    void           *pv;
+    /** The size of the prepared region. */
+    size_t          cb;
+#if defined(__WIN__) || defined(__NT__)
+    /** Handle to the section created to map the file. */
+    HANDLE          hSection;
+#endif
+} KLDRRDRFILEPREP, *PKLDRRDRFILEPREP;
+
+/**
  * The file provier instance for native files.
  */
@@ -57 +72 @@
 {
     /** The file reader vtable. */
-    KLDRRDR     Core;
+    KLDRRDR             Core;
     /** The file handle. */
 #ifdef __OS2__
-    HFILE       File;
+    HFILE               File;
 #elif defined(__WIN__) || defined(__NT__)
-    HANDLE      File;
+    HANDLE              File;
 #else
 # error "Port me!"
 #endif
     /** The current file offset. */
-    off_t       off;
+    off_t               off;
     /** The file size. */
-    off_t       cb;
+    off_t               cb;
+    /** Array where we stuff the mapping area data. */
+    KLDRRDRFILEPREP     aPreps[4];
+    /** The number of current preps. */
+    uint32_t            cPreps;
     /** Number of mapping references. */
-    int32_t     cMappings;
+    int32_t             cMappings;
     /** The memory mapping. */
-    void       *pvMapping;
+    void               *pvMapping;
     /** The filename. */
-    char        szFilename[1];
+    char                szFilename[1];
 } KLDRRDRFILE, *PKLDRRDRFILE;
 
@@ -133 +152 @@
 }
 
+
+/**
+ * Finds a prepared mapping region.
+ *
+ * @returns Pointer to the aPrep entry.
+ * @param   pFile   The instance data.
+ * @param   pv      The base of the region.
+ * @param   cb      The size of the region.
+ */
+static PKLDRRDRFILEPREP kldrRdrFileFindPrepExact(PKLDRRDRFILE pFile, void *pv, size_t cb)
+{
+    int32_t i = pFile->cPreps;
+    while (i-- > 0)
+        if (    pFile->aPreps[i].pv == pv
+            ||  pFile->aPreps[i].cb == cb)
+            return &pFile->aPreps[i];
+    return NULL;
+}
+
+
+/**
+ * Finds a prepared mapping region containing the specified region.
+ *
+ * @returns Pointer to the aPrep entry.
+ * @param   pFile   The instance data.
+ * @param   pv      The base of the sub region.
+ * @param   cb      The size of the sub region.
+ */
+static PKLDRRDRFILEPREP kldrRdrFileFindPrepWithin(PKLDRRDRFILE pFile, void *pv, size_t cb)
+{
+    int32_t i = pFile->cPreps;
+    while (i-- > 0)
+        if ((uintptr_t)pv - (uintptr_t)pFile->aPreps[i].pv < pFile->aPreps[i].cb)
+        {
+            if ((uintptr_t)pv - (uintptr_t)pFile->aPreps[i].pv + cb <= pFile->aPreps[i].cb)
+                return &pFile->aPreps[i];
+            return NULL;
+        }
+    return NULL;
+}
+
+
 /** @copydoc KLDRRDR::pfnUnprepare */
 static int      kldrRdrFileUnprepare(PKLDRRDR pRdr, void *pv, size_t cb)
 {
+    PKLDRRDRFILE        pRdrFile = (PKLDRRDRFILE)pRdr;
+    PKLDRRDRFILEPREP    pPrep = kldrRdrFileFindPrepExact(pRdrFile, pv, cb);
+    if (!pPrep)
+        return KLDR_ERR_INVALID_PARAMETER;
+
     return -1;
 }
@@ -143 +209 @@
 static int      kldrRdrFileUnmap(PKLDRRDR pRdr, void *pv, size_t cb)
 {
+    PKLDRRDRFILE        pRdrFile = (PKLDRRDRFILE)pRdr;
+    PKLDRRDRFILEPREP    pPrep = kldrRdrFileFindPrepExact(pRdrFile, pv, cb);
+    if (!pPrep)
+        return KLDR_ERR_INVALID_PARAMETER;
+
     return -1;
 }
@@ -150 +221 @@
 static int      kldrRdrFileProtect(PKLDRRDR pRdr, void *pv, size_t cb, KLDRPROT enmProt)
 {
+    PKLDRRDRFILE        pRdrFile = (PKLDRRDRFILE)pRdr;
+    PKLDRRDRFILEPREP    pPrep = kldrRdrFileFindPrepExact(pRdrFile, pv, cb);
+    if (!pPrep)
+        return KLDR_ERR_INVALID_PARAMETER;
+
     return -1;
 }
@@ -157 +233 @@
 static int      kldrRdrFileRefreshMap(PKLDRRDR pRdr, void *pv, size_t cb, KLDRPROT enmProt, off_t offFile, size_t cbFile)
 {
+    PKLDRRDRFILE        pRdrFile = (PKLDRRDRFILE)pRdr;
+    PKLDRRDRFILEPREP    pPrep = kldrRdrFileFindPrepExact(pRdrFile, pv, cb);
+    if (!pPrep)
+        return KLDR_ERR_INVALID_PARAMETER;
+
     return -1;
 }
@@ -164 +245 @@
 static int      kldrRdrFileMap(PKLDRRDR pRdr, void *pv, size_t cb, KLDRPROT enmProt, off_t offFile, size_t cbFile)
 {
+    PKLDRRDRFILE        pRdrFile = (PKLDRRDRFILE)pRdr;
+    PKLDRRDRFILEPREP    pPrep = kldrRdrFileFindPrepExact(pRdrFile, pv, cb);
+    if (!pPrep)
+        return KLDR_ERR_INVALID_PARAMETER;
+
     return -1;
 }
@@ -543 +629 @@
     pRdrFile->cb = cb;
     pRdrFile->off = 0;
+    pRdrFile->cMappings = 0;
+    pRdrFile->cPreps = 0;
     kLdrHlpMemCopy(&pRdrFile->szFilename[0], szFilename, cchFilename + 1);