X-Git-Url: https://pd.if.org/git/?a=blobdiff_plain;f=elf%2Fsoname.c;h=8b6c451275f8a8e42d8f50cb58b028d380a7cc07;hb=4efa2cf91ddcce727fd66be55f504510b6400b8b;hp=aa683a53acb3bf726218dc37af3c44e51a396c1a;hpb=489ad6b12bcdfd554acd0908a1770187e76e436d;p=zpackage

diff --git a/elf/soname.c b/elf/soname.c
index aa683a5..8b6c451 100644
--- a/elf/soname.c
+++ b/elf/soname.c
@@ -9,7 +9,8 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include <elf.h>
+
+#include "elf.h"
 
 int main(int ac, char **av) {
 	void *elfbase;
@@ -36,9 +37,14 @@ int main(int ac, char **av) {
 	if (fstat(elffd, &sbuf) == -1) {
 		exit(1);
 	}
+	/* not a regular file? */
 	if (!S_ISREG(sbuf.st_mode)) {
 		exit(1);
 	}
+	/* not at least the size of the elf header? */
+	if (sbuf.st_size < sizeof(Elf64_Ehdr)) {
+		exit(1);
+	}
 
 	elfbase = mmap(0, sbuf.st_size, PROT_READ,MAP_PRIVATE, elffd, 0);
 	if (!elfbase) {