ProSysLib: Dissecting the Process

// PSLNetwork.cpp : Implementation of CPSLNetwork

#include "stdafx.h"
#include "PSLNetwork.h"
#include "PSLIPv4.h"
#include "PSLIPv6.h"
#include "ProSysModule.h"
#include <lm.h>

CPSLNetwork::CPSLNetwork()
{
	// Not needed yet!
	//_Module.InitWinSockets(); // Initialize WinSockets;

	tstring pcName;
	DWORD dwSize = MAX_COMPUTERNAME_LENGTH + 1;
	pcName.resize(dwSize);
	::GetComputerName((LPTSTR)pcName.c_str(), &dwSize);
	m_sPCName = pcName.c_str();
}

HRESULT CPSLNetwork::FinalConstruct()
{
	return S_OK;
}

void CPSLNetwork::FinalRelease()
{
}

////////////////////////////////////////////////////////////////////////
// Interface Implementation;
////////////////////////////////////////////////////////////////////////

STDMETHODIMP CPSLNetwork::get_PCName(BSTR * pValue)
{
	PSL_BEGIN

	*pValue = m_sPCName.copy();
	
	PSL_END
}

STDMETHODIMP CPSLNetwork::get_DomainName(BSTR * pValue)
{
	PSL_BEGIN

	_bstr_t name(_T(""));
	LPWKSTA_INFO_100 pBuffer = NULL;
	DWORD dwStatus = ::NetWkstaGetInfo(NULL, 100, (LPBYTE*)&pBuffer);
	if(dwStatus == NERR_Success)
	{
		name = pBuffer->wki100_langroup;
		::NetApiBufferFree(pBuffer);
	}
	else
		if(dwStatus == ERROR_ACCESS_DENIED)
			return MakeException(exAccessDenied);

	*pValue = name.copy();

	PSL_END
}

STDMETHODIMP CPSLNetwork::IPv4FromString(BSTR Address, IPSLIPv4 ** ppValue)
{
	PSL_BEGIN

	*ppValue = NULL;
	CComObject<CPSLIPv4> * pIPv4 = NULL;
	if(CComObject<CPSLIPv4>::CreateInstance(&pIPv4) == S_OK)
	{
		pIPv4->AddRef();
		bool bValid = true;
		if(Address)
		{
			_bstr_t sAddress(_T(""));
			if(Address)
				sAddress = Address;
			bValid = pIPv4->Initialize(sAddress, true);
		}
		if(bValid)
			*ppValue = CComPtr<IPSLIPv4>(pIPv4);
		else
		{
			pIPv4->Release();
			SetException(exInvalidParameter);
		}
	}

	PSL_END
}

STDMETHODIMP CPSLNetwork::IPv6FromString(BSTR Address, IPSLIPv6 ** ppValue)
{
	PSL_BEGIN

	*ppValue = NULL;
	CComObject<CPSLIPv6> * pIPv6 = NULL;
	if(CComObject<CPSLIPv6>::CreateInstance(&pIPv6) == S_OK)
	{
		pIPv6->AddRef();
		bool bValid = true;
		if(Address)
		{
			_bstr_t sAddress(_T(""));
			if(Address)
				sAddress = Address;
			bValid = pIPv6->Initialize(sAddress, true);
		}
		if(bValid)
			*ppValue = CComPtr<IPSLIPv6>(pIPv6);
		else
		{
			pIPv6->Release();
			SetException(exInvalidParameter);
		}
	}

	PSL_END
}