rtld: Delay DF_1_NODELETE process until object DAG loaded
authorJohn Marino <draco@marino.st>
Mon, 20 Aug 2012 18:26:43 +0000 (20:26 +0200)
committerJohn Marino <draco@marino.st>
Mon, 20 Aug 2012 20:24:25 +0000 (22:24 +0200)
The current code can miss searching dependencies that haven't been loaded
yet.  Delay DF_1_NODELETE processing until object DAG is fully loaded.

Taken-from: FreeBSD SVN 239253 (14 Aug 2012)

libexec/rtld-elf/rtld.c

index 99c1316..4d4aa29 100644 (file)
@@ -1787,6 +1787,27 @@ init_dag(Obj_Entry *root)
     root->dag_inited = true;
 }
 
     root->dag_inited = true;
 }
 
+static void
+process_nodelete(Obj_Entry *root)
+{
+       const Objlist_Entry *elm;
+
+       /*
+        * Walk over object DAG and process every dependent object that
+        * is marked as DF_1_NODELETE. They need to grow their own DAG,
+        * which then should have its reference upped separately.
+        */
+       STAILQ_FOREACH(elm, &root->dagmembers, link) {
+               if (elm->obj != NULL && elm->obj->z_nodelete &&
+                   !elm->obj->ref_nodel) {
+                       dbg("obj %s nodelete", elm->obj->path);
+                       init_dag(elm->obj);
+                       ref_dag(elm->obj);
+                       elm->obj->ref_nodel = true;
+               }
+       }
+}
+
 /*
  * Initialize the dynamic linker.  The argument is the address at which
  * the dynamic linker has been mapped into memory.  The primary task of
 /*
  * Initialize the dynamic linker.  The argument is the address at which
  * the dynamic linker has been mapped into memory.  The primary task of
@@ -1986,12 +2007,6 @@ process_needed(Obj_Entry *obj, Needed_Entry *needed, int flags)
          flags & ~RTLD_LO_NOLOAD);
        if (obj1 == NULL && !ld_tracing && (flags & RTLD_LO_FILTEES) == 0)
            return (-1);
          flags & ~RTLD_LO_NOLOAD);
        if (obj1 == NULL && !ld_tracing && (flags & RTLD_LO_FILTEES) == 0)
            return (-1);
-       if (obj1 != NULL && obj1->z_nodelete && !obj1->ref_nodel) {
-           dbg("obj %s nodelete", obj1->path);
-           init_dag(obj1);
-           ref_dag(obj1);
-           obj1->ref_nodel = true;
-       }
     }
     return (0);
 }
     }
     return (0);
 }
@@ -2857,8 +2872,14 @@ dlopen_object(const char *name, int fd, Obj_Entry *refobj, int lo_flags,
                /* Make list of init functions to call. */
                initlist_add_objects(obj, &obj->next, &initlist);
            }
                /* Make list of init functions to call. */
                initlist_add_objects(obj, &obj->next, &initlist);
            }
+           /*
+            * Process all no_delete objects here, given them own
+            * DAGs to prevent their dependencies from being unloaded.
+            * This has to be done after we have loaded all of the
+            * dependencies, so that we do not miss any.
+            */
+            process_nodelete(obj);
        } else {
        } else {
-
            /*
             * Bump the reference counts for objects on this DAG.  If
             * this is the first dlopen() call for the object that was
            /*
             * Bump the reference counts for objects on this DAG.  If
             * this is the first dlopen() call for the object that was