def gen_connack(flags=0,
rc=0,
proto_ver=4,
properties=b"",
property_helper=True):
if proto_ver == 5:
if property_helper == True:
if properties is not None:
properties = mqtt5_props.gen_uint16_prop(mqtt5_props.PROP_TOPIC_ALIAS_MAXIMUM, 10) \
+ properties + mqtt5_props.gen_uint16_prop(mqtt5_props.PROP_RECEIVE_MAXIMUM, 20)
else:
properties = b""
properties = mqtt5_props.prop_finalise(properties)
packet = struct.pack('!BBBB', 32, 2 + len(properties), flags,
rc) + properties
else:
packet = struct.pack('!BBBB', 32, 2, flags, rc)
return packet
def gen_connack(flags=0, rc=0, proto_ver=4, properties=b""):
if proto_ver == 5:
if properties is not None:
properties = mqtt5_props.gen_uint16_prop(mqtt5_props.PROP_TOPIC_ALIAS_MAXIMUM, 10) + properties
else:
properties = b""
properties = mqtt5_props.prop_finalise(properties)
packet = struct.pack('!BBBB', 32, 2+len(properties), flags, rc) + properties
else:
packet = struct.pack('!BBBB', 32, 2, flags, rc);
return packet
def gen_connect(client_id, clean_session=True, keepalive=60, username=None, password=None, will_topic=None, will_qos=0, will_retain=False, will_payload=b"", proto_ver=4, connect_reserved=False, properties=b"", will_properties=b""):
if (proto_ver&0x7F) == 3 or proto_ver == 0:
remaining_length = 12
elif (proto_ver&0x7F) == 4 or proto_ver == 5:
remaining_length = 10
else:
raise ValueError
if client_id != None:
client_id = client_id.encode("utf-8")
remaining_length = remaining_length + 2+len(client_id)
else:
remaining_length = remaining_length + 2
connect_flags = 0
if connect_reserved:
connect_flags = connect_flags | 0x01
if clean_session:
connect_flags = connect_flags | 0x02
if proto_ver == 5:
if properties == b"":
properties += mqtt5_props.gen_uint16_prop(mqtt5_props.PROP_RECEIVE_MAXIMUM, 20)
properties = mqtt5_props.prop_finalise(properties)
remaining_length += len(properties)
if will_topic != None:
will_topic = will_topic.encode("utf-8")
remaining_length = remaining_length + 2+len(will_topic) + 2+len(will_payload)
connect_flags = connect_flags | 0x04 | ((will_qos&0x03) << 3)
if will_retain:
connect_flags = connect_flags | 32
if proto_ver == 5:
will_properties = mqtt5_props.prop_finalise(will_properties)
remaining_length += len(will_properties)
if username != None:
username = username.encode("utf-8")
remaining_length = remaining_length + 2+len(username)
connect_flags = connect_flags | 0x80
if password != None:
password = password.encode("utf-8")
connect_flags = connect_flags | 0x40
remaining_length = remaining_length + 2+len(password)
rl = pack_remaining_length(remaining_length)
packet = struct.pack("!B"+str(len(rl))+"s", 0x10, rl)
if (proto_ver&0x7F) == 3 or proto_ver == 0:
packet = packet + struct.pack("!H6sBBH", len(b"MQIsdp"), b"MQIsdp", proto_ver, connect_flags, keepalive)
elif (proto_ver&0x7F) == 4 or proto_ver == 5:
packet = packet + struct.pack("!H4sBBH", len(b"MQTT"), b"MQTT", proto_ver, connect_flags, keepalive)
if proto_ver == 5:
packet += properties
if client_id != None:
packet = packet + struct.pack("!H"+str(len(client_id))+"s", len(client_id), bytes(client_id))
else:
packet = packet + struct.pack("!H", 0)
if will_topic != None:
packet += will_properties
packet = packet + struct.pack("!H"+str(len(will_topic))+"s", len(will_topic), will_topic)
if len(will_payload) > 0:
packet = packet + struct.pack("!H"+str(len(will_payload))+"s", len(will_payload), will_payload)
else:
packet = packet + struct.pack("!H", 0)
if username != None:
packet = packet + struct.pack("!H"+str(len(username))+"s", len(username), username)
if password != None:
packet = packet + struct.pack("!H"+str(len(password))+"s", len(password), password)
return packet
def gen_connect(client_id, clean_session=True, keepalive=60, username=None, password=None, will_topic=None, will_qos=0, will_retain=False, will_payload=b"", proto_ver=4, connect_reserved=False, properties=b"", will_properties=b""):
if (proto_ver&0x7F) == 3 or proto_ver == 0:
remaining_length = 12
elif (proto_ver&0x7F) == 4 or proto_ver == 5:
remaining_length = 10
else:
raise ValueError
if client_id != None:
client_id = client_id.encode("utf-8")
remaining_length = remaining_length + 2+len(client_id)
else:
remaining_length = remaining_length + 2
connect_flags = 0
if connect_reserved:
connect_flags = connect_flags | 0x01
if clean_session:
connect_flags = connect_flags | 0x02
if proto_ver == 5:
if properties == b"":
properties += mqtt5_props.gen_uint16_prop(mqtt5_props.PROP_RECEIVE_MAXIMUM, 20)
properties = mqtt5_props.prop_finalise(properties)
remaining_length += len(properties)
if will_topic != None:
will_topic = will_topic.encode("utf-8")
remaining_length = remaining_length + 2+len(will_topic) + 2+len(will_payload)
connect_flags = connect_flags | 0x04 | ((will_qos&0x03) << 3)
if will_retain:
connect_flags = connect_flags | 32
if proto_ver == 5:
will_properties = mqtt5_props.prop_finalise(will_properties)
remaining_length += len(will_properties)
if username != None:
username = username.encode("utf-8")
remaining_length = remaining_length + 2+len(username)
connect_flags = connect_flags | 0x80
if password != None:
password = password.encode("utf-8")
connect_flags = connect_flags | 0x40
remaining_length = remaining_length + 2+len(password)
rl = pack_remaining_length(remaining_length)
packet = struct.pack("!B"+str(len(rl))+"s", 0x10, rl)
if (proto_ver&0x7F) == 3 or proto_ver == 0:
packet = packet + struct.pack("!H6sBBH", len(b"MQIsdp"), b"MQIsdp", proto_ver, connect_flags, keepalive)
elif (proto_ver&0x7F) == 4 or proto_ver == 5:
packet = packet + struct.pack("!H4sBBH", len(b"MQTT"), b"MQTT", proto_ver, connect_flags, keepalive)
if proto_ver == 5:
packet += properties
if client_id != None:
packet = packet + struct.pack("!H"+str(len(client_id))+"s", len(client_id), bytes(client_id))
else:
packet = packet + struct.pack("!H", 0)
if will_topic != None:
packet += will_properties
packet = packet + struct.pack("!H"+str(len(will_topic))+"s", len(will_topic), will_topic)
if len(will_payload) > 0:
packet = packet + struct.pack("!H"+str(len(will_payload))+"s", len(will_payload), will_payload)
else:
packet = packet + struct.pack("!H", 0)
if username != None:
packet = packet + struct.pack("!H"+str(len(username))+"s", len(username), username)
if password != None:
packet = packet + struct.pack("!H"+str(len(password))+"s", len(password), password)
return packet
