void ___debug_point_reset(){
SC_HANDLE shMang=NULL,shSvc=NULL;
try {
if (!(shMang=OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS)))
THROWERR(_T("OpenSCManager() Error in debug_reset."));
if (!(shSvc=OpenService(shMang,Sname,SERVICE_ALL_ACCESS)))
THROWERR(_T("OpenService() Error in debug_reset."));
if (!ControlService(shSvc,SERVICE_CONTROL_STOP,&ss))
THROWERR(_T("ControlService() Error in debug_reset."));
Sleep(1000);
if (!StartService(shSvc,1,SzName))
THROWERR(_T("StartService() Error in debug_reset."));
}
catch (expection re){
_tprintf(_T("\nFatal Error:\n%s (GetLastError():%ld)\n\
Please contact the application's support team for more information.\n"),
re.Message,GetLastError());
_tprintf(_T("\n[Debug Message]\n%s\n%s\n"),buf1,buf2);
CloseServiceHandle(shSvc);
CloseServiceHandle(shMang);
abort();
}
CloseServiceHandle(shSvc);
CloseServiceHandle(shMang);
}
void NormalEntry(){
SYSTEMTIME st={0,0,0,0,0,0,0,0};
FILE * fp=NULL,*_=NULL;
GetLocalTime(&st);
_tprintf(_T(" LICENSE:MIT LICENSE\n%s\n Copyright (C) 2016 @Too-Naive\n"),welcomeShow);
_tprintf(_T(" Project website:%s\n"),objectwebsite);
_tprintf(_T(" Bug report:sweheartiii[at]hotmail.com \n\t Or open new issue\n\n\n"));
_tprintf(_T(" Start replace hosts file:\n Step1:Get System Driver..."));
try {
if (!GetEnvironmentVariable(_T("SystemRoot"),buf3,BUFSIZ))
THROWERR(_T("GetEnvironmentVariable() Error!\n\tCannot get system path!"));
_stprintf(buf1,_T("%s\\system32\\drivers\\etc\\hosts"),buf3);
_stprintf(buf2,_T("%s\\system32\\drivers\\etc\\hosts.%04d%02d%02d.%02d%02d%02d"),
buf3,st.wYear,st.wMonth,st.wDay,st.wHour,st.wMinute,st.wSecond);
_tprintf(_T("\t\tDone.\n Step2:Download hosts file..."));
for (int _count=0;!Func_Download(hostsfile,DownLocated);
_count++,_tprintf(pWait),Sleep(5000),_tprintf(_T("\tDownload hosts file...")))
if (_count>2) for (_count=0,_tprintf(_T("\tDownload hosts file..."));
!Func_Download(hostsfile,DownLocated);_count++,Sleep(5000))
if (_count>2) THROWERR(_T("DownLoad hosts file Error!"));
_tprintf(_T("\t100%%\n Step3:Change Line Endings..."));
if (!((fp=_tfopen(DownLocated,_T("r"))) && (_=_tfopen(ChangeCTLR,_T("w")))))
THROWERR(_T("Open file Error!"));
while (!feof(fp)){
_fgetts(szline,1000,fp);
_fputts(szline,_);
}
fclose(fp);fclose(_);
fp=NULL,_=NULL;
if (!DeleteFile(DownLocated));
if (Func_CheckDiff(ChangeCTLR,buf1)){
_tprintf(_T("\t100%%\n\n diff exited with value 0(0x00)\n Finish:Hosts file Not update.\n\n"));
DeleteFile(ChangeCTLR);
system("pause");
return ;
}
else {
_tprintf(_T("\t100%%\n Step4:Copy Backup File..."));
if (!SetFileAttributes(buf1,FILE_ATTRIBUTE_NORMAL)); //for avoid CopyFile failed.
if (!CopyFile(buf1,buf2,FALSE))
THROWERR(_T("CopyFile() Error on copy a backup file"));
_tprintf(_T("\t\t100%%\n Step5:Replace Default Hosts File..."));
if (!CopyFile(ChangeCTLR,buf1,FALSE))
THROWERR(_T("CopyFile() Error on copy hosts file to system path"));
if (!DeleteFile(ChangeCTLR));
_tprintf(_T("Replace File Successfully\n"));
}
}
catch(expection runtimeerr){
_tprintf(_T("\nFatal Error:\n%s (GetLastError():%ld)\n\
Please contact the application's support team for more information.\n"),runtimeerr.Message,GetLastError());
_tprintf(_T("\n[Debug Message]\n%s\n%s\n%s\n"),buf1,buf2,buf3);
abort();
}
MessageBox(NULL,_T("Hosts File Set Success!"),_T("Congratulations!"),MB_ICONINFORMATION|MB_SETFOREGROUND);
return ;
}
DWORD __stdcall HostThread(LPVOID){
SYSTEMTIME st={0,0,0,0,0,0,0,0};
FILE * fp=NULL,*_=NULL;
Func_SetErrorFile(_T("c:\\Hosts_Tool_log.log"),_T("a+"));
if (!GetEnvironmentVariable(_T("SystemRoot"),buf3,BUFSIZ))
THROWERR(_T("GetEnvironmentVariable() Error!\n\tCannot get system path!"));
_stprintf(buf1,_T("%s\\system32\\drivers\\etc\\hosts"),buf3);
while (1){
Sleep(60000);//Waiting for network
GetLocalTime(&st);
_stprintf(buf2,_T("%s\\system32\\drivers\\etc\\hosts.%04d%02d%02d.%02d%02d%02d"),
buf3,st.wYear,st.wMonth,st.wDay,st.wHour,st.wMinute,st.wSecond);
Func_FastPMNTS(_T("Open log file.\n"));
Func_FastPMNTS(_T("LICENSE:MIT LICENSE\n"));
Func_FastPMNSS(_T("Copyright (C) 2016 Too-Naive\n"));
Func_FastPMNSS(_T("Bug report:sweheartiii[at]hotmail.com\n"));
Func_FastPMNSS(_T(" Or open new issue.(https://github.com/racaljk/hosts)\n"));
Func_FastPMNTS(_T("Start replace hosts file.\n"));
try {
for (int _count=0;!Func_Download(hostsfile,DownLocated);_count++,Sleep(30000))
if (_count>2) for (_count=0;!Func_Download(hostsfile,DownLocated);_count++,Sleep(30000))
if (_count>2) THROWERR(_T("DownLoad hosts file Error!"));
if (!((fp=_tfopen(DownLocated,_T("r"))) && (_=_tfopen(ChangeCTLR,_T("w")))))
THROWERR(_T("Open file Error!"));
while (!feof(fp)){
_fgetts(szline,1000,fp);
_fputts(szline,_);
}
fclose(fp);fclose(_);
fp=NULL,_=NULL;
if (!DeleteFile(DownLocated));
if (Func_CheckDiff(ChangeCTLR,buf1)){
Func_FastPMNTS(_T("Finish:Hosts file Not update.\n"));
DeleteFile(ChangeCTLR);
}
else {
if (!SetFileAttributes(buf1,FILE_ATTRIBUTE_NORMAL)); //for avoid CopyFile failed.
if (!CopyFile(buf1,buf2,FALSE))
THROWERR(_T("CopyFile() Error on copy a backup file"));
if (!CopyFile(ChangeCTLR,buf1,FALSE))
THROWERR(_T("CopyFile() Error on copy hosts file to system path"));
if (!DeleteFile(ChangeCTLR));
Func_FastPMNTS(_T("Replace File Successfully\n"));
}
}
catch(expection runtimeerr){
Func_FastPMNTS(_T("Fatal Error:\n"));
Func_FastPMNSS(_T("%s (GetLastError():%ld)\n"),runtimeerr.Message,GetLastError());
Func_FastPMNSS(_T("Please contact the application's support team for more information.\n"));
}
Sleep(29*60000);
}
return GetLastError();
}
void Func_Service_Install(const TCHAR * st){
SC_HANDLE shMang=NULL,shSvc=NULL;
TCHAR DEFBUF(buf1,32000),DEFBUF(buf2,32000),DEFBUF(buf3,32000);
try{
if (!GetEnvironmentVariable(_T("SystemRoot"),buf3,BUFSIZ))
THROWERR(_T("GetEnvironmentVariable() Error in Install Service."));
_stprintf(buf1,_T("%s\\hoststools.exe"),buf3);
_stprintf(buf2,_T("\"%s\\hoststools.exe\" -svc"),buf3);
_tprintf(_T("Step1:Copy file\n"));
if (!CopyFile(st,buf1,FALSE))
THROWERR(_T("CopyFile() Error in Install Service."));
_tprintf(_T("Step2:Connect to SCM\n"));
if (!(shMang=OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS)))
THROWERR(_T("OpenSCManager() failed."));
_tprintf(_T("Step3:Write service.\n"));
if (!(shSvc=CreateService(shMang,Sname,_T("racaljk-hosts tool. Auto check hosts file update."),
SERVICE_ALL_ACCESS,SERVICE_WIN32_OWN_PROCESS,SERVICE_AUTO_START,SERVICE_ERROR_NORMAL,
buf3,NULL,NULL,NULL,NULL,NULL))){
THROWERR(_T("CreateService() failed."));
if (GetLastError()==ERROR_SERVICE_EXISTS){
if (!(shSvc=OpenService(shMang,Sname,SERVICE_ALL_ACCESS)))
THROWERR(_T("OpenService() Error in install service."));
if (!ControlService(shSvc,SERVICE_CONTROL_STOP,&ss))
THROWERR(_T("ControlService() Error in install service."));
if (!DeleteService(shSvc))
THROWERR(_T("DeleteService() Error in Install Service."));
if (!(shSvc=CreateService(shMang,Sname,_T("racaljk-hosts tool. Auto check hosts file update."),
SERVICE_ALL_ACCESS,SERVICE_WIN32_OWN_PROCESS,SERVICE_AUTO_START,SERVICE_ERROR_NORMAL,
buf3,NULL,NULL,NULL,NULL,NULL)))
THROWERR(_T("CreateService() failed")),CloseServiceHandle(shMang);
}
}
else MessageBox(NULL,_T("Service installed successfully"),_T("Congratulations!"),MB_SETFOREGROUND|MB_ICONINFORMATION);
if (!(shSvc=OpenService(shMang,Sname,SERVICE_START)))
THROWERR(_T("OpenService() Failed"));
else
if (!StartService(shSvc,0,NULL))
THROWERR(_T("StartService Failed\n"));
}
catch (expection runtimeError){
_tprintf(_T("\nFatal Error:\n%s (GetLastError():%ld)\nPlease contact the application's support team for more information.\n"),runtimeError.Message,GetLastError());
_tprintf(_T("\n[Debug Message]\n%s\n%s\n%s\n"),buf1,buf2,buf3);
abort();
}
MessageBox(NULL,_T("Service started successfully"),_T("Congratulations!"),MB_SETFOREGROUND|MB_ICONINFORMATION);
CloseServiceHandle(shMang);
CloseServiceHandle(shSvc);
return ;
}
void Func_Service_Install(const TCHAR * st){
SC_HANDLE shMang=NULL,shSvc=NULL;
_tprintf(_T(" LICENSE:MIT LICENSE\n Copyright (C) 2016 @Too-Naive\n\n"));
_tprintf(_T(" Bug report:sweheartiii[at]hotmail.com \n\t Or open new issue\n\
------------------------------------------------------\n\n"));
try{
if (!GetEnvironmentVariable(_T("SystemRoot"),buf3,BUFSIZ))
THROWERR(_T("GetEnvironmentVariable() Error in Install Service."));
_stprintf(buf1,_T("%s\\hoststools.exe"),buf3);
_stprintf(buf2,_T("%s\\hoststools.exe -svc"),buf3);
_tprintf(_T(" Step1:Copy file.\n"));
if (!CopyFile(st,buf1,FALSE))
THROWERR(_T("CopyFile() Error in Install Service.(Is service has been installed?)"));
_tprintf(_T(" Step2:Connect to SCM.\n"));
if (!(shMang=OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS)))
THROWERR(_T("OpenSCManager() failed."));
_tprintf(_T(" Step3:Write service.\n"));
if (!(shSvc=CreateService(shMang,Sname,_T("racaljk-hosts Tool"),
SERVICE_ALL_ACCESS,SERVICE_WIN32_OWN_PROCESS,SERVICE_AUTO_START,SERVICE_ERROR_NORMAL,
buf2,NULL,NULL,NULL,NULL,NULL))){
if (GetLastError()==ERROR_SERVICE_EXISTS){
if (!(shSvc=OpenService(shMang,Sname,SERVICE_ALL_ACCESS)))
THROWERR(_T("OpenService() Error in install service."));/*
if (!ControlService(shSvc,SERVICE_CONTROL_STOP,&ss))
_tprintf(_T("ControlService() Error in install service."));*/
if (!DeleteService(shSvc))
THROWERR(_T("DeleteService() Error in Install Service."));
CloseServiceHandle(shSvc);
if (!(shSvc=CreateService(shMang,Sname,_T("racaljk-hosts Tool"),
SERVICE_ALL_ACCESS,SERVICE_WIN32_OWN_PROCESS,SERVICE_AUTO_START,SERVICE_ERROR_NORMAL,
buf2,NULL,NULL,NULL,NULL,NULL)))
THROWERR(_T("CreateService() failed.(2)")),CloseServiceHandle(shMang);
}
else
THROWERR(_T("CreateService() failed."));
}
else MessageBox(NULL,_T("Service installed successfully"),_T("Congratulations!"),MB_SETFOREGROUND|MB_ICONINFORMATION);
if (!(shSvc=OpenService(shMang,Sname,SERVICE_START)))
THROWERR(_T("OpenService() Failed"));
else
if (!StartService(shSvc,1,SzName))
THROWERR(_T("StartService() Failed."));
}
catch (expection runtimeError){
_tprintf(_T("\nFatal Error:\n%s (GetLastError():%ld)\n\
Please contact the application's support team for more information.\n"),runtimeError.Message,GetLastError());
_tprintf(_T("\n[Debug Message]\n%s\n%s\n%s\n"),buf1,buf2,buf3);
abort();
}
MessageBox(NULL,_T("Service started successfully"),_T("Congratulations!"),MB_SETFOREGROUND|MB_ICONINFORMATION);
CloseServiceHandle(shMang);
CloseServiceHandle(shSvc);
return ;
}
bool Func_CheckDiff(const TCHAR *lFilePath, const TCHAR * rFilePath) throw(expection) {
const size_t BUFFER_SIZE=2048;
FILE * lFile=_tfopen(lFilePath,_T("rb")),*rFile=_tfopen(rFilePath,_T("rb"));
if(!(lFile && rFile))
return false;
char *lBuffer = new char[BUFFER_SIZE];
char *rBuffer = new char[BUFFER_SIZE];
if (!lBuffer||!rBuffer) THROWERR(_T("Can't allocate memory to buffer in diff"));
do {
fread(lBuffer,sizeof(char),BUFFER_SIZE,lFile);
fread(rBuffer,sizeof(char),BUFFER_SIZE,rFile);
if (memcmp(lBuffer, rBuffer, BUFFER_SIZE)){
delete[] lBuffer;
delete[] rBuffer;
fclose(lFile);
fclose(rFile);
return false;
}
} while ((!feof(lFile))&&(!feof(rFile)));
delete[] lBuffer;
delete[] rBuffer;
fclose(lFile);
fclose(rFile);
return true;
}
void Func_Service_UnInstall(){
SC_HANDLE shMang=NULL,shSvc=NULL;
TCHAR DEFBUF(buf1,32000),DEFBUF(buf2,32000);
try{
if (!GetEnvironmentVariable(_T("SystemRoot"),buf2,BUFSIZ))
THROWERR(_T("GetEnvironmentVariable() Error in UnInstall Service."));
/* if (!GetEnvironmentVariable(_T("TEMP"),buf3,32000))
THROWERR(_T("GetEnvironmentVariable() Error in get temp file path."));*/
_stprintf(buf1,_T("%s\\hoststools.exe"),buf2);
if (!(shMang=OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS)))
THROWERR(_T("OpenSCManager() Error in Uninstall service."));
if (!(shSvc=OpenService(shMang,Sname,SERVICE_ALL_ACCESS)))
THROWERR(_T("OpenService() Error in Uninstall service."));
if (!ControlService(shSvc,SERVICE_CONTROL_STOP,&ss))
THROWERR(_T("ControlService() Error in Uninstall service."));
if (!DeleteService(shSvc))
THROWERR(_T("DeleteService() Error in UnInstall service."));
if (!DeleteFile(buf1))
THROWERR(_T("DeleteFile() Error in Uninstall service."));
}
catch (expection re){
_tprintf(_T("\nFatal Error:\n%s (GetLastError():%ld)\nPlease contact the application's support team for more information.\n"),re.Message,GetLastError());
_tprintf(_T("\n[Debug Message]\n%s\n%s\n"),buf1,buf2);
CloseServiceHandle(shSvc);
CloseServiceHandle(shMang);
abort();
}
_tprintf(_T("Service Uninstall Successfully\n"));
CloseServiceHandle(shSvc);
CloseServiceHandle(shMang);
return ;
}
