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 |
7 |
// RCS-ID: $Id: http.cpp 44660 2007-03-07 23:07:17Z VZ $ |
8 |
// Copyright: (c) 1997, 1998 Guilhem Lavaux |
9 |
// Licence: wxWindows licence |
10 |
///////////////////////////////////////////////////////////////////////////// |
11 |
|
12 |
// For compilers that support precompilation, includes "wx.h". |
13 |
#include "wx/wxprec.h" |
14 |
|
15 |
#ifdef __BORLANDC__ |
16 |
#pragma hdrstop |
17 |
#endif |
18 |
|
19 |
#if wxUSE_PROTOCOL_HTTP |
20 |
|
21 |
#include <stdio.h> |
22 |
#include <stdlib.h> |
23 |
|
24 |
#ifndef WX_PRECOMP |
25 |
#include "wx/string.h" |
26 |
#include "wx/app.h" |
27 |
#endif |
28 |
|
29 |
#include "wx/tokenzr.h" |
30 |
#include "wx/socket.h" |
31 |
#include "wx/protocol/protocol.h" |
32 |
#include "wx/url.h" |
33 |
#include "wx/protocol/http.h" |
34 |
#include "wx/sckstrm.h" |
35 |
|
36 |
IMPLEMENT_DYNAMIC_CLASS(wxHTTP, wxProtocol) |
37 |
IMPLEMENT_PROTOCOL(wxHTTP, wxT("http"), wxT("80"), true) |
38 |
|
39 |
wxHTTP::wxHTTP() |
40 |
: wxProtocol() |
41 |
{ |
42 |
m_addr = NULL; |
43 |
m_read = false; |
44 |
m_proxy_mode = false; |
45 |
m_post_buf = wxEmptyString; |
46 |
m_http_response = 0; |
47 |
|
48 |
SetNotify(wxSOCKET_LOST_FLAG); |
49 |
} |
50 |
|
51 |
wxHTTP::~wxHTTP() |
52 |
{ |
53 |
ClearHeaders(); |
54 |
|
55 |
delete m_addr; |
56 |
} |
57 |
|
58 |
void wxHTTP::ClearHeaders() |
59 |
{ |
60 |
m_headers.clear(); |
61 |
} |
62 |
|
63 |
wxString wxHTTP::GetContentType() |
64 |
{ |
65 |
return GetHeader(wxT("Content-Type")); |
66 |
} |
67 |
|
68 |
void wxHTTP::SetProxyMode(bool on) |
69 |
{ |
70 |
m_proxy_mode = on; |
71 |
} |
72 |
|
73 |
wxHTTP::wxHeaderIterator wxHTTP::FindHeader(const wxString& header) |
74 |
{ |
75 |
wxHeaderIterator it = m_headers.begin(); |
76 |
for ( wxHeaderIterator en = m_headers.end(); it != en; ++it ) |
77 |
{ |
78 |
if ( wxStricmp(it->first, header) == 0 ) |
79 |
break; |
80 |
} |
81 |
|
82 |
return it; |
83 |
} |
84 |
|
85 |
wxHTTP::wxHeaderConstIterator wxHTTP::FindHeader(const wxString& header) const |
86 |
{ |
87 |
wxHeaderConstIterator it = m_headers.begin(); |
88 |
for ( wxHeaderConstIterator en = m_headers.end(); it != en; ++it ) |
89 |
{ |
90 |
if ( wxStricmp(it->first, header) == 0 ) |
91 |
break; |
92 |
} |
93 |
|
94 |
return it; |
95 |
} |
96 |
|
97 |
void wxHTTP::SetHeader(const wxString& header, const wxString& h_data) |
98 |
{ |
99 |
if (m_read) { |
100 |
ClearHeaders(); |
101 |
m_read = false; |
102 |
} |
103 |
|
104 |
wxHeaderIterator it = FindHeader(header); |
105 |
if (it != m_headers.end()) |
106 |
it->second = h_data; |
107 |
else |
108 |
m_headers[header] = h_data; |
109 |
} |
110 |
|
111 |
wxString wxHTTP::GetHeader(const wxString& header) const |
112 |
{ |
113 |
wxHeaderConstIterator it = FindHeader(header); |
114 |
|
115 |
return it == m_headers.end() ? wxGetEmptyString() : it->second; |
116 |
} |
117 |
|
118 |
wxString wxHTTP::GenerateAuthString(const wxString& user, const wxString& pass) const |
119 |
{ |
120 |
static const char *base64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; |
121 |
|
122 |
wxString buf; |
123 |
wxString toencode; |
124 |
|
125 |
buf.Printf(wxT("Basic ")); |
126 |
|
127 |
toencode.Printf(wxT("%s:%s"),user.c_str(),pass.c_str()); |
128 |
|
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]); |
134 |
from += 3; |
135 |
len -= 3; |
136 |
} |
137 |
if (len > 0) { // pad the remaining characters |
138 |
buf << wxString::Format(wxT("%c"), base64[(from[0] >> 2) & 0x3f]); |
139 |
if (len == 1) { |
140 |
buf << wxString::Format(wxT("%c="), base64[(from[0] << 4) & 0x30]); |
141 |
} else { |
142 |
buf << wxString::Format(wxT("%c%c"), base64[((from[0] << 4) & 0x30) | ((from[1] >> 4) & 0xf)], base64[(from[1] << 2) & 0x3c]); |
143 |
} |
144 |
buf << wxString::Format(wxT("=")); |
145 |
} |
146 |
|
147 |
return buf; |
148 |
} |
149 |
|
150 |
void wxHTTP::SetPostBuffer(const wxString& post_buf) |
151 |
{ |
152 |
m_post_buf = post_buf; |
153 |
} |
154 |
|
155 |
void wxHTTP::SendHeaders() |
156 |
{ |
157 |
typedef wxStringToStringHashMap::iterator iterator; |
158 |
wxString buf; |
159 |
|
160 |
for (iterator it = m_headers.begin(), en = m_headers.end(); it != en; ++it ) |
161 |
{ |
162 |
buf.Printf(wxT("%s: %s\r\n"), it->first.c_str(), it->second.c_str()); |
163 |
|
164 |
const wxWX2MBbuf cbuf = buf.mb_str(); |
165 |
Write(cbuf, strlen(cbuf)); |
166 |
} |
167 |
} |
168 |
|
169 |
bool wxHTTP::ParseHeaders() |
170 |
{ |
171 |
wxString line; |
172 |
wxStringTokenizer tokenzr; |
173 |
|
174 |
ClearHeaders(); |
175 |
m_read = true; |
176 |
|
177 |
for ( ;; ) |
178 |
{ |
179 |
m_perr = ReadLine(this, line); |
180 |
if (m_perr != wxPROTO_NOERR) |
181 |
return false; |
182 |
|
183 |
if (line.length() == 0) |
184 |
break; |
185 |
|
186 |
wxString left_str = line.BeforeFirst(':'); |
187 |
m_headers[left_str] = line.AfterFirst(':').Strip(wxString::both); |
188 |
} |
189 |
return true; |
190 |
} |
191 |
|
192 |
bool wxHTTP::Connect(const wxString& host, unsigned short port) |
193 |
{ |
194 |
wxIPV4address *addr; |
195 |
|
196 |
if (m_addr) { |
197 |
delete m_addr; |
198 |
m_addr = NULL; |
199 |
Close(); |
200 |
} |
201 |
|
202 |
m_addr = addr = new wxIPV4address(); |
203 |
|
204 |
if (!addr->Hostname(host)) { |
205 |
delete m_addr; |
206 |
m_addr = NULL; |
207 |
m_perr = wxPROTO_NETERR; |
208 |
return false; |
209 |
} |
210 |
|
211 |
if ( port ) |
212 |
addr->Service(port); |
213 |
else if (!addr->Service(wxT("http"))) |
214 |
addr->Service(80); |
215 |
|
216 |
SetHeader(wxT("Host"), host); |
217 |
|
218 |
return true; |
219 |
} |
220 |
|
221 |
bool wxHTTP::Connect(wxSockAddress& addr, bool WXUNUSED(wait)) |
222 |
{ |
223 |
if (m_addr) { |
224 |
delete m_addr; |
225 |
Close(); |
226 |
} |
227 |
|
228 |
m_addr = addr.Clone(); |
229 |
|
230 |
wxIPV4address *ipv4addr = wxDynamicCast(&addr, wxIPV4address); |
231 |
if (ipv4addr) |
232 |
SetHeader(wxT("Host"), ipv4addr->OrigHostname()); |
233 |
|
234 |
return true; |
235 |
} |
236 |
|
237 |
bool wxHTTP::BuildRequest(const wxString& path, wxHTTP_Req req) |
238 |
{ |
239 |
const wxChar *request; |
240 |
|
241 |
switch (req) |
242 |
{ |
243 |
case wxHTTP_GET: |
244 |
request = wxT("GET"); |
245 |
break; |
246 |
|
247 |
case wxHTTP_POST: |
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() ) ); |
251 |
break; |
252 |
|
253 |
default: |
254 |
return false; |
255 |
} |
256 |
|
257 |
m_http_response = 0; |
258 |
|
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")); |
262 |
|
263 |
// Send authentication information |
264 |
if (!m_username.empty() || !m_password.empty()) { |
265 |
SetHeader(wxT("Authorization"), GenerateAuthString(m_username, m_password)); |
266 |
} |
267 |
|
268 |
SaveState(); |
269 |
|
270 |
// we may use non blocking sockets only if we can dispatch events from them |
271 |
SetFlags( wxIsMainThread() && wxApp::IsMainLoopRunning() ? wxSOCKET_NONE |
272 |
: wxSOCKET_BLOCK ); |
273 |
Notify(false); |
274 |
|
275 |
wxString buf; |
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)); |
279 |
SendHeaders(); |
280 |
Write("\r\n", 2); |
281 |
|
282 |
if ( req == wxHTTP_POST ) { |
283 |
Write(m_post_buf.mbc_str(), m_post_buf.Len()); |
284 |
m_post_buf = wxEmptyString; |
285 |
} |
286 |
|
287 |
wxString tmp_str; |
288 |
m_perr = ReadLine(this, tmp_str); |
289 |
if (m_perr != wxPROTO_NOERR) { |
290 |
RestoreState(); |
291 |
return false; |
292 |
} |
293 |
|
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")); |
299 |
RestoreState(); |
300 |
return true; |
301 |
} |
302 |
|
303 |
wxStringTokenizer token(tmp_str,wxT(' ')); |
304 |
wxString tmp_str2; |
305 |
bool ret_value; |
306 |
|
307 |
token.NextToken(); |
308 |
tmp_str2 = token.NextToken(); |
309 |
|
310 |
m_http_response = wxAtoi(tmp_str2); |
311 |
|
312 |
switch (tmp_str2[0u]) |
313 |
{ |
314 |
case wxT('1'): |
315 |
/* INFORMATION / SUCCESS */ |
316 |
break; |
317 |
|
318 |
case wxT('2'): |
319 |
/* SUCCESS */ |
320 |
break; |
321 |
|
322 |
case wxT('3'): |
323 |
/* REDIRECTION */ |
324 |
break; |
325 |
|
326 |
default: |
327 |
m_perr = wxPROTO_NOFILE; |
328 |
RestoreState(); |
329 |
return false; |
330 |
} |
331 |
|
332 |
ret_value = ParseHeaders(); |
333 |
RestoreState(); |
334 |
return ret_value; |
335 |
} |
336 |
|
337 |
class wxHTTPStream : public wxSocketInputStream |
338 |
{ |
339 |
public: |
340 |
wxHTTP *m_http; |
341 |
size_t m_httpsize; |
342 |
unsigned long m_read_bytes; |
343 |
|
344 |
wxHTTPStream(wxHTTP *http) : wxSocketInputStream(*http), m_http(http) {} |
345 |
size_t GetSize() const { return m_httpsize; } |
346 |
virtual ~wxHTTPStream(void) { m_http->Abort(); } |
347 |
|
348 |
protected: |
349 |
size_t OnSysRead(void *buffer, size_t bufsize); |
350 |
|
351 |
DECLARE_NO_COPY_CLASS(wxHTTPStream) |
352 |
}; |
353 |
|
354 |
size_t wxHTTPStream::OnSysRead(void *buffer, size_t bufsize) |
355 |
{ |
356 |
if (m_httpsize > 0 && m_read_bytes >= m_httpsize) |
357 |
{ |
358 |
m_lasterror = wxSTREAM_EOF; |
359 |
return 0; |
360 |
} |
361 |
|
362 |
size_t ret = wxSocketInputStream::OnSysRead(buffer, bufsize); |
363 |
m_read_bytes += ret; |
364 |
|
365 |
if (m_httpsize==(size_t)-1 && m_lasterror == wxSTREAM_READ_ERROR ) |
366 |
{ |
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 ; |
372 |
} |
373 |
|
374 |
return ret; |
375 |
} |
376 |
|
377 |
bool wxHTTP::Abort(void) |
378 |
{ |
379 |
return wxSocketClient::Close(); |
380 |
} |
381 |
|
382 |
wxInputStream *wxHTTP::GetInputStream(const wxString& path) |
383 |
{ |
384 |
wxHTTPStream *inp_stream; |
385 |
|
386 |
wxString new_path; |
387 |
|
388 |
m_perr = wxPROTO_CONNERR; |
389 |
if (!m_addr) |
390 |
return NULL; |
391 |
|
392 |
// We set m_connected back to false so wxSocketBase will know what to do. |
393 |
#ifdef __WXMAC__ |
394 |
wxSocketClient::Connect(*m_addr , false ); |
395 |
wxSocketClient::WaitOnConnect(10); |
396 |
|
397 |
if (!wxSocketClient::IsConnected()) |
398 |
return NULL; |
399 |
#else |
400 |
if (!wxProtocol::Connect(*m_addr)) |
401 |
return NULL; |
402 |
#endif |
403 |
|
404 |
if (!BuildRequest(path, m_post_buf.empty() ? wxHTTP_GET : wxHTTP_POST)) |
405 |
return NULL; |
406 |
|
407 |
inp_stream = new wxHTTPStream(this); |
408 |
|
409 |
if (!GetHeader(wxT("Content-Length")).empty()) |
410 |
inp_stream->m_httpsize = wxAtoi(WXSTRINGCAST GetHeader(wxT("Content-Length"))); |
411 |
else |
412 |
inp_stream->m_httpsize = (size_t)-1; |
413 |
|
414 |
inp_stream->m_read_bytes = 0; |
415 |
|
416 |
Notify(false); |
417 |
SetFlags(wxSOCKET_BLOCK | wxSOCKET_WAITALL); |
418 |
|
419 |
return inp_stream; |
420 |
} |
421 |
|
422 |
#endif // wxUSE_PROTOCOL_HTTP |