From 44e0f43eeac53648adb4734862f9b36d80853420 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mauricio=20V=C3=A1squez?= <mauricio@kinvolk.io>
Date: Thu, 21 May 2020 11:50:52 -0500
Subject: [PATCH] Fix KFUNC_PROBE return value
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

The KFUNC_PROBE macro is using "void" as return type, this is causing problems
in some tools that have a filtering enable that returns 0.

Reproducer: (Notice that it requires BTF support)

```
$ python opensnoop.py --pid 5
/virtual/main.c:33:21: error: void function '____kretfunc__do_sys_open' should not return a value [-Wreturn-type]
    if (pid != 5) { return 0; }
                    ^      ~
1 error generated.
...
```

Signed-off-by: Mauricio Vásquez <mauricio@kinvolk.io>
---
 src/cc/export/helpers.h | 4 ++--
 tools/klockstat.py      | 6 +++---
 tools/opensnoop.py      | 2 ++
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/cc/export/helpers.h b/src/cc/export/helpers.h
index 79a3879ae35d..2b73d1c7f6e0 100644
--- a/src/cc/export/helpers.h
+++ b/src/cc/export/helpers.h
@@ -1012,7 +1012,7 @@ int raw_tracepoint__##event(struct bpf_raw_tracepoint_args *ctx)
 #define BPF_PROG(name, args...)                                 \
 int name(unsigned long long *ctx);                              \
 __attribute__((always_inline))                                  \
-static void ____##name(unsigned long long *ctx, ##args);        \
+static int ____##name(unsigned long long *ctx, ##args);         \
 int name(unsigned long long *ctx)                               \
 {                                                               \
         _Pragma("GCC diagnostic push")                          \
@@ -1021,7 +1021,7 @@ int name(unsigned long long *ctx)                               \
         _Pragma("GCC diagnostic pop")                           \
         return 0;                                               \
 }                                                               \
-static void ____##name(unsigned long long *ctx, ##args)
+static int ____##name(unsigned long long *ctx, ##args)
 
 #define KFUNC_PROBE(event, args...) \
         BPF_PROG(kfunc__ ## event, args)
diff --git a/tools/klockstat.py b/tools/klockstat.py
index 540dd4e7d1b9..7cb15ad3b839 100755
--- a/tools/klockstat.py
+++ b/tools/klockstat.py
@@ -352,17 +352,17 @@ def stack_id_err(stack_id):
 program_kfunc = """
 KFUNC_PROBE(mutex_unlock, void *lock)
 {
-    do_mutex_unlock_enter();
+    return do_mutex_unlock_enter();
 }
 
 KRETFUNC_PROBE(mutex_lock, void *lock, int ret)
 {
-    do_mutex_lock_return();
+    return do_mutex_lock_return();
 }
 
 KFUNC_PROBE(mutex_lock, void *lock)
 {
-    do_mutex_lock_enter(ctx, 3);
+    return do_mutex_lock_enter(ctx, 3);
 }
 
 """
diff --git a/tools/opensnoop.py b/tools/opensnoop.py
index 995443e3faf3..28fe7559beea 100755
--- a/tools/opensnoop.py
+++ b/tools/opensnoop.py
@@ -197,6 +197,8 @@
     data.ret   = ret;
 
     events.perf_submit(ctx, &data, sizeof(data));
+
+    return 0:
 }
 """