yuanbiao
/
emqx
1
0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

format

This commit is contained in:
Ery Lee 2015-03-08 18:15:30 +08:00
parent b3178ffcd5
commit 00f39607f1
2 changed files with 49 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
apps/emqtt/src/emqtt_parser.erl
View File

@ -33,7 +33,6 @@
%% API %% API
-export([init/0, parse/2]). -export([init/0, parse/2]).
%%%----------------------------------------------------------------------------- %%%-----------------------------------------------------------------------------
%% @doc %% @doc
%% Initialize a parser. %% Initialize a parser.

28
apps/emqtt/src/emqtt_serialiser.erl
View File

@ -33,6 +33,13 @@
%% API %% API
-export([serialise/1]). -export([serialise/1]).
%%------------------------------------------------------------------------------
%% @doc
%% Serialise MQTT Packet.
%%
%% @end
%%------------------------------------------------------------------------------
-spec serialise(mqtt_packet()) -> binary().
serialise(#mqtt_packet{header = Header = #mqtt_packet_header{type = Type}, serialise(#mqtt_packet{header = Header = #mqtt_packet_header{type = Type},
variable = Variable, variable = Variable,
payload = Payload}) -> payload = Payload}) ->
@ -44,13 +51,14 @@ serialise_header(#mqtt_packet_header{type = Type,
dup = Dup, dup = Dup,
qos = Qos, qos = Qos,
retain = Retain}, retain = Retain},
{VariableBin, PayloadBin}) {VariableBin, PayloadBin}) when ?CONNECT =< Type andalso Type =< ?DISCONNECT ->
when is_integer(Type) andalso ?CONNECT =< Type andalso Type =< ?DISCONNECT ->
Len = size(VariableBin) + size(PayloadBin), Len = size(VariableBin) + size(PayloadBin),
true = (Len =< ?MAX_LEN), true = (Len =< ?MAX_LEN),
LenBin = serialise_len(Len), LenBin = serialise_len(Len),
<<Type:4, (opt(Dup)):1, (opt(Qos)):2, (opt(Retain)):1, <<Type:4, (opt(Dup)):1, (opt(Qos)):2, (opt(Retain)):1,
LenBin/binary, VariableBin/binary, PayloadBin/binary>>. LenBin/binary,
VariableBin/binary,
PayloadBin/binary>>.
serialise_variable(?CONNECT, #mqtt_packet_connect{client_id = ClientId, serialise_variable(?CONNECT, #mqtt_packet_connect{client_id = ClientId,
proto_ver = ProtoVer, proto_ver = ProtoVer,
@ -87,8 +95,7 @@ serialise_variable(?CONNECT, #mqtt_packet_connect{client_id = ClientId,
{VariableBin, <<PayloadBin1/binary, UserPasswd/binary>>}; {VariableBin, <<PayloadBin1/binary, UserPasswd/binary>>};
serialise_variable(?CONNACK, #mqtt_packet_connack{ack_flags = AckFlags, serialise_variable(?CONNACK, #mqtt_packet_connack{ack_flags = AckFlags,
return_code = ReturnCode}, return_code = ReturnCode}, undefined) ->
undefined) ->
{<<AckFlags:8, ReturnCode:8>>, <<>>}; {<<AckFlags:8, ReturnCode:8>>, <<>>};
serialise_variable(?SUBSCRIBE, #mqtt_packet_subscribe{packet_id = PacketId, serialise_variable(?SUBSCRIBE, #mqtt_packet_subscribe{packet_id = PacketId,
@ -96,16 +103,14 @@ serialise_variable(?SUBSCRIBE, #mqtt_packet_subscribe{packet_id = PacketId,
{<<PacketId:16/big>>, serialise_topics(Topics)}; {<<PacketId:16/big>>, serialise_topics(Topics)};
serialise_variable(?SUBACK, #mqtt_packet_suback{packet_id = PacketId, serialise_variable(?SUBACK, #mqtt_packet_suback{packet_id = PacketId,
qos_table = QosTable}, qos_table = QosTable}, undefined) ->
undefined) ->
{<<PacketId:16/big>>, << <<Q:8>> || Q <- QosTable >>}; {<<PacketId:16/big>>, << <<Q:8>> || Q <- QosTable >>};
serialise_variable(?UNSUBSCRIBE, #mqtt_packet_unsubscribe{ serialise_variable(?UNSUBSCRIBE, #mqtt_packet_unsubscribe{packet_id = PacketId,
packet_id = PacketId, topics = Topics }, undefined) -> topics = Topics }, undefined) ->
{<<PacketId:16/big>>, serialise_topics(Topics)}; {<<PacketId:16/big>>, serialise_topics(Topics)};
serialise_variable(?UNSUBACK, #mqtt_packet_unsuback{packet_id = PacketId}, serialise_variable(?UNSUBACK, #mqtt_packet_unsuback{packet_id = PacketId}, undefined) ->
undefined) ->
{<<PacketId:16/big>>, <<>>}; {<<PacketId:16/big>>, <<>>};
serialise_variable(?PUBLISH, #mqtt_packet_publish{topic_name = TopicName, serialise_variable(?PUBLISH, #mqtt_packet_publish{topic_name = TopicName,
@ -157,3 +162,4 @@ opt(false) -> 0;
opt(true) -> 1; opt(true) -> 1;
opt(X) when is_integer(X) -> X; opt(X) when is_integer(X) -> X;
opt(B) when is_binary(B) -> 1. opt(B) when is_binary(B) -> 1.