Changeset 2858

Timestamp:

Nov 6, 2006, 1:23:14 AM (19 years ago)

Author:

bird

Message:

bugfixing - dinner break.

Location:

trunk/kLdr

Files:

• Makefile.kmk (modified) (2 diffs)
• kLdr-win.def (modified) (1 diff)
• kLdr.h (modified) (1 diff)
• kLdrHlp.h (modified) (1 diff)
• kLdrInternal.h (modified) (1 diff)
• kLdrMod.c (modified) (1 diff)
• kLdrModPE.c (modified) (5 diffs)
• kLdrRdrFile.c (modified) (11 diffs)

trunk/kLdr/Makefile.kmk

@@ -68 +68 @@
  kLdr_TOOL = GCC3
  kLdr_TOOL = VCC70
- kLdr_CFLAGS = -W3 -Zl
+ kLdr_CFLAGS = -W3 -Zl -ML
  kLdr_ASFLAGS = -f win
- kLdr_LDFLAGS = -Entry:DllMain@12
+ kLdr_LDFLAGS = -Entry:DllMain@12 -Debug
  kLdr_DEFS = __WIN__
  kLdr_SDKS = WIN32SDK
  #kLdr_SDKS.x86 = WIN32SDK
  #kLdr_SDKS.amd64 = WIN64SDK
+ kLdr_LIBS = \
+        $$(PATH_TOOL_VCC70_LIB)/LIBC.lib
 else
  ifneq ($(filter os2,$(BUILD_TARGET)),)
@@ -138 +140 @@
         kLdrHlpHeap.c \
 
+#
+# Heap testcase.
+#
+PROGRAMS += tstkLdrMod
+tstkLdrMod_TEMPLATE = TST
+tstkLdrMod_SOURCES = \
+        tstkLdrMod.c
+ifneq ($(filter win win32 win64 nt,$(BUILD_TARGET)),)
+tstkLdrMod_LIBS = $(TARGET_kLdr:.dll=.lib)
+else
+tstkLdrMod_LIBS = $(TARGET_kLdr)
+endif
+
+
 
 # generate rules.

trunk/kLdr/kLdr-win.def

@@ -38 +38 @@
     kLdrRdrTell
     kLdrRdrName
+    kLdrRdrPageSize
+    kLdrRdrPrepare
+    kLdrRdrMap
+    kLdrRdrRefreshMap
+    kLdrRdrProtect
+    kLdrRdrUnmap
+    kLdrRdrUnprepare
+    kLdrRdrDone
 
     ; The module interpreter API
+    kLdrModOpen
     kLdrModOpenFromRdr
-    kLdrModOpen
+    kLdrModOpenNative
+    kLdrModClose
+    kLdrModQuerySymbol
+    kLdrModEnumSymbols
+    kLdrModGetImport
+    kLdrModNumberOfImports
+    kLdrModCanExecuteOn
+    kLdrModGetStackInfo
+    kLdrModQueryMainEntrypoint
+    kLdrModEnumDbgInfo
+    kLdrModHasDbgInfo
+    kLdrModMap
+    kLdrModUnmap
+    kLdrModAllocTLS
+    kLdrModFreeTLS
+    kLdrModReload
+    kLdrModFixupMapping
+    kLdrModCallInit
+    kLdrModCallTerm
+    kLdrModCallThread
+    kLdrModSize
+    kLdrModGetBits
+    kLdrModRelocateBits
 
     ; Process Bootstrapping

trunk/kLdr/kLdr.h

@@ -928 +928 @@
  */
 #ifdef __OS2__
-# define KLDR_ERR_BASE          0x7face000
+# define KLDR_ERR_BASE          420000
 #elif defined(__WIN__)
-# define KLDR_ERR_BASE          0x7face000
+# define KLDR_ERR_BASE          420000
 #else
 # error "port me"

trunk/kLdr/kLdrHlp.h

@@ -38 +38 @@
 #define KLDR_MAX(a, b) ((a) >= (b) ? (a) : (b))
 /** Calculate the offset of a structure member. */
-#define KLDR_OFFSETOF(strct, memb)  ( (size_t)( ((strct *)0)->memb ) )
+#define KLDR_OFFSETOF(strct, memb)  ( (size_t)( &((strct *)0)->memb ) )
 /** Align a size_t value. */
 #define KLDR_ALIGN_Z(val, align)    ( ((val) + ((align) - 1)) & ~(size_t)((align) - 1) )

trunk/kLdr/kLdrInternal.h

