From: Tomas Krizek <tomas.krizek@nic.cz>
Date: Thu, 15 Nov 2018 12:48:53 +0000 (+0100)
Subject: pytests: import test_query_flood_garbage (test9)
X-Git-Tag: v3.2.0~18^2~34
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=d0cfac15432a243c3ffdd3df4c5d6b68375a7de5;p=thirdparty%2Fknot-resolver.git

pytests: import test_query_flood_garbage (test9)
---

diff --git a/tests/pytests/test_conn_mgmt.py b/tests/pytests/test_conn_mgmt.py
index 0253187df..28730f3b0 100644
--- a/tests/pytests/test_conn_mgmt.py
+++ b/tests/pytests/test_conn_mgmt.py
@@ -169,3 +169,24 @@ def test_query_flood_no_recv(make_kresd_sock):
 
     sock2 = make_kresd_sock()
     utils.ping_alive(sock2)  # resolver must stay alive
+
+
+def test_query_flood_garbage(make_kresd_sock):
+    """Flood resolver with correctly prefixed garbage of maximum size."""
+    # TODO - despite the fact that kresd closes TCP connection, it seems to be
+    # error in TCP stream parsing. Kresd closes TCP connection because of
+    # message length in TCP prefix is lesser then length of the fixed message
+    # header, it shouldn't happen.
+
+    gbuff = utils.get_prefixed_garbage(65533)
+    buff = gbuff * 100
+    end_time = time.time() + utils.MAX_TIMEOUT
+    sock1 = make_kresd_sock()
+
+    with utils.expect_kresd_close(rst_ok=True):  # connection must be closed
+        while time.time() < end_time:
+            sock1.sendall(buff)
+            time.sleep(0.5)
+
+    sock2 = make_kresd_sock()
+    utils.ping_alive(sock2)  # resolver must stay alive