router/mib/mibaccess.cpp

/*
 *  MibAccess.cpp
 *
 *      The original code by Stas Khirman modified by Lauri Pesonen, December, 2000:
 *      
 *      SnmpUtilVarBindFree(), SnmpUtilOidNCmp() and SnmpUtilOidCpy() now loaded from
 *      "snmpapi.dll" dynamically instead of linking statically.
 *      
 *      MibII ctor now takes a parameter whether to load Winsock or not.
 *      WSAStartup maintains an internal reference counter so it would have been ok
 *      to let it load always.
 *      
 *      Fixed a bug where the return value of LoadLibrary() was compared against
 *      HINSTANCE_ERROR instead of NULL.
 *      
 *      Removed some type conversion warnings by casting.
 *      
 *      Added a check in MibExtLoad ctor that the function entry points were found.
 *      
 *      Added a check in GetIPMask() and GetIPAddress() that the library was loaded
 *      before accessing the functions.
 *      
 *      Changed the return type of GetIPAddress() and GetIPMask() from BOOL  to void
 *      as they always returned TRUE.
 *      
 */

/************************************************************************/
/*      Copyright (C) Stas Khirman 1998.  All rights reserved.          */
/*      Written by Stas Khirman (staskh@rocketmail.com).                */
/*                                        and                                                                                       */
/*                 Raz Galili (razgalili@hotmail.com)                               */
/*                                                                      */
/*      Free software: no warranty; use anywhere is ok; spread the      */
/*      sources; note any modifications; share variations and           */
/*      derivatives (including sending to staskh@rocketmail.com).       */
/*                                                                      */
/************************************************************************/

/*
 *  MibAccess.cpp - ip router
 *
 *  Basilisk II (C) 1997-2001 Christian Bauer
 *
 *  Windows platform specific code copyright (C) Lauri Pesonen
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 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 General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */


#include "sysdeps.h"
#include "mibaccess.h"
#include "../dynsockets.h"
#include "../dump.h"

#if DEBUG
#pragma optimize("",off)
#endif

#include "debug.h"

MibExtLoad::MibExtLoad( LPSTR MibDllName, LPSTR SnmpDllName )
{
        
        m_Init = NULL;

        m_InitEx = NULL;
        m_Query = NULL; 
        m_Trap = NULL;

        m_hInst_snmputil = NULL;

        m_SnmpUtilVarBindFree = NULL;
        m_SnmpUtilOidNCmp = NULL;
        m_SnmpUtilOidCpy = NULL;
        
        m_hInst = LoadLibrary( MibDllName );
        if(!m_hInst) {
                D(bug("MIB: library %s could not be loaded.\r\n", MibDllName));
                return;
        }
        D(bug("MIB: library %s loaded ok.\r\n", MibDllName));

        m_Init  =       (pSnmpExtensionInit)GetProcAddress(m_hInst ,"SnmpExtensionInit");
        m_InitEx=       (pSnmpExtensionInitEx)GetProcAddress(m_hInst ,"SnmpExtensionInitEx");
        m_Query =       (pSnmpExtensionQuery)GetProcAddress(m_hInst ,"SnmpExtensionQuery");
        m_Trap  =       (pSnmpExtensionTrap)GetProcAddress(m_hInst ,"SnmpExtensionTrap");

        if( !m_Init || !m_InitEx || !m_Query || !m_Trap )
        {
                D(bug("MIB: required entry points not found in library %s.\r\n", MibDllName));
                FreeLibrary( m_hInst );
                m_hInst = NULL;
        }

        m_hInst_snmputil = LoadLibrary( SnmpDllName );
        if(!m_hInst_snmputil){
                D(bug("MIB: library %s could not be loaded.\r\n", SnmpDllName));
                FreeLibrary( m_hInst );
                m_hInst = NULL;
                return;
        }
        D(bug("MIB: library %s loaded ok.\r\n", SnmpDllName));

        m_SnmpUtilVarBindFree = (VOID (SNMP_FUNC_TYPE *)(SnmpVarBind *))GetProcAddress( m_hInst_snmputil, "SnmpUtilVarBindFree" );
        m_SnmpUtilOidNCmp = (SNMPAPI (SNMP_FUNC_TYPE *)(AsnObjectIdentifier *, AsnObjectIdentifier *, UINT))GetProcAddress( m_hInst_snmputil, "SnmpUtilOidNCmp" );
        m_SnmpUtilOidCpy = (SNMPAPI (SNMP_FUNC_TYPE *)(AsnObjectIdentifier *, AsnObjectIdentifier *))GetProcAddress( m_hInst_snmputil, "SnmpUtilOidCpy" );

        if( !m_SnmpUtilVarBindFree || !m_SnmpUtilOidNCmp || !m_SnmpUtilOidCpy )
        {
                D(bug("MIB: required entry points not found in library %s.\r\n", SnmpDllName));
                FreeLibrary( m_hInst );
                FreeLibrary( m_hInst_snmputil );
                m_hInst = NULL;
                m_hInst_snmputil = NULL;
        }

        #undef SNMP_FreeVarBind
        #undef SNMP_oidncmp
        #undef SNMP_oidcpy

        #define SNMP_FreeVarBind   m_SnmpUtilVarBindFree
        #define SNMP_oidncmp       m_SnmpUtilOidNCmp
        #define SNMP_oidcpy        m_SnmpUtilOidCpy
}

