1 |
|
/* ============================================================================ |
2 |
|
* Douglas Thrift's Search Engine License |
3 |
|
* |
4 |
< |
* Copyright (C) 2002, Douglas Thrift. All Rights Reserved. |
4 |
> |
* Copyright (C) 2002-2003, Douglas Thrift. All Rights Reserved. |
5 |
|
* Redistribution and use in source and binary forms, with or without |
6 |
|
* modification, are permitted provided that the following conditions are met: |
7 |
|
* |
75 |
|
#endif // _WIN32 |
76 |
|
} |
77 |
|
|
78 |
< |
bool HttpHandler::handle(URL &url, bool head) |
78 |
> |
bool HttpHandler::handle(URL &url, const string referer, bool head) |
79 |
|
{ |
80 |
|
bool answer = false; |
81 |
|
|
132 |
|
delete [] port; |
133 |
|
} |
134 |
|
|
135 |
< |
// putline("Referer: " + ¿referer?); |
135 |
> |
if (referer != "") |
136 |
> |
{ |
137 |
> |
putline("Referer: " + referer); |
138 |
> |
} |
139 |
> |
|
140 |
|
putline("Connection: close"); |
141 |
|
putline(); |
142 |
|
|
281 |
|
unsigned bytes = left > BUFSIZ ? BUFSIZ : left; |
282 |
|
unsigned received; |
283 |
|
|
284 |
< |
if ((received = recv(http, buffer, bytes, 0)) == SOCKET_ERROR) |
281 |
< |
{ |
282 |
< |
error(program + ": Recv"); |
283 |
< |
exit(1); |
284 |
< |
} |
285 |
< |
else if (received != bytes) |
284 |
> |
while (true) |
285 |
|
{ |
286 |
< |
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) |
286 |
> |
if ((received = recv(http, buffer, bytes, 0)) == SOCKET_ERROR) |
287 |
|
{ |
288 |
|
error(program + ": Recv"); |
289 |
|
exit(1); |
290 |
|
} |
291 |
+ |
else if (received != bytes) |
292 |
+ |
{ |
293 |
+ |
left -= received; |
294 |
+ |
page += buffer; |
295 |
+ |
|
296 |
+ |
memset(buffer, 0, BUFSIZ + 1); |
297 |
+ |
|
298 |
+ |
bytes -= received; |
299 |
+ |
} |
300 |
+ |
else |
301 |
+ |
{ |
302 |
+ |
break; |
303 |
+ |
} |
304 |
|
} |
305 |
|
|
306 |
|
page += buffer; |
324 |
|
unsigned bytes = left > BUFSIZ ? BUFSIZ : left; |
325 |
|
unsigned received; |
326 |
|
|
327 |
< |
if ((received = recv(http, buffer, bytes, 0)) == SOCKET_ERROR) |
322 |
< |
{ |
323 |
< |
error(program + ": Recv"); |
324 |
< |
exit(1); |
325 |
< |
} |
326 |
< |
else if (received != bytes) |
327 |
> |
while (true) |
328 |
|
{ |
329 |
< |
left -= received; |
330 |
< |
page += buffer; |
330 |
< |
|
331 |
< |
memset(buffer, 0, BUFSIZ + 1); |
332 |
< |
|
333 |
< |
bytes -= received; |
334 |
< |
if (recv(http, buffer, bytes, 0) == SOCKET_ERROR) |
329 |
> |
if ((received = recv(http, buffer, bytes, 0)) == |
330 |
> |
SOCKET_ERROR) |
331 |
|
{ |
332 |
|
error(program + ": Recv"); |
333 |
|
exit(1); |
334 |
|
} |
335 |
+ |
else if (received != bytes) |
336 |
+ |
{ |
337 |
+ |
left -= received; |
338 |
+ |
page += buffer; |
339 |
+ |
|
340 |
+ |
memset(buffer, 0, BUFSIZ + 1); |
341 |
+ |
|
342 |
+ |
bytes -= received; |
343 |
+ |
} |
344 |
+ |
else |
345 |
+ |
{ |
346 |
+ |
break; |
347 |
+ |
} |
348 |
|
} |
349 |
|
|
350 |
|
page += buffer; |