@@ -403 +403 @@
 
 
+/** @name Module interpreter method tables
+ * @{ */
+extern KLDRMODOPS       g_kLdrModPEOps;
+/** @} */
+
 /** @} */
 #ifdef __cplusplus

trunk/kLdr/kLdrMod.c

@@ -183 +183 @@
         rc = KLDR_ERR_LE_NOT_SUPPORTED;
     else if (u.u32 == IMAGE_NT_SIGNATURE)
-        rc = KLDR_ERR_PE_NOT_SUPPORTED;
+        rc = g_kLdrModPEOps.pfnCreate(&g_kLdrModPEOps, pRdr, offHdr, ppMod);
     else if (u.u32 == IMAGE_ELF_SIGNATURE)
         rc = KLDR_ERR_ELF_NOT_SUPPORTED;

trunk/kLdr/kLdrModPE.c

@@ -204 +204 @@
      */
     cchFilename = kLdrHlpStrLen(kLdrRdrName(pRdr));
-    cb = sizeof(KLDRMOD)
-       + s.FileHdr.NumberOfSections * sizeof(KLDRSEG)
-       + sizeof(KLDRMODPE) - sizeof(IMAGE_SECTION_HEADER)
-       + s.FileHdr.NumberOfSections * sizeof(IMAGE_SECTION_HEADER)
+    cb = KLDR_ALIGN_Z(KLDR_OFFSETOF(KLDRMODPE, aShdrs[s.FileHdr.NumberOfSections]), 16)
+       + KLDR_OFFSETOF(KLDRMOD, aSegments[s.FileHdr.NumberOfSections + 1])
        + cchFilename + 1;
     pModPE = (PKLDRMODPE)kldrHlpAlloc(cb);
     if (!pModPE)
         return KLDR_ERR_NO_MEMORY;
+    *ppModPE = pModPE;
 
     /* KLDRMOD */
-    pMod = (PKLDRMOD)((uint8_t *)pModPE + sizeof(KLDRMOD) + (s.FileHdr.NumberOfSections + 1) * sizeof(KLDRSEG));
+    pMod = (PKLDRMOD)((uint8_t *)pModPE + KLDR_ALIGN_Z(KLDR_OFFSETOF(KLDRMODPE, aShdrs[s.FileHdr.NumberOfSections]), 16));
     pMod->pvData = pModPE;
     pMod->pRdr = pRdr;
@@ -252 +251 @@
     pModPE->Hdrs.Signature = s.Signature;
     pModPE->Hdrs.FileHeader = s.FileHdr;
-    *ppModPE = pModPE;
 
     /*
@@ -811 +809 @@
 
 /** @copydoc kLdrModEnumSymbols */
