| % (C) 2019 by Harald Welte <laforge@gnumonks.org> |
| % |
| % 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 <https://www.gnu.org/licenses/>. |
| % |
| % 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. |