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"
36 IMPLEMENT_DYNAMIC_CLASS(wxHTTP
, wxProtocol
)
37 IMPLEMENT_PROTOCOL(wxHTTP
, wxT("http"), wxT("80"), true)
45 m_post_buf
= wxEmptyString
;
48 SetNotify(wxSOCKET_LOST_FLAG
);
58 void wxHTTP::ClearHeaders()
63 wxString
wxHTTP::GetContentType()
65 return GetHeader(wxT("Content-Type"));
68 void wxHTTP::SetProxyMode(bool on
)
73 wxHTTP::wxHeaderIterator
wxHTTP::FindHeader(const wxString
& header
)
75 wxHeaderIterator it
= m_headers
.begin();
76 for ( wxHeaderIterator en
= m_headers
.end(); it
!= en
; ++it
)
78 if ( wxStricmp(it
->first
, header
) == 0 )
85 wxHTTP::wxHeaderConstIterator
wxHTTP::FindHeader(const wxString
& header
) const
87 wxHeaderConstIterator it
= m_headers
.begin();
88 for ( wxHeaderConstIterator en
= m_headers
.end(); it
!= en
; ++it
)
90 if ( wxStricmp(it
->first
, header
) == 0 )
97 void wxHTTP::SetHeader(const wxString
& header
, const wxString
& h_data
)
104 wxHeaderIterator it
= FindHeader(header
);
105 if (it
!= m_headers
.end())
108 m_headers
[header
] = h_data
;
111 wxString
wxHTTP::GetHeader(const wxString
& header
) const
113 wxHeaderConstIterator it
= FindHeader(header
);
115 return it
== m_headers
.end() ? wxGetEmptyString() : it
->second
;
118 wxString
wxHTTP::GenerateAuthString(const wxString
& user
, const wxString
& pass
) const
120 static const char *base64
= "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
125 buf
.Printf(wxT("Basic "));
127 toencode
.Printf(wxT("%s:%s"),user
.c_str(),pass
.c_str());
129 size_t len
= toencode
.length();
130 const wxChar
*from
= toencode
.c_str();
131 while (len
>= 3) { // encode full blocks first
132 buf
<< wxString::Format(wxT("%c%c"), base64
[(from
[0] >> 2) & 0x3f], base64
[((from
[0] << 4) & 0x30) | ((from
[1] >> 4) & 0xf)]);
133 buf
<< wxString::Format(wxT("%c%c"), base64
[((from
[1] << 2) & 0x3c) | ((from
[2] >> 6) & 0x3)], base64
[from
[2] & 0x3f]);
137 if (len
> 0) { // pad the remaining characters
138 buf
<< wxString::Format(wxT("%c"), base64
[(from
[0] >> 2) & 0x3f]);
140 buf
<< wxString::Format(wxT("%c="), base64
[(from
[0] << 4) & 0x30]);
142 buf
<< wxString::Format(wxT("%c%c"), base64
[((from
[0] << 4) & 0x30) | ((from
[1] >> 4) & 0xf)], base64
[(from
[1] << 2) & 0x3c]);
150 void wxHTTP::SetPostBuffer(const wxString
& post_buf
)
152 m_post_buf
= post_buf
;
155 void wxHTTP::SendHeaders()
157 typedef wxStringToStringHashMap::iterator iterator
;
160 for (iterator it
= m_headers
.begin(), en
= m_headers
.end(); it
!= en
; ++it
)
162 buf
.Printf(wxT("%s: %s\r\n"), it
->first
.c_str(), it
->second
.c_str());
164 const wxWX2MBbuf cbuf
= buf
.mb_str();
165 Write(cbuf
, strlen(cbuf
));
169 bool wxHTTP::ParseHeaders()
172 wxStringTokenizer tokenzr
;
179 m_perr
= ReadLine(this, line
);
180 if (m_perr
!= wxPROTO_NOERR
)
183 if (line
.length() == 0)
186 wxString left_str
= line
.BeforeFirst(':');
187 m_headers
[left_str
] = line
.AfterFirst(':').Strip(wxString::both
);
192 bool wxHTTP::Connect(const wxString
& host
, unsigned short port
)
202 m_addr
= addr
= new wxIPV4address();
204 if (!addr
->Hostname(host
)) {
207 m_perr
= wxPROTO_NETERR
;
213 else if (!addr
->Service(wxT("http")))
216 SetHeader(wxT("Host"), host
);
221 bool wxHTTP::Connect(wxSockAddress
& addr
, bool WXUNUSED(wait
))
228 m_addr
= addr
.Clone();
230 wxIPV4address
*ipv4addr
= wxDynamicCast(&addr
, wxIPV4address
);
232 SetHeader(wxT("Host"), ipv4addr
->OrigHostname());
237 bool wxHTTP::BuildRequest(const wxString
& path
, wxHTTP_Req req
)
239 const wxChar
*request
;
244 request
= wxT("GET");
248 request
= wxT("POST");
249 if ( GetHeader( wxT("Content-Length") ).IsNull() )
250 SetHeader( wxT("Content-Length"), wxString::Format( wxT("%lu"), (unsigned long)m_post_buf
.Len() ) );
259 // If there is no User-Agent defined, define it.
260 if (GetHeader(wxT("User-Agent")).IsNull())
261 SetHeader(wxT("User-Agent"), wxT("wxWidgets 2.x"));
263 // Send authentication information
264 if (!m_username
.empty() || !m_password
.empty()) {
265 SetHeader(wxT("Authorization"), GenerateAuthString(m_username
, m_password
));
270 // we may use non blocking sockets only if we can dispatch events from them
271 SetFlags( wxIsMainThread() && wxApp::IsMainLoopRunning() ? wxSOCKET_NONE
276 buf
.Printf(wxT("%s %s HTTP/1.0\r\n"), request
, path
.c_str());
277 const wxWX2MBbuf pathbuf
= wxConvLocal
.cWX2MB(buf
);
278 Write(pathbuf
, strlen(wxMBSTRINGCAST pathbuf
));
282 if ( req
== wxHTTP_POST
) {
283 Write(m_post_buf
.mbc_str(), m_post_buf
.Len());
284 m_post_buf
= wxEmptyString
;
288 m_perr
= ReadLine(this, tmp_str
);
289 if (m_perr
!= wxPROTO_NOERR
) {
294 if (!tmp_str
.Contains(wxT("HTTP/"))) {
295 // TODO: support HTTP v0.9 which can have no header.
296 // FIXME: tmp_str is not put back in the in-queue of the socket.
297 SetHeader(wxT("Content-Length"), wxT("-1"));
298 SetHeader(wxT("Content-Type"), wxT("none/none"));
303 wxStringTokenizer
token(tmp_str
,wxT(' '));
308 tmp_str2
= token
.NextToken();
310 m_http_response
= wxAtoi(tmp_str2
);
312 switch ( tmp_str2
[0u].GetValue() )
315 /* INFORMATION / SUCCESS */
327 m_perr
= wxPROTO_NOFILE
;
332 ret_value
= ParseHeaders();
337 class wxHTTPStream
: public wxSocketInputStream
342 unsigned long m_read_bytes
;
344 wxHTTPStream(wxHTTP
*http
) : wxSocketInputStream(*http
), m_http(http
) {}
345 size_t GetSize() const { return m_httpsize
; }
346 virtual ~wxHTTPStream(void) { m_http
->Abort(); }
349 size_t OnSysRead(void *buffer
, size_t bufsize
);
351 DECLARE_NO_COPY_CLASS(wxHTTPStream
)
354 size_t wxHTTPStream::OnSysRead(void *buffer
, size_t bufsize
)
356 if (m_httpsize
> 0 && m_read_bytes
>= m_httpsize
)
358 m_lasterror
= wxSTREAM_EOF
;
362 size_t ret
= wxSocketInputStream::OnSysRead(buffer
, bufsize
);
365 if (m_httpsize
==(size_t)-1 && m_lasterror
== wxSTREAM_READ_ERROR
)
367 // if m_httpsize is (size_t) -1 this means read until connection closed
368 // which is equivalent to getting a READ_ERROR, for clients however this
369 // must be translated into EOF, as it is the expected way of signalling
370 // end end of the content
371 m_lasterror
= wxSTREAM_EOF
;
377 bool wxHTTP::Abort(void)
379 return wxSocketClient::Close();
382 wxInputStream
*wxHTTP::GetInputStream(const wxString
& path
)
384 wxHTTPStream
*inp_stream
;
388 m_perr
= wxPROTO_CONNERR
;
392 // We set m_connected back to false so wxSocketBase will know what to do.
394 wxSocketClient::Connect(*m_addr
, false );
395 wxSocketClient::WaitOnConnect(10);
397 if (!wxSocketClient::IsConnected())
400 if (!wxProtocol::Connect(*m_addr
))
404 if (!BuildRequest(path
, m_post_buf
.empty() ? wxHTTP_GET
: wxHTTP_POST
))
407 inp_stream
= new wxHTTPStream(this);
409 if (!GetHeader(wxT("Content-Length")).empty())
410 inp_stream
->m_httpsize
= wxAtoi(WXSTRINGCAST
GetHeader(wxT("Content-Length")));
412 inp_stream
->m_httpsize
= (size_t)-1;
414 inp_stream
->m_read_bytes
= 0;
417 SetFlags(wxSOCKET_BLOCK
| wxSOCKET_WAITALL
);
422 #endif // wxUSE_PROTOCOL_HTTP