anlicheng/sdlan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
anlicheng 2026-02-05 22:58:25 +08:00
parent 83361ef1b2
commit 6f2322174d
1 changed files with 47 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
apps/sdlan/src/sdlan_network.erl
View File

@ -399,21 +399,46 @@ handle_cast({policy_request, Sock, {ClientIp, ClientPort}, #sdl_policy_request{c
%% 分片逻辑，index必须从0开始
maybe
{ok, #endpoint{client_id = ClientId, session_token = ST}} ?= maps:find(Mac, Endpoints),
PolicyResponse = sdlan_pb:encode_msg(#sdl_policy_response {
network_id = NetworkId,
src_identity_id = SrcIdentityId,
dst_identity_id = DstIdentityId,
version = Version,
total_num = 1,
index = 0,
rules = <<1, 80:16, 2, 9090:16>>
}),
PolicyResponsePkt = <<?PACKET_POLICY_RESPONSE, PolicyResponse/binary>>,
logger:debug("[sdlan_network] will send policy response: ~p", [PolicyResponsePkt]),
gen_udp:send(Sock, ClientIp, ClientPort, PolicyResponsePkt)
Bin = <<1, 80:16, 2, 9090:16>>,
RulesBin = iolist_to_binary(lists:map(fun(_Id) -> Bin end, lists:seq(1, 1000))),
case byte_size(RulesBin) > 1200 of
true ->
%% 分组
Groups = chunk_rules(RulesBin, 1200),
logger:debug("[sdlan_network] policy_response, groups: ~p", [Groups]),
TotalNum = length(Groups),
Fragments = lists:zip(lists:seq(0, TotalNum - 1), Groups),
lists:foreach(fun({Idx, ChunkRulesBin}) ->
FragmentPolicyResponse = sdlan_pb:encode_msg(#sdl_policy_response {
network_id = NetworkId,
src_identity_id = SrcIdentityId,
dst_identity_id = DstIdentityId,
version = Version,
total_num = TotalNum,
index = Idx,
rules = ChunkRulesBin
}),
FragmentPolicyResponsePkt = <<?PACKET_POLICY_RESPONSE, FragmentPolicyResponse/binary>>,
logger:debug("[sdlan_network] will send policy response: ~p", [FragmentPolicyResponsePkt]),
gen_udp:send(Sock, ClientIp, ClientPort, FragmentPolicyResponsePkt)
end, Fragments);
false ->
%% 小于1200字节不分组
PolicyResponse = sdlan_pb:encode_msg(#sdl_policy_response {
network_id = NetworkId,
src_identity_id = SrcIdentityId,
dst_identity_id = DstIdentityId,
version = Version,
total_num = 1,
index = 0,
rules = RulesBin
}),
PolicyResponsePkt = <<?PACKET_POLICY_RESPONSE, PolicyResponse/binary>>,
logger:debug("[sdlan_network] will send policy response: ~p", [PolicyResponsePkt]),
gen_udp:send(Sock, ClientIp, ClientPort, PolicyResponsePkt)
end
end,
{noreply, State}.
%% @private
@ -595,4 +620,12 @@ endpoint_peers(ExcludeMacs, Endpoints) when is_list(ExcludeMacs), is_map(Endpoin
-spec gen_session_token() -> binary().
gen_session_token() ->
Bytes = crypto:strong_rand_bytes(32),
base64:encode(Bytes).
base64:encode(Bytes).
%% 对rules进行分组
chunk_rules(<<>>, _) ->
[];
chunk_rules(Bin, Size) when byte_size(Bin) =< Size ->
[Bin];
chunk_rules(<<Head:Size/binary, Tail/binary>>, Size) ->
[Head | chunk_rules(Tail, Size)].