Open source winsock TCP/IP library V5: UDPSocketAsync.cpp Source 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 "UDPSocketAsync.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 CUDPSocketAsync_Class "CUDPSocketAsync"
00050 
00051 CUDPSocketAsync::CUDPSocketAsync(BOOL bRawSocket) : CUDPSocket(bRawSocket), 
00052                                                     CAsyncSocket()
00053 {
00054     try
00055     {
00056         //Set our name
00057         SetName(CUDPSocketAsync_Class);
00058     }
00059     ERROR_HANDLER("CUDPSocketAsync")
00060 }
00061 
00062 CUDPSocketAsync::~CUDPSocketAsync()
00063 {
00064     try
00065     {
00066         //We need to close it here
00067         Close();
00068     }
00069     ERROR_HANDLER("~CUDPSocketAsync")
00070 }
00071 
00072 BOOL CUDPSocketAsync::Listen()
00073 {
00074     try
00075     {
00076         //Quit if not ok
00077         if (!CheckSocketValid())
00078             return FALSE;
00079 
00080         //Set async mode
00081         int iResult;
00082         iResult=InternalWSAAsyncSelect(WM_SOCKET_GENERAL,
00083                                        FD_READ);
00084 
00085         //Check result
00086         if (iResult)
00087         {
00088             //Report it
00089             SetLastError("Listen");
00090 
00091             //Exit
00092             return FALSE;
00093         }
00094         else
00095             return TRUE;
00096     }
00097     ERROR_HANDLER_RETURN("Listen",FALSE)
00098 }
00099 
00100 BOOL CUDPSocketAsync::SetAsync()
00101 {
00102     try
00103     {
00104         //Quit if not ok
00105         if (!CheckSocketValid())
00106             return FALSE;
00107 
00108         //Set event to read / write 
00109         int iResult;
00110         iResult=InternalWSAAsyncSelect(WM_SOCKET_GENERAL,
00111                                        FD_WRITE | FD_READ);
00112 
00113         //Check result
00114         if (iResult)
00115         {
00116             //Report it
00117             SetLastError("SetAsync");
00118 
00119             //Exit
00120             return FALSE;
00121         }
00122         else
00123             return TRUE;
00124     }
00125     ERROR_HANDLER_RETURN("SetAsync",FALSE)
00126 }
00127 
00128 BOOL CUDPSocketAsync::OnSocketConnect(int iErrorCode)
00129 {
00130     //First set async again
00131     return SetAsync();
00132 }
00133 
00134 void CUDPSocketAsync::Accepted()
00135 {
00136     try
00137     {
00138         //Add the socket to a list
00139         AddSocketToList();
00140 
00141         //Go to async mode
00142         SetAsync();
00143     }
00144     ERROR_HANDLER("CTCPSocketAsync")
00145 }
00146 
00147 SOCKET CUDPSocketAsync::GetAsyncHandle()const
00148 {
00149     return GetHandle();
00150 }
00151 
00152 BOOL CUDPSocketAsync::Create()
00153 {
00154     try
00155     {
00156         if (!CUDPSocket::Create())
00157             return FALSE;
00158         else
00159         {
00160             //Add the socket to a list
00161             AddSocketToList();
00162 
00163             //Exit
00164             return TRUE;
00165         }
00166     }
00167     ERROR_HANDLER_RETURN("Create",FALSE)
00168 }
00169 
00170 BOOL CUDPSocketAsync::Close()
00171 {
00172     try
00173     {
00174         //Quit if not ok
00175         if (!ValidSocket())
00176             return FALSE;
00177 
00178         //Kill the timer
00179         CAsyncSocket::SocketClosing();
00180 
00181         //Remove from socket list
00182         RemoveSocketFromList();
00183 
00184         //Delegate call
00185         return CUDPSocket::Close();
00186     }
00187     ERROR_HANDLER_RETURN("Close",FALSE)
00188 }
00189 
00190 BOOL CUDPSocketAsync::Send(IP aDestinationAddress, 
00191                            unsigned short usDestinationPort,
00192                            const char* pBuffer,
00193                            unsigned short usBufferLength)
00194 {
00195     try
00196     {
00197         //Delegate call
00198         return CUDPSocket::Send(aDestinationAddress,
00199                                 usDestinationPort,
00200                                 pBuffer,
00201                                 usBufferLength);
00202     }
00203     ERROR_HANDLER_RETURN("Send",GetErrorCode())
00204 }
00205 
00206 BOOL CUDPSocketAsync::Send(const std::string& rDestinationAddress, 
00207                            unsigned short usDestinationPort,
00208                            const char* pBuffer,
00209                            unsigned short usBufferLength)
00210 {
00211     try
00212     {
00213         //Delegate call
00214         return CUDPSocket::Send(rDestinationAddress,
00215                                 usDestinationPort,
00216                                 pBuffer,
00217                                 usBufferLength);
00218     }
00219     ERROR_HANDLER_RETURN("Send",GetErrorCode())
00220 }
00221 
00222 KOMODIA_NAMESPACE_END
Retail products

Documentation

UDPSocketAsync.cpp
