Don't advertise auth mechanisms too early

If "starttls_required: true" is specified for c2s connections,
authentication mechanisms shouldn't be offered before negotiating the
TLS connection.
This commit is contained in:
Holger Weiss 2014-10-23 10:04:14 +02:00
parent 0b22277b11
commit 2d748115ee
1 changed files with 19 additions and 13 deletions

View File

@ -383,6 +383,9 @@ wait_for_stream({xmlstreamstart, _Name, Attrs}, StateData) ->
send_header(StateData, Server, <<"1.0">>, DefaultLang), send_header(StateData, Server, <<"1.0">>, DefaultLang),
case StateData#state.authenticated of case StateData#state.authenticated of
false -> false ->
TLS = StateData#state.tls,
TLSEnabled = StateData#state.tls_enabled,
TLSRequired = StateData#state.tls_required,
SASLState = SASLState =
cyrsasl:server_new( cyrsasl:server_new(
<<"jabber">>, Server, <<"">>, [], <<"jabber">>, Server, <<"">>, [],
@ -398,12 +401,21 @@ wait_for_stream({xmlstreamstart, _Name, Attrs}, StateData) ->
ejabberd_auth:check_password_with_authmodule( ejabberd_auth:check_password_with_authmodule(
U, Server, P, D, DG) U, Server, P, D, DG)
end), end),
Mechs = lists:map(fun (S) -> Mechs =
#xmlel{name = <<"mechanism">>, case TLSEnabled or not TLSRequired of
attrs = [], true ->
children = [{xmlcdata, S}]} Ms = lists:map(fun (S) ->
end, #xmlel{name = <<"mechanism">>,
cyrsasl:listmech(Server)), attrs = [],
children = [{xmlcdata, S}]}
end,
cyrsasl:listmech(Server)),
[#xmlel{name = <<"mechanisms">>,
attrs = [{<<"xmlns">>, ?NS_SASL}],
children = Ms}];
false ->
[]
end,
SockMod = SockMod =
(StateData#state.sockmod):get_sockmod( (StateData#state.sockmod):get_sockmod(
StateData#state.socket), StateData#state.socket),
@ -421,9 +433,6 @@ wait_for_stream({xmlstreamstart, _Name, Attrs}, StateData) ->
_ -> _ ->
[] []
end, end,
TLS = StateData#state.tls,
TLSEnabled = StateData#state.tls_enabled,
TLSRequired = StateData#state.tls_required,
TLSFeature = TLSFeature =
case (TLS == true) andalso case (TLS == true) andalso
(TLSEnabled == false) andalso (TLSEnabled == false) andalso
@ -448,10 +457,7 @@ wait_for_stream({xmlstreamstart, _Name, Attrs}, StateData) ->
#xmlel{name = <<"stream:features">>, #xmlel{name = <<"stream:features">>,
attrs = [], attrs = [],
children = children =
TLSFeature ++ CompressFeature ++ TLSFeature ++ CompressFeature ++ Mechs
[#xmlel{name = <<"mechanisms">>,
attrs = [{<<"xmlns">>, ?NS_SASL}],
children = Mechs}]
++ ++
ejabberd_hooks:run_fold(c2s_stream_features, ejabberd_hooks:run_fold(c2s_stream_features,
Server, [], [Server])}), Server, [], [Server])}),