Explorar o código

Merge pull request #12673 from zmstone/0308-fix-base64-compatibility-on-otp25

fix(mgmt): avoid using base64:decode/2 and encode/2
SergeTupchiy %!s(int64=2) %!d(string=hai) anos
pai
achega
b673ad3e5c

+ 3 - 0
apps/emqx_management/include/emqx_mgmt.hrl

@@ -15,3 +15,6 @@
 %%--------------------------------------------------------------------
 
 -define(DEFAULT_ROW_LIMIT, 100).
+
+-define(URL_PARAM_INTEGER, url_param_integer).
+-define(URL_PARAM_BINARY, url_param_binary).

+ 11 - 12
apps/emqx_management/src/emqx_mgmt_api.erl

@@ -17,13 +17,12 @@
 -module(emqx_mgmt_api).
 
 -include_lib("stdlib/include/qlc.hrl").
+-include("emqx_mgmt.hrl").
 
 -elvis([{elvis_style, dont_repeat_yourself, #{min_complexity => 100}}]).
 
 -define(LONG_QUERY_TIMEOUT, 50000).
 
--define(CONT_BASE64_OPTS, #{mode => urlsafe, padding => false}).
-
 -export([
     paginate/3
 ]).
@@ -151,19 +150,19 @@ decode_continuation(none, _Encoding) ->
 decode_continuation(end_of_data, _Encoding) ->
     %% Clients should not send "after=end_of_data" back to the server
     error;
-decode_continuation(Cont, none) ->
-    Cont;
-decode_continuation(Cont, base64) ->
-    base64:decode(Cont, ?CONT_BASE64_OPTS).
+decode_continuation(Cont, ?URL_PARAM_INTEGER) ->
+    binary_to_integer(Cont);
+decode_continuation(Cont, ?URL_PARAM_BINARY) ->
+    emqx_utils:hexstr_to_bin(Cont).
 
 encode_continuation(none, _Encoding) ->
     none;
 encode_continuation(end_of_data, _Encoding) ->
     end_of_data;
-encode_continuation(Cont, none) ->
-    emqx_utils_conv:bin(Cont);
-encode_continuation(Cont, base64) ->
-    base64:encode(emqx_utils_conv:bin(Cont), ?CONT_BASE64_OPTS).
+encode_continuation(Cont, ?URL_PARAM_INTEGER) ->
+    integer_to_binary(Cont);
+encode_continuation(Cont, ?URL_PARAM_BINARY) ->
+    emqx_utils:bin_to_hexstr(Cont, lower).
 
 %%--------------------------------------------------------------------
 %% Node Query
@@ -663,7 +662,7 @@ parse_pager_params(Params) ->
             false
     end.
 
--spec parse_cont_pager_params(map(), none | base64) ->
+-spec parse_cont_pager_params(map(), ?URL_PARAM_INTEGER | ?URL_PARAM_BINARY) ->
     #{limit := pos_integer(), continuation := none | end_of_table | binary()} | false.
 parse_cont_pager_params(Params, Encoding) ->
     Cont = continuation(Params, Encoding),
@@ -675,7 +674,7 @@ parse_cont_pager_params(Params, Encoding) ->
             false
     end.
 
--spec encode_cont_pager_params(map(), none | base64) -> map().
+-spec encode_cont_pager_params(map(), ?URL_PARAM_INTEGER | ?URL_PARAM_BINARY) -> map().
 encode_cont_pager_params(#{continuation := Cont} = Meta, ContEncoding) ->
     Meta1 = maps:remove(continuation, Meta),
     Meta1#{last => encode_continuation(Cont, ContEncoding)};

+ 3 - 21
apps/emqx_management/src/emqx_mgmt_api_clients.erl

@@ -1135,7 +1135,7 @@ remove_live_sessions(Rows) ->
     ).
 
 list_client_msgs(MsgType, ClientID, QString) ->
-    case parse_cont_pager_params(QString, MsgType) of
+    case emqx_mgmt_api:parse_cont_pager_params(QString, cont_encoding(MsgType)) of
         false ->
             {400, #{code => <<"INVALID_PARAMETER">>, message => <<"after_limit_invalid">>}};
         PagerParams = #{} ->
@@ -1147,28 +1147,10 @@ list_client_msgs(MsgType, ClientID, QString) ->
             end
     end.
 
-parse_cont_pager_params(QString, MsgType) ->
-    case emqx_mgmt_api:parse_cont_pager_params(QString, cont_encoding(MsgType)) of
-        false ->
-            false;
-        PagerParams ->
-            maybe_cast_cont(MsgType, PagerParams)
-    end.
-
-maybe_cast_cont(inflight_msgs, #{continuation := Cont} = PagerParams) when is_binary(Cont) ->
-    try
-        PagerParams#{continuation => emqx_utils_conv:int(Cont)}
-    catch
-        _:_ ->
-            false
-    end;
-maybe_cast_cont(_, PagerParams) ->
-    PagerParams.
-
 %% integer packet id
-cont_encoding(inflight_msgs) -> none;
+cont_encoding(inflight_msgs) -> ?URL_PARAM_INTEGER;
 %% binary message id
-cont_encoding(mqueue_msgs) -> base64.
+cont_encoding(mqueue_msgs) -> ?URL_PARAM_BINARY.
 
 %%--------------------------------------------------------------------
 %% QueryString to Match Spec