CTCPPortScannerLinear Class Reference

#include <TCPPortScannerLinear.h>

Inheritance diagram for CTCPPortScannerLinear:

[legend]

Collaboration diagram for CTCPPortScannerLinear:

[legend]

List of all members.

Public Types
typedef std::set< unsigned short >	PortsList
enum	_LogPriority { lpDebug, lpMessage, lpCritical, lpError }
typedef enum CErrorHandler::_LogPriority	LogPriority
Public Member Functions
virtual BOOL	Scan (const std::string &rDestinationAddress)
virtual BOOL	Scan (IP aTarget)
	CTCPPortScannerLinear ()
virtual	~CTCPPortScannerLinear ()
void	SetMaxSockets (int iMaxSockets)
int	GetMaxSockets () const
void	SetConnectionTimeout (int iMS)
int	GetConnectionTimeout () const
const char *	GetData (unsigned short usPort) const
char *	GetDataForModify (unsigned short usPort) const
const PortsList &	GetPortList () const
BOOL	IsDone () const
BOOL	IsError () const
void	SetReceiveData (int iMaxDataSize, DWORD dwTimeout)
DWORD	GetReceiveDataTimeout () const
int	GetReceiveDataSize () const
virtual BOOL	StopScan ()
void	AddPorts (unsigned short usFromPort, unsigned short usToPort)
void	SetLocalLog (CErrorLog *pLog)
Static Public Member Functions
static std::string	GetCurrentDateTime ()
static std::string	ErrorCodeToString (DWORD dwErrorCode)
static void	SetLog (CErrorLog *pLog)
Protected Member Functions
virtual void	OnPort (unsigned short usPort, BOOL bConnected)
virtual void	OnPortData (unsigned short usPort, const char *pData, int iDataSize)
virtual void	TCPScanDone (BOOL bError)
virtual void	TCPScanDoneLib (BOOL bError)
virtual void	PortScanned (unsigned short usPort)
void	ScanDone (BOOL bError, BOOL bSpawnThread=TRUE)
BOOL	CanScan ()
int	AdjustNumberOfSockets (int iIncrement)
unsigned short	GetNextPort ()
void	SocketDone (unsigned short usPort, BOOL bConnected)
void	SocketDone (unsigned short usPort, char *pData, int iDataSize)
void	DeletePortsMap ()
virtual CTCPSocketAsync *	AllocateSocket (unsigned short usPort) const
virtual void	DestroySocket (CTCPSocketAsync *pSocket) const
BOOL	NewSocket ()
void	SetError (BOOL bError)
void	SetDone (BOOL bDone)
void	ResetSocketCount ()
void	ResetPair ()
void	ResetData ()
void	SetTarget (IP aTarget)
IP	GetTarget () const
BOOL	IsFinished () const
const std::string &	GetClassName () const
void	WriteMessage (const std::string &rMethod, const std::string &rMessage, LogPriority aPriority=lpMessage)
void	WriteMessage (const std::string &rMethod, const std::string &rMessage, DWORD dwAdditionalData, LogPriority aPriority=lpMessage)
virtual void	ReportError (const std::string &rMethod) const
virtual void	ReportError (const std::string &rMethod, const std::string &rMessage) const
virtual void	ReportError (const std::string &rMethod, const std::string &rMessage, DWORD dwAdditionalData) const
virtual void	ReportError (const std::string &rMethod, int iErrorCode) const
virtual void	ReportError (const std::string &rMethod, const std::string &rMessage, int iErrorCode) const
virtual void	ReportErrorOS (const std::string &rMethod, const std::string &rMessage) const
void	SetName (const std::string &rName) const
Static Protected Member Functions
static void	WriteStaticMessage (const std::string &rClass, const std::string &rMethod, const std::string &rMessage, LogPriority aPriority=lpMessage)
static void	WriteStaticMessage (const std::string &rClass, const std::string &rMethod, const std::string &rMessage, DWORD dwAdditionalData, LogPriority aPriority=lpMessage)
static void	ReportStaticError (const std::string &rClass, const std::string &rMethod)
static void	ReportStaticError (const std::string &rClass, const std::string &rMethod, const std::string &rMessage)
static void	ReportStaticError (const std::string &rClass, const std::string &rMethod, const std::string &rMessage, DWORD dwAdditionalData)

---

Detailed Description

Definition at line 47 of file TCPPortScannerLinear.h.

---

Member Typedef Documentation

