% (C) 2019 by Harald Welte % % 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 . % % Additional Permission under GNU AGPL version 3 section 7: % % If you modify this Program, or any covered work, by linking or % combining it with runtime libraries of Erlang/OTP as released by % Ericsson on https://www.erlang.org (or a modified version of these % libraries), containing parts covered by the terms of the Erlang Public % License (https://www.erlang.org/EPLICENSE), the licensors of this % Program grant you additional permission to convey the resulting work % without the need to license the runtime libraries of Erlang/OTP under % the GNU Affero General Public License. Corresponding Source for a % non-source form of such a combination shall include the source code % for the parts of the runtime libraries of Erlang/OTP used as well as % that of the covered work. -module(osmo_dia2gsup). -behavior(gen_server). -include_lib("diameter/include/diameter.hrl"). -include_lib("diameter/include/diameter_gen_base_rfc6733.hrl"). %-include_lib("diameter_settings.hrl"). -export([main/1]). % API -export([start_link/0]). -export([start/0, stop/0]). % gen_server callbacks -export([init/1, handle_call/3, handle_cast/2, handle_info/2]). -export([code_change/3, terminate/2]). -define(SERVER, ?MODULE). % Diameter application definitions -define(DIA_STATS_TAB, iwf_stats). -define(DIA_STATS_COUNTERS, [event_OK, event_ERR]). -define(SVC_NAME, ?MODULE). -define(APP_ALIAS, ?MODULE). -define(CALLBACK_MOD, server_cb). -define(DIAMETER_DICT_HSS, diameter_3gpp_ts29_272). -define(APPID_S6, #'diameter_base_Vendor-Specific-Application-Id'{'Vendor-Id'=10515, 'Auth-Application-Id'=[16777251]}). -define(SERVICE(Name), [{'Origin-Host', application:get_env(osmo_dia2gsup, origin_host, "hss.localdomain")}, {'Origin-Realm', application:get_env(osmo_dia2gsup, origin_realm, "localdomain")}, {'Vendor-Id', application:get_env(osmo_dia2gsup, vendor_id, 0)}, {'Origin-State-Id', diameter:origin_state_id()}, {'Product-Name', "osmo_dia2gsup"}, {'Auth-Application-Id', []}, {'Vendor-Specific-Application-Id', [?APPID_S6]}, {application, [{alias, ?APP_ALIAS}, {dictionary, ?DIAMETER_DICT_HSS}, {module, ?CALLBACK_MOD}] }]). %% ------------------------------------------------------------------ %% API %% ------------------------------------------------------------------ start_link() -> gen_server:start_link({local, ?SERVER}, ?MODULE, [], []). start() -> application:ensure_all_started(?MODULE), start_link(). stop() -> gen_server:cast(?SERVER, stop). main(_Args) -> application:ensure_all_started(?MODULE), timer:sleep(infinity). %% ------------------------------------------------------------------ %% gen_server Function Definitions %% ------------------------------------------------------------------ %% @callback gen_server init(State) -> % DIAMETER side SvcName = ?MODULE, diameter:start_service(SvcName, ?SERVICE(SvcName)), Ip = application:get_env(osmo_dia2gsup, diameter_ip, "127.0.0.8"), Port = application:get_env(osmo_dia2gsup, diameter_port, 3868), Proto = application:get_env(osmo_dia2gsup, diameter_proto, sctp), ConnectTimer = application:get_env(osmo_dia2gsup, diameter_connect_timer, 30000), listen(?SVC_NAME, {address, Proto, element(2,inet:parse_address(Ip)), Port}, {timer, ConnectTimer}), lager:info("Diameter HSS Application started on IP ~s, ~p port ~p~n", [Ip, Proto, Port]), {ok, State}. %% @callback gen_server handle_call(_Req, _From, State) -> {noreply, State}. %% @callback gen_server handle_cast(stop, State) -> {stop, normal, State}; handle_cast(_req, State) -> {noreply, State}. %% @callback gen_server handle_info(_Info, State) -> {noreply, State}. %% @callback gen_server code_change(_OldVsn, State, _Extra) -> {ok, State}. %% @callback gen_server terminate(normal, _State) -> diameter:stop_service(?SVC_NAME), lager:info("Diameter HSS Application stopped.~n"), ok; terminate(shutdown, _State) -> ok; terminate({shutdown, _Reason}, _State) -> ok; terminate(_Reason, _State) -> ok. %% ------------------------------------------------------------------ %% Internal Function Definitions %% ------------------------------------------------------------------ listen(Name, {address, Protocol, IPAddr, Port}, {timer, ConnectTimer}) -> TransOpts = [{transport_module, tmod(Protocol)}, {transport_config, [{reuseaddr, true}, {ip, IPAddr}, {port, Port}]}, {connect_timer, ConnectTimer}], {ok, _} = diameter:add_transport(Name, {listen, TransOpts}). tmod(tcp) -> diameter_tcp; tmod(sctp) -> diameter_sctp.