% Osmocom Erlang utility functions % (C) 2011 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 http://www.erlang.org (or a modified version of these % libraries), containing parts covered by the terms of the Erlang Public % License (http://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_util). -author('Harald Welte '). -export([digit_list2int/1, int2digit_list/1]). -export([reload_config/0]). -export([tuple_walk/3, tuple_walk_print_cb/3]). -export([make_prim/4, make_prim/3]). -export([pointcode2int/1, pointcode2int/2, pointcode_fmt/2]). -export([asn_val/1]). -export([get_env/2, get_env/3]). -include("osmo_util.hrl"). -compile({parse_transform, exprecs}). -export_records([primitive]). % Convert a list of digits to an integer value digit_list2int(Int, []) -> Int; digit_list2int(Int, [Digit|Tail]) -> digit_list2int(Int*10 + Digit, Tail). digit_list2int(Digits) when is_list(Digits) -> digit_list2int(0, Digits). % Convert an integer value into a list of decimal digits int2digit_list(0, Digits) when is_list(Digits) -> Digits; int2digit_list(Int, Digits) when is_integer(Int), is_list(Digits) -> Digit = Int rem 10, int2digit_list(Int div 10, [Digit|Digits]). int2digit_list(Int) when is_integer(Int) -> int2digit_list(Int, []). % reload configuration of an application reload_config() -> case init:get_argument(config) of {ok, [ Files ]} -> ConfFiles = [begin S = filename:basename(F,".config"), filename:join(filename:dirname(F), S ++ ".config") end || F <- Files], % Move sys.config to the head of the list Config = lists:sort(fun("sys.config", _) -> true; (_, _) -> false end, ConfFiles), OldEnv = application_controller:prep_config_change(), Apps = [{application, A, make_appl(A)} || {A,_,_} <- application:which_applications()], application_controller:change_application_data(Apps, Config), application_controller:config_change(OldEnv); _ -> {ok, []} end. make_appl(App) when is_atom(App) -> AppList = element(2,application:get_all_key(App)), FullName = code:where_is_file(atom_to_list(App) ++ ".app"), case file:consult(FullName) of {ok, [{application, _, Opts}]} -> Env = proplists:get_value(env, Opts, []), lists:keyreplace(env, 1, AppList, {env, Env}); {error, _Reason} -> lists:keyreplace(env, 1, AppList, {env, []}) end. % Walk a named tuple and (recursively) all its fields, call user-supplied % callback for each of them tuple_walk(Tpl, TupleCb, Args) when is_tuple(Tpl), is_function(TupleCb), is_list(Args) -> tuple_walk([], Tpl, TupleCb, Args). tuple_walk(Path, Tpl, TupleCb, Args) when is_list(Path), is_tuple(Tpl), is_list(Args) -> % call Callback RetVal = TupleCb(Path, Tpl, Args), if is_tuple(RetVal) -> [TplName|TplList] = tuple_to_list(RetVal), NewTplList = tuple_fieldlist_walk(Path, TplName, TplList, TupleCb, Args), list_to_tuple([TplName|NewTplList]); true -> RetVal end; tuple_walk(Path, TplL, TupleCb, Args) when is_list(Path), is_list(TplL), is_list(Args) -> tuple_walk_list(Path, TplL, TupleCb, Args, []). tuple_walk_list(_Path, [], _TupleCb, _Args, OutList) -> OutList; tuple_walk_list(Path, [Head|Tail], TupleCb, Args, OutList) -> if is_tuple(Head) -> NewHead = tuple_walk(Path, Head, TupleCb, Args); is_list(Head) -> NewHead = tuple_walk(Path, Head, TupleCb, Args); true -> NewHead = Head end, tuple_walk_list(Path, Tail, TupleCb, Args, OutList++[NewHead]). tuple_fieldlist_walk(Path, TplName, FieldList, TupleCb, Args) -> tuple_fieldlist_walk(Path, TplName, FieldList, TupleCb, Args, []). tuple_fieldlist_walk(_Path, _TplName, [], _TplCb, _Args, OutList) -> OutList; tuple_fieldlist_walk(Path, TplName, [Head|List], TupleCb, Args, OutList) -> if is_tuple(Head) -> NewHead = tuple_walk(Path++[TplName], Head, TupleCb, Args); is_list(Head) -> NewHead = tuple_walk(Path++[TplName], Head, TupleCb, Args); true -> NewHead = Head end, tuple_fieldlist_walk(Path, TplName, List, TupleCb, Args, OutList++[NewHead]). tuple_walk_print_cb(Path, Tpl, _Args) when is_list(Path), is_tuple(Tpl) -> io:format("~p:~p~n", [Path, Tpl]), Tpl. % helper function to create a #primitive record make_prim(Subsys, GenName, SpecName) -> make_prim(Subsys, GenName, SpecName, []). make_prim(Subsys, GenName, SpecName, Param) -> #primitive{subsystem = Subsys, gen_name = GenName, spec_name = SpecName, parameters = Param}. % parse a 3-tuple pointcode into a raw integer pointcode2int(Int) when is_integer(Int) -> Int; pointcode2int(undefined) -> undefined; pointcode2int(#pointcode{repr=Type, value=Value}) -> pointcode2int(Type, Value); pointcode2int({Std, Param}) -> pointcode2int(Std, Param). pointcode2int(itu, {A, B, C}) -> <> = <>, PcInt; pointcode2int(ansi, {A, B, C}) -> <> = <>, PcInt; pointcode2int(ttc, {A, B, C}) -> <> = <>, PcInt. % format a point-code into a 3-tuple according to the standard used pointcode_fmt(Std, P) when is_binary(P) -> <> = P, pointcode_fmt(Std, PcInt); pointcode_fmt(itu, PcInt) when is_integer(PcInt) -> <> = <>, {pointcode, itu, {A, B, C}}; pointcode_fmt(ansi, PcInt) -> <> = <>, {pointcode, ansi, {A, B, C}}; pointcode_fmt(ttc, PcInt) -> <> = <>, {pointcode, ttc, {A, B, C}}. asn_val(undefined) -> asn1_NOVALUE; asn_val([]) -> asn1_NOVALUE; asn_val(Foo) -> Foo. % wrapper around application:get_env() thwowing exception on undef get_env(App, Var) when is_atom(App), is_atom(Var) -> case application:get_env(App, Var) of undefined -> throw(undefined); {ok, Value} -> Value end. get_env(App, Var, Default) -> case application:get_env(App, Var) of undefined -> Default; {ok, Value} -> Value end.