%%--------------------------------------------------------------------
%%
%% %CopyrightBegin%
%%
%% Copyright Ericsson AB 1999-2016. All Rights Reserved.
%%
%% Licensed under the Apache License, Version 2.0 (the "License");
%% you may not use this file except in compliance with the License.
%% You may obtain a copy of the License at
%%
%% http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing, software
%% distributed under the License is distributed on an "AS IS" BASIS,
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%% See the License for the specific language governing permissions and
%% limitations under the License.
%%
%% %CopyrightEnd%
%%
%%
%%-----------------------------------------------------------------
%% File: orber_interceptors.erl
%%
%% Description:
%% This file contains the code for calling interceptors
%%
%%-----------------------------------------------------------------
-module(orber_interceptors).
-include_lib("orber/include/corba.hrl").
%%-----------------------------------------------------------------
%% External exports
%%-----------------------------------------------------------------
-export([call_send_message_interceptors/2, call_receive_message_interceptors/1,
call_request_interceptors/2]).
-export([push_system_message_interceptor/2, pop_system_message_interceptor/1,
create_interceptor_table/0]).
%%-----------------------------------------------------------------
%% Internal exports
%%-----------------------------------------------------------------
-export([]).
%%-----------------------------------------------------------------
%% External interface functions
%%-----------------------------------------------------------------
call_receive_message_interceptors(Bytes) ->
case getInMessageInterceptors() of
[] ->
Bytes;
Interceptors ->
apply_message_interceptors(Interceptors, receive_message, corba:create_nil_objref(),
lists:flatten(Bytes))
end.
call_send_message_interceptors(ObjRef, Bytes) ->
case getOutMessageInterceptors() of
[] ->
Bytes;
Interceptors ->
apply_message_interceptors(Interceptors, send_message, ObjRef, lists:flatten(Bytes))
end.
call_request_interceptors(in, Bytes) ->
case getInRequestInterceptors() of
[] ->
Bytes;
Interceptors ->
Bytes
end;
call_request_interceptors(out, Bytes) ->
case getOutRequestInterceptors() of
[] ->
Bytes;
Interceptors ->
Bytes
end.
create_interceptor_table() ->
%% Should be replicated mnesia
ets:new(orber_interceptors, [protected, named_table, set]),
ets:insert(orber_interceptors, {message_in_interceptors, []}),
ets:insert(orber_interceptors, {message_out_interceptors, []}).
push_system_message_interceptor(in, Mod) ->
case ets:lookup(orber_interceptors, message_in_interceptors) of
[{_, Interceptors}] ->
ets:insert(orber_interceptors, {message_in_interceptors, [Mod | Interceptors]});
_ ->
corba:raise(#'INTERNAL'{completion_status=?COMPLETED_NO})
end;
push_system_message_interceptor(out, Mod) ->
case ets:lookup(orber_interceptors, message_out_interceptors) of
[{_, Interceptors}] ->
ets:insert(orber_interceptors, {message_out_interceptors, Interceptors ++ [Mod]});
_ ->
corba:raise(#'INTERNAL'{completion_status=?COMPLETED_NO})
end.
pop_system_message_interceptor(in) ->
case ets:lookup(orber_interceptors, message_in_interceptors) of
[{_, []}] ->
ok;
[{_, [_ | Interceptors]}] ->
ets:insert(orber_interceptors, {message_in_interceptors, Interceptors});
_ ->
corba:raise(#'INTERNAL'{completion_status=?COMPLETED_NO})
end;
pop_system_message_interceptor(out) ->
case ets:lookup(orber_interceptors, message_out_interceptors) of
[{_, []}] ->
ok;
[{_, Interceptors}] ->
ets:insert(orber_interceptors, {message_out_interceptors, lists:droplast(Interceptors)});
_ ->
corba:raise(#'INTERNAL'{completion_status=?COMPLETED_NO})
end.
%%-----------------------------------------------------------------
%% Internal functions
%%-----------------------------------------------------------------
getInMessageInterceptors() ->
case ets:lookup(orber_interceptors, message_in_interceptors) of
[{_, Interceptors}] ->
Interceptors;
_ ->
corba:raise(#'INTERNAL'{completion_status=?COMPLETED_NO})
end.
getOutMessageInterceptors() ->
case ets:lookup(orber_interceptors, message_out_interceptors) of
[{_, Interceptors}] ->
Interceptors;
_ ->
corba:raise(#'INTERNAL'{completion_status=?COMPLETED_NO})
end.
getInRequestInterceptors() ->
[].
getOutRequestInterceptors() ->
[].
apply_message_interceptors([], F, ObjRef, Bytes) ->
Bytes;
apply_message_interceptors([M | Rest], F, ObjRef, Bytes) ->
apply_message_interceptors(Rest, F, ObjRef, apply(M, F, [ObjRef, Bytes])).
