tres/secure_channel: Fix to avoid crash when receive malformed packet

This commit is contained in:
Eishun Kondoh 2018-05-02 15:39:58 +09:00
parent bc3c65cfa9
commit 6bba512271
5 changed files with 66 additions and 29 deletions

View file

@ -6,13 +6,20 @@
-export([insert/3, insert/4,
update/3, get/2,
delete/2, is_exists/2,
is_empty/1]).
is_empty/1, aged_out/1]).
-define(AGE_TIME_MS, 1000).
-define(TABLE, xact_kv).
-define(ENTRY, xact_entry).
-define(TABLE_OPTS, [set, protected, {keypos, #?ENTRY.xid}]).
-record(?ENTRY, {xid = 0, pending = nil, orig = nil, from = nil}).
-record(?ENTRY, {
xid = 0,
pending = nil,
orig = nil,
from = nil,
inserted_at = 0
}).
-spec create() -> reference().
create() ->
@ -24,7 +31,11 @@ drop(Tid) ->
-spec insert(reference(), integer(), map()) -> true.
insert(Tid, Xid, Orig) ->
ets:insert(Tid, #?ENTRY{xid = Xid, orig = Orig}).
ets:insert(Tid, #?ENTRY{
xid = Xid,
orig = Orig,
inserted_at = os:system_time(milli_seconds)
}).
-spec insert(reference(), integer(), map(), term()) -> true.
insert(Tid, Xid, Orig, From) ->
@ -58,8 +69,17 @@ is_empty(Tid) ->
_ -> false
end.
-spec aged_out(reference()) -> integer().
aged_out(Tid) ->
MatchSpec = ms_for_aged_entries(),
ets:select_delete(Tid, MatchSpec).
%% Private functions
ms_for_aged_entries() ->
Now = os:system_time(milli_seconds),
ets:fun2ms(fun(#?ENTRY{inserted_at = T1}) -> Now - T1 > ?AGE_TIME_MS end).
ms_for_exists(Xid) ->
ets:fun2ms(fun(#?ENTRY{xid = TXid}) when TXid == Xid -> true end).