Preliminary support for multiple domains (e.g. IPv6).

1 files changed, 106 insertions, 51 deletions

diff --git a/lib/snmp/src/agent/snmp_target_mib.erl b/lib/snmp/src/agent/snmp_target_mib.erl
index 270a5fd5b6..c76e4bb173 100644
--- a/lib/snmp/src/agent/snmp_target_mib.erl
+++ b/lib/snmp/src/agent/snmp_target_mib.erl
@@ -30,12 +30,13 @@
 	 add_params/5, delete_params/1]).
 -export([check_target_addr/1, check_target_params/1]).
 
--include("snmp_types.hrl").
--include("snmp_tables.hrl").
--include("SNMP-TARGET-MIB.hrl").
--include("SNMPv2-TC.hrl").
--include("SNMPv2-TM.hrl").
--include("SNMP-FRAMEWORK-MIB.hrl").
+-include_lib("snmp/include/snmp_types.hrl").
+-include_lib("snmp/include/snmp_tables.hrl").
+-include_lib("snmp/include/SNMP-TARGET-MIB.hrl").
+-include_lib("snmp/include/SNMPv2-TC.hrl").
+-include_lib("snmp/include/SNMPv2-TM.hrl").
+-include_lib("snmp/include/SNMP-FRAMEWORK-MIB.hrl").
+-include_lib("snmp/include/TRANSPORT-ADDRESS-MIB.hrl").
 
 -define(VMODULE,"TARGET-MIB").
 -include("snmp_verbosity.hrl").
@@ -139,39 +140,53 @@ read_target_config_files(Dir) ->
 %%  {Name, Ip, Udp, Timeout, RetryCount, TagList, Params, EngineId,
 %%   TMask, MMS}
 %%-----------------------------------------------------------------
-check_target_addr({Name, Ip, Udp, Timeout, RetryCount, TagList,
-                   Params, EngineId, TMask, MMS}) ->
+default_domain() ->
+    snmpUDPDomain.
+
+check_target_addr({Name, Domain, Ip, Udp, Timeout, RetryCount, TagList,
+		   Params, EngineId, TMask, MMS}) ->
     ?vtrace("check target address with:"
-	"~n   Name:       ~s"
-	"~n   Ip:         ~p"
-	"~n   Udp:        ~p"
-	"~n   Timeout:    ~p"
-	"~n   RetryCount: ~p"
-	"~n   TagList:    ~p"
-	"~n   Params:     ~p"
-	"~n   EngineId:   ~p"
-	"~n   TMask:      ~p"
-	"~n   MMS:        ~p",
-    [Name,Ip,Udp,Timeout,RetryCount,
-     TagList,Params,EngineId,TMask,MMS]),
+	    "~n   Name:       ~s"
+	    "~n   Domain:     ~p"
+	    "~n   Ip:         ~p"
+	    "~n   Udp:        ~p"
+	    "~n   Timeout:    ~p"
+	    "~n   RetryCount: ~p"
+	    "~n   TagList:    ~p"
+	    "~n   Params:     ~p"
+	    "~n   EngineId:   ~p"
+	    "~n   TMask:      ~p"
+	    "~n   MMS:        ~p",
+	    [Name, 
+	     Domain, Ip, Udp, 
+	     Timeout, RetryCount,
+	     TagList, Params, EngineId, TMask, MMS]),
     snmp_conf:check_string(Name,{gt,0}),
-    snmp_conf:check_ip(Ip),
+    snmp_conf:check_domain(Domain),
+    snmp_conf:check_ip(Domain, Ip),
     snmp_conf:check_integer(Udp, {gt, 0}),
     snmp_conf:check_integer(Timeout, {gte, 0}),
     snmp_conf:check_integer(RetryCount, {gte,0}),
     snmp_conf:check_string(TagList),
     snmp_conf:check_string(Params),
     check_engine_id(EngineId),
-    TAddr = Ip ++ [Udp div 256, Udp rem 256],
-    check_mask(TMask, TAddr),
+    TAddress = snmp_conf:mk_taddress(Domain, Ip, Udp),
+    TDomain  = snmp_conf:mk_tdomain(Domain), 
+    check_mask(TDomain, TMask, TAddress),
     snmp_conf:check_packet_size(MMS),
     ?vtrace("check target address done",[]),
