ntservapp.cpp

本程序是实现中国移动中国联通的网关程序．代码比较完整．

// NTService.cpp
// 
// This is the main program file containing the entry point.

#include "NTServApp.h"
#include "myservice.h"
BYTE isLogFile;

int main(int argc, char* argv[])
{
 	FILE* pf;
	char lpFileName[MAX_PATH];
	char szName[64];
	char szSubName[64];

	GetModuleFileName(NULL,lpFileName,MAX_PATH);
	sprintf(strrchr(lpFileName,'\\')+1,"ServiceName.ini\0");
	pf=fopen(lpFileName,"r");
	if(pf==NULL)
	{
		strcpy(szSubName,"SUB SERVICE");
		strcpy(szName,"SUB SERVICE");
	}
	else
	{
	if(fgets(szSubName,64,pf)==NULL)
	{
		AddLog("服务配置文件读取错误：%s\n",lpFileName);
		fclose(pf);
		return 0;
	}
	if(fgets(szName,64,pf)==NULL)
	{
		AddLog("服务配置文件读取错误：%s\n",lpFileName);
		fclose(pf);
		return 0;
	}
	if(fgets(lpFileName,64,pf)==NULL)isLogFile=0;
	else isLogFile=atoi(lpFileName);

	fclose(pf);	
	if(szName[strlen(szName)-1]<=0x20)szName[strlen(szName)-1]=0;
	if(szSubName[strlen(szSubName)-1]<=0x20)szSubName[strlen(szSubName)-1]=0;
	}

	CMyService MyService(szName,szSubName);
	// Parse for standard arguments (install, uninstall, version etc.)
	if (!MyService.ParseStandardArgs(argc, argv)) {

		// Didn't find any standard args so start the service
		// Uncomment the DebugBreak line below to enter the debugger
		// when the service is started.
		//DebugBreak();
		MyService.StartService();
	}

	// When we get here, the service has been stopped
	return MyService.m_Status.dwWin32ExitCode;
}

void AddLog(const char* format,...)
{
	FILE* pfLog;
	char current[36];
	char tp[36];
	struct _timeb timebuffer;
	char lpFileName[MAX_PATH];
	if(isLogFile==0)return;
	_strdate(current);
	_ftime(&timebuffer);
	strcpy(tp,ctime(&(timebuffer.time)));
	current[2]='_';
	current[5]=0;
	tp[strlen(tp)-1]=0;
	GetModuleFileName(NULL,lpFileName,MAX_PATH);
	sprintf(strrchr(lpFileName,'\\')+1,
	"Service%s_%.2s.txt\0",strrchr(tp,' ')+1,current);
	current[2]='-';
	tp[strrchr(tp,' ')-tp]=0;
	pfLog=fopen(lpFileName,"a");
	va_list args;
	va_start(args,format);
#if _DEBUG
	vfprintf(stdout,format,args);
#endif
	fprintf(pfLog,"%s<%s.%.3u> ",current,
		strrchr(tp,' ')+1,timebuffer.millitm);
	vfprintf(pfLog,format,args);
	va_end(args);
	fflush(pfLog);
	fclose(pfLog);
}