Merge branch 'master' of 10.3.0.4:public/netxms
[public/netxms.git] / src / server / tools / nxconfig / nxconfig.cpp
1 // nxconfig.cpp : Defines the class behaviors for the application.
2 //
3
4 #include "stdafx.h"
5 #include "nxconfig.h"
6
7 #include "MainFrm.h"
8 #include "ConfigWizard.h"
9 #include "IntroPage.h"
10 #include "DBSelectPage.h"
11 #include "ODBCPage.h"
12 #include "PollCfgPage.h"
13 #include "SMTPPage.h"
14 #include "SummaryPage.h"
15 #include "ProcessingPage.h"
16 #include "ConfigFilePage.h"
17 #include "WinSrvPage.h"
18 #include "SrvDepsPage.h"
19 #include "LoggingPage.h"
20 #include "FinishPage.h"
21
22 #include <winsock2.h>
23 #include <afxsock.h> // MFC socket extensions
24 #include <iphlpapi.h>
25 #include <iprtrmib.h>
26 #include <rtinfo.h>
27
28 #ifdef _DEBUG
29 #define new DEBUG_NEW
30 #undef THIS_FILE
31 static char THIS_FILE[] = __FILE__;
32 #endif
33
34 /////////////////////////////////////////////////////////////////////////////
35 // CNxconfigApp
36
37 BEGIN_MESSAGE_MAP(CNxconfigApp, CWinApp)
38 //{{AFX_MSG_MAP(CNxconfigApp)
39 ON_COMMAND(ID_FILE_CFG_WIZARD, OnFileCfgWizard)
40 //}}AFX_MSG_MAP
41 END_MESSAGE_MAP()
42
43 /////////////////////////////////////////////////////////////////////////////
44 // CNxconfigApp construction
45
46 CNxconfigApp::CNxconfigApp()
47 {
48 // TODO: add construction code here,
49 // Place all significant initialization in InitInstance
50 }
51
52 /////////////////////////////////////////////////////////////////////////////
53 // The one and only CNxconfigApp object
54
55 CNxconfigApp appNxConfig;
56
57 /////////////////////////////////////////////////////////////////////////////
58 // CNxconfigApp initialization
59
60 BOOL CNxconfigApp::InitInstance()
61 {
62 HKEY hKey;
63 DWORD dwSize, dwData = 0;
64
65 if (!AfxSocketInit())
66 {
67 AfxMessageBox(IDP_SOCKETS_INIT_FAILED);
68 return FALSE;
69 }
70
71 // Read installation data from registry
72 if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("Software\\NetXMS\\Server"), 0,
73 KEY_QUERY_VALUE, &hKey) == ERROR_SUCCESS)
74 {
75 dwSize = sizeof(DWORD);
76 RegQueryValueEx(hKey, _T("ServerIsConfigured"), NULL, NULL, (BYTE *)&dwData, &dwSize);
77
78 dwSize = (MAX_PATH - 16) * sizeof(TCHAR);
79 if (RegQueryValueEx(hKey, _T("InstallPath"), NULL, NULL,
80 (BYTE *)m_szInstallDir, &dwSize) != ERROR_SUCCESS)
81 {
82 AfxMessageBox(_T("Unable to determine NetXMS installation directory"));
83 m_szInstallDir[0] = 0;
84 }
85
86 RegCloseKey(hKey);
87 }
88 else
89 {
90 AfxMessageBox(_T("Unable to determine NetXMS installation directory"));
91 m_szInstallDir[0] = 0;
92 }
93
94 // Parse command line
95 if (!_tcsicmp(m_lpCmdLine, _T("--create-agent-config")))
96 {
97 CreateAgentConfig();
98 return FALSE;
99 }
100
101 // Check if server is already configured or we cannot determine
102 // installation directory
103 if ((dwData != 0) || (m_szInstallDir[0] == 0))
104 {
105 if ((dwData != 0) && (_tcsicmp(m_lpCmdLine, _T("--configure-if-needed"))))
106 AfxMessageBox(_T("Server already configured"));
107 return FALSE;
108 }
109
110 AfxEnableControlContainer();
111
112 // Standard initialization
113 // If you are not using these features and wish to reduce the size
114 // of your final executable, you should remove from the following
115 // the specific initialization routines you do not need.
116
117 // Change the registry key under which our settings are stored.
118 SetRegistryKey(_T("NetXMS"));
119
120 // To create the main window, this code creates a new frame window
121 // object and then sets it as the application's main window object.
122 CMainFrame* pFrame = new CMainFrame;
123 m_pMainWnd = pFrame;
124
125 // create and load the frame with its resources
126 pFrame->LoadFrame(IDR_MAINFRAME,
127 WS_OVERLAPPEDWINDOW | FWS_ADDTOTITLE, NULL,
128 NULL);
129
130 // The one and only window has been initialized, so show and update it.
131 //pFrame->ShowWindow(SW_SHOW);
132 //pFrame->UpdateWindow();
133
134 pFrame->PostMessage(WM_COMMAND, ID_FILE_CFG_WIZARD);
135
136 return TRUE;
137 }
138
139 /////////////////////////////////////////////////////////////////////////////
140 // CNxconfigApp message handlers
141
142 void CNxconfigApp::OnFileCfgWizard()
143 {
144 CConfigWizard dlg(_T("Configure NetXMS Server"), m_pMainWnd, 0);
145 CIntroPage pgIntro;
146 CConfigFilePage pgConfigFile;
147 CDBSelectPage pgSelectDB;
148 CODBCPage pgCheckODBC;
149 CPollCfgPage pgPollConfig;
150 CSMTPPage pgSMTP;
151 CLoggingPage pgLogging;
152 CWinSrvPage pgWinSrv;
153 CSrvDepsPage pgSrvDeps;
154 CSummaryPage pgSummary;
155 CProcessingPage pgProcessing;
156 CFinishPage pgFinish;
157
158 dlg.m_psh.dwFlags |= PSH_WIZARD;
159 dlg.AddPage(&pgIntro);
160 dlg.AddPage(&pgConfigFile);
161 dlg.AddPage(&pgSelectDB);
162 dlg.AddPage(&pgCheckODBC);
163 dlg.AddPage(&pgPollConfig);
164 dlg.AddPage(&pgSMTP);
165 dlg.AddPage(&pgLogging);
166 dlg.AddPage(&pgWinSrv);
167 dlg.AddPage(&pgSrvDeps);
168 dlg.AddPage(&pgSummary);
169 dlg.AddPage(&pgProcessing);
170 dlg.AddPage(&pgFinish);
171
172 if (dlg.DoModal() == ID_WIZFINISH)
173 {
174 HKEY hKey;
175 DWORD dwData = 1;
176
177 if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("Software\\NetXMS\\Server"), 0,
178 KEY_ALL_ACCESS, &hKey) == ERROR_SUCCESS)
179 {
180 RegSetValueEx(hKey, _T("ServerIsConfigured"), 0, REG_DWORD, (BYTE *)&dwData, sizeof(DWORD));
181 RegCloseKey(hKey);
182 }
183 }
184
185 m_pMainWnd->PostMessage(WM_COMMAND, ID_APP_EXIT);
186 }
187
188
189 //
190 // Create configuration file for local agent
191 //
192
193 void CNxconfigApp::CreateAgentConfig()
194 {
195 FILE *fp;
196 time_t currTime;
197 DWORD dwSize;
198 IP_ADAPTER_INFO *pBuffer, *pInfo;
199 IP_ADDR_STRING *pAddr;
200 TCHAR szAddrList[4096], szFile[MAX_PATH];
201
202 _sntprintf(szFile, MAX_PATH, _T("%s\\etc\\nxagentd.conf"), m_szInstallDir);
203 if (_taccess(szFile, 0) == 0)
204 return; // File already exist, we shouldn't overwrite it
205
206 // Get local interface list
207 szAddrList[0] = 0;
208 if (GetAdaptersInfo(NULL, &dwSize) == ERROR_BUFFER_OVERFLOW)
209 {
210 pBuffer = (IP_ADAPTER_INFO *)malloc(dwSize);
211 if (GetAdaptersInfo(pBuffer, &dwSize) == ERROR_SUCCESS)
212 {
213 for(pInfo = pBuffer; pInfo != NULL; pInfo = pInfo->Next)
214 {
215 // Read all IP addresses for adapter
216 for(pAddr = &pInfo->IpAddressList; pAddr != NULL; pAddr = pAddr->Next)
217 {
218 if (strcmp(pAddr->IpAddress.String, "0.0.0.0"))
219 {
220 if (szAddrList[0] != 0)
221 _tcscat(szAddrList, _T(", "));
222 #ifdef UNICODE
223 WCHAR ipaddr[32];
224 MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, pAddr->IpAddress.String, -1, ipaddr, 32);
225 wcscat(szAddrList, ipaddr);
226 #else
227 strcat(szAddrList, pAddr->IpAddress.String);
228 #endif
229 }
230 }
231 }
232 }
233 free(pBuffer);
234 }
235
236 fp = _tfopen(szFile, _T("w"));
237 if (fp != NULL)
238 {
239 currTime = time(NULL);
240 _ftprintf(fp, _T("#\n# NetXMS agent configuration file\n# Created by server installer at %s#\n\n"), _tctime(&currTime));
241 if (szAddrList[0] != 0)
242 {
243 _ftprintf(fp, _T("LogFile = {syslog}\nMasterServers = 127.0.0.1, %s\n"), szAddrList);
244 }
245 else
246 {
247 _ftprintf(fp, _T("LogFile = {syslog}\nMasterServers = 127.0.0.1\n"));
248 }
249 _ftprintf(fp, _T("FileStore = %s\\var\n"), m_szInstallDir);
250 _ftprintf(fp, _T("RequireAuthentication = no\n"));
251 _ftprintf(fp, _T("SubAgent = winperf.nsm\nSubAgent = portcheck.nsm\n"));
252 fclose(fp);
253 }
254 }