1173

Committed 28 Oct 2025 11:02AM UTC coverage: 33.768% (-0.02%) from 33.79%

Build # 1173

Build Type

push

github

Committed by

badlop

Commit Message

CHANGELOG.md: Update to 25.10

Run Details

15513 of 45940 relevant lines covered (33.77%)

1078.01 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

57.53

/src/extauth.erl

%%%-------------------------------------------------------------------
%%% Created : 7 May 2018 by Evgeny Khramtsov <ekhramtsov@process-one.net>
%%%
%%%
%%% ejabberd, Copyright (C) 2002-2025   ProcessOne
%%%
%%% This program is free software; you can redistribute it and/or
%%% modify it under the terms of the GNU General Public License as
%%% published by the Free Software Foundation; either version 2 of the
%%% License, or (at your option) any later version.
%%%
%%% This program is distributed in the hope that it will be useful,
%%% but WITHOUT ANY WARRANTY; without even the implied warranty of
%%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
%%% General Public License for more details.
%%%
%%% You should have received a copy of the GNU General Public License along
%%% with this program; if not, write to the Free Software Foundation, Inc.,
%%% 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
%%%
%%%-------------------------------------------------------------------
-module(extauth).

-ifndef(GEN_SERVER).
-define(GEN_SERVER, gen_server).
-endif.
-behaviour(?GEN_SERVER).

-define(CALL_TIMEOUT, timer:seconds(30)).

%% API
-export([start/1, stop/1, reload/1, start_link/2]).
-export([check_password/3, set_password/3, try_register/3, remove_user/2,
         remove_user/3, user_exists/2, check_certificate/3]).
-export([prog_name/1, pool_name/1, worker_name/2, pool_size/1]).
-export([init/1, handle_call/3, handle_cast/2, handle_info/2,
         terminate/2, code_change/3]).

-include("logger.hrl").

-record(state, {port :: port(),
                prog :: string(),
                start_time :: integer(),
                os_pid :: integer() | undefined}).

%%%===================================================================
%%% API
%%%===================================================================
start(Host) ->
    extauth_sup:start(Host).

stop(Host) ->
    extauth_sup:stop(Host).

reload(Host) ->
    extauth_sup:reload(Host).

start_link(Name, Prog) ->
    ?GEN_SERVER:start_link({local, Name}, ?MODULE, [Prog], []).

check_password(User, Server, Password) ->
    call_port(Server, [<<"auth">>, User, Server, Password]).

check_certificate(User, Server, Certificate) ->
    call_port(Server, [<<"certauth">>, User, Server, Certificate]).

user_exists(User, Server) ->
    call_port(Server, [<<"isuser">>, User, Server]).

set_password(User, Server, Password) ->
    call_port(Server, [<<"setpass">>, User, Server, Password]).

try_register(User, Server, Password) ->
    call_port(Server, [<<"tryregister">>, User, Server, Password]).

remove_user(User, Server) ->
    call_port(Server, [<<"removeuser">>, User, Server]).

remove_user(User, Server, Password) ->
    call_port(Server, [<<"removeuser3">>, User, Server, Password]).

-spec prog_name(binary()) -> string() | undefined.
prog_name(Host) ->
    ejabberd_option:extauth_program(Host).

-spec pool_name(binary()) -> atom().
pool_name(Host) ->
    case ejabberd_option:extauth_pool_name(Host) of
        undefined ->
            list_to_atom("extauth_pool_" ++ binary_to_list(Host));
        Name ->
            list_to_atom("extauth_pool_" ++ binary_to_list(Name))
    end.

-spec worker_name(atom(), integer()) -> atom().
worker_name(Pool, N) ->
    list_to_atom(atom_to_list(Pool) ++ "_" ++ integer_to_list(N)).

-spec pool_size(binary()) -> pos_integer().
pool_size(Host) ->
    case ejabberd_option:extauth_pool_size(Host) of
        undefined -> misc:logical_processors();
        Size ->
            Size
    end.

%%%===================================================================
%%% gen_server callbacks
%%%===================================================================
init([Prog]) ->
    process_flag(trap_exit, true),
    {Port, OSPid} = start_port(Prog),
    Time = curr_time(),
    {ok, #state{port = Port, start_time = Time,
                prog = Prog, os_pid = OSPid}}.