MibExtLoad::~MibExtLoad()
{
        if( m_hInst ) {
                FreeLibrary( m_hInst );
                m_hInst = NULL;
        }
        if( m_hInst_snmputil ) {
                FreeLibrary( m_hInst_snmputil );
                m_hInst_snmputil = NULL;
        }
}

BOOL MibExtLoad::Init(DWORD dwTimeZeroReference,HANDLE *hPollForTrapEvent,AsnObjectIdentifier *supportedView)
{
        if(m_hInst && m_Init)
                return m_Init(dwTimeZeroReference,hPollForTrapEvent,supportedView);
        return FALSE;
}
BOOL MibExtLoad::InitEx(AsnObjectIdentifier *supportedView)
{
        if(m_hInst && m_InitEx)
                return m_InitEx(supportedView);
        
        return FALSE;
}

BOOL MibExtLoad::Query(BYTE requestType,OUT RFC1157VarBindList *variableBindings,
                                           AsnInteger *errorStatus,AsnInteger *errorIndex)
{
        if(m_hInst && m_Query)
                return m_Query(requestType,variableBindings,errorStatus,errorIndex);
        
        return FALSE;
}

BOOL MibExtLoad::Trap(AsnObjectIdentifier *enterprise, AsnInteger *genericTrap,
                                          AsnInteger *specificTrap, AsnTimeticks *timeStamp, 
                                          RFC1157VarBindList  *variableBindings)
{
        if(m_hInst && m_Trap)
                return m_Trap(enterprise, genericTrap,specificTrap, timeStamp, variableBindings);
        
        return FALSE;
}

MibII::MibII( bool load_winsock ):MibExtLoad("inetmib1.dll","snmpapi.dll")
{
  WSADATA wsa;
        m_load_winsock = load_winsock;
        if(load_winsock) {
                int err = _WSAStartup( 0x0101, &wsa );  
        }
}

MibII::~MibII()
{
  if(m_load_winsock) _WSACleanup();
}

BOOL MibII::Init()
{
        HANDLE PollForTrapEvent;
        AsnObjectIdentifier SupportedView;

        return MibExtLoad::Init(GetTickCount(),&PollForTrapEvent,&SupportedView);

}


