fix

apps/sdlan/src/sdlan_network.erl

@@ -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>>,
+        Bin = <<1, 80:16, 2, 9090:16>>,
-        logger:debug("[sdlan_network] will send policy response: ~p", [PolicyResponsePkt]),
+        RulesBin = iolist_to_binary(lists:map(fun(_Id) -> Bin end, lists:seq(1, 1000))),
-        gen_udp:send(Sock, ClientIp, ClientPort, PolicyResponsePkt)
+        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)].