nt_servc.cc 15.7 KB
Newer Older
unknown's avatar
unknown committed
1 2 3 4 5 6 7
/* ------------------------------------------------------------------------
   Windows NT Service class library
   Copyright Abandoned 1998 Irena Pancirov - Irnet Snc
   This file is public domain and comes with NO WARRANTY of any kind
 -------------------------------------------------------------------------- */
#include <windows.h>
#include <process.h>
8
#include <stdio.h>
unknown's avatar
unknown committed
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
#include "nt_servc.h"


static NTService *pService;

/* ------------------------------------------------------------------------

 -------------------------------------------------------------------------- */
NTService::NTService()
{

    bOsNT	     = FALSE;
    //service variables
    ServiceName      = NULL;
    hExitEvent	     = 0;
    bPause	     = FALSE;
    bRunning	     = FALSE;
    hThreadHandle    = 0;
    fpServiceThread  = NULL;

    //time-out variables
    nStartTimeOut    = 15000;
31
    nStopTimeOut     = 86400000;
unknown's avatar
unknown committed
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55
    nPauseTimeOut    = 5000;
    nResumeTimeOut   = 5000;

    //install variables
    dwDesiredAccess  = SERVICE_ALL_ACCESS;
    dwServiceType    = SERVICE_WIN32_OWN_PROCESS;
    dwStartType      = SERVICE_AUTO_START;
    dwErrorControl   = SERVICE_ERROR_NORMAL;
    szLoadOrderGroup = NULL;
    lpdwTagID	     = NULL;
    szDependencies   = NULL;

    my_argc	     = 0;
    my_argv	     = NULL;
    hShutdownEvent   = 0;
    nError	     = 0;
    dwState	     = 0;
}

/* ------------------------------------------------------------------------

 -------------------------------------------------------------------------- */
NTService::~NTService()
{
56
  if (ServiceName != NULL) delete[] ServiceName;
unknown's avatar
unknown committed
57 58 59 60
}
/* ------------------------------------------------------------------------

 -------------------------------------------------------------------------- */
61

unknown's avatar
unknown committed
62 63 64 65 66 67 68 69 70 71 72 73 74
BOOL NTService::GetOS()
{
  bOsNT = FALSE;
  memset(&osVer, 0, sizeof(OSVERSIONINFO));
  osVer.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
  if (GetVersionEx(&osVer))
  {
    if (osVer.dwPlatformId == VER_PLATFORM_WIN32_NT)
      bOsNT = TRUE;
  }
  return bOsNT;
}

75

unknown's avatar
unknown committed
76 77 78 79 80 81
/* ------------------------------------------------------------------------
 Init()  Registers the main service thread with the service manager

    ServiceThread - pointer to the main programs entry function
		    when the service is started
 -------------------------------------------------------------------------- */
82

unknown's avatar
unknown committed
83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
long NTService::Init(LPCSTR szInternName,void *ServiceThread)
{

  pService = this;

  fpServiceThread = (THREAD_FC)ServiceThread;
  ServiceName = new char[lstrlen(szInternName)+1];
  lstrcpy(ServiceName,szInternName);

  SERVICE_TABLE_ENTRY stb[] =
  {
    { (char *)szInternName,(LPSERVICE_MAIN_FUNCTION) ServiceMain} ,
    { NULL, NULL }
  };

  return StartServiceCtrlDispatcher(stb); //register with the Service Manager
}
100 101


unknown's avatar
unknown committed
102 103 104 105 106 107 108
/* ------------------------------------------------------------------------
  Install() - Installs the service with Service manager
  nError values:
	0  success
	1  Can't open the Service manager
	2  Failed to create service
 -------------------------------------------------------------------------- */
109 110 111 112 113

BOOL NTService::Install(int startType, LPCSTR szInternName,
			LPCSTR szDisplayName,
			LPCSTR szFullPath, LPCSTR szAccountName,
			LPCSTR szPassword)
