motiejus/zig
1
Fork 0
Code Releases Activity

tls.Client: don't read if we don't need more data

Browse Source
This commit is contained in:
Nameless
2023-04-04 14:21:49 -05:00
committed by Andrew Kelley
parent 55a8b7e1fa
commit fde05b10b3
1 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
lib/std/crypto/tls/Client.zig
View File

@@ -923,20 +923,22 @@ pub fn readvAdvanced(c: *Client, stream: anytype, iovecs: []const std.os.iovec)
if (partial_cleartext.len > 0) {
const amt = @intCast(u15, vp.put(partial_cleartext));
c.partial_cleartext_idx += amt;
if (amt < partial_cleartext.len) {
// We still have cleartext left so we cannot issue another read() call yet.
assert(vp.total == amt);
return amt;
if (c.partial_ciphertext_end == c.partial_ciphertext_idx) {
// The buffer is now empty.
c.partial_cleartext_idx = 0;
c.partial_ciphertext_idx = 0;
c.partial_ciphertext_end = 0;
}
if (c.received_close_notify) {
c.partial_ciphertext_end = 0;
assert(vp.total == amt);
return amt;
}
if (c.partial_ciphertext_end == c.partial_ciphertext_idx) {
c.partial_cleartext_idx = 0;
c.partial_ciphertext_idx = 0;
c.partial_ciphertext_end = 0;
} else if (amt <= partial_cleartext.len) {
// We don't need more data, so don't call read.
assert(vp.total == amt);
return amt;
}
}