clientdlg.cpp

初学VC,编写的一个机房管理系统,功能基本上完善,但代码可读性不强.

// clientDlg.cpp : implementation file
//
#include "stdafx.h"
#include "client.h"
#include "clientDlg.h"
#include "afxsock.h"
#include "TaskKeyMgr.h"
#include "ChangePwdDlg.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
#define IDC_HAND            MAKEINTRESOURCE(32649)
#define WM_MY_TRAY_NOTIFICATION WM_USER+100
#define WM_CLOSE_WINDOW WM_USER+101
CString glbxuehao;///////////////全局学号
bool tagofhand=true;
extern CWnd *pclientdlg;////////////发消息的窗口指针////////////////
/////////////////////////////////////////////////////////////////////////////
bool tagofuse=false;//////关机标记////
bool tagofmouse=false;
bool tagofmsg=true;
CRect rect;
bool tsi1=true,tcl1=true,tte1=true,tsi2=true,tcl2=true,tte2=true;///////这些变量都是用来显示效果的标记
void glbshutdown()
{
	CTaskKeyMgr::Disable(CTaskKeyMgr::ALL, FALSE);/////把所有的禁用恢复正常
	///////////////////关机//////////////////////
	OSVERSIONINFO OsInfo;
	OsInfo.dwOSVersionInfoSize=sizeof(OSVERSIONINFO);
	GetVersionEx(&OsInfo);
	if(OsInfo.dwPlatformId==VER_PLATFORM_WIN32_NT)
	{
		HANDLE hToken;
		TOKEN_PRIVILEGES tkp;
		if(!OpenProcessToken(GetCurrentProcess(),
			TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY,&hToken))
		{
			MessageBox(NULL,"系统出现故障，请重启!","警告",MB_OK);
			return;
		}
		LookupPrivilegeValue(NULL,SE_SHUTDOWN_NAME,&tkp.Privileges[0].Luid);
		tkp.PrivilegeCount=1;
		tkp.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED;
		AdjustTokenPrivileges(hToken,FALSE,&tkp,0,(PTOKEN_PRIVILEGES)NULL,0);
		if(GetLastError()!=ERROR_SUCCESS)
		{
				MessageBox(NULL,"系统出现故障，请重启!","警告",MB_OK);
			return;
		}
		if(!ExitWindowsEx(EWX_POWEROFF|EWX_FORCE,0))
		{
				MessageBox(NULL,"系统出现故障，请重启!","警告",MB_OK);
		}
	}
}

UINT waitmsg(void *cs)
{	
	SOCKADDR_IN addClient;
	int len=sizeof(SOCKADDR);
	char recvbuf[50];
	SOCKET	sockSrv=(SOCKET)cs;
	while(1)
	{
			SOCKET sockConn=accept(sockSrv,(SOCKADDR*)&addClient,&len);
			int tagrecv=recv(sockConn,recvbuf,50,0);
			if ((tagrecv == 0) || (tagrecv == SOCKET_ERROR))
			{
					AfxMessageBox("与主机的连接失败,现在将重新启动",MB_OK);
					glbshutdown();
			}
			if(recvbuf[0]=='7')////时间快到了
			{
				shutdown(sockConn, 2);
				closesocket(sockConn);
				if(tagofmsg)
				MessageBox(NULL,"你卡上的余额快不足了,为了您的正常上机,请到前台充卡!","提醒",MB_OK|MB_ICONINFORMATION);
				tagofmsg=false;
			}
			else if(recvbuf[0]=='8')/////////////锁屏///////////
			{
					shutdown(sockConn, 2);
					closesocket(sockConn);
					////////////屏蔽//////////////
					pclientdlg->SendMessage(WM_CLOSE_WINDOW,0,0);
					tagofuse=false;///关机的标记
			}
			else if(recvbuf[0]=='9')
			{
				///////////接收消息//////////
				shutdown(sockConn, 2);
				closesocket(sockConn);
				char *cmsg;
				int len1=recvbuf[1];
				cmsg=new char[len1+1];
				for(int i=0;i<len1;i++)
				cmsg[i]=recvbuf[i+2];
				cmsg[i]='\0';
				CString csmsg;
				csmsg.Format("%s",cmsg);
				delete cmsg;
				MessageBox(NULL,csmsg,"管理员提醒您",MB_OK|MB_ICONINFORMATION);
							
			}
			else if(recvbuf[0]=='a')/////////////关机
			{
				shutdown(sockConn, 2);
				closesocket(sockConn);
				glbshutdown();
			}

			else
			{	
				shutdown(sockConn, 2);
				closesocket(sockConn);
			}
			}

    WSACleanup();
	ExitThread(0);
	return 1;

}

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CClientDlg dialog

