From 67c522e9da4c5d9eb0dc3c9bca789222fc5a4fbf Mon Sep 17 00:00:00 2001
From: Andri Yngvason <andri@yngvason.is>
Date: Sun, 30 Apr 2023 16:41:01 +0000
Subject: [PATCH] server: Implement TLS on WebSocket

---
 src/server.c | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/server.c b/src/server.c
index 69c11e9..001ca87 100644
--- a/src/server.c
+++ b/src/server.c
@@ -1255,6 +1255,14 @@ static void on_connection(void* obj)
 	if (server->socket_type == NVNC__SOCKET_WEBSOCKET)
 	{
 		client->net_stream = stream_ws_new(fd, on_client_event, client);
+
+		if (server->tls_creds && stream_upgrade_to_tls(
+					client->net_stream,
+					client->server->tls_creds) < 0) {
+			nvnc_log(NVNC_LOG_DEBUG, "Failed to upgrade new stream to TLS");
+			stream_destroy(client->net_stream);
+			goto tls_failure;
+		}
 	}
 	else
 #endif
@@ -1295,6 +1303,7 @@ buffer_failure:
 	stream_destroy(client->net_stream);
 stream_failure:
 	close(fd);
+tls_failure:
 accept_failure:
 	free(client);
 }