基于ejabberd的类WhatsApp预认证注册模块选型咨询

嘿，针对你这个类WhatsApp注册流程的需求，我来给你详细拆解下在ejabberd里的实现方案，完全不需要额外搭建其他服务器哦：

核心结论：用自定义ejabberd模块 + 开放权限的IQ Handler就够了 #

首先纠正一个误解：Hooks和IQ handlers并非只适用于已注册用户，只要你在注册IQ Handler时设置允许未认证用户访问，就能处理未注册用户的请求。

1. 实现步骤对应的代码位置 #

你需要创建一个自定义ejabberd模块，在里面实现两个核心IQ处理逻辑：请求OTP、验证OTP并完成注册。

示例模块代码框架

-module(whatsapp_reg).
-behaviour(ejabberd_gen_iq_handler).

-export([start/0, process_iq/3, init/0]).

% 自定义XML命名空间，客户端请求时要指定这个
-define(NS_WHATSAPP_REG, <<"urn:whatsapp:reg">>).

init() ->
    % 创建存储OTP的ETS表，支持并发读取
    ets:new(otp_store, [named_table, public, {read_concurrency, true}]),
    ok.

start() ->
    % 注册IQ Handler，设置access为all，允许未认证用户请求
    ejabberd_iq:register_iq_handler(ejabberd_sm, ?NS_WHATSAPP_REG, ?MODULE, process_iq, all),
    ok.

% 处理请求OTP的IQ
process_iq(_From, _To, #iq{type = set, sub_el = #xmlel{name = <<"request_otp">>, children = Children}}) ->
    % 从XML中提取手机号
    Phone = extract_phone_from_xml(Children),
    % 生成6位随机安全OTP
    OTP = generate_secure_otp(),
    % 存储OTP，设置2分钟过期时间
    ExpireTime = erlang:system_time(second) + 120,
    ets:insert(otp_store, {Phone, OTP, ExpireTime}),
    % 启动定时器，到期后清理过期OTP
    ejabberd_timer:send_after(120000, self(), {expire_otp, Phone}),
    % 调用短信网关发送OTP（用ejabberd自带的httpc模块发起请求即可）
    send_otp_to_sms_gateway(Phone, OTP),
    % 返回OTP已发送的响应
    {iq_result, #xmlel{name = <<"otp_sent">>, attrs = [{<<"xmlns">>, ?NS_WHATSAPP_REG}]}};

% 处理验证OTP的IQ
process_iq(_From, _To, #iq{type = set, sub_el = #xmlel{name = <<"verify_otp">>, children = Children}}) ->
    % 提取手机号和输入的OTP
    {Phone, InputOTP} = extract_phone_otp_from_xml(Children),
    % 验证OTP有效性
    case validate_otp(Phone, InputOTP) of
        valid ->
            % 创建ejabberd用户（可以用OTP作为临时密码，后续让用户修改）
            Domain = <<"your_xmpp_domain">>,
            ejabberd_auth:create_user(Phone, Domain, InputOTP),
            % 清理OTP记录
            ets:delete(otp_store, Phone),
            % 返回注册成功响应
            {iq_result, #xmlel{name = <<"reg_success">>, attrs = [{<<"xmlns">>, ?NS_WHATSAPP_REG}]}};
        invalid ->
            % 返回错误响应
            {iq_error, ?ERR_NOT_ALLOWED}
    end;

% 处理其他无效请求
process_iq(_From, _To, _IQ) ->
    {iq_error, ?ERR_BAD_REQUEST}.

% 辅助函数：生成安全OTP
generate_secure_otp() ->
    Bytes = crypto:strong_rand_bytes(3),
    integer_to_binary(binary:decode_unsigned(Bytes) rem 1000000).

% 辅助函数：验证OTP
validate_otp(Phone, InputOTP) ->
    case ets:lookup(otp_store, Phone) of
        [{Phone, StoredOTP, ExpireTime}] when ExpireTime > erlang:system_time(second), StoredOTP =:= InputOTP ->
            valid;
        _ ->
            invalid
    end.

% 其他辅助函数：XML解析、短信发送等，根据你的实际需求实现
extract_phone_from_xml(Children) -> ...
extract_phone_otp_from_xml(Children) -> ...
send_otp_to_sms_gateway(Phone, OTP) -> ...

2. Hooks的补充使用 #

如果需要在注册完成后触发额外逻辑（比如同步用户数据到其他系统、发送欢迎消息），可以使用ejabberd的register_user钩子，在自定义模块里添加：

start() ->
    ejabberd_hooks:add(register_user, ?MY_DOMAIN, ?MODULE, on_user_register, 50),
    % 之前的IQ Handler注册代码...
    ok.

on_user_register(Username, Domain, _Password) ->
    % 这里写注册后的逻辑，比如记录日志、调用外部API
    ok.

3. 模块加载与配置 #

把这个模块编译成.beam文件放到ejabberd的模块目录，然后在ejabberd.yml里添加：

modules:
  whatsapp_reg: {}

4. 是否需要额外服务器？ #

完全不需要！ejabberd本身支持用Erlang编写自定义模块处理所有注册流程，哪怕你需要调用短信网关，也可以用ejabberd自带的httpc模块发起HTTP请求。如果你的业务逻辑有复杂的非XMPP需求，再考虑拆分，但单纯的手机号+OTP注册完全可以在ejabberd内实现。

注意事项 #

• 手机号要统一格式（比如带国家码），避免重复注册；
• OTP存储可以用Mnesia或Redis替代ETS，避免ejabberd重启后丢失数据；
• 要做好请求频率限制，防止恶意刷OTP；
• 生产环境中建议对IQ请求做加密处理。

内容的提问来源于stack exchange，提问作者Shubham1164