drgn/scripts/gen_tests_elf_py.py
Omar Sandoval 87b7292aa5 Relicense drgn from GPLv3+ to LGPLv2.1+
drgn is currently licensed as GPLv3+. Part of the long term vision for
drgn is that other projects can use it as a library providing
programmatic interfaces for debugger functionality. A more permissive
license is better suited to this goal. We decided on LGPLv2.1+ as a good
balance between software freedom and permissiveness.

All contributors not employed by Meta were contacted via email and
consented to the license change. The only exception was the author of
commit c4fbf7e589 ("libdrgn: fix for compilation error"), who did not
respond. That commit reverted a single line of code to one originally
written by me in commit 640b1c011d ("libdrgn: embed DWARF index in
DWARF info cache").

Signed-off-by: Omar Sandoval <osandov@osandov.com>
2022-11-01 17:05:16 -07:00

74 lines
1.7 KiB
Python
Executable File

#!/usr/bin/env python3
# Copyright (c) Meta Platforms, Inc. and affiliates.
# SPDX-License-Identifier: LGPL-2.1-or-later
import argparse
from pathlib import Path
import re
import sys
def main() -> None:
argparse.ArgumentParser(
description="Generate tests/elf.py from libdrgn/include/elf.h"
).parse_args()
contents = Path("libdrgn/include/elf.h").read_text()
contents = re.sub(r"/\*.*?\*/", "", contents, flags=re.DOTALL)
contents = re.sub(r"\\\n", "", contents)
enums = {
name: []
for name in (
"ET",
"PT",
"SHN",
"SHT",
"STB",
"STT",
"STV",
)
}
for match in re.finditer(
r"^\s*#\s*define\s+(?P<enum>"
+ "|".join(enums)
+ r")_(?P<name>\w+)\s+(?P<value>0x[0-9a-fA-F]+|[0-9]+)",
contents,
re.MULTILINE,
):
enum = match.group("enum")
name = match.group("name")
value = int(match.group("value"), 0)
enums[enum].append((name, value))
f = sys.stdout
f.write(
"""\
# Copyright (c) Meta Platforms, Inc. and affiliates.
# SPDX-License-Identifier: LGPL-2.1-or-later
# Generated by scripts/gen_tests_elf_py.py.
import enum
from typing import Text
"""
)
for type_name, constants in enums.items():
assert constants
f.write(f"\n\nclass {type_name}(enum.IntEnum):\n")
for name, value in constants:
f.write(f" {name} = 0x{value:X}\n")
f.write(
f"""
@classmethod
def str(cls, value: int) -> Text:
try:
return f"{type_name}_{{cls(value).name}}"
except ValueError:
return hex(value)
"""
)
if __name__ == "__main__":
main()