1 files changed, 3 insertions, 1 deletions

diff --git a/drivers/video/atafb.c b/drivers/video/atafb.c
index 37624f74e88b..f3aada20fa02 100644
--- a/drivers/video/atafb.c
+++ b/drivers/video/atafb.c
@@ -52,7 +52,6 @@
 #include <linux/errno.h>
 #include <linux/string.h>
 #include <linux/mm.h>
-#include <linux/slab.h>
 #include <linux/delay.h>
 #include <linux/init.h>
 #include <linux/interrupt.h>
@@ -2242,6 +2241,9 @@ static int ext_setcolreg(unsigned int regno, unsigned int red,
         if (!external_vgaiobase)
                 return 1;
 
+        if (regno > 255)
+                return 1;
+
         switch (external_card_type) {
         case IS_VGA:
                 OUTB(0x3c8, regno);