ejabberd分析(五)+订阅/添加好友

模块ejabberd_c2s中，状态为session_established2。

用户发送iq set 消息到服务器

<iq id="M8QF3-52" type="set"><query xmlns="jabber:iq:roster"><item jid="1555215557@kinglong" name="1555215557"><group>Friends</group></item></query></iq>

服务器端匹配到

case Name of ...... To = xml:get_attr_s("to", Attrs),     ToJID = case To of         "" ->             jlib:make_jid(User, Server, "");         _ ->             jlib:string_to_jid(To)         end,  ...... "iq" -> case jlib:iq_query_info(NewEl) of                 #iq{xmlns = Xmlns} = IQ .... _ -> ejabberd_hooks:run(                   user_send_packet,                   Server,                   [FromJID, ToJID, NewEl]), check_privacy_route(FromJID, StateData, FromJID, ToJID, NewEl),

上面这段代码需要注意ToJID的解析，发送给服务器的消息一般是默认不带to 属性的，这种情况下会自动将to 补上(这关系到包的路由)。

ejabberd_hooks:run(user_send_packet   会记录用户发送的消息到日志。

check_privacy_route 中对权限进行效验，通过后调用ejabberd_router:route(FromRoute, To, Packet) 进行路由。

在ejabberd_router:route中调用do_route/3：

ejabberd_hooks:run_fold(filter_packet,先用包过滤模块进行过滤，通过后在路由表中查找路由到的模块的Pid：

case mnesia:dirty_read(route, LDstDomain) of

注：此处会以LDstDomain = To#jid.lserver, 去router表查找是交由本地处理，还是交由其他server 处理。

在模块ejabberd_local 中使用ejabberd_router:register_route(Host, {apply, ?MODULE, route}) 对当前domain进行了登记，所以会路由到此模块，并调用相应的route函数。

mnesia:write(#route{domain = LDomain,                             pid = Pid,                             local_hint = LocalHint}) {apply, Module, Function} -> Module:Function(From, To, Packet);在ejabberd_local:route/3中调用do_route(From, To, Packet)。

客户端发送的iq消息初始to为空，但系统已自动补上ToJID(参见前面的说明)，资源也为空的 iq，所以会路由到ejabberd_sm:route/3

 if     To#jid.luser /= "" ->         ejabberd_sm:route(From, To, Packet);

也就是说对于那些在session_established 之后发送的消息都会交由ejabberd_sm去路由。

在ejabberd_sm中由do_route/3 来处理：

do_route(From, To, Packet) -> case LResource of "" -> case Name of ......  "iq" -> process_iq(From, To, Packet);ejabberd_sm 中有一张 sm_iqtable  里面记录了各个命名空间的消息最终由那个模块进行处理。

ejabberd_sm 是一个gen_server 当调用register_iq_handler 接口时会向sm_iqtable 插入记录。

而这些命名空间处理模块的登记是由gen_iq_handler 模块的add_iq_handler/6 来进行的。

-module(gen_iq_handler) add_iq_handler(Component, Host, NS, Module, Function, Type) -> case Type of no_queue -> Component:register_iq_handler(Host, NS, Module, Function, no_queue);jabber:iq:roster 定义的宏为NS_ROSTER，我们在源码目录下查找此命名空间是由哪个模块调用的注册：

在mod_roster:start/2 中有如下代码：

gen_iq_handler:add_iq_handler(ejabberd_sm, Host, ?NS_ROSTER, ?MODULE, process_iq, IQDisc).
可以看出mod_roster模块将自身的process_iq 函数在ejabberd_sm中注册为了jabber:iq:roster 命名空间的处理方法。

process_iq中针对发送到本地的iq调用process_local_iq/3：

process_local_iq(From, To, #iq{type = Type} = IQ) -> case Type of set -> process_iq_set(From, To, IQ); get -> process_iq_get(From, To, IQ) end.
这里会对针对iq 的 type 进行不同处理，我们目前跟踪的是<iq id="M8QF3-52" type="set">

process_iq_set(From, To, #iq{sub_el = SubEl} = IQ) -> {xmlelement, _Name, _Attrs, Els} = SubEl, lists:foreach(fun(El) -> process_item_set(From, To, El) end, Els), IQ#iq{type = result, sub_el = []}. 在process_iq_set 中会遍历每一item子项调用process_item_set/3 进行处理：
process_item_set(From, To, {xmlelement, _Name, Attrs, Els}) -> JID1 = jlib:string_to_jid(xml:get_attr_s("jid", Attrs)), #jid{user = User, luser = LUser, lserver = LServer} = From, case JID1 of error -> ok; _ -> JID = {JID1#jid.user, JID1#jid.server, JID1#jid.resource}, LJID = jlib:jid_tolower(JID1), F = fun() -> Res = mnesia:read({roster, {LUser, LServer, LJID}}), Item = case Res of [] -> #roster{usj = {LUser, LServer, LJID}, us = {LUser, LServer}, jid = JID}; [I] -> I#roster{jid = JID, name = "", groups = [], xs = []} end, Item1 = process_item_attrs(Item, Attrs), Item2 = process_item_els(Item1, Els),  case Item2#roster.subscription of <pre name="code" class="java"> %% 如果是删除   remove ->mnesia:delete({roster, {LUser, LServer, LJID}});
 %%否则添加
_ ->mnesia:write(Item2)end,
%% If the item exist in shared roster, take the%% subscription information from there:
Item3 = ejabberd_hooks:run_fold(roster_process_item,LServer, Item2, [LServer]),
case roster_version_on_db(LServer)
oftrue -> mnesia:write(#roster_version{us = {LUser, LServer}, version = sha:sha(term_to_binary(now()))});
false -> okend,{Item, Item3}end,
case mnesia:transaction(F)
of{atomic, {OldItem, Item}} -> push_item(User, LServer, To, Item),
case Item#roster.subscription
ofremove -> send_unsubscribing_presence(From, OldItem),
ok;_ -> ok end;
E -> ?DEBUG("ROSTER: roster item set error: ~p~n", [E]), ok end end;

以上就是一条添加/订阅 好友的iq 消息在ejabberd 中从接收到最终写入到数据库所经过的路径。