mirror of
https://github.com/processone/ejabberd.git
synced 2024-09-27 14:30:55 +02:00
568909d5bb
* src/configure.ac: Likewise * src/mod_muc/mod_muc_room.erl: Weakened presence filtering, added warning in non-anonymous rooms, room destroying updated to latest JEP-0045, added a number of occupants and room name in room's disco#info reply, miscellaneous internal changes (thanks to Sergei Golovan) * src/mod_muc/mod_muc.erl: Better support for nick unregistration (thanks to Sergei Golovan) * src/ejabberd_zlib/ejabberd_zlib.erl: Zlib support (thanks to Sergei Golovan) * src/ejabberd_zlib/ejabberd_zlib_drv.c: Likewise * src/ejabberd_zlib/Makefile.in: Likewise * src/ejabberd_c2s.erl: Stream compression support (JEP-0138) * src/ejabberd_receiver.erl: Likewise * src/mod_disco.erl: Don't split node name before calling hooks (thanks to Sergei Golovan) * src/mod_configure.erl: Support for configuration using ad-hoc commands (thanks to Sergei Golovan) * src/mod_announce.erl: Support for sending announce messages using ad-hoc commands (thanks to Sergei Golovan) * src/mod_adhoc.erl: Ad-hoc support (JEP-0050) (thanks to Magnus Henoch) * src/adhoc.erl: Likewise * src/adhoc.hrl: Likewise * src/jlib.hrl: Updated (thanks to Sergei Golovan) * src/gen_mod.erl: Added function is_loaded/2 (thanks to Sergei Golovan) * src/ejabberd_service.erl: Changed error message on handshake error (thanks to Sergei Golovan) * src/ejabberd.cfg.example: Updated (thanks to Sergei Golovan) SVN Revision: 486
135 lines
3.3 KiB
Erlang
135 lines
3.3 KiB
Erlang
%%%----------------------------------------------------------------------
|
|
%%% File : ejabberd_zlib.erl
|
|
%%% Author : Alexey Shchepin <alexey@sevcom.net>
|
|
%%% Purpose : Interface to zlib
|
|
%%% Created : 19 Jan 2006 by Alexey Shchepin <alexey@sevcom.net>
|
|
%%% Id : $Id$
|
|
%%%----------------------------------------------------------------------
|
|
|
|
-module(ejabberd_zlib).
|
|
-author('alexey@sevcom.net').
|
|
-vsn('$Revision$ ').
|
|
|
|
-behaviour(gen_server).
|
|
|
|
-export([start/0, start_link/0,
|
|
enable_zlib/2, disable_zlib/1,
|
|
send/2,
|
|
recv/2, recv/3, recv_data/2,
|
|
setopts/2,
|
|
controlling_process/2,
|
|
close/1]).
|
|
|
|
%% Internal exports, call-back functions.
|
|
-export([init/1,
|
|
handle_call/3,
|
|
handle_cast/2,
|
|
handle_info/2,
|
|
code_change/3,
|
|
terminate/2]).
|
|
|
|
-define(DEFLATE, 1).
|
|
-define(INFLATE, 2).
|
|
|
|
-record(zlibsock, {sockmod, socket, zlibport}).
|
|
|
|
start() ->
|
|
gen_server:start({local, ?MODULE}, ?MODULE, [], []).
|
|
|
|
start_link() ->
|
|
gen_server:start_link({local, ?MODULE}, ?MODULE, [], []).
|
|
|
|
init([]) ->
|
|
case erl_ddll:load_driver(ejabberd:get_so_path(), ejabberd_zlib_drv) of
|
|
ok -> ok;
|
|
{error, already_loaded} -> ok
|
|
end,
|
|
Port = open_port({spawn, ejabberd_zlib_drv}, [binary]),
|
|
{ok, Port}.
|
|
|
|
|
|
%%% --------------------------------------------------------
|
|
%%% The call-back functions.
|
|
%%% --------------------------------------------------------
|
|
|
|
handle_call(_, _, State) ->
|
|
{noreply, State}.
|
|
|
|
handle_cast(_, State) ->
|
|
{noreply, State}.
|
|
|
|
handle_info({'EXIT', Port, Reason}, Port) ->
|
|
{stop, {port_died, Reason}, Port};
|
|
|
|
handle_info({'EXIT', _Pid, _Reason}, Port) ->
|
|
{noreply, Port};
|
|
|
|
handle_info(_, State) ->
|
|
{noreply, State}.
|
|
|
|
code_change(_OldVsn, State, _Extra) ->
|
|
{ok, State}.
|
|
|
|
terminate(_Reason, Port) ->
|
|
Port ! {self, close},
|
|
ok.
|
|
|
|
|
|
enable_zlib(SockMod, Socket) ->
|
|
case erl_ddll:load_driver(ejabberd:get_so_path(), ejabberd_zlib_drv) of
|
|
ok -> ok;
|
|
{error, already_loaded} -> ok
|
|
end,
|
|
Port = open_port({spawn, ejabberd_zlib_drv}, [binary]),
|
|
{ok, #zlibsock{sockmod = SockMod, socket = Socket, zlibport = Port}}.
|
|
|
|
disable_zlib(#zlibsock{sockmod = SockMod, socket = Socket, zlibport = Port}) ->
|
|
port_close(Port),
|
|
{SockMod, Socket}.
|
|
|
|
recv(Socket, Length) ->
|
|
recv(Socket, Length, infinity).
|
|
recv(#zlibsock{sockmod = SockMod, socket = Socket} = ZlibSock,
|
|
Length, Timeout) ->
|
|
case SockMod:recv(Socket, Length, Timeout) of
|
|
{ok, Packet} ->
|
|
recv_data(ZlibSock, Packet);
|
|
{error, _Reason} = Error ->
|
|
Error
|
|
end.
|
|
|
|
recv_data(#zlibsock{zlibport = Port} = _ZlibSock, Packet) ->
|
|
case port_control(Port, ?INFLATE, Packet) of
|
|
<<0, In/binary>> ->
|
|
{ok, In};
|
|
<<1, Error/binary>> ->
|
|
{error, binary_to_list(Error)}
|
|
end.
|
|
|
|
send(#zlibsock{sockmod = SockMod, socket = Socket, zlibport = Port},
|
|
Packet) ->
|
|
case port_control(Port, ?DEFLATE, Packet) of
|
|
<<0, Out/binary>> ->
|
|
SockMod:send(Socket, Out);
|
|
<<1, Error/binary>> ->
|
|
{error, binary_to_list(Error)}
|
|
end.
|
|
|
|
|
|
setopts(#zlibsock{sockmod = SockMod, socket = Socket}, Opts) ->
|
|
case SockMod of
|
|
gen_tcp ->
|
|
inet:setopts(Socket, Opts);
|
|
_ ->
|
|
SockMod:setopts(Socket, Opts)
|
|
end.
|
|
|
|
controlling_process(#zlibsock{sockmod = SockMod, socket = Socket}, Pid) ->
|
|
SockMod:controlling_process(Socket, Pid).
|
|
|
|
close(#zlibsock{sockmod = SockMod, socket = Socket, zlibport = Port}) ->
|
|
SockMod:close(Socket),
|
|
port_close(Port).
|
|
|
|
|