1 files changed, 15 insertions, 0 deletions
diff --git a/src/mbedtls.h b/src/mbedtls.h
index e7a9017..00386ee 100644
--- a/src/mbedtls.h
+++ b/src/mbedtls.h
@@ -2,9 +2,14 @@
 #include <stdint.h>
 #include <stdbool.h>
 #include <stddef.h>
+#include <time.h>
 
+#include <poll.h>
+
+#include <mbedtls/ssl.h>
 #include <mbedtls/x509_crt.h>
 #include <mbedtls/entropy.h>
+#include <pthsem.h>
 
 
 // Callback that masks `MBEDTLS_X509_BADCERT_EXPIRED`
@@ -15,3 +20,13 @@ int prne_mbedtls_ssl_recv_cb (void *ctx, unsigned char *buf, size_t len);
 * Workaround for a bug - getrandom() blocks
 */
 void prne_mbedtls_entropy_init (mbedtls_entropy_context *ctx);
+
+/* Convenience Functions
+*/
+
+// Handles mbedtls_ssl_handshake(), mbedtls_ssl_close_notify()
+bool prne_mbedtls_pth_handle (
+	mbedtls_ssl_context *ssl,
+	int(*mbedtls_f)(mbedtls_ssl_context*),
+	const int fd,
+	pth_event_t ev);