don't crash when cache information unavailable

1 files changed, 5 insertions, 2 deletions

diff --git a/litmus/color.c b/litmus/color.c
index 2a942d22a676..c3e2a1072360 100644
--- a/litmus/color.c
+++ b/litmus/color.c
@@ -69,10 +69,13 @@ static int __init init_color(void)
         MPRINT("Cache size: %lu  line-size: %lu  ways: %lu  sets: %lu\n",
                         color_cache_info.size, color_cache_info.line_size,
                         color_cache_info.ways, color_cache_info.sets);
+        if (!color_cache_info.size){
+                printk(KERN_WARNING "No cache information found.\n");
+                return -EINVAL;
+        }
         BUG_ON(color_cache_info.size <= 1048576 ||
                         color_cache_info.ways < 15 ||
-                        color_cache_info.line_size != 64 ||
-                        color_cache_info.sets < 4096);
+                        color_cache_info.line_size != 64);
         setup_mask();
         return init_color_lists();
 }