X-Git-Url: http://vcs.maemo.org/git/?a=blobdiff_plain;f=elf_ops.h;h=122bf10c1f785663ce4a453f4209481bc8cf83ff;hb=b9dea4fbc65fbfad4ea7ffb79e7cf7f280bd7f64;hp=a7d027e1abbf28a5867d9ff2f20b03d42e99c992;hpb=5fe141fd30d35516eac3674f7b62be51ba34543f;p=qemu

diff --git a/elf_ops.h b/elf_ops.h
index a7d027e..122bf10 100644
--- a/elf_ops.h
+++ b/elf_ops.h
@@ -138,13 +138,14 @@ static int glue(load_symbols, SZ)(struct elfhdr *ehdr, int fd, int must_swab)
     return -1;
 }
 
-int glue(load_elf, SZ)(int fd, int64_t virt_to_phys_addend, int must_swab)
+int glue(load_elf, SZ)(int fd, int64_t virt_to_phys_addend,
+                       int must_swab, uint64_t *pentry)
 {
     struct elfhdr ehdr;
     struct elf_phdr *phdr = NULL, *ph;
     int size, i, total_size;
     elf_word mem_size, addr;
-    uint8_t *data;
+    uint8_t *data = NULL;
 
     if (read(fd, &ehdr, sizeof(ehdr)) != sizeof(ehdr))
         goto fail;
@@ -152,6 +153,9 @@ int glue(load_elf, SZ)(int fd, int64_t virt_to_phys_addend, int must_swab)
         glue(bswap_ehdr, SZ)(&ehdr);
     }
 
+    if (pentry)
+   	*pentry = (uint64_t)ehdr.e_entry;
+
     glue(load_symbols, SZ)(&ehdr, fd, must_swab);
 
     size = ehdr.e_phnum * sizeof(phdr[0]);
@@ -176,7 +180,8 @@ int glue(load_elf, SZ)(int fd, int64_t virt_to_phys_addend, int must_swab)
             /* XXX: avoid allocating */
             data = qemu_mallocz(mem_size);
             if (ph->p_filesz > 0) {
-                lseek(fd, ph->p_offset, SEEK_SET);
+                if (lseek(fd, ph->p_offset, SEEK_SET) < 0)
+                    goto fail;
                 if (read(fd, data, ph->p_filesz) != ph->p_filesz)
                     goto fail;
             }
@@ -187,10 +192,13 @@ int glue(load_elf, SZ)(int fd, int64_t virt_to_phys_addend, int must_swab)
             total_size += mem_size;
 
             qemu_free(data);
+            data = NULL;
         }
     }
+    qemu_free(phdr);
     return total_size;
  fail:
+    qemu_free(data);
     qemu_free(phdr);
     return -1;
 }