- fixed memory handling problems when using TLS

hal/tls/mbedtls/tls_mbedtls.c

@@ -474,10 +474,9 @@ TLSSocket_write(TLSSocket self, uint8_t* buf, int size)
 void
 TLSSocket_close(TLSSocket self)
 {
+    if (self) {
         int ret;
 
-    //TODO add timeout?
-
         while ((ret = mbedtls_ssl_close_notify(&(self->ssl))) < 0)
         {
             if ((ret != MBEDTLS_ERR_SSL_WANT_READ) && (ret != MBEDTLS_ERR_SSL_WANT_WRITE))
@@ -493,4 +492,5 @@ TLSSocket_close(TLSSocket self)
         mbedtls_ssl_free(&(self->ssl));
 
         GLOBAL_FREEMEM(self);
+    }
 }

src/mms/iso_server/iso_connection.c

@@ -92,15 +92,16 @@ struct sIsoConnection
 static void
 IsoConnection_releaseAllocatedMemory(IsoConnection self)
 {
-    if (self->socket)
-         Socket_destroy(self->socket);
-
 #if (CONFIG_MMS_SUPPORT_TLS == 1)
     if (IsoServer_getTLSConfiguration(self->isoServer) != NULL) {
         TLSSocket_close(self->tlsSocket);
+        self->tlsSocket = NULL;
     }
 #endif /* (CONFIG_MMS_SUPPORT_TLS == 1) */
 
+    if (self->socket)
+         Socket_destroy(self->socket);
+
      GLOBAL_FREEMEM(self->session);
      GLOBAL_FREEMEM(self->presentation);
      AcseConnection_destroy(self->acseConnection);
@@ -662,8 +663,10 @@ IsoConnection_close(IsoConnection self)
         self->socket = NULL;
 
 #if (CONFIG_MMS_SUPPORT_TLS == 1)
-        if (self->tlsSocket)
+        if (self->tlsSocket) {
             TLSSocket_close(self->tlsSocket);
+            self->tlsSocket = NULL;
+        }
 #endif
 
         Socket_destroy(socket);