SocketBase.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_SPOOFBASE_H__3760FA30_2B52_4F62_9EB6_46640C36E4F1__INCLUDED_) 00037 #define AFX_SPOOFBASE_H__3760FA30_2B52_4F62_9EB6_46640C36E4F1__INCLUDED_ 00038 00039 #if _MSC_VER > 1000 00040 #pragma once 00041 #endif // _MSC_VER > 1000 00042 00043 //windows sockets2 00044 #include <winsock2.h> 00045 #include <ws2tcpip.h> 00046 00047 #include "ErrorHandler.h" 00048 00049 KOMODIA_NAMESPACE_START 00050 00051 class CGenericCriticalSection; 00052 00053 typedef unsigned long IP; 00054 00055 //Handles basic errors 00056 class CSocketBase : public CErrorHandler 00057 { 00058 public: 00059 //Get the error code for compatability 00060 //returns SOCKET_ERROR for Win32 00061 static int GetErrorCode(); 00062 00063 //Get the system last error (needed for threading) 00064 static int GetSystemLastError(); 00065 00066 //Are we initialized 00067 static BOOL IsInitialized(); 00068 00069 //Converts a string to long 00070 static unsigned long StringToLong(const std::string& rAddress); 00071 00072 //Convert long to string (Not thread safe) 00073 static char FAR * LongToString(unsigned long ulAddr); 00074 00075 //Convert long to string but as a standard string (thread safe) 00076 static std::string LongToStdString(unsigned long ulAddr); 00077 00078 //Initialize the sockets 00079 //bMultiThreaded - Do we want multithreaded support 00080 //ulNumberOfThreads - Number of threads to allocate 00081 static BOOL InitializeSockets(BOOL bMultiThreaded=TRUE, 00082 unsigned long ulNumberOfThreads=10); 00083 00084 //Shutdown the sockets 00085 static BOOL ShutdownSockets(); 00086 00087 //Get the last error 00088 int GetLastError()const; 00089 00090 //Get the number of threads 00091 static unsigned long GetNumberOfThreads(); 00092 00093 //Are we multithreaded 00094 static BOOL IsMultiThreaded(); 00095 00096 //Get the critical section 00097 static CGenericCriticalSection* GetGlobalCriticalSection(); 00098 00099 //Resolve a DNS entry (using winsock2 routines) 00100 static unsigned long ResolveDNS(const std::string& rAddress); 00101 00102 //Check if an address is valid 00103 static BOOL ValidAddress(const std::string& rAddress); 00104 00105 //Ctor and Dtor 00106 CSocketBase(); 00107 CSocketBase(const CSocketBase& rBase); 00108 virtual ~CSocketBase(); 00109 protected: 00110 //Shutdown notifier 00111 virtual void NotifyShutdown(); 00112 00113 //Register shutdown class 00114 void RegisterShutdown(CSocketBase* pBase); 00115 00116 //Set the socket last error 00117 void SetLastError(const std::string& rMethod)const; 00118 void SetLastError(const std::string&, 00119 int iErrorCode)const; 00120 00121 //Our critical section 00122 static CGenericCriticalSection* m_pCSection; 00123 private: 00124 //Our DNS cache 00125 typedef std::map<std::string,sockaddr_in> DNSMap; 00126 private: 00127 //Reseolve DNS 00128 static sockaddr_in InternalResolveDNS(const std::string& rAddress); 00129 00130 //Last error we had 00131 mutable int m_iLastError; 00132 00133 //Are we initialized 00134 static BOOL m_bInitialized; 00135 00136 //Class to notify 00137 static CSocketBase* m_pShutdownClass; 00138 00139 //Are we multithreaded 00140 static BOOL m_bMultiThreaded; 00141 00142 //Number of threaded 00143 static int m_ulNumberOfThreads; 00144 00145 //Our DNS map 00146 static DNSMap m_aDNSMap; 00147 00148 //CS for the MAP 00149 static CGenericCriticalSection* m_pCSectionDNS; 00150 }; 00151 00152 KOMODIA_NAMESPACE_END 00153 00154 #endif // !defined(AFX_SPOOFBASE_H__3760FA30_2B52_4F62_9EB6_46640C36E4F1__INCLUDED_) |