fixup attach_edns check for streamtcp (in case of future enhancements and smaller...

git-svn-id: file:///svn/unbound/trunk@4196 be551aaa-1e26-0410-a405-d3ace91eadb9

diff --git a/testcode/streamtcp.c b/testcode/streamtcp.c
index 34b5c0281369228cc4015ea3f4db526ab591f664..f5eb8fc48b1da803e69fe26571feed93fb339ff3 100644 (file)
--- a/testcode/streamtcp.c
+++ b/testcode/streamtcp.c
@@ -143,7 +143,9 @@ write_q(int fd, int udp, SSL* ssl, sldns_buffer* buf, uint16_t id,
                edns.edns_present = 1;
                edns.bits = EDNS_DO;
                edns.udp_size = 4096;
-               attach_edns_record(buf, &edns);
+               if(sldns_buffer_capacity(buf) >=
+                       sldns_buffer_limit(buf)+calc_edns_field_size(&edns))
+                       attach_edns_record(buf, &edns);
        }
 
        /* send it */