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

Signed-off-by: Naveen N. Rao <[email protected]>
drcwr · May 2, 2016 · 0301dda · 0301dda
1 parent 56f2f14
commit 0301dda
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git 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
@@ -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;