[PKT_SCHED] qdisc: Use rtnl registration interface

Signed-off-by: Thomas Graf <tgraf@suug.ch>
Signed-off-by: David S. Miller <davem@davemloft.net>

2 files changed, 8 insertions, 21 deletions

diff --git a/include/net/sch_generic.h b/include/net/sch_generic.h
index 82086392735a..a3f4ddd1d6a8 100644
--- a/include/net/sch_generic.h
+++ b/include/net/sch_generic.h
@@ -5,10 +5,10 @@
 #include <linux/types.h>
 #include <linux/rcupdate.h>
 #include <linux/module.h>
-#include <linux/rtnetlink.h>
 #include <linux/pkt_sched.h>
 #include <linux/pkt_cls.h>
 #include <net/gen_stats.h>
+#include <net/rtnetlink.h>
 
 struct Qdisc_ops;
 struct qdisc_walker;
diff --git a/net/sched/sch_api.c b/net/sched/sch_api.c
index 0b9abea68fd5..b06f20294ac0 100644
--- a/net/sched/sch_api.c
+++ b/net/sched/sch_api.c
@@ -27,7 +27,6 @@
 #include <linux/interrupt.h>
 #include <linux/netdevice.h>
 #include <linux/skbuff.h>
-#include <linux/rtnetlink.h>
 #include <linux/init.h>
 #include <linux/proc_fs.h>
 #include <linux/seq_file.h>
@@ -1239,29 +1238,17 @@ static const struct file_operations psched_fops = {
 
 static int __init pktsched_init(void)
 {
-        struct rtnetlink_link *link_p;
-
-        link_p = rtnetlink_links[PF_UNSPEC];
-
-        /* Setup rtnetlink links. It is made here to avoid
-           exporting large number of public symbols.
-         */
-
-        if (link_p) {
-                link_p[RTM_NEWQDISC-RTM_BASE].doit = tc_modify_qdisc;
-                link_p[RTM_DELQDISC-RTM_BASE].doit = tc_get_qdisc;
-                link_p[RTM_GETQDISC-RTM_BASE].doit = tc_get_qdisc;
-                link_p[RTM_GETQDISC-RTM_BASE].dumpit = tc_dump_qdisc;
-                link_p[RTM_NEWTCLASS-RTM_BASE].doit = tc_ctl_tclass;
-                link_p[RTM_DELTCLASS-RTM_BASE].doit = tc_ctl_tclass;
-                link_p[RTM_GETTCLASS-RTM_BASE].doit = tc_ctl_tclass;
-                link_p[RTM_GETTCLASS-RTM_BASE].dumpit = tc_dump_tclass;
-        }
-
         register_qdisc(&pfifo_qdisc_ops);
         register_qdisc(&bfifo_qdisc_ops);
         proc_net_fops_create("psched", 0, &psched_fops);
 
+        rtnl_register(PF_UNSPEC, RTM_NEWQDISC, tc_modify_qdisc, NULL);
+        rtnl_register(PF_UNSPEC, RTM_DELQDISC, tc_get_qdisc, NULL);
+        rtnl_register(PF_UNSPEC, RTM_GETQDISC, tc_get_qdisc, tc_dump_qdisc);
+        rtnl_register(PF_UNSPEC, RTM_NEWTCLASS, tc_ctl_tclass, NULL);
+        rtnl_register(PF_UNSPEC, RTM_DELTCLASS, tc_ctl_tclass, NULL);
+        rtnl_register(PF_UNSPEC, RTM_GETTCLASS, tc_ctl_tclass, tc_dump_tclass);
+
         return 0;
 }