c/c++编写window服务的授权服务(三)
//实现授权服务的函数体LicenseService.cpp
#pragma once
#include "LicenseService.h"
#include "hex_md5.h"
//安装入口点
void main(int argc, char* argv[])
{
//初始化全局变量
//获取exe所在目录
CurExeDir=GetCurDir();
//printf("\n当前exe所在目录为:%s\n",CurExeDir);
//获取ServerConfig.ini配置参数并初始化全局变量
GetServerConfig_ini();
//开始初始化服务
WCHAR* TServiceName=CharPtr2ACharPtr(ServiceName);
//方法一
SERVICE_TABLE_ENTRY ServiceTable[2];
//设置服务名称和服务入口函数
ServiceTable[0].lpServiceName=TServiceName;
ServiceTable[0].lpServiceProc=(LPSERVICE_MAIN_FUNCTION)ServiceMain;
ServiceTable[1].lpServiceName=NULL;
ServiceTable[1].lpServiceProc=NULL;
//方法二
/*SERVICE_TABLE_ENTRY ServiceTable[]=
{
{TEXT(ServiceName),(LPSERVICE_MAIN_FUNCTION)ServiceMain},
{ NULL,NULL}
};*/
//注册为服务和反注册
//-------------------------------------------------
//引数
if(argc<=1)
{
//usage(使用方法)表示
cout<<"使用方法:"<<endl;
//cout<<"zxz\"zxz"<<ENDL;
//"install"
cout<<"\"install\"";
//"or"
cout<<"or";
//"uninstall"
cout<<"\"uninstall\"";
//"exec"
cout<<"\"exec \"";
cout<<" parameter required in the command line"<<endl;
}
if(argc>1)
{
if(_stricmp("install",argv[1])==0)
{ //install(InstallService)
InstallService();
cout<<"installService Succed!"<<endl;
}
else if(_stricmp("uninstall",argv[1])==0)
{ //uninstall(UninstallService)
UninstallService();
cout<<"UninstallService Succed!"<<endl;
}
else if(_stricmp("exec",argv[1])==0)
{ //exec(ServiceMain)
RunService(TServiceName);
}
else
{ //usage(使用方法)表示
cout<<"使用方法 :"<<endl;
//"install"
cout<<"\"install\"";
//"or"
cout<<"or";
//"uninstall"
cout<<"\"uninstall\"";
//"exec"
cout<<"\"exec \"";
//"or"
cout<<"or";
cout<<" parameter required in the command line"<<endl;
}
}
//-------------------------------------------------
//启动服务的控制分派机线程(启动并执行一个新线程)
StartServiceCtrlDispatcher(ServiceTable);
}
//服务入口点(主线程)
void ServiceMain(int argc,char** argv)
{
WCHAR* TServiceName=CharPtr2ACharPtr(ServiceName);
int success;
//success=InitService();
//if(!success)
//{
// //初始化失败,终止服务
// WriteToLog("ServiceMain::InitService初始化失败,终止服务");
// ServiceStatus.dwCurrentState=SERVICE_STOPPED;
// ServiceStatus.dwWin32ExitCode=-1;
// SetServiceStatus(hStatus,&ServiceStatus);
// //退出ServiceMain
// return;
//}
//开始服务
RunService(TServiceName);
//注册服务控制处理函数
hStatus = RegisterServiceCtrlHandler(TServiceName,(LPHANDLER_FUNCTION)ControlHandler);
if(!hStatus)
{
WriteToLog("ServiceMain函数中注册服务控制处理函数失败!\n");
return;
}
//向SCM报告工作状态
RunState=EnumRunState::Running;
ServiceStatus.dwCurrentState=SERVICE_RUNNING;
ServiceStatus.dwServiceType=SERVICE_WIN32;
ServiceStatus.dwControlsAccepted =
SERVICE_ACCEPT_STOP |
SERVICE_ACCEPT_PAUSE_CONTINUE |
SERVICE_ACCEPT_SHUTDOWN;
/*SetServiceStatus(hStatus,&ServiceStatus);*/
启动工作循环(这个模块在另外一个线程中执行)
//while(RunState==EnumRunState::Running)
//{
// WriteToLog("ServiceMain函数写日志\n");
// Sleep(500);
//}
//创建一个终结事件
terminateEvent = CreateEvent (0,TRUE,FALSE,0);
if (!terminateEvent)
{
terminate(GetLastError());
return;
}
//
success = SendStatusToSCM(SERVICE_START_PENDING,NO_ERROR,0,2,1000);
if (!success)
{
terminate(GetLastError());
return;
}
//初始化
success = InitService();
if (!success)
{
terminate(GetLastError());
return;
}
success = SendStatusToSCM(SERVICE_RUNNING,NO_ERROR,0,0,0);
if (!success)
{
terminate(GetLastError());
return;
}
//等待停止
WaitForSingleObject ( terminateEvent, INFINITE);
terminate(0);
WriteToLog("ServiceMain函数执行完毕!");
}
//服务控制台命令处理函数
void ControlHandler(DWORD request)
{
DWORD currentState = 0;
BOOL success;
switch(request)
{
case SERVICE_CONTROL_STOP:
RunState=EnumRunState::Stoped;
currentState = SERVICE_STOP_PENDING;
ServiceStatus.dwCurrentState=currentState;
SetServiceStatus(hStatus,&ServiceStatus);
StopService();
return;
case SERVICE_CONTROL_PAUSE:
RunState=EnumRunState::Pause;
currentState = SERVICE_PAUSED;
ServiceStatus.dwCurrentState=currentState;
SetServiceStatus(hStatus,&ServiceStatus);
PauseService();
break;
case SERVICE_CONTROL_CONTINUE:
RunState=EnumRunState::Running;
currentState = SERVICE_RUNNING;
ServiceStatus.dwCurrentState=currentState;
SetServiceStatus(hStatus,&ServiceStatus);
ResumeService();
break;
case SERVICE_CONTROL_INTERROGATE:
// it will fall to bottom and send status
break;
// Do nothing in a shutdown. Could do cleanup
// here but it must be very quick.
case SERVICE_CONTROL_SHUTDOWN:
// Do nothing on shutdown
return;
default:
break;
}
//ServiceStatus.dwCurrentState=currentState;
//SetServiceStatus(hStatus,&ServiceStatus);
}
//初始化服务(开始服务时执行)
int InitService()
{
//重新获取ServerConfig.ini配置参数
GetServerConfig_ini();
int result;
WCHAR* TServiceName=CharPtr2ACharPtr(ServiceName);
//写日志信息
result=WriteToLog("Monitoring started.");
//start listening...
//开始一个线程 Start the service's thread
DWORD id;
threadHandle = CreateThread(0, 0,(LPTHREAD_START_ROUTINE)ServiceThread,0, 0, &id);
if (threadHandle==0)
{
result=0;
}
else
{
result=1;
}
WriteToLog("InitService一个线程开始了!");
return result;
}
DWORD WINAPI ServiceThread(LPDWORD param)
{
//线程操作
WriteToLog("线程操作ServiceThread开始!");
StartHostPortListen(host,atoi(port));
WriteToLog("线程操作ServiceThread完成!");
return 0;
}
//实现 开始主机+端口监听函数
void StartHostPortListen(char* ipaddr,int port)
{
//port=9998;
WORD wVersionRequested;
WSADATA wsaData;
int err;
//版本请求
wVersionRequested = MAKEWORD(2, 2);
//初始化WinSock(WSAStartup函数)
err = WSAStartup( wVersionRequested, &wsaData);
if ( err != 0 )
{ //初始化出错了退出函数
return;
}
//初始化成功！但版本不一致! 退出函数
if ( LOBYTE( wsaData.wVersion ) != 2 ||
HIBYTE( wsaData.wVersion ) != 2 ) {
WSACleanup( ); //清除Socket
return; //退出函数
}
//生成一个socket(sockSrv变量)
//SOCK_STREAM 用于TCP协议
//SOCK_DGRAM 用于UDP协议
SOCKET sockSrv=socket(AF_INET,SOCK_STREAM,0); //使用TCP协议
//设置一个地址Socket(addrSrv变量)
SOCKADDR_IN addrSrv;
addrSrv.sin_addr.S_un.S_addr=inet_addr(ipaddr); //本机IP地址
addrSrv.sin_family=AF_INET;
addrSrv.sin_port=htons(port); //端口9998
//绑定函数(指定本地IP地址所使用的端口号时候使用)
//(把sockSrv变量与addrSrv变量绑定在一起)
bind(sockSrv,(SOCKADDR*)&addrSrv,sizeof(SOCKADDR));
//监听端口开始
if(listen(sockSrv,MAX_PENDING_CONNECTS)!=0)
{
printf("监听socket失败:%d\n ",::WSAGetLastError());
return;//ExitProcess(STATUS_FAILED);
}
printf("开始监听...");
//接收来自客户端的请求连接
SOCKET sockClient;
SOCKADDR_IN addrClient;//定义一个客户端Socket地址
int len=sizeof(SOCKADDR);
char recvBuf[100];
int recv_len=-1;
//
sockaddr_in acc_sin;
int acc_sin_len;
acc_sin_len=sizeof(acc_sin);
//
while(true)
{
//接收Socket请求 来自客户端的Connect
sockClient=accept(sockSrv,(struct sockaddr*)&acc_sin, &acc_sin_len);
if(sockClient>=0)
{
//创建一个线程来处理socket请求threadHandle
DWORD id;
HANDLE threadHandle = CreateThread(0, 0,(LPTHREAD_START_ROUTINE)HandleSocket,(LPVOID)sockClient, 0, &id);
//
}
else
{ //没有客户端接入
printf("没有客户端接入2\n");
}
Sleep(200);
}
//关闭Socket变量sockSrv
closesocket(sockSrv);
//清除Socket
WSACleanup();
}
//实现 处理Socket请求函数
DWORD WINAPI HandleSocket(LPVOID lpParam)
{
SOCKET sockClient=(SOCKET)lpParam;
//接收数据并保存到recvBuf变量中
char recvBuf[100];
int recv_len=-1;
bool Exits=false;
while(Exits==false)
{
recv_len=recv(sockClient,recvBuf,100,0);
//recv_len=recvfrom(sockClient,recvBuf,100,0,(SOCKADDR*)&addrClient,&len);
if(recv_len>0)
{
//输出数据
recvBuf[recv_len]='\0'; //在末尾加入字符串结束符号 '\0 '
printf("%s\n",recvBuf);
printf("本次数据接收结束。\n ");
//处理(传入请求类型)
char* ret;
ret=new char[recv_len];
strcpy(ret,recvBuf);
bool rlcmd=false,stop=false;
rlcmd=EqStr(ret,"RequestLicenseCommand");
stop=EqStr(ret,"stop");
if(rlcmd==true)
{ //请求类型为:RequestLicenseCommand
bool LicenseAbled=false;
//计算License是否可用
LicenseAbled=false;//true;
//需进一步判断
LicenseAbled=IsLencense();
//-----
char* retmsg="stop";
if(LicenseAbled==true )//&& recvBuf==reqCmd
{ //可用
retmsg="start";
}
else
{ //不可用
retmsg="stop";
}
WriteToLog(retmsg);
if(send(sockClient,retmsg,strlen(retmsg),0)==SOCKET_ERROR)
{
printf("回发信息报错了: %d\n ",WSAGetLastError());
}
printf("本次数据处理结束。\n ");
}
if(stop==true)
{ //请求类型为:stop
Exits=true; //退出(结束本线程)
}
}
}
printf("退出(结束本线程)\n ");
return 0;
}
//运行服务函数
BOOL RunService(WCHAR* sSvcName)
{
SC_HANDLE schSCManager;
SC_HANDLE scHandle;
BOOL boolRet;
// open to SCM
schSCManager = OpenSCManager(
0,
0,
SC_MANAGER_ALL_ACCESS
);
//Open
scHandle = OpenService(
schSCManager,
sSvcName,
SERVICE_ALL_ACCESS
);
//start
boolRet = StartService(
scHandle,
0,
NULL);
return boolRet;
}
VOID ResumeService()
{
//pauseService=FALSE;
//ResumeThread(threadHandle);
return;
}
VOID PauseService()
{
//pauseService = TRUE;
//SuspendThread(threadHandle);
return;
}
VOID StopService()
{
RunState=EnumRunState::Stoped;//=FALSE;
// Set the event that is holding ServiceMain
// so that ServiceMain can return
SetEvent(terminateEvent);
}
//安装函数
void InstallService()
{
SC_HANDLE schService;
SC_HANDLE schSCManager;
TCHAR szPath[512];
WCHAR* TServiceName=CharPtr2ACharPtr(ServiceName);
//Install Starting...
//获取模块文件路径名称
if(!GetModuleFileName(NULL,szPath,512))
{
return;
}
// open to SCM
schSCManager = OpenSCManager(
0, // pointer to machine name string
0, // pointer to database name string
SC_MANAGER_CREATE_SERVICE // type of access
);
if (!schSCManager)
{
//WriteToLog("OpenScManagerErrorID:"+GetLastError());
return;
}
// Install 新服务
schService = CreateService(
schSCManager, // handle to service control manager database
TServiceName, // pointer to name of service to start
TServiceName, // pointer to display name
SERVICE_ALL_ACCESS,// type of access to service
SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS ,// type of service
SERVICE_DEMAND_START,// when to start service
SERVICE_ERROR_NORMAL,// severity if service fails to start
szPath, // pointer to name of binary file
NULL, // pointer to name of load ordering group
NULL, // pointer to variable to get tag identifier
NULL, // pointer to array of dependency names
NULL, // pointer to account name of service
NULL // pointer to password for service account
);
if (!schService)
{
WriteToLog("服务安装出错了");
//WriteToLog("CreateServiceErrorID:"+GetLastError());
return;
}
else
{
WriteToLog("服务已安装Service installed\n");
}
// clean up
CloseServiceHandle(schService);
CloseServiceHandle(schSCManager);
WriteToLog("安装服务已结束Install Ended...\n");
}
//反安装函数
void UninstallService()
{
SC_HANDLE schService;
SC_HANDLE schSCManager;
BOOL success;
SERVICE_STATUS svcStatus;
WCHAR* TServiceName=CharPtr2ACharPtr(ServiceName);
WriteToLog("反安装开始Uninstall Starting...");
// open to SCM
schSCManager = OpenSCManager(
0,// pointer to machine name string
0,// pointer to database name string
SC_MANAGER_CREATE_SERVICE // type of access
);
if (!schSCManager)
{
//WriteToLog("OpenScManagerErrorID:"+GetLastError());
return;
}
//打开一个服务
schService = OpenService(
schSCManager, // handle to service control manager database
TServiceName, // pointer to name of service to start
SERVICE_ALL_ACCESS | DELETE// type of access to service
);
if (!schService)
{
WriteToLog("OpenServiceErrorID:"+GetLastError());
return;
}
//(if necessary)
success = QueryServiceStatus(schService, &svcStatus);
if (!success)
{
WriteToLog("In QueryServiceStatus ErrorID:"+GetLastError());
return;
}
if (svcStatus.dwCurrentState != SERVICE_STOPPED)
{
WriteToLog("Stopping service...");
success = ControlService(
schService, // handle to service
SERVICE_CONTROL_STOP, // control code
&svcStatus // pointer to service status structure
);
if (!success)
{
WriteToLog("In ControlServiceErrorID:"+GetLastError());
return;
}
}
//等待
do
{
QueryServiceStatus(schService,&svcStatus);
Sleep(500);
}while(SERVICE_STOP_PENDING==svcStatus.dwCurrentState);
if(SERVICE_STOPPED!=svcStatus.dwCurrentState)
{
WriteToLog("Failed to Stop Service ErrorID:"+GetLastError());
return;
}
//删除服务
success = DeleteService(schService);
if (success)
{
WriteToLog("Service removed\n");
}
else
{
WriteToLog("In DeleteService ErrorID:"+GetLastError());
return;
}
//Clean up
CloseServiceHandle(schService);
CloseServiceHandle(schSCManager);
WriteToLog("反安装结束Uninstal Ended...");
}
//实现 发送状态到SCM服务控制管理器
BOOL SendStatusToSCM (DWORD dwCurrentState,
DWORD dwWin32ExitCode,
DWORD dwServiceSpecificExitCode,
DWORD dwCheckPoint,
DWORD dwWaitHint)
{
BOOL success;
SERVICE_STATUS serviceStatus;
// Fill in all of the SERVICE_STATUS fields
serviceStatus.dwServiceType =SERVICE_WIN32_OWN_PROCESS;
serviceStatus.dwCurrentState = dwCurrentState;
// If in the process of something, then accept
// no control events, else accept anything
if (dwCurrentState == SERVICE_START_PENDING)
serviceStatus.dwControlsAccepted = 0;
else
serviceStatus.dwControlsAccepted =
SERVICE_ACCEPT_STOP |
SERVICE_ACCEPT_PAUSE_CONTINUE |
SERVICE_ACCEPT_SHUTDOWN;
// if a specific exit code is defines, set up
// the win32 exit code properly
if (dwServiceSpecificExitCode == 0)
serviceStatus.dwWin32ExitCode =dwWin32ExitCode;
else
serviceStatus.dwWin32ExitCode = ERROR_SERVICE_SPECIFIC_ERROR;
serviceStatus.dwServiceSpecificExitCode =dwServiceSpecificExitCode;
serviceStatus.dwCheckPoint = dwCheckPoint;
serviceStatus.dwWaitHint = dwWaitHint;
// Pass the status record to the SCM
//SERVICE_STATUS_HANDLE serviceStatusHandle;
success = SetServiceStatus (hStatus,&serviceStatus);
if (!success)
{
StopService();
}
return success;
}
//实现 服务终结函数
VOID terminate(DWORD error)
{
// if terminateEvent has been created, close it.
if (terminateEvent)
{
CloseHandle(terminateEvent);
}
// Send a message to the scm to tell about
// stop age
if (hStatus)
{
SendStatusToSCM(SERVICE_STOPPED, error,0, 0, 0);
}
// If the thread has started kill it off
if (threadHandle)
{
CloseHandle(threadHandle);
}
// Do not need to close serviceStatusHandle
}
//----------------------------------------------------
//通用函数的实现
//实现 字符串格式转换 char*转为WCHAR*
WCHAR* CharPtr2ACharPtr(const char* src)
{
// Convert to a wchar_t*
size_t srcsize = strlen(src) + 1;
size_t newsize = srcsize;
size_t convertedChars = 0;
wchar_t *wcstring;
wcstring=new wchar_t[newsize];
mbstowcs_s(&convertedChars, wcstring, srcsize, src, _TRUNCATE);
//wcscat_s(wcstring, L" (wchar_t *)");
//wcout << wcstring << endl;
return wcstring;
}
//实现 写日志
int WriteToLog(char* str)
{
FILE* log;
char* curdir;
if(CurExeDir==NULL)
{
CurExeDir=GetCurDir();
}
curdir=CurExeDir;
char* Logfile=UnionStr(curdir,LogFileName);
log=fopen(Logfile,"a+");
if(log==NULL) return 0;
fprintf(log,"%s\n",str);
fclose(log);
return 1;
}
//实现 获取当前可执行文件所在目录
char* GetCurDir(void)
{
//推荐用如下方法来得到当前程序所在目录比较安全：
//void _splitpath( const char *path, char *drive, char *dir, char *fname, char *ext );
//函数来分解开始提到的_pgmptr，然后再用
//void _makepath( char *path, const char *drive, const char *dir, const char *fname, const char *ext );
//函数来对分解后的路径进行组合。这两个函数结合起来功能强大，使用灵活，
//基本上所有的有关目录和路径方面的操作都可以搞定.
//(1)
char* szFilePath=new char[MAX_PATH];
memset(szFilePath,0,MAX_PATH);
GetModuleFileNameA(NULL,szFilePath,MAX_PATH);
//GetModuleFileNameW(NULL,szFilePath,MAX_PATH);
char* szfp=new char[strlen(szFilePath)+1];
strcpy(szfp,szFilePath);
int srcLen=strlen(szfp)+1;
char* Driver=new char[srcLen];
char* dir=new char[srcLen];
char* fn=new char[srcLen];
char* ext=new char[srcLen];
//
memset(Driver,0,srcLen);
memset(dir,0,srcLen);
memset(fn,0,srcLen);
memset(ext,0,srcLen);
//分解路径
_splitpath(szfp,Driver,dir,fn,ext);
char* dri=new char[strlen(Driver)+1];
strcpy(dri,Driver);
char* dir_t=new char[strlen(dir)+1];
strcpy(dir_t,dir);
//组合目录
int len=strlen(dri)+strlen(dir_t);
char* result=new char[len];
memset(result,0,len);
sprintf(result, "%s\\%s",dri,dir_t);
char* r=new char[strlen(result)+1];
strcpy(r,result);
return r;
}
//实现 替换字符串中特征字符串为指定字符串
int ReplaceStr(char *sSrc, char *sMatchStr, char *sReplaceStr)
{
int StringLen;
int MAX_MSG_LENGTH=strlen(sSrc)+1;
char* caNewString=new char[MAX_MSG_LENGTH];
char *FindPos = strstr(sSrc, sMatchStr);
if( (!FindPos) || (!sMatchStr) )
return -1;
while( FindPos )
{
memset(caNewString, 0, sizeof(caNewString));
StringLen = FindPos - sSrc;
strncpy(caNewString, sSrc, StringLen);
strcat(caNewString, sReplaceStr);
strcat(caNewString, FindPos + strlen(sMatchStr));
strcpy(sSrc, caNewString);
FindPos = strstr(sSrc, sMatchStr);
}
return 0;
}
//实现 合并字符串函数
char* UnionStr(const char* A,const char* B)
{
int len=strlen(A)+strlen(B);
char* result=new char[len];
memset(result,0,len);
sprintf(result, "%s%s",A,B);
char* r=new char[strlen(result)+1];
strcpy(r,result);
return r;
}
//实现 字符串是否相等函数
bool EqStr(const char* A,const char* B)
{
bool r=true;
char* t_A=new char[strlen(A)+1];
strcpy(t_A,A);
char* t_B=new char[strlen(B)+1];
strcpy(t_B,B);
int ALen=strlen(t_A);
int BLen=strlen(t_B);
if(ALen>=BLen) ALen=BLen;
{
for(int i=0;i<ALen;i++)
{
if(t_A[i]!=t_B[i])
{
r=false;
break;
}
}
}
return r;
}
//实现 是数字check函数
bool IsNumber(string str)
{
bool r=true;
const char* t=str.c_str();
int Len=strlen(t);
for(int i=0;i<Len;i++)
{
if(!isdigit(t[i]))
{
r=false;
break;
}
}
return r;
}
//实现 获取当前系统的年-月-日格式的时间
char* WinApi_Year_Month_Day()
{
SYSTEMTIME sys;
GetLocalTime(&sys);
char* s=new char[10];
sprintf(s,"%4d-%02d-%02d",sys.wYear,sys.wMonth,sys.wDay);
return s;
}
//实现 比较日期字符串函数
unsigned long Compare_To_Date(char* datetime1,char* datetime2)
{
//return 3表示日期格式不正确
//return 2表示datetime1>datetime2
//return 1表示datetime1=datetime2
//return 0表示datetime1<datetime2
unsigned short year1;
unsigned short month1;
unsigned short day1;
unsigned short year2;
unsigned short month2;
unsigned short day2;
unsigned long bValidate;
bValidate=Validate_Date(datetime1,&year1,&month1,&day1);
if(bValidate==false){
return 3;
}
bValidate=Validate_Date(datetime2,&year2,&month2,&day2);
if(bValidate==false){
return 3;
}
if(year1<year2) return 0;
if(year1>year2) return 2;
if(month1<month2) return 0;
if(month1>month2) return 2;
if(day1<day2) return 0;
if(day1>day2) return 2;
return 1;
}
//实现 比较当前日期字符串函数
unsigned long Compare_To_Current_Date(char* datetime){
unsigned short year;
unsigned short month;
unsigned short day;
unsigned long bValidate=Validate_Date(datetime,&year,&month,&day);
if(bValidate==false){
return 3;
}
SYSTEMTIME CurTime;
GetLocalTime(&CurTime);
if(CurTime.wYear>year)
return 2;
else if(CurTime.wYear<year)
return 0;
if(CurTime.wMonth>month)
return 2;
else if(CurTime.wMonth<month)
return 0;
if(CurTime.wDay>day)
return 2;
else if(CurTime.wDay<day)
return 0;
return 1;
}
//实现 验证日期字符串格式正确性函数
unsigned long Validate_Date(char* date,unsigned short * year,unsigned short * month,unsigned short * day){
if(strlen(date)<10) return false;
std:string s(date);
string _1=s.substr(4,1);
string _2=s.substr(7,1);
if(!EqStr(_1.c_str(),"-")) return false;
if(!EqStr(_2.c_str(),"-")) return false;
string szyear=s.substr(0,4);//yyyy-mm-dd
string szmonth=s.substr(5,2);
string szday=s.substr(8,2);
if(IsNumber(szyear)==false) return false;
if(IsNumber(szmonth)==false) return false;
if(IsNumber(szday)==false) return false;
unsigned long bResult;
try
{
*year=(unsigned short)atol(szyear.c_str());
*month=(unsigned short)atol(szmonth.c_str());
*day=(unsigned short)atol(szday.c_str());
bResult= true;
}
catch(...)
{
bResult=false;
}
return bResult;
}
//实现 获取机器码函数
char* GetMachineCode(void)
{
char* cpuid=GetCpuId();
char* ret;
for(int i=0;i<=2;i++)
{
ret=md5(cpuid);
cpuid=ret;
}
return ret;
}
//实现 获取Cpu编号函数
char* GetCpuId(void)
{
unsigned long s1,s2;
__asm
{
mov eax,01h
xor edx,edx
cpuid
mov s1,edx
mov s2,eax
}
//CString CPUID;
//CPUID.Format("%08X%08X",s1,s2);
//return CPUID;
char ret[255];
sprintf(ret,"%08X%08X",s1,s2);
char* lpmac=ret;
char* RetVal=new char[strlen(lpmac)+1];
strcpy(RetVal,lpmac);
return RetVal;
}
//实现 md5加密算法函数
char* md5(char* src)
{
char md5[64] = {0};
hex_md5(src, md5, 32);
char* ret;
ret=new char[strlen(md5)+1];
strcpy(ret,md5);
return ret;
}
//实现 是否有使用权的函数
bool IsLencense(void)
{
bool r=true;
char* cpuid=GetCpuId();
char* curdate=WinApi_Year_Month_Day();
//char* todate="2011-12-31";
char* tmpmachinecode=UnionStr(cpuid,todate);
char* regcode=md5(tmpmachinecode);
//printf("%s\n",regcode);printf("%s\n",code);
//WriteToLog(regcode);WriteToLog(code);
if(EqStr(regcode,code)==true)
{
printf("注册码有效(%s=%s)\n",code,regcode);
//时间是否过期
//curdate<=todate
//printf("curdate=%s\n",curdate);printf("todate=%s\n",todate);
int retval=Compare_To_Date(curdate,todate);
//printf("retval=%d\n",retval);
if(retval==0 || retval==1)
{
r=true;
}
else
{
r=false;
}
}
else
{
r=false;
printf("注册码无效(%s=%s)\n",code,regcode);
}
return r;
}
//实现 获取ServerConfig.ini参数函数
bool GetServerConfig_ini(void)
{
char* curdir;
if(CurExeDir==NULL)
{
CurExeDir=GetCurDir();
}
curdir=CurExeDir;
//获取ini文件中的数据
char* inifn="ServerConfig.ini";
int len=strlen(curdir)+strlen(inifn)+1;
char* inipath=new char[len];
sprintf(inipath, "%s\\%s", curdir, inifn);
//printf(inipath);
//---------
char* line;
filepath =inipath;
if(OpenFile()==true)
{
char* line;
while(!EndFile())
{
line=NextLine();
if(line!=NULL)
{
//printf("%s\n",line);
int lineLen=strlen(line);
if(lineLen>0)
{
if(line[0]!=';')
{
if(strstr(line,"LicenseServerHost=")!=NULL)
{
host=line;
}
if(strstr(line,"Port=")!=NULL)
{
port=line;
}
if(strstr(line,"ToDate=")!=NULL)
{
todate=line;
}
if(strstr(line,"LicenseCode=")!=NULL)
{
code=line;
}
}
}
}
}
CloseFile();
}
ReplaceStr(host,"LicenseServerHost=","");
ReplaceStr(port,"Port=","");
ReplaceStr(todate,"ToDate=","");
ReplaceStr(code,"LicenseCode=","");
printf("host->%s\n",host);
printf("port->%s\n",port);
printf("todate->%s\n",todate);
printf("code->%s\n",code);
return true;
}
//实现 打开文件操作
bool OpenFile(void)
{
char* path=filepath;
if((fp=fopen(path,"r"))==NULL)
{
printf("file cannot be opened\n");
return false;
}
return true;
}
//实现 检查文件是否结束函数
int EndFile(void)
{
return feof(fp);
}
//实现 读取下一行函数
char* NextLine()
{
char line[4000];
if(fgets(line,4000,fp)!=NULL)
{
char* ret=new char[strlen(line)+1];
strcpy(ret,line);
return ret;
}
return NULL;
}
//实现 关闭文件函数
int CloseFile(void)
{
if(fp!=NULL)
{
return fclose(fp);
}
return -1;
}
//---the---end---
162
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
