UNPKG

6.86 kBtext/x-cView Raw
1/*
2Copyright (c) 2009-2013 Roger Light <roger@atchoo.org>
3All rights reserved.
4
5Redistribution and use in source and binary forms, with or without
6modification, are permitted provided that the following conditions are met:
7
81. Redistributions of source code must retain the above copyright notice,
9 this list of conditions and the following disclaimer.
102. Redistributions in binary form must reproduce the above copyright
11 notice, this list of conditions and the following disclaimer in the
12 documentation and/or other materials provided with the distribution.
133. Neither the name of mosquitto nor the names of its
14 contributors may be used to endorse or promote products derived from
15 this software without specific prior written permission.
16
17THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
18AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
21LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
22CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
23SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
25CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
26ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27POSSIBILITY OF SUCH DAMAGE.
28*/
29
30#include <assert.h>
31#include <string.h>
32
33#include "mosquitto.h"
34#include "logging_mosq.h"
35#include "memory_mosq.h"
36#include "mqtt3_protocol.h"
37#include "net_mosq.h"
38#include "send_mosq.h"
39#include "util_mosq.h"
40
41#ifdef WITH_BROKER
42#include "mosquitto_broker.h"
43#endif
44
45int _mosquitto_send_connect(struct mosquitto *mosq, uint16_t keepalive, bool clean_session)
46{
47 struct _mosquitto_packet *packet = NULL;
48 int payloadlen;
49 uint8_t will = 0;
50 uint8_t byte;
51 int rc;
52 uint8_t version = PROTOCOL_VERSION_v31;
53
54 assert(mosq);
55 assert(mosq->id);
56
57 packet = _mosquitto_calloc(1, sizeof(struct _mosquitto_packet));
58 if(!packet) return MOSQ_ERR_NOMEM;
59
60 payloadlen = 2+strlen(mosq->id);
61 if(mosq->will){
62 will = 1;
63 assert(mosq->will->topic);
64
65 payloadlen += 2+strlen(mosq->will->topic) + 2+mosq->will->payloadlen;
66 }
67 if(mosq->username){
68 payloadlen += 2+strlen(mosq->username);
69 if(mosq->password){
70 payloadlen += 2+strlen(mosq->password);
71 }
72 }
73
74 packet->command = CONNECT;
75 packet->remaining_length = 12+payloadlen;
76 rc = _mosquitto_packet_alloc(packet);
77 if(rc){
78 _mosquitto_free(packet);
79 return rc;
80 }
81
82 /* Variable header */
83 _mosquitto_write_string(packet, PROTOCOL_NAME_v31, strlen(PROTOCOL_NAME_v31));
84#if defined(WITH_BROKER) && defined(WITH_BRIDGE)
85 if(mosq->bridge && mosq->bridge->try_private && mosq->bridge->try_private_accepted){
86 version |= 0x80;
87 }else{
88 }
89#endif
90 _mosquitto_write_byte(packet, version);
91 byte = (clean_session&0x1)<<1;
92 if(will){
93 byte = byte | ((mosq->will->retain&0x1)<<5) | ((mosq->will->qos&0x3)<<3) | ((will&0x1)<<2);
94 }
95 if(mosq->username){
96 byte = byte | 0x1<<7;
97 if(mosq->password){
98 byte = byte | 0x1<<6;
99 }
100 }
101 _mosquitto_write_byte(packet, byte);
102 _mosquitto_write_uint16(packet, keepalive);
103
104 /* Payload */
105 _mosquitto_write_string(packet, mosq->id, strlen(mosq->id));
106 if(will){
107 _mosquitto_write_string(packet, mosq->will->topic, strlen(mosq->will->topic));
108 _mosquitto_write_string(packet, (const char *)mosq->will->payload, mosq->will->payloadlen);
109 }
110 if(mosq->username){
111 _mosquitto_write_string(packet, mosq->username, strlen(mosq->username));
112 if(mosq->password){
113 _mosquitto_write_string(packet, mosq->password, strlen(mosq->password));
114 }
115 }
116
117 mosq->keepalive = keepalive;
118#ifdef WITH_BROKER
119# ifdef WITH_BRIDGE
120 _mosquitto_log_printf(mosq, MOSQ_LOG_DEBUG, "Bridge %s sending CONNECT", mosq->id);
121# endif
122#else
123 _mosquitto_log_printf(mosq, MOSQ_LOG_DEBUG, "Client %s sending CONNECT", mosq->id);
124#endif
125 return _mosquitto_packet_queue(mosq, packet);
126}
127
128int _mosquitto_send_disconnect(struct mosquitto *mosq)
129{
130 assert(mosq);
131#ifdef WITH_BROKER
132# ifdef WITH_BRIDGE
133 _mosquitto_log_printf(mosq, MOSQ_LOG_DEBUG, "Bridge %s sending DISCONNECT", mosq->id);
134# endif
135#else
136 _mosquitto_log_printf(mosq, MOSQ_LOG_DEBUG, "Client %s sending DISCONNECT", mosq->id);
137#endif
138 return _mosquitto_send_simple_command(mosq, DISCONNECT);
139}
140
141int _mosquitto_send_subscribe(struct mosquitto *mosq, int *mid, bool dup, const char *topic, uint8_t topic_qos)
142{
143 /* FIXME - only deals with a single topic */
144 struct _mosquitto_packet *packet = NULL;
145 uint32_t packetlen;
146 uint16_t local_mid;
147 int rc;
148
149 assert(mosq);
150 assert(topic);
151
152 packet = _mosquitto_calloc(1, sizeof(struct _mosquitto_packet));
153 if(!packet) return MOSQ_ERR_NOMEM;
154
155 packetlen = 2 + 2+strlen(topic) + 1;
156
157 packet->command = SUBSCRIBE | (dup<<3) | (1<<1);
158 packet->remaining_length = packetlen;
159 rc = _mosquitto_packet_alloc(packet);
160 if(rc){
161 _mosquitto_free(packet);
162 return rc;
163 }
164
165 /* Variable header */
166 local_mid = _mosquitto_mid_generate(mosq);
167 if(mid) *mid = (int)local_mid;
168 _mosquitto_write_uint16(packet, local_mid);
169
170 /* Payload */
171 _mosquitto_write_string(packet, topic, strlen(topic));
172 _mosquitto_write_byte(packet, topic_qos);
173
174#ifdef WITH_BROKER
175# ifdef WITH_BRIDGE
176 _mosquitto_log_printf(mosq, MOSQ_LOG_DEBUG, "Bridge %s sending SUBSCRIBE (Mid: %d, Topic: %s, QoS: %d)", mosq->id, local_mid, topic, topic_qos);
177# endif
178#else
179 _mosquitto_log_printf(mosq, MOSQ_LOG_DEBUG, "Client %s sending SUBSCRIBE (Mid: %d, Topic: %s, QoS: %d)", mosq->id, local_mid, topic, topic_qos);
180#endif
181
182 return _mosquitto_packet_queue(mosq, packet);
183}
184
185
186int _mosquitto_send_unsubscribe(struct mosquitto *mosq, int *mid, bool dup, const char *topic)
187{
188 /* FIXME - only deals with a single topic */
189 struct _mosquitto_packet *packet = NULL;
190 uint32_t packetlen;
191 uint16_t local_mid;
192 int rc;
193
194 assert(mosq);
195 assert(topic);
196
197 packet = _mosquitto_calloc(1, sizeof(struct _mosquitto_packet));
198 if(!packet) return MOSQ_ERR_NOMEM;
199
200 packetlen = 2 + 2+strlen(topic);
201
202 packet->command = UNSUBSCRIBE | (dup<<3) | (1<<1);
203 packet->remaining_length = packetlen;
204 rc = _mosquitto_packet_alloc(packet);
205 if(rc){
206 _mosquitto_free(packet);
207 return rc;
208 }
209
210 /* Variable header */
211 local_mid = _mosquitto_mid_generate(mosq);
212 if(mid) *mid = (int)local_mid;
213 _mosquitto_write_uint16(packet, local_mid);
214
215 /* Payload */
216 _mosquitto_write_string(packet, topic, strlen(topic));
217
218#ifdef WITH_BROKER
219# ifdef WITH_BRIDGE
220 _mosquitto_log_printf(mosq, MOSQ_LOG_DEBUG, "Bridge %s sending UNSUBSCRIBE (Mid: %d, Topic: %s)", mosq->id, local_mid, topic);
221# endif
222#else
223 _mosquitto_log_printf(mosq, MOSQ_LOG_DEBUG, "Client %s sending UNSUBSCRIBE (Mid: %d, Topic: %s)", mosq->id, local_mid, topic);
224#endif
225 return _mosquitto_packet_queue(mosq, packet);
226}
227