CClientDlg::CClientDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CClientDlg::IDD, pParent),m_trayIcon(IDR_MENU1)
{
	//{{AFX_DATA_INIT(CClientDlg)
	password = _T("");
	xuehao = _T("");

	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDI_ICON1);
	m_bShowTrayNotifications = TRUE;//zxn

	


}

void CClientDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CClientDlg)
	DDX_Control(pDX, IDC_PICTE, m_te);
	DDX_Control(pDX, IDC_PICCL, m_cl);
	DDX_Control(pDX, IDC_PICSJ, m_sj);
	DDX_Text(pDX, IDC_PASSWORD, password);
	DDX_Text(pDX, IDC_XUEHAO, xuehao);

	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CClientDlg, CDialog)
	//{{AFX_MSG_MAP(CClientDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_SJ, OnSj)
	ON_BN_CLICKED(IDC_TEACHER, OnTeacher)
	ON_BN_CLICKED(IDC_CLASS, OnClass)
	ON_MESSAGE(WM_MY_TRAY_NOTIFICATION, OnTrayNotification)
	ON_WM_CREATE()
	ON_COMMAND(ID_EXIT, OnExit)
	ON_COMMAND(ID_CHANGEPWD, OnChangePwd)
	ON_COMMAND(ID_TIME, OnView)
	ON_COMMAND(ID_ABOUT, OnAbout)
	ON_WM_TIMER()
	ON_WM_SIZE()
	ON_WM_LBUTTONDBLCLK()
	ON_WM_MOUSEMOVE()
	ON_WM_SETCURSOR()
	ON_WM_LBUTTONUP()
	ON_MESSAGE(WM_CLOSE_WINDOW,OnCloseWindow)
	ON_WM_KEYDOWN()
	ON_WM_CLOSE()
	ON_WM_KILLFOCUS()
	ON_WM_CHAR()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CClientDlg message handlers

BOOL CClientDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	int itimer=SetTimer(1,300000,NULL);
	if(itimer==0)
	{
		MessageBox("计时器未能工作,计时没能正常进行,请重启服务器!","提示",MB_OK);
	}
	Initi();
	SetWindowPos(&wndTopMost, 0, 0, 0, 0,SWP_NOSIZE | SWP_NOMOVE);

	//SetWindowPos(&wndTopMost, 0, 0, 0, 0,SWP_NOSIZE | SWP_NOMOVE);
    //            
	CRect WindowRect;
	CRect m_FullScreenRect;
    GetWindowRect(&WindowRect);
    CRect ClientRect;
    RepositionBars(0, 0xffff, AFX_IDW_PANE_FIRST, reposQuery,&ClientRect);
    ClientToScreen(&ClientRect);
    //获取屏幕的分辨率
    int nFullWidth=GetSystemMetrics(SM_CXSCREEN);
    int nFullHeight=GetSystemMetrics(SM_CYSCREEN);

    /*将除控制条外的客户区全屏显示到从(0,0)到
    (nFullWidth, nFullHeight)区域,将(0,0)和(nFullWidth, nFullHeight)
    两个点外扩充原窗口和除控制条之外的客户区位置间的差值, 
    就得到全屏显示的窗口位置*/
    m_FullScreenRect.left=WindowRect.left-ClientRect.left;
    m_FullScreenRect.top=WindowRect.top-ClientRect.top;
    m_FullScreenRect.right=WindowRect.right - ClientRect.right+nFullWidth;
    m_FullScreenRect.bottom=WindowRect.bottom - ClientRect.bottom+nFullHeight;
    WINDOWPLACEMENT wndpl;
    wndpl.length=sizeof(WINDOWPLACEMENT);
    wndpl.flags=0;
    wndpl.showCmd=SW_SHOWNORMAL;
    wndpl.rcNormalPosition=m_FullScreenRect;
    SetWindowPlacement(&wndpl); 
	////////////屏蔽//////////////
	CTaskKeyMgr::Disable(CTaskKeyMgr::TASKBAR,TRUE);
	CTaskKeyMgr::Disable(CTaskKeyMgr::TASKKEYS,TRUE);
	CTaskKeyMgr::Disable(CTaskKeyMgr::TASKMGR,TRUE);
		//////////////
	CEdit *mEdit = (CEdit*)GetDlgItem(IDC_XUEHAO);
	mEdit->SetFocus();