handle_call({cmd, Cmd, EndTime}, _From, State) ->
    Timeout = EndTime - curr_time(),
    if Timeout > 0 ->
            Port = State#state.port,
            port_command(Port, Cmd),
            receive
                {Port, {data, [0, N] = Data}} when N == 0; N == 1 ->
                    ?DEBUG("Received response from external authentication "
                           "program: ~p", [Data]),
                    {reply, decode_bool(N), State};
                {Port, Data} ->
                    ?ERROR_MSG("Received unexpected response from external "
                               "authentication program '~ts': ~p "
                               "(port = ~p, pid = ~w)",
                               [State#state.prog, Data, Port, State#state.os_pid]),
                    {reply, {error, unexpected_response}, State};
                {'EXIT', Port, Reason} ->
                    handle_info({'EXIT', Port, Reason}, State)
            after Timeout ->
                    {stop, normal, State}
            end;
       true ->
            {noreply, State}
    end.

handle_cast(_Msg, State) ->
    {noreply, State}.

handle_info({'EXIT', Port, _Reason}, #state{port = Port,
                                            start_time = Time} = State) ->
    case curr_time() - Time of
        Diff when Diff < 1000 ->
            ?ERROR_MSG("Failed to start external authentication program '~ts'",
                       [State#state.prog]),
            {stop, normal, State};
        _ ->
            ?ERROR_MSG("External authentication program '~ts' has terminated "
                       "unexpectedly (pid=~w), restarting via supervisor...",
                       [State#state.prog, State#state.os_pid]),
            {stop, normal, State}
    end;
handle_info(Info, State) ->
    ?WARNING_MSG("Unexpected info: ~p", [Info]),
    {noreply, State}.

terminate(_Reason, State) ->
    catch port_close(State#state.port),
    ok.

code_change(_OldVsn, State, _Extra) ->
    {ok, State}.

%%%===================================================================
%%% Internal functions
%%%===================================================================
-spec curr_time() -> non_neg_integer().
curr_time() ->
    erlang:monotonic_time(millisecond).

-spec start_port(string()) -> {port(), integer() | undefined}.
start_port(Path) ->
    Port = open_port({spawn, Path}, [{packet, 2}]),
    link(Port),
    case erlang:port_info(Port, os_pid) of
        {os_pid, OSPid} ->
            {Port, OSPid};
        undefined ->
            {Port, undefined}
    end.

call_port(Server, Args) ->
    call_port(Server, Args, ?CALL_TIMEOUT).

call_port(Server, Args, Timeout) ->
    StartTime = erlang:monotonic_time(millisecond),
    Pool = pool_name(Server),
    PoolSize = pool_size(Server),
    I = p1_rand:round_robin(PoolSize),
    Cmd = str:join(Args, <<":">>),
    do_call(Cmd, I, I + PoolSize, Pool, PoolSize,
            StartTime + Timeout, StartTime).

do_call(_, Max, Max, _, _, _, _) ->
    {error, disconnected};
do_call(Cmd, I, Max, Pool, PoolSize, EndTime, CurrTime) ->
    Timeout = EndTime - CurrTime,
    if Timeout > 0 ->
            Proc = worker_name(Pool, (I rem PoolSize) + 1),
            try ?GEN_SERVER:call(Proc, {cmd, Cmd, EndTime}, Timeout)
            catch exit:{timeout, {?GEN_SERVER, call, _}} ->
                    {error, timeout};
                  exit:{_, {?GEN_SERVER, call, _}} ->
                    do_call(Cmd, I+1, Max, Pool, PoolSize, EndTime, curr_time())
            end;
       true ->
            {error, timeout}
    end.

decode_bool(0) -> false;
decode_bool(1) -> true.

1	%%%-------------------------------------------------------------------
2	%%% Created : 7 May 2018 by Evgeny Khramtsov <ekhramtsov@process-one.net>
3	%%%
4	%%%
5	%%% ejabberd, Copyright (C) 2002-2025 ProcessOne
6	%%%
7	%%% This program is free software; you can redistribute it and/or
8	%%% modify it under the terms of the GNU General Public License as
9	%%% published by the Free Software Foundation; either version 2 of the
10	%%% License, or (at your option) any later version.
11	%%%
12	%%% This program is distributed in the hope that it will be useful,
13	%%% but WITHOUT ANY WARRANTY; without even the implied warranty of
14	%%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15	%%% General Public License for more details.
16	%%%
17	%%% You should have received a copy of the GNU General Public License along
18	%%% with this program; if not, write to the Free Software Foundation, Inc.,
19	%%% 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20	%%%
21	%%%-------------------------------------------------------------------
22	-module(extauth).
23
24	-ifndef(GEN_SERVER).
25	-define(GEN_SERVER, gen_server).
26	-endif.
27	-behaviour(?GEN_SERVER).
28
29	-define(CALL_TIMEOUT, timer:seconds(30)).
30
31	%% API
32	-export([start/1, stop/1, reload/1, start_link/2]).
33	-export([check_password/3, set_password/3, try_register/3, remove_user/2,
34	remove_user/3, user_exists/2, check_certificate/3]).
35	-export([prog_name/1, pool_name/1, worker_name/2, pool_size/1]).
36	-export([init/1, handle_call/3, handle_cast/2, handle_info/2,
37	terminate/2, code_change/3]).
38
39	-include("logger.hrl").
40
41	-record(state, {port :: port(),
42	prog :: string(),
43	start_time :: integer(),
44	os_pid :: integer() \| undefined}).
45
46	%%%===================================================================
47	%%% API
48	%%%===================================================================
49	start(Host) ->
50	extauth_sup:start(Host).	1✔
51
52	stop(Host) ->
53	extauth_sup:stop(Host).	×
54
55	reload(Host) ->
56	extauth_sup:reload(Host).	×
57
58	start_link(Name, Prog) ->
59	?GEN_SERVER:start_link({local, Name}, ?MODULE, [Prog], []).	4✔
60
61	check_password(User, Server, Password) ->
62	call_port(Server, [<<"auth">>, User, Server, Password]).	2✔
63
64	check_certificate(User, Server, Certificate) ->
65	call_port(Server, [<<"certauth">>, User, Server, Certificate]).	×
66
67	user_exists(User, Server) ->
68	call_port(Server, [<<"isuser">>, User, Server]).	×
69
70	set_password(User, Server, Password) ->
71	call_port(Server, [<<"setpass">>, User, Server, Password]).	×
72
73	try_register(User, Server, Password) ->
74	call_port(Server, [<<"tryregister">>, User, Server, Password]).	×
75
76	remove_user(User, Server) ->
77	call_port(Server, [<<"removeuser">>, User, Server]).	1✔
78
79	remove_user(User, Server, Password) ->
80	call_port(Server, [<<"removeuser3">>, User, Server, Password]).	×
81
82	-spec prog_name(binary()) -> string() \| undefined.
83	prog_name(Host) ->
84	ejabberd_option:extauth_program(Host).	1✔
85
86	-spec pool_name(binary()) -> atom().
87	pool_name(Host) ->
88	case ejabberd_option:extauth_pool_name(Host) of	4✔
89	undefined ->
90	list_to_atom("extauth_pool_" ++ binary_to_list(Host));	4✔
91	Name ->
92	list_to_atom("extauth_pool_" ++ binary_to_list(Name))	×
93	end.
94
95	-spec worker_name(atom(), integer()) -> atom().
96	worker_name(Pool, N) ->
97	list_to_atom(atom_to_list(Pool) ++ "_" ++ integer_to_list(N)).	7✔
98
99	-spec pool_size(binary()) -> pos_integer().
100	pool_size(Host) ->
101	case ejabberd_option:extauth_pool_size(Host) of	4✔
102	undefined -> misc:logical_processors();	4✔
103	Size ->
104	Size	×
105	end.
106
107	%%%===================================================================
108	%%% gen_server callbacks
109	%%%===================================================================
110	init([Prog]) ->
111	process_flag(trap_exit, true),	4✔
112	{Port, OSPid} = start_port(Prog),	4✔
113	Time = curr_time(),	4✔
114	{ok, #state{port = Port, start_time = Time,	4✔
115	prog = Prog, os_pid = OSPid}}.
116
117	handle_call({cmd, Cmd, EndTime}, _From, State) ->
118	Timeout = EndTime - curr_time(),	3✔
119	if Timeout > 0 ->	3✔
120	Port = State#state.port,	3✔
121	port_command(Port, Cmd),	3✔
122	receive	3✔
123	{Port, {data, [0, N] = Data}} when N == 0; N == 1 ->
124	?DEBUG("Received response from external authentication "	3✔
125	"program: ~p", [Data]),	3✔
126	{reply, decode_bool(N), State};	3✔
127	{Port, Data} ->
128	?ERROR_MSG("Received unexpected response from external "	×
129	"authentication program '~ts': ~p "
130	"(port = ~p, pid = ~w)",
131	[State#state.prog, Data, Port, State#state.os_pid]),	×
132	{reply, {error, unexpected_response}, State};	×
133	{'EXIT', Port, Reason} ->
134	handle_info({'EXIT', Port, Reason}, State)	×
135	after Timeout ->
136	{stop, normal, State}	×
137	end;
138	true ->
139	{noreply, State}	×
140	end.
141
142	handle_cast(_Msg, State) ->
143	{noreply, State}.	×
144
145	handle_info({'EXIT', Port, _Reason}, #state{port = Port,
146	start_time = Time} = State) ->
147	case curr_time() - Time of	×
148	Diff when Diff < 1000 ->
149	?ERROR_MSG("Failed to start external authentication program '~ts'",	×
150	[State#state.prog]),	×
151	{stop, normal, State};	×
152	_ ->
153	?ERROR_MSG("External authentication program '~ts' has terminated "	×
154	"unexpectedly (pid=~w), restarting via supervisor...",
155	[State#state.prog, State#state.os_pid]),	×
156	{stop, normal, State}	×
157	end;
158	handle_info(Info, State) ->
159	?WARNING_MSG("Unexpected info: ~p", [Info]),	×
160	{noreply, State}.	×
161
162	terminate(_Reason, State) ->
163	catch port_close(State#state.port),	4✔
164	ok.	4✔
165
166	code_change(_OldVsn, State, _Extra) ->
167	{ok, State}.	×
168
169	%%%===================================================================
170	%%% Internal functions
171	%%%===================================================================
172	-spec curr_time() -> non_neg_integer().
173	curr_time() ->
174	erlang:monotonic_time(millisecond).	7✔
175
176	-spec start_port(string()) -> {port(), integer() \| undefined}.
177	start_port(Path) ->
178	Port = open_port({spawn, Path}, [{packet, 2}]),	4✔
179	link(Port),	4✔
180	case erlang:port_info(Port, os_pid) of	4✔
181	{os_pid, OSPid} ->
182	{Port, OSPid};	4✔
183	undefined ->
184	{Port, undefined}	×
185	end.
186
187	call_port(Server, Args) ->
188	call_port(Server, Args, ?CALL_TIMEOUT).	3✔
189
190	call_port(Server, Args, Timeout) ->
191	StartTime = erlang:monotonic_time(millisecond),	3✔
192	Pool = pool_name(Server),	3✔
193	PoolSize = pool_size(Server),	3✔
194	I = p1_rand:round_robin(PoolSize),	3✔
195	Cmd = str:join(Args, <<":">>),	3✔
196	do_call(Cmd, I, I + PoolSize, Pool, PoolSize,	3✔
197	StartTime + Timeout, StartTime).
198
199	do_call(_, Max, Max, _, _, _, _) ->
200	{error, disconnected};	×
201	do_call(Cmd, I, Max, Pool, PoolSize, EndTime, CurrTime) ->
202	Timeout = EndTime - CurrTime,	3✔
203	if Timeout > 0 ->	3✔
204	Proc = worker_name(Pool, (I rem PoolSize) + 1),	3✔
205	try ?GEN_SERVER:call(Proc, {cmd, Cmd, EndTime}, Timeout)	3✔
206	catch exit:{timeout, {?GEN_SERVER, call, _}} ->
207	{error, timeout};	×
208	exit:{_, {?GEN_SERVER, call, _}} ->
209	do_call(Cmd, I+1, Max, Pool, PoolSize, EndTime, curr_time())	×
210	end;
211	true ->
212	{error, timeout}	×
213	end.
214
215	decode_bool(0) -> false;	1✔
216	decode_bool(1) -> true.	2✔

processone / ejabberd / 1173

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous