mirror of
https://github.com/varun-r-mallya/Python-BPF.git
synced 2025-12-31 21:06:25 +00:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from enum import Enum, auto
|
|
from typing import Any, Callable, Dict, List, Optional, TypedDict
|
|
from dataclasses import dataclass
|
|
|
|
from pythonbpf.vmlinux_parser.dependency_node import Field
|
|
|
|
|
|
@dataclass
|
|
class AssignmentType(Enum):
|
|
CONSTANT = auto()
|
|
STRUCT = auto()
|
|
ARRAY = auto() # probably won't be used
|
|
FUNCTION_POINTER = auto()
|
|
POINTER = auto() # again, probably won't be used
|
|
|
|
@dataclass
|
|
class FunctionSignature(TypedDict):
|
|
return_type: str
|
|
param_types: List[str]
|
|
varargs: bool
|
|
|
|
|
|
# Thew name of the assignment will be in the dict that uses this class
|
|
@dataclass
|
|
class AssignmentInfo(TypedDict):
|
|
value_type: AssignmentType
|
|
python_type: type
|
|
value: Optional[Any]
|
|
pointer_level: Optional[int]
|
|
signature: Optional[FunctionSignature] # For function pointers
|
|
# The key of the dict is the name of the field.
|
|
# Value is a tuple that contains the global variable representing that field
|
|
# along with all the information about that field as a Field type.
|
|
members: Optional[Dict[str, tuple[str, Field]]] # For structs.
|