From 803154306243573493694f4d8ce89397373ce808 Mon Sep 17 00:00:00 2001
From: "Simon V. Lejel" <sl@dcsl.dk>
Date: Sun, 11 Feb 2024 15:24:59 +0100
Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=A7=20Add=20SSL=20stuff=20to=20client?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/client/main.cc | 42 +++++++++++++++++++++++++++++++++++++++---
 1 file changed, 39 insertions(+), 3 deletions(-)

diff --git a/src/client/main.cc b/src/client/main.cc
index db4317a..26b09d7 100644
--- a/src/client/main.cc
+++ b/src/client/main.cc
@@ -1,13 +1,18 @@
 #include <arpa/inet.h>
-#include <cstring>
 #include <iostream>
 #include <netinet/in.h>
+#include <openssl/err.h>
+#include <openssl/ssl.h>
 #include <ostream>
 #include <sys/socket.h>
 #include <unistd.h>
 
 using std::cout, std::cin, std::endl, std::string;
 
+/*
+ * Remember to build with the flags "-L/usr/lib -lssl -lcrypto"
+ */
+
 int main() {
 
     struct Message {
@@ -19,7 +24,30 @@ int main() {
         }
     };
 
+    // Initialize OpenSSL
+    SSL_library_init();
+    SSL_load_error_strings();
+
+    // Create new SSL context
+    SSL_CTX *ctx = SSL_CTX_new(TLS_client_method());
+    if (!ctx) {
+        // TODO Handle error
+    }
+
+    // Load the server's certificate into context
+    if (SSL_CTX_load_verify_locations(
+            ctx, "~/git/mogens_og_karen/ssl/server.crt", nullptr) <= 0) {
+        // TODO Handle error
+    }
+
+    // Create SSL object
+    SSL *ssl = SSL_new(ctx);
+    if (!ssl) {
+        // TODO Handle error
+    }
+
     int clientSocket = socket(AF_INET, SOCK_STREAM, 0);
+    SSL_set_fd(ssl, clientSocket);
 
     sockaddr_in serverAdress;
     serverAdress.sin_family = AF_INET;
@@ -35,6 +63,11 @@ int main() {
         cout << "CONNECTED!" << endl;
     }
 
+    // Perform SSL handshake
+    if (SSL_connect(ssl) != 1) {
+        // TODO Handle error
+    }
+
     while (true) {
         Message message;
         cout << "Type message to server: ";
@@ -48,10 +81,13 @@ int main() {
         cout << endl;
         message.username = "Client 1";
 
-        send(clientSocket, message.toString().data(),
-             strlen(message.toString().data()), 0);
+        SSL_write(ssl, message.toString().data(), 0);
     }
 
+    // Clean up
+    SSL_free(ssl);
+    SSL_CTX_free(ctx);
+    ERR_free_strings();
     close(clientSocket);
 
     return 0;