/* For PUBACK, PUBCOMP, PUBREC, and PUBREL */
int _mosquitto_send_command_with_mid(struct mosquitto *mosq, uint8_t command, uint16_t mid, bool dup)
{
struct _mosquitto_packet *packet = NULL;
int rc;
assert(mosq);
packet = _mosquitto_calloc(1, sizeof(struct _mosquitto_packet));
if(!packet) return MOSQ_ERR_NOMEM;
packet->command = command;
if(dup){
packet->command |= 8;
}
packet->remaining_length = 2;
rc = _mosquitto_packet_alloc(packet);
if(rc){
_mosquitto_free(packet);
return rc;
}
packet->payload[packet->pos+0] = MOSQ_MSB(mid);
packet->payload[packet->pos+1] = MOSQ_LSB(mid);
//add by lanhuaiyu for log
packet->mid = mid;
return _mosquitto_packet_queue(mosq, packet);
}
/* For PUBACK, PUBCOMP, PUBREC, and PUBREL */
int _mosquitto_send_command_with_mid(struct _mosquitto_core *core, uint8_t command, uint16_t mid, bool dup)
{
struct _mosquitto_packet *packet = NULL;
int rc;
assert(core);
packet = _mosquitto_calloc(1, sizeof(struct _mosquitto_packet));
if(!packet) return MOSQ_ERR_NOMEM;
packet->command = command;
if(dup) {
packet->command |= 8;
}
packet->remaining_length = 2;
rc = _mosquitto_packet_alloc(packet);
if(rc) {
_mosquitto_free(packet);
return rc;
}
packet->payload[packet->pos+0] = MOSQ_MSB(mid);
packet->payload[packet->pos+1] = MOSQ_LSB(mid);
_mosquitto_packet_queue(core, packet);
return MOSQ_ERR_SUCCESS;
}
/* For PUBACK, PUBCOMP, PUBREC, and PUBREL */
int send__command_with_mid(struct mosquitto *mosq, uint8_t command, uint16_t mid, bool dup)
{
struct mosquitto__packet *packet = NULL;
int rc;
assert(mosq);
packet = mosquitto__calloc(1, sizeof(struct mosquitto__packet));
if(!packet) return MOSQ_ERR_NOMEM;
packet->command = command;
if(dup){
packet->command |= 8;
}
packet->remaining_length = 2;
rc = packet__alloc(packet);
if(rc){
mosquitto__free(packet);
return rc;
}
packet->payload[packet->pos+0] = MOSQ_MSB(mid);
packet->payload[packet->pos+1] = MOSQ_LSB(mid);
return packet__queue(mosq, packet);
}
void _mosquitto_write_uint16(struct _mosquitto_packet *packet, uint16_t word)
{
_mosquitto_write_byte(packet, MOSQ_MSB(word));
_mosquitto_write_byte(packet, MOSQ_LSB(word));
}
int32_t WriteUint16(Buffer* buf, uint16_t val)
{
my_assert(buf->_read_pos == 0);
return WriteByte(buf, MOSQ_MSB(val))
|| WriteByte(buf, MOSQ_LSB(val));
}
int mosquitto__socks5_send(struct mosquitto *mosq)
{
struct _mosquitto_packet *packet;
int slen;
int ulen, plen;
if(mosq->state == mosq_cs_socks5_new){
packet = _mosquitto_calloc(1, sizeof(struct _mosquitto_packet));
if(!packet) return MOSQ_ERR_NOMEM;
if(mosq->socks5_username){
packet->packet_length = 4;
}else{
packet->packet_length = 3;
}
packet->payload = _mosquitto_malloc(sizeof(uint8_t)*packet->packet_length);
packet->payload[0] = 0x05;
if(mosq->socks5_username){
packet->payload[1] = 2;
packet->payload[2] = SOCKS_AUTH_NONE;
packet->payload[3] = SOCKS_AUTH_USERPASS;
}else{
packet->payload[1] = 1;
packet->payload[2] = SOCKS_AUTH_NONE;
}
pthread_mutex_lock(&mosq->state_mutex);
mosq->state = mosq_cs_socks5_start;
pthread_mutex_unlock(&mosq->state_mutex);
mosq->in_packet.pos = 0;
mosq->in_packet.packet_length = 2;
mosq->in_packet.to_process = 2;
mosq->in_packet.payload = _mosquitto_malloc(sizeof(uint8_t)*2);
if(!mosq->in_packet.payload){
_mosquitto_free(packet->payload);
_mosquitto_free(packet);
return MOSQ_ERR_NOMEM;
}
return _mosquitto_packet_queue(mosq, packet);
}else if(mosq->state == mosq_cs_socks5_auth_ok){
packet = _mosquitto_calloc(1, sizeof(struct _mosquitto_packet));
if(!packet) return MOSQ_ERR_NOMEM;
packet->packet_length = (int)(7 + strlen(mosq->host));
packet->payload = _mosquitto_malloc(sizeof(uint8_t)*packet->packet_length);
slen = (int)strlen(mosq->host);
packet->payload[0] = 0x05;
packet->payload[1] = 1;
packet->payload[2] = 0;
packet->payload[3] = SOCKS_ATYPE_DOMAINNAME;
packet->payload[4] = slen;
memcpy(&(packet->payload[5]), mosq->host, slen);
packet->payload[5+slen] = MOSQ_MSB(mosq->port);
packet->payload[6+slen] = MOSQ_LSB(mosq->port);
pthread_mutex_lock(&mosq->state_mutex);
mosq->state = mosq_cs_socks5_request;
pthread_mutex_unlock(&mosq->state_mutex);
mosq->in_packet.pos = 0;
mosq->in_packet.packet_length = 5;
mosq->in_packet.to_process = 5;
mosq->in_packet.payload = _mosquitto_malloc(sizeof(uint8_t)*5);
if(!mosq->in_packet.payload){
_mosquitto_free(packet->payload);
_mosquitto_free(packet);
return MOSQ_ERR_NOMEM;
}
return _mosquitto_packet_queue(mosq, packet);
}else if(mosq->state == mosq_cs_socks5_send_userpass){
packet = _mosquitto_calloc(1, sizeof(struct _mosquitto_packet));
if(!packet) return MOSQ_ERR_NOMEM;
ulen = (int)strlen(mosq->socks5_username);
plen = (int)strlen(mosq->socks5_password);
packet->packet_length = 3 + ulen + plen;
packet->payload = _mosquitto_malloc(sizeof(uint8_t)*packet->packet_length);
packet->payload[0] = 0x01;
packet->payload[1] = ulen;
memcpy(&(packet->payload[2]), mosq->socks5_username, ulen);
packet->payload[2+ulen] = plen;
memcpy(&(packet->payload[3+ulen]), mosq->socks5_password, plen);
pthread_mutex_lock(&mosq->state_mutex);
mosq->state = mosq_cs_socks5_userpass_reply;
pthread_mutex_unlock(&mosq->state_mutex);
mosq->in_packet.pos = 0;
mosq->in_packet.packet_length = 2;
mosq->in_packet.to_process = 2;
mosq->in_packet.payload = _mosquitto_malloc(sizeof(uint8_t)*2);
if(!mosq->in_packet.payload){
_mosquitto_free(packet->payload);
_mosquitto_free(packet);
return MOSQ_ERR_NOMEM;
}
return _mosquitto_packet_queue(mosq, packet);
}
return MOSQ_ERR_SUCCESS;
}
