Harald Welte | 5df8338 | 2011-03-08 15:17:32 +0100 | [diff] [blame] | 1 | % VFUK-ONW specific mgw_nat actor callback functions |
| 2 | |
| 3 | % (C) 2011 by Harald Welte <laforge@gnumonks.org> |
| 4 | % (C) 2011 OnWaves |
| 5 | % |
| 6 | % All Rights Reserved |
| 7 | % |
| 8 | % This program is free software; you can redistribute it and/or modify |
| 9 | % it under the terms of the GNU Affero General Public License as |
| 10 | % published by the Free Software Foundation; either version 3 of the |
| 11 | % License, or (at your option) any later version. |
| 12 | % |
| 13 | % This program is distributed in the hope that it will be useful, |
| 14 | % but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 15 | % MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 16 | % GNU General Public License for more details. |
| 17 | % |
| 18 | % You should have received a copy of the GNU Affero General Public License |
| 19 | % along with this program. If not, see <http://www.gnu.org/licenses/>. |
| 20 | |
| 21 | -module(mgw_nat_act_vfuk_onw). |
| 22 | -author("Harald Welte <laforge@gnumonks.org>"). |
| 23 | |
Harald Welte | 99e92c2 | 2011-04-06 17:22:30 +0200 | [diff] [blame] | 24 | -export([rewrite_actor/5, reload_config/0]). |
Harald Welte | 93db48d | 2011-03-10 00:47:36 +0100 | [diff] [blame] | 25 | -export([camelph_twalk_cb/3]). |
| 26 | |
| 27 | -include_lib("osmo_map/include/map.hrl"). |
| 28 | -include_lib("osmo_ss7/include/sccp.hrl"). |
Harald Welte | 5df8338 | 2011-03-08 15:17:32 +0100 | [diff] [blame] | 29 | |
| 30 | % Rewrite at SCTP (root) level: |
| 31 | rewrite_actor(sctp, From, Path, 2, DataBin) -> |
| 32 | try mgw_nat:mangle_rx_data(From, Path, DataBin, fun rewrite_actor/5) of |
| 33 | Val -> |
| 34 | Val |
| 35 | catch error:Error -> |
| 36 | % some parser error, simply forward msg unmodified |
Harald Welte | eee3eac | 2011-03-12 10:34:33 +0100 | [diff] [blame] | 37 | error_logger:error_report([{error, Error}, |
| 38 | {stacktrace, erlang:get_stacktrace()}]), |
Harald Welte | 5df8338 | 2011-03-08 15:17:32 +0100 | [diff] [blame] | 39 | DataBin |
| 40 | end; |
| 41 | |
| 42 | % Rewrite at MAP level: call into map_masq module |
| 43 | rewrite_actor(map, From, Path, 0, MapDec) -> |
| 44 | mangle_map_camel_phase(From, Path, MapDec); |
| 45 | |
| 46 | % Default action: no rewrite |
| 47 | rewrite_actor(_Level, _From, _Path, _MsgType, Msg) -> |
| 48 | Msg. |
| 49 | |
| 50 | |
Harald Welte | eee3eac | 2011-03-12 10:34:33 +0100 | [diff] [blame] | 51 | mangle_map_camel_phase(from_stp, _Path, MapDec) -> |
Harald Welte | 93db48d | 2011-03-10 00:47:36 +0100 | [diff] [blame] | 52 | MapDec; |
| 53 | mangle_map_camel_phase(from_msc, Path, MapDec) -> |
| 54 | % Resolve the Global Title of the SCCP Called Addr |
Harald Welte | eea20e1 | 2011-03-31 09:40:44 +0200 | [diff] [blame] | 55 | {value, #sccp_msg{parameters = SccpPars}} = lists:keysearch(sccp_msg, 1, Path), |
Harald Welte | 93db48d | 2011-03-10 00:47:36 +0100 | [diff] [blame] | 56 | CalledAddr = proplists:get_value(called_party_addr, SccpPars), |
Harald Welte | 99e92c2 | 2011-04-06 17:22:30 +0200 | [diff] [blame] | 57 | {ok, IntTbl} = application:get_env(mgw_nat, int_camel_ph_tbl), |
| 58 | case osmo_ss7_gtt:global_title_match(IntTbl, CalledAddr) of |
Harald Welte | 93db48d | 2011-03-10 00:47:36 +0100 | [diff] [blame] | 59 | false -> |
| 60 | MapDec; |
Harald Welte | 99e92c2 | 2011-04-06 17:22:30 +0200 | [diff] [blame] | 61 | PhaseL -> |
| 62 | #global_title{phone_number = PhoneNum} = CalledAddr#sccp_addr.global_title, |
| 63 | PhoneNumInt = osmo_util:digit_list2int(PhoneNum), |
Harald Welte | eea20e1 | 2011-03-31 09:40:44 +0200 | [diff] [blame] | 64 | io:format("Rewriting Camel Phase List to ~p, GT ~p~n", [PhaseL, PhoneNumInt]), |
Harald Welte | 93db48d | 2011-03-10 00:47:36 +0100 | [diff] [blame] | 65 | osmo_util:tuple_walk(MapDec, fun camelph_twalk_cb/3, [PhaseL]) |
| 66 | end. |
| 67 | |
Harald Welte | 99e92c2 | 2011-04-06 17:22:30 +0200 | [diff] [blame] | 68 | |
Harald Welte | 93db48d | 2011-03-10 00:47:36 +0100 | [diff] [blame] | 69 | % tuple tree walker callback function |
| 70 | camelph_twalk_cb(['begin','MapSpecificPDUs_begin',basicROS,invoke, |
| 71 | 'MapSpecificPDUs_begin_components_SEQOF_basicROS_invoke', |
| 72 | 'UpdateLocationArg'], VC = #'VLR-Capability'{}, [PhaseL|_Args]) -> |
| 73 | % Manipulate the VLR capabilities in UpdateLocationArg |
| 74 | VC#'VLR-Capability'{supportedCamelPhases = PhaseL}; |
| 75 | camelph_twalk_cb(_Path, Msg, _Args) -> |
| 76 | % Default case: simply return the unmodified tuple |
| 77 | Msg. |
Harald Welte | 99e92c2 | 2011-04-06 17:22:30 +0200 | [diff] [blame] | 78 | |
| 79 | |
| 80 | gen_int_camelph_tbl(L) -> |
| 81 | gen_int_camelph_tbl(L, []). |
| 82 | gen_int_camelph_tbl([], Out) -> |
| 83 | Out; |
| 84 | gen_int_camelph_tbl([{GttPart, PhasePart}|Tail], Out) -> |
| 85 | GttMatch = osmo_ss7_gtt:'#new-gtt_match'(GttPart), |
| 86 | % Fixme: use ordered insert! |
| 87 | gen_int_camelph_tbl(Tail, Out ++ [{GttMatch, PhasePart}]). |
| 88 | |
| 89 | reload_config() -> |
| 90 | {ok, CamelPatchTblIn} = application:get_env(mgw_nat, camel_phase_patch_table), |
| 91 | io:format("VFUK-ONW actor: reloading config ~p~n", [CamelPatchTblIn]), |
| 92 | try gen_int_camelph_tbl(CamelPatchTblIn) of |
| 93 | TblOut -> |
| 94 | application:set_env(mgw_nat, int_camel_ph_tbl, TblOut) |
| 95 | catch error:Error -> |
| 96 | error_logger:error_report([{error, Error}, |
| 97 | {stacktrace, erlang:get_stacktrace()}]) |
| 98 | end. |