103 lines
2.8 KiB
C++
103 lines
2.8 KiB
C++
|
#include "stdafx.h"
|
||
|
|
#include "natobjs.h"
|
||
|
|
#include "resource.h"
|
||
|
|
|
||
|
|
|
||
|
|
//-------------------------------------------------------------------------
|
||
|
|
CNATGroup::CNATGroup( CNATServerComputer* pNatComputer, LPCTSTR pszIP,
|
||
|
|
LPCTSTR pszName, DWORD dwSticky, DWORD type ):
|
||
|
|
m_pNatComputer(pNatComputer),
|
||
|
|
m_dwSticky(dwSticky),
|
||
|
|
m_type(type)
|
||
|
|
{
|
||
|
|
m_csIP = pszIP;
|
||
|
|
m_csName = pszName;
|
||
|
|
}
|
||
|
|
|
||
|
|
//-------------------------------------------------------------------------
|
||
|
|
CNATGroup::~CNATGroup()
|
||
|
|
{
|
||
|
|
// clean up the site array
|
||
|
|
EmptySites();
|
||
|
|
}
|
||
|
|
|
||
|
|
//-------------------------------------------------------------------------
|
||
|
|
// Edit the properties of this Site - true if OK
|
||
|
|
BOOL CNATGroup::OnProperties()
|
||
|
|
{
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
//-------------------------------------------------------------------------
|
||
|
|
// This is just a handy way to get the nat machine object to commit
|
||
|
|
void CNATGroup::Commit()
|
||
|
|
{
|
||
|
|
if ( m_pNatComputer )
|
||
|
|
m_pNatComputer->Commit();
|
||
|
|
}
|
||
|
|
|
||
|
|
//-------------------------------------------------------------------------
|
||
|
|
// adds a new site to the list.
|
||
|
|
// think about making this a wizard
|
||
|
|
CNATSite* CNATGroup::NewSite()
|
||
|
|
{
|
||
|
|
CNATSiteComputer* pSC = NULL;
|
||
|
|
// ask the nat server object for a site computer
|
||
|
|
pSC = m_pNatComputer->NewComputer();
|
||
|
|
if ( !pSC )
|
||
|
|
return NULL;
|
||
|
|
|
||
|
|
// for now, make a site with the defaults
|
||
|
|
CNATSite* pS = new CNATSite( this, pSC );
|
||
|
|
if ( pS == NULL )
|
||
|
|
{
|
||
|
|
AfxMessageBox( IDS_LOWMEM );
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Ask the user to edit the group's properties.
|
||
|
|
// Do not add it if they cancel
|
||
|
|
if ( !pS->OnProperties() )
|
||
|
|
{
|
||
|
|
delete pS;
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
// add the object to the end of the array
|
||
|
|
m_rgbSites.Add( pS );
|
||
|
|
return pS;
|
||
|
|
}
|
||
|
|
|
||
|
|
//-------------------------------------------------------------------------
|
||
|
|
// adds an existing site to the list - to be called during a refresh.
|
||
|
|
// this checks the visiblity of the machine on the net as it adds it
|
||
|
|
void CNATGroup::AddSite( CNATSiteComputer* pSiteComputer, LPCTSTR pszPrivateIP, IN LPCTSTR pszName )
|
||
|
|
{
|
||
|
|
// create the new Site Computer object
|
||
|
|
CNATSite* pS = new CNATSite( this, pSiteComputer, pszPrivateIP, pszName );
|
||
|
|
if ( pS == NULL )
|
||
|
|
{
|
||
|
|
AfxMessageBox( IDS_LOWMEM );
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
// add the object to the end of the array
|
||
|
|
m_rgbSites.Add( pS );
|
||
|
|
}
|
||
|
|
|
||
|
|
//-------------------------------------------------------------------------
|
||
|
|
// empties and frees all the groups/sites in the groups list
|
||
|
|
void CNATGroup::EmptySites()
|
||
|
|
{
|
||
|
|
DWORD numSites = m_rgbSites.GetSize();
|
||
|
|
|
||
|
|
// loop the array and free all the group objects
|
||
|
|
for ( DWORD i = 0; i < numSites; i++ )
|
||
|
|
{
|
||
|
|
delete m_rgbSites[i];
|
||
|
|
}
|
||
|
|
|
||
|
|
// empty the array itself
|
||
|
|
m_rgbSites.RemoveAll();
|
||
|
|
}
|