ext/paho.mqtt.c/src/samples/paho_c_sub.c - a71ch-crypto-support - Git at Google

 /*******************************************************************************
  * Copyright (c) 2012, 2019 IBM Corp., and others
  *
  * 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 contribution
  *    Ian Craggs - fix for bug 413429 - connectionLost not called
  *    Guilherme Maciel Ferreira - add keep alive option
  *    Ian Craggs - add full capability
  *******************************************************************************/

 #include "MQTTAsync.h"
 #include "MQTTClientPersistence.h"
 #include "pubsub_opts.h"

 #include <stdio.h>
 #include <signal.h>
 #include <string.h>
 #include <stdlib.h>


 #if defined(_WIN32)
 #include <windows.h>
 #define sleep Sleep
 #else
 #include <sys/time.h>
 #include <unistd.h>
 #endif

 #if defined(_WRS_KERNEL)
 #include <OsWrapper.h>
 #endif

 volatile int finished = 0;
 int subscribed = 0;
 int disconnected = 0;


 void mysleep(int ms)
 {
 	#if defined(_WIN32)
 		Sleep(ms);
 	#else
 		usleep(ms * 1000);
 	#endif
 }

 void cfinish(int sig)
 {
 	signal(SIGINT, NULL);
 	finished = 1;
 }


 struct pubsub_opts opts =
 {
 	0, 0, 0, 0, "\n", 100,  	/* debug/app options */
 	NULL, NULL, 1, 0, 0, /* message options */
 	MQTTVERSION_DEFAULT, NULL, "paho-c-sub", 0, 0, NULL, NULL, "localhost", "1883", NULL, 10, /* MQTT options */
 	NULL, NULL, 0, 0, /* will options */
 	0, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, /* TLS options */
 	0, {NULL, NULL}, /* MQTT V5 options */
 };


 int messageArrived(void *context, char *topicName, int topicLen, MQTTAsync_message *message)
 {
 	size_t delimlen = 0;

 	if (opts.verbose)
 		printf("%d %s\t", message->payloadlen, topicName);
 	if (opts.delimiter)
 		delimlen = strlen(opts.delimiter);
 	if (opts.delimiter == NULL || (message->payloadlen > delimlen &&
 		strncmp(opts.delimiter, &((char*)message->payload)[message->payloadlen - delimlen], delimlen) == 0))
 		printf("%.*s", message->payloadlen, (char*)message->payload);
 	else
 		printf("%.*s%s", message->payloadlen, (char*)message->payload, opts.delimiter);
 	if (message->struct_version == 1 && opts.verbose)
 		logProperties(&message->properties);
 	fflush(stdout);
 	MQTTAsync_freeMessage(&message);
 	MQTTAsync_free(topicName);
 	return 1;
 }


 void onDisconnect(void* context, MQTTAsync_successData* response)
 {
 	disconnected = 1;
 }


 void onSubscribe5(void* context, MQTTAsync_successData5* response)
 {
 	subscribed = 1;
 }

 void onSubscribe(void* context, MQTTAsync_successData* response)
 {
 	subscribed = 1;
 }


 void onSubscribeFailure5(void* context, MQTTAsync_failureData5* response)
 {
 	if (!opts.quiet)
 		fprintf(stderr, "Subscribe failed, rc %s reason code %s\n",
 				MQTTAsync_strerror(response->code),
 				MQTTReasonCode_toString(response->reasonCode));
 	finished = 1;
 }


 void onSubscribeFailure(void* context, MQTTAsync_failureData* response)
 {
 	if (!opts.quiet)
 		fprintf(stderr, "Subscribe failed, rc %s\n",
 			MQTTAsync_strerror(response->code));
 	finished = 1;
 }


 void onConnectFailure5(void* context, MQTTAsync_failureData5* response)
 {
 	if (!opts.quiet)
 		fprintf(stderr, "Connect failed, rc %s reason code %s\n",
 			MQTTAsync_strerror(response->code),
 			MQTTReasonCode_toString(response->reasonCode));
 	finished = 1;
 }


 void onConnectFailure(void* context, MQTTAsync_failureData* response)
 {
 	if (!opts.quiet)
 		fprintf(stderr, "Connect failed, rc %s\n", response ? MQTTAsync_strerror(response->code) : "none");
 	finished = 1;
 }


 void onConnect5(void* context, MQTTAsync_successData5* response)
 {
 	MQTTAsync client = (MQTTAsync)context;
 	MQTTAsync_callOptions copts = MQTTAsync_callOptions_initializer;
 	int rc;

 	if (opts.verbose)
 		printf("Subscribing to topic %s with client %s at QoS %d\n", opts.topic, opts.clientid, opts.qos);

 	copts.onSuccess5 = onSubscribe5;
 	copts.onFailure5 = onSubscribeFailure5;
 	copts.context = client;
 	if ((rc = MQTTAsync_subscribe(client, opts.topic, opts.qos, &copts)) != MQTTASYNC_SUCCESS)
 	{
 		if (!opts.quiet)
 			fprintf(stderr, "Failed to start subscribe, return code %s\n", MQTTAsync_strerror(rc));
 		finished = 1;
 	}
 }


 void onConnect(void* context, MQTTAsync_successData* response)
 {
 	MQTTAsync client = (MQTTAsync)context;
 	MQTTAsync_responseOptions ropts = MQTTAsync_responseOptions_initializer;
 	int rc;

 	if (opts.verbose)
 		printf("Subscribing to topic %s with client %s at QoS %d\n", opts.topic, opts.clientid, opts.qos);

 	ropts.onSuccess = onSubscribe;
 	ropts.onFailure = onSubscribeFailure;
 	ropts.context = client;
 	if ((rc = MQTTAsync_subscribe(client, opts.topic, opts.qos, &ropts)) != MQTTASYNC_SUCCESS)
 	{
 		if (!opts.quiet)
 			fprintf(stderr, "Failed to start subscribe, return code %s\n", MQTTAsync_strerror(rc));
 		finished = 1;
 	}
 }

 MQTTAsync_connectOptions conn_opts = MQTTAsync_connectOptions_initializer;


 void trace_callback(enum MQTTASYNC_TRACE_LEVELS level, char* message)
 {
 	fprintf(stderr, "Trace : %d, %s\n", level, message);
 }


 int main(int argc, char** argv)
 {
 	MQTTAsync client;
 	MQTTAsync_disconnectOptions disc_opts = MQTTAsync_disconnectOptions_initializer;
 	MQTTAsync_createOptions create_opts = MQTTAsync_createOptions_initializer;
 	MQTTAsync_willOptions will_opts = MQTTAsync_willOptions_initializer;
 	MQTTAsync_SSLOptions ssl_opts = MQTTAsync_SSLOptions_initializer;
 	int rc = 0;
 	char* url = NULL;
 	const char* version = NULL;
 	const char* program_name = "paho_c_sub";
 	MQTTAsync_nameValue* infos = MQTTAsync_getVersionInfo();
 #if !defined(_WIN32)
     struct sigaction sa;
 #endif

 	if (argc < 2)
 		usage(&opts, (pubsub_opts_nameValue*)infos, program_name);

 	if (getopts(argc, argv, &opts) != 0)
 		usage(&opts, (pubsub_opts_nameValue*)infos, program_name);

 	if (strchr(opts.topic, '#') || strchr(opts.topic, '+'))
 		opts.verbose = 1;

 	if (opts.connection)
 		url = opts.connection;
 	else
 	{
 		url = malloc(100);
 		sprintf(url, "%s:%s", opts.host, opts.port);
 	}
 	if (opts.verbose)
 		printf("URL is %s\n", url);

 	if (opts.tracelevel > 0)
 	{
 		MQTTAsync_setTraceCallback(trace_callback);
 		MQTTAsync_setTraceLevel(opts.tracelevel);
 	}

 	if (opts.MQTTVersion >= MQTTVERSION_5)
 		create_opts.MQTTVersion = MQTTVERSION_5;
 	rc = MQTTAsync_createWithOptions(&client, url, opts.clientid, MQTTCLIENT_PERSISTENCE_NONE,
 			NULL, &create_opts);
 	if (rc != MQTTASYNC_SUCCESS)
 	{
 		if (!opts.quiet)
 			fprintf(stderr, "Failed to create client, return code: %s\n", MQTTAsync_strerror(rc));
 		exit(EXIT_FAILURE);
 	}

 	rc = MQTTAsync_setCallbacks(client, client, NULL, messageArrived, NULL);
 	if (rc != MQTTASYNC_SUCCESS)
 	{
 		if (!opts.quiet)
 			fprintf(stderr, "Failed to set callbacks, return code: %s\n", MQTTAsync_strerror(rc));
 		exit(EXIT_FAILURE);
 	}

 #if defined(_WIN32)
 	signal(SIGINT, cfinish);
 	signal(SIGTERM, cfinish);
 #else
     memset(&sa, 0, sizeof(struct sigaction));
     sa.sa_handler = cfinish;
     sa.sa_flags = 0;

     sigaction(SIGINT, &sa, NULL);
     sigaction(SIGTERM, &sa, NULL);
 #endif

 	if (opts.MQTTVersion == MQTTVERSION_5)
 	{
 		MQTTAsync_connectOptions conn_opts5 = MQTTAsync_connectOptions_initializer5;
 		conn_opts = conn_opts5;
 		conn_opts.onSuccess5 = onConnect5;
 		conn_opts.onFailure5 = onConnectFailure5;
 		conn_opts.cleanstart = 1;
 	}
 	else
 	{
 		conn_opts.onSuccess = onConnect;
 		conn_opts.onFailure = onConnectFailure;
 		conn_opts.cleansession = 1;
 	}
 	conn_opts.keepAliveInterval = opts.keepalive;
 	conn_opts.username = opts.username;
 	conn_opts.password = opts.password;
 	conn_opts.MQTTVersion = opts.MQTTVersion;
 	conn_opts.context = client;
 	conn_opts.automaticReconnect = 1;

 	if (opts.will_topic) 	/* will options */
 	{
 		will_opts.message = opts.will_payload;
 		will_opts.topicName = opts.will_topic;
 		will_opts.qos = opts.will_qos;
 		will_opts.retained = opts.will_retain;
 		conn_opts.will = &will_opts;
 	}

 	if (opts.connection && (strncmp(opts.connection, "ssl://", 6) == 0 ||
 			strncmp(opts.connection, "wss://", 6) == 0))
 	{
 		ssl_opts.verify = (opts.insecure) ? 0 : 1;
 		ssl_opts.CApath = opts.capath;
 		ssl_opts.keyStore = opts.cert;
 		ssl_opts.trustStore = opts.cafile;
 		ssl_opts.privateKey = opts.key;
 		ssl_opts.privateKeyPassword = opts.keypass;
 		ssl_opts.enabledCipherSuites = opts.ciphers;
 		conn_opts.ssl = &ssl_opts;
 	}

 	if ((rc = MQTTAsync_connect(client, &conn_opts)) != MQTTASYNC_SUCCESS)
 	{
 		if (!opts.quiet)
 			fprintf(stderr, "Failed to start connect, return code %s\n", MQTTAsync_strerror(rc));
 		exit(EXIT_FAILURE);
 	}

 	while (!subscribed)
 		mysleep(100);

 	if (finished)
 		goto exit;

 	while (!finished)
 		mysleep(100);

 	disc_opts.onSuccess = onDisconnect;
 	if ((rc = MQTTAsync_disconnect(client, &disc_opts)) != MQTTASYNC_SUCCESS)
 	{
 		if (!opts.quiet)
 			fprintf(stderr, "Failed to start disconnect, return code: %s\n", MQTTAsync_strerror(rc));
 		exit(EXIT_FAILURE);
 	}

 	while (!disconnected)
 		mysleep(100);

 exit:
 	MQTTAsync_destroy(&client);

 	return EXIT_SUCCESS;
 }
	/*******************************************************************************
	* Copyright (c) 2012, 2019 IBM Corp., and others
	*
	* 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 contribution
	* Ian Craggs - fix for bug 413429 - connectionLost not called
	* Guilherme Maciel Ferreira - add keep alive option
	* Ian Craggs - add full capability
	*******************************************************************************/

	#include "MQTTAsync.h"
	#include "MQTTClientPersistence.h"
	#include "pubsub_opts.h"

	#include <stdio.h>
	#include <signal.h>
	#include <string.h>
	#include <stdlib.h>


	#if defined(_WIN32)
	#include <windows.h>
	#define sleep Sleep
	#else
	#include <sys/time.h>
	#include <unistd.h>
	#endif

	#if defined(_WRS_KERNEL)
	#include <OsWrapper.h>
	#endif

	volatile int finished = 0;
	int subscribed = 0;
	int disconnected = 0;


	void mysleep(int ms)
	{
	#if defined(_WIN32)
	Sleep(ms);
	#else
	usleep(ms * 1000);
	#endif
	}

	void cfinish(int sig)
	{
	signal(SIGINT, NULL);
	finished = 1;
	}


	struct pubsub_opts opts =
	{
	0, 0, 0, 0, "\n", 100, /* debug/app options */
	NULL, NULL, 1, 0, 0, /* message options */
	MQTTVERSION_DEFAULT, NULL, "paho-c-sub", 0, 0, NULL, NULL, "localhost", "1883", NULL, 10, /* MQTT options */
	NULL, NULL, 0, 0, /* will options */
	0, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, /* TLS options */
	0, {NULL, NULL}, /* MQTT V5 options */
	};


	int messageArrived(void context, char topicName, int topicLen, MQTTAsync_message *message)
	{
	size_t delimlen = 0;

	if (opts.verbose)
	printf("%d %s\t", message->payloadlen, topicName);
	if (opts.delimiter)
	delimlen = strlen(opts.delimiter);
	if (opts.delimiter == NULL \|\| (message->payloadlen > delimlen &&
	strncmp(opts.delimiter, &((char*)message->payload)[message->payloadlen - delimlen], delimlen) == 0))
	printf("%.s", message->payloadlen, (char)message->payload);
	else
	printf("%.s%s", message->payloadlen, (char)message->payload, opts.delimiter);
	if (message->struct_version == 1 && opts.verbose)
	logProperties(&message->properties);
	fflush(stdout);
	MQTTAsync_freeMessage(&message);
	MQTTAsync_free(topicName);
	return 1;
	}


	void onDisconnect(void* context, MQTTAsync_successData* response)
	{
	disconnected = 1;
	}


	void onSubscribe5(void* context, MQTTAsync_successData5* response)
	{
	subscribed = 1;
	}

	void onSubscribe(void* context, MQTTAsync_successData* response)
	{
	subscribed = 1;
	}


	void onSubscribeFailure5(void* context, MQTTAsync_failureData5* response)
	{
	if (!opts.quiet)
	fprintf(stderr, "Subscribe failed, rc %s reason code %s\n",
	MQTTAsync_strerror(response->code),
	MQTTReasonCode_toString(response->reasonCode));
	finished = 1;
	}


	void onSubscribeFailure(void* context, MQTTAsync_failureData* response)
	{
	if (!opts.quiet)
	fprintf(stderr, "Subscribe failed, rc %s\n",
	MQTTAsync_strerror(response->code));
	finished = 1;
	}


	void onConnectFailure5(void* context, MQTTAsync_failureData5* response)
	{
	if (!opts.quiet)
	fprintf(stderr, "Connect failed, rc %s reason code %s\n",
	MQTTAsync_strerror(response->code),
	MQTTReasonCode_toString(response->reasonCode));
	finished = 1;
	}


	void onConnectFailure(void* context, MQTTAsync_failureData* response)
	{
	if (!opts.quiet)
	fprintf(stderr, "Connect failed, rc %s\n", response ? MQTTAsync_strerror(response->code) : "none");
	finished = 1;
	}


	void onConnect5(void* context, MQTTAsync_successData5* response)
	{
	MQTTAsync client = (MQTTAsync)context;
	MQTTAsync_callOptions copts = MQTTAsync_callOptions_initializer;
	int rc;

	if (opts.verbose)
	printf("Subscribing to topic %s with client %s at QoS %d\n", opts.topic, opts.clientid, opts.qos);

	copts.onSuccess5 = onSubscribe5;
	copts.onFailure5 = onSubscribeFailure5;
	copts.context = client;
	if ((rc = MQTTAsync_subscribe(client, opts.topic, opts.qos, &copts)) != MQTTASYNC_SUCCESS)
	{
	if (!opts.quiet)
	fprintf(stderr, "Failed to start subscribe, return code %s\n", MQTTAsync_strerror(rc));
	finished = 1;
	}
	}


	void onConnect(void* context, MQTTAsync_successData* response)
	{
	MQTTAsync client = (MQTTAsync)context;
	MQTTAsync_responseOptions ropts = MQTTAsync_responseOptions_initializer;
	int rc;

	if (opts.verbose)
	printf("Subscribing to topic %s with client %s at QoS %d\n", opts.topic, opts.clientid, opts.qos);

	ropts.onSuccess = onSubscribe;
	ropts.onFailure = onSubscribeFailure;
	ropts.context = client;
	if ((rc = MQTTAsync_subscribe(client, opts.topic, opts.qos, &ropts)) != MQTTASYNC_SUCCESS)
	{
	if (!opts.quiet)
	fprintf(stderr, "Failed to start subscribe, return code %s\n", MQTTAsync_strerror(rc));
	finished = 1;
	}
	}

	MQTTAsync_connectOptions conn_opts = MQTTAsync_connectOptions_initializer;


	void trace_callback(enum MQTTASYNC_TRACE_LEVELS level, char* message)
	{
	fprintf(stderr, "Trace : %d, %s\n", level, message);
	}


	int main(int argc, char** argv)
	{
	MQTTAsync client;
	MQTTAsync_disconnectOptions disc_opts = MQTTAsync_disconnectOptions_initializer;
	MQTTAsync_createOptions create_opts = MQTTAsync_createOptions_initializer;
	MQTTAsync_willOptions will_opts = MQTTAsync_willOptions_initializer;
	MQTTAsync_SSLOptions ssl_opts = MQTTAsync_SSLOptions_initializer;
	int rc = 0;
	char* url = NULL;
	const char* version = NULL;
	const char* program_name = "paho_c_sub";
	MQTTAsync_nameValue* infos = MQTTAsync_getVersionInfo();
	#if !defined(_WIN32)
	struct sigaction sa;
	#endif

	if (argc < 2)
	usage(&opts, (pubsub_opts_nameValue*)infos, program_name);

	if (getopts(argc, argv, &opts) != 0)
	usage(&opts, (pubsub_opts_nameValue*)infos, program_name);

	if (strchr(opts.topic, '#') \|\| strchr(opts.topic, '+'))
	opts.verbose = 1;

	if (opts.connection)
	url = opts.connection;
	else
	{
	url = malloc(100);
	sprintf(url, "%s:%s", opts.host, opts.port);
	}
	if (opts.verbose)
	printf("URL is %s\n", url);

	if (opts.tracelevel > 0)
	{
	MQTTAsync_setTraceCallback(trace_callback);
	MQTTAsync_setTraceLevel(opts.tracelevel);
	}

	if (opts.MQTTVersion >= MQTTVERSION_5)
	create_opts.MQTTVersion = MQTTVERSION_5;
	rc = MQTTAsync_createWithOptions(&client, url, opts.clientid, MQTTCLIENT_PERSISTENCE_NONE,
	NULL, &create_opts);
	if (rc != MQTTASYNC_SUCCESS)
	{
	if (!opts.quiet)
	fprintf(stderr, "Failed to create client, return code: %s\n", MQTTAsync_strerror(rc));
	exit(EXIT_FAILURE);
	}

	rc = MQTTAsync_setCallbacks(client, client, NULL, messageArrived, NULL);
	if (rc != MQTTASYNC_SUCCESS)
	{
	if (!opts.quiet)
	fprintf(stderr, "Failed to set callbacks, return code: %s\n", MQTTAsync_strerror(rc));
	exit(EXIT_FAILURE);
	}

	#if defined(_WIN32)
	signal(SIGINT, cfinish);
	signal(SIGTERM, cfinish);
	#else
	memset(&sa, 0, sizeof(struct sigaction));
	sa.sa_handler = cfinish;
	sa.sa_flags = 0;

	sigaction(SIGINT, &sa, NULL);
	sigaction(SIGTERM, &sa, NULL);
	#endif

	if (opts.MQTTVersion == MQTTVERSION_5)
	{
	MQTTAsync_connectOptions conn_opts5 = MQTTAsync_connectOptions_initializer5;
	conn_opts = conn_opts5;
	conn_opts.onSuccess5 = onConnect5;
	conn_opts.onFailure5 = onConnectFailure5;
	conn_opts.cleanstart = 1;
	}
	else
	{
	conn_opts.onSuccess = onConnect;
	conn_opts.onFailure = onConnectFailure;
	conn_opts.cleansession = 1;
	}
	conn_opts.keepAliveInterval = opts.keepalive;
	conn_opts.username = opts.username;
	conn_opts.password = opts.password;
	conn_opts.MQTTVersion = opts.MQTTVersion;
	conn_opts.context = client;
	conn_opts.automaticReconnect = 1;

	if (opts.will_topic) /* will options */
	{
	will_opts.message = opts.will_payload;
	will_opts.topicName = opts.will_topic;
	will_opts.qos = opts.will_qos;
	will_opts.retained = opts.will_retain;
	conn_opts.will = &will_opts;
	}

	if (opts.connection && (strncmp(opts.connection, "ssl://", 6) == 0 \|\|
	strncmp(opts.connection, "wss://", 6) == 0))
	{
	ssl_opts.verify = (opts.insecure) ? 0 : 1;
	ssl_opts.CApath = opts.capath;
	ssl_opts.keyStore = opts.cert;
	ssl_opts.trustStore = opts.cafile;
	ssl_opts.privateKey = opts.key;
	ssl_opts.privateKeyPassword = opts.keypass;
	ssl_opts.enabledCipherSuites = opts.ciphers;
	conn_opts.ssl = &ssl_opts;
	}

	if ((rc = MQTTAsync_connect(client, &conn_opts)) != MQTTASYNC_SUCCESS)
	{
	if (!opts.quiet)
	fprintf(stderr, "Failed to start connect, return code %s\n", MQTTAsync_strerror(rc));
	exit(EXIT_FAILURE);
	}

	while (!subscribed)
	mysleep(100);

	if (finished)
	goto exit;

	while (!finished)
	mysleep(100);

	disc_opts.onSuccess = onDisconnect;
	if ((rc = MQTTAsync_disconnect(client, &disc_opts)) != MQTTASYNC_SUCCESS)
	{
	if (!opts.quiet)
	fprintf(stderr, "Failed to start disconnect, return code: %s\n", MQTTAsync_strerror(rc));
	exit(EXIT_FAILURE);
	}

	while (!disconnected)
	mysleep(100);

	exit:
	MQTTAsync_destroy(&client);

	return EXIT_SUCCESS;
	}