mirror of
https://github.com/varun-r-mallya/Python-BPF.git
synced 2025-12-31 21:06:25 +00:00
add base for ir gen
This commit is contained in:
@ -147,3 +147,23 @@ class DependencyHandler:
|
||||
int: The number of nodes
|
||||
"""
|
||||
return len(self._nodes)
|
||||
|
||||
def __getitem__(self, name: str) -> DependencyNode:
|
||||
"""
|
||||
Get a node by name using dictionary-style access.
|
||||
|
||||
Args:
|
||||
name: The name of the node to retrieve
|
||||
|
||||
Returns:
|
||||
DependencyNode: The node with the given name
|
||||
|
||||
Raises:
|
||||
KeyError: If no node with the given name exists
|
||||
|
||||
Example:
|
||||
node = handler["some-dep_node_name"]
|
||||
"""
|
||||
if name not in self._nodes:
|
||||
raise KeyError(f"No node with name '{name}' found")
|
||||
return self._nodes[name]
|
||||
|
||||
@ -8,10 +8,28 @@ class IRGenerator:
|
||||
def __init__(self, module, handler: DependencyHandler):
|
||||
self.module = module
|
||||
self.handler: DependencyHandler = handler
|
||||
self.generated: list[str] = []
|
||||
if not handler.is_ready:
|
||||
raise ImportError(
|
||||
"Semantic analysis of vmlinux imports failed. Cannot generate IR"
|
||||
)
|
||||
for struct in handler:
|
||||
print(struct)
|
||||
self.struct_processor(struct)
|
||||
print()
|
||||
|
||||
def struct_processor(self, struct):
|
||||
if struct.name not in self.generated:
|
||||
print(f"IR generating for {struct.name}")
|
||||
print(f"Struct is {struct}")
|
||||
for dependency in struct.depends_on:
|
||||
if dependency not in self.generated:
|
||||
dep_node_from_dependency = self.handler[dependency]
|
||||
self.struct_processor(dep_node_from_dependency)
|
||||
self.generated.append(dependency)
|
||||
# write actual processor logic here after assuming all dependencies are resolved
|
||||
# this part cannot yet resolve circular dependencies. Gets stuck on an infinite loop during that.
|
||||
self.generated.append(struct.name)
|
||||
|
||||
|
||||
def struct_name_generator(self, ):
|
||||
pass
|
||||
|
||||
Reference in New Issue
Block a user