typedef enum CErrorHandler::_LogPriority CErrorHandler::LogPriority [inherited]

typedef std::set<unsigned short> CTCPPortScanner::PortsList [inherited]

Definition at line 59 of file TCPPortScanner.h.

---

Member Enumeration Documentation

enum CErrorHandler::_LogPriority [inherited]

Enumerator:

lpDebug
lpMessage
lpCritical
lpError

Definition at line 53 of file ErrorHandler.h.

    {
        lpDebug,
        lpMessage,
        lpCritical,
        lpError
    } LogPriority;

---

Constructor & Destructor Documentation

CTCPPortScannerLinear::CTCPPortScannerLinear

(

)

Definition at line 51 of file TCPPortScannerLinear.cpp.

                                             : CTCPPortScanner()
{
    try
    {
        //Set our name
        SetName(CTCPPortScannerLinear_Class);
    }
    ERROR_HANDLER("CTCPPortScannerLinear")
}

CTCPPortScannerLinear::~CTCPPortScannerLinear

(

)

[virtual]

Definition at line 61 of file TCPPortScannerLinear.cpp.

{
}

---

Member Function Documentation

void CTCPPortScanner::AddPorts	(	unsigned short	usFromPort,
		unsigned short	usToPort
	)			[inherited]

Definition at line 243 of file TCPPortScanner.cpp.

{
    try
    {
        //Check we can insert them
        if (!(usFromPort &&
              usToPort &&
              usFromPort<=usToPort))
            return;

        //Insert them
        ScanPair aData;
        aData.usFromPort=usFromPort;
        aData.usToPort=usToPort;

        //Lock
        CCriticalAutoRelease aRelease(m_pCSection,TRUE);

        m_aPortsToScan.push_back(aData);
    }
    ERROR_HANDLER("AddPorts")
}

int CTCPPortScanner::AdjustNumberOfSockets

(

int

iIncrement

)

[protected, inherited]

Definition at line 585 of file TCPPortScanner.cpp.

{
    try
    {
        //Lock the data
        CCriticalAutoRelease aRelease(m_pCSection,TRUE);

        //Readjust
        m_iActiveSockets+=iIncrement;

        //Return it
        return m_iActiveSockets;
    }
    ERROR_HANDLER_RETURN("AdjustNumberOfSockets",0)
}

CTCPSocketAsync * CTCPPortScanner::AllocateSocket

(

unsigned short

usPort

)

const [protected, virtual, inherited]

Reimplemented in CTCPPortScannerStealth, and CUDPScanner.

Definition at line 793 of file TCPPortScanner.cpp.

{
    try
    {
        CTCPScanner* pSocket;
        pSocket=new CTCPScanner((CTCPPortScanner*)this,
                                usPort,
                                GetReceiveDataSize(),
                                GetReceiveDataTimeout());

        //Do we have receive data ?
        if (!GetReceiveDataSize())
            //Disable receive
            pSocket->DisableEvents(CAsyncSocket::aeReceive);

        if (!pSocket->Create())
        {
            //Delete it
            delete pSocket;

            //Error
            return NULL;
        }
        else
        {
            //Set timeout
            if (GetConnectionTimeout())
                if (!pSocket->SetConnectionTimeout(GetConnectionTimeout()))
                {   
                    //Delete it
                    delete pSocket;

                    //Error
                    return NULL;
                }

            //Done
            return pSocket;
        }           
    }
    ERROR_HANDLER_RETURN("AllocateSocket",NULL)
}

BOOL CTCPPortScanner::CanScan

(

)

[protected, inherited]

Definition at line 331 of file TCPPortScanner.cpp.

{
    try
    {
        //Lock the data
        CCriticalAutoRelease aRelease(m_pCSection,TRUE);

        //Do we have a pair
        if (!m_aCurrentPair.usFromPort ||
            !m_aCurrentPair.usToPort ||
            m_aCurrentPair.usFromPort>m_aCurrentPair.usToPort)
            //Get a pair
            if (m_aPortsToScan.begin()==m_aPortsToScan.end())
                return FALSE;
            else
            {
                //Get the pair and remove it
                m_aCurrentPair=m_aPortsToScan.front();
                m_aPortsToScan.pop_front();
            }

        return TRUE;
    }
    ERROR_HANDLER_RETURN("CanScan",FALSE)
}

void CTCPPortScanner::DeletePortsMap

(

)

[protected, inherited]

