./ct_report/coverage/mod_bind2.COVER.html

1 %% @doc Implement Bind2, allowing the inline specification of:
2 %% resource binding, stream management, carbons and csi.
3 %%
4 %% There are two steps to this: on sasl2_success, we check if there was a bind2 inline request,
5 %% and also, if there was a previous SM resumption that has already succeeded – if resumption
6 %% succeeded, bind2 must be entirely skipped.
7 %%
8 %% If we can proceed with binding, we run a similar algorithm that c2s runs itself: we verify the
9 %% user is allowed to start a session, we open the session changing the c2s data, and we finalise
10 %% the operation with handle_state_after_packet or maybe_retry_state. But in the case of bind2,
11 %% once verifications are successful, we need to modify the c2s data to append for example the csi
12 %% state or the carbons tag in the session info, before the session is actually established, so that
13 %% session establishment is atomic to changes in the c2s data.
14 -module(mod_bind2).
15 -xep([{xep, 386}, {version, "0.4.0"}, {status, partial}]).
16
17 -include("jlib.hrl").
18
19 -define(XMLNS_BIND_2, {<<"xmlns">>, ?NS_BIND_2}).
20
21 -behaviour(gen_mod).
22
23 %% gen_mod callbacks
24 -export([start/2, stop/1, deps/2, hooks/1, supported_features/0]).
25
26 %% hooks handlers
27 -export([
28 sasl2_stream_features/3,
29 sasl2_start/3,
30 sasl2_success/3
31 ]).
32
33 -export([get_bind_request/1, append_inline_bound_answer/3]).
34
35 %% gen_mod
36 -spec start(mongooseim:host_type(), gen_mod:module_opts()) -> ok.
37 start(_HostType, _Opts) ->
38
:-(
ok.
39
40 -spec stop(mongooseim:host_type()) -> ok.
41 stop(_HostType) ->
42
:-(
ok.
43
44 -spec deps(mongooseim:host_type(), gen_mod:module_opts()) -> gen_mod_deps:deps().
45 deps(_HostType, Opts) ->
46
:-(
[{mod_sasl2, Opts, hard}].
47
48 -spec hooks(mongooseim:host_type()) -> gen_hook:hook_list().
49 hooks(HostType) ->
50
:-(
[
51 {sasl2_stream_features, HostType, fun ?MODULE:sasl2_stream_features/3, #{}, 50},
52 {sasl2_start, HostType, fun ?MODULE:sasl2_start/3, #{}, 50},
53 {sasl2_success, HostType, fun ?MODULE:sasl2_success/3, #{}, 50} %% after SM
54 ].
55
56 -spec supported_features() -> [atom()].
57 supported_features() ->
58
:-(
[dynamic_domains].
59
60
61 %% Hook handlers
62 -spec sasl2_stream_features(Acc, #{c2s_data := mongoose_c2s:data()}, gen_hook:extra()) ->
63 {ok, Acc} when Acc :: [exml:element()].
64 sasl2_stream_features(Acc, #{c2s_data := C2SData}, _) ->
65
:-(
Bind2Feature = feature(C2SData),
66
:-(
{ok, lists:keystore(feature_name(), #xmlel.name, Acc, Bind2Feature)}.
67
68 -spec sasl2_start(SaslAcc, #{stanza := exml:element()}, gen_hook:extra()) ->
69 {ok, SaslAcc} when SaslAcc :: mongoose_acc:t().
70 sasl2_start(SaslAcc, #{stanza := El}, _) ->
71
:-(
case exml_query:path(El, [{element_with_ns, <<"bind">>, ?NS_BIND_2}]) of
72 undefined ->
73
:-(
{ok, SaslAcc};
74 BindRequest ->
75
:-(
{ok, mod_sasl2:put_inline_request(SaslAcc, ?MODULE, BindRequest)}
76 end.
77
78 -spec sasl2_success(SaslAcc, mod_sasl2:c2s_state_data(), gen_hook:extra()) ->
79 {ok, SaslAcc} when SaslAcc :: mongoose_acc:t().
80 sasl2_success(SaslAcc, _, #{host_type := HostType}) ->
81
:-(
case mod_sasl2:get_inline_request(SaslAcc, ?MODULE, undefined) of
82 undefined ->
83
:-(
{ok, SaslAcc};
84 #{request := BindRequest} ->
85
:-(
Resource = generate_lresource(BindRequest),
86
:-(
maybe_bind(SaslAcc, Resource, HostType)
87 end.
88
89 -spec maybe_bind(SaslAcc, jid:lresource(), mongooseim:host_type()) ->
90 {ok, SaslAcc} when SaslAcc :: mongoose_acc:t().
91 maybe_bind(SaslAcc, Resource, HostType) ->
92
:-(
{SaslAcc1, SaslStateData1, HookParams} = build_new_c2sdata_saslstatedata_and_hookparams(SaslAcc, Resource),
93
:-(
case mongoose_c2s:verify_user(session_established, HostType, HookParams, SaslAcc1) of
94 {ok, SaslAcc2} ->
95
:-(
Bound = create_bind_response(<<"bound">>),
96
:-(
SaslAcc3 = mod_sasl2:update_inline_request(SaslAcc2, ?MODULE, Bound, success),
97
:-(
SaslAcc4 = mongoose_hooks:bind2_enable_features(HostType, SaslAcc3, SaslStateData1),
98
:-(
#{c2s_data := C2SData2} = SaslStateData2 = mod_sasl2:get_state_data(SaslAcc4),
99
:-(
{ok, SaslAcc5, C2SData3} = mongoose_c2s:maybe_open_session(session_established, {ok, SaslAcc4}, C2SData2),
100
:-(
SaslStateData3 = SaslStateData2#{c2s_data => C2SData3, c2s_state => session_established},
101
:-(
SaslAcc6 = mod_sasl2:set_state_data(SaslAcc5, SaslStateData3),
102
:-(
{ok, SaslAcc6};
103 {stop, SaslAcc1} ->
104
:-(
bind2_failed(SaslAcc1)
105 end.
106
107 -spec build_new_c2sdata_saslstatedata_and_hookparams(mongoose_acc:t(), jid:lresource()) ->
108 {mongoose_acc:t(), mod_sasl2:c2s_state_data(), mongoose_c2s_hooks:params()}.
109 build_new_c2sdata_saslstatedata_and_hookparams(SaslAcc, Resource) ->
110
:-(
SaslStateData1 = mod_sasl2:get_state_data(SaslAcc),
111
:-(
#{c2s_data := C2SData1, c2s_state := C2SState1} = SaslStateData1,
112
:-(
C2SData2 = mongoose_c2s:replace_resource(C2SData1, Resource),
113
:-(
SaslStateData2 = SaslStateData1#{c2s_data => C2SData2},
114
:-(
HookParams = mongoose_c2s:hook_arg(C2SData2, C2SState1, internal, bind2, undefined),
115
:-(
{mod_sasl2:set_state_data(SaslAcc, SaslStateData2), SaslStateData2, HookParams}.
116
117 bind2_failed(SaslAcc) ->
118 %% The XEP does not specify what to do if the resource wasn't bound
119 %% so we just set failed here and move along with SASL2
120
:-(
Error = create_bind_response(<<"failed">>),
121
:-(
{ok, mod_sasl2:update_inline_request(SaslAcc, ?MODULE, Error, failure)}.
122
123 create_bind_response(Answer) ->
124
:-(
#xmlel{name = Answer, attrs = [?XMLNS_BIND_2]}.
125
126 %% Helpers
127 -spec generate_lresource(exml:element()) -> jid:lresource().
128 generate_lresource(BindRequest) ->
129
:-(
GeneratedResource = mongoose_c2s:generate_random_resource(),
130
:-(
case exml_query:path(BindRequest, [{element, <<"tag">>}, cdata], not_provided) of
131 not_provided ->
132
:-(
GeneratedResource;
133 Tag ->
134
:-(
case jid:resourceprep(Tag) of
135 error ->
136 %% The XEP does not specify how to fail in case of a bad resource requested,
137 %% so we decide to ignore the tag and simply generate our own resource
138
:-(
GeneratedResource;
139 Prefix -> %% https://xmpp.org/extensions/xep-0386.html#identifiers
140
:-(
<<Prefix/binary, "/", GeneratedResource/binary>>
141 end
142 end.
143
144 -spec feature(mongoose_c2s:data()) -> exml:element().
145 feature(C2SData) ->
146
:-(
Inlines = mongoose_hooks:bind2_stream_features(C2SData, []),
147
:-(
InlineElem = inlines(Inlines),
148
:-(
#xmlel{name = feature_name(),
149 attrs = [?XMLNS_BIND_2],
150 children = [InlineElem]}.
151
152 -spec inlines([exml:element()]) -> exml:element().
153 inlines(Inlines) ->
154
:-(
#xmlel{name = <<"inline">>, children = Inlines}.
155
156 -spec feature_name() -> binary().
157 feature_name() ->
158
:-(
<<"bind">>.
159
160 -spec get_bind_request(mongoose_acc:t()) -> mod_sasl2:inline_request().
161 get_bind_request(SaslAcc) ->
162
:-(
mod_sasl2:get_inline_request(SaslAcc, ?MODULE).
163
164 -spec append_inline_bound_answer(mongoose_acc:t(), mod_sasl2:inline_request(), exml:element()) ->
165 mongoose_acc:t().
166 append_inline_bound_answer(SaslAcc, #{response := BoundEl = #xmlel{children = Children}}, Response) ->
167
:-(
BoundEl1 = BoundEl#xmlel{children = [Response | Children]},
168
:-(
mod_sasl2:append_inline_response(SaslAcc, ?MODULE, BoundEl1).
Line Hits Source