Merge git://git.kernel.org/pub/scm/linux/kernel/git/ebiederm/sysctl
[linux-2.6.git] / fs / proc / proc_sysctl.c
index e971ccc..21d836f 100644 (file)
@@ -6,7 +6,9 @@
 #include <linux/poll.h>
 #include <linux/proc_fs.h>
 #include <linux/security.h>
+#include <linux/sched.h>
 #include <linux/namei.h>
+#include <linux/mm.h>
 #include <linux/module.h>
 #include "internal.h"
 
@@ -33,12 +35,10 @@ static struct ctl_table root_table[] = {
        { }
 };
 static struct ctl_table_root sysctl_table_root = {
-       .default_set.dir.list = LIST_HEAD_INIT(sysctl_table_root.default_set.dir.list),
        .default_set.dir.header = {
                {{.count = 1,
                  .nreg = 1,
-                 .ctl_table = root_table,
-                 .ctl_entry = LIST_HEAD_INIT(sysctl_table_root.default_set.dir.header.ctl_entry),}},
+                 .ctl_table = root_table }},
                .ctl_table_arg = root_table,
                .root = &sysctl_table_root,
                .set = &sysctl_table_root.default_set,
@@ -52,7 +52,6 @@ static int sysctl_follow_link(struct ctl_table_header **phead,
        struct ctl_table **pentry, struct nsproxy *namespaces);
 static int insert_links(struct ctl_table_header *head);
 static void put_links(struct ctl_table_header *header);
-static int sysctl_check_dups(struct ctl_dir *dir, struct ctl_table *table);
 
 static void sysctl_print_dir(struct ctl_dir *dir)
 {
@@ -76,33 +75,89 @@ static int namecmp(const char *name1, int len1, const char *name2, int len2)
        return cmp;
 }
 