Definition at line 481 of file TCPPortScanner.cpp.

{
    try
    {
        //Get the iterator
        DataMap::iterator aIterator;
        aIterator=m_aPortsData.begin();

        //Start
        while (aIterator!=m_aPortsData.end())
        {
            //Save the iterator
            DataMap::iterator aBackupIterator;
            aBackupIterator=aIterator;

            //Advance it
            ++aIterator;

            //Delete the data
            delete [] aBackupIterator->second;

            //Remove the data
            m_aPortsData.erase(aBackupIterator);
        }
    }
    ERROR_HANDLER("DeletePortsMap")
}

void CTCPPortScanner::DestroySocket

(

CTCPSocketAsync *

pSocket

)

const [protected, virtual, inherited]

Reimplemented in CTCPPortScannerStealth, and CUDPScanner.

Definition at line 836 of file TCPPortScanner.cpp.

{
    try
    {
        delete pSocket;
    }
    ERROR_HANDLER("DestroySocket")
}

std::string CErrorHandler::ErrorCodeToString

(

DWORD

dwErrorCode

)

[static, inherited]

Definition at line 122 of file ErrorHandler.cpp.

{
    try
    {
        //Get the error string
        LPVOID lpMsgBuf;
        FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | 
                      FORMAT_MESSAGE_FROM_SYSTEM | 
                      FORMAT_MESSAGE_IGNORE_INSERTS,
                      NULL,
                      dwErrorCode,
                      MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
                      (LPTSTR) &lpMsgBuf,
                      0,
                      NULL);

        //Save it
        std::string sMessage;
        sMessage+=(char*)lpMsgBuf;
        
        //Release the buffer
        LocalFree(lpMsgBuf);

        //Done
        return sMessage;
    }
    catch (...)
    {
        return "Unknown";
    }
}

const std::string & CErrorHandler::GetClassName

(

)

const [protected, inherited]

Reimplemented in CSocketThreadManager.

Definition at line 516 of file ErrorHandler.cpp.

{
    return m_sClassName;
}

int CTCPPortScanner::GetConnectionTimeout

(

)

const [inherited]

Definition at line 732 of file TCPPortScanner.cpp.

{
    return m_iConnectionTimeout;
}

std::string CErrorHandler::GetCurrentDateTime

(

)

[static, inherited]

Definition at line 521 of file ErrorHandler.cpp.

{
    try
    {
        //Our string
        std::string sDate;

        //Our tmp buf
        char cTmp[128];

        //Get date
        _strdate(cTmp);
        sDate=cTmp;
        sDate+=' ';

        //Get time
        _strtime(cTmp);
        sDate+=cTmp;

        //Done
        return sDate;
    }
    ERROR_HANDLER_STATIC_RETURN(CErrorHandler_Class,"GetCurrentDateTime","")
}

const char * CTCPPortScanner::GetData

(

unsigned short

usPort

)

const [inherited]

Definition at line 693 of file TCPPortScanner.cpp.

{
    try
    {
        //Try to look for the data
        DataMap::const_iterator aIterator;
        aIterator=m_aPortsData.find(usPort);

        //Do we have it
        if (aIterator!=m_aPortsData.end())
            return aIterator->second;
        else
            return NULL;
    }
    ERROR_HANDLER_RETURN("GetData",NULL)
}

char * CTCPPortScanner::GetDataForModify

(

unsigned short

usPort

)

const [inherited]

Definition at line 710 of file TCPPortScanner.cpp.

{
    try
    {
        //Try to look for the data
        DataMap::const_iterator aIterator;
        aIterator=m_aPortsData.find(usPort);

        //Do we have it
        if (aIterator!=m_aPortsData.end())
            return aIterator->second;
        else
            return NULL;
    }
    ERROR_HANDLER_RETURN("GetDataForModify",NULL)
}

int CTCPPortScanner::GetMaxSockets

(

)

const [inherited]

Definition at line 742 of file TCPPortScanner.cpp.

{
    return m_iMaxSockets;
}

unsigned short CTCPPortScanner::GetNextPort

(

)

[protected, inherited]

Definition at line 297 of file TCPPortScanner.cpp.

{
    try
    {
        //Lock the data
        CCriticalAutoRelease aRelease(m_pCSection,TRUE);

        //Do we have a pair
        if (!m_aCurrentPair.usFromPort ||
            !m_aCurrentPair.usToPort ||
            m_aCurrentPair.usFromPort>m_aCurrentPair.usToPort)
            //Get a pair
            if (m_aPortsToScan.begin()==m_aPortsToScan.end())
                return 0;
            else
            {
                //Get the pair and remove it
                m_aCurrentPair=m_aPortsToScan.front();
                m_aPortsToScan.pop_front();
            }

        //Get the pair we want to send
        unsigned short usPort;
        usPort=m_aCurrentPair.usFromPort;

        //Increase the port
        ++m_aCurrentPair.usFromPort;

        //Done
        return usPort;
    }
    ERROR_HANDLER_RETURN("GetNextPort",0)
}

const CTCPPortScanner::PortsList & CTCPPortScanner::GetPortList

(

)

const [inherited]

Definition at line 673 of file TCPPortScanner.cpp.

{
    return m_aPortList;
}

int CTCPPortScanner::GetReceiveDataSize

(

)

const [inherited]

Definition at line 292 of file TCPPortScanner.cpp.

{
    return m_iReceiveData;
}

DWORD CTCPPortScanner::GetReceiveDataTimeout

(

)

const [inherited]

Definition at line 287 of file TCPPortScanner.cpp.

{
    return m_dwTimeout;
}

IP CTCPPortScanner::GetTarget

(

)

const [protected, inherited]

Definition at line 771 of file TCPPortScanner.cpp.

{
    return m_aTarget;
}

BOOL CTCPPortScanner::IsDone

(

)

const [inherited]

Definition at line 267 of file TCPPortScanner.cpp.

{
    return m_bDone;
}

BOOL CTCPPortScanner::IsError

(

)

const [inherited]

Definition at line 357 of file TCPPortScanner.cpp.

{
    return m_bError;
}

BOOL CTCPPortScanner::IsFinished

(

)

const [protected, inherited]

Definition at line 853 of file TCPPortScanner.cpp.

{
    return m_bFinished;
}

BOOL CTCPPortScanner::NewSocket

(

)

[protected, inherited]

Definition at line 509 of file TCPPortScanner.cpp.

{
    try
    {
        //Are we done
        if (m_bDone ||
            m_bFinished)
            return TRUE;

        //Get a port
        unsigned short usPort;
        usPort=GetNextPort();

        //Do we have it
        if (!usPort)
        {
            //Are we done
            if (AdjustNumberOfSockets(-1)<=0)
                ScanDone(FALSE);

            //Exit
            return TRUE;
        }

        //Create the socket
        CTCPSocketAsync* pSocket;
        pSocket=AllocateSocket(usPort);
        
        //Create it
        if (!pSocket)
        {
            //Are we done
            if (AdjustNumberOfSockets(-1)<=0 &&
                !CanScan())
            {
                //Run the scan
                ScanDone(FALSE);

                //Exit
                return TRUE;
            }
            else
                return FALSE;
        }

        //Connect it
        if (!pSocket->Connect(0,
                              m_aTarget,
                              usPort))
        {
            //Dispose the socket
            DestroySocket(pSocket);

            //Are we done
            if (AdjustNumberOfSockets(-1)<=0 &&
                !CanScan())
            {
                //Run the scan
                ScanDone(FALSE);

                //Exit
                return TRUE;
            }
            else
                return FALSE;
        }

        //We scanned the port
        PortScanned(usPort);

        //Done
        return TRUE;
    }
    ERROR_HANDLER_RETURN("NewSocket",FALSE)
}

void CTCPPortScanner::OnPort	(	unsigned short	usPort,
		BOOL	bConnected
	)			[protected, virtual, inherited]

Definition at line 678 of file TCPPortScanner.cpp.

{
}

void CTCPPortScanner::OnPortData	(	unsigned short	usPort,
		const char *	pData,
		int	iDataSize
	)			[protected, virtual, inherited]

Definition at line 683 of file TCPPortScanner.cpp.

{
}

void CTCPPortScanner::PortScanned

(

unsigned short

usPort

)

[protected, virtual, inherited]

Reimplemented in CTCPPortScannerStealth.

Definition at line 849 of file TCPPortScanner.cpp.

{
}

void CErrorHandler::ReportError	(	const std::string &	rMethod,
		const std::string &	rMessage,
		int	iErrorCode
	)			const [protected, virtual, inherited]

