Changeset 2870 for trunk/kLdr/testcase/tst-0-driver.c

Timestamp:

Nov 12, 2006, 6:38:28 AM (19 years ago)

Author:

bird

Message:

Test all kLdrDyld apis.

File:

• trunk/kLdr/testcase/tst-0-driver.c (modified) (4 diffs)

trunk/kLdr/testcase/tst-0-driver.c

@@ -30 +30 @@
 *******************************************************************************/
 #include <kLdr.h>
+#include "tst.h"
 #include <stdarg.h>
 #include <stdio.h>
@@ -35 +36 @@
 #include <string.h>
 
+
+/*******************************************************************************
+*   Defined Constants And Macros                                               *
+*******************************************************************************/
+/** Select the appropriate KLDRSYMKIND bit define. */
+#define MY_KLDRSYMKIND_BITS     ( sizeof(void *) == 4 ? KLDRSYMKIND_32BIT : KLDRSYMKIND_64BIT )
 
 
@@ -67 +74 @@
     const char *pszErrInit = "Error, szErr wasn't zapped";
     char szErr[512];
+    char szBuf[512];
+    char *psz;
+    size_t cch;
     HKLDRMOD hMod;
     int rc;
 
+    /*
+     * The first thing to do is a simple load / unload test
+     * using the tst-0-a library (it'll drag in tst-0-d).
+     */
+    printf("tst-0-driver: Basic API test using 'tst-0-a'...\n");
     hMod = (HKLDRMOD)0xffffeeee;
     strcpy(szErr, pszErrInit);