+/* Called under sysctl_lock */
 static struct ctl_table *find_entry(struct ctl_table_header **phead,
        struct ctl_dir *dir, const char *name, int namelen)
 {
        struct ctl_table_header *head;
        struct ctl_table *entry;
+       struct rb_node *node = dir->root.rb_node;
 
-       list_for_each_entry(head, &dir->list, ctl_entry) {
-               if (head->unregistering)
-                       continue;
-               for (entry = head->ctl_table; entry->procname; entry++) {
-                       const char *procname = entry->procname;
-                       if (namecmp(procname, strlen(procname), name, namelen) == 0) {
-                               *phead = head;
-                               return entry;
-                       }
+       while (node)
+       {
+               struct ctl_node *ctl_node;
+               const char *procname;
+               int cmp;
+
+               ctl_node = rb_entry(node, struct ctl_node, node);
+               head = ctl_node->header;
+               entry = &head->ctl_table[ctl_node - head->node];
+               procname = entry->procname;
+
+               cmp = namecmp(name, namelen, procname, strlen(procname));
+               if (cmp < 0)
+                       node = node->rb_left;
+               else if (cmp > 0)
+                       node = node->rb_right;
+               else {
+                       *phead = head;
+                       return entry;
                }
        }
        return NULL;
 }
 
+static int insert_entry(struct ctl_table_header *head, struct ctl_table *entry)
+{
+       struct rb_node *node = &head->node[entry - head->ctl_table].node;
+       struct rb_node **p = &head->parent->root.rb_node;
+       struct rb_node *parent = NULL;
+       const char *name = entry->procname;
+       int namelen = strlen(name);
+
+       while (*p) {
+               struct ctl_table_header *parent_head;
+               struct ctl_table *parent_entry;
+               struct ctl_node *parent_node;
+               const char *parent_name;
+               int cmp;
+
+               parent = *p;
+               parent_node = rb_entry(parent, struct ctl_node, node);
+               parent_head = parent_node->header;
+               parent_entry = &parent_head->ctl_table[parent_node - parent_head->node];
+               parent_name = parent_entry->procname;
+
+               cmp = namecmp(name, namelen, parent_name, strlen(parent_name));
+               if (cmp < 0)
+                       p = &(*p)->rb_left;
+               else if (cmp > 0)
+                       p = &(*p)->rb_right;
+               else {
+                       printk(KERN_ERR "sysctl duplicate entry: ");
+                       sysctl_print_dir(head->parent);
+                       printk(KERN_CONT "/%s\n", entry->procname);
+                       return -EEXIST;
+               }
+       }
+
+       rb_link_node(node, parent, p);
+       return 0;
+}
+
+static void erase_entry(struct ctl_table_header *head, struct ctl_table *entry)
+{
+       struct rb_node *node = &head->node[entry - head->ctl_table].node;
+
+       rb_erase(node, &head->parent->root);
+}
+
 static void init_header(struct ctl_table_header *head,
        struct ctl_table_root *root, struct ctl_table_set *set,
-       struct ctl_table *table)
+       struct ctl_node *node, struct ctl_table *table)
 {
        head->ctl_table = table;
        head->ctl_table_arg = table;
-       INIT_LIST_HEAD(&head->ctl_entry);
        head->used = 0;
        head->count = 1;
        head->nreg = 1;
@@ -110,28 +165,42 @@ static void init_header(struct ctl_table_header *head,
        head->root = root;
        head->set = set;
        head->parent = NULL;
+       head->node = node;
+       if (node) {
+               struct ctl_table *entry;
+               for (entry = table; entry->procname; entry++, node++) {
+                       rb_init_node(&node->node);
+                       node->header = head;
+               }
+       }
 }
 
 static void erase_header(struct ctl_table_header *head)
 {
-       list_del_init(&head->ctl_entry);
+       struct ctl_table *entry;
+       for (entry = head->ctl_table; entry->procname; entry++)
+               erase_entry(head, entry);
 }
 
 static int insert_header(struct ctl_dir *dir, struct ctl_table_header *header)
 {
+       struct ctl_table *entry;
        int err;
 
-       err = sysctl_check_dups(dir, header->ctl_table);
-       if (err)
-               return err;
-
        dir->header.nreg++;
        header->parent = dir;
        err = insert_links(header);
        if (err)
                goto fail_links;
-       list_add_tail(&header->ctl_entry, &header->parent->list);
+       for (entry = header->ctl_table; entry->procname; entry++) {
+               err = insert_entry(header, entry);
+               if (err)
+                       goto fail;
+       }
        return 0;
+fail:
+       erase_header(header);
+       put_links(header);
 fail_links:
        header->parent = NULL;
        drop_sysctl_table(&dir->header);
@@ -241,19 +310,14 @@ static struct ctl_table *lookup_entry(struct ctl_table_header **phead,
        return entry;
 }
 
-static struct ctl_table_header *next_usable_entry(struct ctl_dir *dir,
-                                                 struct list_head *tmp)
+static struct ctl_node *first_usable_entry(struct rb_node *node)
 {
-       struct ctl_table_header *head;
-
-       for (tmp = tmp->next; tmp != &dir->list; tmp = tmp->next) {
-               head = list_entry(tmp, struct ctl_table_header, ctl_entry);
+       struct ctl_node *ctl_node;
 
-               if (!head->ctl_table->procname ||
-                   !use_table(head))
-                       continue;
-
-               return head;
+       for (;node; node = rb_next(node)) {
+               ctl_node = rb_entry(node, struct ctl_node, node);
+               if (use_table(ctl_node->header))
+                       return ctl_node;
        }
        return NULL;
 }
@@ -261,14 +325,17 @@ static struct ctl_table_header *next_usable_entry(struct ctl_dir *dir,
 static void first_entry(struct ctl_dir *dir,
        struct ctl_table_header **phead, struct ctl_table **pentry)
 {
-       struct ctl_table_header *head;
+       struct ctl_table_header *head = NULL;
        struct ctl_table *entry = NULL;
+       struct ctl_node *ctl_node;
 
        spin_lock(&sysctl_lock);
-       head = next_usable_entry(dir, &dir->list);
+       ctl_node = first_usable_entry(rb_first(&dir->root));
        spin_unlock(&sysctl_lock);
-       if (head)
-               entry = head->ctl_table;
+       if (ctl_node) {
+               head = ctl_node->header;
+               entry = &head->ctl_table[ctl_node - head->node];
+       }
        *phead = head;
        *pentry = entry;
 }
@@ -277,15 +344,17 @@ static void next_entry(struct ctl_table_header **phead, struct ctl_table **pentr
 {
        struct ctl_table_header *head = *phead;
        struct ctl_table *entry = *pentry;
+       struct ctl_node *ctl_node = &head->node[entry - head->ctl_table];
 
-       entry++;
-       if (!entry->procname) {
-               spin_lock(&sysctl_lock);
-               unuse_table(head);
-               head = next_usable_entry(head->parent, &head->ctl_entry);
-               spin_unlock(&sysctl_lock);
-               if (head)
-                       entry = head->ctl_table;
+       spin_lock(&sysctl_lock);
+       unuse_table(head);
+
+       ctl_node = first_usable_entry(rb_next(&ctl_node->node));
+       spin_unlock(&sysctl_lock);
+       head = NULL;
+       if (ctl_node) {
+               head = ctl_node->header;
+               entry = &head->ctl_table[ctl_node - head->node];
        }
        *phead = head;
        *pentry = entry;
@@ -384,10 +453,12 @@ static struct dentry *proc_sys_lookup(struct inode *dir, struct dentry *dentry,
        if (!p)
                goto out;
 
-       ret = sysctl_follow_link(&h, &p, current->nsproxy);
-       err = ERR_PTR(ret);
-       if (ret)
-               goto out;
+       if (S_ISLNK(p->mode)) {
+               ret = sysctl_follow_link(&h, &p, current->nsproxy);
+               err = ERR_PTR(ret);
+               if (ret)
+                       goto out;
+       }
 
        err = ERR_PTR(-ENOMEM);
        inode = proc_sys_make_inode(dir->i_sb, h ? h : head, p);
@@ -456,20 +527,32 @@ static ssize_t proc_sys_write(struct file *filp, const char __user *buf,
 
 static int proc_sys_open(struct inode *inode, struct file *filp)
 {
+       struct ctl_table_header *head = grab_header(inode);
        struct ctl_table *table = PROC_I(inode)->sysctl_entry;
 
+       /* sysctl was unregistered */
+       if (IS_ERR(head))
+               return PTR_ERR(head);
+
        if (table->poll)
                filp->private_data = proc_sys_poll_event(table->poll);
 
+       sysctl_head_finish(head);
+
        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_header *head = grab_header(inode);
        struct ctl_table *table = PROC_I(inode)->sysctl_entry;
-       unsigned long event = (unsigned long)filp->private_data;
        unsigned int ret = DEFAULT_POLLMASK;
+       unsigned long event;
+
+       /* sysctl was unregistered */
+       if (IS_ERR(head))
+               return POLLERR | POLLHUP;
 
        if (!table->proc_handler)
                goto out;
@@ -477,6 +560,7 @@ static unsigned int proc_sys_poll(struct file *filp, poll_table *wait)
        if (!table->poll)
                goto out;
 
+       event = (unsigned long)filp->private_data;
        poll_wait(filp, &table->poll->wait, wait);
 
        if (event != atomic_read(&table->poll->event)) {
@@ -485,6 +569,8 @@ static unsigned int proc_sys_poll(struct file *filp, poll_table *wait)
        }
 
 out:
+       sysctl_head_finish(head);
+
        return ret;
 }
 
@@ -534,10 +620,12 @@ static int proc_sys_link_fill_cache(struct file *filp, void *dirent,
        int err, ret = 0;
        head = sysctl_head_grab(head);
 
-       /* It is not an error if we can not follow the link ignore it */
-       err = sysctl_follow_link(&head, &table, current->nsproxy);
-       if (err)
-               goto out;
+       if (S_ISLNK(table->mode)) {
+               /* It is not an error if we can not follow the link ignore it */
+               err = sysctl_follow_link(&head, &table, current->nsproxy);
+               if (err)
+                       goto out;
+       }
 
        ret = proc_sys_fill_cache(filp, dirent, filldir, head, table);
 out:
@@ -767,9 +855,9 @@ static struct ctl_dir *find_subdir(struct ctl_dir *dir,
        entry = find_entry(&head, dir, name, namelen);
        if (!entry)
                return ERR_PTR(-ENOENT);
-       if (S_ISDIR(entry->mode))
-               return container_of(head, struct ctl_dir, header);
-       return ERR_PTR(-ENOTDIR);
+       if (!S_ISDIR(entry->mode))
+               return ERR_PTR(-ENOTDIR);
+       return container_of(head, struct ctl_dir, header);
 }
 
 static struct ctl_dir *new_dir(struct ctl_table_set *set,
@@ -777,30 +865,45 @@ static struct ctl_dir *new_dir(struct ctl_table_set *set,
 {
        struct ctl_table *table;
        struct ctl_dir *new;
+       struct ctl_node *node;
        char *new_name;
 
-       new = kzalloc(sizeof(*new) + sizeof(struct ctl_table)*2 +
-                     namelen + 1, GFP_KERNEL);
+       new = kzalloc(sizeof(*new) + sizeof(struct ctl_node) +
+                     sizeof(struct ctl_table)*2 +  namelen + 1,
+                     GFP_KERNEL);
        if (!new)
                return NULL;
 
-       table = (struct ctl_table *)(new + 1);
+       node = (struct ctl_node *)(new + 1);
+       table = (struct ctl_table *)(node + 1);
        new_name = (char *)(table + 2);
        memcpy(new_name, name, namelen);
        new_name[namelen] = '\0';
-       INIT_LIST_HEAD(&new->list);
        table[0].procname = new_name;
        table[0].mode = S_IFDIR|S_IRUGO|S_IXUGO;
-       init_header(&new->header, set->dir.header.root, set, table);
+       init_header(&new->header, set->dir.header.root, set, node, table);
 
        return new;
 }
 
+/**
+ * get_subdir - find or create a subdir with the specified name.
+ * @dir:  Directory to create the subdirectory in
+ * @name: The name of the subdirectory to find or create
+ * @namelen: The length of name
+ *
+ * Takes a directory with an elevated reference count so we know that
+ * if we drop the lock the directory will not go away.  Upon success
+ * the reference is moved from @dir to the returned subdirectory.
+ * Upon error an error code is returned and the reference on @dir is
+ * simply dropped.
+ */
 static struct ctl_dir *get_subdir(struct ctl_dir *dir,
                                  const char *name, int namelen)
 {
        struct ctl_table_set *set = dir->header.set;
        struct ctl_dir *subdir, *new = NULL;
+       int err;
 
        spin_lock(&sysctl_lock);
        subdir = find_subdir(dir, name, namelen);
@@ -816,13 +919,17 @@ static struct ctl_dir *get_subdir(struct ctl_dir *dir,
        if (!new)
                goto failed;
 
+       /* Was the subdir added while we dropped the lock? */
        subdir = find_subdir(dir, name, namelen);
        if (!IS_ERR(subdir))
                goto found;
        if (PTR_ERR(subdir) != -ENOENT)
                goto failed;
 
-       if (insert_header(dir, &new->header))
+       /* Nope.  Use the our freshly made directory entry. */
+       err = insert_header(dir, &new->header);
+       subdir = ERR_PTR(err);
+       if (err)
                goto failed;
        subdir = new;
 found:
@@ -864,10 +971,6 @@ static int sysctl_follow_link(struct ctl_table_header **phead,
        struct ctl_dir *dir;
        int ret;
 
-       /* Get out quickly if not a link */
-       if (!S_ISLNK((*pentry)->mode))
-               return 0;
-
        ret = 0;
        spin_lock(&sysctl_lock);
        root = (*pentry)->data;
@@ -892,40 +995,6 @@ static int sysctl_follow_link(struct ctl_table_header **phead,
        return ret;
 }
 
-static int sysctl_check_table_dups(struct ctl_dir *dir, struct ctl_table *old,
-       struct ctl_table *table)
-{
-       struct ctl_table *entry, *test;
-       int error = 0;
-
-       for (entry = old; entry->procname; entry++) {
-               for (test = table; test->procname; test++) {
-                       if (strcmp(entry->procname, test->procname) == 0) {
-                               printk(KERN_ERR "sysctl duplicate entry: ");
-                               sysctl_print_dir(dir);
-                               printk(KERN_CONT "/%s\n", test->procname);
-                               error = -EEXIST;
-                       }
-               }
-       }
-       return error;
-}
-
-static int sysctl_check_dups(struct ctl_dir *dir, struct ctl_table *table)
-{
-       struct ctl_table_header *head;
-       int error = 0;
-
-       list_for_each_entry(head, &dir->list, ctl_entry) {
-               if (head->unregistering)
-                       continue;
-               if (head->parent != dir)
-                       continue;
-               error = sysctl_check_table_dups(dir, head->ctl_table, table);
-       }
-       return error;
-}
-
 static int sysctl_err(const char *path, struct ctl_table *table, char *fmt, ...)
 {
        struct va_format vaf;
@@ -977,6 +1046,7 @@ static struct ctl_table_header *new_links(struct ctl_dir *dir, struct ctl_table
 {
        struct ctl_table *link_table, *entry, *link;
        struct ctl_table_header *links;
+       struct ctl_node *node;
        char *link_name;
        int nr_entries, name_bytes;
 
@@ -988,6 +1058,7 @@ static struct ctl_table_header *new_links(struct ctl_dir *dir, struct ctl_table
        }
 
        links = kzalloc(sizeof(struct ctl_table_header) +
+                       sizeof(struct ctl_node)*nr_entries +
                        sizeof(struct ctl_table)*(nr_entries + 1) +
                        name_bytes,
                        GFP_KERNEL);
@@ -995,7 +1066,8 @@ static struct ctl_table_header *new_links(struct ctl_dir *dir, struct ctl_table
        if (!links)
                return NULL;
 
-       link_table = (struct ctl_table *)(links + 1);
+       node = (struct ctl_node *)(links + 1);
+       link_table = (struct ctl_table *)(node + nr_entries);
        link_name = (char *)&link_table[nr_entries + 1];
 
        for (link = link_table, entry = table; entry->procname; link++, entry++) {
@@ -1006,7 +1078,7 @@ static struct ctl_table_header *new_links(struct ctl_dir *dir, struct ctl_table
                link->data = link_root;
                link_name += len;
        }
-       init_header(links, dir->header.root, dir->header.set, link_table);
+       init_header(links, dir->header.root, dir->header.set, node, link_table);
        links->nreg = nr_entries;
 
        return links;
@@ -1128,21 +1200,29 @@ struct ctl_table_header *__register_sysctl_table(
        const char *path, struct ctl_table *table)
 {
        struct ctl_table_root *root = set->dir.header.root;
-       struct ctl_table_header *links = NULL;
        struct ctl_table_header *header;
        const char *name, *nextname;
        struct ctl_dir *dir;
+       struct ctl_table *entry;
+       struct ctl_node *node;
+       int nr_entries = 0;
+
+       for (entry = table; entry->procname; entry++)
+               nr_entries++;
 
-       header = kzalloc(sizeof(struct ctl_table_header), GFP_KERNEL);
+       header = kzalloc(sizeof(struct ctl_table_header) +
+                        sizeof(struct ctl_node)*nr_entries, GFP_KERNEL);
        if (!header)
                return NULL;
 
-       init_header(header, root, set, table);
+       node = (struct ctl_node *)(header + 1);
+       init_header(header, root, set, node, table);
        if (sysctl_check_table(path, table))
                goto fail;
 
        spin_lock(&sysctl_lock);
        dir = &set->dir;
+       /* Reference moved down the diretory tree get_subdir */
        dir->header.nreg++;
        spin_unlock(&sysctl_lock);
 
@@ -1177,12 +1257,28 @@ fail_put_dir_locked:
        drop_sysctl_table(&dir->header);
        spin_unlock(&sysctl_lock);
 fail:
-       kfree(links);
        kfree(header);
        dump_stack();
        return NULL;
 }
 
+/**
+ * register_sysctl - register a sysctl table
+ * @path: The path to the directory the sysctl table is in.
+ * @table: the table structure
+ *
+ * Register a sysctl table. @table should be a filled in ctl_table
+ * array. A completely 0 filled entry terminates the table.
+ *
+ * See __register_sysctl_table for more details.
+ */
+struct ctl_table_header *register_sysctl(const char *path, struct ctl_table *table)
+{
+       return __register_sysctl_table(&sysctl_table_root.default_set,
+                                       path, table);
+}
+EXPORT_SYMBOL(register_sysctl);
+
 static char *append_path(const char *path, char *pos, const char *name)
 {
        int namelen;
@@ -1487,15 +1583,14 @@ void setup_sysctl_set(struct ctl_table_set *set,
        struct ctl_table_root *root,
        int (*is_seen)(struct ctl_table_set *))
 {
-       memset(set, sizeof(*set), 0);
+       memset(set, 0, sizeof(*set));
        set->is_seen = is_seen;
-       INIT_LIST_HEAD(&set->dir.list);
-       init_header(&set->dir.header, root, set, root_table);
+       init_header(&set->dir.header, root, set, NULL, root_table);
 }
 
 void retire_sysctl_set(struct ctl_table_set *set)
 {
-       WARN_ON(!list_empty(&set->dir.list));
+       WARN_ON(!RB_EMPTY_ROOT(&set->dir.root));
 }
 
 int __init proc_sys_init(void)