Merge branch 'bmk/snmp/snmp424_integration/r16' into maint

1 files changed, 120 insertions, 87 deletions

diff --git a/lib/snmp/src/agent/snmpa_mib.erl b/lib/snmp/src/agent/snmpa_mib.erl
index 575a018c0c..031309b990 100644
--- a/lib/snmp/src/agent/snmpa_mib.erl
+++ b/lib/snmp/src/agent/snmpa_mib.erl
@@ -1,7 +1,7 @@
 %%
 %% %CopyrightBegin%
 %%
-%% Copyright Ericsson AB 1996-2012. All Rights Reserved.
+%% Copyright Ericsson AB 1996-2013. All Rights Reserved.
 %%
 %% The contents of this file are subject to the Erlang Public License,
 %% Version 1.1, (the "License"); you may not use this file except in
@@ -18,7 +18,6 @@
 %%
 -module(snmpa_mib).
 
-%% c(snmpa_mib).
 
 %%%-----------------------------------------------------------------
 %%% This module implements a MIB server.
@@ -75,12 +74,14 @@
 %% Internal Data structures
 %%
 %%   State
-%%       data - is the MIB data (defined in snmpa_mib_data)
+%%       data - is the MIB data (defined in mib_data module)
 %%       meo  - mib entry override
 %%       teo  - trap (notification) entry override
 %%-----------------------------------------------------------------
--record(state, {data, meo, teo, backup, 
-		cache, cache_tmr, cache_autogc, cache_gclimit, cache_age}).
+-record(state, 
+	{data, meo, teo, backup, 
+	 cache, cache_tmr, cache_autogc, cache_gclimit, cache_age, 
+	 data_mod}).
 
 
 
@@ -224,9 +225,9 @@ info(MibServer, Type) ->
     call(MibServer, {info, Type}).
 
 dump(MibServer) ->
-    call(MibServer, dump).
+    dump(MibServer, io).
 
-dump(MibServer, File) when is_list(File) ->
+dump(MibServer, File) when (File =:= io) orelse is_list(File) ->
     call(MibServer, {dump, File}).
 
 backup(MibServer, BackupDir) when is_list(BackupDir) ->
@@ -256,7 +257,7 @@ init([Prio, Mibs, Opts]) ->
 do_init(Prio, Mibs, Opts) ->
     process_flag(priority, Prio),
     process_flag(trap_exit, true),
-    put(sname,ms),
+    put(sname, ms),
     put(verbosity, ?vvalidate(get_verbosity(Opts))),
     ?vlog("starting",[]),
 
@@ -289,13 +290,19 @@ do_init(Prio, Mibs, Opts) ->
     MeOverride = get_me_override(Opts),
     TeOverride = get_te_override(Opts),
     MibStorage = get_mib_storage(Opts),
-    Data       = snmpa_mib_data:new(MibStorage),
-    ?vtrace("init -> mib data created",[]),
-    case (catch mib_operations(load_mib, Mibs, Data, 
+    MibDataMod = get_data_mod(Opts), 
+    ?vtrace("init -> try create mib data with"
+	    "~n   MeOverride: ~p"
+	    "~n   TeOverride: ~p"
+	    "~n   MibStorage: ~p", [MeOverride, TeOverride, MibStorage]),
+    Data       = MibDataMod:new(MibStorage),
+    ?vdebug("init -> mib data created", []),
+    case (catch mib_operations(MibDataMod, 
+			       load_mib, Mibs, Data, 
 			       MeOverride, TeOverride, true)) of
 	{ok, Data2} ->
 	    ?vdebug("started",[]),
-	    snmpa_mib_data:sync(Data2),
+	    MibDataMod:sync(Data2),
 	    ?vdebug("mib data synced",[]),
 	    {ok, #state{data          = Data2, 
 			teo           = TeOverride, 
@@ -304,7 +311,8 @@ do_init(Prio, Mibs, Opts) ->
 			cache_tmr     = CacheGcTimer, 
 			cache_autogc  = CacheAutoGC,
 			cache_gclimit = CacheGcLimit,
-			cache_age     = CacheAge}};
+			cache_age     = CacheAge, 
+			data_mod      = MibDataMod}};
 	{'aborted at', Mib, _NewData, Reason} ->
 	    ?vinfo("failed loading mib ~p: ~p",[Mib,Reason]),
 	    {error, {Mib, Reason}}
