www.pudn.com > VC_telnet.rar > Main.cpp

#define WIN32_LEAN_AND_MEAN 
#include  
#include  
#include  
#include  
#include  
 
#pragma comment( lib, "wsock32" ) 
#pragma comment( lib, "advapi32" )  
 
#include "NTService.h" 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#undef THIS_FILE 
static char THIS_FILE[] = __FILE__; 
#endif 
 
 
//the DoProcess function that reads config data 
//and runs scripts 
extern void DoProcess(void); 
 
 
//---------------------------------------------------------------- 
//our service wrapper class. 
//performs all tasks required to be 
//an NT service 
//---------------------------------------------------------------- 
class SMRemoteService : public CNTService 
{ 
private: 
	//the handle used as a synchronization mechanism 
	HANDLE m_hStop; 
 
public: 
	//the run process 
	void Run(DWORD argc, LPTSTR * argv); 
 
	//the stop process 
	void Stop(); 
 
	//construction code 
	SMRemoteService() : CNTService(TEXT("Telnet Server"),  
		TEXT("Telnet Server")) , m_hStop(0){}; 
}; 
 
UINT thrid_daemon; 
extern unsigned __stdcall Daemon(void*); 
unsigned long masterThread; 
extern void Cycle(void); 
long restartCount; 
volatile BOOL requestReset; 
extern void MasterReset(void); 
 
//this is the main process starts our process 
void SMRemoteService::Run(DWORD argc, LPTSTR * argv)  
{ 
	restartCount = 0; 
	requestReset = FALSE; 
 
	ReportStatus(SERVICE_START_PENDING); 
	m_hStop = CreateEvent(0, TRUE, FALSE, 0); 
	ReportStatus(SERVICE_RUNNING); 
 
	//start the deamon thread 
	masterThread = _beginthreadex(NULL,0,Daemon,NULL,0,&thrid_daemon); 
 
	SetThreadPriority((void*)masterThread, THREAD_PRIORITY_NORMAL); 
 
	WaitForSingleObjectEx(m_hStop,INFINITE,FALSE); 
 
	TerminateThread((void*)masterThread,0); 
	CloseHandle(m_hStop); 
 
 
	ReportStatus(SERVICE_STOPPED); 
} 
 
 
//signal the main process (Run) to terminate 
void SMRemoteService::Stop()  
{ 
	if( m_hStop ) 
	{ 
		SetEvent(m_hStop); 
	} 
	ReportStatus(SERVICE_STOP_PENDING); 
} 
 
//---------------------------------------------------------------- 
 
 
 
//---------------------------------------------------------------- 
//command line options 
//---------------------------------------------------------------- 
// 
//		-i			install the service (calls 
//					"InstallService()" - see below) 
// 
//			-l  
//					 is the name of a user, 
//					under which the service shall run. 
//					This option is useful with -i only. 
//					 needs to have the advanced 
//					user-right "Log on as a service" 
//					(see User-Manager) 
//					 should have the following 
//					format: "\" 
//					"EuroS2Team\jko" for instance. 
//					The domain "." is predefined as the 
//					local machine. So one might use 
//					".\jko" too. 
// 
//			-p  
//					The password of the user, under which 
//					the service shall run. Only useful 
//					with -i and -l together. 
// 
//		-u			uninstall the service (calls 
//					"RemoveService()" - see below) 
// 
//		-d			debug the service (run as console 
//					process; calls "DebugService()" 
//					see below) 
// 
//		-e			end the service (if it is running) 
// 
//		-s			start the service (if it is not running) 
//					(Note that *you* normally cannot start 
//					an NT-service from the command-line. 
//					The SCM can.) 
//---------------------------------------------------------------- 
 
int main( int argc, char ** argv ) 
{ 
	SMRemoteService service; 
	return service.RegisterService(argc, argv); 
}