1 /////////////////////////////////////////////////////////////////////////////
2 // Name: src/common/http.cpp
3 // Purpose: HTTP protocol
4 // Author: Guilhem Lavaux
5 // Modified by: Simo Virokannas (authentication, Dec 2005)
6 // Created: August 1997
8 // Copyright: (c) 1997, 1998 Guilhem Lavaux
9 // Licence: wxWindows licence
10 /////////////////////////////////////////////////////////////////////////////
12 // For compilers that support precompilation, includes "wx.h".
13 #include "wx/wxprec.h"
19 #if wxUSE_PROTOCOL_HTTP
25 #include "wx/string.h"
29 #include "wx/tokenzr.h"
30 #include "wx/socket.h"
31 #include "wx/protocol/protocol.h"
33 #include "wx/protocol/http.h"
34 #include "wx/sckstrm.h"
35 #include "wx/thread.h"
38 // ----------------------------------------------------------------------------
40 // ----------------------------------------------------------------------------
42 IMPLEMENT_DYNAMIC_CLASS(wxHTTP
, wxProtocol
)
43 IMPLEMENT_PROTOCOL(wxHTTP
, wxT("http"), wxT("80"), true)
53 SetNotify(wxSOCKET_LOST_FLAG
);
63 void wxHTTP::ClearHeaders()
68 void wxHTTP::ClearCookies()
73 wxString
wxHTTP::GetContentType() const
75 return GetHeader(wxT("Content-Type"));
78 void wxHTTP::SetProxyMode(bool on
)
83 wxHTTP::wxHeaderIterator
wxHTTP::FindHeader(const wxString
& header
)
85 wxHeaderIterator it
= m_headers
.begin();
86 for ( wxHeaderIterator en
= m_headers
.end(); it
!= en
; ++it
)
88 if ( header
.CmpNoCase(it
->first
) == 0 )
95 wxHTTP::wxHeaderConstIterator
wxHTTP::FindHeader(const wxString
& header
) const
97 wxHeaderConstIterator it
= m_headers
.begin();
98 for ( wxHeaderConstIterator en
= m_headers
.end(); it
!= en
; ++it
)
100 if ( header
.CmpNoCase(it
->first
) == 0 )
107 wxHTTP::wxCookieIterator
wxHTTP::FindCookie(const wxString
& cookie
)
109 wxCookieIterator it
= m_cookies
.begin();
110 for ( wxCookieIterator en
= m_cookies
.end(); it
!= en
; ++it
)
112 if ( cookie
.CmpNoCase(it
->first
) == 0 )
119 wxHTTP::wxCookieConstIterator
wxHTTP::FindCookie(const wxString
& cookie
) const
121 wxCookieConstIterator it
= m_cookies
.begin();
122 for ( wxCookieConstIterator en
= m_cookies
.end(); it
!= en
; ++it
)
124 if ( cookie
.CmpNoCase(it
->first
) == 0 )
131 void wxHTTP::SetHeader(const wxString
& header
, const wxString
& h_data
)
138 wxHeaderIterator it
= FindHeader(header
);
139 if (it
!= m_headers
.end())
142 m_headers
[header
] = h_data
;
145 wxString
wxHTTP::GetHeader(const wxString
& header
) const
147 wxHeaderConstIterator it
= FindHeader(header
);
149 return it
== m_headers
.end() ? wxGetEmptyString() : it
->second
;
152 wxString
wxHTTP::GetCookie(const wxString
& cookie
) const
154 wxCookieConstIterator it
= FindCookie(cookie
);
156 return it
== m_cookies
.end() ? wxGetEmptyString() : it
->second
;
159 wxString
wxHTTP::GenerateAuthString(const wxString
& user
, const wxString
& pass
) const
161 // TODO: Use wxBase64Encode() now that we have it instead of reproducing it
163 static const char *base64
= "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
168 buf
.Printf(wxT("Basic "));
170 toencode
.Printf(wxT("%s:%s"),user
.c_str(),pass
.c_str());
172 size_t len
= toencode
.length();
173 const wxChar
*from
= toencode
.c_str();
174 while (len
>= 3) { // encode full blocks first
175 buf
<< wxString::Format(wxT("%c%c"), base64
[(from
[0] >> 2) & 0x3f], base64
[((from
[0] << 4) & 0x30) | ((from
[1] >> 4) & 0xf)]);
176 buf
<< wxString::Format(wxT("%c%c"), base64
[((from
[1] << 2) & 0x3c) | ((from
[2] >> 6) & 0x3)], base64
[from
[2] & 0x3f]);
180 if (len
> 0) { // pad the remaining characters
181 buf
<< wxString::Format(wxT("%c"), base64
[(from
[0] >> 2) & 0x3f]);
183 buf
<< wxString::Format(wxT("%c="), base64
[(from
[0] << 4) & 0x30]);
185 buf
<< wxString::Format(wxT("%c%c"), base64
[((from
[0] << 4) & 0x30) | ((from
[1] >> 4) & 0xf)], base64
[(from
[1] << 2) & 0x3c]);
193 void wxHTTP::SetPostBuffer(const wxString
& post_buf
)
195 // Use To8BitData() for backwards compatibility in this deprecated method.
196 // The new code should use the other overload or SetPostText() and specify
197 // the encoding to use for the text explicitly.
198 wxScopedCharBuffer scb
= post_buf
.To8BitData();
201 m_postBuffer
.Clear();
202 m_postBuffer
.AppendData(scb
.data(), scb
.length());
207 wxHTTP::SetPostBuffer(const wxString
& contentType
,
208 const wxMemoryBuffer
& data
)
211 m_contentType
= contentType
;
213 return !m_postBuffer
.IsEmpty();
217 wxHTTP::SetPostText(const wxString
& contentType
,
218 const wxString
& data
,
219 const wxMBConv
& conv
)
221 wxScopedCharBuffer scb
= data
.mb_str(conv
);
225 m_postBuffer
.Clear();
226 m_postBuffer
.AppendData(scb
.data(), scb
.length());
227 m_contentType
= contentType
;
232 void wxHTTP::SendHeaders()
234 typedef wxStringToStringHashMap::iterator iterator
;
237 for (iterator it
= m_headers
.begin(), en
= m_headers
.end(); it
!= en
; ++it
)
239 buf
.Printf(wxT("%s: %s\r\n"), it
->first
.c_str(), it
->second
.c_str());
241 const wxWX2MBbuf cbuf
= buf
.mb_str();
242 Write(cbuf
, strlen(cbuf
));
246 bool wxHTTP::ParseHeaders()
249 wxStringTokenizer tokenzr
;
257 m_lastError
= ReadLine(this, line
);
258 if (m_lastError
!= wxPROTO_NOERR
)
264 wxString left_str
= line
.BeforeFirst(':');
265 if(!left_str
.CmpNoCase("Set-Cookie"))
267 wxString cookieName
= line
.AfterFirst(':').Strip(wxString::both
).BeforeFirst('=');
268 wxString cookieValue
= line
.AfterFirst(':').Strip(wxString::both
).AfterFirst('=').BeforeFirst(';');
269 m_cookies
[cookieName
] = cookieValue
;
272 m_headers
[left_str
] = line
.AfterFirst(':').Strip(wxString::both
);
276 m_headers
[left_str
] = line
.AfterFirst(':').Strip(wxString::both
);
282 bool wxHTTP::Connect(const wxString
& host
, unsigned short port
)
291 m_addr
= addr
= new wxIPV4address();
293 if (!addr
->Hostname(host
)) {
295 m_lastError
= wxPROTO_NETERR
;
301 else if (!addr
->Service(wxT("http")))
304 wxString hostHdr
= host
;
305 if ( port
&& port
!= 80 )
306 hostHdr
<< wxT(":") << port
;
307 SetHeader(wxT("Host"), hostHdr
);
309 m_lastError
= wxPROTO_NOERR
;
313 bool wxHTTP::Connect(const wxSockAddress
& addr
, bool WXUNUSED(wait
))
320 m_addr
= addr
.Clone();
322 wxIPV4address
*ipv4addr
= wxDynamicCast(&addr
, wxIPV4address
);
325 wxString hostHdr
= ipv4addr
->OrigHostname();
326 unsigned short port
= ipv4addr
->Service();
327 if ( port
&& port
!= 80 )
328 hostHdr
<< wxT(":") << port
;
329 SetHeader(wxT("Host"), hostHdr
);
332 m_lastError
= wxPROTO_NOERR
;
336 bool wxHTTP::BuildRequest(const wxString
& path
, wxHTTP_Req req
)
338 const wxChar
*request
;
343 request
= wxT("GET");
347 request
= wxT("POST");
348 // Content length must be correct, so always set, possibly
349 // overriding the value set explicitly by a previous call to
350 // SetHeader("Content-Length").
351 if ( !m_postBuffer
.IsEmpty() )
354 len
<< m_postBuffer
.GetDataLen();
356 SetHeader(wxS("Content-Length"), len
);
359 // However if the user had explicitly set the content type, don't
360 // override it with the content type passed to SetPostText().
361 if ( !m_contentType
.empty() && GetContentType().empty() )
362 SetHeader(wxS("Content-Type"), m_contentType
);
371 // If there is no User-Agent defined, define it.
372 if ( GetHeader(wxT("User-Agent")).empty() )
373 SetHeader(wxT("User-Agent"), wxT("wxWidgets 2.x"));
375 // Send authentication information
376 if (!m_username
.empty() || !m_password
.empty()) {
377 SetHeader(wxT("Authorization"), GenerateAuthString(m_username
, m_password
));
382 // we may use non blocking sockets only if we can dispatch events from them
383 SetFlags( wxIsMainThread() && wxApp::IsMainLoopRunning() ? wxSOCKET_NONE
388 buf
.Printf(wxT("%s %s HTTP/1.0\r\n"), request
, path
.c_str());
389 const wxWX2MBbuf pathbuf
= buf
.mb_str();
390 Write(pathbuf
, strlen(pathbuf
));
394 if ( req
== wxHTTP_POST
) {
395 if ( !m_postBuffer
.IsEmpty() )
396 Write(m_postBuffer
.GetData(), m_postBuffer
.GetDataLen());
398 m_postBuffer
.Clear();
402 m_lastError
= ReadLine(this, tmp_str
);
403 if (m_lastError
!= wxPROTO_NOERR
) {
408 if (!tmp_str
.Contains(wxT("HTTP/"))) {
409 // TODO: support HTTP v0.9 which can have no header.
410 // FIXME: tmp_str is not put back in the in-queue of the socket.
411 m_lastError
= wxPROTO_NOERR
;
412 SetHeader(wxT("Content-Length"), wxT("-1"));
413 SetHeader(wxT("Content-Type"), wxT("none/none"));
418 wxStringTokenizer
token(tmp_str
,wxT(' '));
423 tmp_str2
= token
.NextToken();
425 m_http_response
= wxAtoi(tmp_str2
);
427 switch ( tmp_str2
[0u].GetValue() )
430 /* INFORMATION / SUCCESS */
442 m_lastError
= wxPROTO_NOFILE
;
447 m_lastError
= wxPROTO_NOERR
;
448 ret_value
= ParseHeaders();
453 bool wxHTTP::Abort(void)
455 return wxSocketClient::Close();
458 // ----------------------------------------------------------------------------
459 // wxHTTPStream and wxHTTP::GetInputStream
460 // ----------------------------------------------------------------------------
462 class wxHTTPStream
: public wxSocketInputStream
467 unsigned long m_read_bytes
;
469 wxHTTPStream(wxHTTP
*http
) : wxSocketInputStream(*http
), m_http(http
) {}
470 size_t GetSize() const { return m_httpsize
; }
471 virtual ~wxHTTPStream(void) { m_http
->Abort(); }
474 size_t OnSysRead(void *buffer
, size_t bufsize
);
476 wxDECLARE_NO_COPY_CLASS(wxHTTPStream
);
479 size_t wxHTTPStream::OnSysRead(void *buffer
, size_t bufsize
)
481 if (m_read_bytes
>= m_httpsize
)
483 m_lasterror
= wxSTREAM_EOF
;
487 size_t ret
= wxSocketInputStream::OnSysRead(buffer
, bufsize
);
490 if (m_httpsize
==(size_t)-1 && m_lasterror
== wxSTREAM_READ_ERROR
)
492 // if m_httpsize is (size_t) -1 this means read until connection closed
493 // which is equivalent to getting a READ_ERROR, for clients however this
494 // must be translated into EOF, as it is the expected way of signalling
495 // end end of the content
496 m_lasterror
= wxSTREAM_EOF
;
502 wxInputStream
*wxHTTP::GetInputStream(const wxString
& path
)
504 wxHTTPStream
*inp_stream
;
508 m_lastError
= wxPROTO_CONNERR
; // all following returns share this type of error
512 // We set m_connected back to false so wxSocketBase will know what to do.
514 wxSocketClient::Connect(*m_addr
, false );
515 wxSocketClient::WaitOnConnect(10);
517 if (!wxSocketClient::IsConnected())
520 if (!wxProtocol::Connect(*m_addr
))
524 if (!BuildRequest(path
, m_postBuffer
.IsEmpty() ? wxHTTP_GET
: wxHTTP_POST
))
527 inp_stream
= new wxHTTPStream(this);
529 if (!GetHeader(wxT("Content-Length")).empty())
530 inp_stream
->m_httpsize
= wxAtoi(GetHeader(wxT("Content-Length")));
532 inp_stream
->m_httpsize
= (size_t)-1;
534 inp_stream
->m_read_bytes
= 0;
537 SetFlags(wxSOCKET_BLOCK
| wxSOCKET_WAITALL
);
539 // no error; reset m_lastError
540 m_lastError
= wxPROTO_NOERR
;
544 #endif // wxUSE_PROTOCOL_HTTP