tests: fix a few tests to use the right register on powerpc

Signed-off-by: Naveen N. Rao <naveen.n.rao@linux.vnet.ibm.com>

tests/python/test_trace2.py

@@ -15,7 +15,11 @@ struct Counters { u64 stat1; };
 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;

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;