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

fix(mgmt): avoid using base64:decode/2 and encode/2
This commit is contained in:
SergeTupchiy 2024-03-11 13:23:37 +02:00 committed by GitHub
commit b673ad3e5c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 17 additions and 33 deletions

View File

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

View File

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

View File

@ -1135,7 +1135,7 @@ remove_live_sessions(Rows) ->
). ).
list_client_msgs(MsgType, ClientID, QString) -> 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 -> false ->
{400, #{code => <<"INVALID_PARAMETER">>, message => <<"after_limit_invalid">>}}; {400, #{code => <<"INVALID_PARAMETER">>, message => <<"after_limit_invalid">>}};
PagerParams = #{} -> PagerParams = #{} ->
@ -1147,28 +1147,10 @@ list_client_msgs(MsgType, ClientID, QString) ->
end end
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 %% integer packet id
cont_encoding(inflight_msgs) -> none; cont_encoding(inflight_msgs) -> ?URL_PARAM_INTEGER;
%% binary message id %% binary message id
cont_encoding(mqueue_msgs) -> base64. cont_encoding(mqueue_msgs) -> ?URL_PARAM_BINARY.
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
%% QueryString to Match Spec %% QueryString to Match Spec