modules/policy: set QTRACE flag for current query too

lib/layer/iterate.c

@@ -754,12 +754,14 @@ static int resolve(kr_layer_t *ctx, knot_pkt_t *pkt)
 	if (!query) {
 		return ctx->state;
 	}
-#ifndef NDEBUG
+
+#ifndef NDEBUG /* These might be useful during debugging. */
 	if (query->flags & QUERY_TRACE) {
 		VERBOSE_MSG("<= answer received:\n");
 		kr_pkt_print(pkt);
 	}
 #endif
+
 	if (query->flags & (QUERY_RESOLVED|QUERY_BADCOOKIE_AGAIN)) {
 		return ctx->state;
 	}

modules/policy/policy.lua

@@ -229,7 +229,9 @@ function policy.enforce(state, req, action)
 			return kres.DONE
 		end
 	elseif action == policy.QTRACE then
+		local qry = req:current()
 		req.options = bit.bor(req.options, kres.query.TRACE)
+		qry.flags = bit.bor(qry.flags, kres.query.TRACE)
 	elseif type(action) == 'function' then
 		return action(state, req)
 	end