// ServerSocketDlg.cpp : implementation file
//

#include "stdafx.h"
#include <atlconv.h>
#include "ServerSocket.h"
#include "ServerSocketDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

const int SOCK_TCP	= 0;
const int SOCK_UDP  = 1;

BYTE fk_HeartMsg[38];

CString Wukong1Ip;
CString LocalIp;

WORD CheckSum(LPBYTE buf,int len);//校驗和函數
DWORD LocalTimeAsms();//獲取當前的時間，以毫秒記
DWORD IpConv(CString ip);	//IP轉換


/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

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()

/////////////////////////////////////////////////////////////////////////////
// CServerSocketDlg dialog

CServerSocketDlg::CServerSocketDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CServerSocketDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CServerSocketDlg)
	m_strPort = _T("2000");
	m_nSockType = SOCK_TCP;	// default TCP
	m_uTimerID=1;
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CServerSocketDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CServerSocketDlg)
	DDX_Control(pDX, IDC_TXT_MESSAGE, m_ctlMessage);
	DDX_Control(pDX, IDC_MESSAGE_LIST, m_ctlMsgList);
	DDX_Control(pDX, IDC_SVR_PORTINC, m_ctlPortInc);
	DDX_Text(pDX, IDC_SVR_PORT, m_strPort);
	DDX_Radio(pDX, IDC_TCP, m_nSockType);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CServerSocketDlg, CDialog)
	//{{AFX_MSG_MAP(CServerSocketDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BTN_START, OnBtnStart)
	ON_BN_CLICKED(IDC_BTN_STOP, OnBtnStop)
	ON_WM_DESTROY()
	ON_BN_CLICKED(IDC_BTN_SEND, OnBtnSend)
	ON_WM_TIMER()
	//}}AFX_MSG_MAP
	ON_MESSAGE(WM_UPDATE_CONNECTION, OnUpdateConnection)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CServerSocketDlg message handlers


//消息翻譯
BOOL CServerSocketDlg::PreTranslateMessage(MSG* pMsg) 
{
	if (pMsg->message == WM_KEYDOWN)
	{
		int nVirtKey = (int) pMsg->wParam;
		if (nVirtKey == VK_ESCAPE)
			return TRUE;
		if (nVirtKey == VK_RETURN && (GetFocus()->m_hWnd  == m_ctlMessage.m_hWnd))
		{
			if (m_pCurServer->IsOpen())
				OnBtnSend();
			return TRUE;
		}
	}
	
	return CDialog::PreTranslateMessage(pMsg);
}

///////////////////////////////////////////////////////////////////////////////
// PickNextAvailable : this is useful only for TCP socket該函數只對TCP使用

//獲得下一個客戶連接
void CServerSocketDlg::PickNextAvailable()
{
	m_pCurServer = NULL;
	for(int i=0; i<MAX_CONNECTION; i++)
	{
		if (!m_SocketManager[i].IsOpen())
		{
			m_pCurServer = &m_SocketManager[i];
			break;
		}
	}
}


