mirror of
https://github.com/msgpack/msgpack-c.git
synced 2025-03-28 01:11:14 +01:00
Merge branch 'master' of github.com:msgpack/msgpack
This commit is contained in:
commit
134c27c900
@ -36,7 +36,7 @@ msgpack.beam: msgpack.erl
|
|||||||
erlc $<
|
erlc $<
|
||||||
|
|
||||||
test: msgpack.beam
|
test: msgpack.beam
|
||||||
erl -s msgpack test -s init stop
|
erl -noshell -s msgpack test -s init stop
|
||||||
|
|
||||||
clean:
|
clean:
|
||||||
-rm *.beam
|
-rm *.beam
|
||||||
|
@ -2,18 +2,8 @@ MessagePack for Erlang
|
|||||||
======================
|
======================
|
||||||
Binary-based efficient object serialization library.
|
Binary-based efficient object serialization library.
|
||||||
|
|
||||||
## Status
|
see wiki ( http://redmine.msgpack.org/projects/msgpack/wiki/QuickStartErlang ) for details
|
||||||
|
|
||||||
still in development.
|
# Status
|
||||||
|
|
||||||
TODOs:
|
0.1.0 released.
|
||||||
- decide string specification.
|
|
||||||
|
|
||||||
## Installation
|
|
||||||
|
|
||||||
## Example
|
|
||||||
|
|
||||||
## License
|
|
||||||
|
|
||||||
|
|
||||||
-
|
|
||||||
|
@ -18,38 +18,91 @@
|
|||||||
-module(msgpack).
|
-module(msgpack).
|
||||||
-author('kuenishi+msgpack@gmail.com').
|
-author('kuenishi+msgpack@gmail.com').
|
||||||
|
|
||||||
%% tuples, atoms are not supported. lists, integers, double, and so on.
|
%% tuples, atoms are not supported. lists, integers, double, and so on.
|
||||||
%% see http://msgpack.sourceforge.jp/spec for
|
%% see http://msgpack.sourceforge.jp/spec for
|
||||||
%% supported formats. APIs are almost compatible
|
%% supported formats. APIs are almost compatible
|
||||||
%% for C API (http://msgpack.sourceforge.jp/c:doc)
|
%% for C API (http://msgpack.sourceforge.jp/c:doc)
|
||||||
%% except buffering functions (both copying and zero-copying).
|
%% except buffering functions (both copying and zero-copying).
|
||||||
-export([pack/1, unpack/1, unpack_all/1, test/0]).
|
-export([pack/1, unpack/1, unpack_all/1]).
|
||||||
|
-export([pack_map/1]).
|
||||||
-include_lib("eunit/include/eunit.hrl").
|
|
||||||
|
|
||||||
% compile:
|
% compile:
|
||||||
% erl> c(msgpack).
|
% erl> c(msgpack).
|
||||||
% erl> S = <some term>.
|
% erl> S = <some term>.
|
||||||
% erl> {S, <<>>} = msgpack:unpack( msgpack:pack(S) ).
|
% erl> {S, <<>>} = msgpack:unpack( msgpack:pack(S) ).
|
||||||
|
-type reason() :: enomem | badarg | no_code_matches.
|
||||||
|
-type msgpack_term() :: [msgpack_term()] | {[{msgpack_term(),msgpack_term()}]} | integer() | float().
|
||||||
|
|
||||||
|
% ===== external APIs ===== %
|
||||||
|
-spec pack(Term::msgpack_term()) -> binary().
|
||||||
|
pack(O) when is_integer(O) andalso O < 0 ->
|
||||||
|
pack_int_(O);
|
||||||
|
pack(O) when is_integer(O) ->
|
||||||
|
pack_uint_(O);
|
||||||
|
pack(O) when is_float(O) ->
|
||||||
|
pack_double(O);
|
||||||
|
pack(nil) ->
|
||||||
|
pack_nil();
|
||||||
|
pack(Bool) when is_atom(Bool) ->
|
||||||
|
pack_bool(Bool);
|
||||||
|
pack(Bin) when is_binary(Bin) ->
|
||||||
|
pack_raw(Bin);
|
||||||
|
pack(List) when is_list(List) ->
|
||||||
|
pack_array(List);
|
||||||
|
pack({Map}) when is_list(Map) ->
|
||||||
|
pack_map(Map);
|
||||||
|
pack(Map) when is_tuple(Map), element(1,Map)=:=dict ->
|
||||||
|
pack_map(dict:to_list(Map));
|
||||||
|
pack(_O) ->
|
||||||
|
{error, undefined}.
|
||||||
|
|
||||||
-type reason() :: enomem.
|
% unpacking.
|
||||||
|
% if failed in decoding and not end, get more data
|
||||||
|
% and feed more Bin into this function.
|
||||||
|
% TODO: error case for imcomplete format when short for any type formats.
|
||||||
|
-spec unpack( binary() )->
|
||||||
|
{msgpack_term(), binary()} | {more, non_neg_integer()} | {error, reason()}.
|
||||||
|
unpack(Bin) when not is_binary(Bin)->
|
||||||
|
{error, badarg};
|
||||||
|
unpack(Bin) when bit_size(Bin) >= 8 ->
|
||||||
|
<< Flag:8/unsigned-integer, Payload/binary >> = Bin,
|
||||||
|
unpack_(Flag, Payload);
|
||||||
|
unpack(<<>>)-> % when bit_size(Bin) < 8 ->
|
||||||
|
{more, 1}.
|
||||||
|
|
||||||
|
-spec unpack_all( binary() ) -> [msgpack_term()].
|
||||||
|
unpack_all(Data)->
|
||||||
|
case unpack(Data) of
|
||||||
|
{ Term, Binary } when bit_size(Binary) =:= 0 ->
|
||||||
|
[Term];
|
||||||
|
{ Term, Binary } when is_binary(Binary) ->
|
||||||
|
[Term|unpack_all(Binary)]
|
||||||
|
end.
|
||||||
|
|
||||||
|
pack_map(M)->
|
||||||
|
case length(M) of
|
||||||
|
Len when Len < 16 ->
|
||||||
|
<< 2#1000:4, Len:4/integer-unit:1, (pack_map_(M, <<>>))/binary >>;
|
||||||
|
Len when Len < 16#10000 -> % 65536
|
||||||
|
<< 16#DE:8, Len:16/big-unsigned-integer-unit:1, (pack_map_(M, <<>>))/binary >>;
|
||||||
|
Len ->
|
||||||
|
<< 16#DF:8, Len:32/big-unsigned-integer-unit:1, (pack_map_(M, <<>>))/binary >>
|
||||||
|
end.
|
||||||
|
|
||||||
|
% ===== internal APIs ===== %
|
||||||
|
|
||||||
% positive fixnum
|
% positive fixnum
|
||||||
pack_uint_(N) when is_integer( N ) , N < 128 ->
|
pack_uint_(N) when is_integer( N ) , N < 128 ->
|
||||||
<< 2#0:1, N:7 >>;
|
<< 2#0:1, N:7 >>;
|
||||||
% uint 8
|
% uint 8
|
||||||
pack_uint_( N ) when is_integer( N ) andalso N < 256 ->
|
pack_uint_( N ) when is_integer( N ) andalso N < 256 ->
|
||||||
<< 16#CC:8, N:8 >>;
|
<< 16#CC:8, N:8 >>;
|
||||||
|
|
||||||
% uint 16
|
% uint 16
|
||||||
pack_uint_( N ) when is_integer( N ) andalso N < 65536 ->
|
pack_uint_( N ) when is_integer( N ) andalso N < 65536 ->
|
||||||
<< 16#CD:8, N:16/big-unsigned-integer-unit:1 >>;
|
<< 16#CD:8, N:16/big-unsigned-integer-unit:1 >>;
|
||||||
|
|
||||||
% uint 32
|
% uint 32
|
||||||
pack_uint_( N ) when is_integer( N ) andalso N < 16#FFFFFFFF->
|
pack_uint_( N ) when is_integer( N ) andalso N < 16#FFFFFFFF->
|
||||||
<< 16#CE:8, N:32/big-unsigned-integer-unit:1 >>;
|
<< 16#CE:8, N:32/big-unsigned-integer-unit:1 >>;
|
||||||
|
|
||||||
% uint 64
|
% uint 64
|
||||||
pack_uint_( N ) when is_integer( N )->
|
pack_uint_( N ) when is_integer( N )->
|
||||||
<< 16#CF:8, N:64/big-unsigned-integer-unit:1 >>.
|
<< 16#CF:8, N:64/big-unsigned-integer-unit:1 >>.
|
||||||
@ -58,21 +111,20 @@ pack_uint_( N ) when is_integer( N )->
|
|||||||
pack_int_( N ) when is_integer( N ) , N >= -32->
|
pack_int_( N ) when is_integer( N ) , N >= -32->
|
||||||
<< 2#111:3, N:5 >>;
|
<< 2#111:3, N:5 >>;
|
||||||
% int 8
|
% int 8
|
||||||
pack_int_( N ) when is_integer( N ) , N >= -256 ->
|
pack_int_( N ) when is_integer( N ) , N > -128 ->
|
||||||
<< 16#D0:8, N:8 >>;
|
<< 16#D0:8, N:8/big-signed-integer-unit:1 >>;
|
||||||
% int 16
|
% int 16
|
||||||
pack_int_( N ) when is_integer( N ), N >= -65536 ->
|
pack_int_( N ) when is_integer( N ), N > -32768 ->
|
||||||
<< 16#D1:8, N:16/big-signed-integer-unit:1 >>;
|
<< 16#D1:8, N:16/big-signed-integer-unit:1 >>;
|
||||||
% int 32
|
% int 32
|
||||||
pack_int_( N ) when is_integer( N ), N >= -16#FFFFFFFF ->
|
pack_int_( N ) when is_integer( N ), N > -16#FFFFFFFF ->
|
||||||
<< 16#D2:8, N:32/big-signed-integer-unit:1 >>;
|
<< 16#D2:8, N:32/big-signed-integer-unit:1 >>;
|
||||||
% int 64
|
% int 64
|
||||||
pack_int_( N ) when is_integer( N )->
|
pack_int_( N ) when is_integer( N )->
|
||||||
<< 16#D3:8, N:64/big-signed-integer-unit:1 >>.
|
<< 16#D3:8, N:64/big-signed-integer-unit:1 >>.
|
||||||
|
|
||||||
% nil
|
% nil
|
||||||
pack_nil()->
|
pack_nil()-> << 16#C0:8 >>.
|
||||||
<< 16#C0:8 >>.
|
|
||||||
% pack_true / pack_false
|
% pack_true / pack_false
|
||||||
pack_bool(true)-> << 16#C3:8 >>;
|
pack_bool(true)-> << 16#C3:8 >>;
|
||||||
pack_bool(false)-> << 16#C2:8 >>.
|
pack_bool(false)-> << 16#C2:8 >>.
|
||||||
@ -85,16 +137,12 @@ pack_bool(false)-> << 16#C2:8 >>.
|
|||||||
pack_double(F) when is_float(F)->
|
pack_double(F) when is_float(F)->
|
||||||
<< 16#CB:8, F:64/big-float-unit:1 >>.
|
<< 16#CB:8, F:64/big-float-unit:1 >>.
|
||||||
|
|
||||||
power(N,0) when is_integer(N) -> 1;
|
|
||||||
power(N,D) when is_integer(N) and is_integer(D) -> N * power(N, D-1).
|
|
||||||
|
|
||||||
% raw bytes
|
% raw bytes
|
||||||
pack_raw(Bin) when is_binary(Bin)->
|
pack_raw(Bin) when is_binary(Bin)->
|
||||||
MaxLen = power(2,16),
|
|
||||||
case byte_size(Bin) of
|
case byte_size(Bin) of
|
||||||
Len when Len < 6->
|
Len when Len < 6->
|
||||||
<< 2#101:3, Len:5, Bin/binary >>;
|
<< 2#101:3, Len:5, Bin/binary >>;
|
||||||
Len when Len < MaxLen ->
|
Len when Len < 16#10000 -> % 65536
|
||||||
<< 16#DA:8, Len:16/big-unsigned-integer-unit:1, Bin/binary >>;
|
<< 16#DA:8, Len:16/big-unsigned-integer-unit:1, Bin/binary >>;
|
||||||
Len ->
|
Len ->
|
||||||
<< 16#DB:8, Len:32/big-unsigned-integer-unit:1, Bin/binary >>
|
<< 16#DB:8, Len:32/big-unsigned-integer-unit:1, Bin/binary >>
|
||||||
@ -102,80 +150,53 @@ pack_raw(Bin) when is_binary(Bin)->
|
|||||||
|
|
||||||
% list / tuple
|
% list / tuple
|
||||||
pack_array(L) when is_list(L)->
|
pack_array(L) when is_list(L)->
|
||||||
MaxLen = power(2,16),
|
|
||||||
case length(L) of
|
case length(L) of
|
||||||
Len when Len < 16 ->
|
Len when Len < 16 ->
|
||||||
<< 2#1001:4, Len:4/integer-unit:1, (pack_array_(L))/binary >>;
|
<< 2#1001:4, Len:4/integer-unit:1, (pack_array_(L, <<>>))/binary >>;
|
||||||
Len when Len < MaxLen ->
|
Len when Len < 16#10000 -> % 65536
|
||||||
<< 16#DC:8, Len:16/big-unsigned-integer-unit:1,(pack_array_(L))/binary >>;
|
<< 16#DC:8, Len:16/big-unsigned-integer-unit:1,(pack_array_(L, <<>>))/binary >>;
|
||||||
Len ->
|
Len ->
|
||||||
<< 16#DD:8, Len:32/big-unsigned-integer-unit:1,(pack_array_(L))/binary >>
|
<< 16#DD:8, Len:32/big-unsigned-integer-unit:1,(pack_array_(L, <<>>))/binary >>
|
||||||
end.
|
end.
|
||||||
pack_array_([])-> <<>>;
|
pack_array_([], Acc) -> Acc;
|
||||||
pack_array_([Head|Tail])->
|
pack_array_([Head|Tail], Acc) ->
|
||||||
<< (pack_object(Head))/binary, (pack_array_(Tail))/binary >>.
|
pack_array_(Tail, <<Acc/binary, (pack(Head))/binary>>).
|
||||||
|
|
||||||
unpack_array_(<<>>, 0)-> [];
|
% FIXME! this should be tail-recursive and without lists:reverse/1
|
||||||
unpack_array_(Remain, 0) when is_binary(Remain)-> [Remain];
|
unpack_array_(<<>>, 0, RetList) -> {lists:reverse(RetList), <<>>};
|
||||||
unpack_array_(Bin, RestLen) when is_binary(Bin)->
|
unpack_array_(Remain, 0, RetList) when is_binary(Remain)-> {lists:reverse(RetList), Remain};
|
||||||
{Term, Rest} = unpack(Bin),
|
unpack_array_(<<>>, RestLen, _RetList) when RestLen > 0 -> {more, RestLen};
|
||||||
[Term|unpack_array_(Rest, RestLen-1)].
|
unpack_array_(Bin, RestLen, RetList) when is_binary(Bin)->
|
||||||
|
case unpack(Bin) of
|
||||||
pack_map({dict,M})->
|
{more, Len} -> {more, Len+RestLen-1};
|
||||||
MaxLen = power(2,16),
|
{Term, Rest}-> unpack_array_(Rest, RestLen-1, [Term|RetList])
|
||||||
case dict:size(M) of
|
|
||||||
Len when Len < 16 ->
|
|
||||||
<< 2#1001:4, Len:4/integer-unit:1, (pack_map_(dict:to_list(M))) >>;
|
|
||||||
Len when Len < MaxLen ->
|
|
||||||
<< 16#DE:8, Len:16/big-unsigned-integer-unit:1, (pack_map_(dict:to_list(M))) >>;
|
|
||||||
Len ->
|
|
||||||
<< 16#DF:8, Len:32/big-unsigned-integer-unit:1, (pack_map_(dict:to_list(M))) >>
|
|
||||||
end.
|
end.
|
||||||
|
|
||||||
pack_map_([])-> <<>>;
|
pack_map_([], Acc) -> Acc;
|
||||||
pack_map_([{Key,Value}|Tail]) ->
|
pack_map_([{Key,Value}|Tail], Acc) ->
|
||||||
<< (pack_object(Key)),(pack_object(Value)),(pack_map_(Tail)) >>.
|
pack_map_(Tail, << Acc/binary, (pack(Key))/binary, (pack(Value))/binary>>).
|
||||||
|
|
||||||
unpack_map_(<<>>, 0)-> [];
|
% FIXME: write test for unpack_map/1
|
||||||
unpack_map_(Bin, 0) when is_binary(Bin)-> [Bin];
|
-spec unpack_map_(binary(), non_neg_integer(), [{term(), msgpack_term()}])->
|
||||||
unpack_map_(Bin, Len) when is_binary(Bin) and is_integer(Len) ->
|
{more, non_neg_integer()} | { any(), binary()}.
|
||||||
{ Key, Rest } = unpack(Bin),
|
unpack_map_(Bin, 0, Acc) when is_binary(Bin) -> {{lists:reverse(Acc)}, Bin};
|
||||||
{ Value, Rest2 } = unpack(Rest),
|
unpack_map_(Bin, Len, Acc) when is_binary(Bin) and is_integer(Len) ->
|
||||||
[{Key,Value}|unpack_map_(Rest2,Len-1)].
|
case unpack(Bin) of
|
||||||
|
{ more, MoreLen } -> { more, MoreLen+Len-1 };
|
||||||
|
{ Key, Rest } ->
|
||||||
|
case unpack(Rest) of
|
||||||
|
{more, MoreLen} -> { more, MoreLen+Len-1 };
|
||||||
|
{ Value, Rest2 } ->
|
||||||
|
unpack_map_(Rest2,Len-1,[{Key,Value}|Acc])
|
||||||
|
end
|
||||||
|
end.
|
||||||
|
|
||||||
pack_object(O) when is_integer(O) andalso O < 0 ->
|
% {more, <remaining byte size>
|
||||||
pack_int_(O);
|
-spec unpack_(Flag::integer(), Payload::binary())->
|
||||||
pack_object(O) when is_integer(O) ->
|
{more, pos_integer()} | {msgpack_term(), binary()} | {error, reason()}.
|
||||||
pack_uint_(O);
|
unpack_(Flag, Payload)->
|
||||||
pack_object(O) when is_float(O)->
|
PayloadLen = byte_size(Payload),
|
||||||
pack_double(O);
|
case Flag of
|
||||||
pack_object(nil) ->
|
|
||||||
pack_nil();
|
|
||||||
pack_object(Bool) when is_atom(Bool) ->
|
|
||||||
pack_bool(Bool);
|
|
||||||
pack_object(Bin) when is_binary(Bin)->
|
|
||||||
pack_raw(Bin);
|
|
||||||
pack_object(List) when is_list(List)->
|
|
||||||
pack_array(List);
|
|
||||||
pack_object({dict, Map})->
|
|
||||||
pack_map({dict, Map});
|
|
||||||
pack_object(_) ->
|
|
||||||
undefined.
|
|
||||||
|
|
||||||
pack(Obj)->
|
|
||||||
pack_object(Obj).
|
|
||||||
|
|
||||||
|
|
||||||
% unpacking.
|
|
||||||
% if failed in decoding and not end, get more data
|
|
||||||
% and feed more Bin into this function.
|
|
||||||
% TODO: error case for imcomplete format when short for any type formats.
|
|
||||||
-spec unpack( binary() )-> {term(), binary()} | {more, non_neg_integer()} | {error, reason()}.
|
|
||||||
unpack(Bin) when not is_binary(Bin)->
|
|
||||||
{error, badard};
|
|
||||||
unpack(Bin) when bit_size(Bin) >= 8 ->
|
|
||||||
<< Flag:8/unsigned-integer, Payload/binary >> = Bin,
|
|
||||||
case Flag of
|
|
||||||
16#C0 ->
|
16#C0 ->
|
||||||
{nil, Payload};
|
{nil, Payload};
|
||||||
16#C2 ->
|
16#C2 ->
|
||||||
@ -183,86 +204,101 @@ unpack(Bin) when bit_size(Bin) >= 8 ->
|
|||||||
16#C3 ->
|
16#C3 ->
|
||||||
{true, Payload};
|
{true, Payload};
|
||||||
|
|
||||||
16#CA -> % 32bit float
|
16#CA when PayloadLen >= 4 -> % 32bit float
|
||||||
<< Return:32/float-unit:1, Rest/binary >> = Payload,
|
<< Return:32/float-unit:1, Rest/binary >> = Payload,
|
||||||
{Return, Rest};
|
{Return, Rest};
|
||||||
16#CB -> % 64bit float
|
16#CA ->
|
||||||
|
{more, 4-PayloadLen}; % at least more
|
||||||
|
|
||||||
|
16#CB when PayloadLen >= 8 -> % 64bit float
|
||||||
<< Return:64/float-unit:1, Rest/binary >> = Payload,
|
<< Return:64/float-unit:1, Rest/binary >> = Payload,
|
||||||
{Return, Rest};
|
{Return, Rest};
|
||||||
|
16#CB ->
|
||||||
|
{more, 8-PayloadLen};
|
||||||
|
|
||||||
16#CC -> % uint 8
|
16#CC when PayloadLen >= 1 -> % uint 8
|
||||||
<< Int:8/unsigned-integer, Rest/binary >> = Payload,
|
<< Int:8/unsigned-integer, Rest/binary >> = Payload,
|
||||||
{Int, Rest};
|
{Int, Rest};
|
||||||
16#CD -> % uint 16
|
16#CC ->
|
||||||
|
{more, 1};
|
||||||
|
|
||||||
|
16#CD when PayloadLen >= 2 -> % uint 16
|
||||||
<< Int:16/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
<< Int:16/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
||||||
{Int, Rest};
|
{Int, Rest};
|
||||||
16#CE ->
|
16#CD ->
|
||||||
|
{more, 2-PayloadLen};
|
||||||
|
|
||||||
|
16#CE when PayloadLen >= 4 ->
|
||||||
<< Int:32/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
<< Int:32/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
||||||
{Int, Rest};
|
{Int, Rest};
|
||||||
16#CF ->
|
16#CE ->
|
||||||
|
{more, 4-PayloadLen}; % at least more
|
||||||
|
|
||||||
|
16#CF when PayloadLen >= 8 ->
|
||||||
<< Int:64/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
<< Int:64/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
||||||
{Int, Rest};
|
{Int, Rest};
|
||||||
|
16#CF ->
|
||||||
|
{more, 8-PayloadLen};
|
||||||
|
|
||||||
16#D0 -> % int 8
|
16#D0 when PayloadLen >= 1 -> % int 8
|
||||||
<< Int:8/big-signed-integer-unit:1, Rest/binary >> = Payload,
|
<< Int:8/big-signed-integer-unit:1, Rest/binary >> = Payload,
|
||||||
{Int, Rest};
|
{Int, Rest};
|
||||||
16#D1 -> % int 16
|
16#D0 ->
|
||||||
|
{more, 1};
|
||||||
|
|
||||||
|
16#D1 when PayloadLen >= 2 -> % int 16
|
||||||
<< Int:16/big-signed-integer-unit:1, Rest/binary >> = Payload,
|
<< Int:16/big-signed-integer-unit:1, Rest/binary >> = Payload,
|
||||||
{Int, Rest};
|
{Int, Rest};
|
||||||
16#D2 -> % int 32
|
16#D1 ->
|
||||||
|
{more, 2-PayloadLen};
|
||||||
|
|
||||||
|
16#D2 when PayloadLen >= 4 -> % int 32
|
||||||
<< Int:32/big-signed-integer-unit:1, Rest/binary >> = Payload,
|
<< Int:32/big-signed-integer-unit:1, Rest/binary >> = Payload,
|
||||||
{Int, Rest};
|
{Int, Rest};
|
||||||
16#D3 -> % int 64
|
16#D2 ->
|
||||||
|
{more, 4-PayloadLen};
|
||||||
|
|
||||||
|
16#D3 when PayloadLen >= 8 -> % int 64
|
||||||
<< Int:64/big-signed-integer-unit:1, Rest/binary >> = Payload,
|
<< Int:64/big-signed-integer-unit:1, Rest/binary >> = Payload,
|
||||||
{Int, Rest};
|
{Int, Rest};
|
||||||
16#DA -> % raw 16
|
16#D3 ->
|
||||||
|
{more, 8-PayloadLen};
|
||||||
|
|
||||||
|
16#DA when PayloadLen >= 2 -> % raw 16
|
||||||
<< Len:16/unsigned-integer-unit:1, Rest/binary >> = Payload,
|
<< Len:16/unsigned-integer-unit:1, Rest/binary >> = Payload,
|
||||||
<< Return:Len/binary, Remain/binary >> = Rest,
|
<< Return:Len/binary, Remain/binary >> = Rest,
|
||||||
{Return, Remain};
|
{Return, Remain};
|
||||||
16#DB -> % raw 32
|
16#DA ->
|
||||||
|
{more, 16-PayloadLen};
|
||||||
|
|
||||||
|
16#DB when PayloadLen >= 4 -> % raw 32
|
||||||
<< Len:32/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
<< Len:32/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
||||||
<< Return:Len/binary, Remain/binary >> = Rest,
|
<< Return:Len/binary, Remain/binary >> = Rest,
|
||||||
{Return, Remain};
|
{Return, Remain};
|
||||||
16#DC -> % array 16
|
16#DB ->
|
||||||
|
{more, 4-PayloadLen};
|
||||||
|
|
||||||
|
16#DC when PayloadLen >= 2 -> % array 16
|
||||||
<< Len:16/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
<< Len:16/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
||||||
Array=unpack_array_(Rest, Len),
|
unpack_array_(Rest, Len, []);
|
||||||
case length(Array) of
|
16#DC ->
|
||||||
Len -> {Array, <<>>};
|
{more, 2-PayloadLen};
|
||||||
_ ->
|
|
||||||
{Return, RemainRest} = lists:split(Len, Array),
|
16#DD when PayloadLen >= 4 -> % array 32
|
||||||
[Remain] = RemainRest,
|
|
||||||
{Return, Remain}
|
|
||||||
end;
|
|
||||||
16#DD -> % array 32
|
|
||||||
<< Len:32/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
<< Len:32/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
||||||
Array=unpack_array_(Rest, Len),
|
unpack_array_(Rest, Len, []);
|
||||||
case length(Array) of
|
16#DD ->
|
||||||
Len -> {Array, <<>>};
|
{more, 4-PayloadLen};
|
||||||
_ ->
|
|
||||||
{Return, RemainRest} = lists:split(Len, Array),
|
16#DE when PayloadLen >= 2 -> % map 16
|
||||||
[Remain] = RemainRest,
|
|
||||||
{Return, Remain}
|
|
||||||
end;
|
|
||||||
16#DE -> % map 16
|
|
||||||
<< Len:16/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
<< Len:16/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
||||||
Array=unpack_map_(Rest, Len),
|
unpack_map_(Rest, Len, []);
|
||||||
case length(Array) of
|
16#DE ->
|
||||||
Len -> { dict:from_list(Array), <<>>};
|
{more, 2-PayloadLen};
|
||||||
_ ->
|
|
||||||
{Return, RemainRest} = lists:split(Len, Array),
|
16#DF when PayloadLen >= 4 -> % map 32
|
||||||
[Remain] = RemainRest,
|
|
||||||
{dict:from_list(Return), Remain}
|
|
||||||
end;
|
|
||||||
16#DF -> % map 32
|
|
||||||
<< Len:32/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
<< Len:32/big-unsigned-integer-unit:1, Rest/binary >> = Payload,
|
||||||
Array=unpack_map_(Rest, Len),
|
unpack_map_(Rest, Len, []);
|
||||||
case length(Array) of
|
|
||||||
Len -> { dict:from_list(Array), <<>>};
|
|
||||||
_ ->
|
|
||||||
{Return, RemainRest} = lists:split(Len, Array),
|
|
||||||
[Remain] = RemainRest,
|
|
||||||
{dict:from_list(Return), Remain}
|
|
||||||
end;
|
|
||||||
|
|
||||||
% positive fixnum
|
% positive fixnum
|
||||||
Code when Code >= 2#00000000, Code < 2#10000000->
|
Code when Code >= 2#00000000, Code < 2#10000000->
|
||||||
@ -273,7 +309,7 @@ unpack(Bin) when bit_size(Bin) >= 8 ->
|
|||||||
{(Code - 16#100), Payload};
|
{(Code - 16#100), Payload};
|
||||||
|
|
||||||
Code when Code >= 2#10100000 , Code < 2#11000000 ->
|
Code when Code >= 2#10100000 , Code < 2#11000000 ->
|
||||||
% 101XXXXX for FixRaw
|
% 101XXXXX for FixRaw
|
||||||
Len = Code rem 2#10100000,
|
Len = Code rem 2#10100000,
|
||||||
<< Return:Len/binary, Remain/binary >> = Payload,
|
<< Return:Len/binary, Remain/binary >> = Payload,
|
||||||
{Return, Remain};
|
{Return, Remain};
|
||||||
@ -281,52 +317,39 @@ unpack(Bin) when bit_size(Bin) >= 8 ->
|
|||||||
Code when Code >= 2#10010000 , Code < 2#10100000 ->
|
Code when Code >= 2#10010000 , Code < 2#10100000 ->
|
||||||
% 1001XXXX for FixArray
|
% 1001XXXX for FixArray
|
||||||
Len = Code rem 2#10010000,
|
Len = Code rem 2#10010000,
|
||||||
Array=unpack_array_(Payload, Len),
|
unpack_array_(Payload, Len, []);
|
||||||
case length(Array) of
|
|
||||||
Len -> { Array, <<>>};
|
|
||||||
_ ->
|
|
||||||
{Return, RemainRest} = lists:split(Len, Array),
|
|
||||||
[Remain] = RemainRest,
|
|
||||||
{Return, Remain}
|
|
||||||
end;
|
|
||||||
|
|
||||||
Code when Code >= 2#10000000 , Code < 2#10010000 ->
|
Code when Code >= 2#10000000 , Code < 2#10010000 ->
|
||||||
% 1000XXXX for FixMap
|
% 1000XXXX for FixMap
|
||||||
Len = Code rem 2#10000000,
|
Len = Code rem 2#10000000,
|
||||||
Array=unpack_map_(Payload, Len),
|
unpack_map_(Payload, Len, []);
|
||||||
case length(Array) of
|
|
||||||
Len -> { dict:from_list(Array), <<>>};
|
|
||||||
_ ->
|
|
||||||
{Return, RemainRest} = lists:split(Len, Array),
|
|
||||||
[Remain] = RemainRest,
|
|
||||||
{dict:from_list(Return), Remain}
|
|
||||||
end;
|
|
||||||
|
|
||||||
_Other ->
|
_Other ->
|
||||||
erlang:display(_Other),
|
|
||||||
{error, no_code_matches}
|
{error, no_code_matches}
|
||||||
end;
|
|
||||||
unpack(_)-> % when bit_size(Bin) < 8 ->
|
|
||||||
{more, 8}.
|
|
||||||
|
|
||||||
unpack_all(Data)->
|
|
||||||
case unpack(Data) of
|
|
||||||
{ Term, Binary } when bit_size(Binary) =:= 0 ->
|
|
||||||
[Term];
|
|
||||||
{ Term, Binary } when is_binary(Binary) ->
|
|
||||||
[Term|unpack_all(Binary)]
|
|
||||||
end.
|
end.
|
||||||
|
|
||||||
|
% ===== test codes ===== %
|
||||||
|
-include_lib("eunit/include/eunit.hrl").
|
||||||
-ifdef(EUNIT).
|
-ifdef(EUNIT).
|
||||||
|
|
||||||
|
compare_all([], [])-> ok;
|
||||||
|
compare_all([], R)-> {toomuchrhs, R};
|
||||||
|
compare_all(L, [])-> {toomuchlhs, L};
|
||||||
|
compare_all([LH|LTL], [RH|RTL]) ->
|
||||||
|
LH=RH,
|
||||||
|
compare_all(LTL, RTL).
|
||||||
|
|
||||||
test_data()->
|
test_data()->
|
||||||
[0, 1, 2, 123, 512, 1230, 678908, 16#FFFFFFFFFF,
|
[true, false, nil,
|
||||||
|
0, 1, 2, 123, 512, 1230, 678908, 16#FFFFFFFFFF,
|
||||||
-1, -23, -512, -1230, -567898, -16#FFFFFFFFFF,
|
-1, -23, -512, -1230, -567898, -16#FFFFFFFFFF,
|
||||||
123.123, -234.4355, 1.0e-34, 1.0e64,
|
123.123, -234.4355, 1.0e-34, 1.0e64,
|
||||||
[23, 234, 0.23],
|
[23, 234, 0.23],
|
||||||
"hogehoge", "243546rf7g68h798j",
|
<<"hogehoge">>, <<"243546rf7g68h798j", 0, 23, 255>>,
|
||||||
<<"hoasfdafdas][">>,
|
<<"hoasfdafdas][">>,
|
||||||
[0,42,"sum", [1,2]], [1,42, nil, [3]]
|
[0,42, <<"sum">>, [1,2]], [1,42, nil, [3]],
|
||||||
|
-234, -40000, -16#10000000, -16#100000000,
|
||||||
|
42
|
||||||
].
|
].
|
||||||
|
|
||||||
basic_test()->
|
basic_test()->
|
||||||
@ -339,34 +362,66 @@ port_test()->
|
|||||||
{[Tests],<<>>} = msgpack:unpack(msgpack:pack([Tests])),
|
{[Tests],<<>>} = msgpack:unpack(msgpack:pack([Tests])),
|
||||||
Port = open_port({spawn, "ruby ../test/crosslang.rb"}, [binary]),
|
Port = open_port({spawn, "ruby ../test/crosslang.rb"}, [binary]),
|
||||||
true = port_command(Port, msgpack:pack(Tests) ),
|
true = port_command(Port, msgpack:pack(Tests) ),
|
||||||
%Port ! {self, {command, msgpack:pack(Tests)}}, ... not owner
|
|
||||||
receive
|
receive
|
||||||
{Port, {data, Data}}-> {Tests, <<>>}=msgpack:unpack(Data)
|
{Port, {data, Data}}-> {Tests, <<>>}=msgpack:unpack(Data)
|
||||||
after 1024-> ?assert(false) end,
|
after 1024-> ?assert(false) end,
|
||||||
port_close(Port).
|
port_close(Port).
|
||||||
|
|
||||||
|
test_p(Len,Term,OrigBin,Len) ->
|
||||||
|
{Term, <<>>}=msgpack:unpack(OrigBin);
|
||||||
|
test_p(I,_,OrigBin,Len) when I < Len->
|
||||||
|
<<Bin:I/binary, _/binary>> = OrigBin,
|
||||||
|
{more, N}=msgpack:unpack(Bin),
|
||||||
|
?assert(0 < N),
|
||||||
|
?assert(N < Len).
|
||||||
|
|
||||||
|
partial_test()-> % error handling test.
|
||||||
|
Term = lists:seq(0, 45),
|
||||||
|
Bin=msgpack:pack(Term),
|
||||||
|
BinLen = byte_size(Bin),
|
||||||
|
[test_p(X, Term, Bin, BinLen) || X <- lists:seq(0,BinLen)].
|
||||||
|
|
||||||
|
long_test()->
|
||||||
|
Longer = lists:seq(0, 655),
|
||||||
|
% Longest = lists:seq(0,12345),
|
||||||
|
{Longer, <<>>} = msgpack:unpack(msgpack:pack(Longer)),
|
||||||
|
% {Longest, <<>>} = msgpack:unpack(msgpack:pack(Longest)).
|
||||||
|
ok.
|
||||||
|
|
||||||
|
map_test()->
|
||||||
|
Ints = lists:seq(0, 65),
|
||||||
|
Map = {[ {X, X*2} || X <- Ints ] ++ [{<<"hage">>, 324}, {43542, [nil, true, false]}]},
|
||||||
|
{Map2, <<>>} = msgpack:unpack(msgpack:pack(Map)),
|
||||||
|
?assertEqual(Map, Map2),
|
||||||
|
ok.
|
||||||
|
|
||||||
unknown_test()->
|
unknown_test()->
|
||||||
Tests = [0, 1, 2, 123, 512, 1230, 678908,
|
Tests = [0, 1, 2, 123, 512, 1230, 678908,
|
||||||
-1, -23, -512, -1230, -567898,
|
-1, -23, -512, -1230, -567898,
|
||||||
% "hogehoge", "243546rf7g68h798j",
|
<<"hogehoge">>, <<"243546rf7g68h798j">>,
|
||||||
123.123 %-234.4355, 1.0e-34, 1.0e64,
|
123.123,
|
||||||
% [23, 234, 0.23]
|
-234.4355, 1.0e-34, 1.0e64,
|
||||||
% [0,42,"sum", [1,2]], [1,42, nil, [3]]
|
[23, 234, 0.23],
|
||||||
|
[0,42,<<"sum">>, [1,2]], [1,42, nil, [3]],
|
||||||
|
{[{1,2},{<<"hoge">>,nil}]},
|
||||||
|
-234, -50000,
|
||||||
|
42
|
||||||
],
|
],
|
||||||
Port = open_port({spawn, "ruby testcase_generator.rb"}, [binary]),
|
Port = open_port({spawn, "ruby testcase_generator.rb"}, [binary]),
|
||||||
%Port ! {self, {command, msgpack:pack(Tests)}}, ... not owner
|
|
||||||
receive
|
receive
|
||||||
{Port, {data, Data}}->
|
{Port, {data, Data}}->
|
||||||
Tests=msgpack:unpack_all(Data)
|
compare_all(Tests, msgpack:unpack_all(Data))
|
||||||
% io:format("~p~n", [Tests])
|
|
||||||
after 1024-> ?assert(false) end,
|
after 1024-> ?assert(false) end,
|
||||||
port_close(Port).
|
port_close(Port).
|
||||||
|
|
||||||
test_([]) -> 0;
|
test_([]) -> 0;
|
||||||
test_([S|Rest])->
|
test_([Before|Rest])->
|
||||||
Pack = msgpack:pack(S),
|
Pack = msgpack:pack(Before),
|
||||||
% io:format("testing: ~p => ~p~n", [S, Pack]),
|
{After, <<>>} = msgpack:unpack( Pack ),
|
||||||
{S, <<>>} = msgpack:unpack( Pack ),
|
?assertEqual(Before, After),
|
||||||
1+test_(Rest).
|
1+test_(Rest).
|
||||||
|
|
||||||
|
other_test()->
|
||||||
|
{more,1}=msgpack:unpack(<<>>).
|
||||||
|
|
||||||
-endif.
|
-endif.
|
||||||
|
@ -39,10 +39,14 @@ end
|
|||||||
|
|
||||||
objs = [0, 1, 2, 123, 512, 1230, 678908,
|
objs = [0, 1, 2, 123, 512, 1230, 678908,
|
||||||
-1, -23, -512, -1230, -567898,
|
-1, -23, -512, -1230, -567898,
|
||||||
# "hogehoge", "243546rf7g68h798j",
|
"hogehoge", "243546rf7g68h798j",
|
||||||
123.123, #-234.4355, 1.0e-34, 1.0e64,
|
123.123,
|
||||||
# [23, 234, 0.23]
|
-234.4355, 1.0e-34, 1.0e64,
|
||||||
# [0,42,"sum", [1,2]], [1,42, nil, [3]]
|
[23, 234, 0.23],
|
||||||
|
[0,42,"sum", [1,2]], [1,42, nil, [3]],
|
||||||
|
{ 1 => 2, "hoge" => nil },
|
||||||
|
-234, -50000,
|
||||||
|
42
|
||||||
]
|
]
|
||||||
begin
|
begin
|
||||||
objs.each do |obj|
|
objs.each do |obj|
|
||||||
|
@ -1,3 +1,10 @@
|
|||||||
# coding: utf-8
|
# coding: utf-8
|
||||||
from _msgpack import *
|
from _msgpack import *
|
||||||
|
|
||||||
|
# alias for compatibility to simplejson/marshal/pickle.
|
||||||
|
load = unpack
|
||||||
|
loads = unpackb
|
||||||
|
|
||||||
|
dump = pack
|
||||||
|
dumps = packb
|
||||||
|
|
||||||
|
@ -14,7 +14,7 @@ except ImportError:
|
|||||||
from distutils.command.build_ext import build_ext
|
from distutils.command.build_ext import build_ext
|
||||||
have_cython = False
|
have_cython = False
|
||||||
|
|
||||||
version = '0.1.3'
|
version = '0.1.4'
|
||||||
|
|
||||||
# take care of extension modules.
|
# take care of extension modules.
|
||||||
if have_cython:
|
if have_cython:
|
||||||
|
Loading…
x
Reference in New Issue
Block a user