/*

* Copyright (c) 2005-2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description:  

*

*/

// myclientDlg.cpp : implementation file

//

#include "stdafx.h"

#include "myclient.h"

#include "myclientDlg.h"

#include <statsocket.h>

#ifdef _DEBUG

#define new DEBUG_NEW

#undef THIS_FILE

static char THIS_FILE[] = __FILE__;

#endif

/////////////////////////////////////////////////////////////////////////////

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

/////////////////////////////////////////////////////////////////////////////

// CMyclientDlg dialog

CMyclientDlg::CMyclientDlg(CWnd* pParent /*=NULL*/)

	: CDialog(CMyclientDlg::IDD, pParent)

{

	//{{AFX_DATA_INIT(CMyclientDlg)

		// NOTE: the ClassWizard will add member initialization here

	//}}AFX_DATA_INIT

	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);

}

void CMyclientDlg::DoDataExchange(CDataExchange* pDX)

{

	CDialog::DoDataExchange(pDX);

	//{{AFX_DATA_MAP(CMyclientDlg)

	DDX_Control(pDX, IDC_INFO, m_Info);

	DDX_Control(pDX, IDC_RECEIVE, m_Receive);

	DDX_Control(pDX, IDC_CHKLISTEN, m_ChkListen);

	DDX_Control(pDX, IDC_PORT, m_Port);

	DDX_Control(pDX, IDC_MACHINE, m_Machine);

	DDX_Control(pDX, IDC_CONNECT, m_Connect);

	DDX_Control(pDX, IDC_SEND, m_Send);

	DDX_Control(pDX, IDC_SENDMSG, m_SendMsg);

	DDX_Control(pDX, IDC_RECVLIST, m_RecvList);

	//}}AFX_DATA_MAP

}

BEGIN_MESSAGE_MAP(CMyclientDlg, CDialog)

	//{{AFX_MSG_MAP(CMyclientDlg)

	ON_WM_SYSCOMMAND()

	ON_WM_PAINT()

	ON_WM_QUERYDRAGICON()

	ON_BN_CLICKED(IDC_SEND, OnSend)

	ON_BN_CLICKED(IDC_CONNECT, OnConnect)

	ON_BN_CLICKED(IDC_RECEIVE, OnReceive)

	ON_BN_CLICKED(IDC_CHKLISTEN, OnChklisten)

	ON_BN_CLICKED(IDC_INFO, OnInfo)

	//}}AFX_MSG_MAP

END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////

// CMyclientDlg message handlers

BOOL CMyclientDlg::OnInitDialog()

{

	CDialog::OnInitDialog();

	// IDM_ABOUTBOX must be in the system command range.

	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);			// Set big icon

	SetIcon(m_hIcon, FALSE);		// Set small icon

	hThreadHandle = 0;

	pSocket = NULL;

	m_Machine.SetWindowText("LON-PHILH03");

	m_Port.SetWindowText("3001");

	m_SendMsg.SetWindowText("<B><S><E>");

	Prepare();

	ResetDeviceInfo();

	return TRUE;  // return TRUE  unless you set the focus to a control

}

void CMyclientDlg::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 CMyclientDlg::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();

	}

}

HCURSOR CMyclientDlg::OnQueryDragIcon()

{

	return (HCURSOR) m_hIcon;

}

void CMyclientDlg::OnSend() 

{

	if (pSocket)

	{

		char szData[256];

		m_SendMsg.GetWindowText(szData, 256);

		if (*szData)

		{

			CWaitCursor oWait;

			if (pSocket->Send(SymbianScript, szData, strlen(szData)) != ITS_OK)

			{

				AddString("Could not send.  Try to re-connect.");

				Prepare();

			}

		}

	}

	else

		AddString("Bad socket");

}

void CMyclientDlg::OnReceive() 