///////////////////////////////////////////////////////////////////////////////
// 啟動服務器
bool CServerSocketDlg::StartServer()
{
	bool bSuccess = false;
	SetTimer(m_uTimerID,1000,NULL);//lqy add
	//判斷服務器是否為空
	if (m_pCurServer != NULL)
	{
		//如果是TCP連接
		if (m_nSockType == SOCK_TCP)
		{
			// no smart addressing - we use connection oriented
			//不設定smart地址
			m_pCurServer->SetSmartAddressing( false );
			//創建TCP socket
			bSuccess = m_pCurServer->CreateSocket( m_strPort, AF_INET, SOCK_STREAM, SO_REUSEADDR); // TCP
		}
		//如果是UDP連接
		else
		{
			m_pCurServer->SetSmartAddressing( true );
			//創建UDP socket
			bSuccess = m_pCurServer->CreateSocket( m_strPort, AF_INET, SOCK_DGRAM, SO_BROADCAST); // UDP
		}
		//判斷是否創建成功
		if (bSuccess && m_pCurServer->WatchComm())
		{
			//設定相關控件屬性
			GetDlgItem(IDC_BTN_SEND)->EnableWindow( TRUE );
			GetDlgItem(IDC_BTN_STOP)->EnableWindow( TRUE );
			NextDlgCtrl();
			GetDlgItem(IDC_BTN_START)->EnableWindow( FALSE );
			GetDlgItem(IDC_TCP)->EnableWindow( FALSE );
			GetDlgItem(IDC_UDP)->EnableWindow( FALSE );
			CString strServer, strAddr;
			//獲得本地機器名
			m_pCurServer->GetLocalName( strServer.GetBuffer(256), 256);
			strServer.ReleaseBuffer();
			//獲得本地IP地址
			m_pCurServer->GetLocalAddress( strAddr.GetBuffer(256), 256);
			strAddr.ReleaseBuffer();
			CString strMsg = _T("服務器: ") + strServer;
					strMsg += _T(", @IP地址: ") + strAddr;
					strMsg += _T(" 運行在端口 ") + m_strPort + CString("\r\n");
			m_pCurServer->AppendMessage( strMsg );
		}
	}

	return bSuccess;
}

//對話框初始化
BOOL CServerSocketDlg::OnInitDialog()
{
	ASSERT( GetDlgItem(IDC_BTN_SEND) != NULL );
	ASSERT( GetDlgItem(IDC_BTN_START) != NULL );
	ASSERT( GetDlgItem(IDC_BTN_STOP) != NULL );

	CDialog::OnInitDialog();

	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);		// 設定小圖標

	//lqy add
	Wukong1Ip="192.168.1.9";
	LocalIp = "127.0.0.1";
	
	//增加其他的初始化工作
	m_ctlPortInc.SetRange32( 2000, 4500);
	GetDlgItem(IDC_BTN_SEND)->EnableWindow( FALSE );
	GetDlgItem(IDC_BTN_STOP)->EnableWindow( FALSE );

	//設定最大連接的屬性
	for(int i=0; i<MAX_CONNECTION; i++)
	{
		m_SocketManager[i].SetMessageWindow( &m_ctlMsgList );
		m_SocketManager[i].SetServerState( true );	// 作為服務器運行
	}

	PickNextAvailable();

	return TRUE;  
}

void CServerSocketDlg::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 CServerSocketDlg::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
	{
		CDialog::OnPaint();
	}
}

///////////////////////////////////////////////////////////////////////////////
// OnUpdateConnection
// This message is sent by server manager to indicate connection status

//事件處理
LRESULT CServerSocketDlg::OnUpdateConnection(WPARAM wParam, LPARAM lParam)
{
	UINT uEvent = (UINT) wParam;
	CSocketManager* pManager = reinterpret_cast<CSocketManager*>( lParam );

	// 如果式不是TCP，則返回
	if (m_nSockType != SOCK_TCP)
		return 0L;

	if ( pManager != NULL)
	{
		// 服務器socket已經連接上
		if (uEvent == EVT_CONSUCCESS)
		{
			PickNextAvailable();
			StartServer();
		}
		else if (uEvent == EVT_CONFAILURE || uEvent == EVT_CONDROP)
		{
			pManager->StopComm();
			if (m_pCurServer == NULL)
			{
				PickNextAvailable();
				StartServer();
			}
		}
	}

	return 1L;
}

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

//啟動服務按鈕事件
void CServerSocketDlg::OnBtnStart() 
{
	//數據更新
	UpdateData();
	StartServer();
}

void CServerSocketDlg::OnBtnStop() 
{
	// 斷開服務器
	m_pCurServer->StopComm();
	if (!m_pCurServer->IsOpen())
	{
		GetDlgItem(IDC_BTN_START)->EnableWindow( TRUE );
		PrevDlgCtrl();
		GetDlgItem(IDC_BTN_STOP)->EnableWindow( FALSE );
		GetDlgItem(IDC_TCP)->EnableWindow( TRUE );
		GetDlgItem(IDC_UDP)->EnableWindow( TRUE );
	}
}

