#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <limits.h>
#include <elwix.h>
#include <aitmqtt.h>


int
main()
{
	mqtt_msg_t *m;
	mqtt_subscr_t s[4], *ss;
	mqtthdr_connflgs_t flg;
	mqtthdr_connack_t cack;
	u_short ka, msgID;
	int i, len;
	char cid[BUFSIZ], user[BUFSIZ], pass[BUFSIZ], topic[BUFSIZ], *t, *msg, *pub;
	struct mqtthdr *hdr;
	u_char *qoses;

	/* conn* */
	m = mqtt_msgCONNECT("MRYN", 4, 0, "aaaaa", NULL, "bbb", NULL, 0, 0, 0);
	printf("connect=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	t = msg = NULL;
	cack = mqtt_readCONNECT(m, &ka, cid, sizeof cid, user, sizeof user, pass, sizeof pass, &t, &msg);
	flg.flags = cack.reserved;
	printf("read connect %d flags:: clean=%d will=%d qos=%d retain=%d pass=%d user=%d\n", cack.retcode, 
			flg.clean_sess, flg.will_flg, flg.will_qos, flg.will_retain, flg.password, flg.username);
	if (flg.reserved) {
		printf("Error:: mqtt_readCONNECT() #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
		mqtt_msgFree(&m, 0);
		return 1;
	}
	printf("++> KA=%d sec, ConnID=%s User=%s Pass=%s Will_Topic=%s Will_Message=%s\n", ka, 
			cid, user, pass, t, msg);
	if (t)
		e_free(t);
	if (msg)
		e_free(msg);
	mqtt_msgFree(&m, 0);

	m = mqtt_msgCONNACK(1);
	printf("connack=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	printf("read connack=%d\n", mqtt_readCONNACK(m));
	mqtt_msgFree(&m, 0);
	getchar();

	/* pub* */
	m = mqtt_msgPUBLISH("AAA/bbb/CCC/ddd", 7, 0, 2, 0, "OLE!!!", 7);
	printf("publish=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	if (!(len = mqtt_readPUBLISH(m, topic, sizeof topic, &msgID, (void**) &pub))) {
		printf("Error:: mqtt_readPUBLISH() #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
		mqtt_msgFree(&m, 0);
		return 2;
	} else
		hdr = (struct mqtthdr*) m->msg_base;
	printf("read publish: dup=%d qos=%d retain=%d\n", hdr->mqtt_msg.dup, hdr->mqtt_msg.qos, hdr->mqtt_msg.retain);
	printf("++> topic=%s MessageID=%d DATA=(%d)%s\n", topic, msgID, len, pub);
	mqtt_msgFree(&m, 0);
	e_free(pub);

	m = mqtt_msgPUBACK(10);
	printf("puback=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	printf("read puback=%d\n", mqtt_readPUBACK(m));
	mqtt_msgFree(&m, 0);

	m = mqtt_msgPUBREC(11);
	printf("pubrec=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	printf("read pubrec=%d\n", mqtt_readPUBREC(m));
	mqtt_msgFree(&m, 0);

	m = mqtt_msgPUBREL(12);
	printf("pubrel=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	printf("read pubrel=%d\n", mqtt_readPUBREL(m));
	mqtt_msgFree(&m, 0);

	m = mqtt_msgPUBCOMP(13);
	printf("pubcomp=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	printf("read pubcomp=%d\n", mqtt_readPUBCOMP(m));
	mqtt_msgFree(&m, 0);
	getchar();

	/* sub* */
	memset(s, 0, sizeof s);
	s[0].sub_topic.msg_len = 3;
	s[0].sub_topic.msg_base = "a/b";
	s[0].sub_qos = MQTT_QOS_ACK;
	s[1].sub_topic.msg_len = 3;
	s[1].sub_topic.msg_base = "c/d";
	s[1].sub_qos = MQTT_QOS_ONCE;
	s[2].sub_topic.msg_len = 7;
	s[2].sub_topic.msg_base = "x/y/z/Q";
	s[2].sub_qos = MQTT_QOS_EXACTLY;
	m = mqtt_msgSUBSCRIBE(s, 10);
	printf("subscribe=%d/%p\n", m->msg_len, m);
	/*
	for (i = 0; i < m->msg_len; i++)
		printf("%d\n", ((u_char*) m->msg_base)[i]);
		*/
	if (mqtt_readSUBSCRIBE(m, &msgID, &ss) == -1) {
		printf("Error:: mqtt_readSUBSCRIBE() #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
		mqtt_msgFree(&m, 0);
		return 3;
	}
	printf("read subscribe: dup=%d qos=%d retain=%d\n", hdr->mqtt_msg.dup, hdr->mqtt_msg.qos, hdr->mqtt_msg.retain);
	printf("++> MessageID=%d\n", msgID);
	for (i = 0; ss[i].sub_topic.msg_base; i++)
		printf(" >>> QoS=%d Topic(%d)=%s\n", ss[i].sub_qos, ss[i].sub_topic.msg_len, 
				(char*) ss[i].sub_topic.msg_base);
	mqtt_msgFree(&m, 0);
	mqtt_subFree(&ss);

	m = mqtt_msgSUBACK(s, 10);
	printf("suback=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	printf("read suback=%d\n", (len = mqtt_readSUBACK(m, &msgID, &qoses)));
	for (i = 0; i < len; i++)
		printf(" >>> QoS=%d\n", qoses[i]);
	mqtt_msgFree(&m, 0);
	e_free(qoses);

	m = mqtt_msgUNSUBSCRIBE(s, 10, 0, 1);
	printf("unsubscribe=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	if (mqtt_readUNSUBSCRIBE(m, &msgID, &ss) == -1) {
		printf("Error:: mqtt_readUNSUBSCRIBE() #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
		mqtt_msgFree(&m, 0);
		return 3;
	}
	printf("read unsubscribe: dup=%d qos=%d retain=%d\n", hdr->mqtt_msg.dup, hdr->mqtt_msg.qos, hdr->mqtt_msg.retain);
	printf("++> MessageID=%d\n", msgID);
	for (i = 0; ss[i].sub_topic.msg_base; i++)
		printf(" >>> Topic(%d)=%s\n", ss[i].sub_topic.msg_len, (char*) ss[i].sub_topic.msg_base);
	mqtt_msgFree(&m, 0);
	mqtt_subFree(&ss);

	m = mqtt_msgUNSUBACK(10);
	printf("unsuback=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	printf("read unsuback=%d\n", mqtt_readUNSUBACK(m));
	mqtt_msgFree(&m, 0);
	getchar();

	/* ping* */
	m = mqtt_msgPINGREQ();
	printf("pingreq=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	printf("read pingreq=%d\n", mqtt_readPINGREQ(m));
	mqtt_msgFree(&m, 0);
	m = mqtt_msgPINGRESP();
	printf("pingresp=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	printf("read pingresp=%d\n", mqtt_readPINGRESP(m));
	mqtt_msgFree(&m, 0);

	m = mqtt_msgDISCONNECT();
	printf("disconnect=%d/%p\n", m->msg_len, m);
//	for (i = 0; i < m->msg_len; i++)
//		printf("%d\n", ((u_char*) m->msg_base)[i]);
	printf("read disconnect=%d\n", mqtt_readDISCONNECT(m));
	mqtt_msgFree(&m, 0);
	return 0;
}