Definition at line 154 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Get the log
        CErrorLog* pLog;
        pLog=GetLog();

        //Convert the error code
        char aTmp[11];
        sprintf(aTmp,"%d",iErrorCode);

        //Get the string for it
        std::string sError;
        sError=rMessage;
        sError+=", and Socket error: ";
        sError+=aTmp;
        sError+=", ";
        sError+=ErrorCodeToString(iErrorCode);

        //Report to the log
        pLog->ReportError(m_sClassName,
                          rMethod,
                          sError);
    }
    ERROR_UNKNOWN("ReportError")
}

void CErrorHandler::ReportError	(	const std::string &	rMethod,
		int	iErrorCode
	)			const [protected, virtual, inherited]

Definition at line 187 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Get the log
        CErrorLog* pLog;
        pLog=GetLog();

        //Convert the error code
        char aTmp[11];
        sprintf(aTmp,"%d",iErrorCode);

        //Get the string for it
        std::string sError;
        sError="Socket error: ";
        sError+=aTmp;
        sError+=", ";
        sError+=ErrorCodeToString(iErrorCode);

        //Report to the log
        pLog->ReportError(m_sClassName,
                          rMethod,
                          sError);
    }
    ERROR_UNKNOWN("ReportError")
}

void CErrorHandler::ReportError	(	const std::string &	rMethod,
		const std::string &	rMessage,
		DWORD	dwAdditionalData
	)			const [protected, virtual, inherited]

Definition at line 275 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Get the log
        CErrorLog* pLog;
        pLog=GetLog();

        //Convert the number
        char aTmp[11];
        ltoa(dwAdditionalData,aTmp,10);

        //Create the new message
        std::string sNewMessage(rMessage);
        sNewMessage+="Additional data: ";
        sNewMessage+=aTmp;

        //Report to the log
        pLog->ReportError(m_sClassName,
                          rMethod,
                          sNewMessage);
    }
    ERROR_UNKNOWN("ReportError")
}

void CErrorHandler::ReportError	(	const std::string &	rMethod,
		const std::string &	rMessage
	)			const [protected, virtual, inherited]

Definition at line 218 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        CErrorLog* pLog;
        pLog=GetLog();

        //Report to the log
        pLog->ReportError(m_sClassName,
                          rMethod,
                          rMessage);
    }
    ERROR_UNKNOWN("ReportError")
}

void CErrorHandler::ReportError

(

const std::string &

rMethod

)

const [protected, virtual, inherited]

Definition at line 311 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
#ifdef WIN32
        //Get the last error
        DWORD dwLastError;
        dwLastError=GetLastError();

        //Report the error
        GetLog()->ReportError(m_sClassName,
                              rMethod,
                              ErrorCodeToString(dwLastError));

#else
        GetLog()->ReportCatchError(m_sClassName,
                                   rMethod,
                                   "Unknown error!");
#endif
    }
    ERROR_UNKNOWN("ReportError")
}

void CErrorHandler::ReportErrorOS	(	const std::string &	rMethod,
		const std::string &	rMessage
	)			const [protected, virtual, inherited]

Definition at line 237 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Get the last error
        DWORD dwLastError;
        dwLastError=GetLastError();

        //Format the message
        std::string sMessage;
        sMessage=rMessage;
        sMessage+=", with error code: ";

        //Convert the error code
        char aTmp[11];
        itoa(dwLastError,aTmp,10);

        //Add it again
        sMessage+=aTmp;
        sMessage+=" ";
        sMessage+=ErrorCodeToString(dwLastError);
        
        //Get the log
        CErrorLog* pLog;
        pLog=GetLog();

        //Report to the log
        pLog->ReportError(m_sClassName,
                          rMethod,
                          sMessage);
    }
    ERROR_UNKNOWN("ReportErrorOS")
}

void CErrorHandler::ReportStaticError	(	const std::string &	rClass,
		const std::string &	rMethod,
		const std::string &	rMessage,
		DWORD	dwAdditionalData
	)			[static, protected, inherited]

Definition at line 489 of file ErrorHandler.cpp.

{
    if (!m_pLog)
        return;

    try
    {
        //Convert the number
        char aTmp[11];
        ltoa(dwAdditionalData,aTmp,10);

        //Create the new message
        std::string sNewMessage(rMessage);
        sNewMessage+="Additional data: ";
        sNewMessage+=aTmp;

        //Report to the log
        m_pLog->ReportError(rClass,
                            rMethod,
                            sNewMessage);
    }
    ERROR_UNKNOWN("ReportStaticError")
}

void CErrorHandler::ReportStaticError	(	const std::string &	rClass,
		const std::string &	rMethod,
		const std::string &	rMessage
	)			[static, protected, inherited]

Definition at line 472 of file ErrorHandler.cpp.

{
    if (!m_pLog)
        return;

    try
    {
        //Report to the log
        m_pLog->ReportError(rClass,
                            rMethod,
                            rMessage);
    }
    ERROR_UNKNOWN("ReportStaticError")
}

void CErrorHandler::ReportStaticError	(	const std::string &	rClass,
		const std::string &	rMethod
	)			[static, protected, inherited]

Definition at line 446 of file ErrorHandler.cpp.

{
    if (!m_pLog)
        return;

    try
    {
#ifdef WIN32
        //Get the last error
        DWORD dwLastError;
        dwLastError=GetLastError();

        //Report the error
        m_pLog->ReportError(rClass,
                            rMethod,
                            ErrorCodeToString(dwLastError));
#else
        m_pLog->ReportError(rClass,
                            rMethod,
                            "Unknown error!");
#endif
    }
    ERROR_UNKNOWN("ReportStaticError")
}

void CTCPPortScanner::ResetData

(

)

[protected, inherited]

Definition at line 782 of file TCPPortScanner.cpp.

{
    try
    {
        //Clear the sets
        m_aPortAnswers.clear();
        m_aPortList.clear();
    }
    ERROR_HANDLER("ResetData")
}

void CTCPPortScanner::ResetPair

(

)

[protected, inherited]

Definition at line 776 of file TCPPortScanner.cpp.

{
    m_aCurrentPair.usFromPort=0;
    m_aCurrentPair.usToPort=0;
}

void CTCPPortScanner::ResetSocketCount

(

)

[protected, inherited]

Definition at line 761 of file TCPPortScanner.cpp.

{
    m_iActiveSockets=0;
}

BOOL CTCPPortScannerLinear::Scan

(

IP

aTarget

)

[virtual]

Implements CTCPPortScanner.

Definition at line 74 of file TCPPortScannerLinear.cpp.

{
    //Check max sockets are ok
    if (GetMaxSockets()<=0 ||
        !IsDone() ||
        !aTarget)
    {
        //Report it
        ReportError("Scan","Invalid parameters!");

        //Exit
        return FALSE;
    }

    try
    {
        //Reset the pair
        ResetPair();

        //Can we scan
        if (!CanScan())
        {
            //Error
            ReportError("Scan","No ports given!");

            //Error
            SetError(TRUE);

            //Exit
            return FALSE;
        }

        //Set the flags
        SetError(FALSE);
        SetDone(FALSE);

        //Save the number of max sockets
        ResetSocketCount();

        //Delete the port data
        DeletePortsMap();

        //Delete other data
        ResetData();

        //Set our target
        SetTarget(aTarget);

        //Start the loop
        for (int iCounter=0;
             iCounter<GetMaxSockets();
             ++iCounter)
        {
            BOOL bQuit;
            bQuit=FALSE;

            //Try to allocate sockets
            while (!bQuit)
            {
                //Another socket
                AdjustNumberOfSockets(1);

                //Create it
                if (!(bQuit=NewSocket()))
                    ReportError("Scan","Failed to create socket!");
            }
        }

        //Done
        return TRUE;
    }
    ERROR_HANDLER_RETURN("Scan",FALSE)
}

BOOL CTCPPortScannerLinear::Scan

(

const std::string &

rDestinationAddress

)

[virtual]

Implements CTCPPortScanner.

Definition at line 65 of file TCPPortScannerLinear.cpp.

{
    try
    {
        return Scan(CSpoofBase::StringToLong(rDestinationAddress));
    }
    ERROR_HANDLER_RETURN("Scan",FALSE)
}

void CTCPPortScanner::ScanDone	(	BOOL	bError,
		BOOL	bSpawnThread = TRUE
	)			[protected, inherited]

Definition at line 628 of file TCPPortScanner.cpp.

{
    try
    {
        {
            //Lock
            CCriticalAutoRelease aRelease(m_pCSectionDone);

            //What shell we do?
            if (m_bDone ||
                m_bFinished)
                return;

            //Set we are finished
            m_bFinished=TRUE;
        }

        //Thread's data
        ThreadData* pData;
        pData=new ThreadData;

        //Populate it
        pData->bError=bError;
        pData->pScanner=this;
        
        //Do we have a thread
        if (bSpawnThread)
        {
            //Spawn a thread
            CGenericThread* pThread;
            pThread=COSManager::CreateThread(DoneThread);

            //Set to autodelete
            pThread->SetAutoDelete(TRUE);

            //Start it
            pThread->Start((LPVOID)pData);
        }
        else
            DoneThread((LPVOID)pData);
    }
    ERROR_HANDLER("ScanDone")
}

void CTCPPortScanner::SetConnectionTimeout

(

int

iMS

)

[inherited]

Definition at line 727 of file TCPPortScanner.cpp.

{
    m_iConnectionTimeout=iMS;
}

void CTCPPortScanner::SetDone

(

BOOL

bDone

)

[protected, inherited]

Definition at line 752 of file TCPPortScanner.cpp.

{
    m_bDone=bDone;

    //Set the finish flag
    if (!bDone)
        m_bFinished=FALSE;
}

void CTCPPortScanner::SetError

(

BOOL

bError

)

[protected, inherited]

Definition at line 747 of file TCPPortScanner.cpp.

{
    m_bError=bError;
}

void CErrorHandler::SetLocalLog

(

CErrorLog *

pLog

)

[inherited]

Definition at line 441 of file ErrorHandler.cpp.

{
    m_pLocalLog=pLog;
}

void CErrorHandler::SetLog

(

CErrorLog *

pLog

)

[static, inherited]

Definition at line 305 of file ErrorHandler.cpp.

{
    //Save the new log
    m_pLog=pLog;
}

void CTCPPortScanner::SetMaxSockets

(

int

iMaxSockets

)

[inherited]

Definition at line 737 of file TCPPortScanner.cpp.

{
    m_iMaxSockets=iMaxSockets;
}

void CErrorHandler::SetName

(

const std::string &

rName

)

const [protected, inherited]

Definition at line 116 of file ErrorHandler.cpp.

{
    //Save the class name
    m_sClassName=rName;
}

void CTCPPortScanner::SetReceiveData	(	int	iMaxDataSize,
		DWORD	dwTimeout
	)			[inherited]

Definition at line 272 of file TCPPortScanner.cpp.

{
    if (!m_bDone ||
        !m_bFinished)
        return;

    try
    {
        m_iReceiveData=iMaxDataSize;
        m_dwTimeout=dwTimeout;
    }
    ERROR_HANDLER("SetReceiveData")
}

void CTCPPortScanner::SetTarget

(

IP

aTarget

)

[protected, inherited]

Definition at line 766 of file TCPPortScanner.cpp.

{
    m_aTarget=aTarget;
}

void CTCPPortScanner::SocketDone	(	unsigned short	usPort,
		char *	pData,
		int	iDataSize
	)			[protected, inherited]

Definition at line 420 of file TCPPortScanner.cpp.

{
    try
    {
        if (m_bDone ||
            m_bFinished)
            return;

        {
            //Lock the data
            CCriticalAutoRelease aRelease(m_pCSection,TRUE);

            //Did we receive this answer already?
            if (m_aPortAnswers.find(usPort)!=m_aPortAnswers.end())
                return;

            //Add it to the list
            m_aPortList.insert(usPort);

            //And to the existing ports
            m_aPortAnswers.insert(usPort);

            //Add the data to the map
            m_aPortsData.insert(DataMap::value_type(usPort,pData));
        }

        //Inidicate for user
        OnPortData(usPort,
                   pData,
                   iDataSize);

        long lCount;
        lCount=AdjustNumberOfSockets(-1);

        //Decrease our count
        if (!CanScan() &&
            lCount<=0)
            ScanDone(FALSE);
        else
        {
            //Quit flag
            BOOL bQuit;
            bQuit=FALSE;

            //Create the socket
            while (!bQuit)
            {
                //Adjust the number of sockets
                AdjustNumberOfSockets(1);

                //Allocate it
                if (!(bQuit=NewSocket()))
                    ReportError("SocketDone","Error creating the socket!");
            }
        }
    }
    ERROR_HANDLER("SocketDone")
}

void CTCPPortScanner::SocketDone	(	unsigned short	usPort,
		BOOL	bConnected
	)			[protected, inherited]

Definition at line 362 of file TCPPortScanner.cpp.

