fix bug and format.

pull/773/head
pdliyan 2 years ago
parent 9c167c7147
commit 454bf0df58

@ -2,8 +2,9 @@
#![no_main]
use aya_bpf::{
macros::{uprobe, map},
programs::{perf_event, ProbeContext}, maps::PerfEventArray,
macros::{map, uprobe},
maps::PerfEventArray,
programs::ProbeContext,
};
use aya_log_ebpf::{debug, info};
@ -19,9 +20,9 @@ pub struct Args {
a_7: i64,
}
impl Args{
fn new()->Self{
Self{
impl Args {
fn new() -> Self {
Self {
a_0: 0,
a_1: 0,
a_2: 0,
@ -48,7 +49,7 @@ pub fn test_stack_argument(ctx: ProbeContext) -> i32 {
//read argument, and send event
fn try_stack_argument(ctx: ProbeContext) -> Result<i32, i64> {
let args = Args::new();
let mut args = Args::new();
args.a_0 = ctx.arg(0).ok_or(255)?;
args.a_1 = ctx.arg(1).ok_or(255)?;
args.a_2 = ctx.arg(2).ok_or(255)?;
@ -58,7 +59,6 @@ fn try_stack_argument(ctx: ProbeContext) -> Result<i32, i64> {
args.a_6 = ctx.stack_arg(0).ok_or(255)?;
args.a_7 = ctx.stack_arg(1).ok_or(255)?;
EVENTS.output(&ctx, &args, 0);
Ok(0)

@ -17,9 +17,9 @@ pub struct Args {
a_7: i64,
}
impl Args{
fn new()->Self{
Self{
impl Args {
fn new() -> Self {
Self {
a_0: 0,
a_1: 0,
a_2: 0,
@ -74,9 +74,9 @@ async fn stack_argument() {
loop {
let events = buf.read_events(&mut buffer).await.unwrap();
for buf in buffers.iter_mut().task(events.read){
for buf in buffers.iter_mut().task(events.read) {
let ptr = buf.as_ptr() as *const Args;
let data = unsafe{ptr.read_unaligned()};
let data = unsafe { ptr.read_unaligned() };
assert_eq!(data.a_0, 0);
assert_eq!(data.a_1, 1);
assert_eq!(data.a_2, 2);

Loading…
Cancel
Save