home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / wxos2240.zip / wxWindows-2.4.0 / src / common / protocol.cpp < prev    next >
C/C++ Source or Header  |  2002-11-04  |  5KB  |  181 lines
  1. /////////////////////////////////////////////////////////////////////////////
  2. // Name:        protocol.cpp
  3. // Purpose:     Implement protocol base class
  4. // Author:      Guilhem Lavaux
  5. // Modified by:
  6. // Created:     07/07/1997
  7. // RCS-ID:      $Id: protocol.cpp,v 1.15.2.1 2002/11/03 13:07:41 RR Exp $
  8. // Copyright:   (c) 1997, 1998 Guilhem Lavaux
  9. // Licence:     wxWindows license
  10. /////////////////////////////////////////////////////////////////////////////
  11.  
  12. #ifdef __GNUG__
  13. #pragma implementation "protocol.h"
  14. #endif
  15.  
  16. // For compilers that support precompilation, includes "wx.h".
  17. #include "wx/wxprec.h"
  18.  
  19. #ifdef __BORLANDC__
  20.   #pragma hdrstop
  21. #endif
  22.  
  23. #if wxUSE_PROTOCOL
  24.  
  25. #include "wx/protocol/protocol.h"
  26. #include "wx/url.h"
  27. #include "wx/module.h"
  28.  
  29. #include <stdlib.h>
  30.  
  31. /////////////////////////////////////////////////////////////////
  32. // wxProtoInfo
  33. /////////////////////////////////////////////////////////////////
  34.  
  35. /*
  36.  * --------------------------------------------------------------
  37.  * --------- wxProtoInfo CONSTRUCTOR ----------------------------
  38.  * --------------------------------------------------------------
  39.  */
  40.  
  41. wxProtoInfo::wxProtoInfo(const wxChar *name, const wxChar *serv,
  42.                          const bool need_host1, wxClassInfo *info)
  43. {
  44.     m_protoname = name;
  45.     m_servname = serv;
  46.     m_cinfo = info;
  47.     m_needhost = need_host1;
  48.     next = wxURL::ms_protocols;
  49.     wxURL::ms_protocols = this;
  50. }
  51.  
  52. /////////////////////////////////////////////////////////////////
  53. // wxProtocol ///////////////////////////////////////////////////
  54. /////////////////////////////////////////////////////////////////
  55.  
  56. #if wxUSE_SOCKETS
  57. IMPLEMENT_ABSTRACT_CLASS(wxProtocol, wxSocketClient)
  58. #else
  59. IMPLEMENT_ABSTRACT_CLASS(wxProtocol, wxObject)
  60. #endif
  61.  
  62. wxProtocol::wxProtocol()
  63. #if wxUSE_SOCKETS
  64.  : wxSocketClient()
  65. #endif
  66. {
  67. }
  68.  
  69. #if wxUSE_SOCKETS
  70. bool wxProtocol::Reconnect()
  71. {
  72.     wxIPV4address addr;
  73.  
  74.     if (!GetPeer(addr))
  75.     {
  76.         Close();
  77.         return FALSE;
  78.     }
  79.  
  80.     if (!Close())
  81.         return FALSE;
  82.         
  83.     if (!Connect(addr))
  84.         return FALSE;
  85.  
  86.     return TRUE;
  87. }
  88.  
  89. // ----------------------------------------------------------------------------
  90. // Read a line from socket
  91. // ----------------------------------------------------------------------------
  92.  
  93. // TODO ReadLine() should use buffers private to wxProtocol for efficiency!
  94.  
  95. // static
  96. wxProtocolError wxProtocol::ReadLine(wxSocketBase *socket, wxString& result)
  97. {
  98.     result.Empty();
  99.     char ch, chLast = '\0';
  100.     while ( !socket->Read(&ch, sizeof(ch)).Error() )
  101.     {
  102.         switch ( ch )
  103.         {
  104.             case '\r':
  105.                 // remember it, if the following is '\n', we're done
  106.                 chLast = '\r';
  107.                 break;
  108.  
  109.             case '\n':
  110.                 // only ends line if the previous character was '\r'
  111.                 if ( chLast == '\r' )
  112.                 {
  113.                     // EOL found
  114.                     return wxPROTO_NOERR;
  115.                 }
  116.                 //else: fall through
  117.  
  118.             default:
  119.                 // normal char
  120.                 if ( chLast )
  121.                 {
  122.                     result += wxString::FromAscii( chLast );
  123.                     chLast = '\0';
  124.                 }
  125.  
  126.                 result += wxString::FromAscii( ch );
  127.         }
  128.     }
  129.  
  130.     return wxPROTO_NETERR;
  131. }
  132.  
  133. wxProtocolError wxProtocol::ReadLine(wxString& result)
  134. {
  135.     return ReadLine(this, result);
  136. }
  137.  
  138. // old function which only chops '\n' and not '\r\n'
  139. wxProtocolError GetLine(wxSocketBase *sock, wxString& result)
  140. {
  141. #define PROTO_BSIZE 2048
  142.     size_t avail, size;
  143.     char tmp_buf[PROTO_BSIZE], tmp_str[PROTO_BSIZE];
  144.     char *ret;
  145.     bool found;
  146.  
  147.     avail = sock->Read(tmp_buf, PROTO_BSIZE).LastCount();
  148.     if (sock->Error() || avail == 0)
  149.         return wxPROTO_NETERR;
  150.  
  151.     memcpy(tmp_str, tmp_buf, avail);
  152.  
  153.     // Not implemented on all systems
  154.     // ret = (char *)memccpy(tmp_str, tmp_buf, '\n', avail);
  155.     found = FALSE;
  156.     for (ret=tmp_str;ret < (tmp_str+avail); ret++)
  157.         if (*ret == '\n') 
  158.         {
  159.             found = TRUE;
  160.             break;
  161.         }
  162.  
  163.     if (!found)
  164.         return wxPROTO_PROTERR;
  165.         
  166.     *ret = 0;
  167.  
  168.     result = wxString::FromAscii( tmp_str );
  169.     result = result.Left(result.Length()-1);
  170.  
  171.     size = ret-tmp_str+1;
  172.     sock->Unread(&tmp_buf[size], avail-size);
  173.   
  174.     return wxPROTO_NOERR;
  175. #undef PROTO_BSIZE
  176. }
  177. #endif // wxUSE_SOCKETS
  178.  
  179. #endif // wxUSE_PROTOCOL
  180.  
  181.