add zpm-soneed

[zpackage] / elf / soname.c

diff --git a/elf/soname.c b/elf/soname.c
index 99770159b4e3109a0cd0175c7faf87994fb99198..8b6c451275f8a8e42d8f50cb58b028d380a7cc07 100644 (file)
--- 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;
@@ -22,6 +23,13 @@ int main(int ac, char **av) {
        char *name, *dynname;
        Elf64_Dyn *dent;
 
+       if (lstat(av[1], &sbuf) == -1) {
+               exit(1);
+       }
+       if (!S_ISREG(sbuf.st_mode)) {
+               exit(1);
+       }
+
        elffd = open(av[1], O_RDONLY);
        if (elffd == -1) {
                exit(1);
@@ -29,6 +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) {