1 files changed, 131 insertions, 0 deletions
diff --git a/tests/next_cfi.c b/tests/next_cfi.c
new file mode 100644
index 00000000..b923744f
--- /dev/null
+++ b/tests/next_cfi.c
@@ -0,0 +1,131 @@
+/* Test program for dwarf_next_cfi
+   Copyright (C) 2018 Red Hat, Inc.
+   This file is part of elfutils.
+
+   This file is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 3 of the License, or
+   (at your option) any later version.
+
+   elfutils is distributed in the hope that it will be useful, but
+   WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
+
+#include <config.h>
+#include <assert.h>
+#include <inttypes.h>
+#include <error.h>
+#include ELFUTILS_HEADER(dw)
+#include <dwarf.h>
+#include <argp.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <locale.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+void
+handle_section (const unsigned char e_ident[],
+		Elf_Scn *scn, const bool is_eh)
+{
+  if (is_eh)
+    printf (".eh_frame\n");
+  else
+    printf (".debug_frame\n");
+
+  Elf_Data *data = elf_getdata (scn, NULL);
+  if (data == NULL || data->d_buf == NULL)
+    error (EXIT_FAILURE, 0, "no section data");
+
+  int res;
+  Dwarf_Off off;
+  Dwarf_Off next_off = 0;
+  Dwarf_CFI_Entry entry;
+  while ((res = dwarf_next_cfi (e_ident, data, is_eh, off = next_off,
+				&next_off, &entry)) == 0)
+    {
+      printf ("[%" PRId64 "] ", off);
+      if (dwarf_cfi_cie_p (&entry))
+	printf ("CIE augmentation=\"%s\"\n", entry.cie.augmentation);
+      else
+	{
+	  printf ("FDE cie=[%" PRId64 "]\n", entry.fde.CIE_pointer);
+
+	  Dwarf_Off cie_off = entry.fde.CIE_pointer;
+	  Dwarf_Off cie_off_next;
+	  Dwarf_CFI_Entry cie_entry;
+	  if (dwarf_next_cfi (e_ident, data, is_eh, cie_off, &cie_off_next,
+			      &cie_entry) != 0
+	      || !dwarf_cfi_cie_p (&cie_entry))
+	    error (EXIT_FAILURE, 0, "FDE doesn't point to CIE");
+	}
+    }
+
+  if (res < 0)
+    error (EXIT_FAILURE, 0, "dwarf_next_cfi failed: %s\n",
+	   dwarf_errmsg (-1));
+}
+
+int
+main (int argc, char *argv[])
+{
+  if (argc != 2)
+    error (EXIT_FAILURE, 0, "need file name argument");
+
+  const char *file = argv[1];
+  printf ("%s\n", file);
+
+  int fd = open (file, O_RDONLY);
+  if (fd == -1)
+    error (EXIT_FAILURE, errno, "cannot open input file `%s'", file);
+
+  elf_version (EV_CURRENT);
+
+  Elf *elf = elf_begin (fd, ELF_C_READ, NULL);
+  if (elf == NULL)
+    error (EXIT_FAILURE, 0, "cannot create ELF descriptor: %s",
+	   elf_errmsg (-1));
+
+  size_t esize;
+  const unsigned char *ident = (const unsigned char *) elf_getident (elf,
+								     &esize);
+  if (ident == NULL || esize < EI_NIDENT)
+    error (EXIT_FAILURE, 0, "no, or too small, ELF ident");
+
+  GElf_Ehdr ehdr;
+  if (gelf_getehdr (elf, &ehdr) == NULL)
+    error (EXIT_FAILURE, 0, "cannot get the ELF header: %s\n",
+	   elf_errmsg (-1));
+
+  size_t strndx = ehdr.e_shstrndx;
+
+  Elf_Scn *scn = NULL;
+  while ((scn = elf_nextscn (elf, scn)) != NULL)
+    {
+      GElf_Shdr shdr;
+      if (gelf_getshdr (scn, &shdr) != NULL)
+	{
+	  char *name = elf_strptr (elf, strndx, (size_t) shdr.sh_name);
+	  if (name != NULL && shdr.sh_type == SHT_PROGBITS)
+	    {
+	      if (strcmp (name, ".eh_frame") == 0)
+		handle_section (ident, scn, true);
+	      if (strcmp (name, ".debug_frame") == 0)
+		handle_section (ident, scn, false);
+	    }
+	}
+    }
+
+  elf_end (elf);
+  close (fd);
+
+  return 0;
+}