//發送按鈕處理事件，發送聊天內容
void CServerSocketDlg::OnBtnSend() 
{
	//定義緩沖區
	BYTE byBuffer[256] = { 0 };
	CString strText;
	//獲得要發送的信息
	m_ctlMessage.GetWindowText( strText );
	int nLen = strText.GetLength();

	if (nLen > 0)
	{
		//在要發送信息的后面加上回車
		strText += _T("\r\n");
		nLen = strText.GetLength();
		//如果是UDP協議
		if (m_nSockType == SOCK_UDP)
		{
			USES_CONVERSION;
			SockAddrIn sin;
			// 發送廣播信息
			sin.CreateFrom(_T("255.255.255.255"), m_strPort);

			memcpy(byBuffer, (LPSOCKADDR) sin, sin.Size());//lqy 將地址信息放入緩沖區
			//將要發送的信息復制到緩沖區
			strcpy((LPSTR)&byBuffer[sin.Size()], T2CA(strText));//lqy 再放入數據
			nLen += sin.Size();
		}
		else
		{
			USES_CONVERSION;
			strcpy((LPSTR)byBuffer, T2CA(strText));//tcp連接不需要地址信息
		}

		// 發送數據
		if (m_nSockType == SOCK_UDP)
			m_pCurServer->WriteComm( byBuffer, nLen+1, INFINITE);
		else
		{
			// 發送到所有客戶端

			for(int i=0; i<MAX_CONNECTION; i++)
			{
				if (m_SocketManager[i].IsOpen() && (m_pCurServer != &m_SocketManager[i]))
					m_SocketManager[i].WriteComm( byBuffer, nLen+1, INFINITE);
			}
		}
	}	
}

void CServerSocketDlg::OnDestroy() 
{
	KillTimer(m_uTimerID);
	for(int i=0; i<MAX_CONNECTION; i++)
	m_SocketManager[i].StopComm();

	CDialog::OnDestroy();	
}

void CServerSocketDlg::SetHeartMsg(WORD wukongNo,WORD state,WORD weapchanstate)//設置心跳報文
{
	CString WkIp;
	int len = 38;
	//設定報文標識A010
	fk_HeartMsg[0] = 0xF0;
	fk_HeartMsg[1] = 0x10;
	//設定長度34
	fk_HeartMsg[2] = 0x00;
	fk_HeartMsg[3] = 38;
	//設定序號0
	fk_HeartMsg[4] = 0;
	fk_HeartMsg[5] = 0;
	fk_HeartMsg[6] = 0;
	fk_HeartMsg[7] = 0;
	//設定數據產生時刻
	DWORD localtime=0;
    localtime = LocalTimeAsms();
	fk_HeartMsg[8] = HIBYTE(HIWORD(localtime));
	fk_HeartMsg[9] = LOBYTE(HIWORD(localtime));
	fk_HeartMsg[10] = HIBYTE(LOWORD(localtime));
	fk_HeartMsg[11] = LOBYTE(LOWORD(localtime));
	//設定信源標識ip
	DWORD sip,dip;
	if(wukongNo == 1)	//武控1
		WkIp = Wukong1Ip;	//Fakong1Ip，Fakong2Ip需在程序初始化時
	//預先設定
/*	if(wukongNo == 2)
   		WkIp = Wukong2Ip;//武控2
	if(wukongNo == 3)	//武控1
		WkIp = Wukong3Ip;
	if(wukongNo == 4)	//武控1
		WkIp = Wukong4Ip;	*/
	sip = IpConv(LocalIp);
	dip = IpConv(WkIp);
	fk_HeartMsg[15] = HIBYTE(HIWORD(sip));
	fk_HeartMsg[14] = LOBYTE(HIWORD(sip));
	fk_HeartMsg[13] = HIBYTE(LOWORD(sip));
	fk_HeartMsg[12] = LOBYTE(LOWORD(sip));
	//設定信宿標識ip
	fk_HeartMsg[19] = HIBYTE(HIWORD(dip));
	fk_HeartMsg[18] = LOBYTE(HIWORD(dip));
	fk_HeartMsg[17] = HIBYTE(LOWORD(dip));
	fk_HeartMsg[16] = LOBYTE(LOWORD(dip));
	//設定確認標志
	fk_HeartMsg[20] = 0;
	fk_HeartMsg[21] = 0xFF;
	//保留字
	fk_HeartMsg[22] = 0;
	fk_HeartMsg[23] = 0;
	//通用發控單元狀態 正常
	fk_HeartMsg[24] = 0;
	fk_HeartMsg[25] = 0;
	fk_HeartMsg[26] = 0;
	fk_HeartMsg[27] = 0;
	//通道狀態 正常
	fk_HeartMsg[28] = 0;
	fk_HeartMsg[29] = 0;
	fk_HeartMsg[30] = 0;
	fk_HeartMsg[31] = 0;
	//保留字
	fk_HeartMsg[32] = 0;
	fk_HeartMsg[33] = 0;
	fk_HeartMsg[34] = 0;
	fk_HeartMsg[35] = 0;

	WORD cs = CheckSum(fk_HeartMsg,len);
	fk_HeartMsg[36] = HIBYTE(cs);
	fk_HeartMsg[37] = LOBYTE(cs);



}