@@ -84 +99 @@
     if (!rc)
     {
+        HKLDRMOD hMod2;
+        HKLDRMOD hMod3;
+        printf("tst-0-driver: hMod=%p ('tst-0-a')\n", (void *)hMod);
+
+        /*
+         * Simple test of kLdrDyldFindByName.
+         */
+        hMod2 = (HKLDRMOD)0xffffeeee;
+        rc = kLdrDyldFindByName("tst-0", NULL, NULL, KLDRDYLD_SEARCH_HOST, 0, &hMod2);
+        if (!rc)
+            Failure("kLdrDyldFindByName(\"tst-0\",,,) didn't fail!\n");
+        if (rc && hMod2 != NIL_HKLDRMOD)
+            Failure("hMod2 wasn't set correctly on kLdrDyldFindByName failure!\n");
+
+        hMod2 = (HKLDRMOD)0xffffeeee;
+        rc = kLdrDyldFindByName("tst-0-a", NULL, NULL, KLDRDYLD_SEARCH_HOST, 0, &hMod2);
+        if (rc)
+            Failure("kLdrDyldFindByName(\"tst-0-a\",,,) failed, rc=%d (%#x)\n", rc, rc);
+        if (!rc && hMod2 != hMod)
+            Failure("kLdrDyldFindByName(\"tst-0-a\",,,) returned the wrong module handle: %p instead of %p\n",
+                    (void *)hMod2, (void *)hMod);
+
+        hMod2 = (HKLDRMOD)0xffffeeee;
+        rc = kLdrDyldFindByName("tst-0-d", NULL, NULL, KLDRDYLD_SEARCH_HOST, 0, &hMod2);
+        if (!rc)
+            printf("tst-0-driver: hMod2=%p ('tst-0-d')\n", (void *)hMod2);
+        else
+            Failure("kLdrDyldFindByName(\"tst-0-d\",,,) failed, rc=%d (%#x)\n", rc, rc);
+
+        /*
+         * Get the name and filename for each of the two modules.
+         */
+        rc = kLdrDyldGetName(hMod2, szBuf, sizeof(szBuf));
+        if (!rc)
+        {
+            printf("tst-0-driver: name: '%s' ('tst-0-d')\n", szBuf);
+            psz = strstr(szBuf, "-0-");
+            if (    !psz
+                ||  strnicmp(psz, "-0-d", sizeof("-0-d") - 1))
+                Failure("kLdrDyldGetName(\"tst-0-d\",,,) -> '%s': pattern '-0-d' not found\n", szBuf);
+
+            /* overflow test. */
+            cch = strlen(szBuf);
+            szBuf[cch + 1] = szBuf[cch] = szBuf[cch - 1] = 'x';
+            szBuf[cch + 2] = '\0';
+            rc = kLdrDyldGetName(hMod2, szBuf, cch);
+            if (rc == KLDR_ERR_BUFFER_OVERFLOW)
+            {
+                if (!szBuf[0])
+                    Failure("kLdrDyldGetName didn't return partial result on overflow\n");
+                else if (szBuf[cch - 1])
+                    Failure("kLdrDyldGetName didn't terminate partial result correctly overflow: '%s'\n", szBuf);
+                else if (szBuf[cch] != 'x')
+                    Failure("kLdrDyldGetName exceeded the buffer limit on partial overflow: '%s'\n", szBuf);
+            }
+            else
+                Failure("kLdrDyldGetName(\"tst-0-d\",,,) -> rc=%d (%#x) instead of KLDR_ERR_BUFFER_OVERFLOW\n", rc, rc);
+
+            /* check that we can query the module by the returned name. */
+            rc = kLdrDyldGetName(hMod2, szBuf, sizeof(szBuf));
+            if (!rc)
+            {
+                hMod3 = (HKLDRMOD)0xffffeeee;
+                rc = kLdrDyldFindByName(szBuf, NULL, NULL, KLDRDYLD_SEARCH_HOST, 0, &hMod3);
+                if (rc || hMod3 != hMod2)
+                    Failure("kLdrDyldFindByName(\"%s\",,,) failed, rc=%d (%#x) hMod3=%p hMod2=%p\n",
+                            szBuf, rc, rc, (void *)hMod3, (void *)hMod2);
+            }
+            else
+                Failure("kLdrDyldGetName(\"tst-0-d\",,,) failed (b), rc=%d (%#x)\n", rc, rc);
+        }
+        else
+            Failure("kLdrDyldGetName(\"tst-0-d\",,,) failed, rc=%d (%#x)\n", rc, rc);
+
+        rc = kLdrDyldGetFilename(hMod2, szBuf, sizeof(szBuf));
+        if (!rc)
+        {
+            printf("tst-0-driver: filename: '%s' ('tst-0-d')\n", szBuf);
+
+            /* overflow test. */
+            cch = strlen(szBuf);
+            szBuf[cch + 1] = szBuf[cch] = szBuf[cch - 1] = 'x';
+            szBuf[cch + 2] = '\0';
+            rc = kLdrDyldGetFilename(hMod2, szBuf, cch);
+            if (rc == KLDR_ERR_BUFFER_OVERFLOW)
+            {
+                if (!szBuf[0])
+                    Failure("kLdrDyldGetFilename didn't return partial result on overflow\n");
+                else if (szBuf[cch - 1])
+                    Failure("kLdrDyldGetFilename didn't terminate partial result correctly overflow: '%s'\n", szBuf);
+                else if (szBuf[cch] != 'x')
+                    Failure("kLdrDyldGetFilename exceeded the buffer limit on partial overflow: '%s'\n", szBuf);
+            }
+            else
+                Failure("kLdrDyldGetFilename(\"tst-0-d\",,,) -> rc=%d (%#x) instead of KLDR_ERR_BUFFER_OVERFLOW\n", rc, rc);
+
+            /* check that we can query the module by the returned filename. */
+            rc = kLdrDyldGetFilename(hMod2, szBuf, sizeof(szBuf));
+            if (!rc)
+            {
+                hMod3 = (HKLDRMOD)0xffffeeee;
+                rc = kLdrDyldFindByName(szBuf, NULL, NULL, KLDRDYLD_SEARCH_HOST, 0, &hMod3);
+                if (rc || hMod3 != hMod2)
+                    Failure("kLdrDyldFindByName(\"%s\",,,) failed, rc=%d (%#x) hMod3=%p hMod2=%p\n",
+                            szBuf, rc, rc, (void *)hMod3, (void *)hMod2);
+            }
+            else
+                Failure("kLdrDyldGetName(\"tst-0-d\",,,) failed (b), rc=%d (%#x)\n", rc, rc);
+        }
+        else
+            Failure("kLdrDyldGetFilename(\"tst-0-d\",,,) failed, rc=%d (%#x)\n", rc, rc);
+
+        /* the other module */
+        rc = kLdrDyldGetName(hMod, szBuf, sizeof(szBuf));
+        if (!rc)
+        {
+            printf("tst-0-driver: name: '%s' ('tst-0-a')\n", szBuf);
+            psz = strstr(szBuf, "-0-");
+            if (    !psz
+                ||  strnicmp(psz, "-0-a", sizeof("-0-a") - 1))
+                Failure("kLdrDyldGetName(\"tst-0-a\",,,) -> '%s': pattern '-0-a' not found\n", szBuf);
+
+            /* check that we can query the module by the returned name. */
+            hMod3 = (HKLDRMOD)0xffffeeee;
+            rc = kLdrDyldFindByName(szBuf, NULL, NULL, KLDRDYLD_SEARCH_HOST, 0, &hMod3);
+            if (rc || hMod3 != hMod)
+                Failure("kLdrDyldFindByName(\"%s\",,,) failed, rc=%d (%#x) hMod3=%p hMod=%p\n",
+                        szBuf, rc, rc, (void *)hMod3, (void *)hMod);
+        }
+        else
+            Failure("kLdrDyldGetName(\"tst-0-a\",,,) failed, rc=%d (%#x)\n", rc, rc);
+
+        rc = kLdrDyldGetFilename(hMod, szBuf, sizeof(szBuf));
+        if (!rc)
+        {
+            printf("tst-0-driver: filename: '%s' ('tst-0-a')\n", szBuf);
+
+            /* check that we can query the module by the returned filename. */
+            hMod3 = (HKLDRMOD)0xffffeeee;
+            rc = kLdrDyldFindByName(szBuf, NULL, NULL, KLDRDYLD_SEARCH_HOST, 0, &hMod3);
+            if (rc || hMod3 != hMod)
+                Failure("kLdrDyldFindByName(\"%s\",,,) failed, rc=%d (%#x) hMod3=%p hMod=%p\n",
+                        szBuf, rc, rc, (void *)hMod3, (void *)hMod);
+        }
+        else
+            Failure("kLdrDyldGetFilename(\"tst-0-a\",,,) failed, rc=%d (%#x)\n", rc, rc);
+
+
+        /*
+         * Resolve the symbol exported by each of the two modules and call them.
+         */
+        if (!g_cErrors)
+        {
+            uintptr_t uValue;
+            uint32_t fKind;
+
+            fKind = 0xffeeffee;
+            uValue = ~(uintptr_t)42;
+            rc = kLdrDyldQuerySymbol(hMod, NIL_KLDRMOD_SYM_ORDINAL, MY_NAME("FuncA"), &uValue, &fKind);
+            if (!rc)
+            {
+                if (uValue == ~(uintptr_t)42)
+                    Failure("kLdrDyldQuerySymbol(\"tst-0-a\",,\"FuncA\",): uValue wasn't set.\n");
+                if (fKind == 0xffeeffee)
+                    Failure("kLdrDyldQuerySymbol(\"tst-0-a\",,\"FuncA\",): fKind wasn't set.\n");
+                if (    (fKind & KLDRSYMKIND_BIT_MASK) != KLDRSYMKIND_NO_BIT
+                    &&  (fKind & KLDRSYMKIND_BIT_MASK) != MY_KLDRSYMKIND_BITS)
+                    Failure("fKind=%#x indicates a different code 'bit' mode than we running at.\n", fKind);
+                if (    (fKind & KLDRSYMKIND_TYPE_MASK) != KLDRSYMKIND_NO_TYPE
+                    &&  (fKind & KLDRSYMKIND_TYPE_MASK) != KLDRSYMKIND_CODE)
+                    Failure("fKind=%#x indicates that \"FuncA\" isn't code.\n", fKind);
+                if (fKind & KLDRSYMKIND_FORWARDER)
+                    Failure("fKind=%#x indicates that \"FuncA\" is a forwarder. it isn't.\n", fKind);
+
+                /* call it. */
+                if (!g_cErrors)
+                {
+                    int (*pfnFuncA)(void) = (int (*)(void))uValue;
+                    rc = pfnFuncA();
+                    if (rc != 0x42000042)
+                        Failure("FuncA returned %#x expected 0x42000042\n", rc);
+                }
+
+                /*
+                 * Test kLdrDyldFindByAddress now that we've got an address.
+                 */
+                hMod3 = (HKLDRMOD)0xeeeeffff;
+                rc = kLdrDyldFindByAddress(uValue, &hMod3, NULL, NULL);
+                if (!rc)
+                {
+                    uintptr_t offSegment;
+                    uint32_t iSegment;
+
+                    if (hMod3 != hMod)
+                        Failure("kLdrDyldFindByAddress(%#p/*FuncA*/, \"tst-0-a\",,,) return incorrect hMod3=%p instead of %p.\n",
+                                uValue, hMod3, hMod);
+
+                    hMod3 = (HKLDRMOD)0xeeeeffff;
+                    iSegment = 0x42424242;
+                    rc = kLdrDyldFindByAddress(uValue, &hMod3, &iSegment, &offSegment);
+                    if (!rc)
+                    {
+                        if (hMod3 != hMod)
+                            Failure("Bad hMod3 on 2nd kLdrDyldFindByAddress call.\n");
+                        if (iSegment > 0x1000) /* safe guess */
+                            Failure("Bad iSegment=%#x\n", iSegment);
+                        if (offSegment > 0x100000) /* guesswork */
+                            Failure("Bad offSegment=%p\n", (void *)offSegment);
+                    }
+                    else
+                        Failure("kLdrDyldFindByAddress(%#p/*FuncA*/, \"tst-0-a\",,,) failed (b), rc=%d (%#x)\n",
+                                uValue, rc, rc);
+
+                    /* negative test */
+                    hMod3 = (HKLDRMOD)0xeeeeffff;
+                    iSegment = 0x42424242;
+                    offSegment = 0x87654321;
+                    rc = kLdrDyldFindByAddress(~(uintptr_t)16, &hMod3, &iSegment, &offSegment);
+                    if (!rc)
+                        Failure("negative kLdrDyldFindByAddress test returned successfully!\n");
+                    if (iSegment != ~(uint32_t)0)
+                        Failure("negative kLdrDyldFindByAddress: bad iSegment=%#x\n", iSegment);
+                    if (offSegment != ~(uintptr_t)0)
+                        Failure("negative kLdrDyldFindByAddress: bad offSegment=%p\n", (void *)offSegment);
+                    if (hMod3 != NIL_HKLDRMOD)
+                        Failure("negative kLdrDyldFindByAddress: bad hMod3=%p\n", (void *)hMod3);
+                }
+                else
+                    Failure("kLdrDyldFindByAddress(%#p/*FuncA*/, \"tst-0-a\",,,) failed, rc=%d (%#x)\n",
+                            uValue, rc, rc);
+            }
+            else
+                Failure("kLdrDyldQuerySymbol(\"tst-0-a\",,\"FuncA\",) failed, rc=%d (%#x)\n", rc, rc);
+
+            fKind = 0xffeeffee;
+            uValue = ~(uintptr_t)42;
+            rc = kLdrDyldQuerySymbol(hMod2, NIL_KLDRMOD_SYM_ORDINAL, MY_NAME("FuncD"), &uValue, &fKind);
+            if (!rc)
+            {
+                if (uValue == ~(uintptr_t)42)
+                    Failure("kLdrDyldQuerySymbol(\"tst-0-d\",,\"FuncD\",): uValue wasn't set.\n");
+                if (fKind == 0xffeeffee)
+                    Failure("kLdrDyldQuerySymbol(\"tst-0-d\",,\"FuncD\",): fKind wasn't set.\n");
+                if (    (fKind & KLDRSYMKIND_BIT_MASK) != KLDRSYMKIND_NO_BIT
+                    &&  (fKind & KLDRSYMKIND_BIT_MASK) != MY_KLDRSYMKIND_BITS)
+                    Failure("fKind=%#x indicates a different code 'bit' mode than we running at.\n", fKind);
+                if (    (fKind & KLDRSYMKIND_TYPE_MASK) != KLDRSYMKIND_NO_TYPE
+                    &&  (fKind & KLDRSYMKIND_TYPE_MASK) != KLDRSYMKIND_CODE)
+                    Failure("fKind=%#x indicates that \"FuncD\" isn't code.\n", fKind);
+                if (fKind & KLDRSYMKIND_FORWARDER)
+                    Failure("fKind=%#x indicates that \"FuncD\" is a forwarder. it isn't.\n", fKind);
+
+                /* call it. */
+                if (!g_cErrors)
+                {
+                    int (*pfnFuncD)(void) = (int (*)(void))uValue;
+                    rc = pfnFuncD();
+                    if (rc != 0x42000000)
+                        Failure("FuncD returned %#x expected 0x42000000\n", rc);
+                }
+
+                /* use the address to get the module handle. */
+                hMod3 = (HKLDRMOD)0xeeeeffff;
+                rc = kLdrDyldFindByAddress(uValue, &hMod3, NULL, NULL);
+                if (!rc)
+                {
+                    if (hMod3 != hMod2)
+                        Failure("kLdrDyldFindByAddress(%#p/*FuncD*/,,,) return incorrect hMod3=%p instead of %p.\n",
+                                uValue, hMod3, hMod2);
+                }
+                else
+                    Failure("kLdrDyldFindByAddress(%#p/*FuncD*/,,,) failed, rc=%d (%#x)\n",
+                            uValue, rc, rc);
+            }
+            else
+                Failure("kLdrDyldQuerySymbol(\"tst-0-a\",,\"FuncA\",) failed, rc=%d (%#x)\n", rc, rc);
+
+        }
+
+        /*
+         * Finally unload it.
+         */
         rc = kLdrDyldUnload(hMod);
         if (rc)
             Failure("kLdrDyldUnload() failed. rc=%d (%#x)\n", rc, rc);
+        if (!rc)
+        {
+            rc = kLdrDyldFindByName("tst-0-d", NULL, NULL, KLDRDYLD_SEARCH_HOST, 0, &hMod2);
+            if (rc != KLDR_ERR_MODULE_NOT_FOUND)
+                Failure("kLdrDyldFindByName(\"tst-0-d\",,,) return rc=%d (%#x), expected KLDR_ERR_MODULE_NOT_FOUND\n", rc, rc);
+
+            rc = kLdrDyldFindByName("tst-0-a", NULL, NULL, KLDRDYLD_SEARCH_HOST, 0, &hMod2);
+            if (rc != KLDR_ERR_MODULE_NOT_FOUND)
+                Failure("kLdrDyldFindByName(\"tst-0-a\",,,) return rc=%d (%#x), expected KLDR_ERR_MODULE_NOT_FOUND\n", rc, rc);
+        }
+    }
+
+    /*
+     * Now do what tst-0 would do.
+     */
+    if (!g_cErrors)
+    {
+
     }