From 65f7c078663da387fd2cee8ab401c7ef66081138 Mon Sep 17 00:00:00 2001
From: Determinant <ted.sybil@gmail.com>
Date: Thu, 20 Jun 2019 19:17:11 -0400
Subject: add TLS benchmark

---
 include/salticidae/crypto.h | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

(limited to 'include/salticidae/crypto.h')

diff --git a/include/salticidae/crypto.h b/include/salticidae/crypto.h
index 1d20b22..bcfd9dc 100644
--- a/include/salticidae/crypto.h
+++ b/include/salticidae/crypto.h
@@ -128,6 +128,10 @@ static inline int _tls_pem_with_passwd(char *buf, int size, int, void *) {
     return _size - 1;
 }
 
+static int _skip_CA_check(int, X509_STORE_CTX *) {
+    return 1;
+}
+
 class PKey {
     EVP_PKEY *key;
     friend class TLSContext;
@@ -271,6 +275,11 @@ class TLSContext {
             throw SalticidaeError(SALTI_ERROR_TLS_LOAD_KEY);
     }
 
+    void set_verify(bool skip_ca_check = true, SSL_verify_cb verify_callback = nullptr) {
+        SSL_CTX_set_verify(ctx,
+                SSL_VERIFY_PEER, skip_ca_check ? _skip_CA_check : verify_callback);
+    }
+
     bool check_privkey() {
         return SSL_CTX_check_private_key(ctx) > 0;
     }
@@ -329,13 +338,9 @@ class TLS {
         return SSL_get_error(ssl, ret);
     }
 
-    ~TLS() {
-        if (ssl)
-        {
-            SSL_shutdown(ssl);
-            SSL_free(ssl);
-        }
-    }
+    void shutdown() { SSL_shutdown(ssl); }
+
+    ~TLS() { if (ssl) SSL_free(ssl); }
 };
 
 }
-- 
cgit v1.2.3