1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/netwerk/test/TestRes.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,244 @@
1.4 +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
1.5 +/* This Source Code Form is subject to the terms of the Mozilla Public
1.6 + * License, v. 2.0. If a copy of the MPL was not distributed with this
1.7 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
1.8 +
1.9 +#include "nsIResProtocolHandler.h"
1.10 +#include "nsIServiceManager.h"
1.11 +#include "nsIIOService.h"
1.12 +#include "nsIInputStream.h"
1.13 +#include "nsIComponentManager.h"
1.14 +#include "nsIComponentRegistrar.h"
1.15 +#include "nsIStreamListener.h"
1.16 +#include "nsIEventQueueService.h"
1.17 +#include "nsIURI.h"
1.18 +#include "nsCRT.h"
1.19 +#include "nsNetCID.h"
1.20 +
1.21 +static NS_DEFINE_CID(kIOServiceCID, NS_IOSERVICE_CID);
1.22 +static NS_DEFINE_CID(kEventQueueServiceCID, NS_EVENTQUEUESERVICE_CID);
1.23 +
1.24 +////////////////////////////////////////////////////////////////////////////////
1.25 +
1.26 +nsresult
1.27 +SetupMapping()
1.28 +{
1.29 + nsresult rv;
1.30 +
1.31 + nsCOMPtr<nsIIOService> serv(do_GetService(kIOServiceCID, &rv));
1.32 + if (NS_FAILED(rv)) return rv;
1.33 +
1.34 + nsCOMPtr<nsIProtocolHandler> ph;
1.35 + rv = serv->GetProtocolHandler("res", getter_AddRefs(ph));
1.36 + if (NS_FAILED(rv)) return rv;
1.37 +
1.38 + nsCOMPtr<nsIResProtocolHandler> resPH = do_QueryInterface(ph, &rv);
1.39 + if (NS_FAILED(rv)) return rv;
1.40 +
1.41 + rv = resPH->AppendSubstitution("foo", "file://y|/");
1.42 + if (NS_FAILED(rv)) return rv;
1.43 +
1.44 + rv = resPH->AppendSubstitution("foo", "file://y|/mozilla/dist/win32_D.OBJ/bin/");
1.45 + if (NS_FAILED(rv)) return rv;
1.46 +
1.47 + return rv;
1.48 +}
1.49 +
1.50 +////////////////////////////////////////////////////////////////////////////////
1.51 +
1.52 +nsresult
1.53 +TestOpenInputStream(const char* url)
1.54 +{
1.55 + nsresult rv;
1.56 +
1.57 + nsCOMPtr<nsIIOService> serv(do_GetService(kIOServiceCID, &rv));
1.58 + if (NS_FAILED(rv)) return rv;
1.59 +
1.60 + nsCOMPtr<nsIChannel> channel;
1.61 + rv = serv->NewChannel(url,
1.62 + nullptr, // base uri
1.63 + getter_AddRefs(channel));
1.64 + if (NS_FAILED(rv)) return rv;
1.65 +
1.66 + nsCOMPtr<nsIInputStream> in;
1.67 + rv = channel->Open(getter_AddRefs(in));
1.68 + if (NS_FAILED(rv)) {
1.69 + fprintf(stdout, "failed to OpenInputStream for %s\n", url);
1.70 + return NS_OK;
1.71 + }
1.72 +
1.73 + char buf[1024];
1.74 + while (1) {
1.75 + uint32_t amt;
1.76 + rv = in->Read(buf, sizeof(buf), &amt);
1.77 + if (NS_FAILED(rv)) return rv;
1.78 + if (amt == 0) break; // eof
1.79 +
1.80 + char* str = buf;
1.81 + while (amt-- > 0) {
1.82 + fputc(*str++, stdout);
1.83 + }
1.84 + }
1.85 + nsCOMPtr<nsIURI> uri;
1.86 + char* str;
1.87 +
1.88 + rv = channel->GetOriginalURI(getter_AddRefs(uri));
1.89 + if (NS_FAILED(rv)) return rv;
1.90 + rv = uri->GetSpec(&str);
1.91 + if (NS_FAILED(rv)) return rv;
1.92 + fprintf(stdout, "%s resolved to ", str);
1.93 + free(str);
1.94 +
1.95 + rv = channel->GetURI(getter_AddRefs(uri));
1.96 + if (NS_FAILED(rv)) return rv;
1.97 + rv = uri->GetSpec(&str);
1.98 + if (NS_FAILED(rv)) return rv;
1.99 + fprintf(stdout, "%s\n", str);
1.100 + free(str);
1.101 +
1.102 + return NS_OK;
1.103 +}
1.104 +
1.105 +////////////////////////////////////////////////////////////////////////////////
1.106 +
1.107 +bool gDone = false;
1.108 +nsIEventQueue* gEventQ = nullptr;
1.109 +
1.110 +class Listener : public nsIStreamListener
1.111 +{
1.112 +public:
1.113 + NS_DECL_ISUPPORTS
1.114 +
1.115 + Listener() {}
1.116 + virtual ~Listener() {}
1.117 +
1.118 + NS_IMETHOD OnStartRequest(nsIRequest *request, nsISupports *ctxt) {
1.119 + nsresult rv;
1.120 + nsCOMPtr<nsIURI> uri;
1.121 + nsCOMPtr<nsIChannel> channel = do_QueryInterface(request);
1.122 +
1.123 + rv = channel->GetURI(getter_AddRefs(uri));
1.124 + if (NS_SUCCEEDED(rv)) {
1.125 + char* str;
1.126 + rv = uri->GetSpec(&str);
1.127 + if (NS_SUCCEEDED(rv)) {
1.128 + fprintf(stdout, "Starting to load %s\n", str);
1.129 + free(str);
1.130 + }
1.131 + }
1.132 + return NS_OK;
1.133 + }
1.134 +
1.135 + NS_IMETHOD OnStopRequest(nsIRequest *request, nsISupports *ctxt,
1.136 + nsresult aStatus) {
1.137 + nsresult rv;
1.138 + nsCOMPtr<nsIURI> uri;
1.139 + nsCOMPtr<nsIChannel> channel = do_QueryInterface(request);
1.140 +
1.141 + rv = channel->GetURI(getter_AddRefs(uri));
1.142 + if (NS_SUCCEEDED(rv)) {
1.143 + char* str;
1.144 + rv = uri->GetSpec(&str);
1.145 + if (NS_SUCCEEDED(rv)) {
1.146 + fprintf(stdout, "Ending load %s, status=%x\n", str, aStatus);
1.147 + free(str);
1.148 + }
1.149 + }
1.150 + gDone = true;
1.151 + return NS_OK;
1.152 + }
1.153 +
1.154 + NS_IMETHOD OnDataAvailable(nsIRequest *request, nsISupports *ctxt,
1.155 + nsIInputStream *inStr,
1.156 + uint64_t sourceOffset, uint32_t count) {
1.157 + nsresult rv;
1.158 + char buf[1024];
1.159 + while (count > 0) {
1.160 + uint32_t amt;
1.161 + rv = inStr->Read(buf, sizeof(buf), &amt);
1.162 + count -= amt;
1.163 + char* c = buf;
1.164 + while (amt-- > 0) {
1.165 + fputc(*c++, stdout);
1.166 + }
1.167 + }
1.168 + return NS_OK;
1.169 + }
1.170 +};
1.171 +
1.172 +NS_IMPL_ISUPPORTS(Listener, nsIStreamListener, nsIRequestObserver)
1.173 +
1.174 +nsresult
1.175 +TestAsyncRead(const char* url)
1.176 +{
1.177 + nsresult rv;
1.178 +
1.179 + nsCOMPtr<nsIEventQueueService> eventQService =
1.180 + do_GetService(kEventQueueServiceCID, &rv);
1.181 + if (NS_FAILED(rv)) return rv;
1.182 +
1.183 + rv = eventQService->GetThreadEventQueue(NS_CURRENT_THREAD, &gEventQ);
1.184 + if (NS_FAILED(rv)) return rv;
1.185 +
1.186 + nsCOMPtr<nsIIOService> serv(do_GetService(kIOServiceCID, &rv));
1.187 + if (NS_FAILED(rv)) return rv;
1.188 +
1.189 + nsCOMPtr<nsIChannel> channel;
1.190 + rv = serv->NewChannel(url,
1.191 + nullptr, // base uri
1.192 + getter_AddRefs(channel));
1.193 + if (NS_FAILED(rv)) return rv;
1.194 +
1.195 + nsCOMPtr<nsIStreamListener> listener = new Listener();
1.196 + if (listener == nullptr)
1.197 + return NS_ERROR_OUT_OF_MEMORY;
1.198 + rv = channel->AsyncOpen(nullptr, listener);
1.199 + if (NS_FAILED(rv)) return rv;
1.200 +
1.201 + while (!gDone) {
1.202 + PLEvent* event;
1.203 + rv = gEventQ->GetEvent(&event);
1.204 + if (NS_FAILED(rv)) return rv;
1.205 + rv = gEventQ->HandleEvent(event);
1.206 + if (NS_FAILED(rv)) return rv;
1.207 + }
1.208 +
1.209 + return rv;
1.210 +}
1.211 +
1.212 +int
1.213 +main(int argc, char* argv[])
1.214 +{
1.215 + nsresult rv;
1.216 + {
1.217 + nsCOMPtr<nsIServiceManager> servMan;
1.218 + NS_InitXPCOM2(getter_AddRefs(servMan), nullptr, nullptr);
1.219 + nsCOMPtr<nsIComponentRegistrar> registrar = do_QueryInterface(servMan);
1.220 + NS_ASSERTION(registrar, "Null nsIComponentRegistrar");
1.221 + if (registrar)
1.222 + registrar->AutoRegister(nullptr);
1.223 +
1.224 + NS_ASSERTION(NS_SUCCEEDED(rv), "AutoregisterComponents failed");
1.225 +
1.226 + if (argc < 2) {
1.227 + printf("usage: %s resource://foo/<path-to-resolve>\n", argv[0]);
1.228 + return -1;
1.229 + }
1.230 +
1.231 + rv = SetupMapping();
1.232 + NS_ASSERTION(NS_SUCCEEDED(rv), "SetupMapping failed");
1.233 + if (NS_FAILED(rv)) return rv;
1.234 +
1.235 + rv = TestOpenInputStream(argv[1]);
1.236 + NS_ASSERTION(NS_SUCCEEDED(rv), "TestOpenInputStream failed");
1.237 +
1.238 + rv = TestAsyncRead(argv[1]);
1.239 + NS_ASSERTION(NS_SUCCEEDED(rv), "TestAsyncRead failed");
1.240 + } // this scopes the nsCOMPtrs
1.241 + // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
1.242 + rv = NS_ShutdownXPCOM(nullptr);
1.243 + NS_ASSERTION(NS_SUCCEEDED(rv), "NS_ShutdownXPCOM failed");
1.244 + return rv;
1.245 +}
1.246 +
1.247 +////////////////////////////////////////////////////////////////////////////////
