rtld: Add write-lock to case of filter loading
authorJohn Marino <draco@marino.st>
Sat, 21 Apr 2012 18:52:45 +0000 (20:52 +0200)
committerJohn Marino <draco@marino.st>
Sat, 21 Apr 2012 19:45:34 +0000 (21:45 +0200)
Propagate the current state of rtld_bind_lock to dlopen_object() calls
through the filter loading call chain. This fixes attempts to
write-lock the already locked rtld_bind_lock when filter loading is
initiated by relocation of dlopening dso.

Taken from: FreeBSD SVN 234170 (12 APR 2012)

Do not try to adjust stacks if dlopen_object is called too early.  This
is a follow-up to FreeBSD SVN 233231 which fixed a similar issue with
object initialization code.

Taken from: FreeBSD SVN 233777 (02 APR 2012)

libexec/rtld-elf/rtld.c

index 17528fc..d3fd379 100644 (file)
@@ -87,7 +87,7 @@ static void digest_dynamic(Obj_Entry *, int);
 static Obj_Entry *digest_phdr(const Elf_Phdr *, int, caddr_t, const char *);
 static Obj_Entry *dlcheck(void *);
 static Obj_Entry *dlopen_object(const char *name, int fd, Obj_Entry *refobj,
-    int lo_flags, int mode);
+    int lo_flags, int mode, RtldLockState *lockstate);
 static Obj_Entry *do_load_object(int, const char *, char *, struct stat *, int);
 static int do_search_info(const Obj_Entry *obj, int, struct dl_serinfo *);
 static bool donelist_check(DoneList *, const Obj_Entry *);
@@ -1945,13 +1945,14 @@ unload_filtees(Obj_Entry *obj)
 }
 
 static void
-load_filtee1(Obj_Entry *obj, Needed_Entry *needed, int flags)
+load_filtee1(Obj_Entry *obj, Needed_Entry *needed, int flags,
+    RtldLockState *lockstate)
 {
 
     for (; needed != NULL; needed = needed->next) {
        needed->obj = dlopen_object(obj->strtab + needed->name, -1, obj,
          flags, ((ld_loadfltr || obj->z_loadfltr) ? RTLD_NOW : RTLD_LAZY) |
-         RTLD_LOCAL);
+         RTLD_LOCAL, lockstate);
     }
 }
 
@@ -1961,8 +1962,8 @@ load_filtees(Obj_Entry *obj, int flags, RtldLockState *lockstate)
 
     lock_restart_for_upgrade(lockstate);
     if (!obj->filtees_loaded) {
-       load_filtee1(obj, obj->needed_filtees, flags);
-       load_filtee1(obj, obj->needed_aux_filtees, flags);
+       load_filtee1(obj, obj->needed_filtees, flags, lockstate);
+       load_filtee1(obj, obj->needed_aux_filtees, flags, lockstate);
        obj->filtees_loaded = true;
     }
 }
@@ -2772,7 +2773,7 @@ rtld_dlopen(const char *name, int fd, int mode)
            lo_flags |= RTLD_LO_TRACE;
 
     return (dlopen_object(name, fd, obj_main, lo_flags,
-      mode & (RTLD_MODEMASK | RTLD_GLOBAL)));
+      mode & (RTLD_MODEMASK | RTLD_GLOBAL), NULL));
 }
 
 static void
@@ -2787,17 +2788,20 @@ dlopen_cleanup(Obj_Entry *obj)
 
 static Obj_Entry *
 dlopen_object(const char *name, int fd, Obj_Entry *refobj, int lo_flags,
-    int mode)
+    int mode, RtldLockState *lockstate)
 {
     Obj_Entry **old_obj_tail;
     Obj_Entry *obj;
     Objlist initlist;
-    RtldLockState lockstate;
+    RtldLockState mlockstate;
     int result;
 
     objlist_init(&initlist);
 
-    wlock_acquire(rtld_bind_lock, &lockstate);
+    if (lockstate == NULL && !(lo_flags & RTLD_LO_EARLY)) {
+       wlock_acquire(rtld_bind_lock, &mlockstate);
+       lockstate = &mlockstate;
+    }
     GDB_STATE(RT_ADD,NULL);
 
     old_obj_tail = obj_tail;
@@ -2826,7 +2830,7 @@ dlopen_object(const char *name, int fd, Obj_Entry *refobj, int lo_flags,
            if (result == -1 || (relocate_objects(obj,
             (mode & RTLD_MODEMASK) == RTLD_NOW, &obj_rtld,
              (lo_flags & RTLD_LO_EARLY) ? SYMLOOK_EARLY : 0,
-             &lockstate)) == -1) {
+             lockstate)) == -1) {
                dlopen_cleanup(obj);
                obj = NULL;
            } else if (lo_flags & RTLD_LO_EARLY) {
@@ -2869,27 +2873,32 @@ dlopen_object(const char *name, int fd, Obj_Entry *refobj, int lo_flags,
        name);
     GDB_STATE(RT_CONSISTENT,obj ? &obj->linkmap : NULL);
 
-    map_stacks_exec(&lockstate);
+    if (!(lo_flags & RTLD_LO_EARLY)) {
+       map_stacks_exec(lockstate);
+    }
 
     if (initlist_objects_ifunc(&initlist, (mode & RTLD_MODEMASK) == RTLD_NOW,
       (lo_flags & RTLD_LO_EARLY) ? SYMLOOK_EARLY : 0,
-      &lockstate) == -1) {
+      lockstate) == -1) {
        objlist_clear(&initlist);
        dlopen_cleanup(obj);
-       lock_release(rtld_bind_lock, &lockstate);
+       if (lockstate == &mlockstate)
+           lock_release(rtld_bind_lock, lockstate);
        return (NULL);
     }
 
     if (!(lo_flags & RTLD_LO_EARLY)) {
        /* Call the init functions. */
-       objlist_call_init(&initlist, &lockstate);
+       objlist_call_init(&initlist, lockstate);
     }
     objlist_clear(&initlist);
-    lock_release(rtld_bind_lock, &lockstate);
+    if (lockstate == &mlockstate)
+       lock_release(rtld_bind_lock, lockstate);
     return obj;
 trace:
     trace_loaded_objects(obj);
-    lock_release(rtld_bind_lock, &lockstate);
+    if (lockstate == &mlockstate)
+       lock_release(rtld_bind_lock, lockstate);
     exit(0);
 }