WORD CheckSum(LPBYTE buf,int len)//校驗和函數
{
	DWORD sum=0;
	DWORD tempsum=0;
	WORD tempword=0;
//	LPBYTE temp={0};
	int i=0;
	buf[len-1]=0;
	buf[len-2]=0;
	if((len%2)==0)
	{
		while(i<=len-2)
		{ 	
			sum+=MAKELONG(MAKEWORD(buf[i+1],buf[i]),0);
			i+=2;
		}
	}
/*	if(len%2 ==1)
	{
		len+=1;
		temp[0] = 0;
		for(i=1;i<=len;i++)
			temp[i] = buf[i-1];
		i=0;
		while(i<=len-2)
		{
			sum+=MAKEWORD(temp[i+1],temp[i]);
			i+=2;
		}
	}	*/
	WORD hisum=HIWORD(sum);
	WORD losum=LOWORD(sum);
    while(hisum!=0)
	{
		hisum = HIWORD(sum);
		losum=LOWORD(sum);
		sum = 0;
	//	sum	= hisum&0x0000FFFF+losum&0x0000FFFF;
		sum=MAKELONG(hisum,0)+MAKELONG(losum,0);
	} 

	//對losum取反；
	tempword=(~losum);
	buf[len-1] = LOBYTE(tempword);//恢復數組的校驗位的值
	buf[len-2] = HIBYTE(tempword);

	return ~losum;

}

DWORD LocalTimeAsms( )//獲取當前的時間，以毫秒記
{
	SYSTEMTIME st;
	GetLocalTime(&st);
	DWORD hourtoms	= (st.wHour)*3600*1000;
	DWORD mintoms	=	(st.wMinute)*60*1000;
	DWORD sectoms	= (st.wSecond)*1000;
	return hourtoms+mintoms+(sectoms+st.wMilliseconds);

}

DWORD IpConv(CString ip)	//IP格式轉換
{
	return inet_addr(ip);
}

void CServerSocketDlg::OnTimer(UINT nIDEvent) 
{
	// TODO: Add your message handler code here and/or call default
	SetHeartMsg(1,0,0);
	BYTE byBuffer[256] = { 0 };
	int nLen=0;

	
		if (m_nSockType == SOCK_UDP)
		{
			USES_CONVERSION;
			SockAddrIn sin;
			// 發送廣播信息
			sin.CreateFrom(_T("255.255.255.255"), m_strPort);

			memcpy(byBuffer, (LPSOCKADDR) sin, sin.Size());
			memcpy(&byBuffer[sin.Size()],fk_HeartMsg,38);
			nLen = sin.Size()+38;
		}

		// 發送數據
		if (m_nSockType == SOCK_UDP)
			m_pCurServer->WriteComm( byBuffer, nLen, INFINITE);
	
	CDialog::OnTimer(nIDEvent);
}