📄 timeclientdlg.cpp
字号:
// TimeClientDlg.cpp : 实现文件
//
#include "stdafx.h"
#include "TimeClient.h"
#include "TimeClientDlg.h"
#include ".\timeclientdlg.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
#define WM_SHOWTASK WM_USER+100
// 用于应用程序“关于”菜单项的 CAboutDlg 对话框
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// 对话框数据
enum { IDD = IDD_ABOUTBOX };
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持
// 实现
protected:
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()
// CTimeClientDlg 对话框
CTimeClientDlg::CTimeClientDlg(CWnd* pParent /*=NULL*/)
: CDialog(CTimeClientDlg::IDD, pParent)
, m_freq(0)
, m_ishour(FALSE)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CTimeClientDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
DDX_Control(pDX, IDC_BUTTON_CONN, m_conn);
DDX_Control(pDX, IDC_BUTTON_DISCONN, m_disconn);
DDX_Control(pDX, IDC_EDIT_TIME, m_time);
DDX_Control(pDX, IDC_EDIT_STATE, m_state);
DDX_Control(pDX, IDC_EDIT_IP, m_ipaddr);
DDX_Text(pDX, IDC_EDIT4, m_freq);
DDX_Control(pDX, IDC_BUTTON_SYNC, m_sync);
DDX_Radio(pDX, IDC_RADIO1, m_ishour);
}
BEGIN_MESSAGE_MAP(CTimeClientDlg, CDialog)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//}}AFX_MSG_MAP
ON_BN_CLICKED(IDC_BUTTON_CONN, OnBnClickedButtonConn)
ON_BN_CLICKED(IDC_BUTTON_DISCONN, OnBnClickedButtonDisconn)
ON_BN_CLICKED(IDC_BUTTON_SYNC, OnBnClickedButtonSync)
ON_BN_CLICKED(IDC_BUTTON_MIN, OnBnClickedButtonMin)
ON_BN_CLICKED(IDC_BUTTON_SAVE, OnBnClickedButtonSave)
ON_BN_CLICKED(IDCANCEL, OnBnClickedCancel)
ON_WM_DESTROY()
ON_WM_TIMER()
ON_MESSAGE(WM_SHOWTASK,onShowTask)
END_MESSAGE_MAP()
// CTimeClientDlg 消息处理程序
BOOL CTimeClientDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// 将\“关于...\”菜单项添加到系统菜单中。
// IDM_ABOUTBOX 必须在系统命令范围内。
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
ASSERT(IDM_ABOUTBOX < 0xF000);
CMenu* pSysMenu = GetSystemMenu(FALSE);
if (pSysMenu != NULL)
{
CString strAboutMenu;
strAboutMenu.LoadString(IDS_ABOUTBOX);
if (!strAboutMenu.IsEmpty())
{
pSysMenu->AppendMenu(MF_SEPARATOR);
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
}
}
// 设置此对话框的图标。当应用程序主窗口不是对话框时,框架将自动
// 执行此操作
SetIcon(m_hIcon, TRUE); // 设置大图标
SetIcon(m_hIcon, FALSE); // 设置小图标
// TODO: 在此添加额外的初始化代码
ee=0;
CTime m_nTimer,autoconn,timesync;
m_nTimer=SetTimer(1,1000,NULL);
autoconn=SetTimer(3,20000,NULL);
m_ipaddr.SetWindowText("127.0.0.1");
m_state.SetWindowText("请等待自动连接或手动连接");
LPCTSTR SubPath;
HKEY hKey;
LPBYTE freq;
LPBYTE hour;
freq= new BYTE[100];
hour=new BYTE[100];
char ipaddress[255];
ZeroMemory(&ipaddress,sizeof(ipaddress));
DWORD type=REG_SZ;
DWORD cbData=100;
SubPath="SOFTWARE\\HDBS";
m_ipaddr.GetWindowText(ipaddress,30);
if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,SubPath,0,KEY_READ,&hKey)==ERROR_SUCCESS)
{
RegQueryValueEx(hKey,"IP",0,&type,(LPBYTE)ipaddress,&cbData);
RegQueryValueEx(hKey,"Freq",0,&type,freq,&cbData);
RegQueryValueEx(hKey,"Base",0,&type,hour,&cbData);
m_ipaddr.SetWindowText(ipaddress);
m_freq=atoi((const char*)(LPCTSTR)(CString)freq);
if(atoi((const char*)(LPCTSTR)(CString)hour)==0)
m_ishour=false;
else
m_ishour=true;
long timefreq;
if(m_ishour)
timefreq=m_freq*60*60*1000;
else
timefreq=m_freq*60*1000;
timesync=SetTimer(2,timefreq,NULL);
UpdateData(FALSE);
}
else
{
if(RegCreateKeyEx(HKEY_LOCAL_MACHINE,SubPath,0,NULL,REG_OPTION_NON_VOLATILE,KEY_WRITE,NULL,&hKey,NULL)!=ERROR_SUCCESS ||
RegSetValueEx(hKey,"Freq",0,REG_SZ,(const BYTE*)"30",15)!=ERROR_SUCCESS ||
RegSetValueEx(hKey,"Base",0,REG_SZ,(const BYTE*)"0",3)!=ERROR_SUCCESS ||
RegSetValueEx(hKey,"IP",0,REG_SZ,(const BYTE*)(LPCTSTR)ipaddress,20)!=ERROR_SUCCESS)
{
AfxMessageBox("无法写入注册表...");
//exit(0);
}
m_freq=1;
m_ishour=true;
timesync=SetTimer(2,1800000,NULL);
UpdateData(FALSE);
}
return TRUE; // 除非设置了控件的焦点,否则返回 TRUE
}
void CTimeClientDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialog::OnSysCommand(nID, lParam);
}
if(SC_MINIMIZE==nID)
{
toTray(TRUE);
return;
}
}
// 如果向对话框添加最小化按钮,则需要下面的代码
// 来绘制该图标。对于使用文档/视图模型的 MFC 应用程序,
// 这将由框架自动完成。
void CTimeClientDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // 用于绘制的设备上下文
SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
// 使图标在工作矩形中居中
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// 绘制图标
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
//当用户拖动最小化窗口时系统调用此函数取得光标显示。
HCURSOR CTimeClientDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
void CTimeClientDlg::OnBnClickedButtonConn()
{
// TODO: 在此添加控件通知处理程序代码
char ipaddress[255];
ZeroMemory(&ipaddress,sizeof(ipaddress));
m_ipaddr.GetWindowText(ipaddress,30);
cli.sin_addr.s_addr=inet_addr(ipaddress);
cli.sin_family=AF_INET;
cli.sin_port=123;
//创建socket
clisock=socket(AF_INET,SOCK_STREAM,0);
ee=1;
//启动线程
UpdateData();
long timefreq;
if(m_ishour)
timefreq=m_freq*60*60*1000;
else
timefreq=m_freq*60*1000;
CTime timesync;
timesync=SetTimer(2,timefreq,NULL);
AfxBeginThread(thread,0);
}
void CTimeClientDlg::OnBnClickedButtonDisconn()
{
// TODO: 在此添加控件通知处理程序代码
ee=0;
closesocket(clisock);
m_state.SetWindowText("未连接");
m_disconn.ShowWindow(FALSE);
m_conn.ShowWindow(TRUE);
m_sync.EnableWindow(FALSE);
UpdateData();
CTime autoconn;
autoconn=SetTimer(2,20000,NULL);
}
void CTimeClientDlg::OnBnClickedButtonSync()
{
// TODO: 在此添加控件通知处理程序代码
Timesync ts;
GetLocalTime(&ts.sendbycli);
send(clisock,(char *)&ts,sizeof(ts),0);
}
void CTimeClientDlg::OnBnClickedButtonMin()
{
// TODO: 在此添加控件通知处理程序代码
toTray(TRUE);
}
void CTimeClientDlg::OnBnClickedButtonSave()
{
// TODO: 在此添加控件通知处理程序代码
UpdateData();
HKEY hKey;
LPCTSTR SubPath;
SubPath="SOFTWARE\\HDBS";
int ish;
if(m_ishour)
ish=1;
else
ish=0;
CString sfreq;
CString sish;
char ipaddress[255];
ZeroMemory(&ipaddress,sizeof(ipaddress));
m_ipaddr.GetWindowText(ipaddress,30);
sfreq.Format("%d",m_freq);
sish.Format("%d",ish);
if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,SubPath,0,KEY_READ,&hKey)==ERROR_SUCCESS)
{
RegDeleteKey(HKEY_LOCAL_MACHINE,SubPath);
RegCreateKeyEx(HKEY_LOCAL_MACHINE,SubPath,0,NULL,REG_OPTION_NON_VOLATILE,KEY_WRITE,NULL,&hKey,NULL);
if(RegSetValueEx(hKey,"Freq",0,REG_SZ,(const BYTE*)(LPCTSTR)sfreq,10)!=ERROR_SUCCESS ||
RegSetValueEx(hKey,"Base",0,REG_SZ,(const BYTE*)(LPCTSTR)sish,10)!=ERROR_SUCCESS ||
RegSetValueEx(hKey,"IP",0,REG_SZ,(const BYTE*)(LPCTSTR)ipaddress,20)!=ERROR_SUCCESS)
{
AfxMessageBox("写入注册表出错。");
}
else
{
AfxMessageBox("保存成功。");
long timefreq;
if(m_ishour)
timefreq=m_freq*60*60*1000;
else
timefreq=m_freq*60*1000;
CTime timesync;
timesync=SetTimer(2,timefreq,NULL);
}
}
else
{
AfxMessageBox("写入注册表出错。");
}
}
void CTimeClientDlg::OnBnClickedCancel()
{
// TODO: 在此添加控件通知处理程序代码
ee=0;
closesocket(clisock);
KillTimer(1);
KillTimer(2);
KillTimer(3);
OnCancel();
}
UINT thread(LPVOID v)
{
Timesync ts;
int s=1,addcount=0;
CTimeClientDlg *dlg=(CTimeClientDlg*) AfxGetApp()->GetMainWnd();
dlg->m_conn.ShowWindow(FALSE);
dlg->m_disconn.ShowWindow(TRUE);
//连接到服务器
while(connect(dlg->clisock,(sockaddr*)&(dlg->cli),sizeof(dlg->cli)) && dlg->ee!=0)
{
dlg->m_state.SetWindowText("等待.....");
dlg->m_disconn.SetWindowText("取消");
}
if (dlg->ee==1)
dlg->m_state.SetWindowText("连接成功");
dlg->m_disconn.SetWindowText("断开");
dlg->m_sync.EnableWindow(TRUE);
GetLocalTime(&ts.sendbycli);
send(dlg->clisock,(char *)&ts,sizeof(ts),0);
//循环获得数据
while(s!=SOCKET_ERROR && dlg->ee!=0)
{
//调用recv函数接收数据
s=recv(dlg->clisock,(char *)&ts,sizeof(ts),0);
GetLocalTime(&ts.recbycli);
//adjust time
int year,month,day,hour,minute,second,msecond;
year=(ts.recbysrv.wYear-ts.sendbycli.wYear+ts.recbysrv.wYear-ts.recbycli.wYear)/2;
month=(ts.recbysrv.wMonth-ts.sendbycli.wMonth+ts.recbysrv.wMonth-ts.recbycli.wMonth)/2;
day=(ts.recbysrv.wDay-ts.sendbycli.wDay+ts.recbysrv.wDay-ts.recbycli.wDay)/2;
hour=(ts.recbysrv.wHour-ts.sendbycli.wHour+ts.recbysrv.wHour-ts.recbycli.wHour)/2;
minute=(ts.recbysrv.wMinute-ts.sendbycli.wMinute+ts.recbysrv.wMinute-ts.recbycli.wMinute)/2;
second=(ts.recbysrv.wSecond-ts.sendbycli.wSecond+ts.recbysrv.wSecond-ts.recbycli.wSecond)/2;
msecond=(ts.recbysrv.wMilliseconds-ts.sendbycli.wMilliseconds+ts.recbysrv.wMilliseconds-ts.recbycli.wMilliseconds)/2;
SYSTEMTIME ct,st;
GetLocalTime(&ct);
ct.wMilliseconds+=msecond;
if(ct.wMilliseconds>=1000)
{
ct.wSecond++;
ct.wMilliseconds-=1000;
}
else if(ct.wMilliseconds<0)
{
ct.wSecond--;
ct.wMilliseconds+=1000;
}
ct.wSecond+=second;
if(ct.wSecond>=60)
{
ct.wMinute++;
ct.wSecond-=60;
}
else if(ct.wSecond<0)
{
ct.wMinute--;
ct.wSecond+=60;
}
ct.wMinute+=minute;
if(ct.wMinute>=60)
{
ct.wHour++;
ct.wMinute-=60;
}
else if(ct.wMinute<0)
{
ct.wHour--;
ct.wMinute+=60;
}
ct.wHour+=hour;
if(ct.wHour>=24)
{
ct.wDay++;
ct.wHour-=24;
}
else if(ct.wHour<0)
{
ct.wDay--;
ct.wHour+=24;
}
ct.wDay+=day;
ct.wMonth+=month;
ct.wYear+=year;
SetLocalTime(&ct);
send(dlg->clisock,"Disconnected",100,0);
dlg->m_sync.EnableWindow(FALSE);
dlg->m_conn.ShowWindow(TRUE);
dlg->m_disconn.ShowWindow(FALSE);
dlg->m_state.SetWindowText("同步成功");
closesocket(dlg->clisock);
AfxEndThread(0);
return 0;
}
//发送断开命令
}
void CTimeClientDlg::OnTimer(UINT nIDEvent) //定时器响应函数
{
if(nIDEvent==1)
{
CTime gg; //定义时间变量
CString s,st;
gg=CTime::GetCurrentTime();
//format time to display
s.Format("%04d",gg.GetYear());
st=st+s+"/";
s.Format("%02d",gg.GetMonth());
st=st+s+"/";
s.Format("%02d",gg.GetDay());
st=st+s+" ";
s.Format("%02d",gg.GetHour());
st=st+s+":";
s.Format("%02d",gg.GetMinute());
st=st+s+":";
s.Format("%02d",gg.GetSecond());
st=st+s;
SetDlgItemText(IDC_EDIT_TIME,st);
CDialog::OnTimer(nIDEvent);
}
if(nIDEvent==2)
{
OnBnClickedButtonConn();
}
if(nIDEvent==3)
{
if(ee==0)
{
OnBnClickedButtonConn();
KillTimer(3);
}
}
}
void CTimeClientDlg::OnDestroy()
{
CDialog::OnDestroy();
// TODO: 在此处添加消息处理程序代码
ee=0;
closesocket(clisock);
KillTimer(1);
KillTimer(2);
KillTimer(3);
}
void CTimeClientDlg::toTray(bool show)
{
NOTIFYICONDATA pnid;
pnid.cbSize=(DWORD)sizeof(NOTIFYICONDATA);//该结构体大小
pnid.hWnd=m_hWnd; //窗口句柄
pnid.uID=(UINT)IDR_MAINFRAME; //图标句柄
pnid.uFlags=NIF_ICON|NIF_MESSAGE|NIF_TIP ; /*图标有效|自定义消息有效|鼠标指向显示文字有效*/
pnid.uCallbackMessage=WM_SHOWTASK;//自定义的消息名称
pnid.hIcon=m_hIcon; //图标句柄
strcpy(pnid.szTip,"时间同步客户端");//鼠标指到那所显示的文字
if(show)
{
Shell_NotifyIcon(NIM_ADD,&pnid);//在托盘区添加图标
ShowWindow(SW_HIDE);//隐藏主窗口
}
else
{
Shell_NotifyIcon(NIM_DELETE,&pnid);//删除托盘图标
ShowWindow(SW_SHOW);//显示主窗口
}
}
LRESULT CTimeClientDlg::onShowTask(WPARAM wParam,LPARAM lParam)
//wParam接收的是图标的ID,而lParam接收的是鼠标的行为
{
if(wParam!=IDR_MAINFRAME)
return 1;
switch(lParam)
{
case WM_RBUTTONUP://右键起来时弹出快捷菜单,这里只有一个“关闭”
{
LPPOINT lpoint=new tagPOINT;
::GetCursorPos(lpoint);//得到鼠标位置
CMenu menu;
menu.CreatePopupMenu();//声明一个弹出式菜单
//增加菜单项“关闭”,点击则发送消息WM_DESTROY给主窗口(已
//隐藏),将程序结束。
menu.AppendMenu(MF_STRING,WM_DESTROY,"关闭");
//确定弹出式菜单的位置
menu.TrackPopupMenu(TPM_LEFTALIGN,lpoint->x,lpoint->y,this);
//资源回收
HMENU hmenu=menu.Detach();
menu.DestroyMenu();
delete lpoint;
toTray(FALSE);
}
break;
case WM_LBUTTONDBLCLK://双击左键的处理
{
this->ShowWindow(SW_SHOW);//简单的显示主窗口
CTimeClientDlg *dlg=(CTimeClientDlg*) AfxGetApp()->GetMainWnd();
ShowWindow(SW_RESTORE);
dlg->SetForegroundWindow();
toTray(FALSE);
}
break;
}
return 0;
}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -