creditobject.cpp

Visual_C++[1].NET_Bible1 Visual_C++宝典书中的全部源码

// CreditObject.cpp
#include "stdafx.h"
#include "CreditServer_h.h"
#include "CreditObject.h"

long g_cComponents = 0;

CCreditVerification::CCreditVerification() : m_refCount(1) 
{
	InterlockedIncrement(&g_cComponents);

	HGLOBAL hG = 0;
	IStream* pIS;
	HRESULT hr = CoMarshalInterThreadInterfaceInStream(
		IID_IUnknown, this, &pIS);
	if (SUCCEEDED(hr))
	{
		hr = GetHGlobalFromStream(pIS, &hG);
		if (SUCCEEDED(hr))
		{
			BYTE* pG = (BYTE*)GlobalLock(hG);

			WORD w1 = MAKEWORD(pG[32], pG[33]);
			WORD w2 = MAKEWORD(pG[34], pG[35]);
			m_uliOXID.LowPart = MAKELONG(w1, w2);

			w1 = MAKEWORD(pG[36], pG[37]);
			w2 = MAKEWORD(pG[38], pG[39]);
			m_uliOXID.HighPart = MAKELONG(w1, w2);

			GlobalUnlock(hG);
		}
	}
}

CCreditVerification::~CCreditVerification() 
{
	InterlockedDecrement(&g_cComponents);
}

// ICreditVerification
HRESULT __stdcall CCreditVerification::SetCardType(unsigned char* s)
{
	if (0 == strcmp("Visa", (const char*)s))
		m_type = Visa;
	else if (0 == strcmp("MasterCard", (const char*)s))
		m_type = MasterCard;
	else m_type = Amex;
	return S_OK;
}
HRESULT __stdcall CCreditVerification::VerifyCardNumber(long n) 
{
	if (m_type == Visa && (n % 2))
		return S_OK;
	else if (m_type == MasterCard && (n % 3))
		return S_OK;
	else if (m_type == Amex && (n % 5))
		return S_OK;
	else return E_INVALIDARG;
}

// IUnknown
HRESULT __stdcall CCreditVerification::QueryInterface(REFIID iid, void**pp)
{
	if (NULL == pp)
		return E_INVALIDARG;
	*pp = NULL;

	if (iid == IID_IUnknown) 
		*pp = this;
	else if (iid == IID_ICreditVerification)
		*pp = static_cast<ICreditVerification*>(this);

	if (*pp)
	{
		AddRef();
		return S_OK;
	}
	return E_NOINTERFACE;
}

ULONG __stdcall CCreditVerification::AddRef() 
{
	return InterlockedIncrement(&m_refCount); 
}

ULONG __stdcall CCreditVerification::Release() 
{
	if (InterlockedDecrement(&m_refCount) == 0) 
	{
		delete this;
		return 0;
	}
	return m_refCount;
}