Changeset 159 for trunk/nom/src

Timestamp:

Dec 31, 2006, 10:46:48 AM (19 years ago)

Author:

cinc

Message:

Added functions to register DLLs with the garbage collector.

File:

• trunk/nom/src/nomgc.c (modified) (2 diffs)

trunk/nom/src/nomgc.c

@@ -42 +42 @@
 #include <os2.h>
 
-#include <stdarg.h>
 #include <stdio.h>
 #include <string.h>
+#include <stdlib.h>
 
 /* For nomToken etc. */
@@ -103 +103 @@
 }
 
+
+/*
+  Find a library record in the buffer filled by DosQuerySysState().
+ */
+static qsLrec_t* qsFindModuleRec(const HREGDLL hRegisterDLL,  USHORT hMod){
+  qsLrec_t *       pModRec;
+  int a=0;
+
+  pModRec=hRegisterDLL->pLibRec;
+  while(NULL!=pModRec)
+    {
+      a++;
+      //            printf("%d Checking: %x -> %04X (%s)\n", a, pModRec, pModRec->hmte, pModRec->pName);
+
+      if (NULLHANDLE==pModRec->pObjInfo   && pModRec->ctObj > 0)
+        {
+          pModRec->pObjInfo = (qsLObjrec_t*)((char*)pModRec
+                                             + ((sizeof(qsLrec_t)
+                                             + pModRec->ctImpMod * sizeof(short)
+                                             + strlen((char*)pModRec->pName) + 1    /* filename */
+                                             + 3) & ~3));
+          pModRec->pNextRec = (void*)((char*)pModRec->pObjInfo
+                                    + sizeof(qsLObjrec_t) * pModRec->ctObj);
+        }
+      if(pModRec->hmte==hMod)
+        break;
+
+      pModRec=(qsLrec_t *)pModRec->pNextRec;
+    }
+  return pModRec;
+}
+
+
+#define BUFSIZE 1024*1024
+NOMEXTERN HREGDLL NOMLINK nomBeginRegisterDLLWithGC(void)
+{
+  ULONG rc;
+  HREGDLL hReg=NULLHANDLE;
+  PTIB     ptib;
+  PPIB     ppib;
+  char *  buf;
+
+  rc = DosGetInfoBlocks(&ptib, &ppib);
+  if (rc!=NO_ERROR)
+    return NULLHANDLE;
+
+  buf = malloc(BUFSIZE);
+  if(!buf)
+    return NULLHANDLE;
+
+  memset(buf,0,BUFSIZE);
+
+  rc = DosQuerySysState(QS_PROCESS | QS_SEMAPHORE | QS_MTE | QS_FILESYS | QS_SHMEMORY ,
+                        QS_MTE, /*0x96*/ ppib->pib_ulpid , 1UL, (PCHAR)buf, BUFSIZE);
+  if (rc==NO_ERROR) {
+    hReg=(qsPtrRec_t*) buf;
+  }
+  else
+    free(buf);
+
+  return hReg;
+}
+
+NOMEXTERN void NOMLINK nomEndRegisterDLLWithGC(const HREGDLL hRegisterDLL )
+{
+  free((char*)hRegisterDLL);
+}
+
+
+/*
+  FIXME:
+
+  This function will not find every given DLL because it doesn't follow every import of
+  each DLL.
+  It's only meant for registering the GTK+, GLIB and friends DLLs. This works because GTK2.DLL
+  is directly loaded by the exe and all the friends DLLs are imported by it.
+
+  Feel free to make this function really useful...
+
+  Oh, and some refactoring would be nice, too.
+
+ */
+#define OBJREAD         0x0001L
+#define OBJWRITE        0x0002L
+#define OBJINVALID      0x0080L
+NOMEXTERN BOOL NOMLINK nomRegisterDLLByName(const HREGDLL hRegisterDLL, const char* chrDLLName)
+{
+  qsPrec_t * p;
+  int a=0;
+
+  printf("Trying to register DLL %s\n", chrDLLName);
+
+  p=hRegisterDLL->pProcRec;
+  while(p && p->RecType == 1)
+    {
+      a++;
+      if (p->cLib) {
+        int i;
+
+        for (i=0; i<p->cLib; i++){
+          qsLrec_t * pModRec;
+
+          //printf("%d %04X (p: %04x %04X, %04X) ",i, p->pLibRec[i], p, &p->pLibRec[i], &p->pLibRec);
+
+          pModRec=qsFindModuleRec(hRegisterDLL,  p->pLibRec[i]);
+          if(pModRec){
+            //  printf("DLL name: %s\n", pModRec->pName);
+            if(NULLHANDLE!=strstr( pModRec->pName, chrDLLName))
+              {
+                qsLObjrec_t  *pObjInfo;
+                printf("    --> Found DLL %s\n", pModRec->pName);
+                pObjInfo=pModRec->pObjInfo;
+                if(NULLHANDLE!=pObjInfo)
+                  {
+                    int iObj;
+                    for(iObj=0; iObj<pModRec->ctObj ;iObj++)
+                      {
+                        if (!(pObjInfo[iObj].oflags & OBJWRITE)) continue;
+                        if (!(pObjInfo[iObj].oflags & OBJREAD)) continue;
+                        if ((pObjInfo[iObj].oflags & OBJINVALID)) continue;
+                        printf("    #%d: %04lX, size: %04lX %04lX\n",
+                               iObj, pObjInfo[iObj].oaddr, pObjInfo[iObj].osize, pObjInfo[iObj].oflags); 
+                        nomRegisterDataAreaForGC((char*)pObjInfo[iObj].oaddr,
+                                                 (char*)(pObjInfo[iObj].oaddr+pObjInfo[iObj].osize));
+
+                      }
+                  }
+                return TRUE;
+              }
+            /* Check the imports of this DLL if any */
+            if(pModRec->ctImpMod >0)
+              {
+                int iImps;
+                PUSHORT   pImpHmte;
+
+                pImpHmte=(PUSHORT)((void*)pModRec + sizeof(qsLrec_t));
+
+                for(iImps=0; iImps < pModRec->ctImpMod; iImps++)
+                  {
+                    qsLrec_t * pModImp;
+                    // printf("  Trying import #%d (%04X)\n", iImps, pImpHmte[iImps]);
+                    pModImp=qsFindModuleRec(hRegisterDLL,  pImpHmte[iImps]);
+                    if(pModImp){
+                      //printf("  DLL name: %s\n", pModImp->pName);
+                      if(NULLHANDLE!=strstr( pModImp->pName, chrDLLName))
+                        {
+                          qsLObjrec_t  *pObjInfo;
+                          printf("    --> Found DLL %s\n", pModImp->pName);
+                          pObjInfo=pModImp->pObjInfo;
+                          if(NULLHANDLE!=pObjInfo)
+                            {
+                              int iObj;
+                              for(iObj=0; iObj<pModImp->ctObj ;iObj++)
+                                {
+                                  if (!(pObjInfo[iObj].oflags & OBJWRITE)) continue;
+                                  if (!(pObjInfo[iObj].oflags & OBJREAD)) continue;
+                                  if ((pObjInfo[iObj].oflags & OBJINVALID)) continue;
+                                  
+                                  printf("    #%d: %04lX, size: %04lX %04lX\n",
+                                         iObj, pObjInfo[iObj].oaddr, pObjInfo[iObj].osize, pObjInfo[iObj].oflags); 
+                                  nomRegisterDataAreaForGC((char*)pObjInfo[iObj].oaddr,
+                                                           (char*)(pObjInfo[iObj].oaddr+pObjInfo[iObj].osize));
+                                }
+                            }
+                          return TRUE;
+                        }
+                    }/* for() */
+                  }/* for()*/
+              }/* if() */
+          }
+        }/* For() */
+      }
+      break;
+    }
+  return NULLHANDLE;
+}