varun-r-mallya/python-bpf
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/Python-BPF.git synced 2025-12-31 21:06:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support storing and printing string type

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-09-26 04:17:29 +05:30
parent da8a495da7
commit 737c4d3039
2 changed files with 19 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/execve5.py
View File

@ -27,7 +27,7 @@ def hello(ctx: c_void_p) -> c_int32:
process_id = pid() process_id = pid()
dataobj.pid = process_id dataobj.pid = process_id
dataobj.ts = ts dataobj.ts = ts
print(f"clone called at {ts} by pid {process_id}") print(f"clone called at {ts} by pid {process_id}, str is {strobj}")
events.output(dataobj) events.output(dataobj)
return c_int32(0) return c_int32(0)

21
pythonbpf/bpf_helper_handler.py
View File

@ -75,6 +75,7 @@ def bpf_printk_emitter(call, map_ptr, module, builder, func, local_sym_tab=None,
exprs = [] exprs = []
for value in call.args[0].values: for value in call.args[0].values:
print("Value in f-string:", ast.dump(value))
if isinstance(value, ast.Constant): if isinstance(value, ast.Constant):
if isinstance(value.value, str): if isinstance(value.value, str):
fmt_parts.append(value.value) fmt_parts.append(value.value)
@ -86,10 +87,24 @@ def bpf_printk_emitter(call, map_ptr, module, builder, func, local_sym_tab=None,
"Only string and integer constants are supported in f-string.") "Only string and integer constants are supported in f-string.")
elif isinstance(value, ast.FormattedValue): elif isinstance(value, ast.FormattedValue):
print("Formatted value:", ast.dump(value)) print("Formatted value:", ast.dump(value))
# Assume int for now # TODO: Dirty handling here, only checks for int or str
if isinstance(value.value, ast.Name): if isinstance(value.value, ast.Name):
fmt_parts.append("%lld") if local_sym_tab and value.value.id in local_sym_tab:
exprs.append(value.value) var_ptr, var_type = local_sym_tab[value.value.id]
if isinstance(var_type, ir.IntType):
fmt_parts.append("%lld")
exprs.append(value.value)
elif var_type == ir.PointerType(ir.IntType(8)):
# Case with string
fmt_parts.append("%s")
exprs.append(value.value)
else:
raise NotImplementedError(
"Only integer and pointer types are supported in formatted values.")
print("Formatted value variable:", var_ptr, var_type)
else:
raise ValueError(
f"Variable {value.value.id} not found in local symbol table.")
else: else:
raise NotImplementedError( raise NotImplementedError(
"Only simple variable names are supported in formatted values.") "Only simple variable names are supported in formatted values.")