Test SPTPS messages sent while key renegotation is in progress.

[tinc] / src / sptps_test.c

diff --git a/src/sptps_test.c b/src/sptps_test.c
index 3ee7ab6..79a1a85 100644 (file)
--- a/src/sptps_test.c
+++ b/src/sptps_test.c
@@ -154,9 +154,11 @@ int main(int argc, char *argv[]) {
                                break;
                        if(buf[0] == '^')
                                sptps_send_record(&s, SPTPS_HANDSHAKE, NULL, 0);
-                       else if(buf[0] == '$')
+                       else if(buf[0] == '$') {
                                sptps_force_kex(&s);
-                       else
+                               if(len > 1)
+                                       sptps_send_record(&s, 0, buf, len);
+                       } else
                        if(!sptps_send_record(&s, buf[0] == '!' ? 1 : 0, buf, buf[0] == '\n' ? 0 : buf[0] == '*' ? sizeof buf : len))
                                return 1;
                }