/*-------------------------------------------------------
Upstream MQTTSnPubAck
-------------------------------------------------------*/
void GatewayControlTask::handleSnPubAck(Event* ev, ClientNode* clnode, MQTTSnMessage* msg){
printf(GREEN_FORMAT1, currentDateTime(), "PUBACK", LEFTARROW, clnode->getNodeId()->c_str(), msgPrint(msg));
MQTTSnPubAck* sPubAck = new MQTTSnPubAck();
MQTTPubAck* pubAck = new MQTTPubAck();
sPubAck->absorb(msg);
pubAck->setMessageId(sPubAck->getMsgId());
clnode->setBrokerSendMessage(pubAck);
Event* ev1 = new Event();
ev1->setBrokerSendEvent(clnode);
_res->getBrokerSendQue()->post(ev1);
delete sPubAck;
}
/*-------------------------------------------------------
Downstream MQTTPubAck
-------------------------------------------------------*/
void GatewayControlTask::handlePuback(Event* ev, ClientNode* clnode, MQTTMessage* msg){
MQTTPubAck* mqMsg = static_cast<MQTTPubAck*>(msg);
MQTTSnPubAck* snMsg = clnode->getWaitedPubAck();
if(snMsg){
printf(BLUE_FORMAT1, currentDateTime(), "PUBACK", RIGHTARROW, clnode->getNodeId()->c_str(), msgPrint(snMsg));
if(snMsg->getMsgId() == mqMsg->getMessageId()){
clnode->setWaitedPubAck(0);
clnode->setClientSendMessage(snMsg);
Event* ev1 = new Event();
ev1->setClientSendEvent(clnode);
_res->getClientSendQue()->post(ev1);
}
}
}
void ClientRecvTask::run(){
if(_sp.begin(_res->getArgv()[ARGV_DEVICE_NAME], B57600, O_RDONLY) == -1){
THROW_EXCEPTION(ExFatal, ERRNO_SYS_02, "can't open device.");
}
_zb.setSerialPort(&_sp);
_res->getClientList()->authorize(FILE_NAME_CLIENT_LIST);
while(true){
XBResponse* resp = new XBResponse();
bool eventSetFlg = true;
if(_zb.getResponse(resp)){
Event* ev = new Event();
ClientNode* clnode = _res->getClientList()->getClient(resp->getRemoteAddress64());
if(!clnode){
if(resp->getPayloadPtr()[1] == MQTTSN_TYPE_CONNECT){
ClientNode* node = _res->getClientList()->createNode(resp->getRemoteAddress64());
if(!node){
delete ev;
D_MQTT("Client is not authorized.\n");
continue;
}
MQTTSnConnect* msg = new MQTTSnConnect();
msg->absorb(resp);
node->setAddress16(resp->getRemoteAddress16());
if(msg->getClientId()->size() > 0){
node->setNodeId(msg->getClientId());
}
node->setClientRecvMessage(msg);
ev->setClientRecvEvent(node);
}else if(resp->getPayloadPtr()[1] == MQTTSN_TYPE_SEARCHGW){
MQTTSnSearchGw* msg = new MQTTSnSearchGw();
msg->absorb(resp);
ev->setEvent(msg);
}else{
eventSetFlg = false;
}
}else{
if (resp->getPayloadPtr()[1] == MQTTSN_TYPE_CONNECT){
MQTTSnConnect* msg = new MQTTSnConnect();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
clnode->setAddress16(resp->getRemoteAddress16());
ev->setClientRecvEvent(clnode);
}else if(resp->getPayloadPtr()[1] == MQTTSN_TYPE_PUBLISH){
MQTTSnPublish* msg = new MQTTSnPublish();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if(resp->getPayloadPtr()[1] == MQTTSN_TYPE_PUBACK){
MQTTSnPubAck* msg = new MQTTSnPubAck();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (resp->getPayloadPtr()[1] == MQTTSN_TYPE_CONNECT){
MQTTSnConnect* msg = new MQTTSnConnect();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (resp->getPayloadPtr()[1] == MQTTSN_TYPE_PINGREQ){
MQTTSnPingReq* msg = new MQTTSnPingReq();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (resp->getPayloadPtr()[1] == MQTTSN_TYPE_DISCONNECT){
MQTTSnDisconnect* msg = new MQTTSnDisconnect();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (resp->getPayloadPtr()[1] == MQTTSN_TYPE_REGISTER){
MQTTSnRegister* msg = new MQTTSnRegister();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (resp->getPayloadPtr()[1] == MQTTSN_TYPE_REGACK){
MQTTSnRegAck* msg = new MQTTSnRegAck();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (resp->getPayloadPtr()[1] == MQTTSN_TYPE_UNSUBSCRIBE){
MQTTSnUnsubscribe* msg = new MQTTSnUnsubscribe();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (resp->getPayloadPtr()[1] == MQTTSN_TYPE_SUBSCRIBE){
MQTTSnSubscribe* msg = new MQTTSnSubscribe();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (resp->getPayloadPtr()[1] == MQTTSN_TYPE_WILLTOPIC){
MQTTSnWillTopic* msg = new MQTTSnWillTopic();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (resp->getPayloadPtr()[1] == MQTTSN_TYPE_WILLMSG){
MQTTSnWillMsg* msg = new MQTTSnWillMsg();
msg->absorb(resp);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if(resp->getPayloadPtr()[1] == MQTTSN_TYPE_SEARCHGW){
MQTTSnSearchGw* msg = new MQTTSnSearchGw();
msg->absorb(resp);
ev->setEvent(msg);
}else{
eventSetFlg = false;
}
}
if(eventSetFlg){
_res->getGatewayEventQue()->post(ev);
}else{
delete ev;
}
}
delete resp;
}
}
/*-------------------------------------------------------
* Upstream MQTTSnPublish
-------------------------------------------------------*/
void GatewayControlTask::handleSnPublish(Event* ev, ClientNode* clnode, MQTTSnMessage* msg){
printf(BLUE_FORMAT2, currentDateTime(), "PUBLISH", LEFTARROW, clnode->getNodeId()->c_str(), msgPrint(msg));
MQTTSnPublish* sPublish = new MQTTSnPublish();
MQTTPublish* mqMsg = new MQTTPublish();
sPublish->absorb(msg);
Topic* tp = clnode->getTopics()->getTopic(sPublish->getTopicId());
if(tp || ((sPublish->getFlags() && MQTTSN_TOPIC_TYPE) == MQTTSN_TOPIC_TYPE_SHORT)){
if(tp){
mqMsg->setTopic(tp->getTopicName());
}else{
string str;
mqMsg->setTopic(sPublish->getTopic(&str));
}
if(sPublish->getMsgId()){
MQTTSnPubAck* sPuback = new MQTTSnPubAck();
sPuback->setMsgId(sPublish->getMsgId());
sPuback->setTopicId(sPublish->getTopicId());
if(clnode->getWaitedPubAck()){
delete clnode->getWaitedPubAck();
}
clnode->setWaitedPubAck(sPuback);
mqMsg->setMessageId(sPublish->getMsgId());
}
mqMsg->setQos(sPublish->getQos());
if(sPublish->getFlags() && MQTTSN_FLAG_DUP){
mqMsg->setDup();
}
if(sPublish->getFlags() && MQTTSN_FLAG_RETAIN){
mqMsg->setRetain();
}
mqMsg->setPayload(sPublish->getData() , sPublish->getDataLength());
clnode->setBrokerSendMessage(mqMsg);
Event* ev1 = new Event();
ev1->setBrokerSendEvent(clnode);
_res->getBrokerSendQue()->post(ev1);
}else{
if(sPublish->getMsgId()){
MQTTSnPubAck* sPuback = new MQTTSnPubAck();
sPuback->setMsgId(sPublish->getMsgId());
sPuback->setTopicId(sPublish->getTopicId());
sPuback->setReturnCode(MQTTSN_RC_REJECTED_INVALID_TOPIC_ID);
clnode->setClientSendMessage(sPuback);
Event* ev1 = new Event();
ev1->setClientSendEvent(clnode);
printf(BLUE_FORMAT1, currentDateTime(), "PUBACK", RIGHTARROW, clnode->getNodeId()->c_str(), msgPrint(sPuback));
_res->getClientSendQue()->post(ev1); // Send PubAck INVALID_TOPIC_ID
}
}
delete sPublish;
}
void ClientRecvTask::run(){
NETWORK_CONFIG config;
char param[TOMYFRAME_PARAM_MAX];
bool secure = false; // TCP
#ifdef NETWORK_XBEE
if(_res->getParam("BaudRate",param) == 0){
int val = atoi(param);
switch(val){
case 9600:
config.baudrate = B9600;
break;
case 19200:
config.baudrate =B19200;
break;
case 38400:
config.baudrate =B38400;
break;
case 57600:
config.baudrate =B57600;
break;
case 115200:
config.baudrate = B115200;
break;
default:
THROW_EXCEPTION(ExFatal, ERRNO_APL_01, "Invalid baud rate!"); // ABORT
}
}else{
config.baudrate = B57600;
}
config.flag = O_RDONLY;
if(_res->getParam("SerialDevice",param) == 0){
config.device = strdup(param);
}
if(_res->getParam("SecureConnection",param) == 0){
if(!strcasecmp(param, "YES")){
secure = true; // TLS
}
}
_res->getClientList()->authorize(FILE_NAME_CLIENT_LIST, secure);
_network = new Network();
#endif
#ifdef NETWORK_UDP
if(_res->getParam("BroadcastIP", param) == 0){
config.ipAddress = strdup(param);
}
if(_res->getParam("BroadcastPortNo",param) == 0){
config.gPortNo = atoi(param);
}
if(_res->getParam("GatewayPortNo",param) == 0){
config.uPortNo = atoi(param);
}
_network = _res->getNetwork();
#endif
#ifdef NETWORK_XXXXX
_network = _res->getNetwork();
#endif
if(_network->initialize(config) < 0){
THROW_EXCEPTION(ExFatal, ERRNO_APL_01, "can't open the client port."); // ABORT
}
uint8_t* recvMsg;
int len;
while(true){
bool eventSetFlg = true;
recvMsg = _network->Network::getResponce(&len);
if (recvMsg != 0){
uint8_t offset = 0;
if (recvMsg[0] == 0x01){
offset = 3;
}else{
offset = 1;
}
Event* ev = new Event();
ClientNode* clnode = _res->getClientList()->getClient(_network->getAddrMsb(),_network->getAddrLsb(),
_network->getAddr16());
if(!clnode){
if(*(recvMsg + offset) == MQTTSN_TYPE_CONNECT){
#ifdef NETWORK_XBEE
ClientNode* node = _res->getClientList()->createNode(secure, _network->getAddrMsb(),_network->getAddrLsb(),0);
#endif
#ifdef NETWORK_UDP
ClientNode* node = _res->getClientList()->createNode(secure, _network->getAddrMsb(),_network->getAddrLsb(),
_network->getAddr16());
#endif
#ifdef NETWORK_XXXXX
ClientNode* node = _res->getClientList()->createNode(secure, _network->getAddrMsb(),_network->getAddrLsb(),
_network->getAddr16());
#endif
if(!node){
delete ev;
LOGWRITE("Client is not authorized.\n");
continue;
}
MQTTSnConnect* msg = new MQTTSnConnect();
msg->absorb(recvMsg);
node->setClientAddress16(_network->getAddr16());
if(msg->getClientId()->size() > 0){
node->setNodeId(msg->getClientId());
}
node->setClientRecvMessage(msg);
ev->setClientRecvEvent(node);
}else if(*(recvMsg + offset) == MQTTSN_TYPE_SEARCHGW){
MQTTSnSearchGw* msg = new MQTTSnSearchGw();
msg->absorb(recvMsg);
ev->setEvent(msg);
}else{
eventSetFlg = false;
}
}else{
if (*(recvMsg + offset) == MQTTSN_TYPE_CONNECT){
MQTTSnConnect* msg = new MQTTSnConnect();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
clnode->setClientAddress16(_network->getAddr16());
ev->setClientRecvEvent(clnode);
}else if(*(recvMsg + offset) == MQTTSN_TYPE_PUBLISH){
MQTTSnPublish* msg = new MQTTSnPublish();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if(*(recvMsg + offset) == MQTTSN_TYPE_PUBACK){
MQTTSnPubAck* msg = new MQTTSnPubAck();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if(*(recvMsg + offset) == MQTTSN_TYPE_PUBREC){
MQTTSnPubRec* msg = new MQTTSnPubRec();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if(*(recvMsg + offset) == MQTTSN_TYPE_PUBREL){
MQTTSnPubRel* msg = new MQTTSnPubRel();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if(*(recvMsg + offset) == MQTTSN_TYPE_PUBCOMP){
MQTTSnPubComp* msg = new MQTTSnPubComp();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (*(recvMsg + offset) == MQTTSN_TYPE_CONNECT){
MQTTSnConnect* msg = new MQTTSnConnect();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (*(recvMsg + offset) == MQTTSN_TYPE_PINGREQ){
MQTTSnPingReq* msg = new MQTTSnPingReq();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (*(recvMsg + offset) == MQTTSN_TYPE_DISCONNECT){
MQTTSnDisconnect* msg = new MQTTSnDisconnect();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (*(recvMsg + offset) == MQTTSN_TYPE_REGISTER){
MQTTSnRegister* msg = new MQTTSnRegister();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (*(recvMsg + offset) == MQTTSN_TYPE_REGACK){
MQTTSnRegAck* msg = new MQTTSnRegAck();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (*(recvMsg + offset) == MQTTSN_TYPE_UNSUBSCRIBE){
MQTTSnUnsubscribe* msg = new MQTTSnUnsubscribe();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (*(recvMsg + offset) == MQTTSN_TYPE_SUBSCRIBE){
MQTTSnSubscribe* msg = new MQTTSnSubscribe();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (*(recvMsg + offset) == MQTTSN_TYPE_WILLTOPIC){
MQTTSnWillTopic* msg = new MQTTSnWillTopic();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if (*(recvMsg + offset) == MQTTSN_TYPE_WILLMSG){
MQTTSnWillMsg* msg = new MQTTSnWillMsg();
msg->absorb(recvMsg);
clnode->setClientRecvMessage(msg);
ev->setClientRecvEvent(clnode);
}else if(*(recvMsg + offset) == MQTTSN_TYPE_SEARCHGW){
MQTTSnSearchGw* msg = new MQTTSnSearchGw();
clnode->disconnected();
msg->absorb(recvMsg);
ev->setEvent(msg);
}else{
eventSetFlg = false;
}
}
if(eventSetFlg){
_res->getGatewayEventQue()->post(ev);
}else{
delete ev;
}
}
}
}