unknown's avatar
unknown committed
114
{
115
  BOOL ret_val=FALSE;
unknown's avatar
unknown committed
116 117
  SC_HANDLE newService, scm;

118 119 120 121 122 123
  if (!SeekStatus(szInternName,1))
   return FALSE;

  char szFilePath[_MAX_PATH];
  GetModuleFileName(NULL, szFilePath, sizeof(szFilePath));

unknown's avatar
unknown committed
124
  // open a connection to the SCM
125 126 127
  if (!(scm = OpenSCManager(0, 0,SC_MANAGER_CREATE_SERVICE)))
    printf("Failed to install the service (Couldn't open the SCM)\n");
  else 				// Install the new service
128
  {
129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
    if (!(newService=
	  CreateService(scm,
			szInternName,
			szDisplayName,
			dwDesiredAccess,//default: SERVICE_ALL_ACCESS
			dwServiceType,	//default: SERVICE_WIN32_OWN_PROCESS
		    			//default: SERVICE_AUTOSTART
			(startType == 1 ? SERVICE_AUTO_START :
			 SERVICE_DEMAND_START),
			dwErrorControl,	//default: SERVICE_ERROR_NORMAL
			szFullPath,	//exec full path
			szLoadOrderGroup, //default: NULL
			lpdwTagID,	//default: NULL
			szDependencies,	//default: NULL
			szAccountName,	//default: NULL
			szPassword)))	//default: NULL
      printf("Failed to install the service (Couldn't create service)\n");
     else
147
     {
148 149 150
       printf("Service successfully installed.\n");
       CloseServiceHandle(newService);
       ret_val=TRUE;				// Everything went ok
151
     }
152 153 154
     CloseServiceHandle(scm);
  }
  return ret_val;
unknown's avatar
unknown committed
155
}
156 157


unknown's avatar
unknown committed
158 159 160 161 162 163 164 165
/* ------------------------------------------------------------------------
  Remove() - Removes  the service
  nError values:
	0  success
	1  Can't open the Service manager
	2  Failed to locate service
	3  Failed to delete service
 -------------------------------------------------------------------------- */
166

unknown's avatar
unknown committed
167 168
BOOL NTService::Remove(LPCSTR szInternName)
{
169
  BOOL ret_value=FALSE;
unknown's avatar
unknown committed
170 171
  SC_HANDLE service, scm;

172 173 174
  if (!SeekStatus(szInternName,0))
   return FALSE;

unknown's avatar
unknown committed
175 176 177
  nError=0;

  // open a connection to the SCM
178
  if (!(scm = OpenSCManager(0, 0,SC_MANAGER_CREATE_SERVICE)))
179
  {
180
    printf("Failed to remove the service (Couldn't open the SCM)\n");
181 182
  }
  else
unknown's avatar
unknown committed
183
  {
184
    if ((service = OpenService(scm,szInternName, DELETE)))
unknown's avatar
unknown committed
185
    {
186
      if (!DeleteService(service))
187 188
        printf("Failed to remove the service\n");
      else
189
      {
190
        printf("Service successfully removed.\n");
191 192
	ret_value=TRUE;				// everything went ok
      }
193
      CloseServiceHandle(service);
unknown's avatar
unknown committed
194
    }
195 196 197
    else
      printf("Failed to remove the service (Couldn't open the service)\n");
    CloseServiceHandle(scm);
unknown's avatar
unknown committed
198
  }
199
  return ret_value;
unknown's avatar
unknown committed
200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216
}

/* ------------------------------------------------------------------------
   Stop() - this function should be called before the app. exits to stop
	    the service
 -------------------------------------------------------------------------- */
void NTService::Stop(void)
{
  SetStatus(SERVICE_STOP_PENDING,NO_ERROR, 0, 1, 60000);
  StopService();
  SetStatus(SERVICE_STOPPED, NO_ERROR, 0, 1, 1000);
}

/* ------------------------------------------------------------------------
  ServiceMain() - This is the function that is called from the
		  service manager to start the service
 -------------------------------------------------------------------------- */
217

