]>
Commit | Line | Data |
---|---|---|
f4ada568 GL |
1 | ///////////////////////////////////////////////////////////////////////////// |
2 | // Name: http.cpp | |
3 | // Purpose: HTTP protocol | |
4 | // Author: Guilhem Lavaux | |
5 | // Modified by: | |
6 | // Created: August 1997 | |
7 | // RCS-ID: $Id$ | |
8 | // Copyright: (c) 1997, 1998 Guilhem Lavaux | |
9 | // Licence: wxWindows license | |
10 | ///////////////////////////////////////////////////////////////////////////// | |
11 | ||
12 | #ifdef __GNUG__ | |
13 | #pragma implementation "http.h" | |
14 | #endif | |
15 | ||
fcc6dddd JS |
16 | // For compilers that support precompilation, includes "wx.h". |
17 | #include "wx/wxprec.h" | |
18 | ||
19 | #ifdef __BORLANDC__ | |
20 | #pragma hdrstop | |
21 | #endif | |
22 | ||
35a4dab7 GL |
23 | #if wxUSE_SOCKETS |
24 | ||
fcc6dddd JS |
25 | #ifndef WX_PRECOMP |
26 | #endif | |
27 | ||
f4ada568 GL |
28 | #include <stdio.h> |
29 | #include <stdlib.h> | |
30 | #include "wx/string.h" | |
31 | #include "wx/tokenzr.h" | |
32 | #include "wx/socket.h" | |
33 | #include "wx/protocol/protocol.h" | |
34 | #include "wx/protocol/http.h" | |
35 | #include "wx/sckstrm.h" | |
36 | ||
f4ada568 GL |
37 | #if !USE_SHARED_LIBRARY |
38 | IMPLEMENT_DYNAMIC_CLASS(wxHTTP, wxProtocol) | |
4846abaf | 39 | IMPLEMENT_PROTOCOL(wxHTTP, _T("http"), _T("80"), TRUE) |
f4ada568 GL |
40 | #endif |
41 | ||
42 | #define HTTP_BSIZE 2048 | |
43 | ||
44 | wxHTTP::wxHTTP() | |
45 | : wxProtocol(), | |
46 | m_headers(wxKEY_STRING) | |
47 | { | |
48 | m_addr = NULL; | |
49 | m_read = FALSE; | |
50 | ||
51 | SetNotify(REQ_LOST); | |
52 | } | |
53 | ||
54 | wxHTTP::~wxHTTP() | |
55 | { | |
56 | // wxString isn't a wxObject | |
1f01991f | 57 | wxNode *node = m_headers.First(); |
f4ada568 GL |
58 | wxString *string; |
59 | ||
60 | while (node) { | |
61 | string = (wxString *)node->Data(); | |
62 | delete string; | |
63 | node = node->Next(); | |
64 | } | |
65 | } | |
66 | ||
67 | wxString wxHTTP::GetContentType() | |
68 | { | |
4846abaf | 69 | return GetHeader(_T("Content-Type")); |
f4ada568 GL |
70 | } |
71 | ||
72 | void wxHTTP::SetHeader(const wxString& header, const wxString& h_data) | |
73 | { | |
74 | if (m_read) { | |
75 | m_headers.Clear(); | |
76 | m_read = FALSE; | |
77 | } | |
78 | ||
79 | wxNode *node = m_headers.Find(header); | |
80 | ||
81 | if (!node) | |
82 | m_headers.Append(header, (wxObject *)(new wxString(h_data))); | |
83 | else { | |
84 | wxString *str = (wxString *)node->Data(); | |
85 | (*str) = h_data; | |
86 | } | |
87 | } | |
88 | ||
89 | wxString wxHTTP::GetHeader(const wxString& header) | |
90 | { | |
91 | wxNode *node = m_headers.Find(header); | |
92 | if (!node) | |
ce3ed50d | 93 | return wxEmptyString; |
f4ada568 GL |
94 | |
95 | return *((wxString *)node->Data()); | |
96 | } | |
97 | ||
98 | void wxHTTP::SendHeaders() | |
99 | { | |
100 | wxNode *head = m_headers.First(); | |
101 | ||
53c6e7cc VZ |
102 | while (head) |
103 | { | |
f4ada568 | 104 | wxString *str = (wxString *)head->Data(); |
f4ada568 | 105 | |
53c6e7cc | 106 | wxString buf; |
4846abaf | 107 | buf.Printf(_T("%s: %s\n\r"), head->GetKeyString(), str->GetData()); |
53c6e7cc | 108 | |
fde7d98e | 109 | const wxWX2MBbuf cbuf = buf.mb_str(); |
4846abaf | 110 | Write(cbuf, strlen(cbuf)); |
f4ada568 GL |
111 | |
112 | head = head->Next(); | |
113 | } | |
114 | } | |
115 | ||
116 | bool wxHTTP::ParseHeaders() | |
117 | { | |
118 | wxString line; | |
a737331d | 119 | wxStringTokenizer tokenzr; |
f4ada568 GL |
120 | |
121 | m_headers.Clear(); | |
122 | m_read = TRUE; | |
123 | ||
124 | while (1) { | |
125 | m_error = GetLine(this, line); | |
126 | if (m_error != wxPROTO_NOERR) | |
127 | return FALSE; | |
128 | ||
129 | if (line.Length() == 0) | |
130 | break; | |
131 | ||
a737331d GL |
132 | printf("Header: %s\n", WXSTRINGCAST line); |
133 | tokenzr.SetString(line, " :\t\n\r"); | |
134 | if (!tokenzr.HasMoreToken()) | |
f4ada568 GL |
135 | return FALSE; |
136 | ||
a737331d GL |
137 | wxString left_str = tokenzr.GetNextToken(); |
138 | wxString *str = new wxString(tokenzr.GetNextToken()); | |
f4ada568 GL |
139 | |
140 | m_headers.Append(left_str, (wxObject *) str); | |
141 | } | |
142 | return TRUE; | |
143 | } | |
144 | ||
145 | bool wxHTTP::Connect(const wxString& host) | |
146 | { | |
147 | wxIPV4address *addr; | |
148 | ||
149 | if (m_connected) { | |
150 | delete m_addr; | |
151 | m_addr = NULL; | |
152 | Close(); | |
153 | } | |
154 | ||
155 | m_addr = addr = new wxIPV4address(); | |
156 | ||
157 | if (!addr->Hostname(host)) { | |
158 | delete m_addr; | |
159 | m_addr = NULL; | |
160 | m_error = wxPROTO_NETERR; | |
161 | return FALSE; | |
162 | } | |
163 | ||
4846abaf | 164 | if (!addr->Service(_T("http"))) |
f4ada568 GL |
165 | addr->Service(80); |
166 | ||
167 | return TRUE; | |
168 | } | |
169 | ||
8a2c6ef8 | 170 | bool wxHTTP::Connect(wxSockAddress& addr, bool WXUNUSED(wait)) |
f4ada568 GL |
171 | { |
172 | struct sockaddr *raw_addr; | |
173 | size_t len; | |
174 | ||
175 | m_addr = (wxSockAddress *)(addr.GetClassInfo()->CreateObject()); | |
176 | ||
177 | addr.Build(raw_addr, len); | |
178 | m_addr->Disassemble(raw_addr, len); | |
179 | ||
180 | return TRUE; | |
181 | } | |
182 | ||
183 | bool wxHTTP::BuildRequest(const wxString& path, wxHTTP_Req req) | |
184 | { | |
185 | char *tmp_buf; | |
186 | char buf[HTTP_BSIZE]; | |
fde7d98e | 187 | const wxWX2MBbuf pathbuf = path.mb_str(); |
f4ada568 GL |
188 | |
189 | switch (req) { | |
190 | case wxHTTP_GET: | |
191 | tmp_buf = "GET"; | |
192 | break; | |
193 | default: | |
194 | return FALSE; | |
195 | } | |
196 | ||
41895a05 GL |
197 | SaveState(); |
198 | Notify(FALSE); | |
199 | SetFlags(WAITALL); | |
200 | ||
4846abaf | 201 | sprintf(buf, "%s %s HTTP/1.0\n\r", tmp_buf, (const char*)pathbuf); |
f4ada568 GL |
202 | Write(buf, strlen(buf)); |
203 | SendHeaders(); | |
204 | sprintf(buf, "\n\r"); | |
205 | Write(buf, strlen(buf)); | |
206 | ||
207 | wxString tmp_str; | |
208 | ||
209 | m_error = GetLine(this, tmp_str); | |
41895a05 GL |
210 | if (m_error != wxPROTO_NOERR) { |
211 | RestoreState(); | |
f4ada568 | 212 | return FALSE; |
41895a05 | 213 | } |
f4ada568 | 214 | |
4846abaf | 215 | if (!tmp_str.Contains(_T("HTTP/"))) { |
f4ada568 | 216 | // TODO: support HTTP v0.9 which can have no header. |
4846abaf OK |
217 | SetHeader(_T("Content-Length"), _T("-1")); |
218 | SetHeader(_T("Content-Type"), _T("none/none")); | |
41895a05 | 219 | RestoreState(); |
f4ada568 GL |
220 | return TRUE; |
221 | } | |
222 | ||
4846abaf | 223 | wxStringTokenizer token(tmp_str,_T(' ')); |
f4ada568 | 224 | wxString tmp_str2; |
41895a05 | 225 | bool ret_value; |
f4ada568 GL |
226 | |
227 | token.NextToken(); | |
228 | tmp_str2 = token.NextToken(); | |
229 | ||
4846abaf | 230 | switch (wxAtoi(tmp_str2)) { |
f4ada568 GL |
231 | case 200: |
232 | break; | |
233 | default: | |
234 | m_error = wxPROTO_NOFILE; | |
41895a05 | 235 | RestoreState(); |
f4ada568 GL |
236 | return FALSE; |
237 | } | |
238 | ||
41895a05 GL |
239 | ret_value = ParseHeaders(); |
240 | RestoreState(); | |
241 | return ret_value; | |
f4ada568 GL |
242 | } |
243 | ||
244 | class wxHTTPStream : public wxSocketInputStream { | |
245 | public: | |
246 | wxHTTP *m_http; | |
9a1b2c28 GL |
247 | size_t m_httpsize; |
248 | ||
f4ada568 | 249 | wxHTTPStream(wxHTTP *http) : wxSocketInputStream(*http), m_http(http) {} |
41895a05 | 250 | size_t StreamSize() const { return m_httpsize; } |
f4ada568 GL |
251 | virtual ~wxHTTPStream(void) { m_http->Abort(); } |
252 | }; | |
253 | ||
254 | bool wxHTTP::Abort(void) | |
255 | { | |
256 | return wxSocketClient::Close(); | |
257 | } | |
258 | ||
259 | wxInputStream *wxHTTP::GetInputStream(const wxString& path) | |
260 | { | |
261 | wxHTTPStream *inp_stream = new wxHTTPStream(this); | |
262 | ||
263 | if (!m_addr || m_connected) { | |
264 | m_error = wxPROTO_CONNERR; | |
265 | return NULL; | |
266 | } | |
267 | ||
268 | if (!wxProtocol::Connect(*m_addr)) | |
269 | return NULL; | |
270 | ||
271 | if (!BuildRequest(path, wxHTTP_GET)) | |
272 | return NULL; | |
273 | ||
4846abaf OK |
274 | wxPrintf(_T("Len = %s\n"), WXSTRINGCAST GetHeader(_T("Content-Length"))); |
275 | if (!GetHeader(_T("Content-Length")).IsEmpty()) | |
276 | inp_stream->m_httpsize = wxAtoi(WXSTRINGCAST GetHeader(_T("Content-Length"))); | |
9a1b2c28 | 277 | |
f4ada568 GL |
278 | return inp_stream; |
279 | } | |
35a4dab7 GL |
280 | |
281 | #endif | |
282 | // wxUSE_SOCKETS |