Changeset 2247

Timestamp:

Jan 18, 2009, 5:08:13 AM (17 years ago)

Author:

bird

Message:

kbuild.c/FreeBSD: Alternative method for obtaining the executable address.

File:

• trunk/src/kmk/kbuild.c (modified) (2 diffs)

trunk/src/kmk/kbuild.c

@@ -40 +40 @@
 #if defined(__APPLE__)
 # include <mach-o/dyld.h>
+#endif
+#if defined(__FreeBSD__)
+# include <dlfcn.h>
+# include <sys/link_elf.h>
 #endif
 
@@ -133 +137 @@
     rc = readlink("/proc/curproc/file", szTmp, GET_PATH_MAX - 1);
     if (rc < 0 || rc == GET_PATH_MAX - 1)
+    {
+        /* /proc is optional, try rtdl. */
+        void *hExe = dlopen(NULL, 0);
         rc = -1;
+        if (hExe)
+        {
+            struct link_map const *pLinkMap = 0;
+            if (dlinfo(hExe, RTLD_DI_LINKMAP, &pLinkMap) == 0)
+            {
+                const char *pszImageName = pLinkMap->l_name;
+                size_t cchImageName = strlen(pszImageName);
+                if (cchImageName < GET_PATH_MAX)
+                {
+                    memcpy(szTmp, pszImageName, cchImageName + 1);
+                    rc = 0;
+                }
+            }    
+        }
+    }
     else
         szTmp[rc] = '\0';
 
-#elif defined(__gnu_linux__) /** @todo find proper define... */
+#elif defined(__gnu_linux__) || defined(__linux__)
     rc = readlink("/proc/self/exe", szTmp, GET_PATH_MAX - 1);
     if (rc < 0 || rc == GET_PATH_MAX - 1)