gecko-dev/netwerk/test/TestSyncHTTP.cpp

68 строки
1.7 KiB
C++
Исходник Обычный вид История

#include <nsCOMPtr.h>
#include <nsString.h>
#include <nsIURI.h>
#include <nsIChannel.h>
#include <nsIHTTPChannel.h>
#include <nsIInputStream.h>
#include <nsNetUtil.h>
/*
* This is a very simple program that tests a "blocking" HTTP get. The call to
* OpenInputStream does not block; instead it returns a blocking nsIInputStream,
* from which we read the HTTP response content.
*/
#define RETURN_IF_FAILED(rv, what) \
PR_BEGIN_MACRO \
if (NS_FAILED(rv)) { \
printf(what ": failed - %08x\n", rv); \
return -1; \
} \
PR_END_MACRO
int
main(int argc, char **argv)
{
nsresult rv;
nsCOMPtr<nsIURI> uri;
nsCOMPtr<nsIChannel> channel;
nsCOMPtr<nsIInputStream> inputStream;
PRTime t1, t2;
if (argc < 2) {
printf("Usage: TestSyncHTTP <url>\n");
return -1;
}
rv = NS_NewURI(getter_AddRefs(uri), argv[1]);
RETURN_IF_FAILED(rv, "NS_NewURI");
rv = NS_OpenURI(getter_AddRefs(channel), uri, nsnull, nsnull);
RETURN_IF_FAILED(rv, "NS_OpenURI");
nsCOMPtr<nsIHTTPChannel> httpChannel = do_QueryInterface(channel);
if (httpChannel)
httpChannel->SetOpenInputStreamHasEventQueue(PR_FALSE);
t1 = PR_Now();
rv = channel->OpenInputStream(getter_AddRefs(inputStream));
RETURN_IF_FAILED(rv, "nsIChannel::OpenInputStream");
//
// read the response content...
//
char buf[256];
PRUint32 bytesRead = 1, totalRead = 0;
while (bytesRead > 0) {
rv = inputStream->Read(buf, sizeof buf, &bytesRead);
RETURN_IF_FAILED(rv, "nsIInputStream::Read");
totalRead += bytesRead;
}
t2 = PR_Now();
printf("total read: %u bytes\n", totalRead);
printf("total read time: %0.3f\n", ((double) (t2 - t1))/1000000.0);
return 0;
}