-    
-    Addr = {Name, ?snmpUDPDomain, TAddr, Timeout,
+    Addr = {Name, TDomain, TAddress, Timeout,
 	    RetryCount, TagList, Params,
 	    ?'StorageType_nonVolatile', ?'RowStatus_active', EngineId,
             TMask, MMS}, % Values for Augmenting table in SNMP-COMMUNITY-MIB
     {ok, Addr};
+check_target_addr({Name, Ip, Udp, Timeout, RetryCount, TagList,
+                   Params, EngineId, TMask, MMS}) ->
+    Domain = default_domain(), 
+    check_target_addr({Name, 
+		       Domain, Ip, Udp, 
+		       Timeout, RetryCount, TagList,
+		       Params, EngineId, TMask, MMS});
 check_target_addr({Name, Ip, Udp, Timeout, RetryCount, TagList, Params, 
                    EngineId}) ->
     check_target_addr({Name, Ip, Udp, Timeout, RetryCount, TagList,
@@ -194,11 +209,12 @@ check_engine_id(discovery) ->
 check_engine_id(EngineId) ->
     snmp_conf:check_string(EngineId).
 
-check_mask([], _TAddr) ->
+
+check_mask(_TDomain, [], _TAddress) ->
     ok;
-check_mask(TMask, TAddr) when length(TMask) == length(TAddr) ->
-    snmp_conf:check_taddress(TMask);
-check_mask(TMask, _TAddr) ->
+check_mask(TDomain, TMask, TAddress) when length(TMask) =:= length(TAddress) ->
+    snmp_conf:check_taddress(TDomain, TMask);
+check_mask(_TDomain, TMask, _TAddr) ->
     throw({error, {invalid_mask, TMask}}).
 
 
@@ -261,7 +277,13 @@ table_del_row(Tab, Key) ->
 
 add_addr(Name, Ip, Port, Timeout, Retry, TagList, 
 	 Params, EngineId, TMask, MMS) ->
-    Addr = {Name, Ip, Port, Timeout, Retry, TagList, 
+    Domain = default_domain(), 
+    add_addr(Name, Domain, Ip, Port, Timeout, Retry, TagList, 
+	     Params, EngineId, TMask, MMS).
+
+add_addr(Name, Domain, Ip, Port, Timeout, Retry, TagList, 
+	 Params, EngineId, TMask, MMS) ->
+    Addr = {Name, Domain, Ip, Port, Timeout, Retry, TagList, 
 	    Params, EngineId, TMask, MMS},
     case (catch check_target_addr(Addr)) of
 	{ok, Row} ->
@@ -341,8 +363,11 @@ maybe_create_var(Var) ->
 init_var(Var) -> ets:insert(snmp_agent_table, {Var, 0}).
 
 vars() ->
-    [snmpUnavailableContexts,
-     snmpUnknownContexts].
+    [
+     snmpUnavailableContexts,
+     snmpUnknownContexts
+    ].
+
 
 %%-----------------------------------------------------------------
 %% API functions
@@ -562,6 +587,8 @@ snmpTargetAddrTable(print) ->
 				 Prefix, element(?snmpTargetAddrTDomain, Row),
 				 case element(?snmpTargetAddrTDomain, Row) of
 				     ?snmpUDPDomain -> udp;
+				     ?transportDomainUdpIpv4 -> udpIpv4;
+				     ?transportDomainUdpIpv6 -> udpIpv6;
 				     _ -> undefined
 				 end,
 				 Prefix, element(?snmpTargetAddrTAddress, Row),
@@ -611,13 +638,13 @@ snmpTargetAddrTable(get_next, RowIndex, Cols) ->
     next(snmpTargetAddrTable, RowIndex, Cols);
 snmpTargetAddrTable(set, RowIndex, Cols0) ->
     %% BMK BMK BMK
-    case (catch verify_targetAddrTable_cols(Cols0, [])) of
+    case (catch verify_targetAddrTable_cols(Cols0)) of
 	{ok, Cols} ->
 	    snmp_notification_mib:invalidate_cache(),
 	    %% Add columns for augmenting table snmpTargetAddrExtTable and for
-	    %% target engine ID.  Target engine ID is set to "".  The function
+	    %% target engine ID. Target engine ID is set to "". The function
 	    %% get_target_engine_id will return "" unless a value is set using
-	    %% set_target_engine_id.  If it is "" Informs can't be sent to the
+	    %% set_target_engine_id. If it is "" Informs can't be sent to the
 	    %% target.
 	    NCols = Cols ++ [{?snmpTargetAddrEngineId, ""},
 			     {?snmpTargetAddrTMask, []},
@@ -628,12 +655,12 @@ snmpTargetAddrTable(set, RowIndex, Cols0) ->
 	    Error
     end;
 snmpTargetAddrTable(is_set_ok, RowIndex, Cols0) ->
-    case (catch verify_targetAddrTable_cols(Cols0, [])) of
+    case (catch verify_targetAddrTable_cols(Cols0)) of
 	{ok, Cols} ->
 	    %% Add columns for augmenting table snmpTargetAddrExtTable and for
-	    %% target engine ID.  Target engine ID is set to "".  The function
+	    %% target engine ID. Target engine ID is set to "". The function
 	    %% get_target_engine_id will return "" unless a value is set using
-	    %% set_target_engine_id.  If it is "" Informs can't be sent to the
+	    %% set_target_engine_id. If it is "" Informs can't be sent to the
 	    %% target.
 	    NCols = Cols ++ [{?snmpTargetAddrEngineId, ""},
 			     {?snmpTargetAddrTMask, []},
@@ -647,55 +674,83 @@ snmpTargetAddrTable(Op, Arg1, Arg2) ->
     Db = db(snmpTargetAddrTable), 
     snmp_generic:table_func(Op, Arg1, Arg2, Db).
 
+verify_targetAddrTable_cols(Cols) ->
+    ValidCols0 = verify_targetAddrTable_cols(Cols, []),
+    %% Make a last pass to verify TDomain and TAddress.
+    ValidCols0.
+
 verify_targetAddrTable_cols([], Cols) ->
     {ok, lists:reverse(Cols)};
-verify_targetAddrTable_cols([{Col, Val0}|Cols], Acc) ->
-    Val = verify_targetAddrTable_col(Col, Val0),
-    verify_targetAddrTable_cols(Cols, [{Col, Val}|Acc]).
+verify_targetAddrTable_cols([{Col, Val0}|Cols], ValidCols) ->
+    Val = verify_targetAddrTable_col(Col, Val0, ValidCols),
+    verify_targetAddrTable_cols(Cols, [{Col, Val}|ValidCols]).
 
-verify_targetAddrTable_col(?snmpTargetAddrName, Name) ->
+verify_targetAddrTable_col(?snmpTargetAddrName, Name, _) ->
     case (catch snmp_conf:check_string(Name)) of
 	ok ->
 	    Name;
 	_ ->
 	    wrongValue(?snmpTargetAddrName)
     end;
-verify_targetAddrTable_col(?snmpTargetAddrTAddress, TAddr) ->
-    case (catch snmp_conf:check_taddress(TAddr)) of
+verify_targetAddrTable_col(?snmpTargetAddrTDomain, TDomain, _) ->
+    case (catch snmp_conf:check_tdomain(TDomain)) of
 	ok ->
-	    TAddr;
+	    TDomain;
 	_ ->
-	    wrongValue(?snmpTargetAddrTAddress)
+	    wrongValue(?snmpTargetAddrTDomain)
+    end;
+%% In order to (properly) validate the TAddress, 
+%% the TDomain must already have been validated 
+%% (the format of TAddress depends on TDomain).
+verify_targetAddrTable_col(?snmpTargetAddrTAddress, TAddress, ValidCols) ->
+    case lists:keysearch(?snmpTargetAddrTDomain, 1, ValidCols) of
+	{value, {?snmpTargetAddrTDomain, TDomain}} ->
+	    case (catch snmp_conf:check_taddress(TDomain, TAddress)) of
+		ok ->
+		    TAddress;
+		_ ->
+		    wrongValue(?snmpTargetAddrTAddress)
+	    end;
+	false ->
+	    %% The user did not provide us with a TDomain, which 
+	    %% must mean that he/she intends to use the old domain. 
+	    TDomain = snmp_misc:mk_tdomain(default_domain()),
+	    case (catch snmp_conf:check_taddress(TDomain, TAddress)) of
+		ok ->
+		    TAddress;
+		_ ->
+		    wrongValue(?snmpTargetAddrTAddress)
+	    end	    
     end;
-verify_targetAddrTable_col(?snmpTargetAddrTimeout, Timeout) ->
+verify_targetAddrTable_col(?snmpTargetAddrTimeout, Timeout, _) ->
     case (catch snmp_conf:check_integer(Timeout)) of
 	ok when Timeout >= 0 ->
 	    Timeout;
 	_ ->
 	    wrongValue(?snmpTargetAddrTimeout)
     end;
-verify_targetAddrTable_col(?snmpTargetAddrRetryCount, Retry) ->
+verify_targetAddrTable_col(?snmpTargetAddrRetryCount, Retry, _) ->
     case (catch snmp_conf:check_integer(Retry)) of
 	ok when Retry >= 0 ->
 	    Retry;
 	_ ->
 	    wrongValue(?snmpTargetAddrRetryCount)
     end;
-verify_targetAddrTable_col(?snmpTargetAddrTagList, TagList) ->
+verify_targetAddrTable_col(?snmpTargetAddrTagList, TagList, _) ->
     case (catch snmp_conf:check_string(TagList)) of
 	ok ->
 	    TagList;
 	_ ->
 	    wrongValue(?snmpTargetAddrTagList)
     end;
-verify_targetAddrTable_col(?snmpTargetAddrParams, Params) ->
+verify_targetAddrTable_col(?snmpTargetAddrParams, Params, _) ->
     case (catch snmp_conf:check_string(Params)) of
 	ok ->
 	    Params;	
 	_ ->
 	    wrongValue(?snmpTargetAddrParams)
     end;
-verify_targetAddrTable_col(_, Val) ->
+verify_targetAddrTable_col(_, Val, _) ->
     Val.