1 files changed, 21 insertions, 17 deletions

diff --git a/drivers/ata/libata-eh.c b/drivers/ata/libata-eh.c
index 1d7faab64810..b16c81a24ba3 100644
--- a/drivers/ata/libata-eh.c
+++ b/drivers/ata/libata-eh.c
@@ -2375,6 +2375,22 @@ static int ata_eh_skip_recovery(struct ata_link *link)
         return 1;
 }
 
+static int ata_eh_schedule_probe(struct ata_device *dev)
+{
+        struct ata_eh_context *ehc = &dev->link->eh_context;
+
+        if (!(ehc->i.probe_mask & (1 << dev->devno)) ||
+            (ehc->did_probe_mask & (1 << dev->devno)))
+                return 0;
+
+        ata_eh_detach_dev(dev);
+        ata_dev_init(dev);
+        ehc->did_probe_mask |= (1 << dev->devno);
+        ehc->i.action |= ATA_EH_SOFTRESET;
+
+        return 1;
+}
+
 static int ata_eh_handle_dev_fail(struct ata_device *dev, int err)
 {
         struct ata_eh_context *ehc = &dev->link->eh_context;
@@ -2406,16 +2422,9 @@ static int ata_eh_handle_dev_fail(struct ata_device *dev, int err)
                 if (ata_link_offline(dev->link))
                         ata_eh_detach_dev(dev);
 
-                /* probe if requested */
-                if ((ehc->i.probe_mask & (1 << dev->devno)) &&
-                    !(ehc->did_probe_mask & (1 << dev->devno))) {
-                        ata_eh_detach_dev(dev);
-                        ata_dev_init(dev);
-
+                /* schedule probe if necessary */
+                if (ata_eh_schedule_probe(dev))
                         ehc->tries[dev->devno] = ATA_EH_DEV_TRIES;
-                        ehc->did_probe_mask |= (1 << dev->devno);
-                        ehc->i.action |= ATA_EH_SOFTRESET;
-                }
 
                 return 1;
         } else {
@@ -2492,14 +2501,9 @@ int ata_eh_recover(struct ata_port *ap, ata_prereset_fn_t prereset,
                         if (dev->flags & ATA_DFLAG_DETACH)
                                 ata_eh_detach_dev(dev);
 
-                        if (!ata_dev_enabled(dev) &&
-                            ((ehc->i.probe_mask & (1 << dev->devno)) &&
-                             !(ehc->did_probe_mask & (1 << dev->devno)))) {
-                                ata_eh_detach_dev(dev);
-                                ata_dev_init(dev);
-                                ehc->did_probe_mask |= (1 << dev->devno);
-                                ehc->i.action |= ATA_EH_SOFTRESET;
-                        }
+                        /* schedule probe if necessary */
+                        if (!ata_dev_enabled(dev))
+                                ata_eh_schedule_probe(dev);
                 }
         }