check callback return value for reserve the message

wushangzetom · Apr 18, 2014 · cfbd68b · cfbd68b
1 parent d568e54
commit cfbd68b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/skynet-src/skynet_server.c b/skynet-src/skynet_server.c
@@ -194,8 +194,9 @@ _dispatch_message(struct skynet_context *ctx, struct skynet_message *msg) {
 	CHECKCALLING_BEGIN(ctx)
 	int type = msg->sz >> HANDLE_REMOTE_SHIFT;
 	size_t sz = msg->sz & HANDLE_MASK;
-	ctx->cb(ctx, ctx->cb_ud, type, msg->session, msg->source, msg->data, sz);
-	free(msg->data);
+	if (!ctx->cb(ctx, ctx->cb_ud, type, msg->session, msg->source, msg->data, sz)) {
+		free(msg->data);
+	}
 	CHECKCALLING_END(ctx)
 }