|
@@ -5,6 +5,7 @@
|
|
*
|
|
*
|
|
*/
|
|
*/
|
|
|
|
|
|
|
|
+#include <linux/module.h>
|
|
#include "trace.h"
|
|
#include "trace.h"
|
|
|
|
|
|
int ftrace_profile_enable(int event_id)
|
|
int ftrace_profile_enable(int event_id)
|
|
@@ -14,7 +15,8 @@ int ftrace_profile_enable(int event_id)
|
|
|
|
|
|
mutex_lock(&event_mutex);
|
|
mutex_lock(&event_mutex);
|
|
list_for_each_entry(event, &ftrace_events, list) {
|
|
list_for_each_entry(event, &ftrace_events, list) {
|
|
- if (event->id == event_id && event->profile_enable) {
|
|
|
|
|
|
+ if (event->id == event_id && event->profile_enable &&
|
|
|
|
+ try_module_get(event->mod)) {
|
|
ret = event->profile_enable(event);
|
|
ret = event->profile_enable(event);
|
|
break;
|
|
break;
|
|
}
|
|
}
|
|
@@ -32,6 +34,7 @@ void ftrace_profile_disable(int event_id)
|
|
list_for_each_entry(event, &ftrace_events, list) {
|
|
list_for_each_entry(event, &ftrace_events, list) {
|
|
if (event->id == event_id) {
|
|
if (event->id == event_id) {
|
|
event->profile_disable(event);
|
|
event->profile_disable(event);
|
|
|
|
+ module_put(event->mod);
|
|
break;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|