RTNL_FLAG_DOIT_UNLOCKED = 1,
 };
 
-int __rtnl_register(int protocol, int msgtype,
-                   rtnl_doit_func, rtnl_dumpit_func, unsigned int flags);
 void rtnl_register(int protocol, int msgtype,
                   rtnl_doit_func, rtnl_dumpit_func, unsigned int flags);
 int rtnl_register_module(struct module *owner, int protocol, int msgtype,
 
 EXPORT_SYMBOL_GPL(rtnl_register_module);
 
 /**
- * __rtnl_register - Register a rtnetlink message type
+ * rtnl_register - Register a rtnetlink message type
  * @protocol: Protocol family or PF_UNSPEC
  * @msgtype: rtnetlink message type
  * @doit: Function pointer called for each request message
  * The special protocol family PF_UNSPEC may be used to define fallback
  * function pointers for the case when no entry for the specific protocol
  * family exists.
- *
- * Returns 0 on success or a negative error code.
- */
-int __rtnl_register(int protocol, int msgtype,
-                   rtnl_doit_func doit, rtnl_dumpit_func dumpit,
-                   unsigned int flags)
-{
-       return rtnl_register_internal(NULL, protocol, msgtype,
-                                     doit, dumpit, flags);
-}
-EXPORT_SYMBOL_GPL(__rtnl_register);
-
-/**
- * rtnl_register - Register a rtnetlink message type
- *
- * Identical to __rtnl_register() but panics on failure. This is useful
- * as failure of this function is very unlikely, it can only happen due
- * to lack of memory when allocating the chain to store all message
- * handlers for a protocol. Meant for use in init functions where lack
- * of memory implies no sense in continuing.
  */
 void rtnl_register(int protocol, int msgtype,
                   rtnl_doit_func doit, rtnl_dumpit_func dumpit,
                   unsigned int flags)
 {
-       if (__rtnl_register(protocol, msgtype, doit, dumpit, flags) < 0)
-               panic("Unable to register rtnetlink message handler, "
-                     "protocol = %d, message type = %d\n",
-                     protocol, msgtype);
+       int err;
+
+       err = rtnl_register_internal(NULL, protocol, msgtype, doit, dumpit,
+                                    flags);
+       if (err)
+               pr_err("Unable to register rtnetlink message handler, "
+                      "protocol = %d, message type = %d\n", protocol, msgtype);
 }
 EXPORT_SYMBOL_GPL(rtnl_register);
 
 
 
        rtnl_af_register(&inet6_ops);
 
-       err = __rtnl_register(PF_INET6, RTM_GETLINK, NULL, inet6_dump_ifinfo,
-                             0);
+       err = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_GETLINK,
+                                  NULL, inet6_dump_ifinfo, 0);
        if (err < 0)
                goto errout;
 
-       /* Only the first call to __rtnl_register can fail */
-       __rtnl_register(PF_INET6, RTM_NEWADDR, inet6_rtm_newaddr, NULL, 0);
-       __rtnl_register(PF_INET6, RTM_DELADDR, inet6_rtm_deladdr, NULL, 0);
-       __rtnl_register(PF_INET6, RTM_GETADDR, inet6_rtm_getaddr,
-                       inet6_dump_ifaddr, RTNL_FLAG_DOIT_UNLOCKED);
-       __rtnl_register(PF_INET6, RTM_GETMULTICAST, NULL,
-                       inet6_dump_ifmcaddr, 0);
-       __rtnl_register(PF_INET6, RTM_GETANYCAST, NULL,
-                       inet6_dump_ifacaddr, 0);
-       __rtnl_register(PF_INET6, RTM_GETNETCONF, inet6_netconf_get_devconf,
-                       inet6_netconf_dump_devconf, RTNL_FLAG_DOIT_UNLOCKED);
-
+       err = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_NEWADDR,
+                                  inet6_rtm_newaddr, NULL, 0);
+       if (err < 0)
+               goto errout;
+       err = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_DELADDR,
+                                  inet6_rtm_deladdr, NULL, 0);
+       if (err < 0)
+               goto errout;
+       err = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_GETADDR,
+                                  inet6_rtm_getaddr, inet6_dump_ifaddr,
+                                  RTNL_FLAG_DOIT_UNLOCKED);
+       if (err < 0)
+               goto errout;
+       err = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_GETMULTICAST,
+                                  NULL, inet6_dump_ifmcaddr, 0);
+       if (err < 0)
+               goto errout;
+       err = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_GETANYCAST,
+                                  NULL, inet6_dump_ifacaddr, 0);
+       if (err < 0)
+               goto errout;
+       err = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_GETNETCONF,
+                                  inet6_netconf_get_devconf,
+                                  inet6_netconf_dump_devconf,
+                                  RTNL_FLAG_DOIT_UNLOCKED);
+       if (err < 0)
+               goto errout;
        ipv6_addr_label_rtnl_register();
 
        return 0;
 errout:
