| Current File : //usr/share/doc/net-snmp/html/winservice_8c_source.html |
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
<meta http-equiv="X-UA-Compatible" content="IE=9"/>
<title>net-snmp: winservice.c Source File</title>
<link href="tabs.css" rel="stylesheet" type="text/css"/>
<link href="doxygen.css" rel="stylesheet" type="text/css" />
</head>
<body>
<div id="top"><!-- do not remove this div! -->
<div id="titlearea">
<table cellspacing="0" cellpadding="0">
<tbody>
<tr style="height: 56px;">
<td style="padding-left: 0.5em;">
<div id="projectname">net-snmp
 <span id="projectnumber">5.4.1</span>
</div>
</td>
</tr>
</tbody>
</table>
</div>
<!-- Generated by Doxygen 1.7.6.1 -->
<div id="navrow1" class="tabs">
<ul class="tablist">
<li><a href="index.html"><span>Main Page</span></a></li>
<li><a href="pages.html"><span>Related Pages</span></a></li>
<li><a href="modules.html"><span>Modules</span></a></li>
<li><a href="annotated.html"><span>Data Structures</span></a></li>
<li class="current"><a href="files.html"><span>Files</span></a></li>
<li><a href="examples.html"><span>Examples</span></a></li>
</ul>
</div>
<div id="navrow2" class="tabs2">
<ul class="tablist">
<li><a href="files.html"><span>File List</span></a></li>
<li><a href="globals.html"><span>Globals</span></a></li>
</ul>
</div>
</div>
<div class="header">
<div class="headertitle">
<div class="title">winservice.c</div> </div>
</div><!--header-->
<div class="contents">
<div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">/*</span>
<a name="l00002"></a>00002 <span class="comment"> * Windows Service related function definitions</span>
<a name="l00003"></a>00003 <span class="comment"> * By Raju Krishnappa(raju_krishnappa@yahoo.com)</span>
<a name="l00004"></a>00004 <span class="comment"> *</span>
<a name="l00005"></a>00005 <span class="comment"> */</span>
<a name="l00006"></a>00006
<a name="l00007"></a>00007 <span class="preprocessor">#ifdef WIN32</span>
<a name="l00008"></a>00008 <span class="preprocessor"></span>
<a name="l00009"></a>00009 <span class="preprocessor">#include <windows.h></span>
<a name="l00010"></a>00010 <span class="preprocessor">#include <tchar.h></span>
<a name="l00011"></a>00011
<a name="l00012"></a>00012 <span class="preprocessor">#include <stdio.h></span> <span class="comment">/* sprintf */</span>
<a name="l00013"></a>00013 <span class="preprocessor">#include <process.h></span> <span class="comment">/* beginthreadex */</span>
<a name="l00014"></a>00014
<a name="l00015"></a>00015 <span class="preprocessor">#include <net-snmp/library/winservice.h></span>
<a name="l00016"></a>00016
<a name="l00017"></a>00017 <span class="preprocessor">#ifdef mingw32 </span><span class="comment">/* MinGW doesn't fully support exception handling. */</span>
<a name="l00018"></a>00018
<a name="l00019"></a>00019 <span class="preprocessor">#define TRY if(1)</span>
<a name="l00020"></a>00020 <span class="preprocessor"></span><span class="preprocessor">#define LEAVE goto labelFIN</span>
<a name="l00021"></a>00021 <span class="preprocessor"></span><span class="preprocessor">#define FINALLY do { \</span>
<a name="l00022"></a>00022 <span class="preprocessor">labelFIN: \</span>
<a name="l00023"></a>00023 <span class="preprocessor"> ; \</span>
<a name="l00024"></a>00024 <span class="preprocessor">} while(0); if(1)</span>
<a name="l00025"></a>00025 <span class="preprocessor"></span>
<a name="l00026"></a>00026 <span class="preprocessor">#else</span>
<a name="l00027"></a>00027 <span class="preprocessor"></span>
<a name="l00028"></a>00028 <span class="preprocessor">#define TRY __try</span>
<a name="l00029"></a>00029 <span class="preprocessor"></span><span class="preprocessor">#define LEAVE __leave</span>
<a name="l00030"></a>00030 <span class="preprocessor"></span><span class="preprocessor">#define FINALLY __finally</span>
<a name="l00031"></a>00031 <span class="preprocessor"></span>
<a name="l00032"></a>00032 <span class="preprocessor">#endif </span><span class="comment">/* mingw32 */</span>
<a name="l00033"></a>00033
<a name="l00034"></a>00034 <span class="comment">/*</span>
<a name="l00035"></a>00035 <span class="comment"> * External global variables used here</span>
<a name="l00036"></a>00036 <span class="comment"> */</span>
<a name="l00037"></a>00037
<a name="l00038"></a>00038 <span class="comment">/*</span>
<a name="l00039"></a>00039 <span class="comment"> * Application Name </span>
<a name="l00040"></a>00040 <span class="comment"> * This should be declared by the application, which wants to register as</span>
<a name="l00041"></a>00041 <span class="comment"> * windows service</span>
<a name="l00042"></a>00042 <span class="comment"> */</span>
<a name="l00043"></a>00043 <span class="keyword">extern</span> LPTSTR app_name_long;
<a name="l00044"></a>00044
<a name="l00045"></a>00045 <span class="comment">/*</span>
<a name="l00046"></a>00046 <span class="comment"> * Declare global variable</span>
<a name="l00047"></a>00047 <span class="comment"> */</span>
<a name="l00048"></a>00048
<a name="l00049"></a>00049 <span class="comment">/*</span>
<a name="l00050"></a>00050 <span class="comment"> * Flag to indicate whether process is running as Service </span>
<a name="l00051"></a>00051 <span class="comment"> */</span>
<a name="l00052"></a>00052 BOOL g_fRunningAsService = FALSE;
<a name="l00053"></a>00053
<a name="l00054"></a>00054 <span class="comment">/*</span>
<a name="l00055"></a>00055 <span class="comment"> * Variable to maintain Current Service status </span>
<a name="l00056"></a>00056 <span class="comment"> */</span>
<a name="l00057"></a>00057 <span class="keyword">static</span> SERVICE_STATUS ServiceStatus;
<a name="l00058"></a>00058
<a name="l00059"></a>00059 <span class="comment">/*</span>
<a name="l00060"></a>00060 <span class="comment"> * Service Handle </span>
<a name="l00061"></a>00061 <span class="comment"> */</span>
<a name="l00062"></a>00062 <span class="keyword">static</span> SERVICE_STATUS_HANDLE hServiceStatus = 0L;
<a name="l00063"></a>00063
<a name="l00064"></a>00064 <span class="comment">/*</span>
<a name="l00065"></a>00065 <span class="comment"> * Service Table Entry </span>
<a name="l00066"></a>00066 <span class="comment"> */</span>
<a name="l00067"></a>00067 SERVICE_TABLE_ENTRY ServiceTableEntry[] = {
<a name="l00068"></a>00068 {NULL, ServiceMain}, <span class="comment">/* Service Main function */</span>
<a name="l00069"></a>00069 {NULL, NULL}
<a name="l00070"></a>00070 };
<a name="l00071"></a>00071
<a name="l00072"></a>00072 <span class="comment">/*</span>
<a name="l00073"></a>00073 <span class="comment"> * Handle to Thread, to implement Pause, Resume and Stop functions</span>
<a name="l00074"></a>00074 <span class="comment"> */</span>
<a name="l00075"></a>00075 <span class="keyword">static</span> HANDLE hServiceThread = NULL; <span class="comment">/* Thread Handle */</span>
<a name="l00076"></a>00076
<a name="l00077"></a>00077 <span class="comment">/*</span>
<a name="l00078"></a>00078 <span class="comment"> * Holds calling partys Function Entry point, that should start</span>
<a name="l00079"></a>00079 <span class="comment"> * when entering service mode</span>
<a name="l00080"></a>00080 <span class="comment"> */</span>
<a name="l00081"></a>00081 <span class="keyword">static</span> INT (*ServiceEntryPoint) (INT Argc, LPTSTR Argv[]) = 0L;
<a name="l00082"></a>00082
<a name="l00083"></a>00083 <span class="comment">/*</span>
<a name="l00084"></a>00084 <span class="comment"> * To hold Stop Function address, to be called when STOP request</span>
<a name="l00085"></a>00085 <span class="comment"> * received from the SCM</span>
<a name="l00086"></a>00086 <span class="comment"> */</span>
<a name="l00087"></a>00087 <span class="keyword">static</span> VOID (*StopFunction) (VOID) = 0L;
<a name="l00088"></a>00088
<a name="l00089"></a>00089 VOID
<a name="l00090"></a>00090 ProcessError (WORD eventLogType, LPCTSTR pszMessage, <span class="keywordtype">int</span> useGetLastError, <span class="keywordtype">int</span> quiet);
<a name="l00091"></a>00091
<a name="l00092"></a>00092 <span class="comment">/*</span>
<a name="l00093"></a>00093 <span class="comment"> * To register as Windows Service with SCM(Service Control Manager)</span>
<a name="l00094"></a>00094 <span class="comment"> * Input - Service Name, Service Display Name,Service Description and</span>
<a name="l00095"></a>00095 <span class="comment"> * Service startup arguments</span>
<a name="l00096"></a>00096 <span class="comment"> */</span>
<a name="l00097"></a>00097 <span class="keywordtype">int</span>
<a name="l00098"></a>00098 RegisterService (LPCTSTR lpszServiceName, LPCTSTR lpszServiceDisplayName,
<a name="l00099"></a>00099 LPCTSTR lpszServiceDescription,
<a name="l00100"></a>00100 <a class="code" href="struct__InputParams.html">InputParams</a> * StartUpArg, <span class="keywordtype">int</span> quiet) <span class="comment">/* Startup argument to the service */</span>
<a name="l00101"></a>00101 {
<a name="l00102"></a>00102 TCHAR szServicePath[MAX_PATH]; <span class="comment">/* To hold module File name */</span>
<a name="l00103"></a>00103 TCHAR MsgErrorString[MAX_STR_SIZE]; <span class="comment">/* Message or Error string */</span>
<a name="l00104"></a>00104 TCHAR szServiceCommand[MAX_PATH + 9]; <span class="comment">/* Command to execute */</span>
<a name="l00105"></a>00105 SC_HANDLE hSCManager = NULL;
<a name="l00106"></a>00106 SC_HANDLE hService = NULL;
<a name="l00107"></a>00107 TCHAR szRegAppLogKey[] =
<a name="l00108"></a>00108 <span class="stringliteral">"SYSTEM\\CurrentControlSet\\Services\\EventLog\\Application\\"</span>;
<a name="l00109"></a>00109 TCHAR szRegKey[512];
<a name="l00110"></a>00110 HKEY hKey = NULL; <span class="comment">/* Key to registry entry */</span>
<a name="l00111"></a>00111 HKEY hParamKey = NULL; <span class="comment">/* To store startup parameters */</span>
<a name="l00112"></a>00112 DWORD dwData; <span class="comment">/* Type of logging supported */</span>
<a name="l00113"></a>00113 DWORD i, j; <span class="comment">/* Loop variables */</span>
<a name="l00114"></a>00114 <span class="keywordtype">int</span> exitStatus = 0;
<a name="l00115"></a>00115 GetModuleFileName (NULL, szServicePath, MAX_PATH);
<a name="l00116"></a>00116 TRY
<a name="l00117"></a>00117 {
<a name="l00118"></a>00118
<a name="l00119"></a>00119 <span class="comment">/*</span>
<a name="l00120"></a>00120 <span class="comment"> * Open Service Control Manager handle </span>
<a name="l00121"></a>00121 <span class="comment"> */</span>
<a name="l00122"></a>00122 hSCManager = OpenSCManager (NULL, NULL, SC_MANAGER_CREATE_SERVICE);
<a name="l00123"></a>00123 <span class="keywordflow">if</span> (hSCManager == NULL)
<a name="l00124"></a>00124 {
<a name="l00125"></a>00125 ProcessError (EVENTLOG_ERROR_TYPE, _T (<span class="stringliteral">"Can't open SCM (Service Control Manager)"</span>), 1, quiet);
<a name="l00126"></a>00126 exitStatus = SERVICE_ERROR_SCM_OPEN;
<a name="l00127"></a>00127 LEAVE;
<a name="l00128"></a>00128 }
<a name="l00129"></a>00129
<a name="l00130"></a>00130 <span class="comment">/*</span>
<a name="l00131"></a>00131 <span class="comment"> * Generate the Command to be executed by SCM </span>
<a name="l00132"></a>00132 <span class="comment"> */</span>
<a name="l00133"></a>00133 _snprintf (szServiceCommand, <span class="keyword">sizeof</span>(szServiceCommand), <span class="stringliteral">"%s %s"</span>, szServicePath, _T (<span class="stringliteral">"-service"</span>));
<a name="l00134"></a>00134
<a name="l00135"></a>00135 <span class="comment">/*</span>
<a name="l00136"></a>00136 <span class="comment"> * Create the Desired service </span>
<a name="l00137"></a>00137 <span class="comment"> */</span>
<a name="l00138"></a>00138 hService = CreateService (hSCManager, lpszServiceName, lpszServiceDisplayName,
<a name="l00139"></a>00139 SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS,
<a name="l00140"></a>00140 SERVICE_AUTO_START, SERVICE_ERROR_NORMAL, szServiceCommand,
<a name="l00141"></a>00141 NULL, <span class="comment">/* load-order group */</span>
<a name="l00142"></a>00142 NULL, <span class="comment">/* group member tag */</span>
<a name="l00143"></a>00143 NULL, <span class="comment">/* dependencies */</span>
<a name="l00144"></a>00144 NULL, <span class="comment">/* account */</span>
<a name="l00145"></a>00145 NULL); <span class="comment">/* password */</span>
<a name="l00146"></a>00146 <span class="keywordflow">if</span> (hService == NULL)
<a name="l00147"></a>00147 {
<a name="l00148"></a>00148 _snprintf (MsgErrorString, <span class="keyword">sizeof</span>(MsgErrorString), <span class="stringliteral">"%s %s"</span>,
<a name="l00149"></a>00149 _T (<span class="stringliteral">"Can't create service"</span>), lpszServiceDisplayName);
<a name="l00150"></a>00150 ProcessError (EVENTLOG_ERROR_TYPE, MsgErrorString, 1, quiet);
<a name="l00151"></a>00151
<a name="l00152"></a>00152 exitStatus = SERVICE_ERROR_CREATE_SERVICE;
<a name="l00153"></a>00153 LEAVE;
<a name="l00154"></a>00154 }
<a name="l00155"></a>00155
<a name="l00156"></a>00156 <span class="comment">/*</span>
<a name="l00157"></a>00157 <span class="comment"> * Create registry entries for EventLog </span>
<a name="l00158"></a>00158 <span class="comment"> */</span>
<a name="l00159"></a>00159 <span class="comment">/*</span>
<a name="l00160"></a>00160 <span class="comment"> * Create registry Application event log key </span>
<a name="l00161"></a>00161 <span class="comment"> */</span>
<a name="l00162"></a>00162 _tcscpy (szRegKey, szRegAppLogKey);
<a name="l00163"></a>00163 _tcscat (szRegKey, lpszServiceName);
<a name="l00164"></a>00164
<a name="l00165"></a>00165 <span class="comment">/*</span>
<a name="l00166"></a>00166 <span class="comment"> * Create registry key </span>
<a name="l00167"></a>00167 <span class="comment"> */</span>
<a name="l00168"></a>00168 <span class="keywordflow">if</span> (RegCreateKey (HKEY_LOCAL_MACHINE, szRegKey, &hKey) != ERROR_SUCCESS)
<a name="l00169"></a>00169 {
<a name="l00170"></a>00170 _snprintf (MsgErrorString, <span class="keyword">sizeof</span>(MsgErrorString), <span class="stringliteral">"%s %s"</span>,
<a name="l00171"></a>00171 _T (<span class="stringliteral">"is unable to create registry entries"</span>), lpszServiceDisplayName);
<a name="l00172"></a>00172 ProcessError (EVENTLOG_ERROR_TYPE, MsgErrorString, 1, quiet);
<a name="l00173"></a>00173 exitStatus = SERVICE_ERROR_CREATE_REGISTRY_ENTRIES;
<a name="l00174"></a>00174 LEAVE;
<a name="l00175"></a>00175 }
<a name="l00176"></a>00176
<a name="l00177"></a>00177 <span class="comment">/*</span>
<a name="l00178"></a>00178 <span class="comment"> * Add Event ID message file name to the 'EventMessageFile' subkey </span>
<a name="l00179"></a>00179 <span class="comment"> */</span>
<a name="l00180"></a>00180 RegSetValueEx (hKey, <span class="stringliteral">"EventMessageFile"</span>, 0, REG_EXPAND_SZ,
<a name="l00181"></a>00181 (CONST BYTE *) szServicePath,
<a name="l00182"></a>00182 _tcslen (szServicePath) + <span class="keyword">sizeof</span> (TCHAR));
<a name="l00183"></a>00183
<a name="l00184"></a>00184 <span class="comment">/*</span>
<a name="l00185"></a>00185 <span class="comment"> * Set the supported types flags. </span>
<a name="l00186"></a>00186 <span class="comment"> */</span>
<a name="l00187"></a>00187 dwData = EVENTLOG_ERROR_TYPE | EVENTLOG_WARNING_TYPE | EVENTLOG_INFORMATION_TYPE;
<a name="l00188"></a>00188 RegSetValueEx (hKey, <span class="stringliteral">"TypesSupported"</span>, 0, REG_DWORD,
<a name="l00189"></a>00189 (CONST BYTE *) & dwData, <span class="keyword">sizeof</span> (DWORD));
<a name="l00190"></a>00190
<a name="l00191"></a>00191 <span class="comment">/*</span>
<a name="l00192"></a>00192 <span class="comment"> * Close Registry key </span>
<a name="l00193"></a>00193 <span class="comment"> */</span>
<a name="l00194"></a>00194 RegCloseKey (hKey);
<a name="l00195"></a>00195
<a name="l00196"></a>00196 <span class="comment">/*</span>
<a name="l00197"></a>00197 <span class="comment"> * Set Service Description String and save startup parameters if present</span>
<a name="l00198"></a>00198 <span class="comment"> */</span>
<a name="l00199"></a>00199 <span class="keywordflow">if</span> (lpszServiceDescription != NULL || StartUpArg->Argc > 2)
<a name="l00200"></a>00200 {
<a name="l00201"></a>00201 <span class="comment">/*</span>
<a name="l00202"></a>00202 <span class="comment"> * Create Registry Key path </span>
<a name="l00203"></a>00203 <span class="comment"> */</span>
<a name="l00204"></a>00204 _tcscpy (szRegKey, _T (<span class="stringliteral">"SYSTEM\\CurrentControlSet\\Services\\"</span>));
<a name="l00205"></a>00205 _tcscat (szRegKey, app_name_long);
<a name="l00206"></a>00206 hKey = NULL;
<a name="l00207"></a>00207
<a name="l00208"></a>00208 <span class="comment">/*</span>
<a name="l00209"></a>00209 <span class="comment"> * Open Registry key using Create and Set access. </span>
<a name="l00210"></a>00210 <span class="comment"> */</span>
<a name="l00211"></a>00211 <span class="keywordflow">if</span> (RegOpenKeyEx (HKEY_LOCAL_MACHINE, szRegKey, 0, KEY_WRITE,
<a name="l00212"></a>00212 &hKey) != ERROR_SUCCESS)
<a name="l00213"></a>00213 {
<a name="l00214"></a>00214 _snprintf (MsgErrorString, <span class="keyword">sizeof</span>(MsgErrorString), <span class="stringliteral">"%s %s"</span>,
<a name="l00215"></a>00215 _T (<span class="stringliteral">"is unable to create registry entries"</span>),
<a name="l00216"></a>00216 lpszServiceDisplayName);
<a name="l00217"></a>00217 ProcessError (EVENTLOG_ERROR_TYPE, MsgErrorString, 1, quiet);
<a name="l00218"></a>00218 exitStatus = SERVICE_ERROR_CREATE_REGISTRY_ENTRIES;
<a name="l00219"></a>00219 LEAVE;
<a name="l00220"></a>00220 }
<a name="l00221"></a>00221
<a name="l00222"></a>00222 <span class="comment">/*</span>
<a name="l00223"></a>00223 <span class="comment"> * Create description subkey and the set value </span>
<a name="l00224"></a>00224 <span class="comment"> */</span>
<a name="l00225"></a>00225 <span class="keywordflow">if</span> (lpszServiceDescription != NULL)
<a name="l00226"></a>00226 {
<a name="l00227"></a>00227 <span class="keywordflow">if</span> (RegSetValueEx (hKey, <span class="stringliteral">"Description"</span>, 0, REG_SZ,
<a name="l00228"></a>00228 (CONST BYTE *) lpszServiceDescription,
<a name="l00229"></a>00229 _tcslen (lpszServiceDescription) +
<a name="l00230"></a>00230 <span class="keyword">sizeof</span> (TCHAR)) != ERROR_SUCCESS)
<a name="l00231"></a>00231 {
<a name="l00232"></a>00232 _snprintf (MsgErrorString, <span class="keyword">sizeof</span>(MsgErrorString), <span class="stringliteral">"%s %s"</span>,
<a name="l00233"></a>00233 _T (<span class="stringliteral">"is unable to create registry entries"</span>),
<a name="l00234"></a>00234 lpszServiceDisplayName);
<a name="l00235"></a>00235 ProcessError (EVENTLOG_ERROR_TYPE, MsgErrorString, 1, quiet);
<a name="l00236"></a>00236 exitStatus = SERVICE_ERROR_CREATE_REGISTRY_ENTRIES;
<a name="l00237"></a>00237 LEAVE;
<a name="l00238"></a>00238 };
<a name="l00239"></a>00239 }
<a name="l00240"></a>00240
<a name="l00241"></a>00241 <span class="comment">/*</span>
<a name="l00242"></a>00242 <span class="comment"> * Save startup arguments if they are present </span>
<a name="l00243"></a>00243 <span class="comment"> */</span>
<a name="l00244"></a>00244 <span class="keywordflow">if</span> (StartUpArg->Argc > 2)
<a name="l00245"></a>00245 {
<a name="l00246"></a>00246 <span class="comment">/*</span>
<a name="l00247"></a>00247 <span class="comment"> * Create Subkey parameters </span>
<a name="l00248"></a>00248 <span class="comment"> */</span>
<a name="l00249"></a>00249 <span class="keywordflow">if</span> (RegCreateKeyEx
<a name="l00250"></a>00250 (hKey, <span class="stringliteral">"Parameters"</span>, 0, NULL,
<a name="l00251"></a>00251 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL,
<a name="l00252"></a>00252 &hParamKey, NULL) != ERROR_SUCCESS)
<a name="l00253"></a>00253 {
<a name="l00254"></a>00254 _snprintf (MsgErrorString, <span class="keyword">sizeof</span>(MsgErrorString), <span class="stringliteral">"%s %s"</span>,
<a name="l00255"></a>00255 _T (<span class="stringliteral">"is unable to create registry entries"</span>),
<a name="l00256"></a>00256 lpszServiceDisplayName);
<a name="l00257"></a>00257 ProcessError (EVENTLOG_ERROR_TYPE, MsgErrorString, 1, quiet);
<a name="l00258"></a>00258 exitStatus = SERVICE_ERROR_CREATE_REGISTRY_ENTRIES;
<a name="l00259"></a>00259 LEAVE;
<a name="l00260"></a>00260 }
<a name="l00261"></a>00261
<a name="l00262"></a>00262 <span class="comment">/*</span>
<a name="l00263"></a>00263 <span class="comment"> * Save parameters </span>
<a name="l00264"></a>00264 <span class="comment"> */</span>
<a name="l00265"></a>00265
<a name="l00266"></a>00266 <span class="comment">/*</span>
<a name="l00267"></a>00267 <span class="comment"> * Loop through arguments </span>
<a name="l00268"></a>00268 <span class="comment"> */</span>
<a name="l00269"></a>00269 <span class="keywordflow">if</span> (quiet) <span class="comment">/* Make sure we don't store -quiet arg */</span>
<a name="l00270"></a>00270 i = 3;
<a name="l00271"></a>00271 <span class="keywordflow">else</span>
<a name="l00272"></a>00272 i = 2;
<a name="l00273"></a>00273
<a name="l00274"></a>00274 <span class="keywordflow">for</span> (j = 1; i < StartUpArg->Argc; i++, j++)
<a name="l00275"></a>00275 {
<a name="l00276"></a>00276 _snprintf (szRegKey, <span class="keyword">sizeof</span>(szRegKey), <span class="stringliteral">"%s%d"</span>, _T (<span class="stringliteral">"Param"</span>), j);
<a name="l00277"></a>00277
<a name="l00278"></a>00278 <span class="comment">/*</span>
<a name="l00279"></a>00279 <span class="comment"> * Create registry key </span>
<a name="l00280"></a>00280 <span class="comment"> */</span>
<a name="l00281"></a>00281 <span class="keywordflow">if</span> (RegSetValueEx
<a name="l00282"></a>00282 (hParamKey, szRegKey, 0, REG_SZ,
<a name="l00283"></a>00283 (CONST BYTE *) StartUpArg->Argv[i],
<a name="l00284"></a>00284 _tcslen (StartUpArg->Argv[i]) +
<a name="l00285"></a>00285 sizeof (TCHAR)) != ERROR_SUCCESS)
<a name="l00286"></a>00286 {
<a name="l00287"></a>00287 _snprintf (MsgErrorString, <span class="keyword">sizeof</span>(MsgErrorString), <span class="stringliteral">"%s %s"</span>,
<a name="l00288"></a>00288 _T (<span class="stringliteral">"is unable to create registry entries"</span>),
<a name="l00289"></a>00289 lpszServiceDisplayName);
<a name="l00290"></a>00290 ProcessError (EVENTLOG_ERROR_TYPE, MsgErrorString, 1, quiet);
<a name="l00291"></a>00291 exitStatus = SERVICE_ERROR_CREATE_REGISTRY_ENTRIES;
<a name="l00292"></a>00292 LEAVE;
<a name="l00293"></a>00293 };
<a name="l00294"></a>00294 }
<a name="l00295"></a>00295 }
<a name="l00296"></a>00296
<a name="l00297"></a>00297 <span class="comment">/*</span>
<a name="l00298"></a>00298 <span class="comment"> * Everything is set, delete hKey </span>
<a name="l00299"></a>00299 <span class="comment"> */</span>
<a name="l00300"></a>00300 RegCloseKey (hParamKey);
<a name="l00301"></a>00301 RegCloseKey (hKey);
<a name="l00302"></a>00302 }
<a name="l00303"></a>00303
<a name="l00304"></a>00304 <span class="comment">/*</span>
<a name="l00305"></a>00305 <span class="comment"> * Ready to Log messages </span>
<a name="l00306"></a>00306 <span class="comment"> */</span>
<a name="l00307"></a>00307
<a name="l00308"></a>00308 <span class="comment">/*</span>
<a name="l00309"></a>00309 <span class="comment"> * Successfully registered as service </span>
<a name="l00310"></a>00310 <span class="comment"> */</span>
<a name="l00311"></a>00311 _snprintf (MsgErrorString, <span class="keyword">sizeof</span>(MsgErrorString), <span class="stringliteral">"%s %s"</span>, lpszServiceName,
<a name="l00312"></a>00312 _T (<span class="stringliteral">"successfully registered as a service"</span>));
<a name="l00313"></a>00313
<a name="l00314"></a>00314 <span class="comment">/*</span>
<a name="l00315"></a>00315 <span class="comment"> * Log message to eventlog </span>
<a name="l00316"></a>00316 <span class="comment"> */</span>
<a name="l00317"></a>00317 ProcessError (EVENTLOG_INFORMATION_TYPE, MsgErrorString, 0, quiet);
<a name="l00318"></a>00318 }
<a name="l00319"></a>00319
<a name="l00320"></a>00320 FINALLY
<a name="l00321"></a>00321 {
<a name="l00322"></a>00322 <span class="keywordflow">if</span> (hSCManager)
<a name="l00323"></a>00323 CloseServiceHandle (hSCManager);
<a name="l00324"></a>00324 <span class="keywordflow">if</span> (hService)
<a name="l00325"></a>00325 CloseServiceHandle (hService);
<a name="l00326"></a>00326 <span class="keywordflow">if</span> (hKey)
<a name="l00327"></a>00327 RegCloseKey (hKey);
<a name="l00328"></a>00328 <span class="keywordflow">if</span> (hParamKey)
<a name="l00329"></a>00329 RegCloseKey (hParamKey);
<a name="l00330"></a>00330 }
<a name="l00331"></a>00331 <span class="keywordflow">return</span> (exitStatus);
<a name="l00332"></a>00332 }
<a name="l00333"></a>00333
<a name="l00334"></a>00334 <span class="comment">/*</span>
<a name="l00335"></a>00335 <span class="comment"> * Unregister the service with the Windows SCM </span>
<a name="l00336"></a>00336 <span class="comment"> * Input - ServiceName</span>
<a name="l00337"></a>00337 <span class="comment"> */</span>
<a name="l00338"></a>00338 <span class="keywordtype">int</span>
<a name="l00339"></a>00339 UnregisterService (LPCSTR lpszServiceName, <span class="keywordtype">int</span> quiet)
<a name="l00340"></a>00340 {
<a name="l00341"></a>00341 TCHAR MsgErrorString[MAX_STR_SIZE]; <span class="comment">/* Message or Error string */</span>
<a name="l00342"></a>00342 SC_HANDLE hSCManager = NULL; <span class="comment">/* SCM handle */</span>
<a name="l00343"></a>00343 SC_HANDLE hService = NULL; <span class="comment">/* Service Handle */</span>
<a name="l00344"></a>00344 SERVICE_STATUS sStatus;
<a name="l00345"></a>00345 TCHAR szRegAppLogKey[] =
<a name="l00346"></a>00346 <span class="stringliteral">"SYSTEM\\CurrentControlSet\\Services\\EventLog\\Application\\"</span>;
<a name="l00347"></a>00347 TCHAR szRegKey[512];
<a name="l00348"></a>00348 <span class="keywordtype">int</span> exitStatus = 0;
<a name="l00349"></a>00349 <span class="comment">/* HKEY hKey = NULL; ?* Key to registry entry */</span>
<a name="l00350"></a>00350 TRY
<a name="l00351"></a>00351 {
<a name="l00352"></a>00352 <span class="comment">/*</span>
<a name="l00353"></a>00353 <span class="comment"> * Open Service Control Manager </span>
<a name="l00354"></a>00354 <span class="comment"> */</span>
<a name="l00355"></a>00355 hSCManager = OpenSCManager (NULL, NULL, SC_MANAGER_CREATE_SERVICE);
<a name="l00356"></a>00356 <span class="keywordflow">if</span> (hSCManager == NULL)
<a name="l00357"></a>00357 {
<a name="l00358"></a>00358 ProcessError (EVENTLOG_ERROR_TYPE, _T (<span class="stringliteral">"Can't open SCM (Service Control Manager)"</span>), 1, quiet);
<a name="l00359"></a>00359 exitStatus = SERVICE_ERROR_SCM_OPEN;
<a name="l00360"></a>00360 LEAVE;
<a name="l00361"></a>00361 }
<a name="l00362"></a>00362
<a name="l00363"></a>00363 <span class="comment">/*</span>
<a name="l00364"></a>00364 <span class="comment"> * Open registered service </span>
<a name="l00365"></a>00365 <span class="comment"> */</span>
<a name="l00366"></a>00366 hService = OpenService (hSCManager, lpszServiceName, SERVICE_ALL_ACCESS);
<a name="l00367"></a>00367 <span class="keywordflow">if</span> (hService == NULL)
<a name="l00368"></a>00368 {
<a name="l00369"></a>00369 _snprintf (MsgErrorString, <span class="keyword">sizeof</span>(MsgErrorString), <span class="stringliteral">"%s %s"</span>, _T (<span class="stringliteral">"Can't open service"</span>),
<a name="l00370"></a>00370 lpszServiceName);
<a name="l00371"></a>00371 ProcessError (EVENTLOG_ERROR_TYPE, MsgErrorString, 1, quiet);
<a name="l00372"></a>00372 exitStatus = SERVICE_ERROR_OPEN_SERVICE;
<a name="l00373"></a>00373 LEAVE;
<a name="l00374"></a>00374 }
<a name="l00375"></a>00375
<a name="l00376"></a>00376 <span class="comment">/*</span>
<a name="l00377"></a>00377 <span class="comment"> * Query service status </span>
<a name="l00378"></a>00378 <span class="comment"> * If running stop before deleting </span>
<a name="l00379"></a>00379 <span class="comment"> */</span>
<a name="l00380"></a>00380 <span class="keywordflow">if</span> (QueryServiceStatus (hService, &sStatus))
<a name="l00381"></a>00381 {
<a name="l00382"></a>00382 <span class="keywordflow">if</span> (sStatus.dwCurrentState == SERVICE_RUNNING
<a name="l00383"></a>00383 || sStatus.dwCurrentState == SERVICE_PAUSED)
<a name="l00384"></a>00384 {
<a name="l00385"></a>00385 ControlService (hService, SERVICE_CONTROL_STOP, &sStatus);
<a name="l00386"></a>00386 }
<a name="l00387"></a>00387 };
<a name="l00388"></a>00388
<a name="l00389"></a>00389 <span class="comment">/*</span>
<a name="l00390"></a>00390 <span class="comment"> * Delete the service </span>
<a name="l00391"></a>00391 <span class="comment"> */</span>
<a name="l00392"></a>00392 <span class="keywordflow">if</span> (DeleteService (hService) == FALSE)
<a name="l00393"></a>00393 {
<a name="l00394"></a>00394 _snprintf (MsgErrorString, <span class="keyword">sizeof</span>(MsgErrorString), <span class="stringliteral">"%s %s"</span>, _T (<span class="stringliteral">"Can't delete service"</span>),
<a name="l00395"></a>00395 lpszServiceName);
<a name="l00396"></a>00396
<a name="l00397"></a>00397 <span class="comment">/*</span>
<a name="l00398"></a>00398 <span class="comment"> * Log message to eventlog </span>
<a name="l00399"></a>00399 <span class="comment"> */</span>
<a name="l00400"></a>00400 ProcessError (EVENTLOG_ERROR_TYPE, MsgErrorString, 0, quiet);
<a name="l00401"></a>00401 LEAVE;
<a name="l00402"></a>00402 }
<a name="l00403"></a>00403
<a name="l00404"></a>00404 <span class="comment">/*</span>
<a name="l00405"></a>00405 <span class="comment"> * Log "Service deleted successfully " message to eventlog</span>
<a name="l00406"></a>00406 <span class="comment"> */</span>
<a name="l00407"></a>00407 _snprintf (MsgErrorString, <span class="keyword">sizeof</span>(MsgErrorString), <span class="stringliteral">"%s %s"</span>, lpszServiceName, _T (<span class="stringliteral">"service deleted"</span>));
<a name="l00408"></a>00408 ProcessError (EVENTLOG_INFORMATION_TYPE, MsgErrorString, 0, quiet);
<a name="l00409"></a>00409
<a name="l00410"></a>00410 <span class="comment">/*</span>
<a name="l00411"></a>00411 <span class="comment"> * Delete registry entries for EventLog </span>
<a name="l00412"></a>00412 <span class="comment"> */</span>
<a name="l00413"></a>00413 _tcscpy (szRegKey, szRegAppLogKey);
<a name="l00414"></a>00414 _tcscat (szRegKey, lpszServiceName);
<a name="l00415"></a>00415 RegDeleteKey (HKEY_LOCAL_MACHINE, szRegKey);
<a name="l00416"></a>00416 }
<a name="l00417"></a>00417
<a name="l00418"></a>00418 <span class="comment">/*</span>
<a name="l00419"></a>00419 <span class="comment"> * Delete the handles </span>
<a name="l00420"></a>00420 <span class="comment"> */</span>
<a name="l00421"></a>00421 FINALLY
<a name="l00422"></a>00422 {
<a name="l00423"></a>00423 <span class="keywordflow">if</span> (hService)
<a name="l00424"></a>00424 CloseServiceHandle (hService);
<a name="l00425"></a>00425 <span class="keywordflow">if</span> (hSCManager)
<a name="l00426"></a>00426 CloseServiceHandle (hSCManager);
<a name="l00427"></a>00427 }
<a name="l00428"></a>00428 <span class="keywordflow">return</span> (exitStatus);
<a name="l00429"></a>00429 }
<a name="l00430"></a>00430
<a name="l00431"></a>00431 <span class="comment">/*</span>
<a name="l00432"></a>00432 <span class="comment"> * To write message to Windows Event log</span>
<a name="l00433"></a>00433 <span class="comment"> * Input - Event Type, Message string</span>
<a name="l00434"></a>00434 <span class="comment"> */</span>
<a name="l00435"></a>00435 VOID
<a name="l00436"></a>00436 WriteToEventLog (WORD wType, LPCTSTR pszFormat, ...)
<a name="l00437"></a>00437 {
<a name="l00438"></a>00438 TCHAR szMessage[512];
<a name="l00439"></a>00439 LPTSTR LogStr[1];
<a name="l00440"></a>00440 va_list ArgList;
<a name="l00441"></a>00441 HANDLE hEventSource = NULL;
<a name="l00442"></a>00442 va_start (ArgList, pszFormat);
<a name="l00443"></a>00443 _vsnprintf (szMessage, <span class="keyword">sizeof</span>(szMessage), pszFormat, ArgList);
<a name="l00444"></a>00444 va_end (ArgList);
<a name="l00445"></a>00445 LogStr[0] = szMessage;
<a name="l00446"></a>00446 hEventSource = RegisterEventSource (NULL, app_name_long);
<a name="l00447"></a>00447 <span class="keywordflow">if</span> (hEventSource == NULL)
<a name="l00448"></a>00448 <span class="keywordflow">return</span>;
<a name="l00449"></a>00449 ReportEvent (hEventSource, wType, 0,
<a name="l00450"></a>00450 DISPLAY_MSG, <span class="comment">/* To Just output the text to event log */</span>
<a name="l00451"></a>00451 NULL, 1, 0, LogStr, NULL);
<a name="l00452"></a>00452 DeregisterEventSource (hEventSource);
<a name="l00453"></a>00453 }
<a name="l00454"></a>00454
<a name="l00455"></a>00455 <span class="comment">/*</span>
<a name="l00456"></a>00456 <span class="comment"> * Pre-process the second command-line argument from the user. </span>
<a name="l00457"></a>00457 <span class="comment"> * Service related options are:</span>
<a name="l00458"></a>00458 <span class="comment"> * -register - registers the service</span>
<a name="l00459"></a>00459 <span class="comment"> * -unregister - unregisters the service</span>
<a name="l00460"></a>00460 <span class="comment"> * -service - run as service</span>
<a name="l00461"></a>00461 <span class="comment"> * other command-line arguments are ignored here.</span>
<a name="l00462"></a>00462 <span class="comment"> *</span>
<a name="l00463"></a>00463 <span class="comment"> * Return: Type indicating the option specified</span>
<a name="l00464"></a>00464 <span class="comment"> */</span>
<a name="l00465"></a>00465 INT
<a name="l00466"></a>00466 ParseCmdLineForServiceOption (<span class="keywordtype">int</span> argc, TCHAR * argv[], <span class="keywordtype">int</span> *quiet)
<a name="l00467"></a>00467 {
<a name="l00468"></a>00468 <span class="keywordtype">int</span> nReturn = RUN_AS_CONSOLE; <span class="comment">/* Defualted to run as console */</span>
<a name="l00469"></a>00469
<a name="l00470"></a>00470 <span class="keywordflow">if</span> (argc >= 2)
<a name="l00471"></a>00471 {
<a name="l00472"></a>00472
<a name="l00473"></a>00473 <span class="comment">/*</span>
<a name="l00474"></a>00474 <span class="comment"> * second argument present </span>
<a name="l00475"></a>00475 <span class="comment"> */</span>
<a name="l00476"></a>00476 <span class="keywordflow">if</span> (lstrcmpi (_T (<span class="stringliteral">"-register"</span>), argv[1]) == 0)
<a name="l00477"></a>00477 {
<a name="l00478"></a>00478 nReturn = REGISTER_SERVICE;
<a name="l00479"></a>00479 }
<a name="l00480"></a>00480
<a name="l00481"></a>00481 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (lstrcmpi (_T (<span class="stringliteral">"-unregister"</span>), argv[1]) == 0)
<a name="l00482"></a>00482 {
<a name="l00483"></a>00483 nReturn = UN_REGISTER_SERVICE;
<a name="l00484"></a>00484 }
<a name="l00485"></a>00485
<a name="l00486"></a>00486 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (lstrcmpi (_T (<span class="stringliteral">"-service"</span>), argv[1]) == 0)
<a name="l00487"></a>00487 {
<a name="l00488"></a>00488 nReturn = RUN_AS_SERVICE;
<a name="l00489"></a>00489 }
<a name="l00490"></a>00490 }
<a name="l00491"></a>00491
<a name="l00492"></a>00492 <span class="keywordflow">if</span> (argc >= 3)
<a name="l00493"></a>00493 {
<a name="l00494"></a>00494 <span class="comment">/*</span>
<a name="l00495"></a>00495 <span class="comment"> * third argument present </span>
<a name="l00496"></a>00496 <span class="comment"> */</span>
<a name="l00497"></a>00497 <span class="keywordflow">if</span> (lstrcmpi (_T (<span class="stringliteral">"-quiet"</span>), argv[2]) == 0)
<a name="l00498"></a>00498 {
<a name="l00499"></a>00499 *quiet = 1;
<a name="l00500"></a>00500 }
<a name="l00501"></a>00501 }
<a name="l00502"></a>00502
<a name="l00503"></a>00503 <span class="keywordflow">return</span> nReturn;
<a name="l00504"></a>00504 }
<a name="l00505"></a>00505
<a name="l00506"></a>00506 <span class="comment">/*</span>
<a name="l00507"></a>00507 <span class="comment"> * Write error message to Event Log, console or pop-up window</span>
<a name="l00508"></a>00508 <span class="comment"> *</span>
<a name="l00509"></a>00509 <span class="comment"> * If useGetLastError is 1, the last error returned from GetLastError()</span>
<a name="l00510"></a>00510 <span class="comment"> * is appended to pszMessage, separated by a ": ".</span>
<a name="l00511"></a>00511 <span class="comment"> *</span>
<a name="l00512"></a>00512 <span class="comment"> * eventLogType: MessageBox equivalent:</span>
<a name="l00513"></a>00513 <span class="comment"> * </span>
<a name="l00514"></a>00514 <span class="comment"> * EVENTLOG_INFORMATION_TYPE MB_ICONASTERISK</span>
<a name="l00515"></a>00515 <span class="comment"> * EVENTLOG_WARNING_TYPE MB_ICONEXCLAMATION</span>
<a name="l00516"></a>00516 <span class="comment"> * EVENTLOG_ERROR_TYPE MB_ICONSTOP</span>
<a name="l00517"></a>00517 <span class="comment"> * </span>
<a name="l00518"></a>00518 <span class="comment"> */</span>
<a name="l00519"></a>00519 VOID
<a name="l00520"></a>00520 ProcessError (WORD eventLogType, LPCTSTR pszMessage, <span class="keywordtype">int</span> useGetLastError, <span class="keywordtype">int</span> quiet)
<a name="l00521"></a>00521 {
<a name="l00522"></a>00522 LPTSTR pErrorMsgTemp = NULL;
<a name="l00523"></a>00523 HANDLE hEventSource = NULL;
<a name="l00524"></a>00524 TCHAR pszMessageFull[MAX_STR_SIZE]; <span class="comment">/* Combined pszMessage and GetLastError */</span>
<a name="l00525"></a>00525
<a name="l00526"></a>00526 <span class="comment">/*</span>
<a name="l00527"></a>00527 <span class="comment"> * If useGetLastError enabled, generate text from GetLastError() and append to</span>
<a name="l00528"></a>00528 <span class="comment"> * pszMessageFull</span>
<a name="l00529"></a>00529 <span class="comment"> */</span>
<a name="l00530"></a>00530 <span class="keywordflow">if</span> (useGetLastError) {
<a name="l00531"></a>00531 FormatMessage (FORMAT_MESSAGE_ALLOCATE_BUFFER |
<a name="l00532"></a>00532 FORMAT_MESSAGE_FROM_SYSTEM, NULL, GetLastError (),
<a name="l00533"></a>00533 MAKELANGID (LANG_NEUTRAL, SUBLANG_DEFAULT),
<a name="l00534"></a>00534 (LPTSTR) & pErrorMsgTemp, 0, NULL);
<a name="l00535"></a>00535
<a name="l00536"></a>00536 _snprintf (pszMessageFull, <span class="keyword">sizeof</span>(pszMessageFull), <span class="stringliteral">"%s: %s"</span>, pszMessage, pErrorMsgTemp);
<a name="l00537"></a>00537 <span class="keywordflow">if</span> (pErrorMsgTemp) {
<a name="l00538"></a>00538 LocalFree (pErrorMsgTemp);
<a name="l00539"></a>00539 pErrorMsgTemp = NULL;
<a name="l00540"></a>00540 }
<a name="l00541"></a>00541 }
<a name="l00542"></a>00542 <span class="keywordflow">else</span> {
<a name="l00543"></a>00543 _snprintf (pszMessageFull, <span class="keyword">sizeof</span>(pszMessageFull), <span class="stringliteral">"%s"</span>, pszMessage);
<a name="l00544"></a>00544 }
<a name="l00545"></a>00545
<a name="l00546"></a>00546 hEventSource = RegisterEventSource (NULL, app_name_long);
<a name="l00547"></a>00547 <span class="keywordflow">if</span> (hEventSource != NULL) {
<a name="l00548"></a>00548 pErrorMsgTemp = pszMessageFull;
<a name="l00549"></a>00549
<a name="l00550"></a>00550 <span class="keywordflow">if</span> (ReportEvent (hEventSource,
<a name="l00551"></a>00551 eventLogType,
<a name="l00552"></a>00552 0,
<a name="l00553"></a>00553 DISPLAY_MSG, <span class="comment">/* To Just output the text to event log */</span>
<a name="l00554"></a>00554 NULL,
<a name="l00555"></a>00555 1,
<a name="l00556"></a>00556 0,
<a name="l00557"></a>00557 &pErrorMsgTemp,
<a name="l00558"></a>00558 NULL)) {
<a name="l00559"></a>00559 }
<a name="l00560"></a>00560 <span class="keywordflow">else</span> {
<a name="l00561"></a>00561 FormatMessage (FORMAT_MESSAGE_ALLOCATE_BUFFER |
<a name="l00562"></a>00562 FORMAT_MESSAGE_FROM_SYSTEM, NULL, GetLastError (),
<a name="l00563"></a>00563 MAKELANGID (LANG_NEUTRAL, SUBLANG_DEFAULT),
<a name="l00564"></a>00564 (LPTSTR) & pErrorMsgTemp, 0, NULL);
<a name="l00565"></a>00565
<a name="l00566"></a>00566 fprintf(stderr,<span class="stringliteral">"Could NOT lot to Event Log. Error returned from ReportEvent(): %s\n"</span>,pErrorMsgTemp);
<a name="l00567"></a>00567 <span class="keywordflow">if</span> (pErrorMsgTemp) {
<a name="l00568"></a>00568 LocalFree (pErrorMsgTemp);
<a name="l00569"></a>00569 pErrorMsgTemp = NULL;
<a name="l00570"></a>00570 }
<a name="l00571"></a>00571 }
<a name="l00572"></a>00572 DeregisterEventSource (hEventSource);
<a name="l00573"></a>00573 }
<a name="l00574"></a>00574
<a name="l00575"></a>00575 <span class="keywordflow">if</span> (quiet) {
<a name="l00576"></a>00576 fprintf(stderr,<span class="stringliteral">"%s\n"</span>,pszMessageFull);
<a name="l00577"></a>00577 }
<a name="l00578"></a>00578 <span class="keywordflow">else</span> {
<a name="l00579"></a>00579 <span class="keywordflow">switch</span> (eventLogType) {
<a name="l00580"></a>00580 <span class="keywordflow">case</span> EVENTLOG_INFORMATION_TYPE:
<a name="l00581"></a>00581 MessageBox (NULL, pszMessageFull, app_name_long, MB_ICONASTERISK);
<a name="l00582"></a>00582 <span class="keywordflow">break</span>;
<a name="l00583"></a>00583 <span class="keywordflow">case</span> EVENTLOG_WARNING_TYPE:
<a name="l00584"></a>00584 MessageBox (NULL, pszMessageFull, app_name_long, MB_ICONEXCLAMATION);
<a name="l00585"></a>00585 <span class="keywordflow">break</span>;
<a name="l00586"></a>00586 <span class="keywordflow">case</span> EVENTLOG_ERROR_TYPE:
<a name="l00587"></a>00587 MessageBox (NULL, pszMessageFull, app_name_long, MB_ICONSTOP);
<a name="l00588"></a>00588 <span class="keywordflow">break</span>;
<a name="l00589"></a>00589 <span class="keywordflow">default</span>:
<a name="l00590"></a>00590 MessageBox (NULL, pszMessageFull, app_name_long, EVENTLOG_WARNING_TYPE);
<a name="l00591"></a>00591 <span class="keywordflow">break</span>;
<a name="l00592"></a>00592 }
<a name="l00593"></a>00593 }
<a name="l00594"></a>00594
<a name="l00595"></a>00595 LocalFree (pErrorMsgTemp);
<a name="l00596"></a>00596 }
<a name="l00597"></a>00597
<a name="l00598"></a>00598 <span class="comment">/*</span>
<a name="l00599"></a>00599 <span class="comment"> * To update current service status </span>
<a name="l00600"></a>00600 <span class="comment"> * Sends the current service status to the SCM. Also updates</span>
<a name="l00601"></a>00601 <span class="comment"> * the global service status structure.</span>
<a name="l00602"></a>00602 <span class="comment"> */</span>
<a name="l00603"></a>00603 <span class="keyword">static</span> BOOL
<a name="l00604"></a>00604 UpdateServiceStatus (DWORD dwStatus, DWORD dwErrorCode, DWORD dwWaitHint)
<a name="l00605"></a>00605 {
<a name="l00606"></a>00606 DWORD <span class="keyword">static</span> dwCheckpoint = 1;
<a name="l00607"></a>00607 DWORD dwControls = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_PAUSE_CONTINUE;
<a name="l00608"></a>00608 <span class="keywordflow">if</span> (g_fRunningAsService == FALSE)
<a name="l00609"></a>00609 <span class="keywordflow">return</span> FALSE;
<a name="l00610"></a>00610 ZeroMemory (&ServiceStatus, <span class="keyword">sizeof</span> (ServiceStatus));
<a name="l00611"></a>00611 ServiceStatus.dwServiceType = SERVICE_WIN32;
<a name="l00612"></a>00612 ServiceStatus.dwCurrentState = dwStatus;
<a name="l00613"></a>00613 ServiceStatus.dwWaitHint = dwWaitHint;
<a name="l00614"></a>00614 <span class="keywordflow">if</span> (dwErrorCode)
<a name="l00615"></a>00615 {
<a name="l00616"></a>00616 ServiceStatus.dwWin32ExitCode = ERROR_SERVICE_SPECIFIC_ERROR;
<a name="l00617"></a>00617 ServiceStatus.dwServiceSpecificExitCode = dwErrorCode;
<a name="l00618"></a>00618 }
<a name="l00619"></a>00619
<a name="l00620"></a>00620 <span class="comment">/*</span>
<a name="l00621"></a>00621 <span class="comment"> * special cases that depend on the new state </span>
<a name="l00622"></a>00622 <span class="comment"> */</span>
<a name="l00623"></a>00623 <span class="keywordflow">switch</span> (dwStatus)
<a name="l00624"></a>00624 {
<a name="l00625"></a>00625 <span class="keywordflow">case</span> SERVICE_START_PENDING:
<a name="l00626"></a>00626 dwControls = 0;
<a name="l00627"></a>00627 <span class="keywordflow">break</span>;
<a name="l00628"></a>00628 <span class="keywordflow">case</span> SERVICE_RUNNING:
<a name="l00629"></a>00629 <span class="keywordflow">case</span> SERVICE_STOPPED:
<a name="l00630"></a>00630 dwCheckpoint = 0;
<a name="l00631"></a>00631 <span class="keywordflow">break</span>;
<a name="l00632"></a>00632 }
<a name="l00633"></a>00633 ServiceStatus.dwCheckPoint = dwCheckpoint++;
<a name="l00634"></a>00634 ServiceStatus.dwControlsAccepted = dwControls;
<a name="l00635"></a>00635 <span class="keywordflow">return</span> ReportCurrentServiceStatus ();
<a name="l00636"></a>00636 }
<a name="l00637"></a>00637
<a name="l00638"></a>00638 <span class="comment">/*</span>
<a name="l00639"></a>00639 <span class="comment"> * Reports current Service status to SCM</span>
<a name="l00640"></a>00640 <span class="comment"> */</span>
<a name="l00641"></a>00641 <span class="keyword">static</span> BOOL
<a name="l00642"></a>00642 ReportCurrentServiceStatus ()
<a name="l00643"></a>00643 {
<a name="l00644"></a>00644 <span class="keywordflow">return</span> SetServiceStatus (hServiceStatus, &ServiceStatus);
<a name="l00645"></a>00645 }
<a name="l00646"></a>00646
<a name="l00647"></a>00647 <span class="comment">/*</span>
<a name="l00648"></a>00648 <span class="comment"> * The ServiceMain function to start service.</span>
<a name="l00649"></a>00649 <span class="comment"> */</span>
<a name="l00650"></a>00650 VOID WINAPI
<a name="l00651"></a>00651 ServiceMain (DWORD argc, LPTSTR argv[])
<a name="l00652"></a>00652 {
<a name="l00653"></a>00653 SECURITY_ATTRIBUTES SecurityAttributes;
<a name="l00654"></a>00654 DWORD dwThreadId;
<a name="l00655"></a>00655
<a name="l00656"></a>00656 <span class="comment">/*</span>
<a name="l00657"></a>00657 <span class="comment"> * Input Arguments to function startup </span>
<a name="l00658"></a>00658 <span class="comment"> */</span>
<a name="l00659"></a>00659 DWORD ArgCount = 0;
<a name="l00660"></a>00660 LPTSTR *ArgArray = NULL;
<a name="l00661"></a>00661 TCHAR szRegKey[512];
<a name="l00662"></a>00662 TCHAR szValue[128];
<a name="l00663"></a>00663 DWORD nSize;
<a name="l00664"></a>00664 HKEY hParamKey = NULL; <span class="comment">/* To read startup parameters */</span>
<a name="l00665"></a>00665 DWORD TotalParams = 0;
<a name="l00666"></a>00666 DWORD i;
<a name="l00667"></a>00667 <a class="code" href="struct__InputParams.html">InputParams</a> ThreadInputParams;
<a name="l00668"></a>00668
<a name="l00669"></a>00669 <span class="comment">/*</span>
<a name="l00670"></a>00670 <span class="comment"> * Build the Input parameters to pass to worker thread </span>
<a name="l00671"></a>00671 <span class="comment"> */</span>
<a name="l00672"></a>00672
<a name="l00673"></a>00673 <span class="comment">/*</span>
<a name="l00674"></a>00674 <span class="comment"> * SCM sends Service Name as first arg, increment to point</span>
<a name="l00675"></a>00675 <span class="comment"> * arguments user specified while starting contorl agent</span>
<a name="l00676"></a>00676 <span class="comment"> */</span>
<a name="l00677"></a>00677
<a name="l00678"></a>00678 <span class="comment">/*</span>
<a name="l00679"></a>00679 <span class="comment"> * Read registry parameter </span>
<a name="l00680"></a>00680 <span class="comment"> */</span>
<a name="l00681"></a>00681 ArgCount = 1;
<a name="l00682"></a>00682
<a name="l00683"></a>00683 <span class="comment">/*</span>
<a name="l00684"></a>00684 <span class="comment"> * Create Registry Key path </span>
<a name="l00685"></a>00685 <span class="comment"> */</span>
<a name="l00686"></a>00686 _snprintf (szRegKey, <span class="keyword">sizeof</span>(szRegKey), <span class="stringliteral">"%s%s\\%s"</span>,
<a name="l00687"></a>00687 _T (<span class="stringliteral">"SYSTEM\\CurrentControlSet\\Services\\"</span>), app_name_long,
<a name="l00688"></a>00688 <span class="stringliteral">"Parameters"</span>);
<a name="l00689"></a>00689 <span class="keywordflow">if</span> (RegOpenKeyEx
<a name="l00690"></a>00690 (HKEY_LOCAL_MACHINE, szRegKey, 0, KEY_ALL_ACCESS, &hParamKey) == ERROR_SUCCESS)
<a name="l00691"></a>00691 {
<a name="l00692"></a>00692
<a name="l00693"></a>00693 <span class="comment">/*</span>
<a name="l00694"></a>00694 <span class="comment"> * Read startup Configuration information </span>
<a name="l00695"></a>00695 <span class="comment"> */</span>
<a name="l00696"></a>00696 <span class="comment">/*</span>
<a name="l00697"></a>00697 <span class="comment"> * Find number of subkeys inside parameters </span>
<a name="l00698"></a>00698 <span class="comment"> */</span>
<a name="l00699"></a>00699 <span class="keywordflow">if</span> (RegQueryInfoKey (hParamKey, NULL, NULL, 0,
<a name="l00700"></a>00700 NULL, NULL, NULL, &TotalParams,
<a name="l00701"></a>00701 NULL, NULL, NULL, NULL) == ERROR_SUCCESS)
<a name="l00702"></a>00702 {
<a name="l00703"></a>00703 <span class="keywordflow">if</span> (TotalParams != 0)
<a name="l00704"></a>00704 {
<a name="l00705"></a>00705 ArgCount += TotalParams;
<a name="l00706"></a>00706
<a name="l00707"></a>00707 <span class="comment">/*</span>
<a name="l00708"></a>00708 <span class="comment"> * Allocate memory to hold strings </span>
<a name="l00709"></a>00709 <span class="comment"> */</span>
<a name="l00710"></a>00710 ArgArray = (LPTSTR *) malloc (<span class="keyword">sizeof</span> (LPTSTR) * ArgCount);
<a name="l00711"></a>00711 <span class="keywordflow">if</span> (ArgArray == 0)
<a name="l00712"></a>00712 {
<a name="l00713"></a>00713 WriteToEventLog (EVENTLOG_ERROR_TYPE,
<a name="l00714"></a>00714 _T (<span class="stringliteral">"Resource failure"</span>));
<a name="l00715"></a>00715 <span class="keywordflow">return</span>;
<a name="l00716"></a>00716 }
<a name="l00717"></a>00717
<a name="l00718"></a>00718 <span class="comment">/*</span>
<a name="l00719"></a>00719 <span class="comment"> * Copy first argument </span>
<a name="l00720"></a>00720 <span class="comment"> */</span>
<a name="l00721"></a>00721 ArgArray[0] = _tcsdup (argv[0]);
<a name="l00722"></a>00722 <span class="keywordflow">for</span> (i = 1; i <= TotalParams; i++)
<a name="l00723"></a>00723 {
<a name="l00724"></a>00724
<a name="l00725"></a>00725 <span class="comment">/*</span>
<a name="l00726"></a>00726 <span class="comment"> * Create Subkey value name </span>
<a name="l00727"></a>00727 <span class="comment"> */</span>
<a name="l00728"></a>00728 _snprintf (szRegKey, <span class="keyword">sizeof</span>(szRegKey), <span class="stringliteral">"%s%d"</span>, <span class="stringliteral">"Param"</span>, i);
<a name="l00729"></a>00729
<a name="l00730"></a>00730 <span class="comment">/*</span>
<a name="l00731"></a>00731 <span class="comment"> * Set size </span>
<a name="l00732"></a>00732 <span class="comment"> */</span>
<a name="l00733"></a>00733 nSize = 128;
<a name="l00734"></a>00734 RegQueryValueEx (hParamKey, szRegKey, 0, NULL,
<a name="l00735"></a>00735 (LPBYTE) & szValue, &nSize);
<a name="l00736"></a>00736 ArgArray[i] = _tcsdup (szValue);
<a name="l00737"></a>00737 }
<a name="l00738"></a>00738 }
<a name="l00739"></a>00739 }
<a name="l00740"></a>00740 RegCloseKey (hParamKey);
<a name="l00741"></a>00741 }
<a name="l00742"></a>00742 <span class="keywordflow">if</span> (ArgCount == 1)
<a name="l00743"></a>00743 {
<a name="l00744"></a>00744
<a name="l00745"></a>00745 <span class="comment">/*</span>
<a name="l00746"></a>00746 <span class="comment"> * No statup agrs are given </span>
<a name="l00747"></a>00747 <span class="comment"> */</span>
<a name="l00748"></a>00748 ThreadInputParams.Argc = argc;
<a name="l00749"></a>00749 ThreadInputParams.Argv = argv;
<a name="l00750"></a>00750 }
<a name="l00751"></a>00751
<a name="l00752"></a>00752 <span class="keywordflow">else</span>
<a name="l00753"></a>00753 {
<a name="l00754"></a>00754 ThreadInputParams.Argc = ArgCount;
<a name="l00755"></a>00755 ThreadInputParams.Argv = ArgArray;
<a name="l00756"></a>00756 }
<a name="l00757"></a>00757
<a name="l00758"></a>00758 <span class="comment">/*</span>
<a name="l00759"></a>00759 <span class="comment"> * Register Service Control Handler </span>
<a name="l00760"></a>00760 <span class="comment"> */</span>
<a name="l00761"></a>00761 hServiceStatus = RegisterServiceCtrlHandler (app_name_long, ControlHandler);
<a name="l00762"></a>00762 <span class="keywordflow">if</span> (hServiceStatus == 0)
<a name="l00763"></a>00763 {
<a name="l00764"></a>00764 WriteToEventLog (EVENTLOG_ERROR_TYPE,
<a name="l00765"></a>00765 _T (<span class="stringliteral">"RegisterServiceCtrlHandler failed"</span>));
<a name="l00766"></a>00766 <span class="keywordflow">return</span>;
<a name="l00767"></a>00767 }
<a name="l00768"></a>00768
<a name="l00769"></a>00769 <span class="comment">/*</span>
<a name="l00770"></a>00770 <span class="comment"> * Update the service status to START_PENDING </span>
<a name="l00771"></a>00771 <span class="comment"> */</span>
<a name="l00772"></a>00772 UpdateServiceStatus (SERVICE_START_PENDING, NO_ERROR, SCM_WAIT_INTERVAL);
<a name="l00773"></a>00773
<a name="l00774"></a>00774 <span class="comment">/*</span>
<a name="l00775"></a>00775 <span class="comment"> * Spin of worker thread, which does majority of the work </span>
<a name="l00776"></a>00776 <span class="comment"> */</span>
<a name="l00777"></a>00777 TRY
<a name="l00778"></a>00778 {
<a name="l00779"></a>00779 <span class="keywordflow">if</span> (SetSimpleSecurityAttributes (&SecurityAttributes) == FALSE)
<a name="l00780"></a>00780 {
<a name="l00781"></a>00781 WriteToEventLog (EVENTLOG_ERROR_TYPE,
<a name="l00782"></a>00782 _T (<span class="stringliteral">"Couldn't init security attributes"</span>));
<a name="l00783"></a>00783 LEAVE;
<a name="l00784"></a>00784 }
<a name="l00785"></a>00785 hServiceThread =
<a name="l00786"></a>00786 (<span class="keywordtype">void</span> *) _beginthreadex (&SecurityAttributes, 0,
<a name="l00787"></a>00787 ThreadFunction,
<a name="l00788"></a>00788 (<span class="keywordtype">void</span> *) &ThreadInputParams, 0, &dwThreadId);
<a name="l00789"></a>00789 <span class="keywordflow">if</span> (hServiceThread == NULL)
<a name="l00790"></a>00790 {
<a name="l00791"></a>00791 WriteToEventLog (EVENTLOG_ERROR_TYPE, _T (<span class="stringliteral">"Couldn't start worker thread"</span>));
<a name="l00792"></a>00792 LEAVE;
<a name="l00793"></a>00793 }
<a name="l00794"></a>00794
<a name="l00795"></a>00795 <span class="comment">/*</span>
<a name="l00796"></a>00796 <span class="comment"> * Set Service Status to Running </span>
<a name="l00797"></a>00797 <span class="comment"> */</span>
<a name="l00798"></a>00798 UpdateServiceStatus (SERVICE_RUNNING, NO_ERROR, SCM_WAIT_INTERVAL);
<a name="l00799"></a>00799
<a name="l00800"></a>00800 <span class="comment">/*</span>
<a name="l00801"></a>00801 <span class="comment"> * Wait for termination event and worker thread to</span>
<a name="l00802"></a>00802 <span class="comment"> * * spin down.</span>
<a name="l00803"></a>00803 <span class="comment"> */</span>
<a name="l00804"></a>00804 WaitForSingleObject (hServiceThread, INFINITE);
<a name="l00805"></a>00805 }
<a name="l00806"></a>00806 FINALLY
<a name="l00807"></a>00807 {
<a name="l00808"></a>00808 <span class="comment">/*</span>
<a name="l00809"></a>00809 <span class="comment"> * Release resources </span>
<a name="l00810"></a>00810 <span class="comment"> */</span>
<a name="l00811"></a>00811 UpdateServiceStatus (SERVICE_STOPPED, NO_ERROR, SCM_WAIT_INTERVAL);
<a name="l00812"></a>00812 <span class="keywordflow">if</span> (hServiceThread)
<a name="l00813"></a>00813 CloseHandle (hServiceThread);
<a name="l00814"></a>00814 FreeSecurityAttributes (&SecurityAttributes);
<a name="l00815"></a>00815
<a name="l00816"></a>00816 <span class="comment">/*</span>
<a name="l00817"></a>00817 <span class="comment"> * Delete allocated argument list </span>
<a name="l00818"></a>00818 <span class="comment"> */</span>
<a name="l00819"></a>00819 <span class="keywordflow">if</span> (ArgCount > 1 && ArgArray != NULL)
<a name="l00820"></a>00820 {
<a name="l00821"></a>00821 <span class="comment">/*</span>
<a name="l00822"></a>00822 <span class="comment"> * Delete all strings </span>
<a name="l00823"></a>00823 <span class="comment"> */</span>
<a name="l00824"></a>00824 <span class="keywordflow">for</span> (i = 0; i < ArgCount; i++)
<a name="l00825"></a>00825 {
<a name="l00826"></a>00826 free (ArgArray[i]);
<a name="l00827"></a>00827 }
<a name="l00828"></a>00828 free (ArgArray);
<a name="l00829"></a>00829 }
<a name="l00830"></a>00830 }
<a name="l00831"></a>00831 }
<a name="l00832"></a>00832
<a name="l00833"></a>00833 <span class="comment">/*</span>
<a name="l00834"></a>00834 <span class="comment"> * Function to start as Windows service</span>
<a name="l00835"></a>00835 <span class="comment"> * The calling party should specify their entry point as input parameter</span>
<a name="l00836"></a>00836 <span class="comment"> * Returns TRUE if the Service is started successfully</span>
<a name="l00837"></a>00837 <span class="comment"> */</span>
<a name="l00838"></a>00838 BOOL
<a name="l00839"></a>00839 RunAsService (INT (*ServiceFunction) (INT, LPTSTR *))
<a name="l00840"></a>00840 {
<a name="l00841"></a>00841
<a name="l00842"></a>00842 <span class="comment">/*</span>
<a name="l00843"></a>00843 <span class="comment"> * Set the ServiceEntryPoint </span>
<a name="l00844"></a>00844 <span class="comment"> */</span>
<a name="l00845"></a>00845 ServiceEntryPoint = ServiceFunction;
<a name="l00846"></a>00846
<a name="l00847"></a>00847 <span class="comment">/*</span>
<a name="l00848"></a>00848 <span class="comment"> * By default, mark as Running as a service </span>
<a name="l00849"></a>00849 <span class="comment"> */</span>
<a name="l00850"></a>00850 g_fRunningAsService = TRUE;
<a name="l00851"></a>00851
<a name="l00852"></a>00852 <span class="comment">/*</span>
<a name="l00853"></a>00853 <span class="comment"> * Initialize ServiceTableEntry table </span>
<a name="l00854"></a>00854 <span class="comment"> */</span>
<a name="l00855"></a>00855 ServiceTableEntry[0].lpServiceName = app_name_long; <span class="comment">/* Application Name */</span>
<a name="l00856"></a>00856
<a name="l00857"></a>00857 <span class="comment">/*</span>
<a name="l00858"></a>00858 <span class="comment"> * Call SCM via StartServiceCtrlDispatcher to run as Service </span>
<a name="l00859"></a>00859 <span class="comment"> * * If the function returns TRUE we are running as Service, </span>
<a name="l00860"></a>00860 <span class="comment"> */</span>
<a name="l00861"></a>00861 <span class="keywordflow">if</span> (StartServiceCtrlDispatcher (ServiceTableEntry) == FALSE)
<a name="l00862"></a>00862 {
<a name="l00863"></a>00863 g_fRunningAsService = FALSE;
<a name="l00864"></a>00864
<a name="l00865"></a>00865 <span class="comment">/*</span>
<a name="l00866"></a>00866 <span class="comment"> * Some other error has occurred. </span>
<a name="l00867"></a>00867 <span class="comment"> */</span>
<a name="l00868"></a>00868 WriteToEventLog (EVENTLOG_ERROR_TYPE,
<a name="l00869"></a>00869 _T (<span class="stringliteral">"Couldn't start service - %s"</span>), app_name_long);
<a name="l00870"></a>00870 }
<a name="l00871"></a>00871 <span class="keywordflow">return</span> g_fRunningAsService;
<a name="l00872"></a>00872 }
<a name="l00873"></a>00873
<a name="l00874"></a>00874 <span class="comment">/*</span>
<a name="l00875"></a>00875 <span class="comment"> * Service control handler function</span>
<a name="l00876"></a>00876 <span class="comment"> * Responds to SCM commands/requests</span>
<a name="l00877"></a>00877 <span class="comment"> * This service handles 4 commands</span>
<a name="l00878"></a>00878 <span class="comment"> * - interrogate, pause, continue and stop.</span>
<a name="l00879"></a>00879 <span class="comment"> */</span>
<a name="l00880"></a>00880 VOID WINAPI
<a name="l00881"></a>00881 ControlHandler (DWORD dwControl)
<a name="l00882"></a>00882 {
<a name="l00883"></a>00883 <span class="keywordflow">switch</span> (dwControl)
<a name="l00884"></a>00884 {
<a name="l00885"></a>00885 <span class="keywordflow">case</span> SERVICE_CONTROL_INTERROGATE:
<a name="l00886"></a>00886 ProcessServiceInterrogate ();
<a name="l00887"></a>00887 <span class="keywordflow">break</span>;
<a name="l00888"></a>00888
<a name="l00889"></a>00889 <span class="keywordflow">case</span> SERVICE_CONTROL_PAUSE:
<a name="l00890"></a>00890 ProcessServicePause ();
<a name="l00891"></a>00891 <span class="keywordflow">break</span>;
<a name="l00892"></a>00892
<a name="l00893"></a>00893 <span class="keywordflow">case</span> SERVICE_CONTROL_CONTINUE:
<a name="l00894"></a>00894 ProcessServiceContinue ();
<a name="l00895"></a>00895 <span class="keywordflow">break</span>;
<a name="l00896"></a>00896
<a name="l00897"></a>00897 <span class="keywordflow">case</span> SERVICE_CONTROL_STOP:
<a name="l00898"></a>00898 ProcessServiceStop ();
<a name="l00899"></a>00899 <span class="keywordflow">break</span>;
<a name="l00900"></a>00900 }
<a name="l00901"></a>00901 }
<a name="l00902"></a>00902
<a name="l00903"></a>00903 <span class="comment">/*</span>
<a name="l00904"></a>00904 <span class="comment"> * To stop the service.</span>
<a name="l00905"></a>00905 <span class="comment"> * If a stop function was registered, invoke it,</span>
<a name="l00906"></a>00906 <span class="comment"> * otherwise terminate the worker thread.</span>
<a name="l00907"></a>00907 <span class="comment"> * After stopping, Service status is set to STOP in </span>
<a name="l00908"></a>00908 <span class="comment"> * main loop</span>
<a name="l00909"></a>00909 <span class="comment"> */</span>
<a name="l00910"></a>00910 VOID
<a name="l00911"></a>00911 ProcessServiceStop (VOID)
<a name="l00912"></a>00912 {
<a name="l00913"></a>00913 UpdateServiceStatus (SERVICE_STOP_PENDING, NO_ERROR, SCM_WAIT_INTERVAL);
<a name="l00914"></a>00914
<a name="l00915"></a>00915 <span class="keywordflow">if</span> (StopFunction != NULL)
<a name="l00916"></a>00916 {
<a name="l00917"></a>00917 (*StopFunction) ();
<a name="l00918"></a>00918 }
<a name="l00919"></a>00919
<a name="l00920"></a>00920 <span class="keywordflow">else</span>
<a name="l00921"></a>00921 {
<a name="l00922"></a>00922 TerminateThread (hServiceThread, 0);
<a name="l00923"></a>00923 }
<a name="l00924"></a>00924 }
<a name="l00925"></a>00925
<a name="l00926"></a>00926 <span class="comment">/*</span>
<a name="l00927"></a>00927 <span class="comment"> * Returns the current state of the service to the SCM.</span>
<a name="l00928"></a>00928 <span class="comment"> */</span>
<a name="l00929"></a>00929 VOID
<a name="l00930"></a>00930 ProcessServiceInterrogate (VOID)
<a name="l00931"></a>00931 {
<a name="l00932"></a>00932 ReportCurrentServiceStatus ();
<a name="l00933"></a>00933 }
<a name="l00934"></a>00934
<a name="l00935"></a>00935 <span class="comment">/*</span>
<a name="l00936"></a>00936 <span class="comment"> * To Create a security descriptor with a NULL ACL, which</span>
<a name="l00937"></a>00937 <span class="comment"> * allows unlimited access. Returns a SECURITY_ATTRIBUTES</span>
<a name="l00938"></a>00938 <span class="comment"> * structure that contains the security descriptor.</span>
<a name="l00939"></a>00939 <span class="comment"> * The structure contains a dynamically allocated security</span>
<a name="l00940"></a>00940 <span class="comment"> * descriptor that must be freed either manually, or by</span>
<a name="l00941"></a>00941 <span class="comment"> * calling FreeSecurityAttributes </span>
<a name="l00942"></a>00942 <span class="comment"> */</span>
<a name="l00943"></a>00943 BOOL
<a name="l00944"></a>00944 SetSimpleSecurityAttributes (SECURITY_ATTRIBUTES * pSecurityAttr)
<a name="l00945"></a>00945 {
<a name="l00946"></a>00946 BOOL fReturn = FALSE;
<a name="l00947"></a>00947 SECURITY_DESCRIPTOR *pSecurityDesc = NULL;
<a name="l00948"></a>00948
<a name="l00949"></a>00949 <span class="comment">/*</span>
<a name="l00950"></a>00950 <span class="comment"> * If an invalid address is passed as a parameter, return</span>
<a name="l00951"></a>00951 <span class="comment"> * FALSE right away. </span>
<a name="l00952"></a>00952 <span class="comment"> */</span>
<a name="l00953"></a>00953 <span class="keywordflow">if</span> (!pSecurityAttr)
<a name="l00954"></a>00954 <span class="keywordflow">return</span> FALSE;
<a name="l00955"></a>00955 pSecurityDesc =
<a name="l00956"></a>00956 (SECURITY_DESCRIPTOR *) LocalAlloc (LPTR, SECURITY_DESCRIPTOR_MIN_LENGTH);
<a name="l00957"></a>00957 <span class="keywordflow">if</span> (!pSecurityDesc)
<a name="l00958"></a>00958 <span class="keywordflow">return</span> FALSE;
<a name="l00959"></a>00959 fReturn =
<a name="l00960"></a>00960 InitializeSecurityDescriptor (pSecurityDesc, SECURITY_DESCRIPTOR_REVISION);
<a name="l00961"></a>00961 <span class="keywordflow">if</span> (fReturn != FALSE)
<a name="l00962"></a>00962 {
<a name="l00963"></a>00963 fReturn = SetSecurityDescriptorDacl (pSecurityDesc, TRUE, NULL, FALSE);
<a name="l00964"></a>00964 }
<a name="l00965"></a>00965 <span class="keywordflow">if</span> (fReturn != FALSE)
<a name="l00966"></a>00966 {
<a name="l00967"></a>00967 pSecurityAttr->nLength = <span class="keyword">sizeof</span> (SECURITY_ATTRIBUTES);
<a name="l00968"></a>00968 pSecurityAttr->lpSecurityDescriptor = pSecurityDesc;
<a name="l00969"></a>00969 pSecurityAttr->bInheritHandle = TRUE;
<a name="l00970"></a>00970 }
<a name="l00971"></a>00971
<a name="l00972"></a>00972 <span class="keywordflow">else</span>
<a name="l00973"></a>00973 {
<a name="l00974"></a>00974 <span class="comment">/*</span>
<a name="l00975"></a>00975 <span class="comment"> * Couldn't initialize or set security descriptor. </span>
<a name="l00976"></a>00976 <span class="comment"> */</span>
<a name="l00977"></a>00977 LocalFree (pSecurityDesc);
<a name="l00978"></a>00978 }
<a name="l00979"></a>00979 <span class="keywordflow">return</span> fReturn;
<a name="l00980"></a>00980 }
<a name="l00981"></a>00981
<a name="l00982"></a>00982 <span class="comment">/*</span>
<a name="l00983"></a>00983 <span class="comment"> * This function Frees the security descriptor, if any was created.</span>
<a name="l00984"></a>00984 <span class="comment"> */</span>
<a name="l00985"></a>00985 VOID
<a name="l00986"></a>00986 FreeSecurityAttributes (SECURITY_ATTRIBUTES * pSecurityAttr)
<a name="l00987"></a>00987 {
<a name="l00988"></a>00988 <span class="keywordflow">if</span> (pSecurityAttr && pSecurityAttr->lpSecurityDescriptor)
<a name="l00989"></a>00989 LocalFree (pSecurityAttr->lpSecurityDescriptor);
<a name="l00990"></a>00990 }
<a name="l00991"></a>00991
<a name="l00992"></a>00992 <span class="comment">/*</span>
<a name="l00993"></a>00993 <span class="comment"> * This function runs in the worker thread</span>
<a name="l00994"></a>00994 <span class="comment"> * until an exit is forced, or until the SCM issues the STOP command.</span>
<a name="l00995"></a>00995 <span class="comment"> * Invokes registered service function</span>
<a name="l00996"></a>00996 <span class="comment"> * Returns when called registered function returns</span>
<a name="l00997"></a>00997 <span class="comment"> *</span>
<a name="l00998"></a>00998 <span class="comment"> * Input:</span>
<a name="l00999"></a>00999 <span class="comment"> * lpParam contains argc and argv, pass to service main function </span>
<a name="l01000"></a>01000 <span class="comment"> */</span>
<a name="l01001"></a>01001 DWORD WINAPI
<a name="l01002"></a>01002 ThreadFunction (LPVOID lpParam)
<a name="l01003"></a>01003 {
<a name="l01004"></a>01004 <a class="code" href="struct__InputParams.html">InputParams</a> * pInputArg = (<a class="code" href="struct__InputParams.html">InputParams</a> *) lpParam;
<a name="l01005"></a>01005 <span class="keywordflow">return</span> (*ServiceEntryPoint) (pInputArg->Argc, pInputArg->Argv);
<a name="l01006"></a>01006 }
<a name="l01007"></a>01007
<a name="l01008"></a>01008 <span class="comment">/*</span>
<a name="l01009"></a>01009 <span class="comment"> * This function is called to register an application-specific function</span>
<a name="l01010"></a>01010 <span class="comment"> * which is invoked when the SCM stops the worker thread.</span>
<a name="l01011"></a>01011 <span class="comment"> */</span>
<a name="l01012"></a>01012 VOID
<a name="l01013"></a>01013 RegisterStopFunction (VOID (*StopFunc) (VOID))
<a name="l01014"></a>01014 {
<a name="l01015"></a>01015 StopFunction = StopFunc;
<a name="l01016"></a>01016 }
<a name="l01017"></a>01017
<a name="l01018"></a>01018 <span class="comment">/*</span>
<a name="l01019"></a>01019 <span class="comment"> * SCM pause command invokes this function</span>
<a name="l01020"></a>01020 <span class="comment"> * If the service is not running, this function does nothing.</span>
<a name="l01021"></a>01021 <span class="comment"> * Otherwise, suspend the worker thread and update the status.</span>
<a name="l01022"></a>01022 <span class="comment"> */</span>
<a name="l01023"></a>01023 VOID
<a name="l01024"></a>01024 ProcessServicePause (VOID)
<a name="l01025"></a>01025 {
<a name="l01026"></a>01026 <span class="keywordflow">if</span> (ServiceStatus.dwCurrentState == SERVICE_RUNNING)
<a name="l01027"></a>01027 {
<a name="l01028"></a>01028 UpdateServiceStatus (SERVICE_PAUSE_PENDING, NO_ERROR, SCM_WAIT_INTERVAL);
<a name="l01029"></a>01029
<a name="l01030"></a>01030 <span class="keywordflow">if</span> (SuspendThread (hServiceThread) != -1)
<a name="l01031"></a>01031 {
<a name="l01032"></a>01032 UpdateServiceStatus (SERVICE_PAUSED, NO_ERROR, SCM_WAIT_INTERVAL);
<a name="l01033"></a>01033 }
<a name="l01034"></a>01034 }
<a name="l01035"></a>01035 }
<a name="l01036"></a>01036
<a name="l01037"></a>01037 <span class="comment">/*</span>
<a name="l01038"></a>01038 <span class="comment"> * SCM resume command invokes this function</span>
<a name="l01039"></a>01039 <span class="comment"> * If the service is not paused, this function does nothing.</span>
<a name="l01040"></a>01040 <span class="comment"> * Otherwise, resume the worker thread and update the status.</span>
<a name="l01041"></a>01041 <span class="comment"> */</span>
<a name="l01042"></a>01042 VOID
<a name="l01043"></a>01043 ProcessServiceContinue (VOID)
<a name="l01044"></a>01044 {
<a name="l01045"></a>01045 <span class="keywordflow">if</span> (ServiceStatus.dwCurrentState == SERVICE_PAUSED)
<a name="l01046"></a>01046 {
<a name="l01047"></a>01047 UpdateServiceStatus (SERVICE_CONTINUE_PENDING, NO_ERROR, SCM_WAIT_INTERVAL);
<a name="l01048"></a>01048
<a name="l01049"></a>01049 <span class="keywordflow">if</span> (ResumeThread (hServiceThread) != -1)
<a name="l01050"></a>01050 {
<a name="l01051"></a>01051 UpdateServiceStatus (SERVICE_RUNNING, NO_ERROR, SCM_WAIT_INTERVAL);
<a name="l01052"></a>01052 }
<a name="l01053"></a>01053 }
<a name="l01054"></a>01054 }
<a name="l01055"></a>01055
<a name="l01056"></a>01056 <span class="preprocessor">#endif </span><span class="comment">/* WIN32 */</span>
<a name="l01057"></a>01057
<a name="l01058"></a>01058
</pre></div></div><!-- contents -->
<hr class="footer"/><address class="footer"><small>
Generated by  <a href="http://www.doxygen.org/index.html">
<img class="footer" src="doxygen.png" alt="doxygen"/>
</a> 1.7.6.1
</small></address>
</body>
</html>