/*
* Just like simpleGet but with an explicit port number
*/
static bool get(MprTestGroup *gp, cchar *host, int port, cchar *uri, int expectCode)
{
MprHttp *http;
int code;
http = getHttp(gp);
if (expectCode <= 0) {
expectCode = 200;
}
if (host) {
mprSetHttpDefaultHost(http, host);
}
if (port > 0) {
mprSetHttpDefaultPort(http, getDefaultPort(gp) + port);
}
if (mprHttpRequest(http, "GET", uri, 0) < 0) {
return 0;
}
code = mprGetHttpCode(http);
assert(code == expectCode);
if (code != expectCode) {
mprLog(gp, 0, "get: HTTP response code %d, expected %d", code, expectCode);
return 0;
}
assert(mprGetHttpError(http) != 0);
assert(mprGetHttpContent(http) != 0);
return 1;
}
static void location(MprTestGroup *gp)
{
MprHttp *http;
char *uri;
int rc, code;
http = getHttp(gp);
uri = "/egi/egiProgram?SWITCHES=-l%20http://www.redhat.com/";
rc = httpRequest(http, "GET", uri);
assert(rc == 0);
if (rc != 0) {
return;
}
code = mprGetHttpCode(http);
if (code != 302) {
mprLog(gp, 0, "Client failed for %s, response code: %d, msg %s\n", uri, code, mprGetHttpMessage(http));
}
}
static bool authGet(MprTestGroup *gp, char *uri, char *realm, char *user, char *password, int expectCode)
{
MprHttp *http;
char uriBuf[MPR_MAX_STRING];
cchar *content;
int rc, code, contentLength;
http = getHttp(gp);
if (authType) {
mprSprintf(uriBuf, sizeof(uriBuf), "/%s%s", authType, uri);
mprSetHttpCredentials(http, user, password);
} else {
mprStrcpy(uriBuf, sizeof(uriBuf), uri);
mprResetHttpCredentials(http);
}
rc = httpRequest(http, "GET", uriBuf);
if (!assert(rc == 0)) {
return 0;
}
code = mprGetHttpCode(http);
assert(code == expectCode);
if (code != expectCode) {
mprLog(gp, 0, "Client failed for %s, response code: %d, expected %d, msg %s\n", uriBuf, code, expectCode,
mprGetHttpMessage(http));
return 0;
}
if (expectCode != 200) {
contentLength = mprGetHttpContentLength(http);
content = mprGetHttpContent(http);
if (! assert(content != 0 && contentLength > 0)) {
return 0;
}
}
return 1;
}
static void status(MprTestGroup *gp)
{
MprHttp *http;
char *uri;
int rc, code;
http = getHttp(gp);
/*
* Have egiProgram exit with a 711 status
*/
uri = "/egi/egiProgram?SWITCHES=-s%20711";
rc = httpRequest(http, "GET", uri);
assert(rc == 0);
if (rc != 0) {
return;
}
code = mprGetHttpCode(http);
if (code != 711) {
mprLog(gp, 0, "Client failed for %s, response code: %d, msg %s\n", uri, code, mprGetHttpMessage(http));
}
}
static int doRequest(MprHttp *http, cchar *url, MprList *fields, MprList *files)
{
MprHttpResponse *resp;
MprKeyValue *header;
char buf[MPR_HTTP_BUFSIZE], seqBuf[16], *responseHeaders, *redirect;
int64 contentLen;
int code, next, count, bytes, transCount;
mprAssert(url && *url);
mprLog(http, MPR_DEBUG, "fetch: %s %s", method, url);
/*
* Send the request
*/
count = -1;
transCount = 0;
do {
for (next = 0; (header = mprGetNextItem(headers, &next)) != 0; ) {
mprSetHttpHeader(http, 0, header->key, header->value);
}
if (sequence) {
static int next = 0;
mprItoa(seqBuf, sizeof(seqBuf), next++, 10);
mprSetHttpHeader(http, 1, "X-Http-Seq", seqBuf);
}
if (ranges) {
mprSetHttpHeader(http, 1, "Range", ranges);
}
if (fields) {
mprSetHttpHeader(http, 1, "Content-Type", "application/x-www-form-urlencoded");
}
if (chunkSize) {
mprSetHttpChunked(http, 1);
}
if (setContentLength(http, fields, files) < 0) {
return MPR_ERR_CANT_OPEN;
}
if (mprStartHttpRequest(http, method, url) < 0) {
mprError(http, "Can't process request for \"%s\". %s", url, mprGetHttpError(http));
return MPR_ERR_CANT_OPEN;
}
/*
* This program does not do full-duplex writes with reads. ie. if you have a request that sends and receives
* data in parallel -- http will do the writes first then read the response.
*/
if (files || fields) {
if (writeBody(http, fields, files) < 0) {
mprError(http, "Can't write body data to \"%s\". %s", url, mprGetHttpError(http));
return MPR_ERR_CANT_WRITE;
}
} else {
if (chunkSize) {
mprFinalizeHttpWriting(http);
}
mprWaitForHttpResponse(http, -1);
}
#if WIN
_setmode(fileno(stdout), O_BINARY);
#endif
while ((bytes = mprReadHttp(http, buf, sizeof(buf))) > 0) {
showOutput(http, buf, bytes);
}
mprWaitForHttp(http, MPR_HTTP_STATE_COMPLETE, -1);
/* May not be complete if a disconnect occurs */
if (http->state >= MPR_HTTP_STATE_CONTENT) {
if (mprNeedHttpRetry(http, &redirect)) {
if (redirect) {
url = resolveUrl(http, redirect);
}
count--;
transCount++;
continue;
}
break;
}
} while (++count < http->retries && transCount < 4 && !mprIsExiting(http));
if (count >= http->retries) {
mprError(http, "http: failed \"%s\" request for %s after %d attempt(s)", method, url, count);
return MPR_ERR_TOO_MANY;
}
if (mprIsExiting(http)) {
return MPR_ERR_BAD_STATE;
}
/*
* Request now complete
*/
code = mprGetHttpCode(http);
contentLen = mprGetHttpContentLength(http);
mprLog(http, 6, "Response code %d, content len %d", code, contentLen);
if (http->response) {
if (showCode) {
mprPrintf(http, "%d\n", code);
}
if (showHeaders) {
responseHeaders = mprGetHttpHeaders(http);
resp = http->response;
mprPrintf(http, "%s %d %s\n", resp->protocol, resp->code, resp->message);
mprPrintf(http, "%s\n", responseHeaders);
mprFree(responseHeaders);
}
}
if (code < 0) {
mprError(http, "Can't process request for \"%s\" %s", url, mprGetHttpError(http));
return MPR_ERR_CANT_READ;
} else if (code == 0 && http->protocolVersion == 0) {
;
} else if (!(200 <= code && code <= 206) && !(301 <= code && code <= 304)) {
if (!showCode) {
mprError(http, "Can't process request for \"%s\" (%d) %s", url, code, mprGetHttpError(http));
return MPR_ERR_CANT_READ;
}
}
lock();
if (verbose && noout) {
trace(http, url, fetchCount, method, code, (int) contentLen);
}
unlock();
return 0;
}
