From 6aa8ac70a443f5dece6fe4678319a6a2d4570e6c Mon Sep 17 00:00:00 2001
From: "zhaozhao.zz" <zhaozhao.zz@alibaba-inc.com>
Date: Thu, 29 Mar 2018 23:20:58 +0800
Subject: [PATCH] debug: avoid free client unexpectedly when reload & loadaof

---
 src/debug.c | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/debug.c b/src/debug.c
index a66390db..5a8190c9 100644
--- a/src/debug.c
+++ b/src/debug.c
@@ -345,7 +345,10 @@ NULL
             return;
         }
         emptyDb(-1,EMPTYDB_NO_FLAGS,NULL);
-        if (rdbLoad(server.rdb_filename,NULL) != C_OK) {
+        aeDeleteFileEvent(server.el,c->fd,AE_READABLE);
+        int ret = rdbLoad(server.rdb_filename,NULL);
+        aeCreateFileEvent(server.el,c->fd,AE_READABLE,readQueryFromClient,c);
+        if (ret != C_OK) {
             addReplyError(c,"Error trying to load the RDB dump");
             return;
         }
@@ -354,7 +357,10 @@ NULL
     } else if (!strcasecmp(c->argv[1]->ptr,"loadaof")) {
         if (server.aof_state != AOF_OFF) flushAppendOnlyFile(1);
         emptyDb(-1,EMPTYDB_NO_FLAGS,NULL);
-        if (loadAppendOnlyFile(server.aof_filename) != C_OK) {
+        aeDeleteFileEvent(server.el,c->fd,AE_READABLE);
+        int ret = loadAppendOnlyFile(server.aof_filename);
+        aeCreateFileEvent(server.el,c->fd,AE_READABLE,readQueryFromClient,c);
+        if (ret != C_OK) {
             addReply(c,shared.err);
             return;
         }