/*-------------------------------------------------------
Upstream MQTTSnRegister
-------------------------------------------------------*/
void GatewayControlTask::handleSnRegister(Event* ev, ClientNode* clnode, MQTTSnMessage* msg){
printf(FORMAT2, currentDateTime(), "REGISTER", LEFTARROW, clnode->getNodeId()->c_str(), msgPrint(msg));
MQTTSnRegister* snMsg = new MQTTSnRegister();
MQTTSnRegAck* respMsg = new MQTTSnRegAck();
snMsg->absorb(msg);
respMsg->setMsgId(snMsg->getMsgId());
uint16_t tpId = clnode->getTopics()->createTopic(snMsg->getTopicName());
respMsg->setTopicId(tpId);
respMsg->setReturnCode(MQTTSN_RC_ACCEPTED);
clnode->setClientSendMessage(respMsg);
Event* evrg = new Event();
evrg->setClientSendEvent(clnode);
printf(FORMAT1, currentDateTime(), "REGACK", RIGHTARROW, clnode->getNodeId()->c_str(), msgPrint(respMsg));
_res->getClientSendQue()->post(evrg);
delete snMsg;
}
/*-------------------------------------------------------
Downstream MQTTPublish
-------------------------------------------------------*/
void GatewayControlTask::handlePublish(Event* ev, ClientNode* clnode, MQTTMessage* msg){
MQTTPublish* mqMsg = static_cast<MQTTPublish*>(msg);
MQTTSnPublish* snMsg = new MQTTSnPublish();
string* tp = mqMsg->getTopic();
uint16_t tpId;
if(tp->size() == 2){
tpId = getUint16((uint8_t*)tp);
snMsg->setFlags(MQTTSN_TOPIC_TYPE_SHORT);
}else{
tpId = clnode->getTopics()->getTopicId(tp);
snMsg->setFlags(MQTTSN_TOPIC_TYPE_NORMAL);
}
if(tpId == 0){
/* ----- may be a publish message response of subscribed with '#' or '+' -----*/
tpId = clnode->getTopics()->createTopic(tp);
if(tpId > 0){
MQTTSnRegister* regMsg = new MQTTSnRegister();
regMsg->setTopicId(tpId);
regMsg->setTopicName(tp);
printf(FORMAT2, currentDateTime(), "REGISTER", RIGHTARROW, clnode->getNodeId()->c_str(), msgPrint(regMsg));
clnode->setClientSendMessage(regMsg);
Event* evrg = new Event();
evrg->setClientSendEvent(clnode);
_res->getClientSendQue()->post(evrg); // Send Register first.
}else{
printf("GatewayControlTask Can't create Topic %s\n", tp->c_str());
return;
}
}
snMsg->setTopicId(tpId);
snMsg->setMsgId(mqMsg->getMessageId());
snMsg->setData(mqMsg->getPayload(),mqMsg->getPayloadLength());
snMsg->setQos(mqMsg->getQos());
if(mqMsg->isDup()){
snMsg->setDup();
}
if(mqMsg->isRetain()){
snMsg->setDup();
}
clnode->setClientSendMessage(snMsg);
printf(GREEN_FORMAT1, currentDateTime(), "PUBLISH", RIGHTARROW, clnode->getNodeId()->c_str(), msgPrint(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;
}
}
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;
}
}
}
}
