From c4ef9cf0e8f66f4322ec4f97441931251ac3bef8 Mon Sep 17 00:00:00 2001
From: Otto Moerbeek <otto.moerbeek@open-xchange.com>
Date: Mon, 14 Feb 2022 14:55:10 +0100
Subject: [PATCH] Also store the Lua context for the TCP case

---
 pdns/recursordist/rec-main.hh | 4 ++--
 pdns/recursordist/rec-tcp.cc  | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pdns/recursordist/rec-main.hh b/pdns/recursordist/rec-main.hh
index 81c0460760..da1b8d82fd 100644
--- a/pdns/recursordist/rec-main.hh
+++ b/pdns/recursordist/rec-main.hh
@@ -48,8 +48,8 @@
 //! used to send information to a newborn mthread
 struct DNSComboWriter
 {
-  DNSComboWriter(const std::string& query, const struct timeval& now) :
-    d_mdp(true, query), d_now(now), d_query(query)
+  DNSComboWriter(const std::string& query, const struct timeval& now, shared_ptr<RecursorLua4> luaContext) :
+    d_mdp(true, query), d_now(now), d_query(query), d_luaContext(luaContext)
   {
   }
 
diff --git a/pdns/recursordist/rec-tcp.cc b/pdns/recursordist/rec-tcp.cc
index 4e41c08be3..807b3dce1c 100644
--- a/pdns/recursordist/rec-tcp.cc
+++ b/pdns/recursordist/rec-tcp.cc
@@ -329,7 +329,7 @@ static void handleRunningTCPQuestion(int fd, FDMultiplexer::funcparam_t& var)
       conn->state = TCPConnection::BYTE0;
       std::unique_ptr<DNSComboWriter> dc;
       try {
-        dc = std::make_unique<DNSComboWriter>(conn->data, g_now);
+        dc = std::make_unique<DNSComboWriter>(conn->data, g_now, t_pdl);
       }
       catch (const MOADNSException& mde) {
         g_stats.clientParseError++;
-- 
2.47.2