Make the Process events connector use task watchers instead of hooking the
paths it's interested in.

Signed-off-by: Matt Helsley <[EMAIL PROTECTED]>
---
 drivers/connector/cn_proc.c |   51 +++++++++++++++++++++++++++++++-------------
 fs/exec.c                   |    1 
 include/linux/cn_proc.h     |   21 ------------------
 kernel/exit.c               |    2 -
 kernel/fork.c               |    2 -
 kernel/sys.c                |    8 ------
 6 files changed, 36 insertions(+), 49 deletions(-)

Index: linux-2.6.19/drivers/connector/cn_proc.c
===================================================================
--- linux-2.6.19.orig/drivers/connector/cn_proc.c
+++ linux-2.6.19/drivers/connector/cn_proc.c
@@ -44,19 +44,20 @@ static inline void get_seq(__u32 *ts, in
        *ts = get_cpu_var(proc_event_counts)++;
        *cpu = smp_processor_id();
        put_cpu_var(proc_event_counts);
 }
 
-void proc_fork_connector(struct task_struct *task)
+static int proc_fork_connector(unsigned long clone_flags,
+                              struct task_struct *task)
 {
        struct cn_msg *msg;
        struct proc_event *ev;
        __u8 buffer[CN_PROC_MSG_SIZE];
        struct timespec ts;
 
        if (atomic_read(&proc_event_num_listeners) < 1)
-               return;
+               return 0;
 
        msg = (struct cn_msg*)buffer;
        ev = (struct proc_event*)msg->data;
        get_seq(&msg->seq, &ev->cpu);
        ktime_get_ts(&ts); /* get high res monotonic timestamp */
@@ -70,21 +71,24 @@ void proc_fork_connector(struct task_str
        memcpy(&msg->id, &cn_proc_event_id, sizeof(msg->id));
        msg->ack = 0; /* not used */
        msg->len = sizeof(*ev);
        /*  If cn_netlink_send() failed, the data is not sent */
        cn_netlink_send(msg, CN_IDX_PROC, GFP_KERNEL);
+       return 0;
 }
+DEFINE_TASK_CLONECALL(proc_fork_connector);
 
-void proc_exec_connector(struct task_struct *task)
+static int proc_exec_connector(unsigned long ignore,
+                              struct task_struct *task)
 {
        struct cn_msg *msg;
        struct proc_event *ev;
        struct timespec ts;
        __u8 buffer[CN_PROC_MSG_SIZE];
 
        if (atomic_read(&proc_event_num_listeners) < 1)
-               return;
+               return 0;
 
        msg = (struct cn_msg*)buffer;
        ev = (struct proc_event*)msg->data;
        get_seq(&msg->seq, &ev->cpu);
        ktime_get_ts(&ts); /* get high res monotonic timestamp */
@@ -95,21 +99,23 @@ void proc_exec_connector(struct task_str
 
        memcpy(&msg->id, &cn_proc_event_id, sizeof(msg->id));
        msg->ack = 0; /* not used */
        msg->len = sizeof(*ev);
        cn_netlink_send(msg, CN_IDX_PROC, GFP_KERNEL);
+       return 0;
 }
+DEFINE_TASK_EXECCALL(proc_exec_connector);
 
-void proc_id_connector(struct task_struct *task, int which_id)
+static int process_change_id(unsigned long which_id, struct task_struct *task)
 {
        struct cn_msg *msg;
        struct proc_event *ev;
        __u8 buffer[CN_PROC_MSG_SIZE];
        struct timespec ts;
 
        if (atomic_read(&proc_event_num_listeners) < 1)
-               return;
+               return 0;
 
        msg = (struct cn_msg*)buffer;
        ev = (struct proc_event*)msg->data;
        ev->what = which_id;
        ev->event_data.id.process_pid = task->pid;
@@ -119,47 +125,64 @@ void proc_id_connector(struct task_struc
                ev->event_data.id.e.euid = task->euid;
        } else if (which_id == PROC_EVENT_GID) {
                ev->event_data.id.r.rgid = task->gid;
                ev->event_data.id.e.egid = task->egid;
        } else
-               return;
+               return 0;
        get_seq(&msg->seq, &ev->cpu);
        ktime_get_ts(&ts); /* get high res monotonic timestamp */
        ev->timestamp_ns = timespec_to_ns(&ts);
 
        memcpy(&msg->id, &cn_proc_event_id, sizeof(msg->id));
        msg->ack = 0; /* not used */
        msg->len = sizeof(*ev);
        cn_netlink_send(msg, CN_IDX_PROC, GFP_KERNEL);
+       return 0;
+}
+
+static int proc_change_uid_connector(unsigned long ignore,
+                                    struct task_struct *task)
+{
+       return process_change_id(PROC_EVENT_UID, task);
+}
+DEFINE_TASK_UIDCALL(proc_change_uid_connector);
+
+static int proc_change_gid_connector(unsigned long ignore,
+                                    struct task_struct *task)
+{
+       return process_change_id(PROC_EVENT_GID, task);
 }
+DEFINE_TASK_GIDCALL(proc_change_gid_connector);
 
-void proc_exit_connector(struct task_struct *task)
+static int proc_exit_connector(unsigned long code, struct task_struct *task)
 {
        struct cn_msg *msg;
        struct proc_event *ev;
        __u8 buffer[CN_PROC_MSG_SIZE];
        struct timespec ts;
 
        if (atomic_read(&proc_event_num_listeners) < 1)
-               return;
+               return 0;
 
        msg = (struct cn_msg*)buffer;
        ev = (struct proc_event*)msg->data;
        get_seq(&msg->seq, &ev->cpu);
        ktime_get_ts(&ts); /* get high res monotonic timestamp */
        ev->timestamp_ns = timespec_to_ns(&ts);
        ev->what = PROC_EVENT_EXIT;
        ev->event_data.exit.process_pid = task->pid;
        ev->event_data.exit.process_tgid = task->tgid;
-       ev->event_data.exit.exit_code = task->exit_code;
+       ev->event_data.exit.exit_code = code;
        ev->event_data.exit.exit_signal = task->exit_signal;
 
        memcpy(&msg->id, &cn_proc_event_id, sizeof(msg->id));
        msg->ack = 0; /* not used */
        msg->len = sizeof(*ev);
        cn_netlink_send(msg, CN_IDX_PROC, GFP_KERNEL);
+       return 0;
 }
+DEFINE_TASK_EXITCALL(proc_exit_connector);
 
 /*
  * Send an acknowledgement message to userspace
  *
  * Use 0 for success, EFOO otherwise.
@@ -226,14 +249,12 @@ static void cn_proc_mcast_ctl(void *data
  */
 static int __init cn_proc_init(void)
 {
        int err;
 
-       if ((err = cn_add_callback(&cn_proc_event_id, "cn_proc",
-                                  &cn_proc_mcast_ctl))) {
+       err = cn_add_callback(&cn_proc_event_id, "cn_proc", &cn_proc_mcast_ctl);
+       if (err)
                printk(KERN_WARNING "cn_proc failed to register\n");
-               return err;
-       }
-       return 0;
+       return err;
 }
 
 module_init(cn_proc_init);
Index: linux-2.6.19/kernel/fork.c
===================================================================
--- linux-2.6.19.orig/kernel/fork.c
+++ linux-2.6.19/kernel/fork.c
@@ -40,11 +40,10 @@
 #include <linux/mount.h>
 #include <linux/profile.h>
 #include <linux/rmap.h>
 #include <linux/acct.h>
 #include <linux/tsacct_kern.h>
-#include <linux/cn_proc.h>
 #include <linux/delayacct.h>
 #include <linux/taskstats_kern.h>
 #include <linux/random.h>
 #include <linux/init.h>
 
@@ -1219,11 +1218,10 @@ static struct task_struct *copy_process(
 
        total_forks++;
        spin_unlock(&current->sighand->siglock);
        write_unlock_irq(&tasklist_lock);
        notify_task_watchers(WATCH_TASK_CLONE, clone_flags, p);
-       proc_fork_connector(p);
        return p;
 
 bad_fork_cleanup_namespaces:
        exit_task_namespaces(p);
 bad_fork_cleanup_mm:
Index: linux-2.6.19/kernel/exit.c
===================================================================
--- linux-2.6.19.orig/kernel/exit.c
+++ linux-2.6.19/kernel/exit.c
@@ -30,11 +30,10 @@
 #include <linux/taskstats_kern.h>
 #include <linux/delayacct.h>
 #include <linux/syscalls.h>
 #include <linux/signal.h>
 #include <linux/posix-timers.h>
-#include <linux/cn_proc.h>
 #include <linux/mutex.h>
 #include <linux/futex.h>
 #include <linux/compat.h>
 #include <linux/pipe_fs_i.h>
 #include <linux/resource.h>
@@ -925,11 +924,10 @@ fastcall NORET_TYPE void do_exit(long co
        module_put(task_thread_info(tsk)->exec_domain->module);
        if (tsk->binfmt)
                module_put(tsk->binfmt->module);
 
        tsk->exit_code = code;
-       proc_exit_connector(tsk);
        exit_notify(tsk);
        exit_task_namespaces(tsk);
        /*
         * This must happen late, after the PID is not
         * hashed anymore:
Index: linux-2.6.19/kernel/sys.c
===================================================================
--- linux-2.6.19.orig/kernel/sys.c
+++ linux-2.6.19/kernel/sys.c
@@ -956,11 +956,10 @@ asmlinkage long sys_setregid(gid_t rgid,
            (egid != (gid_t) -1 && egid != old_rgid))
                current->sgid = new_egid;
        current->fsgid = new_egid;
        current->egid = new_egid;
        current->gid = new_rgid;
-       proc_id_connector(current, PROC_EVENT_GID);
        notify_task_watchers(WATCH_TASK_GID, 0, current);
        return 0;
 }
 
 /*
@@ -991,11 +990,10 @@ asmlinkage long sys_setgid(gid_t gid)
                current->egid = current->fsgid = gid;
        }
        else
                return -EPERM;
 
-       proc_id_connector(current, PROC_EVENT_GID);
        notify_task_watchers(WATCH_TASK_GID, 0, current);
        return 0;
 }
   
 static int set_user(uid_t new_ruid, int dumpclear)
@@ -1079,11 +1077,10 @@ asmlinkage long sys_setreuid(uid_t ruid,
        if (ruid != (uid_t) -1 ||
            (euid != (uid_t) -1 && euid != old_ruid))
                current->suid = current->euid;
        current->fsuid = current->euid;
 
-       proc_id_connector(current, PROC_EVENT_UID);
        notify_task_watchers(WATCH_TASK_UID, 0, current);
 
        return security_task_post_setuid(old_ruid, old_euid, old_suid, 
LSM_SETID_RE);
 }
 
@@ -1126,11 +1123,10 @@ asmlinkage long sys_setuid(uid_t uid)
                smp_wmb();
        }
        current->fsuid = current->euid = uid;
        current->suid = new_suid;
 
-       proc_id_connector(current, PROC_EVENT_UID);
        notify_task_watchers(WATCH_TASK_UID, 0, current);
 
        return security_task_post_setuid(old_ruid, old_euid, old_suid, 
LSM_SETID_ID);
 }
 
@@ -1174,11 +1170,10 @@ asmlinkage long sys_setresuid(uid_t ruid
        }
        current->fsuid = current->euid;
        if (suid != (uid_t) -1)
                current->suid = suid;
 
-       proc_id_connector(current, PROC_EVENT_UID);
        notify_task_watchers(WATCH_TASK_UID, 0, current);
 
        return security_task_post_setuid(old_ruid, old_euid, old_suid, 
LSM_SETID_RES);
 }
 
@@ -1226,11 +1221,10 @@ asmlinkage long sys_setresgid(gid_t rgid
        if (rgid != (gid_t) -1)
                current->gid = rgid;
        if (sgid != (gid_t) -1)
                current->sgid = sgid;
 
-       proc_id_connector(current, PROC_EVENT_GID);
        notify_task_watchers(WATCH_TASK_GID, 0, current);
        return 0;
 }
 
 asmlinkage long sys_getresgid(gid_t __user *rgid, gid_t __user *egid, gid_t 
__user *sgid)
@@ -1267,11 +1261,10 @@ asmlinkage long sys_setfsuid(uid_t uid)
                        smp_wmb();
                }
                current->fsuid = uid;
        }
 
-       proc_id_connector(current, PROC_EVENT_UID);
        notify_task_watchers(WATCH_TASK_UID, 0, current);
 
        security_task_post_setuid(old_fsuid, (uid_t)-1, (uid_t)-1, 
LSM_SETID_FS);
 
        return old_fsuid;
@@ -1294,11 +1287,10 @@ asmlinkage long sys_setfsgid(gid_t gid)
                if (gid != old_fsgid) {
                        current->mm->dumpable = suid_dumpable;
                        smp_wmb();
                }
                current->fsgid = gid;
-               proc_id_connector(current, PROC_EVENT_GID);
                notify_task_watchers(WATCH_TASK_GID, 0, current);
        }
        return old_fsgid;
 }
 
Index: linux-2.6.19/fs/exec.c
===================================================================
--- linux-2.6.19.orig/fs/exec.c
+++ linux-2.6.19/fs/exec.c
@@ -1085,11 +1085,10 @@ int search_binary_handler(struct linux_b
                                        fput(bprm->file);
                                bprm->file = NULL;
                                current->did_exec = 1;
                                notify_task_watchers(WATCH_TASK_EXEC, 0,
                                                     current);
-                               proc_exec_connector(current);
                                return retval;
                        }
                        read_lock(&binfmt_lock);
                        put_binfmt(fmt);
                        if (retval != -ENOEXEC || bprm->mm == NULL)
Index: linux-2.6.19/include/linux/cn_proc.h
===================================================================
--- linux-2.6.19.orig/include/linux/cn_proc.h
+++ linux-2.6.19/include/linux/cn_proc.h
@@ -95,27 +95,6 @@ struct proc_event {
                        __u32 exit_code, exit_signal;
                } exit;
        } event_data;
 };
 
-#ifdef __KERNEL__
-#ifdef CONFIG_PROC_EVENTS
-void proc_fork_connector(struct task_struct *task);
-void proc_exec_connector(struct task_struct *task);
-void proc_id_connector(struct task_struct *task, int which_id);
-void proc_exit_connector(struct task_struct *task);
-#else
-static inline void proc_fork_connector(struct task_struct *task)
-{}
-
-static inline void proc_exec_connector(struct task_struct *task)
-{}
-
-static inline void proc_id_connector(struct task_struct *task,
-                                    int which_id)
-{}
-
-static inline void proc_exit_connector(struct task_struct *task)
-{}
-#endif /* CONFIG_PROC_EVENTS */
-#endif /* __KERNEL__ */
 #endif /* CN_PROC_H */

--
-
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to [EMAIL PROTECTED]
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to