123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449 |
- /*******************************************************************************
- * Copyright (c) 2009, 2022 IBM Corp, Ian Craggs
- *
- * 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 / 1000L);
- #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);
- }
- }
- /* call yield a few times until unfinished protocol exchanges are finished */
- count = 0;
- do
- {
- MQTTClient_yield();
- rc = MQTTClient_getPendingDeliveryTokens(c, &tokens);
- assert("getPendingDeliveryTokens rc == 0", rc == MQTTCLIENT_SUCCESS, "rc was %d", rc);
- } while (tokens != NULL && ++count < 10);
- 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;
- }
|