netwerk/test/TestRes.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   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 "nsIResProtocolHandler.h"
   7 #include "nsIServiceManager.h"
   8 #include "nsIIOService.h"
   9 #include "nsIInputStream.h"
  10 #include "nsIComponentManager.h"
  11 #include "nsIComponentRegistrar.h"
  12 #include "nsIStreamListener.h"
  13 #include "nsIEventQueueService.h"
  14 #include "nsIURI.h"
  15 #include "nsCRT.h"
  16 #include "nsNetCID.h"
  17
  18 static NS_DEFINE_CID(kIOServiceCID, NS_IOSERVICE_CID);
  19 static NS_DEFINE_CID(kEventQueueServiceCID, NS_EVENTQUEUESERVICE_CID);
  20
  21 ////////////////////////////////////////////////////////////////////////////////
  22
  23 nsresult
  24 SetupMapping()
  25 {
  26     nsresult rv;
  27
  28     nsCOMPtr<nsIIOService> serv(do_GetService(kIOServiceCID, &rv));
  29     if (NS_FAILED(rv)) return rv;
  30
  31     nsCOMPtr<nsIProtocolHandler> ph;
  32     rv = serv->GetProtocolHandler("res", getter_AddRefs(ph));
  33     if (NS_FAILED(rv)) return rv;
  34
  35     nsCOMPtr<nsIResProtocolHandler> resPH = do_QueryInterface(ph, &rv);
  36     if (NS_FAILED(rv)) return rv;
  37
  38     rv = resPH->AppendSubstitution("foo", "file://y|/");
  39     if (NS_FAILED(rv)) return rv;
  40
  41     rv = resPH->AppendSubstitution("foo", "file://y|/mozilla/dist/win32_D.OBJ/bin/");
  42     if (NS_FAILED(rv)) return rv;
  43
  44     return rv;
  45 }
  46
  47 ////////////////////////////////////////////////////////////////////////////////
  48
  49 nsresult
  50 TestOpenInputStream(const char* url)
  51 {
  52     nsresult rv;
  53
  54     nsCOMPtr<nsIIOService> serv(do_GetService(kIOServiceCID, &rv));
  55     if (NS_FAILED(rv)) return rv;
  56
  57     nsCOMPtr<nsIChannel> channel;
  58     rv = serv->NewChannel(url,
  59                           nullptr, // base uri
  60                           getter_AddRefs(channel));
  61     if (NS_FAILED(rv)) return rv;
  62
  63     nsCOMPtr<nsIInputStream> in;
  64     rv = channel->Open(getter_AddRefs(in));
  65     if (NS_FAILED(rv)) {
  66         fprintf(stdout, "failed to OpenInputStream for %s\n", url);
  67         return NS_OK;
  68     }
  69
  70     char buf[1024];
  71     while (1) {
  72         uint32_t amt;
  73         rv = in->Read(buf, sizeof(buf), &amt);
  74         if (NS_FAILED(rv)) return rv;
  75         if (amt == 0) break;    // eof
  76
  77         char* str = buf;
  78         while (amt-- > 0) {
  79             fputc(*str++, stdout);
  80         }
  81     }
  82     nsCOMPtr<nsIURI> uri;
  83     char* str;
  84
  85     rv = channel->GetOriginalURI(getter_AddRefs(uri));
  86     if (NS_FAILED(rv)) return rv;
  87     rv = uri->GetSpec(&str);
  88     if (NS_FAILED(rv)) return rv;
  89     fprintf(stdout, "%s resolved to ", str);
  90     free(str);
  91
  92     rv = channel->GetURI(getter_AddRefs(uri));
  93     if (NS_FAILED(rv)) return rv;
  94     rv = uri->GetSpec(&str);
  95     if (NS_FAILED(rv)) return rv;
  96     fprintf(stdout, "%s\n", str);
  97     free(str);
  98
  99     return NS_OK;
 100 }
 101
 102 ////////////////////////////////////////////////////////////////////////////////
 103
 104 bool gDone = false;
 105 nsIEventQueue* gEventQ = nullptr;
 106
 107 class Listener : public nsIStreamListener
 108 {
 109 public:
 110     NS_DECL_ISUPPORTS
 111
 112     Listener() {}
 113     virtual ~Listener() {}
 114
 115     NS_IMETHOD OnStartRequest(nsIRequest *request, nsISupports *ctxt) {
 116         nsresult rv;
 117         nsCOMPtr<nsIURI> uri;
 118         nsCOMPtr<nsIChannel> channel = do_QueryInterface(request);
 119
 120         rv = channel->GetURI(getter_AddRefs(uri));
 121         if (NS_SUCCEEDED(rv)) {
 122             char* str;
 123             rv = uri->GetSpec(&str);
 124             if (NS_SUCCEEDED(rv)) {
 125                 fprintf(stdout, "Starting to load %s\n", str);
 126                 free(str);
 127             }
 128         }
 129         return NS_OK;
 130     }
 131
 132     NS_IMETHOD OnStopRequest(nsIRequest *request, nsISupports *ctxt,
 133                              nsresult aStatus) {
 134         nsresult rv;
 135         nsCOMPtr<nsIURI> uri;
 136         nsCOMPtr<nsIChannel> channel = do_QueryInterface(request);
 137
 138         rv = channel->GetURI(getter_AddRefs(uri));
 139         if (NS_SUCCEEDED(rv)) {
 140             char* str;
 141             rv = uri->GetSpec(&str);
 142             if (NS_SUCCEEDED(rv)) {
 143                 fprintf(stdout, "Ending load %s, status=%x\n", str, aStatus);
 144                 free(str);
 145             }
 146         }
 147         gDone = true;
 148         return NS_OK;
 149     }
 150
 151     NS_IMETHOD OnDataAvailable(nsIRequest *request, nsISupports *ctxt,
 152                                nsIInputStream *inStr,
 153                                uint64_t sourceOffset, uint32_t count) {
 154         nsresult rv;
 155         char buf[1024];
 156         while (count > 0) {
 157             uint32_t amt;
 158             rv = inStr->Read(buf, sizeof(buf), &amt);
 159             count -= amt;
 160             char* c = buf;
 161             while (amt-- > 0) {
 162                 fputc(*c++, stdout);
 163             }
 164         }
 165         return NS_OK;
 166     }
 167 };
 168
 169 NS_IMPL_ISUPPORTS(Listener, nsIStreamListener, nsIRequestObserver)
 170
 171 nsresult
 172 TestAsyncRead(const char* url)
 173 {
 174     nsresult rv;
 175
 176     nsCOMPtr<nsIEventQueueService> eventQService =
 177              do_GetService(kEventQueueServiceCID, &rv);
 178     if (NS_FAILED(rv)) return rv;
 179
 180     rv = eventQService->GetThreadEventQueue(NS_CURRENT_THREAD, &gEventQ);
 181     if (NS_FAILED(rv)) return rv;
 182
 183     nsCOMPtr<nsIIOService> serv(do_GetService(kIOServiceCID, &rv));
 184     if (NS_FAILED(rv)) return rv;
 185
 186     nsCOMPtr<nsIChannel> channel;
 187     rv = serv->NewChannel(url,
 188                           nullptr, // base uri
 189                           getter_AddRefs(channel));
 190     if (NS_FAILED(rv)) return rv;
 191
 192     nsCOMPtr<nsIStreamListener> listener = new Listener();
 193     if (listener == nullptr)
 194         return NS_ERROR_OUT_OF_MEMORY;
 195     rv = channel->AsyncOpen(nullptr, listener);
 196     if (NS_FAILED(rv)) return rv;
 197
 198     while (!gDone) {
 199         PLEvent* event;
 200         rv = gEventQ->GetEvent(&event);
 201         if (NS_FAILED(rv)) return rv;
 202         rv = gEventQ->HandleEvent(event);
 203         if (NS_FAILED(rv)) return rv;
 204     }
 205
 206     return rv;
 207 }
 208
 209 int
 210 main(int argc, char* argv[])
 211 {
 212     nsresult rv;
 213     {
 214         nsCOMPtr<nsIServiceManager> servMan;
 215         NS_InitXPCOM2(getter_AddRefs(servMan), nullptr, nullptr);
 216         nsCOMPtr<nsIComponentRegistrar> registrar = do_QueryInterface(servMan);
 217         NS_ASSERTION(registrar, "Null nsIComponentRegistrar");
 218         if (registrar)
 219             registrar->AutoRegister(nullptr);
 220
 221         NS_ASSERTION(NS_SUCCEEDED(rv), "AutoregisterComponents failed");
 222
 223         if (argc < 2) {
 224             printf("usage: %s resource://foo/<path-to-resolve>\n", argv[0]);
 225             return -1;
 226         }
 227
 228         rv = SetupMapping();
 229         NS_ASSERTION(NS_SUCCEEDED(rv), "SetupMapping failed");
 230         if (NS_FAILED(rv)) return rv;
 231
 232         rv = TestOpenInputStream(argv[1]);
 233         NS_ASSERTION(NS_SUCCEEDED(rv), "TestOpenInputStream failed");
 234
 235         rv = TestAsyncRead(argv[1]);
 236         NS_ASSERTION(NS_SUCCEEDED(rv), "TestAsyncRead failed");
 237     } // this scopes the nsCOMPtrs
 238     // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
 239     rv = NS_ShutdownXPCOM(nullptr);
 240     NS_ASSERTION(NS_SUCCEEDED(rv), "NS_ShutdownXPCOM failed");
 241     return rv;
 242 }
 243
 244 ////////////////////////////////////////////////////////////////////////////////