/* Advance to next CU header.

   Copyright (C) 2002-2010, 2016, 2017 Red Hat, Inc.

   This file is part of elfutils.

   Written by Ulrich Drepper <drepper@redhat.com>, 2002.

   This file is free software; you can redistribute it and/or modify

   it under the terms of either

     * the GNU Lesser General Public License as published by the Free

       Software Foundation; either version 3 of the License, or (at

       your option) any later version

   or

     * the GNU General Public License as published by the Free

       Software Foundation; either version 2 of the License, or (at

       your option) any later version

   or both in parallel, as here.

   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 copies of the GNU General Public License and

   the GNU Lesser General Public License along with this program.  If

   not, see <http://www.gnu.org/licenses/>.  */

#ifdef HAVE_CONFIG_H

# include <config.h>

#endif

#include <libdwP.h>

#include <dwarf.h>

int

dwarf_next_unit (Dwarf *dwarf, Dwarf_Off off, Dwarf_Off *next_off,

		 size_t *header_sizep, Dwarf_Half *versionp,

		 Dwarf_Off *abbrev_offsetp, uint8_t *address_sizep,

		 uint8_t *offset_sizep, uint64_t *v4_type_signaturep,

		 Dwarf_Off *v4_type_offsetp)

{

  const bool v4_debug_types = v4_type_signaturep != NULL;

  return __libdw_next_unit (dwarf, v4_debug_types, off, next_off,

			     header_sizep, versionp, NULL,

			     abbrev_offsetp, address_sizep, offset_sizep,

			     v4_type_signaturep, v4_type_offsetp);

}

INTDEF(dwarf_next_unit)

int

internal_function

__libdw_next_unit (Dwarf *dwarf, bool v4_debug_types, Dwarf_Off off,

		   Dwarf_Off *next_off, size_t *header_sizep,

		   Dwarf_Half *versionp, uint8_t *unit_typep,

		   Dwarf_Off *abbrev_offsetp, uint8_t *address_sizep,

		   uint8_t *offset_sizep, uint64_t *unit_id8p,

		   Dwarf_Off *subdie_offsetp)

