249 lines
9.9 KiB
Erlang
249 lines
9.9 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2012-2016 Feng Lee <feng@emqtt.io>.
|
|
%%
|
|
%% Licensed under the Apache License, Version 2.0 (the "License");
|
|
%% you may not use this file except in compliance with the License.
|
|
%% You may obtain a copy of the License at
|
|
%%
|
|
%% http://www.apache.org/licenses/LICENSE-2.0
|
|
%%
|
|
%% Unless required by applicable law or agreed to in writing, software
|
|
%% distributed under the License is distributed on an "AS IS" BASIS,
|
|
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
%% See the License for the specific language governing permissions and
|
|
%% limitations under the License.
|
|
%%--------------------------------------------------------------------
|
|
|
|
%% MQTT Protocol Header
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% MQTT Protocol Version and Levels
|
|
%%--------------------------------------------------------------------
|
|
-define(MQTT_PROTO_V31, 3).
|
|
-define(MQTT_PROTO_V311, 4).
|
|
|
|
-define(PROTOCOL_NAMES, [
|
|
{?MQTT_PROTO_V31, <<"MQIsdp">>},
|
|
{?MQTT_PROTO_V311, <<"MQTT">>}]).
|
|
|
|
-type mqtt_vsn() :: ?MQTT_PROTO_V31 | ?MQTT_PROTO_V311.
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% MQTT QoS
|
|
%%--------------------------------------------------------------------
|
|
-define(QOS_0, 0). %% At most once
|
|
-define(QOS_1, 1). %% At least once
|
|
-define(QOS_2, 2). %% Exactly once
|
|
|
|
-define(QOS0, 0). %% At most once
|
|
-define(QOS1, 1). %% At least once
|
|
-define(QOS2, 2). %% Exactly once
|
|
|
|
-define(IS_QOS(I), (I >= ?QOS0 andalso I =< ?QOS2)).
|
|
|
|
-type mqtt_qos() :: ?QOS0 | ?QOS1 | ?QOS2.
|
|
|
|
-type mqtt_qos_name() :: qos0 | at_most_once |
|
|
qos1 | at_least_once |
|
|
qos2 | exactly_once.
|
|
|
|
-define(QOS_I(Name),
|
|
begin
|
|
(case Name of
|
|
?QOS_0 -> ?QOS_0;
|
|
qos0 -> ?QOS_0;
|
|
at_most_once -> ?QOS_0;
|
|
?QOS_1 -> ?QOS_1;
|
|
qos1 -> ?QOS_1;
|
|
at_least_once -> ?QOS_1;
|
|
?QOS_2 -> ?QOS_2;
|
|
qos2 -> ?QOS_2;
|
|
exactly_once -> ?QOS_2
|
|
end)
|
|
end).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Max ClientId Length. Why 1024? NiDongDe...
|
|
%%--------------------------------------------------------------------
|
|
-define(MAX_CLIENTID_LEN, 1024).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% MQTT Control Packet Types
|
|
%%--------------------------------------------------------------------
|
|
-define(RESERVED, 0). %% Reserved
|
|
-define(CONNECT, 1). %% Client request to connect to Server
|
|
-define(CONNACK, 2). %% Server to Client: Connect acknowledgment
|
|
-define(PUBLISH, 3). %% Publish message
|
|
-define(PUBACK, 4). %% Publish acknowledgment
|
|
-define(PUBREC, 5). %% Publish received (assured delivery part 1)
|
|
-define(PUBREL, 6). %% Publish release (assured delivery part 2)
|
|
-define(PUBCOMP, 7). %% Publish complete (assured delivery part 3)
|
|
-define(SUBSCRIBE, 8). %% Client subscribe request
|
|
-define(SUBACK, 9). %% Server Subscribe acknowledgment
|
|
-define(UNSUBSCRIBE, 10). %% Unsubscribe request
|
|
-define(UNSUBACK, 11). %% Unsubscribe acknowledgment
|
|
-define(PINGREQ, 12). %% PING request
|
|
-define(PINGRESP, 13). %% PING response
|
|
-define(DISCONNECT, 14). %% Client is disconnecting
|
|
|
|
-define(TYPE_NAMES, [
|
|
'CONNECT',
|
|
'CONNACK',
|
|
'PUBLISH',
|
|
'PUBACK',
|
|
'PUBREC',
|
|
'PUBREL',
|
|
'PUBCOMP',
|
|
'SUBSCRIBE',
|
|
'SUBACK',
|
|
'UNSUBSCRIBE',
|
|
'UNSUBACK',
|
|
'PINGREQ',
|
|
'PINGRESP',
|
|
'DISCONNECT']).
|
|
|
|
-type mqtt_packet_type() :: ?RESERVED..?DISCONNECT.
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% MQTT Connect Return Codes
|
|
%%--------------------------------------------------------------------
|
|
-define(CONNACK_ACCEPT, 0). %% Connection accepted
|
|
-define(CONNACK_PROTO_VER, 1). %% Unacceptable protocol version
|
|
-define(CONNACK_INVALID_ID, 2). %% Client Identifier is correct UTF-8 but not allowed by the Server
|
|
-define(CONNACK_SERVER, 3). %% Server unavailable
|
|
-define(CONNACK_CREDENTIALS, 4). %% Username or password is malformed
|
|
-define(CONNACK_AUTH, 5). %% Client is not authorized to connect
|
|
|
|
-type mqtt_connack() :: ?CONNACK_ACCEPT..?CONNACK_AUTH.
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% MQTT Parser and Serializer
|
|
%%--------------------------------------------------------------------
|
|
-define(MAX_LEN, 16#fffffff).
|
|
-define(HIGHBIT, 2#10000000).
|
|
-define(LOWBITS, 2#01111111).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% MQTT Packet Fixed Header
|
|
%%--------------------------------------------------------------------
|
|
-record(mqtt_packet_header, {
|
|
type = ?RESERVED :: mqtt_packet_type(),
|
|
dup = false :: boolean(),
|
|
qos = ?QOS_0 :: mqtt_qos(),
|
|
retain = false :: boolean()}).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% MQTT Packets
|
|
%%--------------------------------------------------------------------
|
|
-type mqtt_client_id() :: binary().
|
|
-type mqtt_packet_id() :: 1..16#ffff | undefined.
|
|
|
|
-record(mqtt_packet_connect, {
|
|
client_id = <<>> :: mqtt_client_id(),
|
|
proto_ver = ?MQTT_PROTO_V311 :: mqtt_vsn(),
|
|
proto_name = <<"MQTT">> :: binary(),
|
|
will_retain = false :: boolean(),
|
|
will_qos = ?QOS_0 :: mqtt_qos(),
|
|
will_flag = false :: boolean(),
|
|
clean_sess = false :: boolean(),
|
|
keep_alive = 60 :: non_neg_integer(),
|
|
will_topic = undefined :: undefined | binary(),
|
|
will_msg = undefined :: undefined | binary(),
|
|
username = undefined :: undefined | binary(),
|
|
password = undefined :: undefined | binary()}).
|
|
|
|
-record(mqtt_packet_connack, {
|
|
ack_flags = ?RESERVED :: 0 | 1,
|
|
return_code :: mqtt_connack() }).
|
|
|
|
-record(mqtt_packet_publish, {
|
|
topic_name :: binary(),
|
|
packet_id :: mqtt_packet_id() }).
|
|
|
|
-record(mqtt_packet_puback, {
|
|
packet_id :: mqtt_packet_id() }).
|
|
|
|
-record(mqtt_packet_subscribe, {
|
|
packet_id :: mqtt_packet_id(),
|
|
topic_table :: list({binary(), mqtt_qos()}) }).
|
|
|
|
-record(mqtt_packet_unsubscribe, {
|
|
packet_id :: mqtt_packet_id(),
|
|
topics :: list(binary()) }).
|
|
|
|
-record(mqtt_packet_suback, {
|
|
packet_id :: mqtt_packet_id(),
|
|
qos_table :: list(mqtt_qos() | 128) }).
|
|
|
|
-record(mqtt_packet_unsuback, {
|
|
packet_id :: mqtt_packet_id() }).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% MQTT Control Packet
|
|
%%--------------------------------------------------------------------
|
|
-record(mqtt_packet, {
|
|
header :: #mqtt_packet_header{},
|
|
variable :: #mqtt_packet_connect{} | #mqtt_packet_connack{}
|
|
| #mqtt_packet_publish{} | #mqtt_packet_puback{}
|
|
| #mqtt_packet_subscribe{} | #mqtt_packet_suback{}
|
|
| #mqtt_packet_unsubscribe{} | #mqtt_packet_unsuback{}
|
|
| mqtt_packet_id() | undefined,
|
|
payload :: binary() | undefined }).
|
|
|
|
-type mqtt_packet() :: #mqtt_packet{}.
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% MQTT Packet Match
|
|
%%--------------------------------------------------------------------
|
|
-define(CONNECT_PACKET(Var),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = ?CONNECT}, variable = Var}).
|
|
|
|
-define(CONNACK_PACKET(ReturnCode),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = ?CONNACK},
|
|
variable = #mqtt_packet_connack{return_code = ReturnCode}}).
|
|
|
|
-define(CONNACK_PACKET(ReturnCode, SessPresent),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = ?CONNACK},
|
|
variable = #mqtt_packet_connack{ack_flags = SessPresent,
|
|
return_code = ReturnCode}}).
|
|
|
|
-define(PUBLISH_PACKET(Qos, PacketId),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = ?PUBLISH,
|
|
qos = Qos},
|
|
variable = #mqtt_packet_publish{packet_id = PacketId}}).
|
|
|
|
-define(PUBLISH_PACKET(Qos, Topic, PacketId, Payload),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = ?PUBLISH,
|
|
qos = Qos},
|
|
variable = #mqtt_packet_publish{topic_name = Topic,
|
|
packet_id = PacketId},
|
|
payload = Payload}).
|
|
|
|
-define(PUBACK_PACKET(Type, PacketId),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = Type},
|
|
variable = #mqtt_packet_puback{packet_id = PacketId}}).
|
|
|
|
-define(PUBREL_PACKET(PacketId),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = ?PUBREL, qos = ?QOS_1},
|
|
variable = #mqtt_packet_puback{packet_id = PacketId}}).
|
|
|
|
-define(SUBSCRIBE_PACKET(PacketId, TopicTable),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = ?SUBSCRIBE, qos = ?QOS_1},
|
|
variable = #mqtt_packet_subscribe{packet_id = PacketId,
|
|
topic_table = TopicTable}}).
|
|
-define(SUBACK_PACKET(PacketId, QosTable),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = ?SUBACK},
|
|
variable = #mqtt_packet_suback{packet_id = PacketId,
|
|
qos_table = QosTable}}).
|
|
-define(UNSUBSCRIBE_PACKET(PacketId, Topics),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = ?UNSUBSCRIBE, qos = ?QOS_1},
|
|
variable = #mqtt_packet_unsubscribe{packet_id = PacketId,
|
|
topics = Topics}}).
|
|
-define(UNSUBACK_PACKET(PacketId),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = ?UNSUBACK},
|
|
variable = #mqtt_packet_unsuback{packet_id = PacketId}}).
|
|
|
|
-define(PACKET(Type),
|
|
#mqtt_packet{header = #mqtt_packet_header{type = Type}}).
|
|
|