unknown's avatar
unknown committed
218 219 220 221
void NTService::ServiceMain(DWORD argc, LPTSTR *argv)
{

  // registration function
222 223 224 225 226
  if (!(pService->hServiceStatusHandle =
	RegisterServiceCtrlHandler(pService->ServiceName,
				   (LPHANDLER_FUNCTION)
				   NTService::ServiceCtrlHandler)))
    goto error;
unknown's avatar
unknown committed
227 228

  // notify SCM of progress
229 230
  if (!pService->SetStatus(SERVICE_START_PENDING,NO_ERROR, 0, 1, 8000))
    goto error;
unknown's avatar
unknown committed
231 232

  // create the exit event
233 234
  if (!(pService->hExitEvent = CreateEvent (0, TRUE, FALSE,0)))
    goto error;
unknown's avatar
unknown committed
235

236 237 238
  if (!pService->SetStatus(SERVICE_START_PENDING,NO_ERROR, 0, 3,
			   pService->nStartTimeOut))
    goto error;
unknown's avatar
unknown committed
239 240 241 242 243 244

  // save start arguments
  pService->my_argc=argc;
  pService->my_argv=argv;

  // start the service
245 246
  if (!pService->StartService())
    goto error;
unknown's avatar
unknown committed
247

248 249 250
  // Check that the service is now running.
  if (!pService->SetStatus(SERVICE_RUNNING,NO_ERROR, 0, 0, 0))
    goto error;
unknown's avatar
unknown committed
251 252 253 254 255

  // wait for exit event
  WaitForSingleObject (pService->hExitEvent, INFINITE);

  // wait for thread to exit
256
  if (WaitForSingleObject (pService->hThreadHandle, INFINITE) == WAIT_TIMEOUT)
257
   CloseHandle(pService->hThreadHandle);
unknown's avatar
unknown committed
258 259

  pService->Exit(0);
260 261 262 263 264
  return;

error:
  pService->Exit(GetLastError());
  return;
unknown's avatar
unknown committed
265 266 267 268 269
}

/* ------------------------------------------------------------------------
   StartService() - starts the appliaction thread
 -------------------------------------------------------------------------- */
270

unknown's avatar
unknown committed
271 272 273
BOOL NTService::StartService()
{
  // Start the real service's thread (application)
274 275 276
  if (!(hThreadHandle = (HANDLE) _beginthread((THREAD_FC)fpServiceThread,0,
					      (void *) this)))
    return FALSE;
unknown's avatar
unknown committed
277 278 279 280 281 282 283 284 285 286 287
  bRunning = TRUE;
  return TRUE;
}
/* ------------------------------------------------------------------------

 -------------------------------------------------------------------------- */
void NTService::StopService()
{
  bRunning=FALSE;

  // Set the event for application
288
  if (hShutdownEvent)
unknown's avatar
unknown committed
289 290 291 292 293 294 295 296 297 298
     SetEvent(hShutdownEvent);

  // Set the event for ServiceMain
  SetEvent(hExitEvent);
}
/* ------------------------------------------------------------------------

 -------------------------------------------------------------------------- */
void NTService::PauseService()
{
299 300
  bPause = TRUE;
  SuspendThread(hThreadHandle);
unknown's avatar
unknown committed
301 302 303 304 305 306
}
/* ------------------------------------------------------------------------

 -------------------------------------------------------------------------- */
void NTService::ResumeService()
{
307 308
  bPause=FALSE;
  ResumeThread(hThreadHandle);
unknown's avatar
unknown committed
309 310 311 312 313
}
/* ------------------------------------------------------------------------

 -------------------------------------------------------------------------- */
BOOL NTService::SetStatus (DWORD dwCurrentState,DWORD dwWin32ExitCode,
314 315
			   DWORD dwServiceSpecificExitCode, DWORD dwCheckPoint,
			   DWORD dwWaitHint)
unknown's avatar
unknown committed
316 317 318 319
{
  BOOL bRet;
  SERVICE_STATUS serviceStatus;

320
  dwState=dwCurrentState;
unknown's avatar
unknown committed
321

322 323
  serviceStatus.dwServiceType	= SERVICE_WIN32_OWN_PROCESS;
  serviceStatus.dwCurrentState = dwCurrentState;
unknown's avatar
unknown committed
324

325 326 327 328 329 330
  if (dwCurrentState == SERVICE_START_PENDING)
    serviceStatus.dwControlsAccepted = 0;	//don't accept control events
  else
    serviceStatus.dwControlsAccepted =    (SERVICE_ACCEPT_STOP |
					   SERVICE_ACCEPT_PAUSE_CONTINUE |
					   SERVICE_ACCEPT_SHUTDOWN);
unknown's avatar
unknown committed
331

332 333 334 335 336
  // if a specific exit code is defined,set up the win32 exit code properly
  if (dwServiceSpecificExitCode == 0)
    serviceStatus.dwWin32ExitCode = dwWin32ExitCode;
  else
    serviceStatus.dwWin32ExitCode = ERROR_SERVICE_SPECIFIC_ERROR;
unknown's avatar
unknown committed
337

338
  serviceStatus.dwServiceSpecificExitCode = dwServiceSpecificExitCode;
unknown's avatar
unknown committed
339

340 341
  serviceStatus.dwCheckPoint = dwCheckPoint;
  serviceStatus.dwWaitHint   = dwWaitHint;
unknown's avatar
unknown committed
342

343 344 345
  // Pass the status to the Service Manager
  if (!(bRet=SetServiceStatus (hServiceStatusHandle, &serviceStatus)))
    StopService();
unknown's avatar
unknown committed
346

347
  return bRet;
unknown's avatar
unknown committed
348 349 350 351 352 353
}
/* ------------------------------------------------------------------------

 -------------------------------------------------------------------------- */
