wechat/wxhelper
5
0
Fork 0
mirror of https://github.com/ttttupup/wxhelper.git synced 2025-04-29 16:29:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed mqtt variable length header issue

Browse Source
This commit is contained in:
npt-1707 2025-04-23 04:08:11 +08:00
parent 24bb7212fa
commit 335a4771fe
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/mongoose.c
View File

@ -3000,21 +3000,20 @@ static int encode_variable_length(uint8_t *buf, size_t value) {
return len; return len;
} }
static uint32_t decode_variable_length(const char *buf, static size_t decode_varint(const uint8_t *buf, size_t len, size_t *value) {
uint32_t *bytes_consumed) { uint32_t multiplier = 1;
uint32_t value = 0, multiplier = 1, offset; size_t offset;
*value = 0;
for (offset = 0; offset < 4; offset++) { for (offset = 0; offset < 4 && offset < len; offset++) {
uint8_t encoded_byte = ((uint8_t *) buf)[offset]; uint8_t encoded_byte = buf[offset];
value += (encoded_byte & 0x7F) * multiplier; *value += (encoded_byte & 0x7F) * multiplier;
multiplier *= 128; multiplier *= 128;
if (!(encoded_byte & 0x80)) break; if (!(encoded_byte & 0x80)) return offset + 1;
} }
if (bytes_consumed != NULL) *bytes_consumed = offset + 1; return 0;
return value;
} }
static int mqtt_prop_type_by_id(uint8_t prop_id) { static int mqtt_prop_type_by_id(uint8_t prop_id) {
@ -3107,8 +3106,8 @@ static void mg_send_mqtt_properties(struct mg_connection *c,
size_t mg_mqtt_next_prop(struct mg_mqtt_message *msg, struct mg_mqtt_prop *prop, size_t mg_mqtt_next_prop(struct mg_mqtt_message *msg, struct mg_mqtt_prop *prop,
size_t ofs) { size_t ofs) {
uint8_t *i = (uint8_t *) msg->dgram.ptr + msg->props_start + ofs; uint8_t *i = (uint8_t *) msg->dgram.ptr + msg->props_start + ofs;
size_t new_pos = ofs; uint8_t *end = (uint8_t *) msg->dgram.ptr + msg->dgram.len;
uint32_t bytes_consumed; size_t new_pos = ofs, len;
prop->id = i[0]; prop->id = i[0];
if (ofs >= msg->dgram.len || ofs >= msg->props_start + msg->props_size) if (ofs >= msg->dgram.len || ofs >= msg->props_start + msg->props_size)
@ -3148,8 +3147,8 @@ size_t mg_mqtt_next_prop(struct mg_mqtt_message *msg, struct mg_mqtt_prop *prop,
new_pos += 2 + prop->val.len; new_pos += 2 + prop->val.len;
break; break;
case MQTT_PROP_TYPE_VARIABLE_INT: case MQTT_PROP_TYPE_VARIABLE_INT:
prop->iv = decode_variable_length((char *) i, &bytes_consumed); len = decode_varint(i, (size_t) (end - i), (size_t *) &prop->iv);
new_pos += bytes_consumed; new_pos = (!len) ? 0 : new_pos + len;
break; break;
default: new_pos = 0; default: new_pos = 0;
} }
@ -3314,7 +3313,8 @@ int mg_mqtt_parse(const uint8_t *buf, size_t len, uint8_t version,
} }
if (p > end) return MQTT_MALFORMED; if (p > end) return MQTT_MALFORMED;
if (version == 5 && p + 2 < end) { if (version == 5 && p + 2 < end) {
m->props_size = decode_variable_length((char *) p, &len_len); len_len = (uint32_t) decode_varint(p, (size_t) (end - p), &m->props_size);
if (!len_len) return MQTT_MALFORMED;
m->props_start = (size_t) (p + len_len - buf); m->props_start = (size_t) (p + len_len - buf);
p += len_len + m->props_size; p += len_len + m->props_size;
} }