Fix potential memleak.

[vlc] / src / network / tls.c

diff --git a/src/network/tls.c b/src/network/tls.c
index 756e2264c3d67f0b74c3e03486d946c865e66864..11e3dd2a24861ce817b2de1bd85aed9ba9252aa8 100644 (file)
--- a/src/network/tls.c
+++ b/src/network/tls.c
@@ -26,7 +26,11 @@
  * libvlc interface to the Transport Layer Security (TLS) plugins.
  */
 
-#include <vlc/vlc.h>
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
+
+#include <vlc_common.h>
 #include "libvlc.h"
 
 #include <vlc_tls.h>
@@ -69,7 +73,7 @@ tls_ServerCreate (vlc_object_t *obj, const char *cert_path,
     if (srv->p_module == NULL)
     {
         msg_Err (srv, "TLS server plugin not available");
-        vlc_object_destroy (srv);
+        vlc_object_release (srv);
         return NULL;
     }
 
@@ -90,7 +94,63 @@ void tls_ServerDelete (tls_server_t *srv)
 
     module_Unneed (srv, srv->p_module);
     vlc_object_detach (srv);
-    vlc_object_destroy (srv);
+    vlc_object_release (srv);
+}
+
+
+/**
+ * Adds one or more certificate authorities from a file.
+ * @return -1 on error, 0 on success.
+ */
+int tls_ServerAddCA (tls_server_t *srv, const char *path)
+{
+    return srv->pf_add_CA (srv, path);
+}
+
+
+/**
+ * Adds one or more certificate revocation list from a file.
+ * @return -1 on error, 0 on success.
+ */
+int tls_ServerAddCRL (tls_server_t *srv, const char *path)
+{
+    return srv->pf_add_CRL (srv, path);
+}
+
+
+tls_session_t *tls_ServerSessionPrepare (tls_server_t *srv)
+{
+    tls_session_t *ses;
+
+    ses = srv->pf_open (srv);
+    if (ses == NULL)
+        return NULL;
+
+    vlc_object_attach (ses, srv);
+    return ses;
+}
+
+
+void tls_ServerSessionClose (tls_session_t *ses)
+{
+    tls_server_t *srv = (tls_server_t *)(ses->p_parent);
+    srv->pf_close (srv, ses);
+}
+
+
+int tls_ServerSessionHandshake (tls_session_t *ses, int fd)
+{
+    ses->pf_set_fd (ses, fd);
+    return 2;
+}
+
+
+int tls_SessionContinueHandshake (tls_session_t *ses)
+{
+    int val = ses->pf_handshake (ses);
+    if (val < 0)
+        tls_ServerSessionClose (ses);
+    return val;
 }
 
 
@@ -108,6 +168,7 @@ tls_session_t *
 tls_ClientCreate (vlc_object_t *obj, int fd, const char *psz_hostname)
 {
     tls_session_t *cl;
+    int val;
 
     cl = (tls_session_t *)vlc_custom_create (obj, sizeof (*cl),
                                              VLC_OBJECT_GENERIC,
@@ -128,13 +189,15 @@ tls_ClientCreate (vlc_object_t *obj, int fd, const char *psz_hostname)
     if (cl->p_module == NULL)
     {
         msg_Err (cl, "TLS client plugin not available");
-        vlc_object_destroy (cl);
+        vlc_object_release (cl);
         return NULL;
     }
 
-    int val = tls_ClientSessionHandshake (cl, fd);
-    while (val > 0)
-        val = tls_SessionContinueHandshake (cl);
+    cl->pf_set_fd (cl, fd);
+
+    do
+        val = cl->pf_handshake (cl);
+    while (val > 0);
 
     if (val == 0)
     {
@@ -145,7 +208,7 @@ tls_ClientCreate (vlc_object_t *obj, int fd, const char *psz_hostname)
     msg_Err (cl, "TLS client session handshake error");
 
     module_Unneed (cl, cl->p_module);
-    vlc_object_destroy (cl);
+    vlc_object_release (cl);
     return NULL;
 }
 
@@ -161,5 +224,5 @@ void tls_ClientDelete (tls_session_t *cl)
 
     module_Unneed (cl, cl->p_module);
     vlc_object_detach (cl);
-    vlc_object_destroy (cl);
+    vlc_object_release (cl);
 }