DNSUDPSocket.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 "DNSUDPSocket.h" 00038 00039 #include "ErrorHandlerMacros.h" 00040 00041 #include "DNSSocket.h" 00042 00043 #ifdef _MEMORY_DEBUG 00044 #define new DEBUG_NEW 00045 #define malloc DEBUG_MALLOC 00046 static char THIS_FILE[] = __FILE__; 00047 #endif 00048 00049 namespace KomodiaDNS 00050 { 00051 00052 #define CDNSUDPSocket_Class "CDNSUDPSocket" 00053 00054 CDNSUDPSocket::CDNSUDPSocket(CDNSSocket* pSocket) : CUDPSocketAsync(FALSE), 00055 m_pDNSSocket(pSocket) 00056 00057 { 00058 try 00059 { 00060 //Set our name 00061 SetName("CDNSUDPSocket"); 00062 } 00063 ERROR_HANDLER("CDNSUDPSocket") 00064 } 00065 00066 CDNSUDPSocket::~CDNSUDPSocket() 00067 { 00068 } 00069 00070 BOOL CDNSUDPSocket::OnSocketReceive(int iErrorCode) 00071 { 00072 try 00073 { 00074 //Check if there's no error 00075 if (!iErrorCode) 00076 { 00077 //Receive the data 00078 char* pBuffer; 00079 pBuffer=new char[2000]; 00080 00081 //How much data did we receive 00082 int iReceived=0; 00083 00084 //Receive 00085 iReceived=Receive(pBuffer,2000); 00086 00087 //If every thing was OK 00088 if (!iReceived) 00089 { 00090 //Dispose of the buffer 00091 delete [] pBuffer; 00092 00093 //Inform of an error 00094 m_pDNSSocket->SocketError(GetSystemLastError()); 00095 00096 //Done 00097 return TRUE; 00098 } 00099 00100 //Inform our father 00101 m_pDNSSocket->SocketReceive(pBuffer, 00102 iReceived); 00103 } 00104 else 00105 //Inform about the error 00106 m_pDNSSocket->SocketError(iErrorCode); 00107 00108 //Always return true 00109 return TRUE; 00110 } 00111 ERROR_HANDLER_RETURN("OnSocketReceive",TRUE) 00112 } 00113 00114 //of namespace 00115 } |