SniffSocket.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 "SniffSocket.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 CSniffSocket_Class "CSniffSocket" 00050 00051 CSniffSocket::CSniffSocket() : CSpoofSocket(TRUE), 00052 CAsyncSocket() 00053 { 00054 try 00055 { 00056 //Set our name 00057 SetName(CSniffSocket_Class); 00058 } 00059 ERROR_HANDLER("CSniffSocket") 00060 } 00061 00062 CSniffSocket::~CSniffSocket() 00063 { 00064 try 00065 { 00066 RemoveSocketFromList(); 00067 } 00068 ERROR_HANDLER("~CSniffSocket") 00069 } 00070 00071 BOOL CSniffSocket::Sniff(BOOL bSniff) 00072 { 00073 try 00074 { 00075 //Try to move to sniff mode 00076 if (CSpoofSocket::Sniff(bSniff)) 00077 return !WSAAsyncSelect(GetHandle(), 00078 GetWindowHandle(), 00079 WM_SOCKET_GENERAL, 00080 FD_READ); 00081 else 00082 return FALSE; 00083 } 00084 ERROR_HANDLER_RETURN("Sniff",FALSE) 00085 } 00086 00087 BOOL CSniffSocket::Create() 00088 { 00089 try 00090 { 00091 //Try to create as regular IP socket 00092 SetProtocol(IPPROTO_IP); 00093 if (CSpoofSocket::Create(IPPROTO_IP)) 00094 { 00095 //OK, we can use this socket 00096 AddSocketToList(); 00097 00098 //Done 00099 return TRUE; 00100 } 00101 else 00102 return FALSE; 00103 } 00104 ERROR_HANDLER_RETURN("Create",FALSE) 00105 } 00106 00107 SOCKET CSniffSocket::GetAsyncHandle() const 00108 { 00109 return GetHandle(); 00110 } 00111 00112 BOOL CSniffSocket::SetAsync() 00113 { 00114 //Do nothing 00115 return TRUE; 00116 } 00117 00118 KOMODIA_NAMESPACE_END |