Blackfin arch: add BUG_ON() checks to make sure we dont overflow the cplb tables

Signed-off-by: Mike Frysinger <vapier.adi@gmail.com>
Signed-off-by: Bryan Wu <cooloney@kernel.org>

1 files changed, 10 insertions, 1 deletions

diff --git a/arch/blackfin/kernel/cplb-nompu/cplbinit.c b/arch/blackfin/kernel/cplb-nompu/cplbinit.c
index afef5c740aa4..e14c37e98ed5 100644
--- a/arch/blackfin/kernel/cplb-nompu/cplbinit.c
+++ b/arch/blackfin/kernel/cplb-nompu/cplbinit.c
@@ -420,8 +420,17 @@ void __init generate_cplb_tables_cpu(unsigned int cpu)
                 }
         }
 
-        /* close tables */
+        /* make sure we locked the kernel start */
+        BUG_ON(cplb.init_i.pos < 2 + cplb_data[ZERO_P].valid);
+        BUG_ON(cplb.init_d.pos < 1 + cplb_data[ZERO_P].valid + cplb_data[L1D_MEM].valid);
+
+        /* make sure we didnt overflow the table */
+        BUG_ON(cplb.init_i.size <= cplb.init_i.pos);
+        BUG_ON(cplb.init_d.size <= cplb.init_d.pos);
+        BUG_ON(cplb.switch_i.size <= cplb.switch_i.pos);
+        BUG_ON(cplb.switch_d.size <= cplb.switch_d.pos);
 
+        /* close tables */
         close_cplbtab(&cplb.init_i);
         close_cplbtab(&cplb.init_d);