[gentoo-commits] gentoo-projects commit in portage-utils/libq: prelink.c

[Mike Frysinger (vapier)]

vapier      11/10/03 03:43:31

  Modified:             prelink.c
  Log:
  filter out elfs by their type

Revision  Changes    Path
1.2                  portage-utils/libq/prelink.c

file : http://sources.gentoo.org/viewvc.cgi/gentoo-projects/portage-utils/libq/prelink.c?rev=1.2&view=markup
plain: http://sources.gentoo.org/viewvc.cgi/gentoo-projects/portage-utils/libq/prelink.c?rev=1.2&content-type=text/plain
diff : http://sources.gentoo.org/viewvc.cgi/gentoo-projects/portage-utils/libq/prelink.c?r1=1.1&r2=1.2

Index: prelink.c
===================================================================
RCS file: /var/cvsroot/gentoo-projects/portage-utils/libq/prelink.c,v
retrieving revision 1.1
retrieving revision 1.2
diff -u -r1.1 -r1.2
--- prelink.c	3 Oct 2011 00:24:22 -0000	1.1
+++ prelink.c	3 Oct 2011 03:43:31 -0000	1.2
@@ -1,7 +1,7 @@
 /*
  * Copyright 2011 Gentoo Foundation
  * Distributed under the terms of the GNU General Public License v2
- * $Header: /var/cvsroot/gentoo-projects/portage-utils/libq/prelink.c,v 1.1 2011/10/03 00:24:22 vapier Exp $
+ * $Header: /var/cvsroot/gentoo-projects/portage-utils/libq/prelink.c,v 1.2 2011/10/03 03:43:31 vapier Exp $
  */
 
 static const char prelink_bin[] = "prelink";
@@ -60,10 +60,11 @@
 
 #ifdef __linux__
 #include <elf.h>
-static bool is_elf(int fd, const char *filename)
+static bool is_prelink_elf(int fd, const char *filename)
 {
-	char header[SELFMAG];
+	unsigned char header[EI_NIDENT + 2];
 	ssize_t len;
+	uint16_t e_type;
 
 	len = read(fd, &header, sizeof(header));
 	if (len == -1) {
@@ -75,10 +76,32 @@
 	if (len < sizeof(header))
 		return false;
 
-	return memcmp(header, ELFMAG, sizeof(header)) == 0 ? true : false;
+	if (memcmp(header, ELFMAG, SELFMAG))
+		return false;
+
+	/* prelink only likes certain types of ELFs */
+	switch (header[EI_DATA]) {
+	case ELFDATA2LSB:
+		e_type = header[EI_NIDENT] | (header[EI_NIDENT + 1] << 8);
+		break;
+	case ELFDATA2MSB:
+		e_type = header[EI_NIDENT + 1] | (header[EI_NIDENT] << 8);
+		break;
+	default:
+		return false;
+	}
+	switch (e_type) {
+	case ET_EXEC:
+	case ET_DYN:
+		return true;
+	default:
+		return false;
+	}
+
+	/* XXX: should we also check OS's/arches that prelink supports ? */
 }
 #else
-static bool is_elf(int fd, const char *filename)
+static bool is_prelink_elf(int fd, const char *filename)
 {
 	return false;
 }
@@ -93,7 +116,7 @@
 {
 	int pipefd[2];
 
-	if (!is_elf(fd, filename))
+	if (!is_prelink_elf(fd, filename))
 		return fd;
 
 	if (pipe(pipefd))