PDA

View Full Version : [C++]Avvio da servizio


4ndr34s
11-03-2009, 19:26
#define _WIN32_WINNT 0x0500
#pragma comment(lib, "advapi32.lib")
#include <iostream.h>
#include <windows.h>
#include <tlhelp32.h>
#include <stdlib.h>
#include <stdio.h>
#include <winsvc.h>
//--------KILL PROCESS--------
BOOL KillProcessByName(char *szProcessToKill){
HANDLE hProcessSnap;
HANDLE hProcess;
PROCESSENTRY32 pe32;
DWORD dwPriorityClass;
hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0); // Takes a snapshot of all the processes
if(hProcessSnap == INVALID_HANDLE_VALUE){
return( FALSE );
}
pe32.dwSize = sizeof(PROCESSENTRY32);
if(!Process32First(hProcessSnap, &pe32)){
CloseHandle(hProcessSnap);
return( FALSE );
}
do{
if(!strcmp(pe32.szExeFile,szProcessToKill)){ // checks if process at current position has the name of to be killed app
hProcess = OpenProcess(PROCESS_TERMINATE,0, pe32.th32ProcessID); // gets handle to process
TerminateProcess(hProcess,0); // Terminate process by handle
CloseHandle(hProcess); // close the handle
}
}while(Process32Next(hProcessSnap,&pe32)); // gets next member of snapshot
CloseHandle(hProcessSnap); // closes the snapshot handle
return( TRUE );
}//--------KILL PROCESS--------


//--------FUNZIONE QUANDO SI CLICCLA SULLA X--------
void onExit( void )
{
KillProcessByName("notepad.exe");// Run cleanup code here!
}//----FUNZIONE QUANDO SI CLICCA SULLA X----



DWORD CountProcesses(CHAR *pProcessName)
{
DWORD dwCount = 0;
HANDLE hSnap = NULL;
PROCESSENTRY32 proc32;

if((hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0)) == INVALID_HANDLE_VALUE)
return -1;
proc32.dwSize=sizeof(PROCESSENTRY32);
while((Process32Next(hSnap, &proc32)) == TRUE)
if(stricmp(proc32.szExeFile,pProcessName) == 0)
++dwCount;
CloseHandle(hSnap);
return dwCount;
}//--------CONTROLLO PROCESSO--------




void ServiceMain(DWORD argc, LPTSTR *argv);
void ServiceCtrlHandler(DWORD nControlCode);
BOOL UpdateServiceStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode,
DWORD dwServiceSpecificExitCode, DWORD dwCheckPoint,
DWORD dwWaitHint);
BOOL StartServiceThread();
DWORD ServiceExecutionThread(LPDWORD param);
HANDLE hServiceThread;
void KillService();

char *strServiceName = "UG Anticheat Monitor";
SERVICE_STATUS_HANDLE nServiceStatusHandle;
HANDLE killServiceEvent;
BOOL nServiceRunning;
DWORD nServiceCurrentStatus;

int main(int argc, char* argv[])
{

SERVICE_TABLE_ENTRY servicetable[]=
{
{strServiceName,(LPSERVICE_MAIN_FUNCTION)ServiceMain},
{NULL,NULL}
};
BOOL success;
success=StartServiceCtrlDispatcher(servicetable);

if(!success)
{
//error occured
}
}

void ServiceMain(DWORD argc, LPTSTR *argv)
{
BOOL success;
nServiceStatusHandle=RegisterServiceCtrlHandler(strServiceName,
(LPHANDLER_FUNCTION)ServiceCtrlHandler);
if(!nServiceStatusHandle)
{
return;
}
success=UpdateServiceStatus(SERVICE_START_PENDING,NO_ERROR,0,1,3000);
if(!success)
{
return;
}
killServiceEvent=CreateEvent(0,TRUE,FALSE,0);
if(killServiceEvent==NULL)
{
return;
}
success=UpdateServiceStatus(SERVICE_START_PENDING,NO_ERROR,0,2,1000);
if(!success)
{
return;
}
success=StartServiceThread();
if(!success)
{
return;
}
nServiceCurrentStatus=SERVICE_RUNNING;
success=UpdateServiceStatus(SERVICE_RUNNING,NO_ERROR,0,0,0);
if(!success)
{
return;
}
WaitForSingleObject(killServiceEvent,INFINITE);
CloseHandle(killServiceEvent);
}



