wt/mongoose
1
0 Fork 0
mirror of https://github.com/cesanta/mongoose.git synced 2026-10-24 11:54:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Read out all pending TLS data

Browse Source
This commit is contained in:
cpq authored and Sergio R. Caprile committed 2026-01-30 10:54:06 -03:00
1 parent 715f98c546
commit b93822fc50
2 files changed
+35 -33
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

mongoose.c
+28 -22
View File
@@ -15037,31 +15037,37 @@ long mg_tls_send(struct mg_connection *c, const void *buf, size_t len) {
}
long mg_tls_recv(struct mg_connection *c, void *buf, size_t len) {
int r = 0;
struct tls_data *tls = (struct tls_data *) c->tls;
unsigned char *recv_buf;
size_t minlen;
long res = 0;
r = mg_tls_recv_record(c);
if (r < 0) {
return r;
while (buf != NULL && len > 0) {
if (tls->recv_len == 0) {
int r = mg_tls_recv_record(c);
if (r < 0) {
if (res == 0) res = r;
break;
}
if (tls->content_type != MG_TLS_APP_DATA) {
tls->recv_len = 0;
mg_tls_drop_record(c);
continue;
}
}
{
uint8_t *recv_buf = &c->rtls.buf[tls->recv_offset];
size_t n = len < tls->recv_len ? len : tls->recv_len;
memmove(buf, recv_buf, n);
tls->recv_offset += n;
tls->recv_len -= n;
res += (long) n;
buf = (char *) buf + n;
len -= n;
if (tls->recv_len == 0) {
mg_tls_drop_record(c);
}
}
}
recv_buf = &c->rtls.buf[tls->recv_offset];
if (tls->content_type != MG_TLS_APP_DATA) {
tls->recv_len = 0;
mg_tls_drop_record(c);
return MG_IO_WAIT;
}
if (buf == NULL || len == 0) return 0L;
minlen = len < tls->recv_len ? len : tls->recv_len;
memmove(buf, recv_buf, minlen);
tls->recv_offset += minlen;
tls->recv_len -= minlen;
if (tls->recv_len == 0) {
mg_tls_drop_record(c);
}
return (long) minlen;
return res;
}
size_t mg_tls_pending(struct mg_connection *c) {
src/tls_builtin.c
+7 -11
View File
@@ -2509,25 +2509,21 @@ long mg_tls_recv(struct mg_connection *c, void *buf, size_t len) {
unsigned char *recv_buf;
size_t minlen;
r = mg_tls_recv_record(c);
if (r < 0) {
return r;
}
recv_buf = &c->rtls.buf[tls->recv_offset];
if (tls->content_type != MG_TLS_APP_DATA) {
for (;;) {
r = mg_tls_recv_record(c);
if (r < 0) return r;
if (tls->content_type == MG_TLS_APP_DATA) break;
tls->recv_len = 0;
mg_tls_drop_record(c);
return MG_IO_WAIT;
}
if (buf == NULL || len == 0) return 0L;
recv_buf = &c->rtls.buf[tls->recv_offset];
minlen = len < tls->recv_len ? len : tls->recv_len;
memmove(buf, recv_buf, minlen);
tls->recv_offset += minlen;
tls->recv_len -= minlen;
if (tls->recv_len == 0) {
mg_tls_drop_record(c);
}
if (tls->recv_len == 0) mg_tls_drop_record(c);
return (long) minlen;
}