The Tor Browser / file comparison
comparison: netwerk/test/TestSyncHTTP.cpp
netwerk/test/TestSyncHTTP.cpp
- branch
- TOR_BUG_3246
- changeset 7
- 129ffea94266
equal
deleted
inserted
replaced
| -1:000000000000 | 0:5e7b6039b72f |
|---|---|
| 1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ | |
| 2 /* This Source Code Form is subject to the terms of the Mozilla Public | |
| 3 * License, v. 2.0. If a copy of the MPL was not distributed with this | |
| 4 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ | |
| 5 | |
| 6 #include <nsCOMPtr.h> | |
| 7 #include <nsString.h> | |
| 8 #include <nsIURI.h> | |
| 9 #include <nsIChannel.h> | |
| 10 #include <nsIHTTPChannel.h> | |
| 11 #include <nsIInputStream.h> | |
| 12 #include <nsNetUtil.h> | |
| 13 | |
| 14 /* | |
| 15 * Test synchronous HTTP. | |
| 16 */ | |
| 17 | |
| 18 #define RETURN_IF_FAILED(rv, what) \ | |
| 19 PR_BEGIN_MACRO \ | |
| 20 if (NS_FAILED(rv)) { \ | |
| 21 printf(what ": failed - %08x\n", rv); \ | |
| 22 return -1; \ | |
| 23 } \ | |
| 24 PR_END_MACRO | |
| 25 | |
| 26 struct TestContext { | |
| 27 nsCOMPtr<nsIURI> uri; | |
| 28 nsCOMPtr<nsIChannel> channel; | |
| 29 nsCOMPtr<nsIInputStream> inputStream; | |
| 30 PRTime t1, t2; | |
| 31 uint32_t bytesRead, totalRead; | |
| 32 | |
| 33 TestContext() | |
| 34 : t1(0), t2(0), bytesRead(0), totalRead(0) | |
| 35 { printf("TestContext [this=%p]\n", (void*)this); } | |
| 36 ~TestContext() | |
| 37 { printf("~TestContext [this=%p]\n", (void*)this); } | |
| 38 }; | |
| 39 | |
| 40 int | |
| 41 main(int argc, char **argv) | |
| 42 { | |
| 43 nsresult rv; | |
| 44 TestContext *c; | |
| 45 int i, nc=0, npending=0; | |
| 46 char buf[256]; | |
| 47 | |
| 48 if (argc < 2) { | |
| 49 printf("Usage: TestSyncHTTP <url-list>\n"); | |
| 50 return -1; | |
| 51 } | |
| 52 | |
| 53 c = new TestContext[argc-1]; | |
| 54 | |
| 55 for (i=0; i<(argc-1); ++i, ++nc) { | |
| 56 rv = NS_NewURI(getter_AddRefs(c[i].uri), argv[i+1]); | |
| 57 RETURN_IF_FAILED(rv, "NS_NewURI"); | |
| 58 | |
| 59 rv = NS_OpenURI(getter_AddRefs(c[i].channel), c[i].uri, nullptr, nullptr); | |
| 60 RETURN_IF_FAILED(rv, "NS_OpenURI"); | |
| 61 | |
| 62 nsCOMPtr<nsIHTTPChannel> httpChannel = do_QueryInterface(c[i].channel); | |
| 63 if (httpChannel) | |
| 64 httpChannel->SetOpenHasEventQueue(false); | |
| 65 | |
| 66 // initialize these fields for reading | |
| 67 c[i].bytesRead = 1; | |
| 68 c[i].totalRead = 0; | |
| 69 } | |
| 70 | |
| 71 for (i=0; i<nc; ++i) { | |
| 72 c[i].t1 = PR_Now(); | |
| 73 | |
| 74 rv = c[i].channel->Open(getter_AddRefs(c[i].inputStream)); | |
| 75 RETURN_IF_FAILED(rv, "nsIChannel::OpenInputStream"); | |
| 76 } | |
| 77 | |
| 78 npending = nc; | |
| 79 while (npending) { | |
| 80 for (i=0; i<nc; ++i) { | |
| 81 // | |
| 82 // read the response content... | |
| 83 // | |
| 84 if (c[i].bytesRead > 0) { | |
| 85 rv = c[i].inputStream->Read(buf, sizeof buf, &c[i].bytesRead); | |
| 86 RETURN_IF_FAILED(rv, "nsIInputStream::Read"); | |
| 87 c[i].totalRead += c[i].bytesRead; | |
| 88 | |
| 89 if (c[i].bytesRead == 0) { | |
| 90 c[i].t2 = PR_Now(); | |
| 91 printf("finished GET of: %s\n", argv[i+1]); | |
| 92 printf("total read: %u bytes\n", c[i].totalRead); | |
| 93 printf("total read time: %0.3f\n", | |
| 94 ((double) (c[i].t2 - c[i].t1))/1000000.0); | |
| 95 npending--; | |
| 96 } | |
| 97 } | |
| 98 } | |
| 99 } | |
| 100 | |
| 101 delete[] c; | |
| 102 | |
| 103 NS_ShutdownXPCOM(nullptr); | |
| 104 return 0; | |
| 105 } |
