diff --git a/tests/python/test_trace2.py b/tests/python/test_trace2.py index 1210c4ec0acc..92a60fcfc1a9 100755 --- a/tests/python/test_trace2.py +++ b/tests/python/test_trace2.py @@ -15,7 +15,11 @@ BPF_TABLE("hash", struct Ptr, struct Counters, stats, 1024); int count_sched(struct pt_regs *ctx) { +#if defined(__powerpc__) + struct Ptr key = {.ptr=ctx->gpr[3]}; +#else struct Ptr key = {.ptr=ctx->bx}; +#endif struct Counters zleaf = {0}; stats.lookup_or_init(&key, &zleaf)->stat1++; return 0; diff --git a/tests/python/test_trace3.c b/tests/python/test_trace3.c index 7e1f8d284d73..50ec3a2964ed 100644 --- a/tests/python/test_trace3.c +++ b/tests/python/test_trace3.c @@ -28,14 +28,22 @@ static u32 log2l(u64 v) { } int probe_blk_start_request(struct pt_regs *ctx) { +#if defined(__powerpc__) + struct Request rq = {.rq = ctx->gpr[3]}; +#else struct Request rq = {.rq = ctx->di}; +#endif struct Time tm = {.start = bpf_ktime_get_ns()}; requests.update(&rq, &tm); return 0; } int probe_blk_update_request(struct pt_regs *ctx) { +#if defined(__powerpc__) + struct Request rq = {.rq = ctx->gpr[3]}; +#else struct Request rq = {.rq = ctx->di}; +#endif struct Time *tm = requests.lookup(&rq); if (!tm) return 0; u64 delta = bpf_ktime_get_ns() - tm->start;