// NetUpgrade.cpp : implementation file
//

#include "stdafx.h"
#include "netsdkdemo.h"
#include "NetUpgrade.h"
#include "NetSDKDemoDlg.h"

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

/////////////////////////////////////////////////////////////////////////////
// CNetUpgrade dialog


CNetUpgrade::CNetUpgrade(CWnd* pParent /*=NULL*/)
	: CDialog(CNetUpgrade::IDD, pParent)
{
	//{{AFX_DATA_INIT(CNetUpgrade)
	m_strFilePath = _T("");
	//}}AFX_DATA_INIT
}


void CNetUpgrade::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CNetUpgrade)
	DDX_Control(pDX, IDC_UPGRADEPROGRESS, m_upgradeProgress);
	DDX_Text(pDX, IDC_FILEPATH, m_strFilePath);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CNetUpgrade, CDialog)
	//{{AFX_MSG_MAP(CNetUpgrade)
	ON_BN_CLICKED(IDC_OPENFILE, OnOpenfile)
	ON_BN_CLICKED(IDC_SENDBIOS, OnSendbios)
	ON_BN_CLICKED(IDC_SENDWEB, OnSendweb)
	ON_BN_CLICKED(IDC_SENDBOOT, OnSendboot)
	ON_BN_CLICKED(IDC_BIOSANDBOOT, OnBiosandboot)
	ON_WM_CLOSE()
	ON_BN_CLICKED(IDC_UP_CANCEL, OnUpCancel)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CNetUpgrade message handlers

void CNetUpgrade::OnOpenfile() 
{
	char buffer[32];
	CString str;
	memset(buffer,0,sizeof(buffer));
	
	//如果當前正在發送數據則不能打開文件。
	if(m_lUpgradeID) 
	{
		return;
	}
	
	CString strFilter = "Upgrade File(*.bin)|*.bin|All Files|*.*|";
	CFileDialog dlg(TRUE, NULL,NULL, OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT, 
		strFilter, this);
	
	if(dlg.DoModal() == IDOK)
	{
		m_strFilePath = dlg.GetPathName();
		
		if(m_File.m_hFile != CFile::hFileNull) 
		{
			m_File.Close();
		}
		
		if(!m_File.Open(m_strFilePath, CFile::modeRead|CFile::typeBinary))
		{
			CString str;
			str.Format("%s %s.",MSG_UG_OPENFILE, m_strFilePath, MSG_UG_FAILED);
			MessageBox(str, MSG_UG_ERROR, MB_ICONERROR);
			return;
		}
		int nFileSize = m_File.GetLength();
		if (nFileSize < 2)
		{
			AfxMessageBox(MSG_FILE_MISSTYPE);
			m_File.Close();
			return;
		}
		
		m_File.Read(buffer,2); //取文件頭兩個字節
		str = buffer;
		if(str == "DH") //Bios文件
		{
			GetDlgItem(IDC_SENDBIOS)->EnableWindow(TRUE);	
			GetDlgItem(IDC_SENDWEB)->EnableWindow(FALSE);
			GetDlgItem(IDC_SENDBOOT)->EnableWindow(FALSE);	
			GetDlgItem(IDC_BIOSANDBOOT)->EnableWindow(FALSE);
		}
		else if(str == "U") //Web文件
		{
			GetDlgItem(IDC_SENDBIOS)->EnableWindow(FALSE);	
			GetDlgItem(IDC_SENDWEB)->EnableWindow(TRUE);
			GetDlgItem(IDC_SENDBOOT)->EnableWindow(FALSE);	
			GetDlgItem(IDC_BIOSANDBOOT)->EnableWindow(FALSE);
		}
		else if(buffer[0] == (char)0xC0 && buffer[1]==(char)0x80) //boot文件
		{
			//boot文件
			if(nFileSize <= 1202180)
			{
				GetDlgItem(IDC_SENDBIOS)->EnableWindow(FALSE);	
				GetDlgItem(IDC_SENDWEB)->EnableWindow(FALSE);
				GetDlgItem(IDC_SENDBOOT)->EnableWindow(TRUE);	
				GetDlgItem(IDC_BIOSANDBOOT)->EnableWindow(FALSE);
			}
			//bios and boot文件
			else
			{
				GetDlgItem(IDC_SENDBIOS)->EnableWindow(FALSE);	
				GetDlgItem(IDC_SENDWEB)->EnableWindow(FALSE);
				GetDlgItem(IDC_BIOSANDBOOT)->EnableWindow(TRUE);	
				GetDlgItem(IDC_SENDBOOT)->EnableWindow(FALSE);
			}
		}
		else
		{
			AfxMessageBox(MSG_FILE_MISSTYPE);
			if(m_File.m_hFile != CFile::hFileNull) 
			{
				m_File.Close();
			}
			GetDlgItem(IDC_SENDBIOS)->EnableWindow(FALSE);	
			GetDlgItem(IDC_SENDWEB)->EnableWindow(FALSE);
			GetDlgItem(IDC_SENDBOOT)->EnableWindow(FALSE);	
			GetDlgItem(IDC_BIOSANDBOOT)->EnableWindow(FALSE);	
			return;
		}
		m_File.Close();
		
		UpdateData(FALSE);
	}	
}

