X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/4846abaf45a5a1b2ba4392fab7fdc45f1518f4d7..f5ba273ecd799f652736ce2bc830283787302a56:/src/common/protocol.cpp diff --git a/src/common/protocol.cpp b/src/common/protocol.cpp index ac5020956f..542752e318 100644 --- a/src/common/protocol.cpp +++ b/src/common/protocol.cpp @@ -17,19 +17,15 @@ #include "wx/wxprec.h" #ifdef __BORLANDC__ -#pragma hdrstop -#endif - -#if wxUSE_SOCKETS - -#ifndef WX_PRECOMP -#include + #pragma hdrstop #endif #include "wx/protocol/protocol.h" #include "wx/url.h" #include "wx/module.h" +#include + ///////////////////////////////////////////////////////////////// // wxProtoInfo ///////////////////////////////////////////////////////////////// @@ -55,18 +51,26 @@ wxProtoInfo::wxProtoInfo(const wxChar *name, const wxChar *serv, // wxProtocol /////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////// +#if wxUSE_SOCKETS IMPLEMENT_ABSTRACT_CLASS(wxProtocol, wxSocketClient) +#else +IMPLEMENT_ABSTRACT_CLASS(wxProtocol, wxObject) +#endif wxProtocol::wxProtocol() +#if wxUSE_SOCKETS : wxSocketClient() +#endif { } +#if wxUSE_SOCKETS bool wxProtocol::Reconnect() { wxIPV4address addr; - if (!GetPeer(addr)) { + if (!GetPeer(addr)) + { Close(); return FALSE; } @@ -74,6 +78,7 @@ bool wxProtocol::Reconnect() return FALSE; if (!Connect(addr)) return FALSE; + return TRUE; } @@ -85,7 +90,7 @@ wxProtocolError GetLine(wxSocketBase *sock, wxString& result) { bool found; avail = sock->Read(tmp_buf, PROTO_BSIZE).LastCount(); - if (sock->LastError() != 0 || avail == 0) + if (sock->Error() || avail == 0) return wxPROTO_NETERR; memcpy(tmp_str, tmp_buf, avail); @@ -107,10 +112,11 @@ wxProtocolError GetLine(wxSocketBase *sock, wxString& result) { result = result.Left(result.Length()-1); size = ret-tmp_str+1; - sock->CreatePushbackBefore(&tmp_buf[size], avail-size); + sock->Unread(&tmp_buf[size], avail-size); return wxPROTO_NOERR; #undef PROTO_BSIZE } +#endif // ---------------------------------------------------------------------- // Module @@ -124,21 +130,29 @@ public: void OnExit(); }; -#if !USE_SHARED_LIBRARY IMPLEMENT_DYNAMIC_CLASS(wxProtocolModule, wxModule) -#endif bool wxProtocolModule::OnInit() { - wxURL::g_proxy = new wxHTTP(); +#if wxUSE_SOCKETS + char *env_http_prox; + + wxURL::g_proxy = NULL; + // Initialize the proxy when HTTP_PROXY is defined + env_http_prox = getenv("HTTP_PROXY"); + if (env_http_prox) + wxURL::SetDefaultProxy(env_http_prox); +#endif + return TRUE; } void wxProtocolModule::OnExit() { - delete wxURL::g_proxy; +#if wxUSE_SOCKETS + if (wxURL::g_proxy) + delete wxURL::g_proxy; wxURL::g_proxy = NULL; +#endif } -#endif - // wxUSE_SOCKETS