testsocketactl.cpp

适合初学者

// TestSocketACtl.cpp : Implementation of the CTestSocketACtrl ActiveX Control class.

#include "stdafx.h"
#include "TestSocketA.h"
#include "TestSocketACtl.h"
#include "TestSocketAPpg.h"

#include "TestSocket.h"


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


IMPLEMENT_DYNCREATE(CTestSocketACtrl, COleControl)


/////////////////////////////////////////////////////////////////////////////
// Message map

BEGIN_MESSAGE_MAP(CTestSocketACtrl, COleControl)
	//{{AFX_MSG_MAP(CTestSocketACtrl)
	ON_WM_CREATE()
	ON_WM_TIMER()
	//}}AFX_MSG_MAP
	ON_OLEVERB(AFX_IDS_VERB_PROPERTIES, OnProperties)
	ON_MESSAGE(WM_DELETE_CLIENT,OnDeleteClient)
END_MESSAGE_MAP()


/////////////////////////////////////////////////////////////////////////////
// Dispatch map

BEGIN_DISPATCH_MAP(CTestSocketACtrl, COleControl)
	//{{AFX_DISPATCH_MAP(CTestSocketACtrl)
	DISP_FUNCTION(CTestSocketACtrl, "CreateTestCon", CreateTestCon, VT_BOOL, VTS_I4 VTS_BSTR VTS_I4)
	//}}AFX_DISPATCH_MAP
	DISP_FUNCTION_ID(CTestSocketACtrl, "AboutBox", DISPID_ABOUTBOX, AboutBox, VT_EMPTY, VTS_NONE)
END_DISPATCH_MAP()


/////////////////////////////////////////////////////////////////////////////
// Event map

BEGIN_EVENT_MAP(CTestSocketACtrl, COleControl)
	//{{AFX_EVENT_MAP(CTestSocketACtrl)
	EVENT_CUSTOM("OnSocketClose", FireOnSocketClose, VTS_NONE)
	//}}AFX_EVENT_MAP
END_EVENT_MAP()


/////////////////////////////////////////////////////////////////////////////
// Property pages

// TODO: Add more property pages as needed.  Remember to increase the count!
BEGIN_PROPPAGEIDS(CTestSocketACtrl, 1)
	PROPPAGEID(CTestSocketAPropPage::guid)
END_PROPPAGEIDS(CTestSocketACtrl)


/////////////////////////////////////////////////////////////////////////////
// Initialize class factory and guid

IMPLEMENT_OLECREATE_EX(CTestSocketACtrl, "TESTSOCKETA.TestSocketACtrl.1",
	0xf4eceb5a, 0x5421, 0x48e7, 0xad, 0xf5, 0xef, 0xd1, 0x28, 0x70, 0x36, 0xce)


/////////////////////////////////////////////////////////////////////////////
// Type library ID and version

IMPLEMENT_OLETYPELIB(CTestSocketACtrl, _tlid, _wVerMajor, _wVerMinor)


/////////////////////////////////////////////////////////////////////////////
// Interface IDs

const IID BASED_CODE IID_DTestSocketA =
		{ 0x2a591068, 0xfcf1, 0x46e0, { 0x9d, 0xd8, 0x12, 0xa6, 0x7f, 0xf2, 0xe8, 0x66 } };
const IID BASED_CODE IID_DTestSocketAEvents =
		{ 0xb7e612a8, 0x4fb, 0x470e, { 0x9e, 0x45, 0x1a, 0x4e, 0xa, 0xc3, 0xe, 0xd7 } };


/////////////////////////////////////////////////////////////////////////////
// Control type information

static const DWORD BASED_CODE _dwTestSocketAOleMisc =
	OLEMISC_ACTIVATEWHENVISIBLE |
	OLEMISC_SETCLIENTSITEFIRST |
	OLEMISC_INSIDEOUT |
	OLEMISC_CANTLINKINSIDE |
	OLEMISC_RECOMPOSEONRESIZE;

IMPLEMENT_OLECTLTYPE(CTestSocketACtrl, IDS_TESTSOCKETA, _dwTestSocketAOleMisc)


/////////////////////////////////////////////////////////////////////////////
// CTestSocketACtrl::CTestSocketACtrlFactory::UpdateRegistry -
// Adds or removes system registry entries for CTestSocketACtrl

