TCPPortScannerMultithreaded.hGo to the documentation of this file.00001 /* 00002 * Copyright (c) 2000-2003 Barak Weichselbaum <barak@komodia.com> 00003 * All rights reserved. 00004 * 00005 * Redistribution and use in source and binary forms, with or without 00006 * modification, are permitted provided that the following conditions 00007 * are met: 00008 * 1. Redistributions of source code must retain the above copyright 00009 * notice, this list of conditions and the following disclaimer. 00010 * 2. Redistributions in binary form must reproduce the above copyright 00011 * notice, this list of conditions and the following disclaimer in the 00012 * documentation and/or other materials provided with the distribution. 00013 * 00014 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 00015 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 00016 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 00017 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 00018 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 00019 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 00020 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 00021 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 00022 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 00023 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 00024 * SUCH DAMAGE. 00025 * 00026 * Contact info: 00027 * ------------- 00028 * 00029 * Site: http://www.komodia.com 00030 * Main contact: barak@komodia.com 00031 * For custom projects, 00032 * consulting, or other 00033 * paid services: sales@komodia.com 00034 */ 00035 00036 #if !defined(AFX_TCPPORTSCANNERMULTITHREADED_H__DC4B6B1C_8905_41A0_8667_255FE94F65F5__INCLUDED_) 00037 #define AFX_TCPPORTSCANNERMULTITHREADED_H__DC4B6B1C_8905_41A0_8667_255FE94F65F5__INCLUDED_ 00038 00039 #if _MSC_VER > 1000 00040 #pragma once 00041 #endif // _MSC_VER > 1000 00042 00043 #include "TCPPortScanner.h" 00044 00045 KOMODIA_NAMESPACE_START 00046 00047 class CThreadPool; 00048 00049 class CTCPPortScannerMultithreaded : public CTCPPortScanner 00050 { 00051 public: 00052 //Get/set max threads (default is 10) 00053 void SetMaxThreads(int iThreads); 00054 int GetMaxThreads()const; 00055 00056 //Set our multithreaded pool 00057 void SetMultithreadedPool(CThreadPool* pPool); 00058 00059 //Do the scan 00060 virtual BOOL Scan(const std::string& rDestinationAddress); 00061 virtual BOOL Scan(IP aTarget); 00062 00063 //Ctor and Dtor 00064 CTCPPortScannerMultithreaded(); 00065 virtual ~CTCPPortScannerMultithreaded(); 00066 private: 00067 //No copy Ctor 00068 CTCPPortScannerMultithreaded(const CTCPPortScannerMultithreaded& rScanner); 00069 00070 //No copy operator 00071 CTCPPortScannerMultithreaded& operator=(const CTCPPortScannerMultithreaded& rScanner); 00072 00073 //Our socket proc 00074 static void SocketProc(LPVOID lpData); 00075 00076 //Our thread pool 00077 CThreadPool* m_pThreadPool; 00078 00079 //Are we the owners of the thread pool ? 00080 BOOL m_bOwners; 00081 00082 //Max threads 00083 int m_iMaxThreads; 00084 }; 00085 00086 KOMODIA_NAMESPACE_END 00087 00088 #endif // !defined(AFX_TCPPORTSCANNERMULTITHREADED_H__DC4B6B1C_8905_41A0_8667_255FE94F65F5__INCLUDED_) |