1 files changed, 5 insertions, 0 deletions

diff --git a/kernel/locking/mutex.c b/kernel/locking/mutex.c
index 5352ce50a97e..0a4a3772fdca 100644
--- a/kernel/locking/mutex.c
+++ b/kernel/locking/mutex.c
@@ -36,6 +36,8 @@
 # include "mutex.h"
 #endif
 
+#include <trace/hooks/dtask.h>
+
 void
 __mutex_init(struct mutex *lock, const char *name, struct lock_class_key *key)
 {
@@ -1002,6 +1004,7 @@ __mutex_lock_common(struct mutex *lock, long state, unsigned int subclass,
 
         waiter.task = current;
 
+        trace_android_vh_mutex_wait_start(lock);
         set_current_state(state);
         for (;;) {
                 /*
@@ -1057,6 +1060,7 @@ __mutex_lock_common(struct mutex *lock, long state, unsigned int subclass,
         spin_lock(&lock->wait_lock);
 acquired:
         __set_current_state(TASK_RUNNING);
+        trace_android_vh_mutex_wait_finish(lock);
 
         if (use_ww_ctx && ww_ctx) {
                 /*
@@ -1087,6 +1091,7 @@ skip_wait:
 
 err:
         __set_current_state(TASK_RUNNING);
+        trace_android_vh_mutex_wait_finish(lock);
         mutex_remove_waiter(lock, &waiter, current);
 err_early_kill:
         spin_unlock(&lock->wait_lock);