void MibII::GetIPAddress( UINT IpArray[], UINT &IpArraySize )
{
        if(!m_hInst) {
                IpArraySize = 0;
                return;
        }
        
        UINT OID_ipAdEntAddr[] = { 1, 3, 6, 1, 2, 1, 4 , 20, 1 ,1 };
        AsnObjectIdentifier MIB_ipAdEntAddr = { sizeof(OID_ipAdEntAddr)/sizeof(UINT), OID_ipAdEntAddr };
        RFC1157VarBindList  varBindList;
        RFC1157VarBind      varBind[1];
        AsnInteger          errorStatus;
        AsnInteger          errorIndex;
        AsnObjectIdentifier MIB_NULL = {0,0};
        BOOL                Exit;
        int                 ret;
        int                 IpCount=0;
        DWORD               dtmp;
        
        varBindList.list = varBind;
        varBindList.len  = 1;
        varBind[0].name  = MIB_NULL;
        SNMP_oidcpy(&varBind[0].name,&MIB_ipAdEntAddr);
        Exit = FALSE;
        
        IpCount=0;
        while(!Exit){
                ret = Query(ASN_RFC1157_GETNEXTREQUEST,&varBindList,&errorStatus,&errorIndex);
                
                if(!ret)
                        Exit=TRUE;
                else{
                        ret = SNMP_oidncmp(&varBind[0].name,&MIB_ipAdEntAddr,MIB_ipAdEntAddr.idLength);
                        if(ret!=0){
                                Exit=TRUE;
                        }
                        else{
                                dtmp = *((DWORD *)varBind[0].value.asnValue.address.stream);
                                IpArray[IpCount] = dtmp;
                                IpCount++;
                                if(IpCount>=(int)IpArraySize)
                                        Exit = TRUE;
                        }
                }
        }
        
        IpArraySize = IpCount;
        
        SNMP_FreeVarBind(&varBind[0]);
}

