fix(asan): heap-use-after-free in pegasus_write_service.cpp (#478)

apache · Mar 31, 2020 · b41c693 · b41c693
1 parent 01312e7
commit b41c693
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/server/pegasus_write_service.cpp b/src/server/pegasus_write_service.cpp
@@ -297,16 +297,18 @@ int pegasus_write_service::duplicate(int64_t decree,
         resp.__set_error(_impl->multi_remove(ctx.decree, rpc.request(), rpc.response()));
         return resp.error;
     }
+    put_rpc put;
+    remove_rpc remove;
     if (request.task_code == dsn::apps::RPC_RRDB_RRDB_PUT ||
         request.task_code == dsn::apps::RPC_RRDB_RRDB_REMOVE) {
         int err = 0;
         if (request.task_code == dsn::apps::RPC_RRDB_RRDB_PUT) {
-            put_rpc rpc(write);
-            err = _impl->batch_put(ctx, rpc.request(), rpc.response());
+            put = put_rpc(write);
+            err = _impl->batch_put(ctx, put.request(), put.response());
         }
         if (request.task_code == dsn::apps::RPC_RRDB_RRDB_REMOVE) {
-            remove_rpc rpc(write);
-            err = _impl->batch_remove(ctx.decree, rpc.request(), rpc.response());
+            remove = remove_rpc(write);
+            err = _impl->batch_remove(ctx.decree, remove.request(), remove.response());
         }
         if (!err) {
             err = _impl->batch_commit(ctx.decree);