1 files changed, 5 insertions, 2 deletions

diff --git a/drivers/ata/libahci.c b/drivers/ata/libahci.c
index 965842a08743..09620c2ffa0f 100644
--- a/drivers/ata/libahci.c
+++ b/drivers/ata/libahci.c
@@ -35,6 +35,7 @@
 #include <linux/kernel.h>
 #include <linux/gfp.h>
 #include <linux/module.h>
+#include <linux/nospec.h>
 #include <linux/blkdev.h>
 #include <linux/delay.h>
 #include <linux/interrupt.h>
@@ -1146,10 +1147,12 @@ static ssize_t ahci_led_store(struct ata_port *ap, const char *buf,
 
         /* get the slot number from the message */
         pmp = (state & EM_MSG_LED_PMP_SLOT) >> 8;
-        if (pmp < EM_MAX_SLOTS)
+        if (pmp < EM_MAX_SLOTS) {
+                pmp = array_index_nospec(pmp, EM_MAX_SLOTS);
                 emp = &pp->em_priv[pmp];
-        else
+        } else {
                 return -EINVAL;
+        }
 
         /* mask off the activity bits if we are in sw_activity
          * mode, user should turn off sw_activity before setting