ViewVC Help
View File | Revision Log | Show Annotations | Download File | View Changeset | Root Listing
root/proj/trunk/Search/HttpHandler.cpp
(Generate patch)

Comparing trunk/Search/HttpHandler.cpp (file contents):
Revision 17 by douglas, 2002-12-09T18:31:11-08:00 vs.
Revision 24 by douglas, 2002-12-22T00:58:21-08:00

# Line 62 | Line 62 | HttpHandler::HttpHandler()
62          }
63   #endif // _WIN32
64  
65 <        begin = 0;
65 >        length = 0;
66 >        chunked = false;
67   }
68  
69   HttpHandler::~HttpHandler()
# Line 131 | Line 132 | bool HttpHandler::handle(URL &url, bool
132                  delete [] port;
133          }
134  
135 + //      putline("Referer: " + ¿referer?);
136 +        putline("Connection: close");
137          putline();
138  
139 <        string line = getline();
139 >        code response;
140 >        string line;
141  
142 <        if (line.find("HTTP/") != 0)
142 >        do
143          {
144 <                return answer;
141 <        }
144 >                line = getline();
145  
146 <        unsigned dot = line.find('.');
147 <        unsigned space = line.find(' ');
146 >                if (line.find("HTTP/") != 0)
147 >                {
148 >                        return answer;
149 >                }
150  
151 <        unsigned major = strtoul(line.substr(5, dot - 5).c_str(), 0, 0);
152 <        unsigned minor = strtoul(line.substr(dot + 1, space - dot - 1).c_str(), 0,
148 <                0);
151 >                unsigned dot = line.find('.');
152 >                unsigned space = line.find(' ');
153  
154 <        if (major > 1 || minor < 1)
155 <        {
156 <                cerr << program << ": Potentially Incompatible Server: HTTP/" << major
153 <                        << "." << minor << "\n";
154 >                unsigned major = strtoul(line.substr(5, dot - 5).c_str(), 0, 10);
155 >                unsigned minor = strtoul(line.substr(dot + 1, space - dot - 1).c_str(),
156 >                        0, 10);
157  
158 <                return answer;
159 <        }
158 >                if (major > 1)
159 >                {
160 >                        cerr << program << ": Potentially Incompatible Server: HTTP/" <<
161 >                                major << "." << minor << "\n";
162 >
163 >                        return answer;
164 >                }
165  
166 <        code response = code(strtoul(line.substr(space + 1).c_str(), 0, 0));
166 >                response = code(strtoul(line.substr(space + 1).c_str(), 0, 10));
167 >
168 >                if (response < ok) do line = getline(); while (line != "");
169 >        }
170 >        while (response < ok);
171  
172          do
173          {
174                  line = getline();
175 +
176 +                if (line != "")
177 +                {
178 +                        unsigned colon = line.find(':');
179 +
180 +                        string field = line.substr(0, colon);
181 +                        string value = line.substr(colon + 1);
182 +
183 +                        while (isspace(value[0])) value.erase(0, 1);
184 +
185 +                        if (field == "Content-Type")
186 +                        {
187 +                                type = value;
188 +                        }
189 +                        else if (field == "Content-Length")
190 +                        {
191 +                                length = strtoul(value.c_str(), 0, 10);
192 +                        }
193 +                        else if (field == "Location")
194 +                        {
195 +                                location = value;
196 +                        }
197 +                        else if (field == "Transfer-Encoding")
198 +                        {
199 +                                chunked = value == "chunked";
200 +                        }
201 +                }
202          }
203          while (line != "");
204  
205          switch (response)
206          {
207          case ok:
208 +                if (debug) cerr << "response = " << response << "\n";
209                  answer = true;
210                  break;
211 <        case choices:
172 <                break;
211 >        case choices:
212          case moved:
174 –                break;
213          case found:
214 +                if (debug) cerr << "response = " << response << "\n"
215 +                        << "location = " << location << "\n";
216 +                location = getLink(location, url);
217                  break;
218          case notfound:
178 –                break;
219          case internal:
220 +                if (debug) cerr << "response = " << response << "\n";
221                  break;
222          default:
223 +                if (debug) cerr << "response = " << response << "\n";
224 +                if (response <= 299)
225 +                {
226 +                        answer = true;
227 +                }
228 +                else if (response <= 399)
229 +                {
230 +                        location = getLink(location, url);
231 +                }
232                  break;
233          }
234  
235 +        if (!head && answer) populate();
236 +
237          return answer;
238   }
239  
240   HttpHandler& HttpHandler::getline(string& line, char endline)
241   {
242 <        int end = page.find(endline, begin);
243 <        int newline = page.find('\n', begin);
242 >        unsigned end = page.find(endline);
243 >        unsigned newline = page.find('\n');
244  
245          if (newline < end || end == string::npos)
246          {
247                  end = newline;
248          }
249  
250 <        line = page.substr(begin, end - begin);
251 <
200 <        if (end == string::npos)
201 <        {
202 <                begin = end;
203 <        }
204 <        else
205 <        {
206 <                begin = end + 1;
207 <        }
250 >        line = page.substr(0, end);
251 >        page.erase(0, (end == string::npos ? end : end + 1));
252  
253          return *this;
254   }
255  
256 < bool HttpHandler::good()
256 > void HttpHandler::clear()
257   {
258 <        bool answer = true;
258 >        closesocket(http);
259  
260 <        if (begin >= page.length())
260 >        type = "";
261 >        length = 0;
262 >        location = "";
263 >        page = "";
264 >        chunked = false;
265 > }
266 >
267 > void HttpHandler::populate()
268 > {
269 >        if (!chunked)
270          {
271 <                answer = false;
271 >                unsigned left = length;
272 >
273 >                while (left > 0)
274 >                {
275 >                        memset(buffer, 0, BUFSIZ + 1);
276 >
277 >                        unsigned bytes = left > BUFSIZ ? BUFSIZ : left;
278 >                        unsigned received;
279 >
280 >                        if ((received = recv(http, buffer, bytes, 0)) == SOCKET_ERROR)
281 >                        {
282 >                                error(program + ": Recv");
283 >                                exit(1);
284 >                        }
285 >                        else if (received != bytes)
286 >                        {
287 >                                left -= received;
288 >                                page += buffer;
289 >
290 >                                memset(buffer, 0, BUFSIZ + 1);
291 >
292 >                                bytes -= received;
293 >                                if (recv(http, buffer, bytes, 0) == SOCKET_ERROR)
294 >                                {
295 >                                        error(program + ": Recv");
296 >                                        exit(1);
297 >                                }
298 >                        }
299 >
300 >                        page += buffer;
301 >                        left -= bytes;
302 >                }
303          }
304 <        else if (begin == string::npos)
304 >        else
305          {
306 <                answer = false;
223 <        }
306 >                unsigned chunk;
307  
308 <        return answer;
309 < }
308 >                do
309 >                {
310 >                        chunk = strtoul(getline().c_str(), 0, 16);
311  
312 < void HttpHandler::clear()
313 < {
314 <        type = "";
315 <        length = 0;
316 <        location = "";
317 <        begin = 0;
318 <        page = "";
312 >                        unsigned left = chunk;
313 >
314 >                        while (left > 0)
315 >                        {
316 >                                memset(buffer, 0, BUFSIZ + 1);
317 >
318 >                                unsigned bytes = left > BUFSIZ ? BUFSIZ : left;
319 >                                unsigned received;
320 >
321 >                                if ((received = recv(http, buffer, bytes, 0)) == SOCKET_ERROR)
322 >                                {
323 >                                        error(program + ": Recv");
324 >                                        exit(1);
325 >                                }
326 >                                else if (received != bytes)
327 >                                {
328 >                                        left -= received;
329 >                                        page += buffer;
330 >
331 >                                        memset(buffer, 0, BUFSIZ + 1);
332 >
333 >                                        bytes -= received;
334 >                                        if (recv(http, buffer, bytes, 0) == SOCKET_ERROR)
335 >                                        {
336 >                                                error(program + ": Recv");
337 >                                                exit(1);
338 >                                        }
339 >                                }
340 >
341 >                                page += buffer;
342 >                                left -= bytes;
343 >                        }
344 >
345 >                        getline();
346 >                        length += chunk;
347 >                }
348 >                while (chunk > 0);
349 >        }
350 >
351 >        for (unsigned index = 0; index < page.length(); index++)
352 >        {
353 >                if (page[index] == '\r' && (index + 1 < page.length()) ? page[index +
354 >                        1] == '\n' : false)
355 >                {
356 >                        page.erase(index, 1);
357 >                }
358 >                else if (page[index] == '\r')
359 >                {
360 >                        page[index] = '\n';
361 >                }
362 >        }
363   }
364  
365   void HttpHandler::putline(const string line)
# Line 266 | Line 394 | string HttpHandler::getline()
394          return line;
395   }
396  
397 < void HttpHandler::error(const string prefix, bool host)
397 > void HttpHandler::error(const string& prefix, bool host)
398   {
399   #ifdef _WIN32
400          string error;

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines