TCPPortScannerLinear.cppGo 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 #include "stdafx.h" 00037 #include "TCPPortScannerLinear.h" 00038 00039 #include "ErrorHandlerMacros.h" 00040 00041 #ifdef _MEMORY_DEBUG 00042 #define new DEBUG_NEW 00043 #define malloc DEBUG_MALLOC 00044 static char THIS_FILE[] = __FILE__; 00045 #endif 00046 00047 KOMODIA_NAMESPACE_START 00048 00049 #define CTCPPortScannerLinear_Class "CTCPPortScannerLinear" 00050 00051 CTCPPortScannerLinear::CTCPPortScannerLinear() : CTCPPortScanner() 00052 { 00053 try 00054 { 00055 //Set our name 00056 SetName(CTCPPortScannerLinear_Class); 00057 } 00058 ERROR_HANDLER("CTCPPortScannerLinear") 00059 } 00060 00061 CTCPPortScannerLinear::~CTCPPortScannerLinear() 00062 { 00063 } 00064 00065 BOOL CTCPPortScannerLinear::Scan(const std::string& rDestinationAddress) 00066 { 00067 try 00068 { 00069 return Scan(CSpoofBase::StringToLong(rDestinationAddress)); 00070 } 00071 ERROR_HANDLER_RETURN("Scan",FALSE) 00072 } 00073 00074 BOOL CTCPPortScannerLinear::Scan(IP aTarget) 00075 { 00076 //Check max sockets are ok 00077 if (GetMaxSockets()<=0 || 00078 !IsDone() || 00079 !aTarget) 00080 { 00081 //Report it 00082 ReportError("Scan","Invalid parameters!"); 00083 00084 //Exit 00085 return FALSE; 00086 } 00087 00088 try 00089 { 00090 //Reset the pair 00091 ResetPair(); 00092 00093 //Can we scan 00094 if (!CanScan()) 00095 { 00096 //Error 00097 ReportError("Scan","No ports given!"); 00098 00099 //Error 00100 SetError(TRUE); 00101 00102 //Exit 00103 return FALSE; 00104 } 00105 00106 //Set the flags 00107 SetError(FALSE); 00108 SetDone(FALSE); 00109 00110 //Save the number of max sockets 00111 ResetSocketCount(); 00112 00113 //Delete the port data 00114 DeletePortsMap(); 00115 00116 //Delete other data 00117 ResetData(); 00118 00119 //Set our target 00120 SetTarget(aTarget); 00121 00122 //Start the loop 00123 for (int iCounter=0; 00124 iCounter<GetMaxSockets(); 00125 ++iCounter) 00126 { 00127 BOOL bQuit; 00128 bQuit=FALSE; 00129 00130 //Try to allocate sockets 00131 while (!bQuit) 00132 { 00133 //Another socket 00134 AdjustNumberOfSockets(1); 00135 00136 //Create it 00137 if (!(bQuit=NewSocket())) 00138 ReportError("Scan","Failed to create socket!"); 00139 } 00140 } 00141 00142 //Done 00143 return TRUE; 00144 } 00145 ERROR_HANDLER_RETURN("Scan",FALSE) 00146 } 00147 00148 KOMODIA_NAMESPACE_END |