Open source winsock TCP/IP library V5: TCPRelayHTTP.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 "TCPRelayHTTP.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 CTCPRelayHTTP_Class "CTCPRelayHTTP"
00050 
00051 CTCPRelayHTTP::CTCPRelayHTTP() : CTCPRelay()
00052 {
00053     try
00054     {
00055         //Set our name
00056         SetName(CTCPRelayHTTP_Class);
00057     }
00058     ERROR_HANDLER("CTCPRelayHTTP")
00059 }
00060 
00061 CTCPRelayHTTP::~CTCPRelayHTTP()
00062 {
00063 }
00064 
00065 BOOL CTCPRelayHTTP::Relay(const std::string& rBindAddress,
00066                           unsigned short usBindPort,
00067                           const std::string& rDestinationAddress,
00068                           unsigned short usDestinationPort)
00069 {
00070     try
00071     {
00072         //Delegate
00073         return CTCPRelay::Relay(rBindAddress,
00074                                 usBindPort,
00075                                 rDestinationAddress,
00076                                 usDestinationPort);
00077     }
00078     ERROR_HANDLER_RETURN("Relay",FALSE)
00079 }
00080 
00081 BOOL CTCPRelayHTTP::Relay(IP aBindAddress,
00082                           unsigned short usBindPort,
00083                           IP aDestinationAddress,
00084                           unsigned short usDestinationPort)
00085 {
00086     try
00087     {
00088         //Are we running
00089         if (IsRunning())
00090             return FALSE;
00091 
00092         //Save the addresses
00093         m_sBindAddress=CSocketBase::LongToStdString(aBindAddress);
00094         m_sTargetAddress=CSocketBase::LongToStdString(aDestinationAddress);
00095 
00096         //Delegate the call
00097         return CTCPRelay::Relay(aBindAddress,
00098                                 usBindPort,
00099                                 aDestinationAddress,
00100                                 usDestinationPort);
00101     }
00102     ERROR_HANDLER_RETURN("Relay",FALSE)
00103 }
00104 
00105 std::string CTCPRelayHTTP::ModifyReceiveString(BOOL bIncoming,
00106                                                const char* pBuffer,
00107                                                unsigned short usBufferSize)const
00108 {
00109     try
00110     {
00111         //Only incoming
00112         if (!bIncoming)
00113             return "";
00114 
00115         //Convert to string
00116         std::string sData;
00117         sData=pBuffer;
00118 
00119         //Modify count
00120         int iModifyCount;
00121         iModifyCount=0;
00122 
00123         //Start to modify it
00124         while (ReplaceString(sData))
00125             ++iModifyCount;
00126 
00127         //Did we modify it
00128         if (iModifyCount)
00129             return sData;
00130         else
00131             return "";
00132     }
00133     ERROR_HANDLER_RETURN("ModifyReceiveString","")
00134 }
00135 
00136 BOOL CTCPRelayHTTP::ReplaceString(std::string& rString)const
00137 {
00138     try
00139     {
00140         //Find the string
00141         std::string::size_type aPos;
00142         aPos=rString.find(m_sBindAddress,
00143                           0);
00144 
00145         //Do we have it
00146         if (aPos!=std::string::npos)
00147         {
00148             std::string sNewString;
00149             sNewString.reserve(rString.size()+30);
00150 
00151             //Take the part before the host
00152             sNewString=rString.substr(0,
00153                                       aPos);
00154 
00155             //Add the new address
00156             sNewString+=m_sTargetAddress;
00157 
00158             //Add what's left
00159             sNewString+=rString.substr(aPos+
00160                                        m_sBindAddress.length(),
00161                                        rString.length()-
00162                                        aPos-
00163                                        m_sBindAddress.length());
00164 
00165             //Give it to our string
00166             rString.swap(sNewString);
00167 
00168             //Done
00169             return TRUE;
00170         }
00171         else
00172             return FALSE;
00173     }
00174     ERROR_HANDLER_RETURN("ReplaceString",FALSE)
00175 }
00176 
00177 KOMODIA_NAMESPACE_END
Retail products

Documentation

TCPRelayHTTP.cpp