void MibII::GetIPMask( UINT IpArray[], UINT &IpArraySize )
{
        if(!m_hInst) {
                IpArraySize = 0;
                return;
        }
        
        UINT OID_ipAdEntMask[] = { 1, 3, 6, 1, 2, 1, 4 , 20, 1 ,3 };
        AsnObjectIdentifier MIB_ipAdEntMask = { sizeof(OID_ipAdEntMask)/sizeof(UINT), OID_ipAdEntMask };
        RFC1157VarBindList  varBindList;
        RFC1157VarBind      varBind[1];
        AsnInteger          errorStatus;
        AsnInteger          errorIndex;
        AsnObjectIdentifier MIB_NULL = {0,0};
        BOOL                Exit;
        int                 ret;
        int                 IpCount=0;
        DWORD               dtmp;
        
        varBindList.list = varBind;
        varBindList.len  = 1;
        varBind[0].name  = MIB_NULL;
        SNMP_oidcpy(&varBind[0].name,&MIB_ipAdEntMask);
        Exit = FALSE;
        
        IpCount=0;
        while(!Exit){
                ret = Query(ASN_RFC1157_GETNEXTREQUEST,&varBindList,&errorStatus,&errorIndex);
                
                if(!ret)
                        Exit=TRUE;
                else{
                        ret = SNMP_oidncmp(&varBind[0].name,&MIB_ipAdEntMask,MIB_ipAdEntMask.idLength);
                        if(ret!=0){
                                Exit=TRUE;
                        }
                        else{
                                dtmp = *((DWORD *)varBind[0].value.asnValue.address.stream);
                                IpArray[IpCount] = dtmp;
                                IpCount++;
                                if(IpCount>=(int)IpArraySize)
                                        Exit = TRUE;
                        }
                }
        }
        
        IpArraySize = IpCount;
        
        SNMP_FreeVarBind(&varBind[0]);
}
Revision:	1.2
Committed:	2004-12-26T23:24:33Z (19 years, 11 months ago) by gbeauche
Branch:	MAIN
CVS Tags:	nigel-build-19, nigel-build-17
Changes since 1.1:	+2 -2 lines
Log Message:	cross compile fixes
#	User	Rev	Content
1	gbeauche	1.1	/*
2			* MibAccess.cpp
3			*
4			* The original code by Stas Khirman modified by Lauri Pesonen, December, 2000:
5			*
6			* SnmpUtilVarBindFree(), SnmpUtilOidNCmp() and SnmpUtilOidCpy() now loaded from
7			* "snmpapi.dll" dynamically instead of linking statically.
8			*
9			* MibII ctor now takes a parameter whether to load Winsock or not.
10			* WSAStartup maintains an internal reference counter so it would have been ok
11			* to let it load always.
12			*
13			* Fixed a bug where the return value of LoadLibrary() was compared against
14			* HINSTANCE_ERROR instead of NULL.
15			*
16			* Removed some type conversion warnings by casting.
17			*
18			* Added a check in MibExtLoad ctor that the function entry points were found.
19			*
20			* Added a check in GetIPMask() and GetIPAddress() that the library was loaded
21			* before accessing the functions.
22			*
23			* Changed the return type of GetIPAddress() and GetIPMask() from BOOL to void
24			* as they always returned TRUE.
25			*
26			*/
27
28			/************************************************************************/
29			/* Copyright (C) Stas Khirman 1998. All rights reserved. */
30			/* Written by Stas Khirman (staskh@rocketmail.com). */
31			/* and */
32			/* Raz Galili (razgalili@hotmail.com) */
33			/* */
34			/* Free software: no warranty; use anywhere is ok; spread the */
35			/* sources; note any modifications; share variations and */
36			/* derivatives (including sending to staskh@rocketmail.com). */
37			/* */
38			/************************************************************************/
39
40			/*
41			* MibAccess.cpp - ip router
42			*
43			* Basilisk II (C) 1997-2001 Christian Bauer
44			*
45			* Windows platform specific code copyright (C) Lauri Pesonen
46			*
47			* This program is free software; you can redistribute it and/or modify
48			* it under the terms of the GNU General Public License as published by
49			* the Free Software Foundation; either version 2 of the License, or
50			* (at your option) any later version.
51			*
52			* This program is distributed in the hope that it will be useful,
53			* but WITHOUT ANY WARRANTY; without even the implied warranty of
54			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
55			* GNU General Public License for more details.
56			*
57			* You should have received a copy of the GNU General Public License
58			* along with this program; if not, write to the Free Software
59			* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
60			*/
61
62
63			#include "sysdeps.h"
64			#include "mibaccess.h"
65	gbeauche	1.2	#include "../dynsockets.h"
66			#include "../dump.h"
67	gbeauche	1.1
68			#if DEBUG
69			#pragma optimize("",off)
70			#endif
71
72			#include "debug.h"
73
74			MibExtLoad::MibExtLoad( LPSTR MibDllName, LPSTR SnmpDllName )
75			{
76
77			m_Init = NULL;
78
79			m_InitEx = NULL;
80			m_Query = NULL;
81			m_Trap = NULL;
82
83			m_hInst_snmputil = NULL;
84
85			m_SnmpUtilVarBindFree = NULL;
86			m_SnmpUtilOidNCmp = NULL;
87			m_SnmpUtilOidCpy = NULL;
88
89			m_hInst = LoadLibrary( MibDllName );
90			if(!m_hInst) {
91			D(bug("MIB: library %s could not be loaded.\r\n", MibDllName));
92			return;
93			}
94			D(bug("MIB: library %s loaded ok.\r\n", MibDllName));
95
96			m_Init = (pSnmpExtensionInit)GetProcAddress(m_hInst ,"SnmpExtensionInit");
97			m_InitEx= (pSnmpExtensionInitEx)GetProcAddress(m_hInst ,"SnmpExtensionInitEx");
98			m_Query = (pSnmpExtensionQuery)GetProcAddress(m_hInst ,"SnmpExtensionQuery");
99			m_Trap = (pSnmpExtensionTrap)GetProcAddress(m_hInst ,"SnmpExtensionTrap");
100
101			if( !m_Init \|\| !m_InitEx \|\| !m_Query \|\| !m_Trap )
102			{
103			D(bug("MIB: required entry points not found in library %s.\r\n", MibDllName));
104			FreeLibrary( m_hInst );
105			m_hInst = NULL;
106			}
107
108			m_hInst_snmputil = LoadLibrary( SnmpDllName );
109			if(!m_hInst_snmputil){
110			D(bug("MIB: library %s could not be loaded.\r\n", SnmpDllName));
111			FreeLibrary( m_hInst );
112			m_hInst = NULL;
113			return;
114			}
115			D(bug("MIB: library %s loaded ok.\r\n", SnmpDllName));
116
117			m_SnmpUtilVarBindFree = (VOID (SNMP_FUNC_TYPE )(SnmpVarBind ))GetProcAddress( m_hInst_snmputil, "SnmpUtilVarBindFree" );
118			m_SnmpUtilOidNCmp = (SNMPAPI (SNMP_FUNC_TYPE )(AsnObjectIdentifier , AsnObjectIdentifier *, UINT))GetProcAddress( m_hInst_snmputil, "SnmpUtilOidNCmp" );
119			m_SnmpUtilOidCpy = (SNMPAPI (SNMP_FUNC_TYPE )(AsnObjectIdentifier , AsnObjectIdentifier *))GetProcAddress( m_hInst_snmputil, "SnmpUtilOidCpy" );
120
121			if( !m_SnmpUtilVarBindFree \|\| !m_SnmpUtilOidNCmp \|\| !m_SnmpUtilOidCpy )
122			{
123			D(bug("MIB: required entry points not found in library %s.\r\n", SnmpDllName));
124			FreeLibrary( m_hInst );
125			FreeLibrary( m_hInst_snmputil );
126			m_hInst = NULL;
127			m_hInst_snmputil = NULL;
128			}
129
130			#undef SNMP_FreeVarBind
131			#undef SNMP_oidncmp
132			#undef SNMP_oidcpy
133
134			#define SNMP_FreeVarBind m_SnmpUtilVarBindFree
135			#define SNMP_oidncmp m_SnmpUtilOidNCmp
136			#define SNMP_oidcpy m_SnmpUtilOidCpy
137			}
138
139			MibExtLoad::~MibExtLoad()
140			{
141			if( m_hInst ) {
142			FreeLibrary( m_hInst );
143			m_hInst = NULL;
144			}
145			if( m_hInst_snmputil ) {
146			FreeLibrary( m_hInst_snmputil );
147			m_hInst_snmputil = NULL;
148			}
149			}
150
151			BOOL MibExtLoad::Init(DWORD dwTimeZeroReference,HANDLE hPollForTrapEvent,AsnObjectIdentifier supportedView)
152			{
153			if(m_hInst && m_Init)
154			return m_Init(dwTimeZeroReference,hPollForTrapEvent,supportedView);
155			return FALSE;
156			}
157			BOOL MibExtLoad::InitEx(AsnObjectIdentifier *supportedView)
158			{
159			if(m_hInst && m_InitEx)
160			return m_InitEx(supportedView);
161
162			return FALSE;
163			}
164
165			BOOL MibExtLoad::Query(BYTE requestType,OUT RFC1157VarBindList *variableBindings,
166			AsnInteger errorStatus,AsnInteger errorIndex)
167			{
168			if(m_hInst && m_Query)
169			return m_Query(requestType,variableBindings,errorStatus,errorIndex);
170
171			return FALSE;
172			}
173
174			BOOL MibExtLoad::Trap(AsnObjectIdentifier enterprise, AsnInteger genericTrap,
175			AsnInteger specificTrap, AsnTimeticks timeStamp,
176			RFC1157VarBindList *variableBindings)
177			{
178			if(m_hInst && m_Trap)
179			return m_Trap(enterprise, genericTrap,specificTrap, timeStamp, variableBindings);
180
181			return FALSE;
182			}
183
184			MibII::MibII( bool load_winsock ):MibExtLoad("inetmib1.dll","snmpapi.dll")
185			{
186			WSADATA wsa;
187			m_load_winsock = load_winsock;
188			if(load_winsock) {
189			int err = _WSAStartup( 0x0101, &wsa );
190			}
191			}
192
193			MibII::~MibII()
194			{
195			if(m_load_winsock) _WSACleanup();
196			}
197
198			BOOL MibII::Init()
199			{
200			HANDLE PollForTrapEvent;
201			AsnObjectIdentifier SupportedView;
202
203			return MibExtLoad::Init(GetTickCount(),&PollForTrapEvent,&SupportedView);
204
205			}
206
207
208			void MibII::GetIPAddress( UINT IpArray[], UINT &IpArraySize )
209			{
210			if(!m_hInst) {
211			IpArraySize = 0;
212			return;
213			}
214
215			UINT OID_ipAdEntAddr[] = { 1, 3, 6, 1, 2, 1, 4 , 20, 1 ,1 };
216			AsnObjectIdentifier MIB_ipAdEntAddr = { sizeof(OID_ipAdEntAddr)/sizeof(UINT), OID_ipAdEntAddr };
217			RFC1157VarBindList varBindList;
218			RFC1157VarBind varBind[1];
219			AsnInteger errorStatus;
220			AsnInteger errorIndex;
221			AsnObjectIdentifier MIB_NULL = {0,0};
222			BOOL Exit;
223			int ret;
224			int IpCount=0;
225			DWORD dtmp;
226
227			varBindList.list = varBind;
228			varBindList.len = 1;
229			varBind[0].name = MIB_NULL;
230			SNMP_oidcpy(&varBind[0].name,&MIB_ipAdEntAddr);
231			Exit = FALSE;
232
233			IpCount=0;
234			while(!Exit){
235			ret = Query(ASN_RFC1157_GETNEXTREQUEST,&varBindList,&errorStatus,&errorIndex);
236
237			if(!ret)
238			Exit=TRUE;
239			else{
240			ret = SNMP_oidncmp(&varBind[0].name,&MIB_ipAdEntAddr,MIB_ipAdEntAddr.idLength);
241			if(ret!=0){
242			Exit=TRUE;
243			}
244			else{
245			dtmp = ((DWORD )varBind[0].value.asnValue.address.stream);
246			IpArray[IpCount] = dtmp;
247			IpCount++;
248			if(IpCount>=(int)IpArraySize)
249			Exit = TRUE;
250			}
251			}
252			}
253
254			IpArraySize = IpCount;
255
256			SNMP_FreeVarBind(&varBind[0]);
257			}
258
259			void MibII::GetIPMask( UINT IpArray[], UINT &IpArraySize )
260			{
261			if(!m_hInst) {
262			IpArraySize = 0;
263			return;
264			}
265
266			UINT OID_ipAdEntMask[] = { 1, 3, 6, 1, 2, 1, 4 , 20, 1 ,3 };
267			AsnObjectIdentifier MIB_ipAdEntMask = { sizeof(OID_ipAdEntMask)/sizeof(UINT), OID_ipAdEntMask };
268			RFC1157VarBindList varBindList;
269			RFC1157VarBind varBind[1];
270			AsnInteger errorStatus;
271			AsnInteger errorIndex;
272			AsnObjectIdentifier MIB_NULL = {0,0};
273			BOOL Exit;
274			int ret;
275			int IpCount=0;
276			DWORD dtmp;
277
278			varBindList.list = varBind;
279			varBindList.len = 1;
280			varBind[0].name = MIB_NULL;
281			SNMP_oidcpy(&varBind[0].name,&MIB_ipAdEntMask);
282			Exit = FALSE;
283
284			IpCount=0;
285			while(!Exit){
286			ret = Query(ASN_RFC1157_GETNEXTREQUEST,&varBindList,&errorStatus,&errorIndex);
287
288			if(!ret)
289			Exit=TRUE;
290			else{
291			ret = SNMP_oidncmp(&varBind[0].name,&MIB_ipAdEntMask,MIB_ipAdEntMask.idLength);
292			if(ret!=0){
293			Exit=TRUE;
294			}
295			else{
296			dtmp = ((DWORD )varBind[0].value.asnValue.address.stream);
297			IpArray[IpCount] = dtmp;
298			IpCount++;
299			if(IpCount>=(int)IpArraySize)
300			Exit = TRUE;
301			}
302			}
303			}
304
305			IpArraySize = IpCount;
306
307			SNMP_FreeVarBind(&varBind[0]);
308			}