Commit | Line | Data |
---|---|---|
5039dede AK |
1 | /* |
2 | ** NetXMS - Network Management System | |
3 | ** Server Library | |
619e5c9b | 4 | ** Copyright (C) 2003-2011 Victor Kirhenshtein |
5039dede AK |
5 | ** |
6 | ** This program is free software; you can redistribute it and/or modify | |
0702ed69 VK |
7 | ** it under the terms of the GNU Lesser General Public License as published by |
8 | ** the Free Software Foundation; either version 3 of the License, or | |
5039dede AK |
9 | ** (at your option) any later version. |
10 | ** | |
11 | ** This program is distributed in the hope that it will be useful, | |
12 | ** but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
14 | ** GNU General Public License for more details. | |
15 | ** | |
0702ed69 | 16 | ** You should have received a copy of the GNU Lesser General Public License |
5039dede AK |
17 | ** along with this program; if not, write to the Free Software |
18 | ** Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. | |
19 | ** | |
20 | ** File: nxsrvapi.h | |
21 | ** | |
22 | **/ | |
23 | ||
24 | #ifndef _nxsrvapi_h_ | |
25 | #define _nxsrvapi_h_ | |
26 | ||
27 | #ifdef _WIN32 | |
28 | #ifdef LIBNXSRV_EXPORTS | |
29 | #define LIBNXSRV_EXPORTABLE __declspec(dllexport) | |
30 | #else | |
31 | #define LIBNXSRV_EXPORTABLE __declspec(dllimport) | |
32 | #endif | |
33 | #else /* _WIN32 */ | |
34 | #define LIBNXSRV_EXPORTABLE | |
35 | #endif | |
36 | ||
37 | #ifndef LIBNXCL_NO_DECLARATIONS | |
38 | #define LIBNXCL_NO_DECLARATIONS 1 | |
39 | #endif | |
40 | #include <nxclapi.h> | |
41 | #include <nxcpapi.h> | |
42 | #include <nms_agent.h> | |
944016d6 | 43 | #include "../libnxsrv/messages.h" |
5039dede AK |
44 | #include <nxsnmp.h> |
45 | #include <netxms_isc.h> | |
46 | ||
47 | ||
48 | // | |
49 | // Default files | |
50 | // | |
51 | ||
52 | #ifdef _WIN32 | |
53 | ||
9796ce45 VK |
54 | #define DEFAULT_CONFIG_FILE _T("C:\\netxmsd.conf") |
55 | ||
56 | #define DEFAULT_SHELL _T("cmd.exe") | |
57 | #define DEFAULT_LOG_FILE _T("C:\\NetXMS.log") | |
58 | #define DEFAULT_DATA_DIR _T("C:\\NetXMS\\var") | |
59 | #define DEFAULT_LIBDIR _T("C:\\NetXMS\\lib") | |
712dd47d | 60 | #define DEFAULT_JAVA_LIBDIR _T("C:\\NetXMS\\lib\\java") |
9796ce45 VK |
61 | #define DEFAULT_DUMP_DIR _T("C:\\") |
62 | ||
63 | #define LDIR_NDD _T("\\ndd") | |
64 | ||
65 | #define DDIR_MIBS _T("\\mibs") | |
66 | #define DDIR_PACKAGES _T("\\packages") | |
67 | #define DDIR_BACKGROUNDS _T("\\backgrounds") | |
68 | #define DDIR_SHARED_FILES _T("\\shared") | |
69 | #define DFILE_KEYS _T("\\server_key") | |
70 | #define DFILE_COMPILED_MIB _T("\\mibs\\netxms.mib") | |
71 | #define DDIR_IMAGES _T("\\images") | |
72 | #define DDIR_FILES _T("\\files") | |
e77e56ca | 73 | #define DDIR_REPORTS _T("\\reports") |
5039dede AK |
74 | |
75 | #else /* _WIN32 */ | |
76 | ||
9796ce45 | 77 | #define DEFAULT_CONFIG_FILE _T("{search}") |
5039dede | 78 | |
9796ce45 | 79 | #define DEFAULT_SHELL _T("/bin/sh") |
5039dede | 80 | |
9796ce45 VK |
81 | #ifndef DATADIR |
82 | #define DATADIR _T("/var/netxms") | |
83 | #endif | |
5039dede | 84 | |
9796ce45 VK |
85 | #ifndef LIBDIR |
86 | #define LIBDIR _T("/usr/lib") | |
87 | #endif | |
5039dede | 88 | |
644f97f5 VK |
89 | #ifndef PKGLIBDIR |
90 | #define PKGLIBDIR _T("/usr/lib/netxms") | |
91 | #endif | |
92 | ||
9796ce45 VK |
93 | #define DEFAULT_LOG_FILE DATADIR _T("/log/netxmsd.log") |
94 | #define DEFAULT_DATA_DIR DATADIR | |
644f97f5 | 95 | #define DEFAULT_LIBDIR PKGLIBDIR |
712dd47d | 96 | #define DEFAULT_JAVA_LIBDIR PKGLIBDIR _T("/java") |
9796ce45 VK |
97 | #define DEFAULT_DUMP_DIR _T("/") |
98 | ||
99 | #define LDIR_NDD _T("/ndd") | |
100 | ||
101 | #define DDIR_MIBS _T("/mibs") | |
102 | #define DDIR_PACKAGES _T("/packages") | |
103 | #define DDIR_BACKGROUNDS _T("/backgrounds") | |
104 | #define DDIR_SHARED_FILES _T("/shared") | |
105 | #define DFILE_KEYS _T("/.server_key") | |
106 | #define DFILE_COMPILED_MIB _T("/mibs/netxms.mib") | |
107 | #define DDIR_IMAGES _T("/images") | |
108 | #define DDIR_FILES _T("/files") | |
e77e56ca | 109 | #define DDIR_REPORTS _T("/reports") |
5039dede AK |
110 | |
111 | #endif /* _WIN32 */ | |
112 | ||
113 | ||
114 | // | |
115 | // Application flags | |
116 | // | |
117 | ||
118 | #define AF_DAEMON 0x00000001 | |
119 | #define AF_USE_SYSLOG 0x00000002 | |
120 | #define AF_ENABLE_NETWORK_DISCOVERY 0x00000004 | |
121 | #define AF_ACTIVE_NETWORK_DISCOVERY 0x00000008 | |
122 | #define AF_LOG_SQL_ERRORS 0x00000010 | |
123 | #define AF_DELETE_EMPTY_SUBNETS 0x00000020 | |
124 | #define AF_ENABLE_SNMP_TRAPD 0x00000040 | |
125 | #define AF_ENABLE_ZONING 0x00000080 | |
126 | #define AF_SYNC_NODE_NAMES_WITH_DNS 0x00000100 | |
127 | #define AF_CHECK_TRUSTED_NODES 0x00000200 | |
3bbc7435 | 128 | #define AF_ENABLE_NXSL_CONTAINER_FUNCS 0x00000400 |
a203ea53 | 129 | #define AF_USE_FQDN_FOR_NODE_NAMES 0x00000800 |
dc4a1aec | 130 | #define AF_APPLY_TO_DISABLED_DCI_FROM_TEMPLATE 0x00001000 |
cbc777ee | 131 | #define AF_WRITE_FULL_DUMP 0x00080000 |
5039dede AK |
132 | #define AF_RESOLVE_NODE_NAMES 0x00100000 |
133 | #define AF_CATCH_EXCEPTIONS 0x00200000 | |
134 | #define AF_INTERNAL_CA 0x00400000 | |
135 | #define AF_DB_LOCKED 0x01000000 | |
136 | #define AF_ENABLE_MULTIPLE_DB_CONN 0x02000000 | |
137 | #define AF_DB_CONNECTION_LOST 0x04000000 | |
138 | #define AF_NO_NETWORK_CONNECTIVITY 0x08000000 | |
b98bccbd | 139 | #define AF_EVENT_STORM_DETECTED 0x10000000 |
5039dede AK |
140 | #define AF_SERVER_INITIALIZED 0x40000000 |
141 | #define AF_SHUTDOWN 0x80000000 | |
142 | ||
5039dede AK |
143 | |
144 | // | |
145 | // Encryption usage policies | |
146 | // | |
147 | ||
148 | #define ENCRYPTION_DISABLED 0 | |
149 | #define ENCRYPTION_ALLOWED 1 | |
150 | #define ENCRYPTION_PREFERRED 2 | |
151 | #define ENCRYPTION_REQUIRED 3 | |
152 | ||
153 | ||
5ad2167d VK |
154 | // |
155 | // Flags for SnmpGet | |
156 | // | |
157 | ||
158 | #define SG_VERBOSE 0x0001 | |
159 | #define SG_STRING_RESULT 0x0002 | |
160 | #define SG_RAW_RESULT 0x0004 | |
161 | #define SG_HSTRING_RESULT 0x0008 | |
4f131f2f | 162 | #define SG_PSTRING_RESULT 0x0010 |
5ad2167d VK |
163 | |
164 | ||
5039dede AK |
165 | // |
166 | // Win32 service and syslog constants | |
167 | // | |
168 | ||
169 | #ifdef _WIN32 | |
170 | ||
171 | #define CORE_SERVICE_NAME _T("NetXMSCore") | |
172 | #define CORE_EVENT_SOURCE _T("NetXMSCore") | |
173 | #define NETXMSD_SYSLOG_NAME CORE_EVENT_SOURCE | |
174 | ||
175 | #else | |
176 | ||
177 | #define NETXMSD_SYSLOG_NAME _T("netxmsd") | |
178 | ||
179 | #endif /* _WIN32 */ | |
180 | ||
181 | ||
182 | // | |
183 | // Single ARP cache entry | |
184 | // | |
185 | ||
186 | typedef struct | |
187 | { | |
188 | DWORD dwIndex; // Interface index | |
189 | DWORD dwIpAddr; | |
190 | BYTE bMacAddr[MAC_ADDR_LENGTH]; | |
191 | } ARP_ENTRY; | |
192 | ||
193 | ||
194 | // | |
195 | // ARP cache structure used by discovery functions and AgentConnection class | |
196 | // | |
197 | ||
198 | typedef struct | |
199 | { | |
200 | DWORD dwNumEntries; | |
201 | ARP_ENTRY *pEntries; | |
202 | } ARP_CACHE; | |
203 | ||
204 | ||
205 | // | |
206 | // Interface information structure used by discovery functions and AgentConnection class | |
207 | // | |
208 | ||
209 | typedef struct | |
210 | { | |
478d4ff4 VK |
211 | TCHAR szName[MAX_DB_STRING]; // Interface display name |
212 | TCHAR szDescription[MAX_DB_STRING]; // Value of ifDescr MIB variable for SNMP agents | |
5039dede AK |
213 | DWORD dwIndex; |
214 | DWORD dwType; | |
eec253a8 VK |
215 | DWORD dwBridgePortNumber; |
216 | DWORD dwSlotNumber; | |
217 | DWORD dwPortNumber; | |
5039dede AK |
218 | DWORD dwIpAddr; |
219 | DWORD dwIpNetMask; | |
220 | BYTE bMacAddr[MAC_ADDR_LENGTH]; | |
221 | int iNumSecondary; // Number of secondary IP's on this interface | |
4c16cdc7 | 222 | bool isPhysicalPort; |
36e44abe | 223 | } NX_INTERFACE_INFO; |
5039dede AK |
224 | |
225 | ||
226 | // | |
227 | // Interface list used by discovery functions and AgentConnection class | |
228 | // | |
229 | ||
98762401 | 230 | class LIBNXSRV_EXPORTABLE InterfaceList |
5039dede | 231 | { |
98762401 VK |
232 | private: |
233 | int m_size; // Number of valid entries | |
234 | int m_allocated; // Number of allocated entries | |
235 | void *m_data; // Can be used by custom enumeration handlers | |
36e44abe | 236 | NX_INTERFACE_INFO *m_interfaces; // Interface entries |
98762401 VK |
237 | |
238 | public: | |
239 | InterfaceList(int initialAlloc = 8); | |
240 | ~InterfaceList(); | |
241 | ||
36e44abe | 242 | void add(NX_INTERFACE_INFO *iface); |
98762401 | 243 | void remove(int index); |
98762401 VK |
244 | |
245 | int getSize() { return m_size; } | |
36e44abe | 246 | NX_INTERFACE_INFO *get(int index) { return ((index >= 0) && (index < m_size)) ? &m_interfaces[index] : NULL; } |
b1bd1fb2 | 247 | NX_INTERFACE_INFO *findByIfIndex(DWORD ifIndex); |
98762401 VK |
248 | |
249 | void setData(void *data) { m_data = data; } | |
250 | void *getData() { return m_data; } | |
251 | }; | |
5039dede AK |
252 | |
253 | ||
77c7801c VK |
254 | // |
255 | // Vlan information | |
256 | // | |
257 | ||
0208aa8c VK |
258 | #define VLAN_PRM_IFINDEX 0 |
259 | #define VLAN_PRM_SLOTPORT 1 | |
5eb3a790 | 260 | #define VLAN_PRM_BPORT 2 |
0208aa8c | 261 | |
77c7801c VK |
262 | class LIBNXSRV_EXPORTABLE VlanInfo |
263 | { | |
264 | private: | |
265 | int m_vlanId; | |
266 | TCHAR *m_name; | |
0208aa8c | 267 | int m_portRefMode; // Port reference mode - by ifIndex or by slot/port |
77c7801c VK |
268 | int m_allocated; |
269 | int m_numPorts; // Number of ports in VLAN | |
0208aa8c | 270 | DWORD *m_ports; // member ports (slot/port pairs or ifIndex) |
b1bd1fb2 VK |
271 | DWORD *m_indexes; // ifIndexes for ports |
272 | DWORD *m_ids; // Interface object IDs for ports | |
77c7801c VK |
273 | |
274 | public: | |
0208aa8c | 275 | VlanInfo(int vlanId, int prm); |
77c7801c VK |
276 | ~VlanInfo(); |
277 | ||
278 | int getVlanId() { return m_vlanId; } | |
0208aa8c | 279 | int getPortReferenceMode() { return m_portRefMode; } |
77c7801c VK |
280 | const TCHAR *getName() { return CHECK_NULL_EX(m_name); } |
281 | int getNumPorts() { return m_numPorts; } | |
282 | DWORD *getPorts() { return m_ports; } | |
b1bd1fb2 VK |
283 | DWORD *getIfIndexes() { return m_indexes; } |
284 | DWORD *getIfIds() { return m_ids; } | |
77c7801c VK |
285 | |
286 | void add(DWORD slot, DWORD port); | |
0208aa8c | 287 | void add(DWORD ifIndex); |
77c7801c | 288 | void setName(const TCHAR *name); |
b1bd1fb2 VK |
289 | |
290 | void prepareForResolve(); | |
291 | void resolvePort(int index, DWORD sp, DWORD ifIndex, DWORD id); | |
77c7801c VK |
292 | }; |
293 | ||
a3050773 VK |
294 | /** |
295 | * Vlan list | |
296 | */ | |
7f632dfe | 297 | class LIBNXSRV_EXPORTABLE VlanList : public RefCountObject |
77c7801c VK |
298 | { |
299 | private: | |
300 | int m_size; // Number of valid entries | |
301 | int m_allocated; // Number of allocated entries | |
302 | void *m_data; // Can be used by custom enumeration handlers | |
303 | VlanInfo **m_vlans; // VLAN entries | |
304 | ||
305 | public: | |
306 | VlanList(int initialAlloc = 8); | |
7f632dfe | 307 | virtual ~VlanList(); |
77c7801c VK |
308 | |
309 | void add(VlanInfo *vlan); | |
0208aa8c | 310 | void addMemberPort(int vlanId, DWORD portId); |
77c7801c VK |
311 | |
312 | int getSize() { return m_size; } | |
313 | VlanInfo *get(int index) { return ((index >= 0) && (index < m_size)) ? m_vlans[index] : NULL; } | |
0208aa8c | 314 | VlanInfo *findById(int id); |
77c7801c VK |
315 | |
316 | void setData(void *data) { m_data = data; } | |
317 | void *getData() { return m_data; } | |
7f632dfe VK |
318 | |
319 | void fillMessage(CSCPMessage *msg); | |
77c7801c VK |
320 | }; |
321 | ||
a3050773 VK |
322 | /** |
323 | * Route information | |
324 | */ | |
5039dede AK |
325 | typedef struct |
326 | { | |
327 | DWORD dwDestAddr; | |
328 | DWORD dwDestMask; | |
329 | DWORD dwNextHop; | |
330 | DWORD dwIfIndex; | |
331 | DWORD dwRouteType; | |
332 | } ROUTE; | |
333 | ||
a3050773 VK |
334 | /** |
335 | * Routing table | |
336 | */ | |
5039dede AK |
337 | typedef struct |
338 | { | |
339 | int iNumEntries; // Number of entries | |
340 | ROUTE *pRoutes; // Route list | |
341 | } ROUTING_TABLE; | |
342 | ||
a3050773 VK |
343 | /** |
344 | * Information about policies installed on agent | |
345 | */ | |
1f385e47 VK |
346 | class LIBNXSRV_EXPORTABLE AgentPolicyInfo |
347 | { | |
348 | private: | |
349 | int m_size; | |
350 | BYTE *m_guidList; | |
351 | int *m_typeList; | |
352 | TCHAR **m_serverList; | |
353 | ||
354 | public: | |
355 | AgentPolicyInfo(CSCPMessage *msg); | |
356 | ~AgentPolicyInfo(); | |
357 | ||
358 | int getSize() { return m_size; } | |
359 | bool getGuid(int index, uuid_t guid); | |
360 | int getType(int index) { return ((index >= 0) && (index < m_size)) ? m_typeList[index] : -1; } | |
361 | const TCHAR *getServer(int index) { return ((index >= 0) && (index < m_size)) ? m_serverList[index] : NULL; } | |
362 | }; | |
363 | ||
a3050773 VK |
364 | /** |
365 | * Agent connection | |
366 | */ | |
5039dede AK |
367 | class LIBNXSRV_EXPORTABLE AgentConnection |
368 | { | |
369 | private: | |
370 | DWORD m_dwAddr; | |
371 | int m_nProtocolVersion; | |
372 | int m_iAuthMethod; | |
08b214c6 | 373 | char m_szSecret[MAX_SECRET_LENGTH]; |
5039dede AK |
374 | time_t m_tLastCommandTime; |
375 | SOCKET m_hSocket; | |
376 | DWORD m_dwNumDataLines; | |
377 | DWORD m_dwRequestId; | |
378 | DWORD m_dwCommandTimeout; | |
7c521895 | 379 | DWORD m_connectionTimeout; |
5039dede AK |
380 | DWORD m_dwRecvTimeout; |
381 | TCHAR **m_ppDataLines; | |
382 | MsgWaitQueue *m_pMsgWaitQueue; | |
383 | BOOL m_bIsConnected; | |
384 | MUTEX m_mutexDataLock; | |
d3a7cf4c | 385 | MUTEX m_mutexSocketWrite; |
5039dede | 386 | THREAD m_hReceiverThread; |
98abc9f1 | 387 | NXCPEncryptionContext *m_pCtx; |
5039dede AK |
388 | int m_iEncryptionPolicy; |
389 | BOOL m_bUseProxy; | |
390 | DWORD m_dwProxyAddr; | |
391 | WORD m_wPort; | |
392 | WORD m_wProxyPort; | |
393 | int m_iProxyAuth; | |
08b214c6 | 394 | char m_szProxySecret[MAX_SECRET_LENGTH]; |
5039dede | 395 | int m_hCurrFile; |
9f6d453a | 396 | TCHAR m_currentFileName[MAX_PATH]; |
bb85e341 VK |
397 | DWORD m_dwDownloadRequestId; |
398 | CONDITION m_condFileDownload; | |
399 | BOOL m_fileDownloadSucceeded; | |
400 | void (*m_downloadProgressCallback)(size_t, void *); | |
401 | void *m_downloadProgressCallbackArg; | |
901a5a9b | 402 | bool m_deleteFileOnDownloadFailure; |
4685a2ad | 403 | bool m_fileUploadInProgress; |
5039dede | 404 | |
a3050773 VK |
405 | void receiverThread(); |
406 | static THREAD_RESULT THREAD_CALL receiverThreadStarter(void *); | |
5039dede AK |
407 | |
408 | protected: | |
7c521895 VK |
409 | void destroyResultData(); |
410 | BOOL sendMessage(CSCPMessage *pMsg); | |
c17f6cbc | 411 | CSCPMessage *waitForMessage(WORD wCode, DWORD dwId, DWORD dwTimeOut) { return m_pMsgWaitQueue->waitForMessage(wCode, dwId, dwTimeOut); } |
7c521895 VK |
412 | DWORD waitForRCC(DWORD dwRqId, DWORD dwTimeOut); |
413 | DWORD setupEncryption(RSA *pServerKey); | |
414 | DWORD authenticate(BOOL bProxyData); | |
415 | DWORD setupProxyConnection(); | |
416 | DWORD getIpAddr() { return ntohl(m_dwAddr); } | |
417 | DWORD prepareFileDownload(const TCHAR *fileName, DWORD rqId, bool append, void (*downloadProgressCallback)(size_t, void *), void *cbArg); | |
5039dede | 418 | |
a3050773 | 419 | virtual void printMsg(const TCHAR *pszFormat, ...); |
f480bdd4 VK |
420 | virtual void onTrap(CSCPMessage *pMsg); |
421 | virtual void onDataPush(CSCPMessage *msg); | |
90284364 | 422 | virtual bool processCustomMessage(CSCPMessage *pMsg); |
f480bdd4 | 423 | virtual void onFileDownload(BOOL success); |
5039dede | 424 | |
a3050773 VK |
425 | void lock() { MutexLock(m_mutexDataLock); } |
426 | void unlock() { MutexUnlock(m_mutexDataLock); } | |
98abc9f1 | 427 | NXCPEncryptionContext *acquireEncryptionContext(); |
5039dede AK |
428 | |
429 | public: | |
5039dede AK |
430 | AgentConnection(DWORD dwAddr, WORD wPort = AGENT_LISTEN_PORT, |
431 | int iAuthMethod = AUTH_NONE, const TCHAR *pszSecret = NULL); | |
432 | virtual ~AgentConnection(); | |
433 | ||
c3acd0f6 | 434 | BOOL connect(RSA *pServerKey = NULL, BOOL bVerbose = FALSE, DWORD *pdwError = NULL, DWORD *pdwSocketError = NULL); |
7c521895 VK |
435 | void disconnect(); |
436 | BOOL isConnected() { return m_bIsConnected; } | |
437 | int getProtocolVersion() { return m_nProtocolVersion; } | |
5039dede | 438 | |
45d84f8a | 439 | SOCKET getSocket() { return m_hSocket; } |
5a9e58a3 | 440 | |
4687826e | 441 | ARP_CACHE *getArpCache(); |
98762401 | 442 | InterfaceList *getInterfaceList(); |
4687826e VK |
443 | ROUTING_TABLE *getRoutingTable(); |
444 | DWORD getParameter(const TCHAR *pszParam, DWORD dwBufSize, TCHAR *pszBuffer); | |
445 | DWORD getList(const TCHAR *pszParam); | |
446 | DWORD getTable(const TCHAR *pszParam, Table **table); | |
7c521895 | 447 | DWORD nop(); |
4687826e | 448 | DWORD execAction(const TCHAR *pszAction, int argc, TCHAR **argv); |
619e5c9b | 449 | DWORD uploadFile(const TCHAR *localFile, const TCHAR *destinationFile = NULL, void (* progressCallback)(INT64, void *) = NULL, void *cbArg = NULL); |
4687826e VK |
450 | DWORD startUpgrade(const TCHAR *pszPkgName); |
451 | DWORD checkNetworkService(DWORD *pdwStatus, DWORD dwIpAddr, int iServiceType, WORD wPort = 0, | |
5039dede | 452 | WORD wProto = 0, const TCHAR *pszRequest = NULL, const TCHAR *pszResponse = NULL); |
cc8ce218 | 453 | DWORD getSupportedParameters(StructArray<NXC_AGENT_PARAM> **paramList, StructArray<NXC_AGENT_TABLE> **tableList); |
4687826e | 454 | DWORD getConfigFile(TCHAR **ppszConfig, DWORD *pdwSize); |
45d84f8a | 455 | DWORD updateConfigFile(const TCHAR *pszConfig); |
7c521895 | 456 | DWORD enableTraps(); |
1f385e47 VK |
457 | DWORD getPolicyInventory(AgentPolicyInfo **info); |
458 | DWORD uninstallPolicy(uuid_t guid); | |
e58b5e54 VK |
459 | |
460 | DWORD generateRequestId() { return m_dwRequestId++; } | |
45d84f8a | 461 | CSCPMessage *customRequest(CSCPMessage *pRequest, const TCHAR *recvFile = NULL, bool appendFile = false, |
f2665675 | 462 | void (*downloadProgressCallback)(size_t, void *) = NULL, void *cbArg = NULL); |
5039dede | 463 | |
7c521895 VK |
464 | DWORD getNumDataLines() { return m_dwNumDataLines; } |
465 | const TCHAR *getDataLine(DWORD dwIndex) { return dwIndex < m_dwNumDataLines ? m_ppDataLines[dwIndex] : _T("(error)"); } | |
5039dede | 466 | |
7c521895 VK |
467 | void setConnectionTimeout(DWORD dwTimeout) { m_connectionTimeout = max(dwTimeout, 1000); } |
468 | DWORD getConnectionTimeout() { return m_connectionTimeout; } | |
45d84f8a VK |
469 | void setCommandTimeout(DWORD dwTimeout) { m_dwCommandTimeout = max(dwTimeout, 500); } |
470 | DWORD getCommandTimeout() { return m_dwCommandTimeout; } | |
7c521895 VK |
471 | void setRecvTimeout(DWORD dwTimeout) { m_dwRecvTimeout = max(dwTimeout, 10000); } |
472 | void setEncryptionPolicy(int iPolicy) { m_iEncryptionPolicy = iPolicy; } | |
473 | void setProxy(DWORD dwAddr, WORD wPort = AGENT_LISTEN_PORT, | |
5039dede | 474 | int iAuthMethod = AUTH_NONE, const TCHAR *pszSecret = NULL); |
7c521895 | 475 | void setPort(WORD wPort) { m_wPort = wPort; } |
08b214c6 | 476 | void setAuthData(int nMethod, const char *pszSecret) { m_iAuthMethod = nMethod; strncpy(m_szSecret, pszSecret, MAX_SECRET_LENGTH); m_szSecret[MAX_SECRET_LENGTH - 1] = 0; } |
901a5a9b | 477 | void setDeleteFileOnDownloadFailure(bool flag) { m_deleteFileOnDownloadFailure = flag; } |
5039dede AK |
478 | }; |
479 | ||
a3050773 VK |
480 | /** |
481 | * Proxy SNMP transport | |
482 | */ | |
5039dede AK |
483 | class LIBNXSRV_EXPORTABLE SNMP_ProxyTransport : public SNMP_Transport |
484 | { | |
485 | protected: | |
486 | AgentConnection *m_pAgentConnection; | |
487 | CSCPMessage *m_pResponse; | |
488 | DWORD m_dwIpAddr; | |
489 | WORD m_wPort; | |
490 | ||
491 | public: | |
492 | SNMP_ProxyTransport(AgentConnection *pConn, DWORD dwIpAddr, WORD wPort); | |
493 | virtual ~SNMP_ProxyTransport(); | |
494 | ||
c4366266 | 495 | virtual int readMessage(SNMP_PDU **ppData, DWORD dwTimeout = INFINITE, |
cd9f247e VK |
496 | struct sockaddr *pSender = NULL, socklen_t *piAddrSize = NULL, |
497 | SNMP_SecurityContext* (*contextFinder)(struct sockaddr *, socklen_t) = NULL); | |
c4366266 | 498 | virtual int sendMessage(SNMP_PDU *pdu); |
5039dede AK |
499 | }; |
500 | ||
501 | ||
502 | // | |
503 | // ISC flags | |
504 | // | |
505 | ||
506 | #define ISCF_IS_CONNECTED ((DWORD)0x00000001) | |
507 | #define ISCF_REQUIRE_ENCRYPTION ((DWORD)0x00000002) | |
508 | ||
509 | ||
510 | // | |
511 | // Inter-server connection (ISC) | |
512 | // | |
513 | ||
514 | class LIBNXSRV_EXPORTABLE ISC | |
515 | { | |
516 | private: | |
517 | DWORD m_flags; | |
518 | DWORD m_addr; | |
519 | WORD m_port; | |
520 | SOCKET m_socket; | |
521 | int m_protocolVersion; | |
522 | DWORD m_requestId; | |
523 | DWORD m_recvTimeout; | |
524 | MsgWaitQueue *m_msgWaitQueue; | |
525 | MUTEX m_mutexDataLock; | |
7b8b337e | 526 | MUTEX m_socketLock; |
5039dede | 527 | THREAD m_hReceiverThread; |
98abc9f1 | 528 | NXCPEncryptionContext *m_ctx; |
5039dede AK |
529 | DWORD m_commandTimeout; |
530 | ||
c17f6cbc | 531 | void ReceiverThread(); |
5039dede AK |
532 | static THREAD_RESULT THREAD_CALL ReceiverThreadStarter(void *); |
533 | ||
534 | protected: | |
535 | void DestroyResultData(void); | |
536 | DWORD SetupEncryption(RSA *pServerKey); | |
537 | DWORD ConnectToService(DWORD service); | |
538 | ||
c17f6cbc VK |
539 | void Lock() { MutexLock(m_mutexDataLock); } |
540 | void Unlock() { MutexUnlock(m_mutexDataLock); } | |
5039dede AK |
541 | |
542 | virtual void PrintMsg(const TCHAR *format, ...); | |
543 | ||
544 | public: | |
545 | ISC(); | |
546 | ISC(DWORD addr, WORD port = NETXMS_ISC_PORT); | |
547 | virtual ~ISC(); | |
548 | ||
549 | DWORD Connect(DWORD service, RSA *serverKey = NULL, BOOL requireEncryption = FALSE); | |
550 | void Disconnect(); | |
551 | ||
552 | BOOL SendMessage(CSCPMessage *msg); | |
c17f6cbc | 553 | CSCPMessage *WaitForMessage(WORD code, DWORD id, DWORD timeOut) { return m_msgWaitQueue->waitForMessage(code, id, timeOut); } |
5039dede AK |
554 | DWORD WaitForRCC(DWORD rqId, DWORD timeOut); |
555 | ||
556 | DWORD Nop(void); | |
557 | }; | |
558 | ||
559 | ||
560 | // | |
561 | // Functions | |
562 | // | |
563 | ||
564 | void LIBNXSRV_EXPORTABLE DestroyArpCache(ARP_CACHE *pArpCache); | |
5039dede AK |
565 | void LIBNXSRV_EXPORTABLE DestroyRoutingTable(ROUTING_TABLE *pRT); |
566 | void LIBNXSRV_EXPORTABLE SortRoutingTable(ROUTING_TABLE *pRT); | |
567 | const TCHAR LIBNXSRV_EXPORTABLE *AgentErrorCodeToText(int iError); | |
568 | ||
569 | void LIBNXSRV_EXPORTABLE WriteLogOther(WORD wType, const TCHAR *format, ...); | |
570 | void LIBNXSRV_EXPORTABLE DbgPrintf(int level, const TCHAR *format, ...); | |
9d88cdc9 | 571 | void LIBNXSRV_EXPORTABLE DbgPrintf2(int level, const TCHAR *format, va_list args); |
c20b2798 | 572 | |
5039dede AK |
573 | void LIBNXSRV_EXPORTABLE SetAgentDEP(int iPolicy); |
574 | ||
575 | const TCHAR LIBNXSRV_EXPORTABLE *ISCErrorCodeToText(DWORD code); | |
576 | ||
5ad2167d VK |
577 | DWORD LIBNXSRV_EXPORTABLE SnmpNewRequestId(); |
578 | DWORD LIBNXSRV_EXPORTABLE SnmpGet(DWORD dwVersion, SNMP_Transport *pTransport, | |
579 | const TCHAR *szOidStr, const DWORD *oidBinary, DWORD dwOidLen, void *pValue, | |
580 | DWORD dwBufferSize, DWORD dwFlags); | |
581 | DWORD LIBNXSRV_EXPORTABLE SnmpEnumerate(DWORD dwVersion, SNMP_Transport *pTransport, const TCHAR *szRootOid, | |
582 | DWORD (* pHandler)(DWORD, SNMP_Variable *, SNMP_Transport *, void *), | |
583 | void *pUserArg, BOOL bVerbose); | |
584 | ||
5039dede AK |
585 | |
586 | // | |
587 | // Variables | |
588 | // | |
589 | ||
590 | extern DWORD LIBNXSRV_EXPORTABLE g_dwFlags; | |
5ad2167d | 591 | extern DWORD LIBNXSRV_EXPORTABLE g_dwSNMPTimeout; |
5039dede | 592 | extern int LIBNXSRV_EXPORTABLE g_nDebugLevel; |
5039dede | 593 | |
89135050 VK |
594 | |
595 | // | |
596 | // Helper finctions for checking server flags | |
597 | // | |
598 | ||
599 | inline bool IsStandalone() | |
600 | { | |
601 | return !(g_dwFlags & AF_DAEMON) ? true : false; | |
602 | } | |
603 | ||
604 | inline bool IsZoningEnabled() | |
605 | { | |
606 | return (g_dwFlags & AF_ENABLE_ZONING) ? true : false; | |
607 | } | |
608 | ||
609 | inline bool IsShutdownInProgress() | |
610 | { | |
611 | return (g_dwFlags & AF_SHUTDOWN) ? true : false; | |
612 | } | |
613 | ||
614 | ||
5039dede | 615 | #endif /* _nxsrvapi_h_ */ |