Energy: Support new function names in new kernel

sock_aio_read/write -> sock_read/write_iter

Change-Id: I868d148ed380417a5a93e35b8928093b236ab192

diff --git a/energy/energy.c b/energy/energy.c
index 49e238a..73ab3b0 100644 (file)
--- a/energy/energy.c
+++ b/energy/energy.c
@@ -680,6 +680,14 @@ static struct kretprobe sock_aio_write_krp = {
        .data_size = sizeof(struct socket *)
 };
 
+#if LINUX_VERSION_CODE >= KERNEL_VERSION(4, 0, 0)
+static const char sock_aio_read[] = "sock_read_iter";
+static const char sock_aio_write[] = "sock_write_iter";
+#else  /* LINUX_VERSION_CODE >= KERNEL_VERSION(4, 0, 0) */
+static const char sock_aio_read[] = "sock_aio_read";
+static const char sock_aio_write[] = "sock_aio_write";
+#endif /* LINUX_VERSION_CODE >= KERNEL_VERSION(4, 0, 0) */
+
 static struct kern_probe wifi_probes[] = {
        {
                .name = "sock_recvmsg",
@@ -690,11 +698,11 @@ static struct kern_probe wifi_probes[] = {
                .rp = &sock_send_krp,
        },
        {
-               .name = "sock_aio_read",
+               .name = sock_aio_read,
                .rp = &sock_aio_read_krp,
        },
        {
-               .name = "sock_aio_write",
+               .name = sock_aio_write,
                .rp = &sock_aio_write_krp,
        }
 };