{

	if (pSocket)

	{

		CWaitCursor oWait;

		char ID = 0;

		unsigned long Length = 0;

		char *pData = NULL;

		if (pSocket->Receive(&ID, &pData, &Length) != ITS_OK)

		{

			AddString("Could not receive.  Try to re-connect.");

			Prepare();

		}

		else

		{

			char szBuffer[120] = {0};

			switch(ID)

			{

			case SymbianScript:

				if ((Length < 80) && pData)

					sprintf(szBuffer, "%ld [%s]", Length, pData);

				else

				{

					sprintf(szBuffer, "Received %ld bytes of data", Length);

					OpenDataInNotepad(pData, Length);

				}

				break;

			case SymbianError:

				if ((Length < 100) && pData)

					sprintf(szBuffer, "ERROR: %ld [%s]", Length, pData);

				else

				{

					sprintf(szBuffer, "ERROR: Received %ld bytes of data", Length);

					OpenDataInNotepad(pData, Length);

				}

				break;

			case SymbianDeviceInfo:

				StoreDeviceInformation(pData, Length);

				DisplayDeviceInformation(pData);

				break;

			};

			AddString(szBuffer);

		}

		if (pData)

		{

			delete [] pData;

			pData = NULL;

		}

	}

	else

		AddString("Bad socket");

}

void CMyclientDlg::OnConnect() 

{

	bool IsConnected = false;

	pSocket = new CSTATSocket;

	if (pSocket)

	{

		CWaitCursor oWait;

		if (pSocket->Initialise() != ITS_OK)

			return;

		char szMachine[256] = {0};

		if (!m_ChkListen.GetCheck())

		{

			// get machine name and port number

			m_Machine.GetWindowText(szMachine, 247);

			strcat(szMachine, ":");

			m_Port.GetWindowText(szMachine + strlen(szMachine), 8);

		}

		else

		{

			m_Port.GetWindowText(szMachine, 8);

		}

		if (pSocket->Connect(szMachine) == ITS_OK)

		{

			AddString("Connected.");

			m_SendMsg.EnableWindow(TRUE);

			m_Send.EnableWindow(TRUE);

			m_Receive.EnableWindow(TRUE);

			m_Connect.EnableWindow(FALSE);

			m_ChkListen.EnableWindow(FALSE);

			m_Machine.EnableWindow(FALSE);

			m_Port.EnableWindow(FALSE);

			m_Info.EnableWindow(TRUE);

		}

		else

			AddString("Could not connect.");

	}

	else

		AddString("Memory error.");

}

void CMyclientDlg::OnChklisten() 

{

	if (m_ChkListen.GetCheck())

	{

		m_Machine.EnableWindow(FALSE);

	}

	else

	{

		m_Machine.EnableWindow(TRUE);

	}

}

void CMyclientDlg::OnOK() 

{

	Prepare();

	// delete once we've finished with it

	DeleteFile(STAT_TEMP_FILE);

	CDialog::OnOK();

}

void CMyclientDlg::OnInfo() 

{

	if (pSocket)

	{

		if (pSocket->Send(SymbianDeviceInfo) != ITS_OK)

		{

			AddString("Could not send.  Try to re-connect.");

			Prepare();

		}

		else

		{

			OnReceive();

		}

	}

}

///////////////////////////////////////////////////////////////////////////////////

// PRIVATE FUNCTIONS

///////////////////////////////////////////////////////////////////////////////////

///////////////////////////////////////////////////////////////////////////////////

// Add a string to the list box

///////////////////////////////////////////////////////////////////////////////////

void CMyclientDlg::AddString(char *szText)

{

	m_RecvList.AddString(szText);

	int nCount = m_RecvList.GetCount();

	if (nCount > 0)

		m_RecvList.SetTopIndex(nCount - 1);

	UpdateWindow();

}

//////////////////////////////////////////////////////////////////////////////////////

// Open a text file in Notepad

void CMyclientDlg::OpenDataInNotepad(char *pContents, unsigned long ulLength)

