From 09c413b1be6adc4240117ffb07bb1914d6481ca5 Mon Sep 17 00:00:00 2001
From: Marc Hartmayer <mhartmay@linux.ibm.com>
Date: Tue, 22 Mar 2022 14:18:45 +0100
Subject: [PATCH] zdump: read_elf_phdrs: handle no program header case
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Handle the case where no program headers are defined (see `man 5 elf`).

Signed-off-by: Marc Hartmayer <mhartmay@linux.ibm.com>
Reviewed-by: Alexander Egorenkov <egorenar@linux.ibm.com>
Reviewed-by: Steffen Eiden <seiden@linux.ibm.com>
Signed-off-by: Jan Höppner <hoeppner@linux.ibm.com>
---
 zdump/df_elf.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/zdump/df_elf.c b/zdump/df_elf.c
index c95cf6d6..257ffaa9 100644
--- a/zdump/df_elf.c
+++ b/zdump/df_elf.c
@@ -69,6 +69,11 @@ Elf64_Phdr *read_elf_phdrs(const struct zg_fh *fh, const Elf64_Ehdr *ehdr, unsig
 	size_t phdrs_size;
 	Elf64_Phdr *phdrs;
 
+	if (!phnum) {
+		*phdr_count = phnum;
+		return NULL;
+	}
+
 	/* Cannot wraparound since `Elf64_Half`` is `uint16_t` */
 	phdrs_size = sizeof(*phdrs) * phnum;
 	phdrs = util_malloc(phdrs_size);