void NTService::ServiceCtrlHandler(DWORD ctrlCode)
{
354
  DWORD  dwState;
unknown's avatar
unknown committed
355

356 357
  if (!pService)
    return;
unknown's avatar
unknown committed
358 359 360

  dwState=pService->dwState;  // get current state

361
  switch(ctrlCode) {
unknown's avatar
unknown committed
362

363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395
#ifdef NOT_USED	 /* do we need this ? */
  case SERVICE_CONTROL_PAUSE:
    if (pService->bRunning && ! pService->bPause)
    {
      dwState = SERVICE_PAUSED;
      pService->SetStatus(SERVICE_PAUSE_PENDING,NO_ERROR, 0, 1,
			  pService->nPauseTimeOut);
      pService->PauseService();
    }
    break;

  case SERVICE_CONTROL_CONTINUE:
    if (pService->bRunning && pService->bPause)
    {
      dwState = SERVICE_RUNNING;
      pService->SetStatus(SERVICE_CONTINUE_PENDING,NO_ERROR, 0, 1,
			  pService->nResumeTimeOut);
      pService->ResumeService();
    }
    break;
#endif

  case SERVICE_CONTROL_SHUTDOWN:
  case SERVICE_CONTROL_STOP:
    dwState = SERVICE_STOP_PENDING;
    pService->SetStatus(SERVICE_STOP_PENDING,NO_ERROR, 0, 1,
			pService->nStopTimeOut);
    pService->StopService();
    break;

  default:
    pService->SetStatus(dwState, NO_ERROR,0, 0, 0);
    break;
unknown's avatar
unknown committed
396 397 398
  }
  //pService->SetStatus(dwState, NO_ERROR,0, 0, 0);
}
399

unknown's avatar
unknown committed
400 401 402
/* ------------------------------------------------------------------------

 -------------------------------------------------------------------------- */
403

unknown's avatar
unknown committed
404 405
void NTService::Exit(DWORD error)
{
406 407
  if (hExitEvent)
    CloseHandle(hExitEvent);
unknown's avatar
unknown committed
408 409 410

  // Send a message to the scm to tell that we stop
  if (hServiceStatusHandle)
411
    SetStatus(SERVICE_STOPPED, error,0, 0, 0);
unknown's avatar
unknown committed
412 413 414 415 416 417

  // If the thread has started kill it ???
  // if (hThreadHandle) CloseHandle(hThreadHandle);

}

418 419 420
/* ------------------------------------------------------------------------

 -------------------------------------------------------------------------- */
421

