From 633bfe00b342fd9fdda217cf3c9301f2962f75ac Mon Sep 17 00:00:00 2001
From: dmiller <dmiller@e0a8ed71-7df4-0310-8962-fdc924857419>
Date: Mon, 3 Jun 2024 19:00:29 +0000
Subject: [PATCH] Allow NSE to connect via DTLS

---
 nse_nsock.cc | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/nse_nsock.cc b/nse_nsock.cc
index 18a75a7bb..82701b1ea 100644
--- a/nse_nsock.cc
+++ b/nse_nsock.cc
@@ -562,9 +562,12 @@ static int connect (lua_State *L, int status, lua_KContext ctx)
           dest->ai_addrlen, port);
       break;
     case SSL:
-      nu->proto = IPPROTO_TCP;
+      if (nu->proto != IPPROTO_UDP) {
+        // Assume TCP unless we're explicitly connecting to a UDP port
+        nu->proto = IPPROTO_TCP;
+      }
       nsock_connect_ssl(nsp, nu->nsiod, callback, nu->timeout, nu,
-          dest->ai_addr, dest->ai_addrlen, IPPROTO_TCP, port, nu->ssl_session);
+          dest->ai_addr, dest->ai_addrlen, nu->proto, port, nu->ssl_session);
       break;
   }