-static int kldrModPEEnumSymbols(PKLDRMOD pMod, uint32_t fFlags, const void *pvBits, KLDRADDR BaseAddress,
-                                PFNKLDRMODENUMSYMS pfnCallback, void *pvUser)
+static int kldrModPEEnumSymbols(PKLDRMOD pMod, const void *pvBits, KLDRADDR BaseAddress,
+                                uint32_t fFlags, PFNKLDRMODENUMSYMS pfnCallback, void *pvUser)
 {
     PKLDRMODPE                      pModPE = (PKLDRMODPE)pMod->pvData;
@@ -890 +888 @@
 
 /** @copydoc kLdrModGetImport */
-static int kldrModPEGetImport(PKLDRMOD pMod, void *pvBits, uint32_t iImport, char *pszName, size_t cchName)
+static int kldrModPEGetImport(PKLDRMOD pMod, const void *pvBits, uint32_t iImport, char *pszName, size_t cchName)
 {
     PKLDRMODPE                      pModPE = (PKLDRMODPE)pMod->pvData;
@@ -1940 +1938 @@
 }
 
+
+/**
+ * The PE module interpreter method table.
+ */
+KLDRMODOPS g_kLdrModPEOps =
+{
+    "PE",
+    NULL,
+    kldrModPECreate,
+    kldrModPEDestroy,
+    kldrModPEQuerySymbol,
+    kldrModPEEnumSymbols,
+    kldrModPEGetImport,
+    kldrModPENumberOfImports,
+    NULL /* can execute one is optional */,
+    kldrModPEGetStackInfo,
+    kldrModPEQueryMainEntrypoint,
+    kldrModPEEnumDbgInfo,
+    kldrModPEHasDbgInfo,
+    kldrModPEMap,
+    kldrModPEUnmap,
+    kldrModPEAllocTLS,
+    kldrModPEFreeTLS,
+    kldrModPEReload,
+    kldrModPEFixupMapping,
+    kldrModPECallInit,
+    kldrModPECallTerm,
+    kldrModPECallThread,
+    kldrModPESize,
+    kldrModPEGetBits,
+    kldrModPERelocateBits,
+    42 /* the end */
+};

trunk/kLdr/kLdrRdrFile.c

@@ -232 +232 @@
 
         SetLastError(0);
-        offLow = SetFilePointer(pRdrFile->File, 0, &offHigh, FILE_BEGIN);
+        offLow = SetFilePointer(pRdrFile->File, 0, &offHigh, FILE_CURRENT);
         rc = GetLastError();
         if (rc)
@@ -325 +325 @@
 #ifdef __OS2__
         ULONG ulNew;
-        APIRET rc = DosSetFilePtr(pRdrFile->File, off, FILE_BEGIN, &ulNew);
+        APIRET rc;
+
+        rc = DosSetFilePtr(pRdrFile->File, off, FILE_BEGIN, &ulNew);
         if (rc)
         {
@@ -333 +335 @@
 
 #elif defined(__WIN__)
-        LONG offHigh = (LONG)(off >> 32);
-        LONG offLow = SetFilePointer(pRdrFile->File, (LONG)off, &offHigh, FILE_BEGIN);
+        LONG offHigh;
+        LONG offLow;
+
+        offHigh = sizeof(off_t) == 4 ? 0 : (off >> 32);
+        offLow = SetFilePointer(pRdrFile->File, (LONG)off, &offHigh, FILE_BEGIN);
         if (    offLow != (LONG)off
-            &&  offHigh != (LONG)(off >> 32))
+            ||  offHigh != (LONG)(sizeof(off_t) == 4 ? 0 : (off >> 32)))
         {
             int rc = GetLastError();
@@ -405 +410 @@
 
 #elif defined(__WIN__)
+    rc = 0;
     if (!CloseHandle(pRdrFile->File))
         rc = GetLastError();
@@ -433 +439 @@
      */
 #ifdef __OS2__
-    ULONG ulAction = 0;
-    FILESTATUS3 Info;
-    APIRET rc;
-    HFILE File = 0;
-    off_t cb;
+    ULONG           ulAction = 0;
+    FILESTATUS3     Info;
+    APIRET          rc;
+    HFILE           File = 0;
+    off_t           cb;
+    char            szFilename[CCHMAXPATH];
 
     if ((uintptr_t)pszFilename >= 0x20000000)
@@ -451 +458 @@
     if (rc)
         return rc;
+
+    rc = DosQueryPathInfo(pszFilename, FIL_QUERYFULLNAME, szFilename, sizeof(szFilename));
+    if (rc)
+    {
+        DosClose(File);
+        return rc;
+    }
 
     rc = DosQueryFileInfo(File, FIL_STANDARD, &Info, sizeof(Info));
@@ -467 +481 @@
     HANDLE              File;
     off_t               cb;
+    char                szFilename[MAX_PATH];
 
     SecAttr.bInheritHandle = FALSE;
@@ -475 +490 @@
         return GetLastError();
 
+    if (!GetFullPathName(pszFilename, sizeof(szFilename), szFilename, NULL))
+    {
+        rc = GetLastError();
+        CloseHandle(File);
+        return rc;
+    }
+
     SetLastError(0);
     Low = GetFileSize(File, &High);
@@ -483 +505 @@
         return rc;
     }
-    cb = ((off_t)High << 32) | Low;
+    if (sizeof(off_t) == 4)
+        cb = High ? 0x7fffffff : Low;
+    else
+        cb = ((off_t)High << 32) | Low;
 
 #else
@@ -493 +518 @@
      * Allocate the reader instance.
      */
-    cchFilename = kLdrHlpStrLen(pszFilename);
+    cchFilename = kLdrHlpStrLen(szFilename);
     pRdrFile = (PKLDRRDRFILE)kldrHlpAlloc(sizeof(*pRdrFile) + cchFilename);
     if (!pRdrFile)
@@ -513 +538 @@
      * Initialize it and return successfully.
      */
+    pRdrFile->Core.u32Magic = KLDRRDR_MAGIC;
     pRdrFile->Core.pOps = &g_kLdrRdrFileOps;
     pRdrFile->File = File;
+    pRdrFile->cb = cb;
     pRdrFile->off = 0;
-    kLdrHlpMemCopy(&pRdrFile->szFilename[0], pszFilename, cchFilename + 1);
+    kLdrHlpMemCopy(&pRdrFile->szFilename[0], szFilename, cchFilename + 1);
 
     *ppRdr = &pRdrFile->Core;