From 97343f3d77b896046de2e0819558d48f55642a65 Mon Sep 17 00:00:00 2001
From: jl777 <jameslee777@yahoo.com>
Date: Fri, 27 Apr 2018 15:18:28 +0300
Subject: [PATCH] Handle MoMdepth overflow

---
 src/komodo_ccdata.h   | 2 ++
 src/rpcblockchain.cpp | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/komodo_ccdata.h b/src/komodo_ccdata.h
index 70e601946..058e38cee 100644
--- a/src/komodo_ccdata.h
+++ b/src/komodo_ccdata.h
@@ -45,6 +45,8 @@ bits256 iguana_merkle(bits256 *tree,int32_t txn_count)
 uint256 komodo_calcMoM(int32_t height,int32_t MoMdepth)
 {
     static uint256 zero; bits256 MoM,*tree; CBlockIndex *pindex; int32_t i;
+    if ( MoMdepth >= height )
+        return(zero);
     tree = (bits256 *)calloc(MoMdepth * 3,sizeof(*tree));
     for (i=0; i<MoMdepth; i++)
     {
diff --git a/src/rpcblockchain.cpp b/src/rpcblockchain.cpp
index 0a3c73c1d..d9cc072e5 100644
--- a/src/rpcblockchain.cpp
+++ b/src/rpcblockchain.cpp
@@ -876,7 +876,7 @@ UniValue calc_MoM(const UniValue& params, bool fHelp)
     LOCK(cs_main);
     height = atoi(params[0].get_str().c_str());
     MoMdepth = atoi(params[1].get_str().c_str());
-    if ( height <= 0 || MoMdepth <= 0 )
+    if ( height <= 0 || MoMdepth <= 0 || MoMdepth >= height )
         throw runtime_error("calc_MoM illegal height or MoMdepth\n");
     //fprintf(stderr,"height_MoM height.%d\n",height);
     MoM = komodo_calcMoM(height,MoMdepth);