//	CWnd *pWnd=GetDlgItem(IDC_CHOOSE);
	
//	pWnd->GetWindowRect(&rect);
	rect.left=514;rect.top=97;rect.right=1024;rect.bottom=677;
	ClipCursor(&rect);
	/////////////////////////


	return FALSE;  
}

void CClientDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CClientDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting
		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		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;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		////////////////显示PIC
		CPaintDC dc(this);
		HBITMAP hbitmap=::LoadBitmap(::AfxGetInstanceHandle(),MAKEINTRESOURCE(IDB_BITMAP4));
		HDC hMemDC=::CreateCompatibleDC(NULL);
		SelectObject(hMemDC,hbitmap);
		::StretchBlt(dc.m_hDC,0,0,m_Width,m_Height,hMemDC,0,0,1024,768,SRCCOPY);
		::DeleteDC(hMemDC);
		::DeleteObject(hbitmap);
		CDialog::OnPaint();
	}
	
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CClientDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}


void CClientDlg::OnSj() 
{
	// TODO: Add your control notification handler code here
	SetWindowPos(&wndNoTopMost, 0, 0, 0, 0,SWP_NOSIZE | SWP_NOMOVE);////避免出现不好的东东
	static l_counter=0;////标记是否关机，如果超过3次，就关机
	l_counter++;
	if(l_counter>9)
	{
		glbshutdown();
	}
	HBITMAP hBitmap;
	sockClient=socket(AF_INET,SOCK_STREAM,0);
	addrSrv.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
	addrSrv.sin_family =AF_INET;
	addrSrv.sin_port=htons(6000);
	currenttime=CTime::GetCurrentTime();
    UpdateData();
	if(xuehao.IsEmpty()||password.IsEmpty())
	{	
		//MessageBox("请输入你的学号和密码!","错误",MB_OK|MB_ICONINFORMATION);
		
		hBitmap = ::LoadBitmap(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDB_PICNO));
		m_sj.SetBitmap(hBitmap);
		return;
	}
	glbxuehao=xuehao;
	CString str;
	str +="0";
	str += char(xuehao.GetLength());
	str += xuehao;
	str += char(password.GetLength());
	str += password;
	str += char(0);
	char* buf = str.GetBuffer(0);
	int ret = 0;
	int error = 0;
	ret = connect(sockClient,(SOCKADDR*)&addrSrv,sizeof(SOCKADDR));
	if(ret == SOCKET_ERROR)
	{
	error = WSAGetLastError();
	if(error == 10061)
	MessageBox("与服务器连接遇到错误,请稍后重试","错误",MB_ICONERROR|MB_OK);
	return ;
	}
	ret = send(sockClient, buf, str.GetLength(), 0);
	if(ret != str.GetLength()){
	MessageBox("与服务器连接遇到错误,请稍后重试","错误",MB_ICONERROR|MB_OK);
	return ;
	}
	char buff[10];
	ret = recv(sockClient, buff, 10, 0);
	if(ret == 0){
	MessageBox("与服务器连接遇到错误,请稍后重试","错误",MB_ICONERROR|MB_OK);
	return ;
	}
	buff[ret] = '\0';
	char tag=buff[0];
	xuehao.Empty();
	password.Empty();
	UpdateData(FALSE);
   switch(tag)
  {
	  case '0':tagofsj=0;tagofuse=true;m_trayIcon.SetIcon(IDI_ICON1);CTaskKeyMgr::Disable(CTaskKeyMgr::ALL, FALSE);::ShowWindow(m_hWnd,SW_HIDE); l_counter=0;break;
	  case '1':{hBitmap=::LoadBitmap(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDB_PICIDPSW));m_sj.SetBitmap(hBitmap);CEdit *mEdit = (CEdit*)GetDlgItem(IDC_XUEHAO);
		  mEdit->SetFocus();break;}
	  case '2':{hBitmap=::LoadBitmap(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDB_PICFREEZE));m_sj.SetBitmap(hBitmap);CEdit *mEdit = (CEdit*)GetDlgItem(IDC_XUEHAO);
		  mEdit->SetFocus();break;}
	  case '3':{hBitmap=::LoadBitmap(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDB_PICTIME));m_sj.SetBitmap(hBitmap);CEdit *mEdit = (CEdit*)GetDlgItem(IDC_XUEHAO);
		  mEdit->SetFocus();break;}
	  default: MessageBox("系统遇到不可知的错误，请重新启动!","提醒",MB_OK|MB_ICONERROR);
  }
}