{

	// save data

	int ret = E_BADFILENAME;

	CFile script_file;

	if (script_file.Open(STAT_TEMP_FILE, CFile::modeCreate | CFile::modeWrite | CFile::shareDenyWrite))

	{

		try

		{

			// might be a zero-length file

			if (pContents && ulLength)

				script_file.Write(pContents, ulLength);

			ret = ITS_OK;

		}

		catch(CFileException *e)

		{

			e->Delete();

			ret = GENERAL_FAILURE;

		}

		script_file.Flush();

		script_file.Close();

	}

	if (ret == ITS_OK)

	{

		// open file

		TCHAR szCurrentDir[MAX_PATH + 1];

		if (GetWindowsDirectory(szCurrentDir, sizeof(szCurrentDir)))

		{

			CString cBuf;

			cBuf = _T("\"");

			cBuf += szCurrentDir;

			cBuf += _T("\\Notepad.exe\" ");

			cBuf += STAT_TEMP_FILE;

			if (!CallProcess(NULL, cBuf.GetBuffer(0), NULL))

				MessageBox(_T("Could not invoke Notepad.exe to view script file"), NULL, MB_OK);

		}

		else

			MessageBox(_T("Could not locate Windows directory"), NULL, MB_OK);

	}

}

//////////////////////////////////////////////////////////////////////////////////////

// Invoke a process

bool CMyclientDlg::CallProcess(LPCTSTR szApplication, LPTSTR szCommandLine, LPCTSTR szDirectory)

{

	bool valid = false;

	STARTUPINFO startInfo = {0};

	startInfo.cb = sizeof(STARTUPINFO);

	PROCESS_INFORMATION procInfo = {0};

	// event attributes for the child process

	SECURITY_ATTRIBUTES eventAttr;

	eventAttr.nLength = sizeof(eventAttr);

	eventAttr.lpSecurityDescriptor = NULL;

	eventAttr.bInheritHandle = TRUE;

	if (CreateProcess(szApplication, szCommandLine, NULL, NULL, FALSE, NULL, NULL,

					  szDirectory, &startInfo, &procInfo))

	{

		valid = true;

		CloseHandle(procInfo.hThread);

		CloseHandle(procInfo.hProcess);

	}

	return valid;

}

//////////////////////////////////////////////////////////////////////////////////////

// Reset dialog buttons

void CMyclientDlg::Prepare()

{

	m_Info.EnableWindow(FALSE);

	m_SendMsg.EnableWindow(FALSE);

	m_Send.EnableWindow(FALSE);

	m_Receive.EnableWindow(FALSE);

	m_Connect.EnableWindow(TRUE);

	m_ChkListen.EnableWindow(TRUE);

	if (pSocket)

	{

		pSocket->Disconnect();

		pSocket->Release();

		delete pSocket;

		pSocket = NULL;

	}

}

//----------------------------------------------------------------------------

// Break up newline delimited list of device info and store

void CMyclientDlg::StoreDeviceInformation(char *info, unsigned long length)

{

	ResetDeviceInfo();

	if (length && info)

	{

		char szInfo[2048] = {0};

		char *pEnd = szInfo;

		strncpy(szInfo, info, length);

		char *p[18] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};

		int i = 0;

		// skip over leading CR/LF's

		while ((*pEnd) && (*pEnd) == '\r' || (*pEnd) == '\n')

		{

			(*pEnd) = (char)0;

			pEnd++;

		}

		// split

		while(*pEnd)

		{

			p[i] = pEnd;

			// step over setting

			while ((*pEnd) && (*pEnd) != '\r' && (*pEnd) != '\n')

				pEnd++;

			// null terminate

			while ((*pEnd) && (*pEnd) == '\r' || (*pEnd) == '\n')

			{

				(*pEnd) = (char)0;

				pEnd++;

			}

			i++;

			if (i > 18)

				break;

		}

		*pEnd = (char)0;