1 files changed, 43 insertions, 0 deletions
diff --git a/src/knot/modules/noudp/noudp.c b/src/knot/modules/noudp/noudp.c
new file mode 100644
index 0000000..c2f064a
--- /dev/null
+++ b/src/knot/modules/noudp/noudp.c
@@ -0,0 +1,43 @@
+/*  Copyright (C) 2017 CZ.NIC, z.s.p.o. <knot-dns@labs.nic.cz>
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "knot/include/module.h"
+
+static bool is_udp(knotd_qdata_t *qdata)
+{
+	return qdata->params->flags & KNOTD_QUERY_FLAG_LIMIT_SIZE;
+}
+
+static knotd_state_t noudp_begin(knotd_state_t state, knot_pkt_t *pkt,
+                                 knotd_qdata_t *qdata, knotd_mod_t *mod)
+{
+	if (is_udp(qdata)) {
+		knot_wire_set_tc(pkt->wire);
+		return KNOTD_STATE_DONE;
+	}
+
+	return state;
+}
+
+int noudp_load(knotd_mod_t *mod)
+{
+	knotd_mod_hook(mod, KNOTD_STAGE_BEGIN, noudp_begin);
+
+	return KNOT_EOK;
+}
+
+KNOTD_MOD_API(noudp, KNOTD_MOD_FLAG_SCOPE_ANY | KNOTD_MOD_FLAG_OPT_CONF,
+              noudp_load, NULL, NULL, NULL);