obj-$(CONFIG_NETDEVSIM) += netdevsim.o
 
 netdevsim-objs := \
-       netdev.o dev.o fib.o sdev.o
+       netdev.o dev.o fib.o sdev.o bus.o
 
 ifeq ($(CONFIG_BPF_SYSCALL),y)
 netdevsim-objs += \
 
--- /dev/null
+// SPDX-License-Identifier: GPL-2.0
+/* Copyright (C) 2017 Netronome Systems, Inc.
+ * Copyright (C) 2019 Mellanox Technologies. All rights reserved
+ */
+
+#include <linux/device.h>
+
+#include "netdevsim.h"
+
+struct bus_type nsim_bus = {
+       .name           = DRV_NAME,
+       .dev_name       = DRV_NAME,
+       .num_vf         = nsim_num_vf,
+};
+
+int nsim_bus_init(void)
+{
+       return bus_register(&nsim_bus);
+}
+
+void nsim_bus_exit(void)
+{
+       bus_unregister(&nsim_bus);
+}
 
 
 static struct dentry *nsim_ddir;
 
-static int nsim_num_vf(struct device *dev)
+int nsim_num_vf(struct device *dev)
 {
        struct netdevsim *ns = to_nsim(dev);
 
        return ns->num_vfs;
 }
 
-static struct bus_type nsim_bus = {
-       .name           = DRV_NAME,
-       .dev_name       = DRV_NAME,
-       .num_vf         = nsim_num_vf,
-};
-
 static int nsim_vfs_enable(struct netdevsim *ns, unsigned int num_vfs)
 {
        ns->vfconfigs = kcalloc(num_vfs, sizeof(struct nsim_vf_config),
        if (err)
                goto err_debugfs_destroy;
 
-       err = bus_register(&nsim_bus);
+       err = nsim_bus_init();
        if (err)
                goto err_sdev_exit;
 
        err = rtnl_link_register(&nsim_link_ops);
        if (err)
-               goto err_unreg_bus;
+               goto err_bus_exit;
 
        return 0;
 
-err_unreg_bus:
-       bus_unregister(&nsim_bus);
+err_bus_exit:
+       nsim_bus_exit();
 err_sdev_exit:
        nsim_sdev_exit();
 err_debugfs_destroy:
 static void __exit nsim_module_exit(void)
 {
        rtnl_link_unregister(&nsim_link_ops);
-       bus_unregister(&nsim_bus);
+       nsim_bus_exit();
        nsim_sdev_exit();
        debugfs_remove_recursive(nsim_ddir);
 }
 
 {
        return container_of(ptr, struct netdevsim, dev);
 }
+
+int nsim_num_vf(struct device *dev);
+
+extern struct bus_type nsim_bus;
+
+int nsim_bus_init(void);
+void nsim_bus_exit(void);