void CClientDlg::OnTeacher() 
{
	SetWindowPos(&wndNoTopMost, 0, 0, 0, 0,SWP_NOSIZE | SWP_NOMOVE);
	static l_counter=0;
	HBITMAP hBitmap;
	l_counter++;
	if(l_counter>9)
	{
		glbshutdown();
	}
	currenttime=CTime::GetCurrentTime();
	sockClient=socket(AF_INET,SOCK_STREAM,0);
	addrSrv.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
	addrSrv.sin_family =AF_INET;
	addrSrv.sin_port=htons(6000);
	int ret = connect(sockClient,(SOCKADDR*)&addrSrv,sizeof(SOCKADDR));
	if(ret == SOCKET_ERROR)
	{
	MessageBox("与服务器连接遇到错误,请稍后重试!","错误",MB_ICONERROR|MB_OK);
	return ;
	}
	char sendbuf[2];
	sprintf(sendbuf,"%d",2);
	sendbuf[1]='\0';
	ret = send(sockClient, sendbuf, strlen(sendbuf)+1, 0);
	if(ret!=2)
	{
		MessageBox("与服务器连接遇到错误,请稍后重试!","错误",MB_ICONERROR|MB_OK);
		return ;
	}
	hBitmap=::LoadBitmap(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDB_PICPROMISE));m_te.SetBitmap(hBitmap);
	char buff[2];
	ret = recv(sockClient, buff, 2, 0);
	if(ret == 0){
		MessageBox("与服务器连接遇到错误,请稍后重试!","错误",MB_ICONERROR|MB_OK);
		return ;
	}
	buff[ret] = '\0';
   char tag=buff[0];
	switch(tag)
  {
	  case '0': tagofsj=2;tagofuse=true;m_trayIcon.SetIcon(IDI_ICON1);CTaskKeyMgr::Disable(CTaskKeyMgr::ALL, FALSE);::ShowWindow(m_hWnd,SW_HIDE);l_counter=0;tagofhand=true;break;
	  case '4': hBitmap=::LoadBitmap(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDB_NOAGAIN));m_te.SetBitmap(hBitmap);break;
	  default: MessageBox("系统遇到不可知的错误，请重新启动!","提醒",MB_OK|MB_ICONERROR);
 }
   

}

void CClientDlg::Initi()
{
	WORD wVersionRequested;
	WSADATA wsaData;
	int err;
    wVersionRequested=MAKEWORD(1,1);
	err=WSAStartup(wVersionRequested,&wsaData);
	if(err!=0){
		MessageBox("初始化失败,请重新启动以解决此问题","错误",MB_ICONERROR|MB_OK);
		return ;
	}
	if(LOBYTE(wsaData.wVersion)!=1|| HIBYTE(wsaData.wVersion)!=1)
	{
		WSACleanup();
		MessageBox("初始化失败,请重新启动以解决此问题","错误",MB_ICONERROR|MB_OK);
		return ;
	}
	/////////////初始化客户端//////////////////