aboutsummaryrefslogtreecommitdiffstats
path: root/src/mgw_nat_usr.erl
blob: 85033335d3e208cbc39fdb6b988c29b87cfdde44 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
% Wrapper code, wrapping sctp_handler.erl into OTP gen_server

% (C) 2011 by Harald Welte <laforge@gnumonks.org>
% (C) 2011 OnWaves
%
% All Rights Reserved
%
% This program is free software; you can redistribute it and/or modify
% it under the terms of the GNU Affero General Public License as
% published by the Free Software Foundation; either version 3 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 Affero General Public License
% along with this program.  If not, see <http://www.gnu.org/licenses/>.

-module(mgw_nat_usr).
-author("Harald Welte <laforge@gnumonks.org>").

-behavior(gen_server).

-export([start_link/1, stop/0, sccp_masq_reset/0, sccp_masq_dump/0,
	 reload_config/0]).
-export([init/1, handle_cast/2, handle_info/2, terminate/2]).


start_link(Params) ->
	gen_server:start_link({local, ?MODULE}, ?MODULE, Params, []).

stop() ->
	gen_server:cast(?MODULE, stop).

sccp_masq_reset() ->
	gen_server:cast(?MODULE, sccp_masq_reset).

sccp_masq_dump() ->
	gen_server:cast(?MODULE, sccp_masq_dump).

reload_config() ->
	gen_server:cast(?MODULE, reload_config).

%% Callback functions of the OTP behavior

init(_Params) ->
	sccp_masq:init(),
	map_masq:config_update(),
	MscLocalIp = get_app_config(msc_local_ip),
	MscLocalPort = get_app_config(msc_local_port),
	MscRemoteIp = get_app_config(msc_remote_ip),
	StpRemoteIp = get_app_config(stp_remote_ip),
	StpRemotePort = get_app_config(stp_remote_port),
	RewriteActMod = get_app_config(rewrite_act_mod),
	RewriteActMod:reload_config(),
	io:format("Starting mgw_nat_usr with rewrite actor module ~p~n", [RewriteActMod]),
	SctpHdlrArgs =	[MscLocalIp, MscLocalPort, MscRemoteIp,
			 StpRemoteIp, StpRemotePort, RewriteActMod],
	apply(sctp_handler, init, SctpHdlrArgs).

handle_cast(stop, LoopData) ->
	{stop, normal, LoopData};

handle_cast(sccp_masq_reset, LoopData) ->
	sccp_masq:reset(),
	{noreply, LoopData};

handle_cast(sccp_masq_dump, LoopData) ->
	sccp_masq:dump(),
	{noreply, LoopData};

handle_cast(reload_config, LoopData) ->
	RewriteActMod = get_app_config(rewrite_act_mod),
	RewriteActMod:reload_config(),
	{noreply, LoopData}.


terminate(_Reason, _LoopData) ->
	ok.

% callback for other events like incoming SCTP message
handle_info({sctp, Sock, Ip, Port, Data}, LoopData) ->
	NewL = sctp_handler:handle_sctp(LoopData, {sctp, Sock, Ip, Port, Data}),
	{noreply, NewL}.

get_app_config(Name) ->
	case application:get_env(Name) of
	    undefined ->
		error_logger:error_report([{error, app_cfg_missing},
					   {get_app_config, Name}]),
		throw(app_cfg_missing);
	    {ok, Val} ->
		Val
	end.