BOOL CTestSocketACtrl::CTestSocketACtrlFactory::UpdateRegistry(BOOL bRegister)
{
	// TODO: Verify that your control follows apartment-model threading rules.
	// Refer to MFC TechNote 64 for more information.
	// If your control does not conform to the apartment-model rules, then
	// you must modify the code below, changing the 6th parameter from
	// afxRegApartmentThreading to 0.

	if (bRegister)
		return AfxOleRegisterControlClass(
			AfxGetInstanceHandle(),
			m_clsid,
			m_lpszProgID,
			IDS_TESTSOCKETA,
			IDB_TESTSOCKETA,
			afxRegApartmentThreading,
			_dwTestSocketAOleMisc,
			_tlid,
			_wVerMajor,
			_wVerMinor);
	else
		return AfxOleUnregisterClass(m_clsid, m_lpszProgID);
}


/////////////////////////////////////////////////////////////////////////////
// CTestSocketACtrl::CTestSocketACtrl - Constructor

CTestSocketACtrl::CTestSocketACtrl()
{
	InitializeIIDs(&IID_DTestSocketA, &IID_DTestSocketAEvents);

	// TODO: Initialize your control's instance data here.


}


/////////////////////////////////////////////////////////////////////////////
// CTestSocketACtrl::~CTestSocketACtrl - Destructor

CTestSocketACtrl::~CTestSocketACtrl()
{
	// TODO: Cleanup your control's instance data here.
}


/////////////////////////////////////////////////////////////////////////////
// CTestSocketACtrl::OnDraw - Drawing function

void CTestSocketACtrl::OnDraw(
			CDC* pdc, const CRect& rcBounds, const CRect& rcInvalid)
{
	// TODO: Replace the following code with your own drawing code.
	pdc->FillRect(rcBounds, CBrush::FromHandle((HBRUSH)GetStockObject(WHITE_BRUSH)));
	pdc->Ellipse(rcBounds);
}


/////////////////////////////////////////////////////////////////////////////
// CTestSocketACtrl::DoPropExchange - Persistence support

void CTestSocketACtrl::DoPropExchange(CPropExchange* pPX)
{
	ExchangeVersion(pPX, MAKELONG(_wVerMinor, _wVerMajor));
	COleControl::DoPropExchange(pPX);

	// TODO: Call PX_ functions for each persistent custom property.

}


/////////////////////////////////////////////////////////////////////////////
// CTestSocketACtrl::OnResetState - Reset control to default state

void CTestSocketACtrl::OnResetState()
{
	COleControl::OnResetState();  // Resets defaults found in DoPropExchange

	// TODO: Reset any other control state here.
}


/////////////////////////////////////////////////////////////////////////////
// CTestSocketACtrl::AboutBox - Display an "About" box to the user

void CTestSocketACtrl::AboutBox()
{
	CDialog dlgAbout(IDD_ABOUTBOX_TESTSOCKETA);
	dlgAbout.DoModal();
}


/////////////////////////////////////////////////////////////////////////////
// CTestSocketACtrl message handlers




int CTestSocketACtrl::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{
	if (COleControl::OnCreate(lpCreateStruct) == -1)
		return -1;
	
	// TODO: Add your specialized creation code here
	//ms 网络初始化
	m_pTS=NULL;
	
	
	return 0;
}

void CTestSocketACtrl::OnTimer(UINT nIDEvent) 
{
	// TODO: Add your message handler code here and/or call default
	if(m_pTS && (!m_pTS->m_bConnect)) 
	{
		TRACE("OnTimer Reconnect Server\n");
		m_pTS->Connect(m_szServerIP,m_iPort);
		AfxMessageBox("OnTimeer");
	}
	COleControl::OnTimer(nIDEvent);
}

LRESULT CTestSocketACtrl::OnDeleteClient(WPARAM WParam,LPARAM LParam)
{
//	TRACE("Socket OnClose\n");
	FireOnSocketClose();

	m_pTS->Close();
	delete m_pTS;
	
	CreateTestCon(m_iProcessType,m_szServiceIP,m_iPort);
	
	return 0L;
}


BOOL CTestSocketACtrl::CreateTestCon(long iProcessType, LPCTSTR pszServerIP, long iPort) 
{
	// TODO: Add your dispatch handler code here
	if(!AfxSocketInit())
	{
//		TRACE("Socket Initial Error!");
		return FALSE;
	}
	if(m_pTS)
		return FALSE;
	m_pTS=new CTestSocket(this);
	if(!m_pTS->Create())
	{
		delete m_pTS;
		m_pTS=NULL;
		return FALSE;
	}
	m_pTS->Connect(pszServerIP,iPort);
	AfxMessageBox("connect");
	m_szServerIP=pszServerIP;
	
	return TRUE;
}