void CALLBACK UpgradeCallBackFunc(LONG lLoginID, LONG lUpgradechannel, int nTotalSize, int nSendSize, DWORD dwUser)
{
	CNetUpgrade * pThis = (CNetUpgrade *)dwUser;
	if (-1 == nSendSize)
	{
		pThis->SetProgressPos(-1);
	}
	else if (-2 == nSendSize)
	{
		pThis->SetProgressPos(-2);
	}
	else
	{
		int nProgress = 100 * nSendSize/nTotalSize;
		pThis->SetProgressPos(nProgress);
	}	
	return;
}

void CNetUpgrade::ProcessSend()
{
	if (m_lUpgradeID)
	{
		return; //已經在升級了
	}
	LONG lUpgradeID = CLIENT_StartUpgrade(m_pDevice->LoginID, m_strFilePath.GetBuffer(0),
		UpgradeCallBackFunc, (DWORD)this);
	if (!lUpgradeID)
	{
		((CNetSDKDemoDlg *)GetParent())->LastError();//Zhengdh 06.11.24
		MessageBox(MSG_UG_STARTFAILED);
	}
	else
	{
		m_bStopped = FALSE;
		if ( !CLIENT_SendUpgrade(lUpgradeID))
		{
			m_bStopped = TRUE;
			((CNetSDKDemoDlg *)GetParent())->LastError();//Zhengdh 06.11.24
			MessageBox(MSG_UG_SENDFAILED);
			CLIENT_StopUpgrade(lUpgradeID);
		}
		else
		{
			m_lUpgradeID = lUpgradeID;
		}
	}
}

void CNetUpgrade::OnSendbios() 
{
	ProcessSend();
}

void CNetUpgrade::OnSendweb() 
{
	ProcessSend();
}

void CNetUpgrade::OnSendboot() 
{
	ProcessSend();
}

void CNetUpgrade::OnBiosandboot() 
{
	ProcessSend();
}

void CNetUpgrade::SetProgressPos(int npos)
{
	if (m_bStopped)
	{
		return;
	}
	if (-1 == npos) 
	{
		if (m_lUpgradeID)
		{
			BOOL bret = CLIENT_StopUpgrade(m_lUpgradeID);
			if (!bret) 
			{
				MessageBox(MSG_UG_STOPFAILED);
			}
			else
			{
				MessageBox(MSG_UG_DONE);
				EndDialog(0);
			}
			m_lUpgradeID = 0;
			m_upgradeProgress.SetPos(0);
			GetDlgItem(IDC_UPPROGRESS)->SetWindowText(MSG_UP_PROGRESS);
		}
	}
	else if (-2 == npos)
	{
		MessageBox("Send upgrade data error!");
		if (m_lUpgradeID)
		{
			BOOL bret = CLIENT_StopUpgrade(m_lUpgradeID);
			if (!bret) 
			{
				MessageBox(MSG_UG_STOPFAILED);
			}
			m_lUpgradeID = 0;
			m_upgradeProgress.SetPos(0);
			GetDlgItem(IDC_UPPROGRESS)->SetWindowText(MSG_UP_PROGRESS);
		}
	}
	else
	{
		m_upgradeProgress.SetPos(npos);
		if (npos >= 100)
		{
			GetDlgItem(IDC_UPPROGRESS)->SetWindowText(MSG_TRANSFER_DONE);
		}
	}
}

BOOL CNetUpgrade::OnInitDialog() 
{
	CDialog::OnInitDialog();
	
	g_SetWndStaticText(this);
	m_lUpgradeID = 0;
	m_bStopped = TRUE;
	
	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}

void CNetUpgrade::OnClose() 
{
	// TODO: Add your message handler code here and/or call default
	
	CDialog::OnClose();
}

void CNetUpgrade::OnUpCancel() 
{
	if (m_lUpgradeID)
	{
		m_bStopped = TRUE;
		BOOL bret = CLIENT_StopUpgrade(m_lUpgradeID);
		if (!bret)
		{
			MessageBox(MSG_UG_STOPFAILED);
			EndDialog(0);
		}
		else
		{
			m_upgradeProgress.SetPos(0);
			m_lUpgradeID = 0;
			GetDlgItem(IDC_UPPROGRESS)->SetWindowText(MSG_UP_PROGRESS);
		}
	}
	else
	{
		EndDialog(0);
	}
}