mirror of
https://github.com/eclipse/paho.mqtt-sn.embedded-c.git
synced 2025-12-15 08:26:51 +01:00
Update & BugFix Add GatewayTester
Signed-off-by: tomoaki <tomoaki@tomy-tech.com>
This commit is contained in:
tomoaki
388
MQTTSNGateway/GatewayTester/src/LSubscribeManager.cpp
Normal file
388
MQTTSNGateway/GatewayTester/src/LSubscribeManager.cpp
Normal file
@@ -0,0 +1,388 @@
|
||||
/**************************************************************************************
|
||||
* Copyright (c) 2016, Tomoaki Yamaguchi
|
||||
*
|
||||
* All rights reserved. This program and the accompanying materials
|
||||
* are made available under the terms of the Eclipse Public License v1.0
|
||||
* and Eclipse Distribution License v1.0 which accompany this distribution.
|
||||
*
|
||||
* The Eclipse Public License is available at
|
||||
* http://www.eclipse.org/legal/epl-v10.html
|
||||
* and the Eclipse Distribution License is available at
|
||||
* http://www.eclipse.org/org/documents/edl-v10.php.
|
||||
*
|
||||
* Contributors:
|
||||
* Tomoaki Yamaguchi - initial API and implementation and/or initial documentation
|
||||
**************************************************************************************/
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "LMqttsnClientApp.h"
|
||||
#include "LTimer.h"
|
||||
#include "LScreen.h"
|
||||
#include "LGwProxy.h"
|
||||
#include "LMqttsnClient.h"
|
||||
#include "LSubscribeManager.h"
|
||||
|
||||
using namespace std;
|
||||
using namespace linuxAsyncClient;
|
||||
|
||||
extern void setUint16(uint8_t* pos, uint16_t val);
|
||||
extern uint16_t getUint16(const uint8_t* pos);
|
||||
extern LMqttsnClient* theClient;
|
||||
extern SUBSCRIBE_LIST;
|
||||
extern LScreen* theScreen;
|
||||
#define SUB_DONE 1
|
||||
#define SUB_READY 0
|
||||
/*========================================
|
||||
Class SubscribeManager
|
||||
=======================================*/
|
||||
LSubscribeManager::LSubscribeManager()
|
||||
{
|
||||
_first = 0;
|
||||
_last = 0;
|
||||
}
|
||||
|
||||
LSubscribeManager::~LSubscribeManager()
|
||||
{
|
||||
SubElement* elm = _first;
|
||||
SubElement* sav = 0;
|
||||
while (elm)
|
||||
{
|
||||
sav = elm->next;
|
||||
if (elm != 0)
|
||||
{
|
||||
free(elm);
|
||||
}
|
||||
elm = sav;
|
||||
}
|
||||
}
|
||||
|
||||
void LSubscribeManager::onConnect(void)
|
||||
{
|
||||
ASSERT("\033[0m\033[0;32m Attempting OnConnect.....\033[0m\033[0;37m\n");
|
||||
if (_first == 0)
|
||||
{
|
||||
for (uint8_t i = 0; theOnPublishList[i].topic != 0; i++)
|
||||
{
|
||||
subscribe(theOnPublishList[i].topic, theOnPublishList[i].pubCallback, theOnPublishList[i].qos);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
SubElement* elm = _first;
|
||||
SubElement* pelm;
|
||||
do
|
||||
{
|
||||
pelm = elm;
|
||||
if (elm->msgType == MQTTSN_TYPE_SUBSCRIBE)
|
||||
{
|
||||
elm->done = SUB_READY;
|
||||
elm->retryCount = MQTTSN_RETRY_COUNT;
|
||||
subscribe(elm->topicName, elm->callback, elm->qos);
|
||||
}
|
||||
elm = pelm->next;
|
||||
} while (pelm->next);
|
||||
}
|
||||
|
||||
while (!theClient->getSubscribeManager()->isDone())
|
||||
{
|
||||
theClient->getGwProxy()->getMessage();
|
||||
}
|
||||
ASSERT("\033[0m\033[0;32m OnConnect complete\033[0m\033[0;37m\n");
|
||||
ASSERT("\033[0m\033[0;32m Test is Ready.\033[0m\033[0;37m\n");
|
||||
}
|
||||
|
||||
bool LSubscribeManager::isDone(void)
|
||||
{
|
||||
SubElement* elm = _first;
|
||||
SubElement* prevelm;
|
||||
while (elm)
|
||||
{
|
||||
prevelm = elm;
|
||||
if (elm->done == SUB_READY)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
elm = prevelm->next;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
void LSubscribeManager::send(SubElement* elm)
|
||||
{
|
||||
if (elm->done == SUB_DONE)
|
||||
{
|
||||
return;
|
||||
}
|
||||
uint8_t msg[MQTTSN_MAX_MSG_LENGTH + 1];
|
||||
if (elm->topicType == MQTTSN_TOPIC_TYPE_NORMAL)
|
||||
{
|
||||
msg[0] = 5 + strlen(elm->topicName);
|
||||
strcpy((char*) msg + 5, elm->topicName);
|
||||
}
|
||||
else
|
||||
{
|
||||
msg[0] = 7;
|
||||
setUint16(msg + 5, elm->topicId);
|
||||
}
|
||||
msg[1] = elm->msgType;
|
||||
msg[2] = elm->qos | elm->topicType;
|
||||
if (elm->retryCount == MQTTSN_RETRY_COUNT)
|
||||
{
|
||||
elm->msgId = theClient->getGwProxy()->getNextMsgId();
|
||||
}
|
||||
|
||||
if ((elm->retryCount < MQTTSN_RETRY_COUNT) && elm->msgType == MQTTSN_TYPE_SUBSCRIBE)
|
||||
{
|
||||
msg[2] = msg[2] | MQTTSN_FLAG_DUP;
|
||||
}
|
||||
|
||||
setUint16(msg + 3, elm->msgId);
|
||||
|
||||
theClient->getGwProxy()->connect();
|
||||
theClient->getGwProxy()->writeMsg(msg);
|
||||
theClient->getGwProxy()->resetPingReqTimer();
|
||||
elm->sendUTC = time(NULL);
|
||||
elm->retryCount--;
|
||||
}
|
||||
|
||||
void LSubscribeManager::subscribe(const char* topicName, TopicCallback onPublish, uint8_t qos)
|
||||
{
|
||||
uint8_t topicType = MQTTSN_TOPIC_TYPE_NORMAL;
|
||||
if ( strlen(topicName) <= 2)
|
||||
{
|
||||
topicType = MQTTSN_TOPIC_TYPE_SHORT;
|
||||
}
|
||||
SubElement* elm = add(MQTTSN_TYPE_SUBSCRIBE, topicName, 0, topicType, qos, onPublish);
|
||||
send(elm);
|
||||
}
|
||||
|
||||
void LSubscribeManager::subscribe(uint16_t topicId, TopicCallback onPublish, uint8_t qos, uint8_t topicType)
|
||||
{
|
||||
SubElement* elm = add(MQTTSN_TYPE_SUBSCRIBE, 0, topicId, topicType, qos, onPublish);
|
||||
send(elm);
|
||||
}
|
||||
|
||||
void LSubscribeManager::unsubscribe(const char* topicName)
|
||||
{
|
||||
SubElement* elm = add(MQTTSN_TYPE_UNSUBSCRIBE, topicName, 0, MQTTSN_TOPIC_TYPE_NORMAL, 0, 0);
|
||||
send(elm);
|
||||
}
|
||||
|
||||
void LSubscribeManager::unsubscribe(uint16_t topicId, uint8_t topicType)
|
||||
{
|
||||
SubElement* elm = add(MQTTSN_TYPE_UNSUBSCRIBE, 0, topicId, topicType, 0, 0);
|
||||
send(elm);
|
||||
}
|
||||
|
||||
void LSubscribeManager::checkTimeout(void)
|
||||
{
|
||||
SubElement* elm = _first;
|
||||
|
||||
while (elm)
|
||||
{
|
||||
if (elm->sendUTC + MQTTSN_TIME_RETRY < time(NULL))
|
||||
{
|
||||
if (elm->retryCount >= 0)
|
||||
{
|
||||
send(elm);
|
||||
}
|
||||
else
|
||||
{
|
||||
if ( elm->done == SUB_READY )
|
||||
{
|
||||
if (elm->msgType == MQTTSN_TYPE_SUBSCRIBE)
|
||||
{
|
||||
ASSERT("\033[0m\033[0;31m\n!!!!!! SUBSCRIBE Error !!!!! Topic : %s\033[0m\033[0;37m\n\n", (char*)elm->topicName);
|
||||
}else{
|
||||
ASSERT("\033[0m\033[0;31m\n!!!!!! UNSUBSCRIBE Error !!!!! Topic : %s\033[0m\033[0;37m\n\n", (char*)elm->topicName);
|
||||
}
|
||||
elm->done = SUB_DONE;
|
||||
}
|
||||
}
|
||||
}
|
||||
elm = elm->next;
|
||||
}
|
||||
}
|
||||
|
||||
void LSubscribeManager::responce(const uint8_t* msg)
|
||||
{
|
||||
if (msg[0] == MQTTSN_TYPE_SUBACK)
|
||||
{
|
||||
uint16_t topicId = getUint16(msg + 2);
|
||||
uint16_t msgId = getUint16(msg + 4);
|
||||
uint8_t rc = msg[6];
|
||||
|
||||
LTopicTable* tt = theClient->getGwProxy()->getTopicTable();
|
||||
SubElement* elm = getElement(msgId);
|
||||
if (elm)
|
||||
{
|
||||
if ( rc == MQTTSN_RC_ACCEPTED )
|
||||
{
|
||||
tt->add((char*) elm->topicName, topicId, elm->topicType, elm->callback);
|
||||
getElement(msgId)->done = SUB_DONE;
|
||||
ASSERT("\033[0m\033[0;32m Topic \"%s\" Id : %d was Subscribed. \033[0m\033[0;37m\n\n", getElement(msgId)->topicName, getElement(msgId)->topicId);
|
||||
}
|
||||
else
|
||||
{
|
||||
remove(elm);
|
||||
ASSERT("\033[0m\033[0;31m SUBACK Invalid messageId. %s\033[0m\033[0;37m\n\n", getElement(msgId)->topicName);
|
||||
}
|
||||
}
|
||||
}
|
||||
else if (msg[0] == MQTTSN_TYPE_UNSUBACK)
|
||||
{
|
||||
uint16_t msgId = getUint16(msg + 1);
|
||||
SubElement* elm = getElement(msgId);
|
||||
if (elm)
|
||||
{
|
||||
LTopicTable* tt = theClient->getGwProxy()->getTopicTable();
|
||||
tt->setCallback(elm->topicName, 0);
|
||||
ASSERT("\033[0m\033[0;32m Topic \"%s\" Id : %d was Unsubscribed. \033[0m\033[0;37m\n\n", getElement(msgId)->topicName, getElement(msgId)->topicId);
|
||||
remove(getElement(msgId));
|
||||
}
|
||||
else
|
||||
{
|
||||
ASSERT("\033[0m\033[0;31m UNSUBACK Invalid messageId. %s\033[0m\033[0;37m\n\n", getElement(msgId)->topicName);
|
||||
remove(getElement(msgId));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/* SubElement operations */
|
||||
|
||||
SubElement* LSubscribeManager::add(uint8_t msgType, const char* topicName, uint16_t topicId, uint8_t topicType,
|
||||
uint8_t qos, TopicCallback callback)
|
||||
{
|
||||
SubElement* elm = 0;
|
||||
if (topicName )
|
||||
{
|
||||
elm = getElement(topicName, msgType);
|
||||
}
|
||||
else
|
||||
{
|
||||
elm = getElement(topicId, topicType);
|
||||
}
|
||||
if ( elm == 0 )
|
||||
{
|
||||
elm = (SubElement*) calloc(1, sizeof(SubElement));
|
||||
if (elm == 0)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
if (_last == 0)
|
||||
{
|
||||
_first = elm;
|
||||
_last = elm;
|
||||
}
|
||||
else
|
||||
{
|
||||
elm->prev = _last;
|
||||
_last->next = elm;
|
||||
_last = elm;
|
||||
}
|
||||
}
|
||||
|
||||
elm->msgType = msgType;
|
||||
elm->callback = callback;
|
||||
elm->topicName = topicName;
|
||||
elm->topicId = topicId;
|
||||
elm->topicType = topicType;
|
||||
|
||||
if (qos == 1)
|
||||
{
|
||||
elm->qos = MQTTSN_FLAG_QOS_1;
|
||||
}
|
||||
else if (qos == 2)
|
||||
{
|
||||
elm->qos = MQTTSN_FLAG_QOS_2;
|
||||
}
|
||||
else
|
||||
{
|
||||
elm->qos = MQTTSN_FLAG_QOS_0;
|
||||
}
|
||||
elm->msgId = 0;
|
||||
elm->retryCount = MQTTSN_RETRY_COUNT;
|
||||
elm->done = SUB_READY;
|
||||
elm->sendUTC = 0;
|
||||
|
||||
return elm;
|
||||
}
|
||||
|
||||
void LSubscribeManager::remove(SubElement* elm)
|
||||
{
|
||||
if (elm)
|
||||
{
|
||||
if (elm->prev == 0)
|
||||
{
|
||||
_first = elm->next;
|
||||
if (elm->next != 0)
|
||||
{
|
||||
elm->next->prev = 0;
|
||||
_last = elm->next;
|
||||
}
|
||||
free(elm);
|
||||
}
|
||||
else
|
||||
{
|
||||
if ( elm->next == 0 )
|
||||
{
|
||||
_last = elm->prev;
|
||||
}
|
||||
elm->prev->next = elm->next;
|
||||
free(elm);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
SubElement* LSubscribeManager::getElement(uint16_t msgId)
|
||||
{
|
||||
SubElement* elm = _first;
|
||||
while (elm)
|
||||
{
|
||||
if (elm->msgId == msgId)
|
||||
{
|
||||
return elm;
|
||||
}
|
||||
else
|
||||
{
|
||||
elm = elm->next;
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
SubElement* LSubscribeManager::getElement(const char* topicName, uint8_t msgType)
|
||||
{
|
||||
SubElement* elm = _first;
|
||||
while (elm)
|
||||
{
|
||||
if (strcmp(elm->topicName, topicName) == 0 && elm->msgType == msgType)
|
||||
{
|
||||
return elm;
|
||||
}
|
||||
else
|
||||
{
|
||||
elm = elm->next;
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
SubElement* LSubscribeManager::getElement(uint16_t topicId, uint8_t topicType)
|
||||
{
|
||||
SubElement* elm = _first;
|
||||
while (elm)
|
||||
{
|
||||
if (elm->topicId == topicId && elm->topicType == topicType)
|
||||
{
|
||||
return elm;
|
||||
}
|
||||
else
|
||||
{
|
||||
elm = elm->next;
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user