1 files changed, 7 insertions, 4 deletions

diff --git a/drivers/rapidio/rio-scan.c b/drivers/rapidio/rio-scan.c
index 45d14cd6b356..e655756e1185 100644
--- a/drivers/rapidio/rio-scan.c
+++ b/drivers/rapidio/rio-scan.c
@@ -947,7 +947,7 @@ static int rio_enum_complete(struct rio_mport *port)
  */
 static int __devinit
 rio_disc_peer(struct rio_net *net, struct rio_mport *port, u16 destid,
-              u8 hopcount)
+              u8 hopcount, struct rio_dev *prev, int prev_port)
 {
         u8 port_num, route_port;
         struct rio_dev *rdev;
@@ -957,6 +957,9 @@ rio_disc_peer(struct rio_net *net, struct rio_mport *port, u16 destid,
         if ((rdev = rio_setup_device(net, port, destid, hopcount, 0))) {
                 /* Add device to the global and bus/net specific list. */
                 list_add_tail(&rdev->net_list, &net->devices);
+                rdev->prev = prev;
+                if (prev && rio_is_switch(prev))
+                        prev->rswitch->nextdev[prev_port] = rdev;
         } else
                 return -1;
 
@@ -998,8 +1001,8 @@ rio_disc_peer(struct rio_net *net, struct rio_mport *port, u16 destid,
                                 if (ndestid == RIO_ANY_DESTID(port->sys_size))
                                         continue;
                                 rio_unlock_device(port, destid, hopcount);
-                                if (rio_disc_peer
-                                    (net, port, ndestid, hopcount + 1) < 0)
+                                if (rio_disc_peer(net, port, ndestid,
+                                        hopcount + 1, rdev, port_num) < 0)
                                         return -1;
                         }
                 }
@@ -1291,7 +1294,7 @@ int __devinit rio_disc_mport(struct rio_mport *mport)
                                                    mport->host_deviceid);
 
                 if (rio_disc_peer(net, mport, RIO_ANY_DESTID(mport->sys_size),
-                                        0) < 0) {
+                                        0, NULL, 0) < 0) {
                         printk(KERN_INFO
                                "RIO: master port %d device has failed discovery\n",
                                mport->id);