Re: [PATCH] sysctl: add support for poll()

From: Lucas De Marchi
Date: Wed Jun 01 2011 - 22:52:11 EST


CC'ing people as suggested by get_maintainer.pl

On Wed, Jun 1, 2011 at 9:14 AM, Lucas De Marchi
<lucas.demarchi@xxxxxxxxxxxxxx> wrote:
> Adding support for poll() in sysctl fs allows userspace to receive
> notifications when an entry in sysctl changes. This way it's possible to
> know when hostname/domainname is changed due to the respective syscall
> has been called or its file under /proc/sys has been written to.
>
> Signed-off-by: Lucas De Marchi <lucas.demarchi@xxxxxxxxxxxxxx>
> ---
>  fs/proc/proc_sysctl.c   |   40 ++++++++++++++++++++++++++++++++++++++++
>  include/linux/sysctl.h  |    8 ++++++++
>  include/linux/utsname.h |   16 ++++++++++++++++
>  kernel/sys.c            |    2 ++
>  kernel/utsname_sysctl.c |   36 ++++++++++++++++++++++++++++++++++++
>  5 files changed, 102 insertions(+), 0 deletions(-)
>
> diff --git a/fs/proc/proc_sysctl.c b/fs/proc/proc_sysctl.c
> index f50133c..2e5d3ec 100644
> --- a/fs/proc/proc_sysctl.c
> +++ b/fs/proc/proc_sysctl.c
> @@ -3,6 +3,7 @@
>  */
>  #include <linux/init.h>
>  #include <linux/sysctl.h>
> +#include <linux/poll.h>
>  #include <linux/proc_fs.h>
>  #include <linux/security.h>
>  #include <linux/namei.h>
> @@ -176,6 +177,43 @@ static ssize_t proc_sys_write(struct file *filp, const char __user *buf,
>        return proc_sys_call_handler(filp, (void __user *)buf, count, ppos, 1);
>  }
>
> +static int proc_sys_open(struct inode *inode, struct file *filp)
> +{
> +       struct ctl_table *table = PROC_I(inode)->sysctl_entry;
> +
> +       if (table->poll) {
> +               unsigned long event = atomic_read(&table->poll->event);
> +
> +               filp->private_data = (void *)event;
> +       }
> +
> +       return 0;
> +}
> +
> +static unsigned int proc_sys_poll(struct file *filp, poll_table *wait)
> +{
> +       struct inode *inode = filp->f_path.dentry->d_inode;
> +       struct ctl_table *table = PROC_I(inode)->sysctl_entry;
> +       unsigned long event = (unsigned long)filp->private_data;
> +       unsigned int ret = POLLIN | POLLRDNORM;
> +
> +       if (!table->proc_handler)
> +               goto out;
> +
> +       if (!table->poll)
> +               goto out;
> +
> +       poll_wait(filp, &table->poll->wait, wait);
> +
> +       if (event != atomic_read(&table->poll->event)) {
> +               filp->private_data = (void *)(unsigned long)atomic_read(
> +                                               &table->poll->event);
> +               ret = POLLIN | POLLRDNORM | POLLERR | POLLPRI;
> +       }
> +
> +out:
> +       return ret;
> +}
>
>  static int proc_sys_fill_cache(struct file *filp, void *dirent,
>                                filldir_t filldir,
> @@ -367,6 +405,8 @@ static int proc_sys_getattr(struct vfsmount *mnt, struct dentry *dentry, struct
>  }
>
>  static const struct file_operations proc_sys_file_operations = {
> +       .open           = proc_sys_open,
> +       .poll           = proc_sys_poll,
>        .read           = proc_sys_read,
>        .write          = proc_sys_write,
>        .llseek         = default_llseek,
> diff --git a/include/linux/sysctl.h b/include/linux/sysctl.h
> index 11684d9..96c89ba 100644
> --- a/include/linux/sysctl.h
> +++ b/include/linux/sysctl.h
> @@ -25,6 +25,7 @@
>  #include <linux/kernel.h>
>  #include <linux/types.h>
>  #include <linux/compiler.h>
> +#include <linux/wait.h>
>
>  struct completion;
>
> @@ -1011,6 +1012,12 @@ extern int proc_do_large_bitmap(struct ctl_table *, int,
>  * cover common cases.
>  */
>
> +/* Support for userspace poll() to watch for changes */
> +struct ctl_table_poll {
> +       atomic_t event;
> +       wait_queue_head_t wait;
> +};
> +
>  /* A sysctl table is an array of struct ctl_table: */
>  struct ctl_table
>  {
> @@ -1021,6 +1028,7 @@ struct ctl_table
>        struct ctl_table *child;
>        struct ctl_table *parent;       /* Automatically set */
>        proc_handler *proc_handler;     /* Callback for text formatting */
> +       struct ctl_table_poll *poll;
>        void *extra1;
>        void *extra2;
>  };
> diff --git a/include/linux/utsname.h b/include/linux/utsname.h
> index 4e5b021..c714ed7 100644
> --- a/include/linux/utsname.h
> +++ b/include/linux/utsname.h
> @@ -37,6 +37,14 @@ struct new_utsname {
>  #include <linux/nsproxy.h>
>  #include <linux/err.h>
>
> +enum uts_proc {
> +       UTS_PROC_OSTYPE,
> +       UTS_PROC_OSRELEASE,
> +       UTS_PROC_VERSION,
> +       UTS_PROC_HOSTNAME,
> +       UTS_PROC_DOMAINNAME,
> +};
> +
>  struct user_namespace;
>  extern struct user_namespace init_user_ns;
>
> @@ -80,6 +88,14 @@ static inline struct uts_namespace *copy_utsname(unsigned long flags,
>  }
>  #endif
>
> +#ifdef CONFIG_PROC_SYSCTL
> +extern void uts_proc_notify(enum uts_proc proc);
> +#else
> +static inline void uts_proc_notify(enum uts_proc proc)
> +{
> +}
> +#endif
> +
>  static inline struct new_utsname *utsname(void)
>  {
>        return &current->nsproxy->uts_ns->name;
> diff --git a/kernel/sys.c b/kernel/sys.c
> index e4128b2..ada9cd7 100644
> --- a/kernel/sys.c
> +++ b/kernel/sys.c
> @@ -1211,6 +1211,7 @@ SYSCALL_DEFINE2(sethostname, char __user *, name, int, len)
>                memset(u->nodename + len, 0, sizeof(u->nodename) - len);
>                errno = 0;
>        }
> +       uts_proc_notify(UTS_PROC_HOSTNAME);
>        up_write(&uts_sem);
>        return errno;
>  }
> @@ -1261,6 +1262,7 @@ SYSCALL_DEFINE2(setdomainname, char __user *, name, int, len)
>                memset(u->domainname + len, 0, sizeof(u->domainname) - len);
>                errno = 0;
>        }
> +       uts_proc_notify(UTS_PROC_DOMAINNAME);
>        up_write(&uts_sem);
>        return errno;
>  }
> diff --git a/kernel/utsname_sysctl.c b/kernel/utsname_sysctl.c
> index a2cd77e..e96b766 100644
> --- a/kernel/utsname_sysctl.c
> +++ b/kernel/utsname_sysctl.c
> @@ -13,6 +13,7 @@
>  #include <linux/uts.h>
>  #include <linux/utsname.h>
>  #include <linux/sysctl.h>
> +#include <linux/wait.h>
>
>  static void *get_uts(ctl_table *table, int write)
>  {
> @@ -51,12 +52,28 @@ static int proc_do_uts_string(ctl_table *table, int write,
>        uts_table.data = get_uts(table, write);
>        r = proc_dostring(&uts_table,write,buffer,lenp, ppos);
>        put_uts(table, write, uts_table.data);
> +
> +       if (write) {
> +               atomic_inc(&table->poll->event);
> +               wake_up_interruptible(&table->poll->wait);
> +       }
> +
>        return r;
>  }
>  #else
>  #define proc_do_uts_string NULL
>  #endif
>
> +static struct ctl_table_poll hostname_poll = {
> +       .event          = ATOMIC_INIT(0),
> +       .wait           = __WAIT_QUEUE_HEAD_INITIALIZER(hostname_poll.wait),
> +};
> +
> +static struct ctl_table_poll domainname_poll = {
> +       .event          = ATOMIC_INIT(0),
> +       .wait           = __WAIT_QUEUE_HEAD_INITIALIZER(domainname_poll.wait),
> +};
> +
>  static struct ctl_table uts_kern_table[] = {
>        {
>                .procname       = "ostype",
> @@ -85,6 +102,7 @@ static struct ctl_table uts_kern_table[] = {
>                .maxlen         = sizeof(init_uts_ns.name.nodename),
>                .mode           = 0644,
>                .proc_handler   = proc_do_uts_string,
> +               .poll           = &hostname_poll,
>        },
>        {
>                .procname       = "domainname",
> @@ -92,6 +110,7 @@ static struct ctl_table uts_kern_table[] = {
>                .maxlen         = sizeof(init_uts_ns.name.domainname),
>                .mode           = 0644,
>                .proc_handler   = proc_do_uts_string,
> +               .poll           = &domainname_poll,
>        },
>        {}
>  };
> @@ -105,6 +124,23 @@ static struct ctl_table uts_root_table[] = {
>        {}
>  };
>
> +#ifdef CONFIG_PROC_SYSCTL
> +/*
> + * Notify userspace about a change in a certain entry of uts_kern_table,
> + * identified by the parameter proc.
> + */
> +void uts_proc_notify(enum uts_proc proc)
> +{
> +       struct ctl_table *table = &uts_kern_table[proc];
> +
> +       if (!table->poll)
> +               return;
> +
> +       atomic_inc(&table->poll->event);
> +       wake_up_interruptible(&table->poll->wait);
> +}
> +#endif
> +
>  static int __init utsname_sysctl_init(void)
>  {
>        register_sysctl_table(uts_root_table);
> --
> 1.7.5.2
>
>
--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to majordomo@xxxxxxxxxxxxxxx
More majordomo info at http://vger.kernel.org/majordomo-info.html
Please read the FAQ at http://www.tux.org/lkml/