--- mqtt/src/Attic/conn.c 2011/12/05 14:11:47 1.1.2.6 +++ mqtt/src/Attic/conn.c 2011/12/05 16:55:40 1.1.2.8 @@ -275,12 +275,9 @@ mqtt_readCONNECT(mqtt_msg_t * __restrict buf, u_short if (!buf || !buf->msg_base || !buf->msg_len || !psConnID || !connLen) return flg; - hdr = (struct mqtthdr*) buf->msg_base; - if (hdr->mqtt_msg.type != MQTT_TYPE_CONNECT) { - mqtt_SetErr(EINVAL, "Error:: wrong command #%d", hdr->mqtt_msg.type); + hdr = _mqtt_readHEADER(buf, MQTT_TYPE_CONNECT, &ret, &len); + if (!hdr) return flg; - } else - len = mqtt_decodeLen(hdr->mqtt_len, &ret); if (len < 12) { mqtt_SetErr(EINVAL, "Error:: short message length %d", len); return flg; @@ -391,4 +388,40 @@ mqtt_readCONNECT(mqtt_msg_t * __restrict buf, u_short } return flg; +} + +/* + * mqtt_readCONNACK() Read CONNACK message + * + * @buf = Message buffer + * return: -1 error or >-1 CONNECT message return code + */ +u_char +mqtt_readCONNACK(mqtt_msg_t * __restrict buf) +{ + int len, ret; + struct mqtthdr *hdr; + mqtthdr_connack_t *ack; + caddr_t pos; + + if (!buf || !buf->msg_base || !buf->msg_len) + return (u_char) -1; + + hdr = _mqtt_readHEADER(buf, MQTT_TYPE_CONNACK, &ret, &len); + if (!hdr) + return (u_char) -1; + if (len < sizeof(mqtthdr_connack_t)) { + mqtt_SetErr(EINVAL, "Error:: short message length %d", len); + return (u_char) -1; + } else { + pos = buf->msg_base + ret + 1; + ack = (mqtthdr_connack_t*) pos; + } + + if (ack->retcode > MQTT_RETCODE_DENIED) { + mqtt_SetErr(EINVAL, "Error:: invalid retcode %u", ack->retcode); + return (u_char) -1; + } + + return ack->retcode; }