summaryrefslogtreecommitdiff
path: root/lib/mqtt.c
diff options
context:
space:
mode:
authorDaniel Stenberg <daniel@haxx.se>2022-03-29 08:43:13 +0200
committerDaniel Stenberg <daniel@haxx.se>2022-03-29 08:43:14 +0200
commit87da3e845be515ddbb1542d0d17e66e843610d3a (patch)
tree262ae31fb09e73e2bc93b9bd0b05fa09b2ab5757 /lib/mqtt.c
parent301bd979652a53482b8910845a9643298c6029f7 (diff)
downloadcurl-87da3e845be515ddbb1542d0d17e66e843610d3a.tar.gz
mqtt: better handling of TCP disconnect mid-message
Reported-by: Jenny Heino Bug: https://hackerone.com/reports/1521610 Closes #8644
Diffstat (limited to 'lib/mqtt.c')
-rw-r--r--lib/mqtt.c8
1 files changed, 7 insertions, 1 deletions
diff --git a/lib/mqtt.c b/lib/mqtt.c
index e79bd3b48..9bcbaa195 100644
--- a/lib/mqtt.c
+++ b/lib/mqtt.c
@@ -732,8 +732,14 @@ static CURLcode mqtt_doing(struct Curl_easy *data, bool *done)
case MQTT_FIRST:
/* Read the initial byte only */
result = Curl_read(data, sockfd, (char *)&mq->firstbyte, 1, &nread);
- if(!nread)
+ if(result)
+ break;
+ else if(!nread) {
+ failf(data, "Connection disconnected");
+ *done = TRUE;
+ result = CURLE_RECV_ERROR;
break;
+ }
Curl_debug(data, CURLINFO_HEADER_IN, (char *)&mq->firstbyte, 1);
/* remember the first byte */
mq->npacket = 0;