netwerk/test/TestStreamLoader.cpp

Wed, 31 Dec 2014 06:09:35 +0100

author
Michael Schloh von Bennewitz <michael@schloh.com>
date
Wed, 31 Dec 2014 06:09:35 +0100
changeset 0
6474c204b198
permissions
-rw-r--r--

Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.

     1 #include <stdio.h>
     2 #include "TestCommon.h"
     3 #include "nsNetUtil.h"
     4 #include "nsThreadUtils.h"
     5 #include "prlog.h"
     6 #include "mozilla/Attributes.h"
     8 #if defined(PR_LOGGING)
     9 //
    10 // set NSPR_LOG_MODULES=Test:5
    11 //
    12 static PRLogModuleInfo *gTestLog = nullptr;
    13 #endif
    14 #define LOG(args) PR_LOG(gTestLog, PR_LOG_DEBUG, args)
    16 class MyStreamLoaderObserver MOZ_FINAL : public nsIStreamLoaderObserver
    17 {
    18 public:
    19   NS_DECL_ISUPPORTS
    20   NS_DECL_NSISTREAMLOADEROBSERVER
    21 };
    23 NS_IMPL_ISUPPORTS(MyStreamLoaderObserver, nsIStreamLoaderObserver)
    25 NS_IMETHODIMP
    26 MyStreamLoaderObserver::OnStreamComplete(nsIStreamLoader *loader,
    27                                          nsISupports     *ctxt,
    28                                          nsresult         status,
    29                                          uint32_t         resultLen,
    30                                          const uint8_t   *result)
    31 {
    32   LOG(("OnStreamComplete [status=%x resultLen=%u]\n", status, resultLen));
    34   nsCOMPtr<nsIRequest> request;
    35   loader->GetRequest(getter_AddRefs(request));
    36   LOG(("  request=%p\n", request.get()));
    38   QuitPumpingEvents();
    39   return NS_OK;
    40 }
    42 int main(int argc, char **argv)
    43 {
    44   if (test_common_init(&argc, &argv) != 0)
    45     return -1;
    47   if (argc < 2) {
    48     printf("usage: %s <url>\n", argv[0]);
    49     return -1;
    50   }
    52 #if defined(PR_LOGGING)
    53   gTestLog = PR_NewLogModule("Test");
    54 #endif
    56   nsresult rv = NS_InitXPCOM2(nullptr, nullptr, nullptr);
    57   if (NS_FAILED(rv))
    58     return -1;
    60   {
    61     nsCOMPtr<nsIURI> uri;
    62     rv = NS_NewURI(getter_AddRefs(uri), nsDependentCString(argv[1]));
    63     if (NS_FAILED(rv))
    64       return -1;
    66     nsCOMPtr<nsIChannel> chan;
    67     rv = NS_NewChannel(getter_AddRefs(chan), uri);
    68     if (NS_FAILED(rv))
    69       return -1;
    71     nsCOMPtr<nsIStreamLoaderObserver> observer = new MyStreamLoaderObserver();
    72     if (!observer)
    73       return -1;
    75     nsCOMPtr<nsIStreamLoader> loader;
    76     rv = NS_NewStreamLoader(getter_AddRefs(loader), observer);
    77     if (NS_FAILED(rv))
    78       return -1;
    80     rv = chan->AsyncOpen(loader, nullptr);
    81     if (NS_FAILED(rv))
    82       return -1;
    84     PumpEvents();
    85   } // this scopes the nsCOMPtrs
    86   // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
    87   NS_ShutdownXPCOM(nullptr);
    88   return 0;
    89 }

mercurial