+       rtnl_unregister_all(PF_INET6);
        rtnl_af_unregister(&inet6_ops);
        unregister_netdevice_notifier(&ipv6_dev_notf);
 errlo:
 
 
 void __init ipv6_addr_label_rtnl_register(void)
 {
-       __rtnl_register(PF_INET6, RTM_NEWADDRLABEL, ip6addrlbl_newdel,
-                       NULL, RTNL_FLAG_DOIT_UNLOCKED);
-       __rtnl_register(PF_INET6, RTM_DELADDRLABEL, ip6addrlbl_newdel,
-                       NULL, RTNL_FLAG_DOIT_UNLOCKED);
-       __rtnl_register(PF_INET6, RTM_GETADDRLABEL, ip6addrlbl_get,
-                       ip6addrlbl_dump, RTNL_FLAG_DOIT_UNLOCKED);
+       rtnl_register(PF_INET6, RTM_NEWADDRLABEL, ip6addrlbl_newdel,
+                     NULL, RTNL_FLAG_DOIT_UNLOCKED);
+       rtnl_register(PF_INET6, RTM_DELADDRLABEL, ip6addrlbl_newdel,
+                     NULL, RTNL_FLAG_DOIT_UNLOCKED);
+       rtnl_register(PF_INET6, RTM_GETADDRLABEL, ip6addrlbl_get,
+                     ip6addrlbl_dump, RTNL_FLAG_DOIT_UNLOCKED);
 }
-
 
        if (ret)
                goto out_kmem_cache_create;
 
-       ret = __rtnl_register(PF_INET6, RTM_GETROUTE, NULL, inet6_dump_fib,
-                             0);
+       ret = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_GETROUTE, NULL,
+                                  inet6_dump_fib, 0);
        if (ret)
                goto out_unregister_subsys;
 
 
        if (ret)
                goto fib6_rules_init;
 
-       ret = -ENOBUFS;
-       if (__rtnl_register(PF_INET6, RTM_NEWROUTE, inet6_rtm_newroute, NULL, 0) ||
-           __rtnl_register(PF_INET6, RTM_DELROUTE, inet6_rtm_delroute, NULL, 0) ||
-           __rtnl_register(PF_INET6, RTM_GETROUTE, inet6_rtm_getroute, NULL,
-                           RTNL_FLAG_DOIT_UNLOCKED))
+       ret = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_NEWROUTE,
+                                  inet6_rtm_newroute, NULL, 0);
+       if (ret < 0)
+               goto out_register_late_subsys;
+
+       ret = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_DELROUTE,
+                                  inet6_rtm_delroute, NULL, 0);
+       if (ret < 0)
+               goto out_register_late_subsys;
+
+       ret = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_GETROUTE,
+                                  inet6_rtm_getroute, NULL,
+                                  RTNL_FLAG_DOIT_UNLOCKED);
+       if (ret < 0)
                goto out_register_late_subsys;
 
        ret = register_netdevice_notifier(&ip6_route_dev_notifier);
        return ret;
 
 out_register_late_subsys:
+       rtnl_unregister_all(PF_INET6);
        unregister_pernet_subsys(&ip6_route_net_late_ops);
 fib6_rules_init:
        fib6_rules_cleanup();