@@ -315,32 +323,34 @@ do_init(Prio, Mibs, Opts) ->
 %% Returns: {ok, NewMibData} | {'aborted at', Mib, NewData, Reason}
 %% Args: Operation is load_mib | unload_mib.
 %%----------------------------------------------------------------------
-mib_operations(Operation, Mibs, Data, MeOverride, TeOverride) ->
-    mib_operations(Operation, Mibs, Data, MeOverride, TeOverride, false). 
+mib_operations(Mod, Operation, Mibs, Data, MeOverride, TeOverride) ->
+    mib_operations(Mod, Operation, Mibs, Data, MeOverride, TeOverride, false). 
 
 
-mib_operations(_Operation, [], Data, _MeOverride, _TeOverride, _Force) ->
+mib_operations(_Mod, _Operation, [], Data, _MeOverride, _TeOverride, _Force) ->
     {ok, Data};
-mib_operations(Operation, [Mib|Mibs], Data0, MeOverride, TeOverride, Force) ->
+mib_operations(Mod, Operation, [Mib|Mibs], Data0, MeOverride, TeOverride, Force) ->
     ?vtrace("mib operations ~p on"
-	"~n   Mibs: ~p"
-	"~n   with "
-	"~n   MeOverride: ~p"
-	"~n   TeOverride: ~p"
-	"~n   Force:      ~p", [Operation,Mibs,MeOverride,TeOverride,Force]),
-    Data = mib_operation(Operation, Mib, Data0, MeOverride, TeOverride, Force),
-    mib_operations(Operation, Mibs, Data, MeOverride, TeOverride, Force).
-
-mib_operation(Operation, Mib, Data0, MeOverride, TeOverride, Force) 
+	    "~n   Mibs: ~p"
+	    "~n   with "
+	    "~n   MeOverride: ~p"
+	    "~n   TeOverride: ~p"
+	    "~n   Force:      ~p", 
+	    [Operation, Mibs, MeOverride, TeOverride, Force]),
+    Data = mib_operation(Mod, 
+			 Operation, Mib, Data0, MeOverride, TeOverride, Force),
+    mib_operations(Mod, Operation, Mibs, Data, MeOverride, TeOverride, Force).
+
+mib_operation(Mod, Operation, Mib, Data0, MeOverride, TeOverride, Force) 
   when is_list(Mib) ->
     ?vtrace("mib operation on mib ~p", [Mib]),
-    case apply(snmpa_mib_data, Operation, [Data0,Mib,MeOverride,TeOverride]) of
-	{error, 'already loaded'} when (Operation =:= load_mib) andalso  
+    case apply(Mod, Operation, [Data0, Mib, MeOverride, TeOverride]) of
+	{error, already_loaded} when (Operation =:= load_mib) andalso  
 				       (Force =:= true) ->
 	    ?vlog("ignore mib ~p -> already loaded", [Mib]),
 	    Data0;
-	{error, 'not loaded'} when (Operation =:= unload_mib) andalso 
-				   (Force =:= true) ->
+	{error, not_loaded} when (Operation =:= unload_mib) andalso 
+				 (Force =:= true) ->
 	    ?vlog("ignore mib ~p -> not loaded", [Mib]),
 	    Data0;
 	{error, Reason} ->
@@ -350,7 +360,7 @@ mib_operation(Operation, Mib, Data0, MeOverride, TeOverride, Force)
 	{ok, Data} ->
 	    Data
     end;
-mib_operation(_Op, Mib, Data, _MeOverride, _TeOverride, _Force) ->
+mib_operation(_Mod, _Op, Mib, Data, _MeOverride, _TeOverride, _Force) ->
     throw({'aborted at', Mib, Data, bad_mibname}).
 
 
