/
CallMeSleep.cpp
105 lines (84 loc) · 2.92 KB
/
CallMeSleep.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
// CallMeSleep.cpp : Defines the class behaviors for the application.
//
#include "stdafx.h"
#include "CallMeSleep.h"
#include "CallMeSleepDlg.h"
#include "ServiceModule.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
extern "C" time_t getTime(int hour, int min, int sec);
/////////////////////////////////////////////////////////////////////////////
// CCallMeSleepApp
BEGIN_MESSAGE_MAP(CCallMeSleepApp, CWinApp)
//{{AFX_MSG_MAP(CCallMeSleepApp)
// NOTE - the ClassWizard will add and remove mapping macros here.
// DO NOT EDIT what you see in these blocks of generated code!
//}}AFX_MSG
ON_COMMAND(ID_HELP, CWinApp::OnHelp)
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CCallMeSleepApp construction
CCallMeSleepApp::CCallMeSleepApp()
{
// TODO: add construction code here,
// Place all significant initialization in InitInstance
}
/////////////////////////////////////////////////////////////////////////////
// The one and only CCallMeSleepApp object
CCallMeSleepApp theApp;
/////////////////////////////////////////////////////////////////////////////
// CCallMeSleepApp initialization
BOOL CCallMeSleepApp::InitInstance()
{
AfxEnableControlContainer();
// Standard initialization
// If you are not using these features and wish to reduce the size
// of your final executable, you should remove from the following
// the specific initialization routines you do not need.
#ifdef _AFXDLL
Enable3dControls(); // Call this when using MFC in a shared DLL
#else
Enable3dControlsStatic(); // Call this when linking to MFC statically
#endif
// m_lpCmdLine = "22:00:00-06:00:00"; //for test
if(*m_lpCmdLine != 0) {
int hour = 0, min = 0, sec = 0;
int hour2 = 0, min2 = 0, sec2 = 0;
_stscanf(m_lpCmdLine, _T("%d:%d:%d-%d:%d:%d"), &hour, &min, &sec, &hour2, &min2, &sec2);
time_t start = getTime(hour, min, sec), end = getTime(hour2, min2, sec2);
time_t now = time(NULL);
struct tm *tm_now = localtime(&now);
if(hour > hour2 || (hour == hour2 && min > min2) || (hour == hour2 && min == min2 && sec > sec2))
{
if(tm_now->tm_hour <= hour2) {
start -= 24*60*60;
} else {
end += 24*60*60;
}
}
_Module.SetTime(start, end);
_Module.Init(m_hInstance, _T("CallMeSleep"), NULL);
_Module.m_bService = TRUE;
_Module.Start();
} else {
CCallMeSleepDlg dlg;
m_pMainWnd = &dlg;
int nResponse = dlg.DoModal();
if (nResponse == IDOK)
{
// TODO: Place code here to handle when the dialog is
// dismissed with OK
}
else if (nResponse == IDCANCEL)
{
// TODO: Place code here to handle when the dialog is
// dismissed with Cancel
}
}
// Since the dialog has been closed, return FALSE so that we exit the
// application, rather than start the application's message pump.
return FALSE;
}