From: Guinevere Larsen <guinevere@redhat.com>
To: William Ferreira <wqferr@gmail.com>, gdb-patches@sourceware.org
Subject: Re: [PATCH] [gdb] Create script to convert old tests into Dwarf::assemble calls.
Date: Tue, 17 Dec 2024 13:43:01 -0300 [thread overview]
Message-ID: <c3f6ec7a-2f38-48ff-bc33-295aa8ca1d54@redhat.com> (raw)
In-Reply-To: <20241205135550.9320-1-wqferr@gmail.com>
Thanks for working on this amazing script! This is really fancy, and I
really like what you got going
I have some specific feedback on some lines, but in general I really
like where the script is going!
On 12/5/24 10:55 AM, William Ferreira wrote:
> PR testsuite/32261 requests a script that could convert old .S-based
> tests (that were made before dwarf.exp existed) into the new
> Dwarf::assemble calls in Tcl. This commit is an initial implementation
> of such a script. Python was chosen for convenience, and only relies on
> a single external library.
>
> Usage: the script operates not on .S files, but on ELF files with DWARF
> information. To convert an old test, one must run said test via
> `make check-gdb TESTS=testname` in their build directory. This will
> produce, as a side effect, an ELF file the test used as an inferior, at
> gdb/testsuite/outputs/testname/testname. This ELF file is this script's
> input.
>
> Reliability: not counting the limitations listed below, the script seems
> functional enough to be worthy of discussion in the mailing list. I have
> been testing it with different tests that already use Dwarf::assemble,
> to see if the script can produce a similar call to it. Indeed, in the
> cases that I've tested (including some more complex ones, marked with an
> asterisk below) it is able to produce comparable output to the original
> exp file. Of course, it can't reproduce the complex code *before* the
> Dwarf::assemble call. Values calculated then are simply inlined.
>
> The following .exp files have been tried in this way and their outputs
> highly resemble the original:
> - gdb.dwarf2/dynarr-ptr
> - gdb.dwarf2/void-type
> - gdb.dwarf2/ada-thick-pointer
> - gdb.dwarf2/atomic-type
> - gdb.dwarf2/dw2-entry-points (*)
>
> The following .exp files DO NOT WORK with this script:
> - gdb.dwarf2/cu-no-addrs
> - aranges not supported.
> - Compile unit hi_pc and low_pc hardcoded, prone to user error
> due to forgetting to replace with variables.
>
> The following .exp files work, with caveats addressed in the limitations
> section below.
> - gdb.dwarf2/cpp-linkage-name
> - Works correctly except for one attribute of the form SPECIAL_expr.
> - gdb.dwarf2/dw2-unusual-field-names
> - Same as above, with two instances of SPECIAL_expr.
> - gdb.dwarf2/implptr-optimized-out
> - Same as above, with two instances of SPECIAL_expr.
>
> Currently the script has the following known limitations:
> - It does not support Tcl outside a small boilerplate and
> Dwarf::assemble calls.
I don't think we need to support anything other than creating the
dwarf::assemble stuff. Everything else is usual boilerplate from the
testsuite, so we're pretty familiar with it (or at least have plenty of
things to double-check) while dwarf assembler isn't as common and much
more complex.
> - It does not support line tables.
> - It does not use $srcfile and other variables in the call to
> Dwarf::assemble (since it can't know where it is safe to substitute).
> - It does not support "fission" type DWARFs (in fact I still have no
> clue what those are).
> - It does not support cu {label LABEL} {} CUs, mostly because I couldn't
> find the information using pyelftools.
> - It sometimes outputs empty CUs at the start and end of the call. This
> might be a problem with my machine, but I've checked with DWARF dumps
> and they are indeed in the input ELF files generated by
> `make check-gdb`.
> - It does not support attributes with the forms DW_FORM_block* and
> DW_FORM_exprloc. This is mostly not a concern of the difficulty of
> the implementation, but of it being an incomplete feature and, thus,
> more susceptible to users forgetting to correct its mistakes or
> unfinished values (please see discussion started by Guinevere at
> comment 23 https://sourceware.org/bugzilla/show_bug.cgi?id=32261#c23).
> The incompleteness of this feature is easy to demonstrate: any call to
> gdb_target_symbol, a common tool used in these attributes, needs a
> symbol name that is erased after compilation. There is no way to guess
> where that address being referenced in a DW_OP_addr comes from, and it
> can't be hard coded since it can change depending on the machine
> compiling it.
>
> Please bring up any further shortcomings this script may have with your
> expectations.
>
> Bug: https://sourceware.org/bugzilla/show_bug.cgi?id=32261
> ---
> gdb/testsuite/lib/asm_to_dwarf_assembler.py | 830 ++++++++++++++++++++
> 1 file changed, 830 insertions(+)
> create mode 100644 gdb/testsuite/lib/asm_to_dwarf_assembler.py
>
> diff --git a/gdb/testsuite/lib/asm_to_dwarf_assembler.py b/gdb/testsuite/lib/asm_to_dwarf_assembler.py
> new file mode 100644
> index 00000000000..ea86c19c805
> --- /dev/null
> +++ b/gdb/testsuite/lib/asm_to_dwarf_assembler.py
> @@ -0,0 +1,830 @@
> +# Copyright 2024 Free Software Foundation, Inc.
> +
> +# This program 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.
> +#
> +# This program 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/>.
> +
> +# Due to the pyelftools dependency, this script requires Python version
> +# 3.10 or greater to run.
> +
> +"""A utility to convert ELF files with DWARF info to Dwarf::assemble code.
> +
> +Usage:
> + python ./asm_to_dwarf_assembler.py <path/to/elf/file>
> +
> +Dependencies:
> + Python >= 3.10
> + pyelftools >= 0.31
> +
> +Notes:
> +- Line tables are not currently supported.
> +- Non-contiguous subprograms are not currently supported.
> +- If you want to use $srcfile or similar, you must edit the references to the
> + file name manually, including DW_AT_name attributes on compile units.
> +- If run with binaries generated by make check-gdb, it may include an
> + additional compile_unit before and after the actual compile units. This is
> + an artifact of the normal compilation process, as these CUs are indeed in
> + the generated DWARF in some cases.
> +"""
> +
> +import sys
> +import argparse
> +import elftools
> +import re
> +
> +from io import BytesIO, IOBase
> +from typing import Any, Optional, Iterable, List, Union, Annotated
> +from functools import cache
> +from dataclasses import dataclass
> +from copy import copy
> +from datetime import datetime
> +from logging import getLogger
> +
> +from elftools import dwarf, elf
> +from elftools.elf.elffile import ELFFile
> +from elftools.dwarf.compileunit import CompileUnit as RawCompileUnit
> +from elftools.dwarf.die import DIE as RawDIE, AttributeValue
> +from elftools.common.exceptions import DWARFError
> +
> +
> +logger = getLogger(__file__)
> +
> +
> +EXPR_ATTRIBUTE_FORMS = [
> + "DW_FORM_exprloc",
> + "DW_FORM_block",
> + "DW_FORM_block1",
> + "DW_FORM_block2",
> + "DW_FORM_block4",
> +]
Since these aren't supported, does it make sense to keep them here?
> +
> +
> +@dataclass
> +class DWARFOperation:
> + """Values of the form DW_OP_<name>."""
> + suffix: Annotated[str, 'Part of the name that succeeds "DW_OP_".']
> + code: Annotated[int, "Numeric representation of the OP."]
> + num_operands: Annotated[int, "Number of operands this operator expects."]
> +
> +
> +dwarf_operations: Annotated[
> + dict[int, DWARFOperation],
> + "Table of all DWARF operations as per table 7.9 of the DWARF5 spec."
> +] = {}
> +def _register_op(suffix: str, code: int, num_operands: int) -> None:
> + assert code not in dwarf_operations, "Duplicate operation code found."
> + dwarf_operations[code] = DWARFOperation(suffix, code, num_operands)
> +
> +def _register_all_ops() -> None:
> + # Op codes 0x01 and 0x02 are reserved.
> +
> + _register_op("addr", 0x03, 1)
> +
> + # Op codes 0x04 and 0x05 are reserved.
> +
> + _register_op("deref", 0x06, 0)
> +
> + # Op code 0x07 is reserved.
> +
> + _register_op("const1u", 0x08, 1)
> + _register_op("const1s", 0x09, 1)
> + _register_op("const2u", 0x0a, 1)
> + _register_op("const2s", 0x0b, 1)
> + _register_op("const4u", 0x0c, 1)
> + _register_op("const4s", 0x0d, 1)
> + _register_op("const8u", 0x0e, 1)
> + _register_op("const8s", 0x0f, 1)
> + _register_op("constu", 0x10, 1)
> + _register_op("consts", 0x11, 1)
> +
> + _register_op("dup", 0x12, 0)
> + _register_op("drop", 0x13, 0)
> + _register_op("over", 0x14, 0)
> + _register_op("pick", 0x15, 1)
> + _register_op("swap", 0x16, 0)
> + _register_op("rot", 0x17, 0)
> + _register_op("xderef", 0x18, 0)
> +
> + _register_op("abs", 0x19, 0)
> + _register_op("and", 0x1a, 0)
> + _register_op("div", 0x1b, 0)
> + _register_op("minus", 0x1c, 0)
> + _register_op("mod", 0x1d, 0)
> + _register_op("mul", 0x1e, 0)
> + _register_op("neg", 0x1f, 0)
> + _register_op("not", 0x20, 0)
> + _register_op("or", 0x21, 0)
> + _register_op("plus", 0x22, 0)
> + _register_op("plus_uconst", 0x23, 1)
> + _register_op("shl", 0x24, 0)
> + _register_op("shr", 0x25, 0)
> + _register_op("shra", 0x26, 0)
> + _register_op("xor", 0x27, 0)
> + _register_op("bra", 0x28, 1)
> + _register_op("eq", 0x29, 0)
> + _register_op("ge", 0x2a, 0)
> + _register_op("gt", 0x2b, 0)
> + _register_op("le", 0x2c, 0)
> + _register_op("lt", 0x2d, 0)
> + _register_op("ne", 0x2e, 0)
> + _register_op("skip", 0x2f, 0)
> +
> + for lit_nr in range(32):
> + _register_op(f"lit{lit_nr}", 0x30 + lit_nr, 0)
> +
> + for reg_nr in range(32):
> + _register_op(f"reg{reg_nr}", 0x50 + reg_nr, 0)
> +
> + for breg_nr in range(32):
> + _register_op(f"breg{breg_nr}", 0x70 + breg_nr, 0)
> +
> + _register_op("regx", 0x90, 1)
> + _register_op("fbregx", 0x91, 1)
> + _register_op("bregx", 0x92, 2)
> + _register_op("piece", 0x93, 1)
> + _register_op("deref_size", 0x94, 1)
> + _register_op("xderef_size", 0x95, 1)
> + _register_op("nop", 0x96, 0)
> + _register_op("push_object_address", 0x97, 0)
> +
> + _register_op("call2", 0x98, 1)
> + _register_op("call4", 0x99, 1)
> + _register_op("call_ref", 0x9a, 1)
> + _register_op("form_tls_address", 0x9b, 0)
> + _register_op("call_frame_cfa", 0x9c, 0)
> +
> + _register_op("bit_piece", 0x9d, 2)
> + _register_op("implicit_value", 0x9e, 2)
> + _register_op("stack_value", 0x9f, 0)
> + _register_op("implicit_pointer", 0xa0, 2)
> + _register_op("addrx", 0xa1, 1)
> + _register_op("constx", 0xa2, 1)
> +
> + _register_op("entry_value", 0xa3, 2)
> + _register_op("const_type", 0xa4, 3)
> + _register_op("regval_type", 0xa5, 2)
> + _register_op("deref_type", 0xa6, 2)
> + _register_op("xderef_type", 0xa7, 2)
> +
> + _register_op("convert", 0xa8, 1)
> + _register_op("reinterpret", 0xa9, 1)
> +
> +
> +_register_all_ops()
Similar question to here, if I understood later code correctly. Do we
want to keep this if we're not doing location expressions?
> +
> +
> +copyright_notice_template = """
> +# Copyright 2024-{current_year} Free Software Foundation, Inc.
If the script is adding copyright, it will be for a new file, and so
copyright will always start on the year when the script is run. So it
can just be {current_year}
> +#
> +# This program 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.
> +#
> +# This program 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/>.
> +""".lstrip() # Strip to remove initial \n, but not the last one.
> +
> +# Workaround for my editor not to freak out over unclosed braces.
> +lbrace, rbrace = "{", "}"
> +
> +@cache
> +def get_indent_str(indent_count: int) -> str:
> + """Get whitespace string to prepend to another for indenting."""
> + indent = (indent_count // 2) * "\t"
> + if indent_count % 2 == 1:
> + indent += " "
> + return indent
> +
> +
> +def indent(line: str, indent_count: int) -> str:
> + """Indent line by indent_count levels."""
> + return get_indent_str(indent_count) + line
> +
> +
> +def labelify_str(s: str) -> str:
> + """Make s appropriate for a label name."""
> + # Replace "*" with the literal word "ptr".
> + s = s.replace("*", "ptr")
> +
> + # Replace any non-"word" characters by "_".
> + s = re.sub(r"\W", "_", s)
> +
> + # Remove consecutive "_"s.
> + s = re.sub(r"__+", "_", s)
> +
> + return s
> +
> +
> +class DWARFAttribute:
> + """Storage unit for a single DWARF attribute.
> +
> + All its values are strings that are usually passed on
> + directly to format. The exceptions to this are attributes
> + with int values with DW_FORM_ref4 or DW_FORM_ref_addr form.
> + Their values are interpreted as the global offset of the DIE
> + being referenced, which are looked up dynamically to fetch
> + their labels.
> + """
> + def __init__(
> + self,
> + die_offset: int,
> + name: str,
> + value: str | bytes | int | bool,
> + form=None,
> + ):
> + self.die_offset = die_offset
> + self.name = name
> + self.value = value
> + self.form = form
> +
> + def _format_expr_value(self) -> str:
> + # This code is left here in case this is feature is desired in the
> + # future. It is partly functional, being able to decode operations
> + # but not operands due to their variable length in bytes.
> +
> + # self.form = "SPECIAL_expr"
> + # lines = [lbrace]
> + # curr_idx = 0
> + # while curr_idx < len(self.value):
> + # curr_op_code = self.value[curr_idx]
> + #
> + # try:
> + # curr_op: DWARFOperation = dwarf_operations[curr_op_code]
> + # except KeyError as err:
> + # raise KeyError(
> + # f"Unknown op: {curr_op_code}, item {curr_idx} of"
> + # f" attribute {self.name}, DIE at offset"
> + # f" {self.die_offset}."
> + # ) from err
> + # # Empty arg here so it inserts a space before the first argument,
> + # # only if there is indeed a first argument.
> + # args = [""]
> + # for arg_nr in range(curr_op.num_operands):
> + # curr_idx += 1
> + # # TODO decode bytes according to specific operation.
> + # # The following line only reads the next BYTE of the sequence,
> + # # and needs to be adjusted to read the correct number of
> + # # bytes.
> + # # args.append(str(self.value[curr_idx]))
> + # op_line = "DW_OP_" + curr_op.suffix + " ".join(args)
> + # lines.append(op_line)
> + # curr_idx += 1
> + #
> + # lines.append(rbrace)
> + # return "\n".join(lines)
> + self.form = "SPECIAL_expr"
> + return "{ TODO: Fill expr list }"
Maybe, rather than "TODO" you could add "MANUAL" or something, to make
it clear that this is a manual process and expected to be so? (naming is
hard, but I think todo implies that this will be improved soon, and I
don't think that'd be the case?)
> +
> + def _needs_escaping(self, str_value: str) -> bool:
> + charset = set(str_value)
> + return bool(charset.intersection({"{", "}", " ", "\t"}))
> +
> + def _format_str(self, str_value: str) -> str:
> + if self._needs_escaping(str_value):
> + escaped_str = str(str_value)
> + # Replace single escape (which is itself escaped because of regex)
> + # with a double escape (which doesn't mean anything to regex so
> + # it doesn't need escaping).
> + escaped_str = re.sub(r"\\", r"\\", escaped_str)
> + escaped_str = re.sub("([{}])", r"\\\1", escaped_str)
> + return "{" + escaped_str + "}"
> + else:
> + return str_value
> +
> + def _format_value(
> + self,
> + offset_die_lookup: dict[int, "DWARFDIE"],
> + indent_count: int = 0
> + ) -> str:
> + if self.form in EXPR_ATTRIBUTE_FORMS:
> + return self._format_expr_value()
> + elif isinstance(self.value, bool):
> + return str(int(self.value))
> + elif isinstance(self.value, int):
> + if self.form == "DW_FORM_ref4":
> + # ref4-style referencing label.
> + die: "DWARFDIE" = offset_die_lookup[self.value]
> + return ":$" + die.tcl_label
> + elif self.form == "DW_FORM_ref_addr":
> + # ref_addr-style referencing label.
> + die: "DWARFDIE" = offset_die_lookup[self.value]
> + return "%$" + die.tcl_label
> + else:
> + return str(self.value)
> + elif isinstance(self.value, bytes):
> + return self._format_str(self.value.decode("ascii"))
> + elif isinstance(self.value, str):
> + return self._format_str(self.value)
> + else:
> + raise NotImplementedError(
> + "Unknown data type: " + str(type(self.value))
> + )
> +
> + def format(
> + self,
> + offset_die_lookup: dict[int, "DWARFDIE"],
> + indent_count: int = 0
> + ) -> str:
> + """Format the attribute in the form {name value form}.
> +
> + If form is DW_FORM_exprloc or DW_FORM_block, see next section on
> + DWARFOperations.
> +
> + If it isn't, value is formatted as follows:
> + If bool, use "1" if True, "0" if False.
> + If int:
> + If form is DW_FORM_ref4, use ":$label" where label is the
> + tcl_label of the DWARFDIE at offset "value".
> + If form is DW_FORM_ref_addr, use "%$label" where label is
> + the tcl_label of the DWARFDIE at offset "value".
> + Else, use value directly.
> + If bytes, use value.decode("ascii")
> + If str, use value directly.
> + Any other type results in a NotImplementedError being raised.
> +
> + Regarding DW_FORM_exprloc and DW_FORM_block:
> + The form is replaced with SPECIAL_expr.
> + The entries in the value are interpreted and decoded using the
> + dwarf_operations dictionary, and replaced with their names where
> + applicable.
> + """
> + s = lbrace
> + s += self.name + " "
> + s += self._format_value(offset_die_lookup)
> +
> + # Only explicitly state form if it's not a reference.
> + if self.form not in [None, "DW_FORM_ref4", "DW_FORM_ref_addr"]:
> + s += " " + self.form
> +
> + s += rbrace
> + return indent(s, indent_count)
> +
> +
> +class DWARFDIE:
> + """This script's parsed version of a RawDIE."""
> + def __init__(
> + self,
> + offset: int,
> + tag: str,
> + attrs: dict[str, DWARFAttribute],
> + tcl_label: Optional[str] = None
> + ):
> + self.offset = offset
> + self.tag = tag
> + self.attrs = copy(attrs)
> + self.children = []
> + self.tcl_label = tcl_label
I think it would be worth a comment here or in format lines, regarding
what tcl_label is actually about.
it might not be self-evident what these are, especially for people
unfamiliar with the assembler.
> +
> + def format_lines(
> + self,
> + offset_die_lookup: dict[int, "DWARFDIE"],
> + indent_count: int = 0
> + ) -> list[str]:
> + """Get the list of lines that represent this DIE in Dwarf assembler."""
> + die_lines = []
> +
> + # Prepend label to first line, if it's set.
> + if self.tcl_label:
> + first_line_start = self.tcl_label + ": "
> + else:
> + first_line_start = ""
> +
> + # First line, including label.
> + first_line = indent(
> + first_line_start + self.tag + " " + lbrace,
> + indent_count
> + )
> + die_lines.append(first_line)
> +
> + # Format attributes, if any.
> + if self.attrs:
> + for attr_name, attr in self.attrs.items():
> + attr_line = attr.format(
> + offset_die_lookup,
> + indent_count=indent_count+1
> + )
> + die_lines.append(attr_line)
> + die_lines.append(indent(rbrace, indent_count))
> + else:
> + # Don't create a new line, just append and immediately close the
> + # brace on the last line.
> + die_lines[-1] += rbrace
> +
> + # Format children, if any.
> + if self.children:
> + # Only open a new brace if there are any children for the
> + # current DIE.
> + die_lines[-1] += " " + lbrace
> + for child in self.children:
> + child_lines = child.format_lines(
> + offset_die_lookup,
> + indent_count=indent_count+1
> + )
> + die_lines.extend(child_lines)
> + die_lines.append(indent(rbrace, indent_count))
> +
> + return die_lines
> +
> + def format(
> + self,
> + offset_die_lookup: dict[int, "DWARFDIE"],
> + indent_count: int = 0
> + ) -> str:
> + """Join result from format_lines into a single str."""
> + return "\n".join(self.format_lines(offset_die_lookup, indent_count))
> +
> + def name(self) -> Optional[str]:
> + """Get DW_AT_name (if present) decoded as ASCII."""
> + raw_value = self.attrs.get("DW_AT_name")
> + if raw_value is None:
> + return None
> + else:
> + return raw_value.value.decode("ascii")
> +
> + def type_name(self) -> str:
> + """Name of Dwarf tag, with the "DW_TAG_" prefix removed."""
> + return re.sub("DW_TAG_", "", self.tag)
> +
> +class DWARFCompileUnit(DWARFDIE):
> + """Wrapper subclass for CU DIEs.
> +
> + This is necessary due to the special format CUs take in Dwarf::assemble.
> +
> + Instead of simply:
> + DW_TAG_compile_unit {
> + <attributes>
> + } {
> + <children>
> + }
> +
> + CUs are formatted as:
> + cu { <cu_special_vars> } {
> + DW_TAG_compile_unit {
> + <attributes>
> + } {
> + <children>
> + }
> + }
> + """
> +
> + # Default value for parameter dwarf_version defined in dwarf.exp line 1552.
> + default_dwarf_version = 4
> +
> + # Default value for parameter is_fission defined in dwarf.exp line 1556.
> + # Currently not implemented, see comment below.
> + # default_is_fission = False
> +
> + # Tag that signifies a DIE is a compile unit.
> + compile_unit_tag = "DW_TAG_compile_unit"
> +
> + def __init__(
> + self,
> + raw_die: RawDIE,
> + raw_cu: RawCompileUnit,
> + attrs: dict[str, DWARFAttribute],
> + ):
> + """Initialize additional instance variables for CU encoding.
> +
> + The additional instance variables are:
> + - is_64_bit: Optional[bool]
> + default None
> + Whether this CU is 64 bit or not.
> + - dwarf_version: int
> + default DWARFCompileUnit.default_dwarf_version
> + Version of DWARF this CU is using.
> + - addr_size: Optional[int]
> + default None
> + Size of an address in bytes.
> +
> + These variables are used to configure the first parameter of the cu
> + proc (which contains calls to the compile_unit proc in the body of
> + Dwarf::assemble).
> + """
> + super().__init__(
> + raw_die.offset,
> + DWARFCompileUnit.compile_unit_tag,
> + attrs
> + )
> + self.raw_cu = raw_cu
> + self.is_64_bit: Optional[bool] = None
> + self.dwarf_version: int = raw_cu.header.get(
> + "version",
> + DWARFCompileUnit.default_dwarf_version
> + )
> + self.addr_size: Optional[int] = raw_cu.header["address_size"]
> +
> + # Fission is not currently implemented because I don't know where to
> + # fetch this information from.
> + # self.is_fission: bool = self.default_is_fission
> +
> + # CU labels are not currently implemented because I haven't found where
> + # pyelftools exposes this information.
> + # self.cu_label: Optional[str] = None
> +
> + def format_lines(
> + self,
> + offset_die_lookup: dict[int, DWARFDIE],
> + indent_count: int = 0,
> + ) -> list[str]:
> + lines = []
> + lines.append(self._get_header(indent_count))
> + inner_lines = super().format_lines(
> + offset_die_lookup,
> + indent_count + 1
> + )
> + lines += inner_lines
> + lines.append(indent(rbrace, indent_count))
> + return lines
> +
> + def _get_header(self, indent_count: int = 0) -> str:
> + """Assemble the first line of the surrounding 'cu {} {}' proc call."""
> + header = indent("cu " + lbrace, indent_count)
> + cu_params = []
> +
> + if self.is_64_bit is not None:
> + # Convert from True/False to 1/0.
> + param_value = int(self.is_64_bit)
> + cu_params += ["is_64", str(param_value)]
> +
> + if self.dwarf_version != DWARFCompileUnit.default_dwarf_version:
> + cu_params += ["version", str(self.dwarf_version)]
> +
> + if self.addr_size is not None:
> + cu_params += ["addr_size", str(self.addr_size)]
> +
> + # Fission is not currently implemented, see comment above.
> + # if self.is_fission != DWARFCompileUnit.default_is_fission:
> + # # Same as is_64_bit conversion, True/False -> 1/0.
> + # param_value = int(self.is_fission)
> + # cu_params += ["fission", str(param_value)]
> +
> + # CU labels are not currently implemented, see commend above.
> + # if self.cu_label is not None:
> + # cu_params += ["label", self.cu_label]
> +
> + if cu_params:
> + header += " ".join(cu_params)
> +
> + header += rbrace + " " + lbrace
> + return header
> +
> +class DWARFParser:
> + """Converter from pyelftools's DWARF representation to this script's."""
> +
> + def __init__(self, elf_file: IOBase):
> + """Init parser with file opened in binary mode.
> +
> + File can be closed after this function is called.
> + """
> + self.raw_data = BytesIO(elf_file.read())
> + self.elf_data = ELFFile(self.raw_data)
> + self.dwarf_info = self.elf_data.get_dwarf_info()
> + self.offset_to_die: dict[int, DWARFDIE] = {}
> + self.label_to_die: dict[str, DWARFDIE] = {}
> + self.referenced_offsets: set[int] = set()
> + self.raw_cu_list: list[RawCompileUnit] = []
> + self.top_level_dies: list[DWARFDIE] = []
> + self.subprograms: list[DWARFDIE] = []
> + self.taken_labels: set[str] = set()
> +
> + self._read_cus()
> + self._create_necessary_labels()
> +
> + def _read_cus(self):
I think I'd personally rename this to _read_all_cus or something like
that. It is easy to miss the S at the end and typo the function name or
get confused.
> + """Populate self.raw_cu_list with all CUs in self.dwarf_info."""
> + for cu in self.dwarf_info.iter_CUs():
> + self._read_cu(cu)
> +
> + def _read_cu(self, raw_cu: RawCompileUnit):
> + """Read a compile_unit into self.cu_list."""
> + self.raw_cu_list.append(raw_cu)
> + for raw_die in raw_cu.iter_DIEs():
> + if not raw_die.is_null():
> + self._parse_die(raw_die)
> +
> + def _parse_die(self, raw_die: RawDIE) -> DWARFDIE:
> + """Process a single DIE and add it to offset_to_die.
> +
> + Look for DW_FORM_ref4 and DWD_FORM_ref_addr form attributes and replace
> + them with the global offset of the referenced DIE, and marking the
> + referenced DIE as "referenced". This will be used later to assign and
> + use labels.
> +
> + In case the DIE is a top-level DIE, add it to self.top_level_dies.
> +
> + In case the DIE is a subprogram, add it to self.subprograms and call
> + self._use_vars_for_low_and_high_pc_attr with it.
> + """
> + processed_attrs = {}
> + attr_value: AttributeValue
> + for attr_name, attr_value in raw_die.attributes.items():
> + actual_value = attr_value.value
> + if attr_value.form in ("DW_FORM_ref4", "DW_FORM_ref_addr"):
> + referenced_die = raw_die.get_DIE_from_attribute(attr_name)
> + actual_value = referenced_die.offset
> + self.referenced_offsets.add(referenced_die.offset)
> +
> + processed_attrs[attr_name] = DWARFAttribute(
> + raw_die.offset,
> + attr_name,
> + actual_value,
According to the comment above, this should not send actual_value, but
instead the string "referenced". Is this an oversight here or (more
likely) in the comment?
> + attr_value.form
> + )
> +
> + if raw_die.tag == DWARFCompileUnit.compile_unit_tag:
> + # FIXME: This isn't pretty, as it relies on global parser state.
> + # It also only works under the assumption CUs can't be nested.
> + die_cu = self.raw_cu_list[-1]
We try and not get code into GDB that has a FIXME, so this has to be
resolved before going in.
Would it make sense to call _parse_die with the CU that the DIE comes from?
> + processed_die = DWARFCompileUnit(raw_die, die_cu, processed_attrs)
> + else:
> + processed_die = DWARFDIE(
> + raw_die.offset,
> + raw_die.tag,
> + processed_attrs,
> + None
> + )
> +
> + if raw_die.get_parent() is None:
> + # Top level DIE
> + self.top_level_dies.append(processed_die)
> + else:
> + # Setting the parent here assumes the parent was already processed
> + # prior to this DIE being found.
> + # As far as I'm aware, this is always true in DWARF.
> + processed_parent = self.offset_to_die[raw_die.get_parent().offset]
> + processed_parent.children.append(processed_die)
> +
> + if processed_die.tag == "DW_TAG_subprogram":
> + self.subprograms.append(processed_die)
> + self._use_vars_for_low_and_high_pc_attr(processed_die)
> +
> + self.offset_to_die[processed_die.offset] = processed_die
> + return processed_die
> +
> + def _create_necessary_labels(self):
> + """Create labels to DIEs that were referenced by others."""
> + for offset in self.referenced_offsets:
> + die = self.offset_to_die[offset]
> + self._create_label_for_die(die)
> +
> + def _use_vars_for_low_and_high_pc_attr(self, subprogram: DWARFDIE) -> None:
> + """Replace existing PC attributes with Tcl variables.
> +
> + If DW_AT_low_pc exists for this DIE, replace it with accessing the
> + variable whose name is given by self.subprogram_start_var(subprogram).
> +
> + If DW_AT_high_pc exists for this DIE, replace it with accessing the
> + variable whose name is given by self.subprogram_end_var(subprogram).
> + """
> + low_pc_attr_name = "DW_AT_low_pc"
> + if low_pc_attr_name in subprogram.attrs:
> + start = self.subprogram_start_var(subprogram)
> + subprogram.attrs[low_pc_attr_name].value = start
> +
> + high_pc_attr_name = "DW_AT_high_pc"
> + if high_pc_attr_name in subprogram.attrs:
> + end = self.subprogram_end_var(subprogram)
> + subprogram.attrs[high_pc_attr_name].value = end
> +
> + def _create_label_for_die(self, die: DWARFDIE) -> None:
> + """Set tcl_label to a unique string among other DIEs for this parser.
> +
> + As a first attempt, use labelify(die.name()). If the DIE does not have
> + a name, use labelify(die.type_name()).
> +
> + If the chosen initial label is already taken, try again appending "_2".
> + While the attempt is still taken, try again replacing it with "_3", then
> + "_4", and so on.
> +
> + This function also creates an entry on self.label_to_die.
> + """
> + if die.tcl_label is not None:
> + return
> +
> + label = labelify_str(die.name() or die.type_name())
> +
> + # Deduplicate label in case of collision
> + if label in self.taken_labels:
> + suffix_nr = 2
> +
> + # Walrus operator to prevent writing the assembled label_suffix
> + # string literal twice. This could be rewritten by copying the
> + # string literal to the line after the end of the while loop,
> + # but I deemed it would be too frail in case one of them needs
> + # to be changed and the other is forgotten.
> + while (new_label := f"{label}_{suffix_nr}") in self.taken_labels:
> + suffix_nr += 1
> + label = new_label
> +
> + die.tcl_label = label
> + self.label_to_die[label] = die
> + self.taken_labels.add(label)
> +
> + def subprogram_start_var(self, subprogram: DWARFDIE) -> str:
> + """Name of the Tcl variable that holds the low PC for a subprogram."""
> + return f"${subprogram.name()}_start"
> +
> + def subprogram_end_var(self, subprogram: DWARFDIE) -> str:
> + """Name of the Tcl variable that holds the high PC for a subprogram."""
> + return f"${subprogram.name()}_end"
> +
> + def all_labels(self) -> set[str]:
> + """Get a copy of the set of all labels known to the parser so far."""
> + return copy(self.taken_labels)
> +
> +
> +class DWARFAssemblerGenerator:
> + """Class that generates Dwarf::assemble code out of a DWARFParser."""
> +
> + def __init__(self, dwarf_parser: DWARFParser, output=sys.stdout):
> + self.dwarf_parser = dwarf_parser
> + self.output = output
> +
> + def emit(self, line: str, indent_count: int) -> None:
> + """Print a single line indented indent_count times to self.output.
> +
> + If line is empty, it will always print an empty line, even with nonzero
> + indent_count.
> + """
> + if line:
> + line = get_indent_str(indent_count) + line
> + print(line, file=self.output)
> +
> + def generate_die(self, die: DWARFDIE, indent_count: int):
> + """Generate the lines that represent a DIE."""
> + die_lines = die.format(self.dwarf_parser.offset_to_die, indent_count)
> + self.emit(die_lines, 0)
> +
> + def generate(self):
> + indent_count = 0
> +
> + year = datetime.now().year
> + copyright_notice = copyright_notice_template.format(current_year=year)
> + for copyright_line in copyright_notice.split("\n"):
> + self.emit(copyright_line, indent_count)
> +
> + self.emit("load_lib dwarf.exp", indent_count)
> + self.emit("require dwarf2_support", indent_count)
> + self.emit("set asm_file [standard_output_file $srcfile2]", indent_count)
> +
> + self.emit("", 0)
> + self.emit("# TODO: call prepare_for_testing", indent_count)
> +
> + self.emit("", 0)
> + self.emit("Dwarf::assemble $asm_file " + lbrace, indent_count)
Honestly, you can just have "Dwarf::assemble $srcfile2" I think. no need
to set an asm_file
> +
> + # Begin Dwarf::assemble body.
> + indent_count += 1
> + self.emit("global srcdir subdir srcfile", indent_count)
> +
> + all_labels = self.dwarf_parser.all_labels()
> + self.emit("declare_labels " + " ".join(all_labels), indent_count)
> +
> + self.emit("", 0)
> + for subprogram in self.dwarf_parser.subprograms:
> + self.emit(f"get_func_info {subprogram.name()}", indent_count)
> +
> + for die in self.dwarf_parser.top_level_dies:
> + self.generate_die(die, indent_count)
> +
> + # TODO: line table, if it's within scope (it probably isn't).
> +
> + # End Dwarf::assemble body.
> + indent_count -= 1
> + self.emit(rbrace, indent_count)
> +
> +def main(argv):
> + try:
> + filename = argv[1]
> + except IndexError:
> + print("Usage:")
> + print("python ./asm_to_dwarf_assembler.py <path/to/elf/file>")
> + sys.exit(1)
> +
> + try:
> + with open(filename, "rb") as elf_file:
> + parser = DWARFParser(elf_file)
> + except Exception as e:
> + print("Error parsing ELF file. Does it contain DWARF information?")
> + print(str(e))
> + sys.exit(2)
> + generator = DWARFAssemblerGenerator(parser)
> + generator.generate()
> +
> +if __name__ == "__main__":
> + main(sys.argv)
--
Cheers,
Guinevere Larsen
She/Her/Hers
next prev parent reply other threads:[~2024-12-17 16:44 UTC|newest]
Thread overview: 8+ messages / expand[flat|nested] mbox.gz Atom feed top
2024-12-05 13:55 William Ferreira
2024-12-05 17:25 ` Simon Marchi
2024-12-05 17:37 ` Luis Machado
2024-12-05 17:46 ` Simon Marchi
2024-12-05 17:48 ` Luis Machado
2024-12-09 20:30 ` Tom Tromey
2024-12-17 16:43 ` Guinevere Larsen [this message]
2024-12-17 21:36 ` William Ferreira
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=c3f6ec7a-2f38-48ff-bc33-295aa8ca1d54@redhat.com \
--to=guinevere@redhat.com \
--cc=gdb-patches@sourceware.org \
--cc=wqferr@gmail.com \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox