163 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
			
		
		
	
	
			163 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
| MQTT client for lwIP
 | |
| 
 | |
| Author: Erik Andersson
 | |
| 
 | |
| Details of the MQTT protocol can be found at:
 | |
| http://docs.oasis-open.org/mqtt/mqtt/v3.1.1/os/mqtt-v3.1.1-os.html 
 | |
| 
 | |
| -----------------------------------------------------------------
 | |
| 1. Initial steps, reserve memory and make connection to server:
 | |
| 
 | |
| 1.1: Provide storage
 | |
| 
 | |
| Static allocation:
 | |
|   mqtt_client_t static_client;
 | |
|   example_do_connect(&static_client);
 | |
| 
 | |
| Dynamic allocation:
 | |
|   mqtt_client_t *client = mqtt_client_new();
 | |
|   if(client != NULL) {
 | |
|     example_do_connect(&client);
 | |
|   }
 | |
|   
 | |
| 1.2: Establish Connection with server
 | |
| 
 | |
| void example_do_connect(mqtt_client_t *client)
 | |
| {
 | |
|   struct mqtt_connect_client_info_t ci;
 | |
|   err_t err;
 | |
|   
 | |
|   /* Setup an empty client info structure */
 | |
|   memset(&ci, 0, sizeof(ci));
 | |
|   
 | |
|   /* Minimal amount of information required is client identifier, so set it here */ 
 | |
|   ci.client_id = "lwip_test";
 | |
|   
 | |
|   /* Initiate client and connect to server, if this fails immediately an error code is returned
 | |
|      otherwise mqtt_connection_cb will be called with connection result after attempting 
 | |
|      to establish a connection with the server. 
 | |
|      For now MQTT version 3.1.1 is always used */
 | |
|   
 | |
|   err = mqtt_client_connect(client, ip_addr, MQTT_PORT, mqtt_connection_cb, 0, &ci);
 | |
|   
 | |
|   /* For now just print the result code if something goes wrong
 | |
|   if(err != ERR_OK) {
 | |
|     printf("mqtt_connect return %d\n", err);
 | |
|   }
 | |
| }
 | |
| 
 | |
| Connection to server can also be probed by calling mqtt_client_is_connected(client) 
 | |
| 
 | |
| -----------------------------------------------------------------
 | |
| 2. Implementing the connection status callback
 | |
| 
 | |
| 
 | |
| static void mqtt_connection_cb(mqtt_client_t *client, void *arg, mqtt_connection_status_t status)
 | |
| {
 | |
|   err_t err;
 | |
|   if(status == MQTT_CONNECT_ACCEPTED) {
 | |
|     printf("mqtt_connection_cb: Successfully connected\n");
 | |
|     
 | |
|     /* Setup callback for incoming publish requests */
 | |
|     mqtt_set_inpub_callback(client, mqtt_incoming_publish_cb, mqtt_incoming_data_cb, arg);
 | |
|     
 | |
|     /* Subscribe to a topic named "subtopic" with QoS level 1, call mqtt_sub_request_cb with result */ 
 | |
|     err = mqtt_subscribe(client, "subtopic", 1, mqtt_sub_request_cb, arg);
 | |
| 
 | |
|     if(err != ERR_OK) {
 | |
|       printf("mqtt_subscribe return: %d\n", err);
 | |
|     }
 | |
|   } else {
 | |
|     printf("mqtt_connection_cb: Disconnected, reason: %d\n", status);
 | |
|     
 | |
|     /* Its more nice to be connected, so try to reconnect */
 | |
|     example_do_connect(client);
 | |
|   }  
 | |
| }
 | |
| 
 | |
| static void mqtt_sub_request_cb(void *arg, err_t result)
 | |
| {
 | |
|   /* Just print the result code here for simplicity, 
 | |
|      normal behaviour would be to take some action if subscribe fails like 
 | |
|      notifying user, retry subscribe or disconnect from server */
 | |
|   printf("Subscribe result: %d\n", result);
 | |
| }
 | |
| 
 | |
| -----------------------------------------------------------------
 | |
| 3. Implementing callbacks for incoming publish and data
 | |
| 
 | |
| /* The idea is to demultiplex topic and create some reference to be used in data callbacks
 | |
|    Example here uses a global variable, better would be to use a member in arg
 | |
|    If RAM and CPU budget allows it, the easiest implementation might be to just take a copy of
 | |
|    the topic string and use it in mqtt_incoming_data_cb
 | |
| */
 | |
| static int inpub_id;
 | |
| static void mqtt_incoming_publish_cb(void *arg, const char *topic, u32_t tot_len)
 | |
| {
 | |
|   printf("Incoming publish at topic %s with total length %u\n", topic, (unsigned int)tot_len);
 | |
| 
 | |
|   /* Decode topic string into a user defined reference */
 | |
|   if(strcmp(topic, "print_payload") == 0) {
 | |
|     inpub_id = 0;
 | |
|   } else if(topic[0] == 'A') {
 | |
|     /* All topics starting with 'A' might be handled at the same way */
 | |
|     inpub_id = 1;
 | |
|   } else {
 | |
|     /* For all other topics */
 | |
|     inpub_id = 2;
 | |
|   }
 | |
| }
 | |
| 
 | |
| static void mqtt_incoming_data_cb(void *arg, const u8_t *data, u16_t len, u8_t flags)
 | |
| {
 | |
|   printf("Incoming publish payload with length %d, flags %u\n", len, (unsigned int)flags);
 | |
| 
 | |
|   if(flags & MQTT_DATA_FLAG_LAST) {
 | |
|     /* Last fragment of payload received (or whole part if payload fits receive buffer
 | |
|        See MQTT_VAR_HEADER_BUFFER_LEN)  */
 | |
| 
 | |
|     /* Call function or do action depending on reference, in this case inpub_id */
 | |
|     if(inpub_id == 0) {
 | |
|       /* Don't trust the publisher, check zero termination */
 | |
|       if(data[len-1] == 0) {
 | |
|         printf("mqtt_incoming_data_cb: %s\n", (const char *)data);
 | |
|       }
 | |
|     } else if(inpub_id == 1) {
 | |
|       /* Call an 'A' function... */
 | |
|     } else {
 | |
|       printf("mqtt_incoming_data_cb: Ignoring payload...\n");
 | |
|     }
 | |
|   } else {
 | |
|     /* Handle fragmented payload, store in buffer, write to file or whatever */
 | |
|   }
 | |
| }
 | |
| 
 | |
| -----------------------------------------------------------------
 | |
| 4. Using outgoing publish
 | |
| 
 | |
| 
 | |
| void example_publish(mqtt_client_t *client, void *arg)
 | |
| {
 | |
|   const char *pub_payload= "PubSubHubLubJub";
 | |
|   err_t err;
 | |
|   u8_t qos = 2; /* 0 1 or 2, see MQTT specification */
 | |
|   u8_t retain = 0; /* No don't retain such crappy payload... */
 | |
|   err = mqtt_publish(client, "pub_topic", pub_payload, strlen(pub_payload), qos, retain, mqtt_pub_request_cb, arg);
 | |
|   if(err != ERR_OK) {
 | |
|     printf("Publish err: %d\n", err);
 | |
|   }
 | |
| }
 | |
| 
 | |
| /* Called when publish is complete either with sucess or failure */
 | |
| static void mqtt_pub_request_cb(void *arg, err_t result)
 | |
| {
 | |
|   if(result != ERR_OK) {
 | |
|     printf("Publish result: %d\n", result);
 | |
|   }
 | |
| }
 | |
| 
 | |
| -----------------------------------------------------------------
 | |
| 5. Disconnecting
 | |
| 
 | |
| Simply call mqtt_disconnect(client)
 |