extern int trace_mmap_lock_reg(void);
 extern void trace_mmap_lock_unreg(void);
 
-TRACE_EVENT_FN(mmap_lock_start_locking,
+DECLARE_EVENT_CLASS(mmap_lock,
 
        TP_PROTO(struct mm_struct *mm, const char *memcg_path, bool write),
 
                __entry->mm,
                __get_str(memcg_path),
                __entry->write ? "true" : "false"
-       ),
-
-       trace_mmap_lock_reg, trace_mmap_lock_unreg
+       )
 );
 
+#define DEFINE_MMAP_LOCK_EVENT(name)                                    \
+       DEFINE_EVENT_FN(mmap_lock, name,                                \
+               TP_PROTO(struct mm_struct *mm, const char *memcg_path,  \
+                       bool write),                                    \
+               TP_ARGS(mm, memcg_path, write),                         \
+               trace_mmap_lock_reg, trace_mmap_lock_unreg)
+
+DEFINE_MMAP_LOCK_EVENT(mmap_lock_start_locking);
+DEFINE_MMAP_LOCK_EVENT(mmap_lock_released);
+
 TRACE_EVENT_FN(mmap_lock_acquire_returned,
 
        TP_PROTO(struct mm_struct *mm, const char *memcg_path, bool write,
        trace_mmap_lock_reg, trace_mmap_lock_unreg
 );
 
-TRACE_EVENT_FN(mmap_lock_released,
-
-       TP_PROTO(struct mm_struct *mm, const char *memcg_path, bool write),
-
-       TP_ARGS(mm, memcg_path, write),
-
-       TP_STRUCT__entry(
-               __field(struct mm_struct *, mm)
-               __string(memcg_path, memcg_path)
-               __field(bool, write)
-       ),
-
-       TP_fast_assign(
-               __entry->mm = mm;
-               __assign_str(memcg_path, memcg_path);
-               __entry->write = write;
-       ),
-
-       TP_printk(
-               "mm=%p memcg_path=%s write=%s",
-               __entry->mm,
-               __get_str(memcg_path),
-               __entry->write ? "true" : "false"
-       ),
-
-       trace_mmap_lock_reg, trace_mmap_lock_unreg
-);
-
 #endif /* _TRACE_MMAP_LOCK_H */
 
 /* This part must be outside protection */