1 files changed, 21 insertions, 69 deletions

diff --git a/net/sctp/proc.c b/net/sctp/proc.c
index 537545ebcb0e..17d0155d9de3 100644
--- a/net/sctp/proc.c
+++ b/net/sctp/proc.c
@@ -101,25 +101,6 @@ static const struct file_operations sctp_snmp_seq_fops = {
         .release = single_release_net,
 };
 
-/* Set up the proc fs entry for 'snmp' object. */
-int __net_init sctp_snmp_proc_init(struct net *net)
-{
-        struct proc_dir_entry *p;
-
-        p = proc_create("snmp", S_IRUGO, net->sctp.proc_net_sctp,
-                        &sctp_snmp_seq_fops);
-        if (!p)
-                return -ENOMEM;
-
-        return 0;
-}
-
-/* Cleanup the proc fs entry for 'snmp' object. */
-void sctp_snmp_proc_exit(struct net *net)
-{
-        remove_proc_entry("snmp", net->sctp.proc_net_sctp);
-}
-
 /* Dump local addresses of an association/endpoint. */
 static void sctp_seq_dump_local_addrs(struct seq_file *seq, struct sctp_ep_common *epb)
 {
@@ -259,25 +240,6 @@ static const struct file_operations sctp_eps_seq_fops = {
         .release = seq_release_net,
 };
 
-/* Set up the proc fs entry for 'eps' object. */
-int __net_init sctp_eps_proc_init(struct net *net)
-{
-        struct proc_dir_entry *p;
-
-        p = proc_create("eps", S_IRUGO, net->sctp.proc_net_sctp,
-                        &sctp_eps_seq_fops);
-        if (!p)
-                return -ENOMEM;
-
-        return 0;
-}
-
-/* Cleanup the proc fs entry for 'eps' object. */
-void sctp_eps_proc_exit(struct net *net)
-{
-        remove_proc_entry("eps", net->sctp.proc_net_sctp);
-}
-
 struct sctp_ht_iter {
         struct seq_net_private p;
         struct rhashtable_iter hti;
@@ -390,25 +352,6 @@ static const struct file_operations sctp_assocs_seq_fops = {
         .release = seq_release_net,
 };
 
-/* Set up the proc fs entry for 'assocs' object. */
-int __net_init sctp_assocs_proc_init(struct net *net)
-{
-        struct proc_dir_entry *p;
-
-        p = proc_create("assocs", S_IRUGO, net->sctp.proc_net_sctp,
-                        &sctp_assocs_seq_fops);
-        if (!p)
-                return -ENOMEM;
-
-        return 0;
-}
-
-/* Cleanup the proc fs entry for 'assocs' object. */
-void sctp_assocs_proc_exit(struct net *net)
-{
-        remove_proc_entry("assocs", net->sctp.proc_net_sctp);
-}
-
 static int sctp_remaddr_seq_show(struct seq_file *seq, void *v)
 {
         struct sctp_association *assoc;
@@ -488,12 +431,6 @@ static const struct seq_operations sctp_remaddr_ops = {
         .show  = sctp_remaddr_seq_show,
 };
 
-/* Cleanup the proc fs entry for 'remaddr' object. */
-void sctp_remaddr_proc_exit(struct net *net)
-{
-        remove_proc_entry("remaddr", net->sctp.proc_net_sctp);
-}
-
 static int sctp_remaddr_seq_open(struct inode *inode, struct file *file)
 {
         return seq_open_net(inode, file, &sctp_remaddr_ops,
@@ -507,13 +444,28 @@ static const struct file_operations sctp_remaddr_seq_fops = {
         .release = seq_release_net,
 };
 
-int __net_init sctp_remaddr_proc_init(struct net *net)
+/* Set up the proc fs entry for the SCTP protocol. */
+int __net_init sctp_proc_init(struct net *net)
 {
-        struct proc_dir_entry *p;
-
-        p = proc_create("remaddr", S_IRUGO, net->sctp.proc_net_sctp,
-                        &sctp_remaddr_seq_fops);
-        if (!p)
+        net->sctp.proc_net_sctp = proc_net_mkdir(net, "sctp", net->proc_net);
+        if (!net->sctp.proc_net_sctp)
                 return -ENOMEM;
+        if (!proc_create("snmp", S_IRUGO, net->sctp.proc_net_sctp,
+                        &sctp_snmp_seq_fops))
+                goto cleanup;
+        if (!proc_create("eps", S_IRUGO, net->sctp.proc_net_sctp,
+                        &sctp_eps_seq_fops))
+                goto cleanup;
+        if (!proc_create("assocs", S_IRUGO, net->sctp.proc_net_sctp,
+                        &sctp_assocs_seq_fops))
+                goto cleanup;
+        if (!proc_create("remaddr", S_IRUGO, net->sctp.proc_net_sctp,
+                        &sctp_remaddr_seq_fops))
+                goto cleanup;
         return 0;
+
+cleanup:
+        remove_proc_subtree("sctp", net->proc_net);
+        net->sctp.proc_net_sctp = NULL;
+        return -ENOMEM;
 }