Add module load profile hook

diff -Naur -X dontdiff linux/drivers/oprofile/buffer_sync.c up/drivers/oprofile/buffer_sync.c
@@ -67,6 +67,17 @@
 }
 
  
+static int module_load_notify(struct notifier_block * self, unsigned long val, void * data)
+{
+	sync_cpu_buffers();
+	down(&buffer_sem);
+	add_event_entry(ESCAPE_CODE);
+	add_event_entry(DROP_MODULES_CODE);
+	up(&buffer_sem);
+	return 0;
+}
+
+
 static struct notifier_block exit_task_nb = {
 	.notifier_call	= exit_task_notify,
 };

@@ -79,6 +90,10 @@
 	.notifier_call	= mm_notify,
 };
  
+static struct notifier_block module_load_nb = {
+	.notifier_call = module_load_notify,
+};
+
  
 int sync_start(void)
 {

@@ -91,6 +106,9 @@
 	err = profile_event_register(EXEC_UNMAP, &exec_unmap_nb);
 	if (err)
 		goto out3;
+	err = profile_event_register(MODULE_LOADED, &module_load_nb);
+	if (err)
+		goto out4;
 
 	init_timer(&sync_timer);
 	sync_timer.function = timer_ping;

@@ -98,6 +116,8 @@
 	add_timer(&sync_timer);
 out:
 	return err;
+out4:
+	profile_event_unregister(EXEC_UNMAP, &exec_unmap_nb);
 out3:
 	profile_event_unregister(EXIT_MMAP, &exit_mmap_nb);
 out2:

@@ -111,6 +131,7 @@
 	profile_event_unregister(EXIT_TASK, &exit_task_nb);
 	profile_event_unregister(EXIT_MMAP, &exit_mmap_nb);
 	profile_event_unregister(EXEC_UNMAP, &exec_unmap_nb);
+	profile_event_unregister(MODULE_LOADED, &module_load_nb);
 	del_timer_sync(&sync_timer);
 }
 

diff -Naur -X dontdiff linux/drivers/oprofile/event_buffer.h up/drivers/oprofile/event_buffer.h
@@ -24,12 +24,13 @@
  * then one of the following codes, then the
  * relevant data.
  */
-#define ESCAPE_CODE		~0UL
+#define ESCAPE_CODE			~0UL
 #define CTX_SWITCH_CODE 		1
 #define CPU_SWITCH_CODE 		2
 #define COOKIE_SWITCH_CODE 		3
 #define KERNEL_ENTER_SWITCH_CODE	4
 #define KERNEL_EXIT_SWITCH_CODE		5
+#define DROP_MODULES_CODE		6
  
 /* add data to the event buffer */
 void add_event_entry(unsigned long data);

diff -Naur -X dontdiff linux/include/linux/profile.h up/include/linux/profile.h
@@ -23,7 +23,8 @@
 enum profile_type {
 	EXIT_TASK,
 	EXIT_MMAP,
-	EXEC_UNMAP
+	EXEC_UNMAP,
+	MODULE_LOADED,
 };
 
 #ifdef CONFIG_PROFILING

@@ -41,6 +42,9 @@
 /* exit of all vmas for a task */
 void profile_exit_mmap(struct mm_struct * mm);
 
+/* a module was loaded */
+void profile_module_loaded(void);
+
 int profile_event_register(enum profile_type, struct notifier_block * n);
 
 int profile_event_unregister(enum profile_type, struct notifier_block * n);

@@ -60,6 +64,7 @@
 #define profile_exit_task(a) do { } while (0)
 #define profile_exec_unmap(a) do { } while (0)
 #define profile_exit_mmap(a) do { } while (0)
+#define profile_module_loaded do { } while (0)
  
 #endif /* CONFIG_PROFILING */
  

diff -Naur -X dontdiff linux/kernel/module.c up/kernel/module.c
@@ -31,6 +31,7 @@
 #include <linux/errno.h>
 #include <linux/err.h>
 #include <linux/vermagic.h>
+#include <linux/profile.h>
 #include <asm/uaccess.h>
 #include <asm/semaphore.h>
 #include <asm/pgalloc.h>

@@ -1437,6 +1438,8 @@
 	/* Drop lock so they can recurse */
 	up(&module_mutex);
 
+	profile_module_loaded();
+
 	/* Start the module */
 	ret = mod->init();
 	if (ret < 0) {

diff -Naur -X dontdiff linux/kernel/profile.c up/kernel/profile.c
@@ -51,6 +51,7 @@
 static struct notifier_block * exit_task_notifier;
 static struct notifier_block * exit_mmap_notifier;
 static struct notifier_block * exec_unmap_notifier;
+static struct notifier_block * module_load_notifier;
  
 void profile_exit_task(struct task_struct * task)
 {

@@ -58,6 +59,7 @@
 	notifier_call_chain(&exit_task_notifier, 0, task);
 	up_read(&profile_rwsem);
 }
+
  
 void profile_exit_mmap(struct mm_struct * mm)
 {

@@ -66,6 +68,7 @@
 	up_read(&profile_rwsem);
 }
 
+
 void profile_exec_unmap(struct mm_struct * mm)
 {
 	down_read(&profile_rwsem);

@@ -73,6 +76,15 @@
 	up_read(&profile_rwsem);
 }
 
+
+void profile_module_loaded()
+{
+	down_read(&profile_rwsem);
+	notifier_call_chain(&module_load_notifier, 0, 0);
+	up_read(&profile_rwsem);
+}
+
+
 int profile_event_register(enum profile_type type, struct notifier_block * n)
 {
 	int err = -EINVAL;

@@ -89,6 +101,9 @@
 		case EXEC_UNMAP:
 			err = notifier_chain_register(&exec_unmap_notifier, n);
 			break;
+		case MODULE_LOADED:
+			err = notifier_chain_register(&module_load_notifier, n);
+			break;
 	}
  
 	up_write(&profile_rwsem);

@@ -113,6 +128,9 @@
 		case EXEC_UNMAP:
 			err = notifier_chain_unregister(&exec_unmap_notifier, n);
 			break;
+		case MODULE_LOADED:
+			err = notifier_chain_unregister(&module_load_notifier, n);
+			break;
 	}
 
 	up_write(&profile_rwsem);