summaryrefslogtreecommitdiffstats
path: root/trunk/src/doit.c
diff options
context:
space:
mode:
Diffstat (limited to 'trunk/src/doit.c')
-rw-r--r--trunk/src/doit.c270
1 files changed, 270 insertions, 0 deletions
diff --git a/trunk/src/doit.c b/trunk/src/doit.c
new file mode 100644
index 0000000..2208ece
--- /dev/null
+++ b/trunk/src/doit.c
@@ -0,0 +1,270 @@
+/* Copyright (C) 2001, 2003, 2004, 2005 Red Hat, Inc.
+ Written by Jakub Jelinek <jakub@redhat.com>, 2001.
+
+ This program is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 2, or (at your option)
+ any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program; if not, write to the Free Software Foundation,
+ Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */
+
+#include <config.h>
+#include <alloca.h>
+#include <errno.h>
+#include <error.h>
+#include <fcntl.h>
+#include <string.h>
+#include <sys/stat.h>
+#include <time.h>
+#include <unistd.h>
+#include "prelinktab.h"
+
+struct collect_ents
+ {
+ struct prelink_entry **ents;
+ int nents;
+ };
+
+static int
+find_ents (void **p, void *info)
+{
+ struct collect_ents *l = (struct collect_ents *) info;
+ struct prelink_entry *e = * (struct prelink_entry **) p;
+
+ if ((e->type == ET_DYN && e->done == 1)
+ || (e->type == ET_EXEC && e->done == 0 && ! libs_only))
+ l->ents[l->nents++] = e;
+
+ return 1;
+}
+
+static void
+clear_ent_marks (struct prelink_entry *ent)
+{
+ int i;
+
+ ent->u.tmp = 0;
+ for (i = 0; i < ent->ndepends; ++i)
+ clear_ent_marks (ent->depends[i]);
+}
+
+static struct prelink_entry *
+find_unlisted_dependency (struct prelink_entry *ent)
+{
+ int i;
+ struct prelink_entry *ret;
+
+ if (ent->u.tmp == 0)
+ return ent;
+ for (i = 0; i < ent->ndepends; ++i)
+ if ((ret = find_unlisted_dependency (ent->depends[i])) != NULL)
+ return ret;
+ return NULL;
+}
+
+static void
+prelink_ent (struct prelink_entry *ent)
+{
+ int i;
+ DSO *dso;
+ struct stat64 st;
+ struct prelink_link *hardlink;
+ char *move = NULL;
+ size_t movelen = 0;
+ struct prelink_entry *dep;
+
+ for (i = 0; i < ent->ndepends; ++i)
+ if (ent->depends[i]->done == 1)
+ prelink_ent (ent->depends[i]);
+
+ for (i = 0; i < ent->ndepends; ++i)
+ if (ent->depends[i]->done != 2)
+ {
+ ent->done = 0;
+ if (! undo)
+ ent->type = ET_UNPRELINKABLE;
+ if (verbose)
+ error (0, 0, "Could not prelink %s because its dependency %s could not be prelinked",
+ ent->filename, ent->depends[i]->filename);
+ return;
+ }
+ else
+ clear_ent_marks (ent->depends[i]);
+
+ ent->u.tmp = 1;
+ for (i = 0; i < ent->ndepends; ++i)
+ ent->depends[i]->u.tmp = 1;
+
+ if ((dep = find_unlisted_dependency (ent)) != NULL)
+ {
+ ent->done = 0;
+ if (! undo)
+ ent->type = ET_UNPRELINKABLE;
+ if (verbose)
+ error (0, 0, "Could not prelink %s because it doesn't use %s, but one of its dependencies has been prelinked against it",
+ ent->filename, dep->filename);
+ return;
+ }
+
+ if (verbose)
+ {
+ if (dry_run)
+ printf ("Would prelink %s\n", ent->canon_filename);
+ else
+ printf ("Prelinking %s\n", ent->canon_filename);
+ }
+
+ dso = open_dso (ent->canon_filename);
+ if (dso == NULL)
+ goto error_out;
+
+ if (fstat64 (dso->fd, &st) < 0)
+ {
+ error (0, errno, "%s changed during prelinking", ent->filename);
+ goto error_out;
+ }
+
+ if (st.st_dev != ent->dev || st.st_ino != ent->ino)
+ {
+ error (0, 0, "%s changed during prelinking", ent->filename);
+ goto error_out;
+ }
+
+ if (dry_run)
+ close_dso (dso);
+ else
+ {
+ if (prelink_prepare (dso))
+ goto make_unprelinkable;
+ if (ent->type == ET_DYN && relocate_dso (dso, ent->base))
+ goto make_unprelinkable;
+ if (prelink (dso, ent))
+ goto make_unprelinkable;
+ if (update_dso (dso, NULL))
+ {
+ dso = NULL;
+ goto error_out;
+ }
+ }
+ ent->done = 2;
+ ent->flags |= PCF_PRELINKED;
+
+ /* Redo hardlinks. */
+ for (hardlink = ent->hardlink; hardlink; hardlink = hardlink->next)
+ {
+ size_t len;
+
+ if (wrap_lstat64 (hardlink->canon_filename, &st) < 0)
+ {
+ error (0, 0, "Could not stat %s (former hardlink to %s)",
+ hardlink->canon_filename, ent->canon_filename);
+ continue;
+ }
+
+ if (st.st_dev != ent->dev || st.st_ino != ent->ino)
+ {
+ error (0, 0, "%s is no longer hardlink to %s",
+ hardlink->canon_filename, ent->canon_filename);
+ continue;
+ }
+
+ if (verbose)
+ {
+ if (dry_run)
+ printf ("Would link %s to %s\n", hardlink->canon_filename,
+ ent->canon_filename);
+ else
+ printf ("Linking %s to %s\n", hardlink->canon_filename,
+ ent->canon_filename);
+ }
+
+ if (dry_run)
+ continue;
+
+ len = strlen (hardlink->canon_filename);
+ if (len + sizeof (".#prelink#") > movelen)
+ {
+ movelen = len + sizeof (".#prelink#");
+ move = realloc (move, movelen);
+ if (move == NULL)
+ {
+ error (0, ENOMEM, "Could not hardlink %s to %s",
+ hardlink->canon_filename, ent->canon_filename);
+ movelen = 0;
+ continue;
+ }
+ }
+
+ memcpy (mempcpy (move, hardlink->canon_filename, len), ".#prelink#",
+ sizeof (".#prelink#"));
+ if (wrap_rename (hardlink->canon_filename, move) < 0)
+ {
+ error (0, errno, "Could not hardlink %s to %s",
+ hardlink->canon_filename, ent->canon_filename);
+ continue;
+ }
+
+ if (wrap_link (ent->canon_filename, hardlink->canon_filename) < 0)
+ {
+ error (0, errno, "Could not hardlink %s to %s",
+ hardlink->canon_filename, ent->canon_filename);
+
+ if (wrap_rename (move, hardlink->canon_filename) < 0)
+ {
+ error (0, errno, "Could not rename %s back to %s",
+ move, hardlink->canon_filename);
+ }
+ continue;
+ }
+
+ if (wrap_unlink (move) < 0)
+ {
+ error (0, errno, "Could not unlink %s", move);
+ continue;
+ }
+ }
+ free (move);
+
+ if (! dry_run && wrap_stat64 (ent->canon_filename, &st) >= 0)
+ {
+ ent->dev = st.st_dev;
+ ent->ino = st.st_ino;
+ ent->ctime = st.st_ctime;
+ ent->mtime = st.st_mtime;
+ }
+ return;
+
+make_unprelinkable:
+ if (! undo)
+ ent->type = ET_UNPRELINKABLE;
+error_out:
+ ent->done = 0;
+ if (dso)
+ close_dso (dso);
+ return;
+}
+
+void
+prelink_all (void)
+{
+ struct collect_ents l;
+ int i;
+
+ l.ents =
+ (struct prelink_entry **) alloca (prelink_entry_count
+ * sizeof (struct prelink_entry *));
+ l.nents = 0;
+ htab_traverse (prelink_filename_htab, find_ents, &l);
+
+ for (i = 0; i < l.nents; ++i)
+ if (l.ents[i]->done == 1
+ || (l.ents[i]->done == 0 && l.ents[i]->type == ET_EXEC))
+ prelink_ent (l.ents[i]);
+}
generated by cgit 1.2.3-korg (git 2.39.0) at 2024-04-26 16:49:37 +0000