// EDGE2WEB.cpp : Defines the entry point for the console application.
//
#include "stdafx.h"
#include "smtp.h"
#if !defined(OTL_ORA10G_R2)
#define OTL_ORA10G_R2 // Compile OTL 4.0/OCI10gR2
#endif
#define OTL_ORA_SUBSCRIBE // Enable the otl_subscriber interface
// The following two #define's are required for
// the otl_subscriber interface to function
#define OTL_ORA_OCI_ENV_CREATE
#define OTL_ORA_OCI_ENV_CREATE_MODE (OCI_OBJECT|OCI_EVENTS)
#include <otlv4.h> // include the OTL 4.0 header file
char _chhmmss[32];
char* currenttime()
{
// return current system time
time_t now;
struct tm *newtime;
time(&now);
newtime = localtime(&now);
strftime(_chhmmss, 32, "%m/%d/%Y %H:%M:%S ", newtime);
return _chhmmss;
}
SERVICE_STATUS g_ServiceStatus = {0};
SERVICE_STATUS_HANDLE g_StatusHandle = NULL;
HANDLE g_ServiceStopEvent = INVALID_HANDLE_VALUE;
VOID WINAPI ServiceMain (DWORD argc, LPTSTR *argv);
VOID WINAPI ServiceCtrlHandler (DWORD);
DWORD WINAPI ServiceWorkerThread (LPVOID lpParam);
// global variables for service maintenance
const int nBufferSize = 500;
char g_pServiceName[nBufferSize+1];
char g_pExeFile[nBufferSize+1];
// global variables for logging to file
std::streambuf* g_pOldBuf = NULL;
std::ofstream g_out;
char g_pLogFile[nBufferSize+1];
// global variable for configuration file
char g_pCfgFile[nBufferSize+1];
#define SERVICE_NAME "EDGE2PSEWebsiteUpdater"
//////////////////////////////////////////////////////////////////////
//
// Uninstall
//
VOID UnInstall(char* pName)
{
SC_HANDLE schSCManager = OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS);
if (schSCManager==0)
{
long nError = GetLastError();
char pTemp[121];
sprintf(pTemp, "OpenSCManager failed, error code = %d", nError);
//WriteLog(pTemp);
}
else
{
SC_HANDLE schService = OpenService( schSCManager, pName, SERVICE_ALL_ACCESS);
if (schService==0)
{
long nError = GetLastError();
char pTemp[121];
sprintf(pTemp, "OpenService failed, error code = %d", nError);
//WriteLog(pTemp);
}
else
{
if(!DeleteService(schService))
{
char pTemp[121];
sprintf(pTemp, "Failed to delete service %s", pName);
//WriteLog(pTemp);
}
else
{
char pTemp[121];
sprintf(pTemp, "Service %s removed",pName);
//WriteLog(pTemp);
}
CloseServiceHandle(schService);
}
CloseServiceHandle(schSCManager);
}
}
//////////////////////////////////////////////////////////////////////
//
// Install
//
VOID Install(char* pPath, char* pName)
{
SC_HANDLE schSCManager = OpenSCManager( NULL, NULL, SC_MANAGER_CREATE_SERVICE);
if (schSCManager==0)
{
long nError = GetLastError();
char pTemp[121];
sprintf(pTemp, "OpenSCManager failed, error code = %d", nError);
//WriteLog(pTemp);
}
else
{
SC_HANDLE schService = CreateService
(
schSCManager, /* SCManager database */
pName, /* name of service */
pName, /* service name to display */
SERVICE_ALL_ACCESS, /* desired access */
SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS , /* service type */
SERVICE_AUTO_START, /* start type */
SERVICE_ERROR_NORMAL, /* error control type */
pPath, /* service's binary */
NULL, /* no load ordering group */
NULL, /* no tag identifier */
NULL, /* no dependencies */
NULL, /* LocalSystem account */
NULL
); /* no password */
if (schService==0)
{
long nError = GetLastError();
char pTemp[121];
sprintf(pTemp, "Failed to create service %s, error code = %d", pName, nError);
//WriteLog(pTemp);
}
else
{
char pTemp[121];
sprintf(pTemp, "Service %s installed", pName);
//WriteLog(pTemp);
CloseServiceHandle(schService);
}
CloseServiceHandle(schSCManager);
}
}
#ifdef _DEBUG
/* ControlHandler
Custom handler of Ctrl+C and Ctrl+Break, to ensure graceful system exit
*/
BOOL WINAPI ControlHandler(DWORD dwCtrlType)
{
switch( dwCtrlType ) {
case CTRL_BREAK_EVENT: // use Ctrl+C or Ctrl+Break to simulate
case CTRL_C_EVENT: // SERVICE_CONTROL_STOP in debug mode
case CTRL_CLOSE_EVENT:
case CTRL_LOGOFF_EVENT:
case CTRL_SHUTDOWN_EVENT:
::SetEvent(g_ServiceStopEvent);
return TRUE;
}
return FALSE;
}
#endif
int main(int argc, char* argv[])
{
// initialize MFC
if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0))
{
// TODO: change error code to suit your needs
//_tprintf(_T("Fatal Error: MFC initialization failed\n"));
return 1;
}
// Initialize WinSock
WSADATA wsaData;
WORD wVer = MAKEWORD(2,2);
WSAStartup(wVer,&wsaData);
// initialize variables for .exe, .cfg, and .log file names
char pModuleFile[nBufferSize+1];
DWORD dwSize = GetModuleFileName(NULL,pModuleFile,nBufferSize);
pModuleFile[dwSize] = 0;
if(dwSize>4&&pModuleFile[dwSize-4]=='.')
{
sprintf(g_pExeFile,"%s",pModuleFile);
pModuleFile[dwSize-4] = 0;
sprintf(g_pCfgFile,"%s.cfg",pModuleFile);
// logfile is of the form exename_YYYYMM.LOG
// return current system time
time_t now;
char yyyymm[8];
struct tm *newtime;
time(&now);
newtime = localtime(&now);
strftime(yyyymm, 8, "%Y%m", newtime);
sprintf(g_pLogFile,"%s_%s.log",pModuleFile, yyyymm);
} else {
printf("Invalid module file name: %s\r\n", pModuleFile);
return 2;
}
// Read the service name from the configuration file
GetPrivateProfileString("Service Settings","ServiceName",SERVICE_NAME,
g_pServiceName,nBufferSize,g_pCfgFile);
#ifndef _DEBUG
// uninstall service if switch is "-u"
if(argc==2&&_stricmp("-u",argv[1])==0)
{
UnInstall(g_pServiceName);
return 0;
}
// install service if switch is "-i"
else if(argc==2&&_stricmp("-i",argv[1])==0)
{
Install(g_pExeFile, g_pServiceName);
return 0;
}
#endif
// Redirect cout as early as possible so we can use it for logging early errors
// redirect cout to file
g_pOldBuf = cout.rdbuf();
g_out.open(g_pLogFile, ios::out|ios::binary|ios::app);
std::cout.rdbuf(g_out.rdbuf()); // << this is the redirect command!!
cout << currenttime() << "========================" << endl << flush;
cout << currenttime() << "Service [" << g_pServiceName << "] has started." << endl << flush;
if (!AfxSocketInit()) {
cout << currenttime() << "AfxSocketInit failed. Email notification will not work." << endl;
}
#ifndef _DEBUG
SERVICE_TABLE_ENTRY ServiceTable[] =
{
{SERVICE_NAME, (LPSERVICE_MAIN_FUNCTION) ServiceMain},
{NULL, NULL}
};
if (StartServiceCtrlDispatcher (ServiceTable) == FALSE)
{
cout << currenttime() << "StartServiceCtrlDispatcher returned error." << endl << flush;
// restore cout
if(g_pOldBuf) {
cout.rdbuf(g_pOldBuf);
}
return GetLastError ();
}
// you don't get here unless the service is shutdown
cout << currenttime() << "Service [" << g_pServiceName << "] has been stopped." << endl << flush;
//////////////////////////////////////////////////////
// CLEAN-UP code goes here
Sleep(3000);
#else
g_ServiceStopEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
BOOL bCtrlHandled = SetConsoleCtrlHandler(ControlHandler, TRUE);
HANDLE hThread = CreateThread (NULL, 0, ServiceWorkerThread, NULL, 0, NULL);
WaitForSingleObject(hThread, INFINITE);
if(bCtrlHandled) {
SetConsoleCtrlHandler(ControlHandler, FALSE);
}
cout << currenttime() << "ServiceWorkerThread stopped." << endl << flush;
CloseHandle(g_ServiceStopEvent);
#endif
// restore cout
if(g_pOldBuf) {
cout.rdbuf(g_pOldBuf);
}
return 0;
}
VOID WINAPI Servic