422 423
BOOL NTService::SeekStatus(LPCSTR szInternName, int OperationType)
{
424
  BOOL ret_value=FALSE;
425 426 427
  SC_HANDLE service, scm;

  // open a connection to the SCM
428
  if (!(scm = OpenSCManager(0, 0,SC_MANAGER_CREATE_SERVICE)))
429 430 431 432 433 434 435 436 437 438 439
  {
    DWORD ret_error=GetLastError();
    if (ret_error == ERROR_ACCESS_DENIED)
    {
     printf("Install/Remove of the Service Denied!\n");
     if(!is_super_user())
      printf("That operation should be made by an user with Administrator privileges!\n");
    }
    else
     printf("There is a problem for to open the Service Control Manager!\n");
  }
440
  else
441
  {
442
    if (OperationType == 1)
443
    {
444 445
      /* an install operation */
      if ((service = OpenService(scm,szInternName, SERVICE_ALL_ACCESS )))
446
      {
447 448 449 450 451 452 453 454 455 456
	LPQUERY_SERVICE_CONFIG ConfigBuf;
	DWORD dwSize;

	ConfigBuf = (LPQUERY_SERVICE_CONFIG) LocalAlloc(LPTR, 4096);
	printf("The service already exists!\n");
	if (QueryServiceConfig(service,ConfigBuf,4096,&dwSize))
	  printf("The current server installed: %s\n",
		 ConfigBuf->lpBinaryPathName);
	LocalFree(ConfigBuf);
	CloseServiceHandle(service);
457
      }
458 459
      else
	ret_value=TRUE;
460
    }
461
    else
462
    {
463 464 465
      /* a remove operation */
      if (!(service = OpenService(scm,szInternName, SERVICE_ALL_ACCESS )))
	printf("The service doesn't exists!\n");
466 467
      else
      {
468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483
	SERVICE_STATUS ss;

	memset(&ss, 0, sizeof(ss));
	if (QueryServiceStatus(service,&ss))
	{
	  DWORD dwState = ss.dwCurrentState;
	  if (dwState == SERVICE_RUNNING)
	    printf("Failed to remove the service because the service is running\nStop the service and try again\n");
	  else if (dwState == SERVICE_STOP_PENDING)
	    printf("\
Failed to remove the service because the service is in stop pending state!\n\
Wait 30 seconds and try again.\n\
If this condition persist, reboot the machine and try again\n");
	  else
	    ret_value= TRUE;
	}
484 485
	CloseServiceHandle(service);
      }
486 487 488
    }
    CloseServiceHandle(scm);
  }
489

490
  return ret_value;
491
}
unknown's avatar
unknown committed
492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513
/* ------------------------------------------------------------------------
 -------------------------------------------------------------------------- */
BOOL NTService::IsService(LPCSTR ServiceName)
{
  BOOL ret_value=FALSE;
  SC_HANDLE service, scm;
  
  if (scm = OpenSCManager(0, 0,SC_MANAGER_ENUMERATE_SERVICE))
  {
    if ((service = OpenService(scm,ServiceName, SERVICE_ALL_ACCESS )))
    {
      ret_value=TRUE;
      CloseServiceHandle(service);
    }
    CloseServiceHandle(scm);
  }
  return ret_value;
}
/* ------------------------------------------------------------------------
 -------------------------------------------------------------------------- */
BOOL NTService::got_service_option(char **argv, char *service_option)
{
514 515 516 517
  char *option;
  for (option= argv[1]; *option; option++)
    if (!strcmp(option, service_option))
      return TRUE;
unknown's avatar
unknown committed
518 519
  return FALSE;
}
520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570
/* ------------------------------------------------------------------------
 -------------------------------------------------------------------------- */
BOOL NTService::is_super_user()
{
  HANDLE hAccessToken;
  UCHAR InfoBuffer[1024];
  PTOKEN_GROUPS ptgGroups=(PTOKEN_GROUPS)InfoBuffer;
  DWORD dwInfoBufferSize;
  PSID psidAdministrators;
  SID_IDENTIFIER_AUTHORITY siaNtAuthority = SECURITY_NT_AUTHORITY;
  UINT x;
  BOOL ret_value=FALSE;
 
  if(!OpenThreadToken(GetCurrentThread(), TOKEN_QUERY, TRUE,&hAccessToken ))
  {
   if(GetLastError() != ERROR_NO_TOKEN)
    return FALSE;
 
   if(!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hAccessToken))
    return FALSE;
  }
 
  ret_value= GetTokenInformation(hAccessToken,TokenGroups,InfoBuffer,
                                 1024, &dwInfoBufferSize);

  CloseHandle(hAccessToken);
 
  if(!ret_value )
   return FALSE;
 
  if(!AllocateAndInitializeSid(&siaNtAuthority, 2,
      SECURITY_BUILTIN_DOMAIN_RID,
      DOMAIN_ALIAS_RID_ADMINS,
      0, 0, 0, 0, 0, 0,
      &psidAdministrators))
      return FALSE;

  ret_value = FALSE;
 
  for(x=0;x<ptgGroups->GroupCount;x++)
  {
   if( EqualSid(psidAdministrators, ptgGroups->Groups[x].Sid) )
   {
    ret_value = TRUE;
    break;
   }
 
  }
  FreeSid(psidAdministrators);
  return ret_value;
}