{

  /* Note that debug_type units come from .debug_types in DWARF < 5 and

     from .debug_info in DWARF >= 5.  If the user requested the

     v4_type_signature we return from .debug_types always.  If no signature

     is requested we return units (any type) from .debug_info.  */

  const size_t sec_idx = v4_debug_types ? IDX_debug_types : IDX_debug_info;

  /* Maybe there has been an error before.  */

  if (dwarf == NULL)

    return -1;

  /* If we reached the end before don't do anything.  */

  if (off == (Dwarf_Off) -1l

      || unlikely (dwarf->sectiondata[sec_idx] == NULL)

      /* Make sure there is enough space in the .debug_info section

	 for at least the initial word.  We cannot test the rest since

	 we don't know yet whether this is a 64-bit object or not.  */

      || unlikely (off + 4 >= dwarf->sectiondata[sec_idx]->d_size))

    {

      *next_off = (Dwarf_Off) -1l;

      return 1;

    }

  /* This points into the .debug_info or .debug_types section to the

     beginning of the CU entry.  */

  const unsigned char *data = dwarf->sectiondata[sec_idx]->d_buf;

  const unsigned char *bytes = data + off;

  const unsigned char *bytes_end = data + dwarf->sectiondata[sec_idx]->d_size;

  /* The format of the CU header is described in dwarf2p1 7.5.1 and

     changed in DWARFv5 (to include unit type, switch location of some

     fields and add some optional fields).

     1.  A 4-byte or 12-byte unsigned integer representing the length

	 of the .debug_info contribution for that compilation unit, not

	 including the length field itself. In the 32-bit DWARF format,

	 this is a 4-byte unsigned integer (which must be less than

	 0xfffffff0); in the 64-bit DWARF format, this consists of the

	 4-byte value 0xffffffff followed by an 8-byte unsigned integer

	 that gives the actual length (see Section 7.2.2). This field

	 indicates whether this unit is 32-bit of 64-bit DWARF, which

	 affects all other offset fields in this header.

      2. A 2-byte unsigned integer representing the version of the

	 DWARF information for that compilation unit. For DWARF Version

	 2.1, the value in this field is 2 (3 for v3, 4 for v4, 5 for v5).

	 This fields determines the order of the next fields and whether

	 there are any optional fields in this header.

      3. For DWARF 2, 3 and 4 (including v4 type units):

         A 4-byte or 8-byte unsigned offset into the .debug_abbrev

	 section. This offset associates the compilation unit with a

	 particular set of debugging information entry abbreviations. In

	 the 32-bit DWARF format, this is a 4-byte unsigned length; in

	 the 64-bit DWARF format, this is an 8-byte unsigned length (see

	 Section 7.4).

	 For DWARF 5:

	 A 1-byte unsigned integer representing the unit (header) type.

	 This field determines what the optional fields in the header

	 represent.  If this is an unknown unit type then we cannot

	 assume anything about the rest of the unit (header).

      4. For all DWARF versions (including v4 type units):

         A 1-byte unsigned integer representing the size in bytes of

	 an address on the target architecture. If the system uses

	 segmented addressing, this value represents the size of the

	 offset portion of an address. This is the last field in the header

	 for DWARF versions 2, 3 and 4 (except for v4 type units).

      5. For DWARF 5 only (this is field 3 for DWARF 2, 3, 4 and v4 types):

         A 4-byte or 8-byte unsigned offset into the .debug_abbrev

	 section. This offset associates the compilation unit with a

	 particular set of debugging information entry abbreviations. In

	 the 32-bit DWARF format, this is a 4-byte unsigned length; in

	 the 64-bit DWARF format, this is an 8-byte unsigned length.

      6. For v4 type units (this is really field 5 for v4 types) and

         DWARF 5 optional (skeleton, split_compile, type and

         split_type): An 8 byte (opaque) integer constant value. For

         v4 and v5 type units this is the type signature. For skeleton

         and split compile units this is the compilation ID.

      7. For v4 type units (this is really field 6 for v4 types) and

         DWARF 5 optional (type and split_type) and v4 type units:

         A 4-byte or 8-byte unsigned offset. In the 32-bit DWARF format,

         this is a 4-byte unsigned length; in the 64-bit DWARF format,

         this is an 8-byte unsigned length. This is the type DIE offset

	 (which is not necessarily the first DIE in the unit).

  */

  uint64_t length = read_4ubyte_unaligned_inc (dwarf, bytes);

  size_t offset_size = 4;

  /* Lengths of 0xfffffff0 - 0xffffffff are escape codes.  Oxffffffff is

     used to indicate that 64-bit dwarf information is being used, the

     other values are currently reserved.  */

  if (length == DWARF3_LENGTH_64_BIT)

    offset_size = 8;

  else if (unlikely (length >= DWARF3_LENGTH_MIN_ESCAPE_CODE

		     && length <= DWARF3_LENGTH_MAX_ESCAPE_CODE))

    {

    invalid:

      __libdw_seterrno (DWARF_E_INVALID_DWARF);

      return -1;

    }

  if (length == DWARF3_LENGTH_64_BIT)

    {

      /* This is a 64-bit DWARF format.  */

      if (bytes_end - bytes < 8)

	goto invalid;

      length = read_8ubyte_unaligned_inc (dwarf, bytes);

    }

  /* Read the version stamp.  Always a 16-bit value.  */

  if (bytes_end - bytes < 2)

    goto invalid;

  uint_fast16_t version = read_2ubyte_unaligned_inc (dwarf, bytes);

  /* We keep unit_type at zero for older DWARF since we cannot

     easily guess whether it is a compile or partial unit.  */

  uint8_t unit_type = 0;

  if (version >= 5)

    {

      if (bytes_end - bytes < 1)

	goto invalid;

      unit_type = *bytes++;

    }

  /* All these are optional.  */

  Dwarf_Off subdie_off = 0;

  uint64_t sig_id = 0;

  Dwarf_Off abbrev_offset = 0;

  uint8_t address_size = 0;

  if (version < 2 || version > 5

      || (version == 5 && ! (unit_type == DW_UT_compile

			     || unit_type == DW_UT_partial

			     || unit_type == DW_UT_skeleton

			     || unit_type == DW_UT_split_compile

			     || unit_type == DW_UT_type

			     || unit_type == DW_UT_split_type)))

    {

      /* We cannot really know more about the header.  Just report

	 the length of the unit, version and unit type.  */

      goto done;

    }

  /* We have to guess the unit_type. But we don't have a real CUDIE.  */

  if (version < 5)

    unit_type = v4_debug_types ? DW_UT_type : DW_UT_compile;

  /* Now we know how large the header is (should be).  */

  if (unlikely (__libdw_first_die_from_cu_start (off, offset_size, version,

						 unit_type)

		>= dwarf->sectiondata[sec_idx]->d_size))

    {

      *next_off = -1;

      return 1;

    }

  /* The address size.  Always an 8-bit value.

     Comes after abbrev_offset for version < 5, otherwise unit type

     and address size (if a known unit type) comes before abbrev_offset.  */

  if (version >= 5)

    address_size = *bytes++;

  /* Get offset in .debug_abbrev.  Note that the size of the entry

     depends on whether this is a 32-bit or 64-bit DWARF definition.  */

  if (__libdw_read_offset_inc (dwarf, sec_idx, &bytes, offset_size,

			       &abbrev_offset, IDX_debug_abbrev, 0))

    return -1;

  if (version < 5)

    address_size = *bytes++;

  /* Extra fields, signature/id and type offset/padding.  */

  if (v4_debug_types

      || (version >= 5

	  && (unit_type == DW_UT_skeleton || unit_type == DW_UT_split_compile

	      || unit_type == DW_UT_type || unit_type == DW_UT_split_type)))

    {

      sig_id = read_8ubyte_unaligned_inc (dwarf, bytes);

      if ((v4_debug_types

	   || unit_type == DW_UT_type || unit_type == DW_UT_split_type))

	{

	  if (__libdw_read_offset_inc (dwarf, sec_idx, &bytes, offset_size,

				       &subdie_off, sec_idx, 0))

	    return -1;

	  /* Validate that the TYPE_OFFSET points past the header.  */

	  if (unlikely (subdie_off < (size_t) (bytes - (data + off))))

	    goto invalid;

	}

    }

 done:

  if (unit_id8p != NULL)

    *unit_id8p = sig_id;

  if (subdie_offsetp != NULL)

    *subdie_offsetp = subdie_off;

  /* Store the header length.  This is really how much we have read

     from the header.  If we didn't recognize the unit type the

     header might actually be bigger.  */

  if (header_sizep != NULL)

    *header_sizep = bytes - (data + off);

  if (versionp != NULL)

    *versionp = version;

  if (unit_typep != NULL)

    *unit_typep = unit_type;

  if (abbrev_offsetp != NULL)

    *abbrev_offsetp = abbrev_offset;

  if (address_sizep != NULL)

    *address_sizep = address_size;

  /* Store the offset size.  */

  if (offset_sizep != NULL)

    *offset_sizep = offset_size;

  /* The length of the unit doesn't include the length field itself.

     The length field is either, with offset == 4: 2 * 4 - 4 == 4,

     or with offset == 8: 2 * 8 - 4 == 12.  */

  *next_off = off + 2 * offset_size - 4 + length;

  /* This means that the length field is bogus, but return the CU anyway.

     We just won't return anything after this.  */

  if (*next_off <= off)

    *next_off = (Dwarf_Off) -1;

  return 0;

}

int

dwarf_nextcu (Dwarf *dwarf, Dwarf_Off off, Dwarf_Off *next_off,

	      size_t *header_sizep, Dwarf_Off *abbrev_offsetp,

	      uint8_t *address_sizep, uint8_t *offset_sizep)

{

  return INTUSE(dwarf_next_unit) (dwarf, off, next_off, header_sizep, NULL,

				  abbrev_offsetp, address_sizep, offset_sizep,

				  NULL, NULL);

}

INTDEF(dwarf_nextcu)