PRIVATE int HTSC_putBlock (HTStream * me, const char * b, int l)
{
me->cur_size += l;
/*
** If we get a buffer overflow and we are going to PUT or POST the document
** then ask the user whether it is OK to proceed buffering. Otherwise we
** must give up the request. In all other cases we stop if the buffer fills
** up.
*/
if (!me->ignore && me->max_size > 0 && me->cur_size > me->max_size) {
HTMethod method = HTRequest_method(me->request);
if (HTMethod_hasEntity(method)) {
HTAlertCallback *cbf = HTAlert_find(HT_A_CONFIRM);
if ((cbf && (*cbf)(me->request, HT_A_CONFIRM, HT_MSG_BIG_PUT,
NULL, NULL, NULL)))
me->ignore = YES;
else
me->give_up = YES;
} else {
me->give_up = YES;
}
} else if (!me->ensure) {
HTParentAnchor * anchor = HTRequest_anchor(me->request);
int cl = HTAnchor_length(anchor);
if (cl > 0) HTChunk_ensure(me->chunk, cl);
me->ensure = YES;
}
if (!me->give_up) {
HTChunk_putb(me->chunk, b, l);
return HT_OK;
}
return HT_ERROR;
}
PUBLIC HTStream * HTTPRequest_new (HTRequest * request, HTStream * target,
BOOL endHeader, int version)
{
HTStream * me;
if ((me = (HTStream *) HT_CALLOC(1, sizeof(HTStream))) == NULL)
HT_OUTOFMEM("HTTPRequest_new");
me->isa = &HTTPRequestClass;
me->target = target;
me->request = request;
me->version = version;
me->transparent = NO;
/*
** If sending a body in the request then we want a 100 code!
*/
if (HTMethod_hasEntity(HTRequest_method(request)))
HTRequest_addExpect(request, "100-continue", "");
/* Return general HTTP header stream */
return HTTPGen_new(request, me, endHeader, version);
}
PUBLIC char * HTDialog_progressMessage (HTRequest * request, HTAlertOpcode op,
int msgnum, const char * dfault,
void * input)
{
char * result = NULL;
switch (op) {
case HT_PROG_DNS:
StrAllocMCopy(&result, "Looking up ",
input ? (char *) input : "",
NULL);
break;
case HT_PROG_CONNECT:
StrAllocMCopy(&result, "Contacting ",
input ? (char *) input : "",
NULL);
break;
case HT_PROG_ACCEPT:
StrAllocCopy(result, "Waiting for connection...");
break;
case HT_PROG_LOGIN:
StrAllocCopy(result, "Logging in...");
break;
case HT_PROG_READ:
if (request) {
long cl = HTAnchor_length(HTRequest_anchor(request));
if (cl > 0) {
long b_read = HTRequest_bodyRead(request);
double pro = (double) b_read/cl*100;
char buf[10];
char pct[10];
HTNumToStr((unsigned long) cl, buf, 10);
sprintf(pct, "%d%%", (int) pro);
StrAllocMCopy(&result, "Read (", pct, " of ", buf, ")", NULL);
} else {
long b_read = HTRequest_bytesRead(request);
int * raw_read = input ? (int *) input : NULL;
if (b_read > 0) {
char buf[10];
HTNumToStr(b_read, buf, 10);
StrAllocMCopy(&result, "Read ", buf, "bytes", NULL);
} else if (raw_read && *raw_read>0) {
char buf[10];
HTNumToStr(*raw_read, buf, 10);
StrAllocMCopy(&result, "Read ", buf, "bytes", NULL);
} else {
StrAllocCopy(result, "Reading...");
}
}
}
break;
case HT_PROG_WRITE:
if (request && HTMethod_hasEntity(HTRequest_method(request))) {
HTParentAnchor *anchor=HTRequest_anchor(HTRequest_source(request));
long cl = HTAnchor_length(anchor);
if (cl > 0) {
long b_write = HTRequest_bodyWritten(request);
double pro = (double) b_write/cl*100;
char buf[10];
char pct[10];
HTNumToStr((unsigned long) cl, buf, 10);
sprintf(pct, "%d%%", (int) pro);
StrAllocMCopy(&result, "Writing (", pct, " of ", buf, ")", NULL);
} else {
long b_written = HTRequest_bytesWritten(request);
int * raw_written = input ? (int *) input : NULL;
if (b_written > 0) {
char buf[10];
HTNumToStr(b_written>0 ? b_written : 0, buf, 10);
StrAllocMCopy(&result, "Writing ", buf, "bytes", NULL);
} if (raw_written && *raw_written>0) {
char buf[10];
HTNumToStr(*raw_written, buf, 10);
StrAllocMCopy(&result, "Writing ", buf, "bytes", NULL);
} else {
StrAllocCopy(result, "Writing...");
}
}
}
break;
case HT_PROG_DONE:
StrAllocCopy(result, "Done!");
break;
case HT_PROG_INTERRUPT:
StrAllocCopy(result, "Interrupted!");
break;
case HT_PROG_OTHER:
StrAllocCopy(result, "Working - please wait...");
break;
case HT_PROG_TIMEOUT:
StrAllocCopy(result, "Request timeout - server did not respond.");
break;
default:
StrAllocCopy(result, "UNKNOWN PROGRESS STATE");
break;
}
return result;
}
