12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442 |
- /*******************************************************************************
- * Copyright (c) 2009, 2020 IBM Corp.
- *
- * All rights reserved. This program and the accompanying materials
- * are made available under the terms of the Eclipse Public License v2.0
- * and Eclipse Distribution License v1.0 which accompany this distribution.
- *
- * The Eclipse Public License is available at
- * https://www.eclipse.org/legal/epl-2.0/
- * and the Eclipse Distribution License is available at
- * http://www.eclipse.org/org/documents/edl-v10.php.
- *
- * Contributors:
- * Ian Craggs - initial API and implementation and/or initial documentation
- * Ian Craggs - MQTT 3.1.1 support
- * Ian Craggs - change will message test back to using proxy
- * Ian Craggs - MQTT 5.0 support
- *******************************************************************************/
- /**
- * @file
- * Tests for the MQTT C client
- */
- #include "MQTTClient.h"
- #include <string.h>
- #include <stdlib.h>
- #if !defined(_WINDOWS)
- #include <sys/time.h>
- #include <sys/socket.h>
- #include <unistd.h>
- #include <errno.h>
- #else
- #include <windows.h>
- #define setenv(a, b, c) _putenv_s(a, b)
- #endif
- #define ARRAY_SIZE(a) (sizeof(a) / sizeof(a[0]))
- void usage(void)
- {
- printf("help!!\n");
- exit(EXIT_FAILURE);
- }
- struct Options
- {
- char* connection; /**< connection to system under test. */
- char** haconnections;
- char* proxy_connection;
- int hacount;
- int verbose;
- int test_no;
- int MQTTVersion;
- int iterations;
- } options =
- {
- "tcp://localhost:1883",
- NULL,
- "tcp://localhost:1884",
- 0,
- 0,
- 0,
- MQTTVERSION_5,
- 1,
- };
- void getopts(int argc, char** argv)
- {
- int count = 1;
- while (count < argc)
- {
- if (strcmp(argv[count], "--test_no") == 0)
- {
- if (++count < argc)
- options.test_no = atoi(argv[count]);
- else
- usage();
- }
- else if (strcmp(argv[count], "--connection") == 0)
- {
- if (++count < argc)
- {
- options.connection = argv[count];
- printf("\nSetting connection to %s\n", options.connection);
- }
- else
- usage();
- }
- else if (strcmp(argv[count], "--haconnections") == 0)
- {
- if (++count < argc)
- {
- char* tok = strtok(argv[count], " ");
- options.hacount = 0;
- options.haconnections = malloc(sizeof(char*) * 5);
- while (tok)
- {
- options.haconnections[options.hacount] = malloc(strlen(tok) + 1);
- strcpy(options.haconnections[options.hacount], tok);
- options.hacount++;
- tok = strtok(NULL, " ");
- }
- }
- else
- usage();
- }
- else if (strcmp(argv[count], "--proxy_connection") == 0)
- {
- if (++count < argc)
- options.proxy_connection = argv[count];
- else
- usage();
- }
- else if (strcmp(argv[count], "--MQTTversion") == 0)
- {
- if (++count < argc)
- {
- options.MQTTVersion = atoi(argv[count]);
- printf("setting MQTT version to %d\n", options.MQTTVersion);
- }
- else
- usage();
- }
- else if (strcmp(argv[count], "--iterations") == 0)
- {
- if (++count < argc)
- options.iterations = atoi(argv[count]);
- else
- usage();
- }
- else if (strcmp(argv[count], "--verbose") == 0)
- {
- options.verbose = 1;
- printf("\nSetting verbose on\n");
- }
- count++;
- }
- }
- #define LOGA_DEBUG 0
- #define LOGA_INFO 1
- #include <stdarg.h>
- #include <time.h>
- #include <sys/timeb.h>
- void MyLog(int LOGA_level, char* format, ...)
- {
- static char msg_buf[256];
- va_list args;
- #if defined(_WIN32) || defined(_WINDOWS)
- struct timeb ts;
- #else
- struct timeval ts;
- #endif
- struct tm timeinfo;
- if (LOGA_level == LOGA_DEBUG && options.verbose == 0)
- return;
- #if defined(_WIN32) || defined(_WINDOWS)
- ftime(&ts);
- localtime_s(&timeinfo, &ts.time);
- #else
- gettimeofday(&ts, NULL);
- localtime_r(&ts.tv_sec, &timeinfo);
- #endif
- strftime(msg_buf, 80, "%Y%m%d %H%M%S", &timeinfo);
- #if defined(_WIN32) || defined(_WINDOWS)
- sprintf(&msg_buf[strlen(msg_buf)], ".%.3hu ", ts.millitm);
- #else
- sprintf(&msg_buf[strlen(msg_buf)], ".%.3lu ", ts.tv_usec / 1000);
- #endif
- va_start(args, format);
- vsnprintf(&msg_buf[strlen(msg_buf)], sizeof(msg_buf) - strlen(msg_buf), format, args);
- va_end(args);
- printf("%s\n", msg_buf);
- fflush(stdout);
- }
- #if defined(_WIN32) || defined(_WINDOWS)
- #define mqsleep(A) Sleep(1000*A)
- #define START_TIME_TYPE DWORD
- static DWORD start_time = 0;
- START_TIME_TYPE start_clock(void)
- {
- return GetTickCount();
- }
- #elif defined(AIX)
- #define mqsleep sleep
- #define START_TIME_TYPE struct timespec
- START_TIME_TYPE start_clock(void)
- {
- static struct timespec start;
- clock_gettime(CLOCK_REALTIME, &start);
- return start;
- }
- #else
- #define mqsleep sleep
- #define START_TIME_TYPE struct timeval
- /* TODO - unused - remove? static struct timeval start_time; */
- START_TIME_TYPE start_clock(void)
- {
- struct timeval start_time;
- gettimeofday(&start_time, NULL);
- return start_time;
- }
- #endif
- #if defined(_WIN32)
- long elapsed(START_TIME_TYPE start_time)
- {
- return GetTickCount() - start_time;
- }
- #elif defined(AIX)
- #define assert(a)
- long elapsed(struct timespec start)
- {
- struct timespec now, res;
- clock_gettime(CLOCK_REALTIME, &now);
- ntimersub(now, start, res);
- return (res.tv_sec)*1000L + (res.tv_nsec)/1000000L;
- }
- #else
- long elapsed(START_TIME_TYPE start_time)
- {
- struct timeval now, res;
- gettimeofday(&now, NULL);
- timersub(&now, &start_time, &res);
- return (res.tv_sec)*1000 + (res.tv_usec)/1000;
- }
- #endif
- #define assert(a, b, c, d) myassert(__FILE__, __LINE__, a, b, c, d)
- #define assert1(a, b, c, d, e) myassert(__FILE__, __LINE__, a, b, c, d, e)
- int tests = 0;
- int failures = 0;
- FILE* xml;
- START_TIME_TYPE global_start_time;
- char output[3000];
- char* cur_output = output;
- void write_test_result(void)
- {
- long duration = elapsed(global_start_time);
- fprintf(xml, " time=\"%ld.%.3ld\" >\n", duration / 1000, duration % 1000);
- if (cur_output != output)
- {
- fprintf(xml, "%s", output);
- cur_output = output;
- }
- fprintf(xml, "</testcase>\n");
- }
- void myassert(char* filename, int lineno, char* description, int value, char* format, ...)
- {
- ++tests;
- if (!value)
- {
- va_list args;
- ++failures;
- MyLog(LOGA_INFO, "Assertion failed, file %s, line %d, description: %s\n", filename, lineno, description);
- va_start(args, format);
- vprintf(format, args);
- va_end(args);
- cur_output += sprintf(cur_output, "<failure type=\"%s\">file %s, line %d </failure>\n",
- description, filename, lineno);
- }
- else
- MyLog(LOGA_DEBUG, "Assertion succeeded, file %s, line %d, description: %s", filename, lineno, description);
- }
- /*********************************************************************
- Test1: single-threaded client
- *********************************************************************/
- void test1_sendAndReceive(MQTTClient* c, int qos, char* test_topic)
- {
- MQTTClient_deliveryToken dt;
- MQTTClient_message pubmsg = MQTTClient_message_initializer;
- MQTTClient_message* m = NULL;
- char* topicName = NULL;
- int topicLen;
- int i = 0;
- int iterations = 50;
- int rc;
- MQTTResponse resp;
- MQTTProperty property;
- MyLog(LOGA_DEBUG, "%d messages at QoS %d", iterations, qos);
- pubmsg.payload = "a much longer message that we can shorten to the extent that we need to payload up to 11";
- pubmsg.payloadlen = 11;
- pubmsg.qos = qos;
- pubmsg.retained = 0;
- property.identifier = MQTTPROPERTY_CODE_USER_PROPERTY;
- property.value.data.data = "test user property";
- property.value.data.len = (int)strlen(property.value.data.data);
- property.value.value.data = "test user property value";
- property.value.value.len = (int)strlen(property.value.value.data);
- MQTTProperties_add(&pubmsg.properties, &property);
- for (i = 0; i < iterations; ++i)
- {
- if (i % 10 == 0)
- resp = MQTTClient_publish5(c, test_topic, pubmsg.payloadlen, pubmsg.payload, pubmsg.qos, pubmsg.retained,
- &pubmsg.properties, &dt);
- else
- resp = MQTTClient_publishMessage5(c, test_topic, &pubmsg, &dt);
- assert("Good rc from publish", resp.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", resp.reasonCode);
- if (qos > 0)
- {
- rc = MQTTClient_waitForCompletion(c, dt, 5000L);
- assert("Good rc from waitforCompletion", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- }
- rc = MQTTClient_receive(c, &topicName, &topicLen, &m, 5000);
- assert("Good rc from receive", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- if (topicName)
- {
- MyLog(LOGA_DEBUG, "Message received on topic %s is %.*s", topicName, m->payloadlen, (char*)(m->payload));
- if (pubmsg.payloadlen != m->payloadlen ||
- memcmp(m->payload, pubmsg.payload, m->payloadlen) != 0)
- {
- failures++;
- MyLog(LOGA_INFO, "Error: wrong data - received lengths %d %d", pubmsg.payloadlen, m->payloadlen);
- break;
- }
- assert("Property count should be > 0", m->properties.count > 0, "property count was %d", m->properties.count);
- MQTTClient_free(topicName);
- MQTTClient_freeMessage(&m);
- }
- else
- MyLog(LOGA_INFO, "No message received within timeout period\n");
- }
- /* receive any outstanding messages */
- MQTTClient_receive(c, &topicName, &topicLen, &m, 2000);
- while (topicName)
- {
- MyLog(LOGA_INFO, "Message received on topic %s is %.*s.\n", topicName, m->payloadlen, (char*)(m->payload));
- MQTTClient_free(topicName);
- MQTTClient_freeMessage(&m);
- MQTTClient_receive(c, &topicName, &topicLen, &m, 2000);
- }
- MQTTProperties_free(&pubmsg.properties);
- }
- void logProperties(MQTTProperties *props)
- {
- int i = 0;
- for (i = 0; i < props->count; ++i)
- {
- int id = props->array[i].identifier;
- const char* name = MQTTPropertyName(id);
- char* intformat = "Property name %s value %d";
- switch (MQTTProperty_getType(id))
- {
- case MQTTPROPERTY_TYPE_BYTE:
- MyLog(LOGA_INFO, intformat, name, props->array[i].value.byte);
- break;
- case MQTTPROPERTY_TYPE_TWO_BYTE_INTEGER:
- MyLog(LOGA_INFO, intformat, name, props->array[i].value.integer2);
- break;
- case MQTTPROPERTY_TYPE_FOUR_BYTE_INTEGER:
- MyLog(LOGA_INFO, intformat, name, props->array[i].value.integer4);
- break;
- case MQTTPROPERTY_TYPE_VARIABLE_BYTE_INTEGER:
- MyLog(LOGA_INFO, intformat, name, props->array[i].value.integer4);
- break;
- case MQTTPROPERTY_TYPE_BINARY_DATA:
- case MQTTPROPERTY_TYPE_UTF_8_ENCODED_STRING:
- MyLog(LOGA_INFO, "Property name %s value %.*s", name,
- props->array[i].value.data.len, props->array[i].value.data.data);
- break;
- case MQTTPROPERTY_TYPE_UTF_8_STRING_PAIR:
- MyLog(LOGA_INFO, "Property name %s key %.*s value %.*s", name,
- props->array[i].value.data.len, props->array[i].value.data.data,
- props->array[i].value.value.len, props->array[i].value.value.data);
- break;
- }
- }
- }
- int test1(struct Options options)
- {
- int subsqos = 2;
- MQTTClient c;
- MQTTClient_connectOptions opts = MQTTClient_connectOptions_initializer5;
- MQTTClient_willOptions wopts = MQTTClient_willOptions_initializer;
- MQTTProperties props = MQTTProperties_initializer;
- MQTTProperties willProps = MQTTProperties_initializer;
- MQTTProperty property;
- MQTTSubscribe_options subopts = MQTTSubscribe_options_initializer;
- MQTTResponse response = MQTTResponse_initializer;
- int rc = 0;
- char* test_topic = "C client test1";
- MQTTClient_createOptions createOpts = MQTTClient_createOptions_initializer;
- fprintf(xml, "<testcase classname=\"test1\" name=\"single threaded client using receive\"");
- global_start_time = start_clock();
- failures = 0;
- MyLog(LOGA_INFO, "Starting test 1 - single threaded client using receive");
- createOpts.MQTTVersion = MQTTVERSION_5;
- rc = MQTTClient_createWithOptions(&c, options.connection, "single_threaded_test",
- MQTTCLIENT_PERSISTENCE_DEFAULT, NULL, &createOpts);
- assert("good rc from create", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- if (rc != MQTTCLIENT_SUCCESS)
- {
- MQTTClient_destroy(&c);
- goto exit;
- }
- opts.keepAliveInterval = 20;
- opts.cleanstart = 1;
- opts.username = "testuser";
- opts.password = "testpassword";
- opts.MQTTVersion = options.MQTTVersion;
- printf("test MQTT version %d\n", options.MQTTVersion);
- if (options.haconnections != NULL)
- {
- opts.serverURIs = options.haconnections;
- opts.serverURIcount = options.hacount;
- }
- opts.will = &wopts;
- opts.will->message = "will message";
- opts.will->qos = 1;
- opts.will->retained = 0;
- opts.will->topicName = "will topic";
- opts.will = NULL;
- property.identifier = MQTTPROPERTY_CODE_SESSION_EXPIRY_INTERVAL;
- property.value.integer4 = 30;
- MQTTProperties_add(&props, &property);
- property.identifier = MQTTPROPERTY_CODE_USER_PROPERTY;
- property.value.data.data = "test user property";
- property.value.data.len = (int)strlen(property.value.data.data);
- property.value.value.data = "test user property value";
- property.value.value.len = (int)strlen(property.value.value.data);
- MQTTProperties_add(&props, &property);
- MyLog(LOGA_DEBUG, "Connecting");
- response = MQTTClient_connect5(c, &opts, &props, &willProps);
- assert("Good rc from connect", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", response.reasonCode);
- MQTTProperties_free(&props);
- MQTTProperties_free(&willProps);
- if (response.reasonCode != MQTTCLIENT_SUCCESS)
- goto exit;
- if (response.properties)
- {
- logProperties(response.properties);
- MQTTResponse_free(response);
- }
- subopts.retainAsPublished = 1;
- property.identifier = MQTTPROPERTY_CODE_SUBSCRIPTION_IDENTIFIER;
- property.value.integer4 = 33;
- MQTTProperties_add(&props, &property);
- response = MQTTClient_subscribe5(c, test_topic, subsqos, &subopts, &props);
- assert("Good rc from subscribe", response.reasonCode == subsqos, "rc was %d", response.reasonCode);
- MQTTProperties_free(&props);
- if (response.properties)
- {
- logProperties(response.properties);
- MQTTProperties_free(response.properties);
- }
- test1_sendAndReceive(c, 0, test_topic);
- test1_sendAndReceive(c, 1, test_topic);
- test1_sendAndReceive(c, 2, test_topic);
- MyLog(LOGA_DEBUG, "Stopping\n");
- MQTTProperties_free(&props);
- property.identifier = MQTTPROPERTY_CODE_USER_PROPERTY;
- property.value.data.data = "User property name";
- property.value.data.len = (int)strlen(property.value.data.data);
- property.value.value.data = "User property value";
- property.value.value.len = (int)strlen(property.value.value.data);
- MQTTProperties_add(&props, &property);
- response = MQTTClient_unsubscribe5(c, test_topic, &props);
- assert("Unsubscribe successful", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", response.reasonCode);
- MQTTResponse_free(response);
- MQTTProperties_free(&props);
- property.identifier = MQTTPROPERTY_CODE_SESSION_EXPIRY_INTERVAL;
- property.value.integer4 = 0;
- MQTTProperties_add(&props, &property);
- rc = MQTTClient_disconnect5(c, 0, MQTTREASONCODE_SUCCESS, &props);
- assert("Disconnect successful", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- /* Just to make sure we can connect again */
- response = MQTTClient_connect5(c, &opts, NULL, NULL);
- assert("Connect successful", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", response.reasonCode);
- MQTTResponse_free(response);
- rc = MQTTClient_disconnect5(c, 0, MQTTREASONCODE_SUCCESS, &props);
- assert("Disconnect successful", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- MQTTProperties_free(&props);
- MQTTClient_destroy(&c);
- exit:
- MyLog(LOGA_INFO, "TEST1: test %s. %d tests run, %d failures.",
- (failures == 0) ? "passed" : "failed", tests, failures);
- write_test_result();
- return failures;
- }
- /*********************************************************************
- Test2: multi-threaded client using callbacks
- *********************************************************************/
- volatile int test2_arrivedcount = 0;
- int test2_deliveryCompleted = 0;
- MQTTClient_message test2_pubmsg = MQTTClient_message_initializer;
- void test2_deliveryComplete(void* context, MQTTClient_deliveryToken dt)
- {
- ++test2_deliveryCompleted;
- }
- int test2_messageArrived(void* context, char* topicName, int topicLen, MQTTClient_message* message)
- {
- ++test2_arrivedcount;
- MyLog(LOGA_DEBUG, "Callback: %d message received on topic %s is %.*s.",
- test2_arrivedcount, topicName, message->payloadlen, (char*)(message->payload));
- assert("Message structure version should be 1", message->struct_version == 1,
- "message->struct_version was %d", message->struct_version);
- if (message->struct_version == 1)
- {
- const int props_count = 0;
- assert("Properties count should be 0", message->properties.count == props_count,
- "Properties count was %d\n", message->properties.count);
- logProperties(&message->properties);
- }
- if (test2_pubmsg.payloadlen != message->payloadlen ||
- memcmp(message->payload, test2_pubmsg.payload, message->payloadlen) != 0)
- {
- failures++;
- MyLog(LOGA_INFO, "Error: wrong data received lengths %d %d\n", test2_pubmsg.payloadlen, message->payloadlen);
- }
- MQTTClient_free(topicName);
- MQTTClient_freeMessage(&message);
- return 1;
- }
- void test2_sendAndReceive(MQTTClient* c, int qos, char* test_topic)
- {
- MQTTClient_deliveryToken dt;
- int i = 0;
- int iterations = 50;
- MQTTResponse response = MQTTResponse_initializer;
- int wait_seconds = 0;
- test2_deliveryCompleted = 0;
- MyLog(LOGA_INFO, "%d messages at QoS %d", iterations, qos);
- test2_pubmsg.payload = "a much longer message that we can shorten to the extent that we need to";
- test2_pubmsg.payloadlen = 27;
- test2_pubmsg.qos = qos;
- test2_pubmsg.retained = 0;
- for (i = 1; i <= iterations; ++i)
- {
- if (i % 10 == 0)
- response = MQTTClient_publish5(c, test_topic, test2_pubmsg.payloadlen, test2_pubmsg.payload,
- test2_pubmsg.qos, test2_pubmsg.retained, NULL, NULL);
- else
- response = MQTTClient_publishMessage5(c, test_topic, &test2_pubmsg, &dt);
- assert("Good rc from publish", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", response.reasonCode);
- #if defined(_WIN32)
- Sleep(100);
- #else
- usleep(100000L);
- #endif
- wait_seconds = 10;
- while ((test2_arrivedcount < i) && (wait_seconds-- > 0))
- {
- MyLog(LOGA_DEBUG, "Arrived %d count %d", test2_arrivedcount, i);
- #if defined(_WIN32)
- Sleep(1000);
- #else
- usleep(1000000L);
- #endif
- }
- assert("Message Arrived", wait_seconds > 0,
- "Time out waiting for message %d\n", i );
- }
- if (qos > 0)
- {
- /* MQ Telemetry can send a message to a subscriber before the server has
- completed the QoS 2 handshake with the publisher. For QoS 1 and 2,
- allow time for the final delivery complete callback before checking
- that all expected callbacks have been made */
- wait_seconds = 10;
- while ((test2_deliveryCompleted < iterations) && (wait_seconds-- > 0))
- {
- MyLog(LOGA_DEBUG, "Delivery Completed %d count %d", test2_deliveryCompleted, i);
- #if defined(_WIN32)
- Sleep(1000);
- #else
- usleep(1000000L);
- #endif
- }
- assert("All Deliveries Complete", wait_seconds > 0,
- "Number of deliveryCompleted callbacks was %d\n",
- test2_deliveryCompleted);
- }
- }
- int test2(struct Options options)
- {
- char* testname = "test2";
- int subsqos = 2;
- /* TODO - usused - remove ? MQTTClient_deliveryToken* dt = NULL; */
- MQTTClient c;
- MQTTClient_connectOptions opts = MQTTClient_connectOptions_initializer5;
- MQTTProperties props = MQTTProperties_initializer;
- MQTTProperties willProps = MQTTProperties_initializer;
- MQTTResponse response = MQTTResponse_initializer;
- MQTTSubscribe_options subopts = MQTTSubscribe_options_initializer;
- int rc = 0;
- char* test_topic = "C client test2";
- MQTTClient_createOptions createOpts = MQTTClient_createOptions_initializer;
- fprintf(xml, "<testcase classname=\"test1\" name=\"multi-threaded client using callbacks\"");
- MyLog(LOGA_INFO, "Starting test 2 - multi-threaded client using callbacks");
- global_start_time = start_clock();
- failures = 0;
- createOpts.MQTTVersion = MQTTVERSION_5;
- MQTTClient_createWithOptions(&c, options.connection, "multi_threaded_sample",
- MQTTCLIENT_PERSISTENCE_DEFAULT, NULL, &createOpts);
- opts.keepAliveInterval = 20;
- opts.cleanstart = 1;
- opts.MQTTVersion = options.MQTTVersion;
- opts.username = "testuser";
- opts.binarypwd.data = "testpassword";
- opts.binarypwd.len = (int)strlen(opts.binarypwd.data);
- if (options.haconnections != NULL)
- {
- opts.serverURIs = options.haconnections;
- opts.serverURIcount = options.hacount;
- }
- rc = MQTTClient_setCallbacks(c, NULL, NULL, test2_messageArrived, test2_deliveryComplete);
- assert("Good rc from setCallbacks", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- MyLog(LOGA_DEBUG, "Connecting");
- response = MQTTClient_connect5(c, &opts, &props, &willProps);
- assert("Good rc from connect", response.reasonCode == MQTTCLIENT_SUCCESS,
- "rc was %d", response.reasonCode);
- MQTTResponse_free(response);
- if (rc != MQTTCLIENT_SUCCESS)
- goto exit;
- response = MQTTClient_subscribe5(c, test_topic, subsqos, &subopts, &props);
- assert("Good rc from subscribe", response.reasonCode == subsqos, "rc was %d", rc);
- test2_sendAndReceive(c, 0, test_topic);
- test2_sendAndReceive(c, 1, test_topic);
- test2_sendAndReceive(c, 2, test_topic);
- MyLog(LOGA_DEBUG, "Stopping");
- response = MQTTClient_unsubscribe5(c, test_topic, &props);
- assert("Unsubscribe successful", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- rc = MQTTClient_disconnect5(c, 0, MQTTREASONCODE_SUCCESS, &props);
- assert("Disconnect successful", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- MQTTClient_destroy(&c);
- exit:
- MyLog(LOGA_INFO, "%s: test %s. %d tests run, %d failures.",
- (failures == 0) ? "passed" : "failed", testname, tests, failures);
- write_test_result();
- return failures;
- }
- /*********************************************************************
- Test 3: connack return codes
- for AMQTDD, needs an amqtdd.cfg of:
- allow_anonymous false
- password_file passwords
- and a passwords file of:
- Admin:Admin
- *********************************************************************/
- int test3(struct Options options)
- {
- char* testname = "test3";
- int rc;
- MQTTClient c;
- MQTTClient_connectOptions opts = MQTTClient_connectOptions_initializer5;
- MQTTClient_willOptions wopts = MQTTClient_willOptions_initializer;
- MQTTResponse response;
- MQTTClient_createOptions createOpts = MQTTClient_createOptions_initializer;
- fprintf(xml, "<testcase classname=\"test1\" name=\"connack return codes\"");
- global_start_time = start_clock();
- failures = 0;
- MyLog(LOGA_INFO, "Starting test 3 - connack return codes");
- createOpts.MQTTVersion = MQTTVERSION_5;
- #if 0
- /* clientid too long (RC = 2) */
- rc = MQTTClient_create(&c, options.connection, "client_ID_too_long_for_MQTT_protocol_version_3",
- MQTTCLIENT_PERSISTENCE_NONE, NULL);
- assert("good rc from create", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- rc = MQTTClient_connect(c, &opts);
- assert("identifier rejected", rc == 2, "rc was %d\n", rc);
- MQTTClient_destroy(&c);
- #endif
- /* broker unavailable (RC = 3) - TDD when allow_anonymous not set*/
- rc = MQTTClient_createWithOptions(&c, options.connection, "The C Client", MQTTCLIENT_PERSISTENCE_NONE,
- NULL, &createOpts);
- assert("good rc from create", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- #if 0
- rc = MQTTClient_connect(c, &opts);
- assert("broker unavailable", rc == 3, "rc was %d\n", rc);
- /* authentication failure (RC = 4) */
- opts.username = "Admin";
- opts.password = "fred";
- rc = MQTTClient_connect(c, &opts);
- assert("Bad user name or password", rc == 4, "rc was %d\n", rc);
- #endif
- /* authorization failure (RC = 5) */
- opts.username = "Admin";
- opts.password = "Admin";
- /*opts.will = &wopts; "Admin" not authorized to publish to Will topic by default
- opts.will->message = "will message";
- opts.will->qos = 1;
- opts.will->retained = 0;
- opts.will->topicName = "will topic";*/
- response = MQTTClient_connect5(c, &opts, NULL, NULL);
- //assert("Not authorized", rc == 5, "rc was %d\n", rc);
- MQTTResponse_free(response);
- #if 0
- /* successful connection (RC = 0) */
- opts.username = "Admin";
- opts.password = "Admin";
- opts.will = NULL;
- rc = MQTTClient_connect(c, &opts);
- assert("successful connection", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- MQTTClient_disconnect(c, 0);
- MQTTClient_destroy(&c);
- #endif
- /* TODO - unused - remove ? exit: */
- MyLog(LOGA_INFO, "%s: test %s. %d tests run, %d failures.",
- (failures == 0) ? "passed" : "failed", testname, tests, failures);
- write_test_result();
- return failures;
- }
- /*********************************************************************
- Test 4: client persistence 1
- *********************************************************************/
- int test4_run(int qos, int start_mqtt_version, int restore_mqtt_version)
- {
- char* testname = "test 4";
- char* topic = "Persistence test 1";
- int subsqos = 2;
- MQTTClient c;
- MQTTClient_connectOptions opts = MQTTClient_connectOptions_initializer5;
- MQTTClient_message* m = NULL;
- char* topicName = NULL;
- int topicLen;
- MQTTClient_deliveryToken* tokens = NULL;
- int mytoken = -99;
- char buffer[100];
- int count = 3;
- MQTTProperty property;
- MQTTProperties props = MQTTProperties_initializer;
- MQTTProperties pub_props = MQTTProperties_initializer;
- MQTTResponse response = MQTTResponse_initializer;
- MQTTClient_createOptions createOpts = MQTTClient_createOptions_initializer;
- int i, rc;
- failures = 0;
- MyLog(LOGA_INFO, "Starting test 4 - persistence, qos %d, MQTT versions: %s then %s", qos,
- (start_mqtt_version == MQTTVERSION_5) ? "5" : "3.1.1",
- (restore_mqtt_version == MQTTVERSION_5) ? "5" : "3.1.1");
- createOpts.MQTTVersion = start_mqtt_version;
- rc = MQTTClient_createWithOptions(&c, options.connection, "xrctest15_test_4",
- MQTTCLIENT_PERSISTENCE_DEFAULT, NULL, &createOpts);
- assert("Good rc from create", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- /* we might get some tokens back because they may not be cleaned up until
- * we connect cleanstart
- */
- rc = MQTTClient_getPendingDeliveryTokens(c, &tokens);
- assert("getPendingDeliveryTokens rc == 0", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- if (tokens)
- MQTTClient_free(tokens);
- opts.keepAliveInterval = 20;
- opts.reliable = 0;
- opts.MQTTVersion = start_mqtt_version;
- if (options.haconnections != NULL)
- {
- opts.serverURIs = options.haconnections;
- opts.serverURIcount = options.hacount;
- }
- if (start_mqtt_version == MQTTVERSION_5)
- {
- MyLog(LOGA_DEBUG, "Cleanup by connecting clean start, add session expiry > 0\n");
- opts.cleanstart = 1;
- property.identifier = MQTTPROPERTY_CODE_SESSION_EXPIRY_INTERVAL;
- property.value.integer4 = 30; /* in seconds */
- MQTTProperties_add(&props, &property);
- response = MQTTClient_connect5(c, &opts, &props, NULL);
- assert("Good rc from connect", response.reasonCode == MQTTCLIENT_SUCCESS,
- "rc was %d", response.reasonCode);
- MQTTResponse_free(response);
- if (response.reasonCode != MQTTCLIENT_SUCCESS)
- return -1;
- }
- else
- {
- MyLog(LOGA_DEBUG, "Cleanup by connecting clean session, then reconnecting non-cleansession\n");
- opts.cleanstart = 0; /* only applies to MQTT V5 */
- opts.cleansession = 1;
- rc = MQTTClient_connect(c, &opts);
- assert("Good rc from connect", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- rc = MQTTClient_disconnect(c, 1000);
- assert("Good rc from disconnect", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- opts.cleansession = 0;
- rc = MQTTClient_connect(c, &opts);
- assert("Good rc from connect", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- if (rc != MQTTCLIENT_SUCCESS)
- return -1;
- }
- /* subscribe so we can get messages back */
- if (start_mqtt_version == MQTTVERSION_5)
- {
- response = MQTTClient_subscribe5(c, topic, subsqos, NULL, NULL);
- assert("Good rc from subscribe", response.reasonCode == subsqos, "rc was %d", response.reasonCode);
- }
- else
- {
- rc = MQTTClient_subscribe(c, topic, subsqos);
- assert("Good rc from subscribe", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- }
- /* send messages so that we can receive the same ones */
- if (start_mqtt_version == MQTTVERSION_5)
- {
- property.identifier = MQTTPROPERTY_CODE_USER_PROPERTY;
- property.value.data.data = "test user property";
- property.value.data.len = (int)strlen(property.value.data.data);
- property.value.value.data = "test user property value";
- property.value.value.len = (int)strlen(property.value.value.data);
- MQTTProperties_add(&pub_props, &property);
- }
- for (i = 0; i < count; ++i)
- {
- sprintf(buffer, "Message sequence no %d", i);
- if (start_mqtt_version == MQTTVERSION_5)
- {
- response = MQTTClient_publish5(c, topic, 10, buffer, qos, 0, &pub_props, NULL);
- assert("Good rc from publish", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", response.reasonCode);
- }
- else
- {
- rc = MQTTClient_publish(c, topic, 10, buffer, qos, 0, NULL);
- assert("Good rc from publish", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- }
- }
- /* disconnect immediately without receiving the incoming messages */
- if (start_mqtt_version == MQTTVERSION_5)
- rc = MQTTClient_disconnect5(c, 0, MQTTREASONCODE_SUCCESS, NULL); /* now there should be "orphaned" publications */
- else
- rc = MQTTClient_disconnect(c, 0); /* now there should be "orphaned" publications */
- assert("Good rc from disconnect", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- rc = MQTTClient_getPendingDeliveryTokens(c, &tokens);
- assert("getPendingDeliveryTokens rc == 0", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- assert("should get some tokens back", tokens != NULL, "tokens was %p", tokens);
- if (tokens)
- {
- int i = 0;
- while (tokens[i] != -1)
- MyLog(LOGA_DEBUG, "Pending delivery token %d", tokens[i++]);
- assert1("no of tokens should be count", i == count, "no of tokens %d count %d", i, count);
- mytoken = tokens[0];
- MQTTClient_free(tokens);
- }
- MQTTProperties_free(&props);
- MQTTProperties_free(&pub_props);
- MQTTClient_destroy(&c); /* force re-reading persistence on create */
- createOpts.MQTTVersion = restore_mqtt_version;
- rc = MQTTClient_createWithOptions(&c, options.connection, "xrctest15_test_4",
- MQTTCLIENT_PERSISTENCE_DEFAULT, NULL, &createOpts);
- if (start_mqtt_version == MQTTVERSION_5 && restore_mqtt_version == MQTTVERSION_3_1_1)
- {
- assert("Persistence error from create", rc == MQTTCLIENT_PERSISTENCE_ERROR, "rc was %d", rc);
- goto exit;
- }
- else
- assert("Good rc from create", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- rc = MQTTClient_getPendingDeliveryTokens(c, &tokens);
- assert("getPendingDeliveryTokens rc == 0", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- assert("should get some tokens back", tokens != NULL, "tokens was %p", tokens);
- if (tokens)
- {
- int i = 0;
- while (tokens[i] != -1)
- MyLog(LOGA_DEBUG, "Pending delivery token %d", tokens[i++]);
- MQTTClient_free(tokens);
- assert1("no of tokens should be count", i == count, "no of tokens %d count %d", i, count);
- }
- MyLog(LOGA_DEBUG, "Reconnecting");
- opts.cleanstart = 0;
- response = MQTTClient_connect5(c, &opts, NULL, NULL);
- assert("Good rc from connect", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", response.reasonCode);
- MQTTResponse_free(response);
- if (response.reasonCode != MQTTCLIENT_SUCCESS)
- return -1;
- for (i = 0; i < count; ++i)
- {
- int dup = 0;
- do
- {
- dup = 0;
- MQTTClient_receive(c, &topicName, &topicLen, &m, 5000);
- if (m && m->dup)
- {
- assert("No duplicates should be received for qos 2", qos == 1, "qos is %d", qos);
- MyLog(LOGA_DEBUG, "Duplicate message id %d", m->msgid);
- assert("properties are received", m->properties.count > 0, "property count is %d",
- m->properties.count);
- logProperties(&m->properties);
- MQTTClient_freeMessage(&m);
- MQTTClient_free(topicName);
- dup = 1;
- }
- } while (dup == 1);
- assert("should get a message", m != NULL, "m was %p", m);
- if (m)
- {
- MyLog(LOGA_DEBUG, "Received message id %d", m->msgid);
- assert("topicName is correct", strcmp(topicName, topic) == 0, "topicName is %s", topicName);
- if (start_mqtt_version == MQTTVERSION_5)
- assert("properties are received", m->properties.count > 0, "property count is %d",
- m->properties.count);
- else
- assert("properties are not received", m->properties.count == 0, "property count is %d",
- m->properties.count);
- logProperties(&m->properties);
- MQTTClient_freeMessage(&m);
- MQTTClient_free(topicName);
- }
- }
- MQTTClient_yield(); /* allow any unfinished protocol exchanges to finish */
- rc = MQTTClient_getPendingDeliveryTokens(c, &tokens);
- assert("getPendingDeliveryTokens rc == 0", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- assert("should get no tokens back", tokens == NULL, "tokens was %p", tokens);
- rc = MQTTClient_disconnect5(c, 0, MQTTREASONCODE_SUCCESS, NULL);
- MQTTClient_destroy(&c);
- exit:
- MyLog(LOGA_INFO, "%s: test %s. %d tests run, %d failures.",
- (failures == 0) ? "passed" : "failed", testname, tests, failures);
- return failures;
- }
- int test4(struct Options options)
- {
- int rc = 0;
- fprintf(xml, "<testcase classname=\"test4\" name=\"persistence\"");
- global_start_time = start_clock();
- rc = test4_run(1, MQTTVERSION_5, MQTTVERSION_5) +
- test4_run(2, MQTTVERSION_5, MQTTVERSION_5) +
- test4_run(2, MQTTVERSION_3_1_1, MQTTVERSION_5) +
- test4_run(2, MQTTVERSION_5, MQTTVERSION_3_1_1);
- fprintf(xml, " time=\"%ld\" >\n", elapsed(global_start_time) / 1000);
- if (cur_output != output)
- {
- fprintf(xml, "%s", output);
- cur_output = output;
- }
- fprintf(xml, "</testcase>\n");
- return rc;
- }
- /*********************************************************************
- Test 5: disconnect with quiesce timeout should allow exchanges to complete
- *********************************************************************/
- int test5(struct Options options)
- {
- char* testname = "test 5";
- char* topic = "Persistence test 2";
- int subsqos = 2;
- MQTTClient c;
- MQTTClient_connectOptions opts = MQTTClient_connectOptions_initializer5;
- MQTTClient_deliveryToken* tokens = NULL;
- char buffer[100];
- int count = 5;
- MQTTProperty property;
- MQTTProperties props = MQTTProperties_initializer;
- MQTTResponse response = MQTTResponse_initializer;
- int i, rc;
- MQTTClient_createOptions createOpts = MQTTClient_createOptions_initializer;
- fprintf(xml, "<testcase classname=\"test1\" name=\"disconnect with quiesce timeout should allow exchanges to complete\"");
- global_start_time = start_clock();
- failures = 0;
- MyLog(LOGA_INFO, "Starting test 5 - disconnect with quiesce timeout should allow exchanges to complete");
- createOpts.MQTTVersion = MQTTVERSION_5;
- MQTTClient_createWithOptions(&c, options.connection, "xrctest15_test_5", MQTTCLIENT_PERSISTENCE_DEFAULT,
- NULL, &createOpts);
- opts.keepAliveInterval = 20;
- opts.cleanstart = 1;
- opts.reliable = 0;
- opts.MQTTVersion = options.MQTTVersion;
- if (options.haconnections != NULL)
- {
- opts.serverURIs = options.haconnections;
- opts.serverURIcount = options.hacount;
- }
- property.identifier = MQTTPROPERTY_CODE_SESSION_EXPIRY_INTERVAL;
- property.value.integer4 = 30;
- MQTTProperties_add(&props, &property);
- MyLog(LOGA_DEBUG, "Connecting");
- response = MQTTClient_connect5(c, &opts, &props, NULL);
- assert("Good rc from connect", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", response.reasonCode);
- MQTTResponse_free(response);
- if (response.reasonCode != MQTTCLIENT_SUCCESS)
- {
- MQTTClient_destroy(&c);
- goto exit;
- }
- response = MQTTClient_subscribe5(c, topic, subsqos, NULL, NULL);
- assert("Good rc from subscribe", response.reasonCode == subsqos, "rc was %d", response.reasonCode);
- for (i = 0; i < count; ++i)
- {
- sprintf(buffer, "Message sequence no %d", i);
- response = MQTTClient_publish5(c, topic, 10, buffer, 1, 0, NULL, NULL);
- assert("Good rc from publish", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", response.reasonCode);
- }
- MQTTClient_disconnect5(c, 1000, MQTTREASONCODE_SUCCESS, NULL); /* now there should be no "orphaned" publications */
- MyLog(LOGA_DEBUG, "Disconnected");
- rc = MQTTClient_getPendingDeliveryTokens(c, &tokens);
- assert("getPendingDeliveryTokens rc == 0", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- assert("should get no tokens back", tokens == NULL, "tokens was %p", tokens);
- MQTTProperties_free(&props);
- MQTTClient_destroy(&c);
- exit:
- MyLog(LOGA_INFO, "%s: test %s. %d tests run, %d failures.",
- (failures == 0) ? "passed" : "failed", testname, tests, failures);
- write_test_result();
- return failures;
- }
- /*********************************************************************
- Test 6: connectionLost and will message
- *********************************************************************/
- MQTTClient test6_c1, test6_c2;
- volatile int test6_will_message_arrived = 0;
- volatile int test6_connection_lost_called = 0;
- void test6_connectionLost(void* context, char* cause)
- {
- MQTTClient c = (MQTTClient)context;
- MyLog(LOGA_INFO, "%s -> Callback: connection lost", (c == test6_c1) ? "Client-1" : "Client-2");
- test6_connection_lost_called = 1;
- }
- void test6_deliveryComplete(void* context, MQTTClient_deliveryToken token)
- {
- MyLog(LOGA_DEBUG, "Client-2 -> Callback: publish complete for token %d", token);
- }
- char* test6_will_topic = "C Test 2: will topic";
- char* test6_will_message = "will message from Client-1";
- int test6_messageArrived(void* context, char* topicName, int topicLen, MQTTClient_message* m)
- {
- MQTTClient c = (MQTTClient)context;
- MyLog(LOGA_INFO, "%s -> Callback: message received on topic '%s' is '%.*s'",
- (c == test6_c1) ? "Client-1" : "Client-2", topicName, m->payloadlen, (char*)(m->payload));
- if (c == test6_c2 && strcmp(topicName, test6_will_topic) == 0 && memcmp(m->payload, test6_will_message, m->payloadlen) == 0)
- test6_will_message_arrived = 1;
- MQTTClient_free(topicName);
- MQTTClient_freeMessage(&m);
- return 1;
- }
- int test6(struct Options options)
- {
- char* testname = "test6";
- MQTTClient_connectOptions opts = MQTTClient_connectOptions_initializer5;
- MQTTClient_willOptions wopts = MQTTClient_willOptions_initializer;
- MQTTClient_connectOptions opts2 = MQTTClient_connectOptions_initializer5;
- MQTTResponse response = MQTTResponse_initializer;
- int rc, count;
- char* mqttsas_topic = "MQTTSAS topic";
- MQTTClient_createOptions createOpts = MQTTClient_createOptions_initializer;
- failures = 0;
- MyLog(LOGA_INFO, "Starting test 6 - connectionLost and will messages");
- fprintf(xml, "<testcase classname=\"test1\" name=\"connectionLost and will messages\"");
- global_start_time = start_clock();
- opts.keepAliveInterval = 2;
- opts.cleanstart = 1;
- opts.MQTTVersion = options.MQTTVersion;
- opts.will = &wopts;
- opts.will->message = test6_will_message;
- opts.will->qos = 1;
- opts.will->retained = 0;
- opts.will->topicName = test6_will_topic;
- if (options.haconnections != NULL)
- {
- opts.serverURIs = options.haconnections;
- opts.serverURIcount = options.hacount;
- }
- /* Client-1 with Will options */
- createOpts.MQTTVersion = MQTTVERSION_5;
- rc = MQTTClient_createWithOptions(&test6_c1, options.proxy_connection, "Client_1", MQTTCLIENT_PERSISTENCE_DEFAULT,
- NULL, &createOpts);
- assert("good rc from create", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- if (rc != MQTTCLIENT_SUCCESS)
- goto exit;
- rc = MQTTClient_setCallbacks(test6_c1, (void*)test6_c1, test6_connectionLost, test6_messageArrived, test6_deliveryComplete);
- assert("good rc from setCallbacks", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- if (rc != MQTTCLIENT_SUCCESS)
- goto exit;
- /* Connect to the broker */
- response = MQTTClient_connect5(test6_c1, &opts, NULL, NULL);
- assert("good rc from connect", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- MQTTResponse_free(response);
- if (rc != MQTTCLIENT_SUCCESS)
- goto exit;
- /* Client - 2 (multi-threaded) */
- rc = MQTTClient_createWithOptions(&test6_c2, options.connection, "Client_2", MQTTCLIENT_PERSISTENCE_DEFAULT,
- NULL, &createOpts);
- assert("good rc from create", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- /* Set the callback functions for the client */
- rc = MQTTClient_setCallbacks(test6_c2, (void*)test6_c2, test6_connectionLost, test6_messageArrived, test6_deliveryComplete);
- assert("good rc from setCallbacks", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- /* Connect to the broker */
- opts2.keepAliveInterval = 20;
- opts2.cleanstart = 1;
- MyLog(LOGA_INFO, "Connecting Client_2 ...");
- response = MQTTClient_connect5(test6_c2, &opts2, NULL, NULL);
- MQTTResponse_free(response);
- assert("Good rc from connect", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d\n", response.reasonCode);
- response = MQTTClient_subscribe5(test6_c2, test6_will_topic, 2, NULL, NULL);
- assert("Good rc from subscribe", response.reasonCode == MQTTREASONCODE_GRANTED_QOS_2, "rc was %d\n", response.reasonCode);
- /* now send the command which will break the connection and cause the will message to be sent */
- response = MQTTClient_publish5(test6_c1, mqttsas_topic, (int)strlen("TERMINATE"), "TERMINATE", 0, 0, NULL, NULL);
- assert("Good rc from publish", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d\n", response.reasonCode);
- MyLog(LOGA_INFO, "Waiting to receive the will message");
- count = 0;
- while (++count < 40)
- {
- #if defined(_WIN32)
- Sleep(1000L);
- #else
- sleep(1);
- #endif
- if (test6_will_message_arrived == 1 && test6_connection_lost_called == 1)
- break;
- }
- assert("will message arrived", test6_will_message_arrived == 1,
- "will_message_arrived was %d\n", test6_will_message_arrived);
- assert("connection lost called", test6_connection_lost_called == 1,
- "connection_lost_called %d\n", test6_connection_lost_called);
- response = MQTTClient_unsubscribe5(test6_c2, test6_will_topic, NULL);
- assert("Good rc from unsubscribe", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", response.reasonCode);
- rc = MQTTClient_isConnected(test6_c2);
- assert("Client-2 still connected", rc == 1, "isconnected is %d", rc);
- rc = MQTTClient_isConnected(test6_c1);
- assert("Client-1 not connected", rc == 0, "isconnected is %d", rc);
- rc = MQTTClient_disconnect5(test6_c2, 100L, MQTTREASONCODE_SUCCESS, NULL);
- assert("Good rc from disconnect", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- MQTTClient_destroy(&test6_c1);
- MQTTClient_destroy(&test6_c2);
- exit:
- MyLog(LOGA_INFO, "%s: test %s. %d tests run, %d failures.\n",
- (failures == 0) ? "passed" : "failed", testname, tests, failures);
- write_test_result();
- return failures;
- }
- int test6a(struct Options options)
- {
- char* testname = "test6a";
- MQTTClient_connectOptions opts = MQTTClient_connectOptions_initializer5;
- MQTTClient_willOptions wopts = MQTTClient_willOptions_initializer;
- MQTTClient_connectOptions opts2 = MQTTClient_connectOptions_initializer5;
- int rc, count;
- MQTTResponse response = MQTTResponse_initializer;
- char* mqttsas_topic = "MQTTSAS topic";
- MQTTClient_createOptions createOpts = MQTTClient_createOptions_initializer;
- failures = 0;
- MyLog(LOGA_INFO, "Starting test 6 - connectionLost and binary will messages");
- fprintf(xml, "<testcase classname=\"test1\" name=\"connectionLost and binary will messages\"");
- global_start_time = start_clock();
- opts.keepAliveInterval = 2;
- opts.cleanstart = 1;
- opts.MQTTVersion = options.MQTTVersion;
- opts.will = &wopts;
- opts.will->payload.data = test6_will_message;
- opts.will->payload.len = (int)strlen(test6_will_message) + 1;
- opts.will->qos = 1;
- opts.will->retained = 0;
- opts.will->topicName = test6_will_topic;
- if (options.haconnections != NULL)
- {
- opts.serverURIs = options.haconnections;
- opts.serverURIcount = options.hacount;
- }
- /* Client-1 with Will options */
- createOpts.MQTTVersion = MQTTVERSION_5;
- rc = MQTTClient_createWithOptions(&test6_c1, options.proxy_connection, "Client_1", MQTTCLIENT_PERSISTENCE_DEFAULT,
- NULL, &createOpts);
- assert("good rc from create", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- if (rc != MQTTCLIENT_SUCCESS)
- goto exit;
- rc = MQTTClient_setCallbacks(test6_c1, (void*)test6_c1, test6_connectionLost, test6_messageArrived, test6_deliveryComplete);
- assert("good rc from setCallbacks", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- if (rc != MQTTCLIENT_SUCCESS)
- goto exit;
- /* Connect to the broker */
- response = MQTTClient_connect5(test6_c1, &opts, NULL, NULL);
- assert("good rc from connect", response.reasonCode == MQTTCLIENT_SUCCESS,
- "rc was %d\n", response.reasonCode);
- MQTTResponse_free(response);
- if (response.reasonCode != MQTTCLIENT_SUCCESS)
- goto exit;
- /* Client - 2 (multi-threaded) */
- rc = MQTTClient_createWithOptions(&test6_c2, options.connection, "Client_2", MQTTCLIENT_PERSISTENCE_DEFAULT,
- NULL, &createOpts);
- assert("good rc from create", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- /* Set the callback functions for the client */
- rc = MQTTClient_setCallbacks(test6_c2, (void*)test6_c2, test6_connectionLost, test6_messageArrived, test6_deliveryComplete);
- assert("good rc from setCallbacks", rc == MQTTCLIENT_SUCCESS, "rc was %d\n", rc);
- /* Connect to the broker */
- opts2.keepAliveInterval = 20;
- opts2.cleanstart = 1;
- MyLog(LOGA_INFO, "Connecting Client_2 ...");
- response = MQTTClient_connect5(test6_c2, &opts2, NULL, NULL);
- MQTTResponse_free(response);
- assert("Good rc from connect", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d\n", response.reasonCode);
- response = MQTTClient_subscribe5(test6_c2, test6_will_topic, 2, NULL, NULL);
- assert("Good rc from subscribe", response.reasonCode == MQTTREASONCODE_GRANTED_QOS_2, "rc was %d\n", response.reasonCode);
- /* now send the command which will break the connection and cause the will message to be sent */
- response = MQTTClient_publish5(test6_c1, mqttsas_topic, (int)strlen("TERMINATE"), "TERMINATE", 0, 0, NULL, NULL);
- assert("Good rc from publish", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d\n", response.reasonCode);
- MyLog(LOGA_INFO, "Waiting to receive the will message");
- count = 0;
- while (++count < 40)
- {
- #if defined(_WIN32)
- Sleep(1000L);
- #else
- sleep(1);
- #endif
- if (test6_will_message_arrived == 1 && test6_connection_lost_called == 1)
- break;
- }
- assert("will message arrived", test6_will_message_arrived == 1,
- "will_message_arrived was %d\n", test6_will_message_arrived);
- assert("connection lost called", test6_connection_lost_called == 1,
- "connection_lost_called %d\n", test6_connection_lost_called);
- response = MQTTClient_unsubscribe5(test6_c2, test6_will_topic, NULL);
- assert("Good rc from unsubscribe", response.reasonCode == MQTTCLIENT_SUCCESS, "rc was %d", response.reasonCode);
- rc = MQTTClient_isConnected(test6_c2);
- assert("Client-2 still connected", rc == 1, "isconnected is %d", rc);
- rc = MQTTClient_isConnected(test6_c1);
- assert("Client-1 not connected", rc == 0, "isconnected is %d", rc);
- rc = MQTTClient_disconnect5(test6_c2, 100L, MQTTREASONCODE_SUCCESS, NULL);
- assert("Good rc from disconnect", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- MQTTClient_destroy(&test6_c1);
- MQTTClient_destroy(&test6_c2);
- exit:
- MyLog(LOGA_INFO, "%s: test %s. %d tests run, %d failures.\n",
- (failures == 0) ? "passed" : "failed", testname, tests, failures);
- write_test_result();
- return failures;
- }
- int main(int argc, char** argv)
- {
- int rc = 0;
- int (*tests[])() = {NULL, test1, test2, test3, test4, test5, test6, test6a};
- int i;
- xml = fopen("TEST-test1.xml", "w");
- fprintf(xml, "<testsuite name=\"test1\" tests=\"%d\">\n", (int)(ARRAY_SIZE(tests) - 1));
- setenv("MQTT_C_CLIENT_TRACE", "ON", 1);
- setenv("MQTT_C_CLIENT_TRACE_LEVEL", "ERROR", 1);
- getopts(argc, argv);
- for (i = 0; i < options.iterations; ++i)
- {
- if (options.test_no == 0)
- { /* run all the tests */
- for (options.test_no = 1; options.test_no < ARRAY_SIZE(tests); ++options.test_no)
- rc += tests[options.test_no](options); /* return number of failures. 0 = test succeeded */
- }
- else
- rc = tests[options.test_no](options); /* run just the selected test */
- }
- if (rc == 0)
- MyLog(LOGA_INFO, "verdict pass");
- else
- MyLog(LOGA_INFO, "verdict fail");
- fprintf(xml, "</testsuite>\n");
- fclose(xml);
- return rc;
- }
|