@@ -395,15 +405,15 @@ handle_call({update_cache_opts, Key, Value}, _From, State) ->
     {reply, Result, NewState};
 
 handle_call({lookup, Oid}, _From, 
-	    #state{data = Data, cache = Cache} = State) ->
+	    #state{data = Data, cache = Cache, data_mod = Mod} = State) ->
     ?vlog("lookup ~p", [Oid]), 
     Key = {lookup, Oid}, 
     {Reply, NewState} = 
 	case maybe_cache_lookup(Cache, Key) of
 	    ?NO_CACHE ->
-		{snmpa_mib_data:lookup(Data, Oid), State};
+		{Mod:lookup(Data, Oid), State};
 	    [] ->
-		Rep = snmpa_mib_data:lookup(Data, Oid),
+		Rep = Mod:lookup(Data, Oid),
 		ets:insert(Cache, {Key, Rep, timestamp()}),
 		{Rep, maybe_start_cache_gc_timer(State)};
 	    [{Key, Rep, _}] ->
@@ -414,22 +424,23 @@ handle_call({lookup, Oid}, _From,
     ?vdebug("lookup -> Reply: ~p", [Reply]), 
     {reply, Reply, NewState};
 
-handle_call({which_mib, Oid}, _From, #state{data = Data} = State) ->
+handle_call({which_mib, Oid}, _From, 
+	    #state{data = Data, data_mod = Mod} = State) ->
     ?vlog("which_mib ~p",[Oid]),    
-    Reply = snmpa_mib_data:which_mib(Data, Oid),
+    Reply = Mod:which_mib(Data, Oid),
     ?vdebug("which_mib: ~p",[Reply]),    
     {reply, Reply, State};
 
 handle_call({next, Oid, MibView}, _From, 
-	    #state{data = Data, cache = Cache} = State) ->
+	    #state{data = Data, cache = Cache, data_mod = Mod} = State) ->
     ?vlog("next ~p [~p]", [Oid, MibView]), 
     Key = {next, Oid, MibView},
     {Reply, NewState} = 
 	case maybe_cache_lookup(Cache, Key) of
 	    ?NO_CACHE ->
-		{snmpa_mib_data:next(Data, Oid, MibView), State};
+		{Mod:next(Data, Oid, MibView), State};
 	    [] ->    
-		Rep = snmpa_mib_data:next(Data, Oid, MibView),
+		Rep = Mod:next(Data, Oid, MibView),
 		ets:insert(Cache, {Key, Rep, timestamp()}),
 		{Rep, maybe_start_cache_gc_timer(State)};
 	    [{Key, Rep, _}] ->
@@ -441,89 +452,99 @@ handle_call({next, Oid, MibView}, _From,
     {reply, Reply, NewState};
 
 handle_call({load_mibs, Mibs}, _From, 
-	    #state{data  = Data, 
-		   teo   = TeOverride, 
-		   meo   = MeOverride,
-		   cache = Cache} = State) ->
+	    #state{data         = Data, 
+		   teo          = TeOverride, 
+		   meo          = MeOverride,
+		   cache        = Cache, 
+		   data_mod     = Mod} = State) ->
     ?vlog("load mibs ~p",[Mibs]),    
     %% Invalidate cache
     NewCache = maybe_invalidate_cache(Cache),
-    {NData,Reply} = 
-	case (catch mib_operations(load_mib, Mibs, Data,
+    {NData, Reply} = 
+	case (catch mib_operations(Mod, load_mib, Mibs, Data,
 				   MeOverride, TeOverride)) of
 	    {'aborted at', Mib, NewData, Reason} ->
 		?vlog("aborted at ~p for reason ~p",[Mib,Reason]),    
-		{NewData,{error, {'load aborted at', Mib, Reason}}};
+		{NewData, {error, {'load aborted at', Mib, Reason}}};
 	    {ok, NewData} ->
-		{NewData,ok}
+		{NewData, ok}
 	end,
-    snmpa_mib_data:sync(NData),
+    Mod:sync(NData),
     {reply, Reply, State#state{data = NData, cache = NewCache}};
 
 handle_call({unload_mibs, Mibs}, _From, 
-	    #state{data  = Data, 
-		   teo   = TeOverride, 
-		   meo   = MeOverride,
-		   cache = Cache} = State) ->
+	    #state{data         = Data, 
+		   teo          = TeOverride, 
+		   meo          = MeOverride,
+		   cache        = Cache, 
+		   data_mod     = Mod} = State) ->
     ?vlog("unload mibs ~p",[Mibs]),    
     %% Invalidate cache
     NewCache = maybe_invalidate_cache(Cache),
     %% Unload mib(s)
-    {NData,Reply} = 
-	case (catch mib_operations(unload_mib, Mibs, Data,
+    {NData, Reply} = 
+	case (catch mib_operations(Mod, unload_mib, Mibs, Data,
 				   MeOverride, TeOverride)) of
 	    {'aborted at', Mib, NewData, Reason} ->
-		?vlog("aborted at ~p for reason ~p",[Mib,Reason]),    
+		?vlog("aborted at ~p for reason ~p", [Mib,Reason]),    
 		{NewData, {error, {'unload aborted at', Mib, Reason}}};
 	    {ok, NewData} ->
-		{NewData,ok}
+		{NewData, ok}
 	end,
-    snmpa_mib_data:sync(NData),
+    Mod:sync(NData),
     {reply, Reply, State#state{data = NData, cache = NewCache}};
 
-handle_call(which_mibs, _From, #state{data = Data} = State) ->
+handle_call(which_mibs, _From, #state{data = Data, data_mod = Mod} = State) ->
     ?vlog("which mibs",[]),    
-    Reply = snmpa_mib_data:which_mibs(Data),
+    Reply = Mod:which_mibs(Data),
     {reply, Reply, State};
 
-handle_call({whereis_mib, Mib}, _From, #state{data = Data} = State) ->
+handle_call({whereis_mib, Mib}, _From, 
+	    #state{data         = Data, 
+		   data_mod     = Mod} = State) ->
     ?vlog("whereis mib: ~p",[Mib]),    
-    Reply = snmpa_mib_data:whereis_mib(Data, Mib),
+    Reply = Mod:whereis_mib(Data, Mib),
     {reply, Reply, State};
 
 handle_call({register_subagent, Oid, Pid}, _From, 
-	    #state{data = Data, cache = Cache} = State) ->
+	    #state{data      = Data, 
+		   cache     = Cache, 
+		   data_mod  = Mod} = State) ->
     ?vlog("register subagent ~p, ~p",[Oid,Pid]),
     %% Invalidate cache
     NewCache = maybe_invalidate_cache(Cache),
-    case snmpa_mib_data:register_subagent(Data, Oid, Pid) of
+    case Mod:register_subagent(Data, Oid, Pid) of
 	{error, Reason} ->
 	    ?vlog("registration failed: ~p",[Reason]),    
 	    {reply, {error, Reason}, State#state{cache = NewCache}};
-	NewData ->
+	{ok, NewData} ->
 	    {reply, ok, State#state{data = NewData, cache = NewCache}}
     end;
 
 handle_call({unregister_subagent, OidOrPid}, _From, 
-	    #state{data = Data, cache = Cache} = State) ->
+	    #state{data     = Data, 
+		   cache    = Cache, 
+		   data_mod = Mod} = State) ->
     ?vlog("unregister subagent ~p",[OidOrPid]),    
     %% Invalidate cache
     NewCache = maybe_invalidate_cache(Cache),
-    case snmpa_mib_data:unregister_subagent(Data, OidOrPid) of
+    case Mod:unregister_subagent(Data, OidOrPid) of
+	{ok, NewData} ->
+	    {reply, ok, State#state{data = NewData, cache = NewCache}};
 	{ok, NewData, DeletedSubagentPid} ->
 	    {reply, {ok, DeletedSubagentPid}, State#state{data  = NewData, 
 							  cache = NewCache}};
 	{error, Reason} ->
 	    ?vlog("unregistration failed: ~p",[Reason]),    
-	    {reply, {error, Reason}, State#state{cache = NewCache}};
-	NewData ->
-	    {reply, ok, State#state{data = NewData, cache = NewCache}}
+	    {reply, {error, Reason}, State#state{cache = NewCache}}
     end;
 
-handle_call(info, _From, #state{data = Data, cache = Cache} = State) ->
+handle_call(info, _From, #state{data     = Data, 
+				cache    = Cache, 
+				data_mod = Mod} = State) ->
     ?vlog("info",[]),    
     Reply = 
-	case (catch snmpa_mib_data:info(Data)) of
+	case (catch Mod:info(Data)) of
 	    Info when is_list(Info) ->
 		[{cache, size_cache(Cache)} | Info];
 	    E ->
@@ -531,10 +552,12 @@ handle_call(info, _From, #state{data = Data, cache = Cache} = State) ->
 	    end,
     {reply, Reply, State};
 
-handle_call({info, Type}, _From, #state{data = Data} = State) ->
+handle_call({info, Type}, _From, 
+	    #state{data     = Data, 
+		   data_mod = Mod} = State) ->
     ?vlog("info ~p",[Type]),    
     Reply = 
-	case (catch snmpa_mib_data:info(Data, Type)) of
+	case (catch Mod:info(Data, Type)) of
 	    Info when is_list(Info) ->
 		Info;
 	    E ->
@@ -542,21 +565,19 @@ handle_call({info, Type}, _From, #state{data = Data} = State) ->
 	end,
     {reply, Reply, State};
 
-handle_call(dump, _From, State) ->
-    ?vlog("dump",[]),    
-    Reply = snmpa_mib_data:dump(State#state.data),
-    {reply, Reply, State};
-    
-handle_call({dump, File}, _From, #state{data = Data} = State) ->
+handle_call({dump, File}, _From, 
+	    #state{data = Data, data_mod = Mod} = State) ->
     ?vlog("dump on ~s",[File]),    
-    Reply = snmpa_mib_data:dump(Data, File),
+    Reply = Mod:dump(Data, File),
     {reply, Reply, State};
     
 %% This check (that there is no backup already in progress) is also 
 %% done in the master agent process, but just in case a user issues 
 %% a backup call to this process directly, we add a similar check here. 
 handle_call({backup, BackupDir}, From, 
-	    #state{backup = undefined, data = Data} = State) ->
+	    #state{backup   = undefined, 
+		   data     = Data, 
+		   data_mod = Mod} = State) ->
     ?vlog("backup to ~s", [BackupDir]),
     Pid = self(),
     V   = get(verbosity),
@@ -568,7 +589,7 @@ handle_call({backup, BackupDir}, From,
 			  put(sname, ambs),
 			  put(verbosity, V),
 			  Dir   = filename:join([BackupDir]),
-			  Reply = snmpa_mib_data:backup(Data, Dir),
+			  Reply = Mod:backup(Data, Dir),
 			  Pid ! {backup_done, Reply},
 			  unlink(Pid)
 		  end),	
@@ -637,8 +658,8 @@ handle_info(Info, State) ->
     warning_msg("received unknown info: ~n~p", [Info]),
     {noreply, State}.
 
-terminate(_Reason, #state{data = Data}) ->
-    catch snmpa_mib_data:close(Data),
+terminate(_Reason, #state{data = Data, data_mod = Mod}) ->
+    catch Mod:close(Data),
     ok.
 
 
@@ -655,6 +676,11 @@ terminate(_Reason, #state{data = Data}) ->
 %%     S2 = {state, Data, MEO, TEO, B},
 %%     {ok, S2};
 
+code_change({down, Vsn}, #state{data = Data0, data_mod = Mod} = State, Extra) ->
+    Data = Mod:code_change(down, Vsn, Extra, Data0), 
+    {ok, State#state{data = Data}};
+
+
 %% %% upgrade
 %% %% 
 %% code_change(_Vsn, S1, upgrade_from_pre_4_12) ->
@@ -663,8 +689,9 @@ terminate(_Reason, #state{data = Data}) ->
 %%     S2 = #state{data = Data, meo = MEO, teo = TEO, backup = B, cache = Cache},
 %%     {ok, S2};
 
-code_change(_Vsn, State, _Extra) ->
-    {ok, State}.
+code_change(Vsn, #state{data = Data0, data_mod = Mod} = State, Extra) ->
+    Data = Mod:code_change(up, Vsn, Extra, Data0), 
+    {ok, State#state{data = Data}}.
 
 
 %%-----------------------------------------------------------------
@@ -681,7 +708,10 @@ get_te_override(Options) ->
     get_opt(trapentry_override, Options, false).
 
 get_mib_storage(Options) ->
-    get_opt(mib_storage, Options, ets).
+    get_opt(mib_storage, Options).
+
+get_data_mod(Options) ->
+    get_opt(data_module, Options, snmpa_mib_data_tttn).
 
 get_cacheopt_autogc(Cache, CacheOpts) ->
     IsValid = fun(AutoGC) when ((AutoGC =:= true) orelse 
@@ -868,6 +898,9 @@ timestamp() ->
 
 %% ----------------------------------------------------------------
 
+get_opt(Key, Options) ->
+    snmp_misc:get_option(Key, Options).
+
 get_opt(Key, Options, Default) ->
     snmp_misc:get_option(Key, Options, Default).