BOOL UpdateServiceStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode,
DWORD dwServiceSpecificExitCode, DWORD dwCheckPoint,
DWORD dwWaitHint)
{
BOOL success;
SERVICE_STATUS nServiceStatus;
nServiceStatus.dwServiceType=SERVICE_WIN32_OWN_PROCESS;
nServiceStatus.dwCurrentState=dwCurrentState;
if(dwCurrentState==SERVICE_START_PENDING)
{
nServiceStatus.dwControlsAccepted=0;
}
else
{
nServiceStatus.dwControlsAccepted=SERVICE_ACCEPT_STOP
|SERVICE_ACCEPT_SHUTDOWN;
}
if(dwServiceSpecificExitCode==0)
{
nServiceStatus.dwWin32ExitCode=dwWin32ExitCode;
}
else
{
nServiceStatus.dwWin32ExitCode=ERROR_SERVICE_SPECIFIC_ERROR;
}
nServiceStatus.dwServiceSpecificExitCode=dwServiceSpecificExitCode;
nServiceStatus.dwCheckPoint=dwCheckPoint;
nServiceStatus.dwWaitHint=dwWaitHint;

success=SetServiceStatus(nServiceStatusHandle,&nServiceStatus);

if(!success)
{
KillService();
return success;
}
else
return success;
}

BOOL StartServiceThread()
{
DWORD id;
hServiceThread=CreateThread(0,0,
(LPTHREAD_START_ROUTINE)ServiceExecutionThread,
0,0,&id);
if(hServiceThread==0)
{
return false;
}
else
{
nServiceRunning=true;
return true;
}
}

DWORD ServiceExecutionThread(LPDWORD param)
{
while(nServiceRunning)
{
Beep(450,150);
Sleep(4000);
}
return 0;
}

void KillService()
{
nServiceRunning=false;
SetEvent(killServiceEvent);
UpdateServiceStatus(SERVICE_STOPPED,NO_ERROR,0,0,0);
}

void ServiceCtrlHandler(DWORD nControlCode)
{
BOOL success;
switch(nControlCode)
{
case SERVICE_CONTROL_SHUTDOWN:
case SERVICE_CONTROL_STOP:
nServiceCurrentStatus=SERVICE_STOP_PENDING;
success=UpdateServiceStatus(SERVICE_STOP_PENDING,NO_ERROR,0,1,3000);
KillService();
return;
default:
break;
}
UpdateServiceStatus(nServiceCurrentStatus,NO_ERROR,0,0,0);
}


ragazzi avrei un problema, questo codice serve ad avviare un exe essendo un servizio,fino qui tutto bene, solo che vorrei che questo servizio avviasse un altro exe.
solo che c'è un problema: dove ci metto la stringa????

while(nServiceRunning)
{
Beep(450,150);
Sleep(4000);
}
return 0;
io l'ho messa qui scrivendoci system("omefile");
il file lo avvia ma lo avvia sotto il nome di system, poii questo eseguibile lanciato dal servizio non si vede e non fa la propia funzione, perchè??
dove ho sbagliato??'

fero86
11-03-2009, 22:41
indenta, altrimenti non lo leggerá nessuno.

4ndr34s
11-03-2009, 23:07
purtroppo quando ho incollato l'indentazione è andata a farsi benedire :-S

4ndr34s
12-03-2009, 18:06
nessuno sà darmi una mano?

fero86
12-03-2009, 18:28
purtroppo quando ho incollato l'indentazione è andata a farsi benedire :-S apri Visual C++, incolla il codice, fai Ctrl+A, poi Alt+F8, e poi ri-copiaincolla tutto.

cionci
12-03-2009, 19:01
Il problema è che hai usato il tag QUOTE al posto del tag CODE nello scrivere il post.

C'è una API che si chiama CreateProcessAsUser, potrebbe fare al caso tuo.
In alternativa nelle proprietà del servizio puoi impostare, anche solo per verifica, che il servizio venga eseguito attraverso un determinato utente.

fero86
12-03-2009, 20:18
io l'ho messa qui scrivendoci system("omefile");
il file lo avvia ma lo avvia sotto il nome di system, poii questo eseguibile lanciato dal servizio non si vede e non fa la propia funzione, perchè??
dove ho sbagliato??' non si vede perché gira nel desktop della sessione dell'utente SYSTEM, non in quello interattivo; per il fatto che non svolga la sua funzione non saprei, dipende da qual é la sua funzione.

cionci
12-03-2009, 20:23
E' capitato anche a me di avere problemi con un processo lanciato da un servizio che girava come SYSTEM, non ne so il motivo sinceramente. Io lanciavo un programma di backup, ma questo non funzionava. Probabilmente c'è qualche limitazione.

4ndr34s
13-03-2009, 16:54
ok ragazzi risolto se vi puo servire la stringa da utilizzare nei servizi è

Winexec("nomefile",SW_SHOW);