chore: add reading in a do while

atsign-foundation · Aug 3, 2023 · 8366a91 · 8366a91
1 parent b9e7fb4
commit 8366a91
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/src/at_client/connection.c b/src/at_client/connection.c
@@ -169,7 +169,21 @@ int atclient_connection_send(atclient_connection_ctx *ctx, unsigned char *recv,
     }
     // printf("mbedtls_ssl_write: %d\n", ret);
 
-    ret = mbedtls_ssl_read(ctx->ssl, recv, recvlen);
+    memset(recv, 0, recvlen);
+    do
+    {
+        ret = mbedtls_ssl_read(ctx->ssl, recv, recvlen);
+        if(ret < 0)
+        {
+            goto exit;
+        }
+        *olen = ret;
+        // printf("mbedtls_ssl_read: %d\n", ret);
+
+        // size_t bytesavail = mbedtls_ssl_get_bytes_avail(ctx->ssl);
+        // printf("bytes_avail: %lu\n", bytesavail);
+    } while(ret == MBEDTLS_ERR_SSL_WANT_READ);
+
     if(ret < 0)
     {
         goto exit;
@@ -194,7 +208,6 @@ int atclient_connection_send(atclient_connection_ctx *ctx, unsigned char *recv,
     }
 }
 
-
 void atclient_connection_free(atclient_connection_ctx *ctx)
 {
     free(ctx->server_fd);