{
    try
    {
        if (m_bDone ||
            m_bFinished)
            return;

        //Only if connected
        if (bConnected)
        {
            //Lock the data
            CCriticalAutoRelease aRelease(m_pCSection,TRUE);

            //Did we receive this answer already?
            if (m_aPortAnswers.find(usPort)!=m_aPortAnswers.end())
                return;

            //Add it to the list
            m_aPortList.insert(usPort);

            //And to the existing ports
            m_aPortAnswers.insert(usPort);
        }

        //Inicate for user
        OnPort(usPort,
               bConnected);

        long lCount;
        lCount=AdjustNumberOfSockets(-1);

        //Decrease our count
        if (!CanScan() &&
            lCount<=0)
            ScanDone(FALSE);
        else
        {
            //Quit flag
            BOOL bQuit;
            bQuit=FALSE;

            //Create the socket
            while (!bQuit)
            {
                //Adjust the number of sockets
                AdjustNumberOfSockets(1);

                //Allocate it
                if (!(bQuit=NewSocket()))
                    ReportError("SocketDone","Error creating the socket!");
            }
        }
    }
    ERROR_HANDLER("SocketDone")
}

BOOL CTCPPortScanner::StopScan

(

)

[virtual, inherited]

Definition at line 858 of file TCPPortScanner.cpp.

{
    try
    {
        //Call we are done
        ScanDone(TRUE,
                 FALSE);

        //Done
        return TRUE;
    }
    ERROR_HANDLER_RETURN("StopScan",FALSE)
}

void CTCPPortScanner::TCPScanDone

(

BOOL

bError

)

[protected, virtual, inherited]

Definition at line 689 of file TCPPortScanner.cpp.

{
}

void CTCPPortScanner::TCPScanDoneLib

(

BOOL

bError

)

[protected, virtual, inherited]

Reimplemented in CTCPPortScannerStealth, and CUDPScanner.

Definition at line 845 of file TCPPortScanner.cpp.

{
}

void CErrorHandler::WriteMessage	(	const std::string &	rMethod,
		const std::string &	rMessage,
		DWORD	dwAdditionalData,
		LogPriority	aPriority = lpMessage
	)			[protected, inherited]

Definition at line 355 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Convert the number
        char aTmp[11];
        ltoa(dwAdditionalData,aTmp,10);

        //Create the new message
        std::string sNewMessage(rMessage);
        sNewMessage+="Additional data: ";
        sNewMessage+=aTmp;

        //Delegate the call
        GetLog()->WriteMessage(m_sClassName,
                               rMethod,
                               sNewMessage,
                               aPriority);
    }
    ERROR_UNKNOWN("WriteMessage")
}

void CErrorHandler::WriteMessage	(	const std::string &	rMethod,
		const std::string &	rMessage,
		LogPriority	aPriority = lpMessage
	)			[protected, inherited]

Definition at line 337 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Delegate the call
        GetLog()->WriteMessage(m_sClassName,
                               rMethod,
                               rMessage,
                               aPriority);
    }
    ERROR_UNKNOWN("WriteMessage")
}

void CErrorHandler::WriteStaticMessage	(	const std::string &	rClass,
		const std::string &	rMethod,
		const std::string &	rMessage,
		DWORD	dwAdditionalData,
		LogPriority	aPriority = lpMessage
	)			[static, protected, inherited]

Definition at line 403 of file ErrorHandler.cpp.

{
    if (!m_pLog)
        return;

    try
    {
        //Convert the number
        char aTmp[11];
        ltoa(dwAdditionalData,aTmp,10);

        //Create the new message
        std::string sNewMessage(rMessage);
        sNewMessage+="Additional data: ";
        sNewMessage+=aTmp;

        //Delegate the call
        m_pLog->WriteMessage(rClass,
                             rMethod,
                             sNewMessage,
                             aPriority);
    }
    ERROR_UNKNOWN("WriteStaticMessage")
}

void CErrorHandler::WriteStaticMessage	(	const std::string &	rClass,
		const std::string &	rMethod,
		const std::string &	rMessage,
		LogPriority	aPriority = lpMessage
	)			[static, protected, inherited]

Definition at line 383 of file ErrorHandler.cpp.

{
    if (!m_pLog)
        return;

    try
    {
        //Delegate the call
        m_pLog->WriteMessage(rClass,
                             rMethod,
                             rMessage,
                             aPriority);
    }
    ERROR_UNKNOWN("WriteStaticMessage")
}

---

The documentation for this class was generated from the following files:

• TCPPortScannerLinear.h
• TCPPortScannerLinear.cpp