Fix NULL ctx hold/free in execution message handlers

Author: gabsow

src/mr.c

@@ -584,7 +584,7 @@ static void MR_StepDone(Execution* e, void* pd) {
     size_t stepIndex = mr_BufferReaderReadLongLong(&reader, NULL);
 
     RedisModule_ThreadSafeContextLock(mr_staticCtx);
-    RedisModule_FreeString(NULL, payload);
+    RedisModule_FreeString(mr_staticCtx, payload);
     RedisModule_ThreadSafeContextUnlock(mr_staticCtx);
 
     if (MR_PerformStepDoneOp(e, stepIndex) == MR_ClusterGetSize() - 1){
@@ -616,7 +616,7 @@ static void MR_SetRecord(Execution* e, void* pd) {
     Record* r = MR_RecordDeSerialize(&reader);
 
     RedisModule_ThreadSafeContextLock(mr_staticCtx);
-    RedisModule_FreeString(NULL, payload);
+    RedisModule_FreeString(mr_staticCtx, payload);
     RedisModule_ThreadSafeContextUnlock(mr_staticCtx);
 
     if (MR_SetRecordToStep(e, stepIndex, r) > 10000){
@@ -647,7 +647,7 @@ static void MR_PassRecord(RedisModuleCtx *ctx, const char *sender_id, uint8_t ty
     }
 
     /* run the execution on the thread pool */
-    MR_ExecutionAddTask(e, MR_SetRecord, RedisModule_HoldString(NULL, payload));
+    MR_ExecutionAddTask(e, MR_SetRecord, RedisModule_HoldString(mr_staticCtx, payload));
 }
 
 static void MR_SendRecordToSlot(Execution* e, Step* s, Record* r, size_t slot) {
@@ -701,7 +701,7 @@ static void MR_NotifyStepDone(RedisModuleCtx *ctx, const char *sender_id, uint8_
     }
 
     /* run the execution on the thread pool */
-    MR_ExecutionAddTask(e, MR_StepDone, RedisModule_HoldString(NULL, payload));
+    MR_ExecutionAddTask(e, MR_StepDone, RedisModule_HoldString(mr_staticCtx, payload));
 }
 
 static Record* MR_RunReshuffleStep(Execution* e, Step* s) {
@@ -1140,7 +1140,7 @@ static void MR_RecieveExecution(void* pd) {
      * Possible optimization would be to batch multiple
      * payloads into one GIL locking */
     RedisModule_ThreadSafeContextLock(mr_staticCtx);
-    RedisModule_FreeString(NULL, payload);
+    RedisModule_FreeString(mr_staticCtx, payload);
     RedisModule_ThreadSafeContextUnlock(mr_staticCtx);
 
     /* Finish deserializing the execution, we need to
@@ -1154,7 +1154,7 @@ static void MR_RecieveExecution(void* pd) {
 static void MR_NewExecutionRecieved(RedisModuleCtx *ctx, const char *sender_id, uint8_t type, RedisModuleString* payload) {
     /* We can directly move the job to the thread pool.
      * We need to deserialize the execution and reply to the initiator. */
-    mr_thpool_add_work(mrCtx.executionsThreadPool, MR_RecieveExecution, RedisModule_HoldString(NULL, payload));
+    mr_thpool_add_work(mrCtx.executionsThreadPool, MR_RecieveExecution, RedisModule_HoldString(mr_staticCtx, payload));
 }
 
 static void MR_ExecutionStepSerialize(mr_BufferWriter* buffWriter, Step* s) {