Move everything to the toplevel
This commit is contained in:
parent
efe56d1011
commit
b9e56b5add
3
.gitignore
vendored
3
.gitignore
vendored
|
|
@ -1,5 +1,6 @@
|
||||||
|
.envrc
|
||||||
.venv
|
.venv
|
||||||
*.so
|
*.so
|
||||||
*.dylib
|
*.dylib
|
||||||
**/target/
|
target/
|
||||||
*.egg-info
|
*.egg-info
|
||||||
|
|
|
||||||
16
Cargo.toml
16
Cargo.toml
|
|
@ -1,6 +1,12 @@
|
||||||
[workspace]
|
[package]
|
||||||
resolver = "2"
|
name = "goblin-pyo3"
|
||||||
|
version = "0.0.2"
|
||||||
|
edition = "2021"
|
||||||
|
|
||||||
members = [
|
[lib]
|
||||||
"crates/goblin-pyo3"
|
name = "oelf"
|
||||||
]
|
crate-type = ["cdylib"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
goblin = "0.7.1"
|
||||||
|
pyo3 = { version = "0.20.0", features = ["extension-module"] }
|
||||||
|
|
|
||||||
|
|
@ -1,12 +0,0 @@
|
||||||
[package]
|
|
||||||
name = "goblin-pyo3"
|
|
||||||
version = "0.0.2"
|
|
||||||
edition = "2021"
|
|
||||||
|
|
||||||
[lib]
|
|
||||||
name = "oelf"
|
|
||||||
crate-type = ["cdylib"]
|
|
||||||
|
|
||||||
[dependencies]
|
|
||||||
goblin = "0.7.1"
|
|
||||||
pyo3 = { version = "0.20.0", features = ["extension-module"] }
|
|
||||||
|
|
@ -1,347 +0,0 @@
|
||||||
import sys
|
|
||||||
|
|
||||||
import apsw
|
|
||||||
import apsw.shell
|
|
||||||
import goblin
|
|
||||||
|
|
||||||
from dataclasses import dataclass
|
|
||||||
from enum import Flag, auto
|
|
||||||
from typing import Any, Callable, Iterator, Sequence, Tuple, cast
|
|
||||||
|
|
||||||
connection = apsw.Connection(":memory:")
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class Generator:
|
|
||||||
"""A generator for the virtual table SQLite module.
|
|
||||||
|
|
||||||
This class is needed because apsw wants to assign columns and
|
|
||||||
column_access to the generator function itself."""
|
|
||||||
|
|
||||||
columns: Sequence[str]
|
|
||||||
column_access: apsw.ext.VTColumnAccess
|
|
||||||
callable: Callable[[], Iterator[dict[str, Any]]]
|
|
||||||
|
|
||||||
def __call__(self) -> Iterator[dict[str, Any]]:
|
|
||||||
"""Call the generator should return an iterator of dictionaries.
|
|
||||||
|
|
||||||
The dictionaries should have keys that match the column names."""
|
|
||||||
return self.callable()
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def make_generator(
|
|
||||||
columns: list[str], generator: Callable[[], Iterator[dict[str, Any]]]
|
|
||||||
):
|
|
||||||
"""Create a generator from a callable that returns
|
|
||||||
an iterator of dictionaries."""
|
|
||||||
return Generator(columns, apsw.ext.VTColumnAccess.By_Name, generator)
|
|
||||||
|
|
||||||
|
|
||||||
class CacheFlag(Flag):
|
|
||||||
NONE = 0
|
|
||||||
DYNAMIC_ENTRIES = auto()
|
|
||||||
HEADERS = auto()
|
|
||||||
INSTRUCTIONS = auto()
|
|
||||||
SECTIONS = auto()
|
|
||||||
EXPORTS = auto()
|
|
||||||
IMPORTS = auto()
|
|
||||||
SYMBOLS = auto()
|
|
||||||
RPATHS = auto()
|
|
||||||
LIBS = auto()
|
|
||||||
STRINGS = auto()
|
|
||||||
VERSION_REQUIREMENTS = auto()
|
|
||||||
VERSION_DEFINITIONS = auto()
|
|
||||||
DWARF_DIE = auto()
|
|
||||||
DWARF_DIE_CALL_GRAPH = auto()
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_string(cls, str: str):
|
|
||||||
"""Convert a string to a CacheFlag.
|
|
||||||
|
|
||||||
This also specially handles 'ALL' which returns all the flags."""
|
|
||||||
if str == "ALL":
|
|
||||||
return cls.ALL()
|
|
||||||
try:
|
|
||||||
return cls[str]
|
|
||||||
except KeyError:
|
|
||||||
raise ValueError(f"{str} is not a valid CacheFlag")
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def ALL(cls):
|
|
||||||
retval = cls.NONE
|
|
||||||
for member in cls.__members__.values():
|
|
||||||
retval |= member
|
|
||||||
return retval
|
|
||||||
|
|
||||||
|
|
||||||
def register_generator(
|
|
||||||
connection: apsw.Connection,
|
|
||||||
generator: Generator,
|
|
||||||
table_name: str,
|
|
||||||
generator_flag: CacheFlag,
|
|
||||||
cache_flags: CacheFlag,
|
|
||||||
) -> None:
|
|
||||||
"""Register a virtual table generator.
|
|
||||||
|
|
||||||
This method does a bit of duplicate work which checks if we need to cache
|
|
||||||
the given generator.
|
|
||||||
|
|
||||||
If so we rename the table with a prefix 'raw' and then create a temp table"""
|
|
||||||
original_table_name = table_name
|
|
||||||
if generator_flag in cache_flags:
|
|
||||||
table_name = f"raw_{table_name}"
|
|
||||||
|
|
||||||
apsw.ext.make_virtual_module(connection, table_name, generator)
|
|
||||||
|
|
||||||
if generator_flag in cache_flags:
|
|
||||||
connection.execute(
|
|
||||||
f"""CREATE TABLE {original_table_name}
|
|
||||||
AS SELECT * FROM {table_name};"""
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def register_headers(
|
|
||||||
gob: goblin.Object, connection: apsw.Connection, cache_flags: CacheFlag
|
|
||||||
) -> None:
|
|
||||||
def dynamic_entries_generator() -> Iterator[dict[str, Any]]:
|
|
||||||
header = gob.header
|
|
||||||
yield {
|
|
||||||
"path": gob.path,
|
|
||||||
"magic": header.magic,
|
|
||||||
"cputype": header.cputype,
|
|
||||||
"cpusubtype": header.cpusubtype,
|
|
||||||
"filetype": header.filetype,
|
|
||||||
"ncmds": header.ncmds,
|
|
||||||
"sizeofcmds": header.sizeofcmds,
|
|
||||||
"flags": header.flags,
|
|
||||||
"reserved": header.reserved,
|
|
||||||
}
|
|
||||||
|
|
||||||
generator = Generator.make_generator(
|
|
||||||
[
|
|
||||||
"path",
|
|
||||||
"magic",
|
|
||||||
"cputype",
|
|
||||||
"cpusubtype",
|
|
||||||
"filetype",
|
|
||||||
"ncmds",
|
|
||||||
"sizeofcmds",
|
|
||||||
"flags",
|
|
||||||
"reserved",
|
|
||||||
],
|
|
||||||
dynamic_entries_generator,
|
|
||||||
)
|
|
||||||
|
|
||||||
register_generator(
|
|
||||||
connection,
|
|
||||||
generator,
|
|
||||||
"macho_headers",
|
|
||||||
CacheFlag.HEADERS,
|
|
||||||
cache_flags,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def register_symbols(
|
|
||||||
gob: goblin.Object, connection: apsw.Connection, cache_flags: CacheFlag
|
|
||||||
) -> None:
|
|
||||||
def dynamic_entries_generator() -> Iterator[dict[str, Any]]:
|
|
||||||
for sym in gob.symbols():
|
|
||||||
yield {
|
|
||||||
"name": sym.name,
|
|
||||||
"type": sym.typ,
|
|
||||||
"global": sym.is_global,
|
|
||||||
"weak": sym.weak,
|
|
||||||
"undefined": sym.undefined,
|
|
||||||
"stab": sym.stab,
|
|
||||||
}
|
|
||||||
|
|
||||||
generator = Generator.make_generator(
|
|
||||||
["name", "type", "global", "weak", "undefined", "stab"],
|
|
||||||
dynamic_entries_generator,
|
|
||||||
)
|
|
||||||
|
|
||||||
register_generator(
|
|
||||||
connection,
|
|
||||||
generator,
|
|
||||||
"macho_symbols",
|
|
||||||
CacheFlag.SYMBOLS,
|
|
||||||
cache_flags,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def register_sections(
|
|
||||||
gob: goblin.Object, connection: apsw.Connection, cache_flags: CacheFlag
|
|
||||||
) -> None:
|
|
||||||
def dynamic_entries_generator() -> Iterator[dict[str, Any]]:
|
|
||||||
for sect in gob.sections():
|
|
||||||
yield {
|
|
||||||
"name": sect.name,
|
|
||||||
"segment": sect.segment,
|
|
||||||
"addr": sect.addr,
|
|
||||||
"size": sect.size,
|
|
||||||
"offset": sect.offset,
|
|
||||||
"align": sect.align,
|
|
||||||
"reloff": sect.reloff,
|
|
||||||
"nreloc": sect.nreloc,
|
|
||||||
"flags": sect.flags,
|
|
||||||
}
|
|
||||||
|
|
||||||
generator = Generator.make_generator(
|
|
||||||
[
|
|
||||||
"name",
|
|
||||||
"segment",
|
|
||||||
"addr",
|
|
||||||
"size",
|
|
||||||
"offset",
|
|
||||||
"align",
|
|
||||||
"reloff",
|
|
||||||
"nreloc",
|
|
||||||
"flags",
|
|
||||||
],
|
|
||||||
dynamic_entries_generator,
|
|
||||||
)
|
|
||||||
|
|
||||||
register_generator(
|
|
||||||
connection,
|
|
||||||
generator,
|
|
||||||
"macho_sections",
|
|
||||||
CacheFlag.SECTIONS,
|
|
||||||
cache_flags,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def register_exports(
|
|
||||||
gob: goblin.Object, connection: apsw.Connection, cache_flags: CacheFlag
|
|
||||||
) -> None:
|
|
||||||
def dynamic_entries_generator() -> Iterator[dict[str, Any]]:
|
|
||||||
for exp in gob.exports():
|
|
||||||
yield {
|
|
||||||
"name": exp.name,
|
|
||||||
"size": exp.size,
|
|
||||||
"offset": exp.offset,
|
|
||||||
"type": str(exp.info.typ),
|
|
||||||
"address": exp.info.address,
|
|
||||||
"flags": exp.info.flags,
|
|
||||||
"lib": exp.info.lib,
|
|
||||||
"lib_symbol_name": exp.info.lib_symbol_name,
|
|
||||||
}
|
|
||||||
|
|
||||||
generator = Generator.make_generator(
|
|
||||||
[
|
|
||||||
"name",
|
|
||||||
"size",
|
|
||||||
"offset",
|
|
||||||
"type",
|
|
||||||
"address",
|
|
||||||
"flags",
|
|
||||||
"lib",
|
|
||||||
"lib_symbol_name",
|
|
||||||
],
|
|
||||||
dynamic_entries_generator,
|
|
||||||
)
|
|
||||||
|
|
||||||
register_generator(
|
|
||||||
connection,
|
|
||||||
generator,
|
|
||||||
"macho_exports",
|
|
||||||
CacheFlag.EXPORTS,
|
|
||||||
cache_flags,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def register_imports(
|
|
||||||
gob: goblin.Object, connection: apsw.Connection, cache_flags: CacheFlag
|
|
||||||
) -> None:
|
|
||||||
def dynamic_entries_generator() -> Iterator[dict[str, Any]]:
|
|
||||||
for imp in gob.imports():
|
|
||||||
yield {
|
|
||||||
"name": imp.name,
|
|
||||||
"dylib": imp.dylib,
|
|
||||||
"lazy": imp.is_lazy,
|
|
||||||
"offset": imp.offset,
|
|
||||||
"size": imp.size,
|
|
||||||
"address": imp.address,
|
|
||||||
"addend": imp.addend,
|
|
||||||
"is_weak": imp.is_weak,
|
|
||||||
"start_of_sequence_offset": imp.start_of_sequence_offset,
|
|
||||||
}
|
|
||||||
|
|
||||||
generator = Generator.make_generator(
|
|
||||||
[
|
|
||||||
"name",
|
|
||||||
"dylib",
|
|
||||||
"lazy",
|
|
||||||
"offset",
|
|
||||||
"size",
|
|
||||||
"address",
|
|
||||||
"addend",
|
|
||||||
"is_weak",
|
|
||||||
"start_of_sequence_offset",
|
|
||||||
],
|
|
||||||
dynamic_entries_generator,
|
|
||||||
)
|
|
||||||
|
|
||||||
register_generator(
|
|
||||||
connection,
|
|
||||||
generator,
|
|
||||||
"macho_imports",
|
|
||||||
CacheFlag.IMPORTS,
|
|
||||||
cache_flags,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def register_rpaths(
|
|
||||||
gob: goblin.Object, connection: apsw.Connection, cache_flags: CacheFlag
|
|
||||||
) -> None:
|
|
||||||
def dynamic_entries_generator() -> Iterator[dict[str, Any]]:
|
|
||||||
for rpath in gob.rpaths:
|
|
||||||
yield {"path": g.path, "rpath": rpath}
|
|
||||||
|
|
||||||
generator = Generator.make_generator(
|
|
||||||
["path", "rpath"],
|
|
||||||
dynamic_entries_generator,
|
|
||||||
)
|
|
||||||
|
|
||||||
register_generator(
|
|
||||||
connection,
|
|
||||||
generator,
|
|
||||||
"macho_rpaths",
|
|
||||||
CacheFlag.RPATHS,
|
|
||||||
cache_flags,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def register_libs(
|
|
||||||
gob: goblin.Object, connection: apsw.Connection, cache_flags: CacheFlag
|
|
||||||
) -> None:
|
|
||||||
def dynamic_entries_generator() -> Iterator[dict[str, Any]]:
|
|
||||||
for lib in gob.libs:
|
|
||||||
yield {"path": g.path, "lib": lib}
|
|
||||||
|
|
||||||
generator = Generator.make_generator(
|
|
||||||
["path", "lib"],
|
|
||||||
dynamic_entries_generator,
|
|
||||||
)
|
|
||||||
|
|
||||||
register_generator(
|
|
||||||
connection,
|
|
||||||
generator,
|
|
||||||
"macho_libs",
|
|
||||||
CacheFlag.RPATHS,
|
|
||||||
cache_flags,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
path = sys.argv[1]
|
|
||||||
g = goblin.Object(path)
|
|
||||||
register_headers(g, connection, CacheFlag.HEADERS)
|
|
||||||
register_symbols(g, connection, CacheFlag.SYMBOLS)
|
|
||||||
register_sections(g, connection, CacheFlag.SECTIONS)
|
|
||||||
register_exports(g, connection, CacheFlag.EXPORTS)
|
|
||||||
register_imports(g, connection, CacheFlag.IMPORTS)
|
|
||||||
register_rpaths(g, connection, CacheFlag.RPATHS)
|
|
||||||
register_libs(g, connection, CacheFlag.LIBS)
|
|
||||||
|
|
||||||
shell = apsw.shell.Shell(db=connection, stdin=sys.stdin)
|
|
||||||
shell.command_prompt(["ölf> "])
|
|
||||||
shell.cmdloop()
|
|
||||||
|
|
@ -1,17 +0,0 @@
|
||||||
[project]
|
|
||||||
name = "oelf"
|
|
||||||
version = "0.1"
|
|
||||||
requires-python = ">=3.8"
|
|
||||||
classifiers = [
|
|
||||||
"Programming Language :: Rust",
|
|
||||||
"Programming Language :: Python :: Implementation :: CPython",
|
|
||||||
"Programming Language :: Python :: Implementation :: PyPy",
|
|
||||||
]
|
|
||||||
|
|
||||||
dependencies = [
|
|
||||||
"apsw",
|
|
||||||
]
|
|
||||||
|
|
||||||
[build-system]
|
|
||||||
requires = ["setuptools"]
|
|
||||||
build-backend = "setuptools.build_meta"
|
|
||||||
Loading…
Reference in a new issue