netwerk/test/TestCallbacks.cpp

   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 "TestCommon.h"
   7 #include <stdio.h>
   8 #ifdef WIN32
   9 #include <windows.h>
  10 #endif
  11 #include "nspr.h"
  12 #include "nscore.h"
  13 #include "nsCOMPtr.h"
  14 #include "nsIIOService.h"
  15 #include "nsIServiceManager.h"
  16 #include "nsIStreamListener.h"
  17 #include "nsIInputStream.h"
  18 #include "nsIChannel.h"
  19 #include "nsIURL.h"
  20 #include "nsIInterfaceRequestor.h"
  21 #include "nsIInterfaceRequestorUtils.h"
  22 #include "nsIDNSService.h"
  23 #include "mozilla/Attributes.h"
  24
  25 #include "nsISimpleEnumerator.h"
  26 #include "nsNetUtil.h"
  27 #include "nsStringAPI.h"
  28
  29 static NS_DEFINE_CID(kIOServiceCID,              NS_IOSERVICE_CID);
  30
  31 static bool gError = false;
  32 static int32_t gKeepRunning = 0;
  33
  34 #define NS_IEQUALS_IID \
  35     { 0x11c5c8ee, 0x1dd2, 0x11b2, \
  36       { 0xa8, 0x93, 0xbb, 0x23, 0xa1, 0xb6, 0x27, 0x76 }}
  37
  38 class nsIEquals : public nsISupports {
  39 public:
  40     NS_DECLARE_STATIC_IID_ACCESSOR(NS_IEQUALS_IID)
  41     NS_IMETHOD Equals(void *aPtr, bool *_retval) = 0;
  42 };
  43
  44 NS_DEFINE_STATIC_IID_ACCESSOR(nsIEquals, NS_IEQUALS_IID)
  45
  46 class ConsumerContext MOZ_FINAL : public nsIEquals {
  47
  48     ~ConsumerContext() {}
  49
  50 public:
  51     NS_DECL_THREADSAFE_ISUPPORTS
  52
  53     ConsumerContext() { }
  54
  55     NS_IMETHOD Equals(void *aPtr, bool *_retval) {
  56         *_retval = true;
  57         if (aPtr != this) *_retval = false;
  58         return NS_OK;
  59     }
  60 };
  61
  62 NS_IMPL_ISUPPORTS(ConsumerContext, nsIEquals)
  63
  64 class Consumer : public nsIStreamListener {
  65
  66     virtual ~Consumer();
  67
  68 public:
  69     NS_DECL_THREADSAFE_ISUPPORTS
  70     NS_DECL_NSIREQUESTOBSERVER
  71     NS_DECL_NSISTREAMLISTENER
  72
  73     Consumer();
  74     nsresult Init(nsIURI *aURI, nsIChannel *aChannel, nsISupports *aContext);
  75     nsresult Validate(nsIRequest *request, nsISupports *aContext);
  76
  77     // member data
  78     bool    mOnStart; // have we received an OnStart?
  79     bool    mOnStop;  // have we received an onStop?
  80     int32_t mOnDataCount; // number of times OnData was called.
  81     nsCOMPtr<nsIURI>     mURI;
  82     nsCOMPtr<nsIChannel> mChannel;
  83     nsCOMPtr<nsIEquals>  mContext;
  84 };
  85
  86 // nsISupports implementation
  87 NS_IMPL_ISUPPORTS(Consumer, nsIStreamListener, nsIRequestObserver)
  88
  89
  90 // nsIRequestObserver implementation
  91 NS_IMETHODIMP
  92 Consumer::OnStartRequest(nsIRequest *request, nsISupports* aContext) {
  93     fprintf(stderr, "Consumer::OnStart() -> in\n\n");
  94
  95     if (mOnStart) {
  96         fprintf(stderr, "INFO: multiple OnStarts received\n");
  97     }
  98     mOnStart = true;
  99
 100     nsresult rv = Validate(request, aContext);
 101     if (NS_FAILED(rv)) return rv;
 102
 103     fprintf(stderr, "Consumer::OnStart() -> out\n\n");
 104     return rv;
 105 }
 106
 107 NS_IMETHODIMP
 108 Consumer::OnStopRequest(nsIRequest *request, nsISupports *aContext,
 109                         nsresult aStatus) {
 110     fprintf(stderr, "Consumer::OnStop() -> in\n\n");
 111
 112     if (!mOnStart) {
 113         gError = true;
 114         fprintf(stderr, "ERROR: No OnStart received\n");
 115     }
 116
 117     if (mOnStop) {
 118         fprintf(stderr, "INFO: multiple OnStops received\n");
 119     }
 120
 121     fprintf(stderr, "INFO: received %d OnData()s\n", mOnDataCount);
 122
 123     mOnStop = true;
 124
 125     nsresult rv = Validate(request, aContext);
 126     if (NS_FAILED(rv)) return rv;
 127
 128     fprintf(stderr, "Consumer::OnStop() -> out\n\n");
 129     return rv;
 130 }
 131
 132
 133 // nsIStreamListener implementation
 134 NS_IMETHODIMP
 135 Consumer::OnDataAvailable(nsIRequest *request, nsISupports *aContext,
 136                           nsIInputStream *aIStream,
 137                           uint64_t aOffset, uint32_t aLength) {
 138     fprintf(stderr, "Consumer::OnData() -> in\n\n");
 139
 140     if (!mOnStart) {
 141         gError = true;
 142         fprintf(stderr, "ERROR: No OnStart received\n");
 143     }
 144
 145     mOnDataCount += 1;
 146
 147     nsresult rv = Validate(request, aContext);
 148     if (NS_FAILED(rv)) return rv;
 149
 150     fprintf(stderr, "Consumer::OnData() -> out\n\n");
 151     return rv;
 152 }
 153
 154 // Consumer implementation
 155 Consumer::Consumer() {
 156     mOnStart = mOnStop = false;
 157     mOnDataCount = 0;
 158     gKeepRunning++;
 159 }
 160
 161 Consumer::~Consumer() {
 162     fprintf(stderr, "Consumer::~Consumer -> in\n\n");
 163
 164     if (!mOnStart) {
 165         gError = true;
 166         fprintf(stderr, "ERROR: Never got an OnStart\n");
 167     }
 168
 169     if (!mOnStop) {
 170         gError = true;
 171         fprintf(stderr, "ERROR: Never got an OnStop \n");
 172     }
 173
 174     fprintf(stderr, "Consumer::~Consumer -> out\n\n");
 175     if (--gKeepRunning == 0)
 176       QuitPumpingEvents();
 177 }
 178
 179 nsresult
 180 Consumer::Init(nsIURI *aURI, nsIChannel* aChannel, nsISupports *aContext) {
 181     mURI     = aURI;
 182     mChannel = aChannel;
 183     mContext = do_QueryInterface(aContext);
 184     return NS_OK;
 185 }
 186
 187 nsresult
 188 Consumer::Validate(nsIRequest* request, nsISupports *aContext) {
 189     nsresult rv = NS_OK;
 190     nsCOMPtr<nsIURI> uri;
 191     nsCOMPtr<nsIChannel> aChannel = do_QueryInterface(request);
 192
 193     rv = aChannel->GetURI(getter_AddRefs(uri));
 194     if (NS_FAILED(rv)) return rv;
 195
 196     bool same = false;
 197
 198     rv = mURI->Equals(uri, &same);
 199     if (NS_FAILED(rv)) return rv;
 200
 201     if (!same)
 202         fprintf(stderr, "INFO: URIs do not match\n");
 203
 204     rv = mContext->Equals((void*)aContext, &same);
 205     if (NS_FAILED(rv)) return rv;
 206
 207     if (!same) {
 208         gError = true;
 209         fprintf(stderr, "ERROR: Contexts do not match\n");
 210     }
 211     return rv;
 212 }
 213
 214 nsresult StartLoad(const char *);
 215
 216 int main(int argc, char *argv[]) {
 217     if (test_common_init(&argc, &argv) != 0)
 218         return -1;
 219
 220     nsresult rv = NS_OK;
 221     bool cmdLineURL = false;
 222
 223     if (argc > 1) {
 224         // run in signle url mode
 225         cmdLineURL = true;
 226     }
 227
 228     rv = NS_InitXPCOM2(nullptr, nullptr, nullptr);
 229     if (NS_FAILED(rv)) return -1;
 230
 231     if (cmdLineURL) {
 232         rv = StartLoad(argv[1]);
 233     } else {
 234         rv = StartLoad("http://badhostnamexyz/test.txt");
 235     }
 236     if (NS_FAILED(rv)) return -1;
 237
 238     // Enter the message pump to allow the URL load to proceed.
 239     PumpEvents();
 240
 241     NS_ShutdownXPCOM(nullptr);
 242     if (gError) {
 243         fprintf(stderr, "\n\n-------ERROR-------\n\n");
 244     }
 245     return 0;
 246 }
 247
 248 nsresult StartLoad(const char *aURISpec) {
 249     nsresult rv = NS_OK;
 250
 251     // create a context
 252     ConsumerContext *context = new ConsumerContext;
 253     nsCOMPtr<nsISupports> contextSup = do_QueryInterface(context, &rv);
 254     if (NS_FAILED(rv)) return rv;
 255
 256
 257     nsCOMPtr<nsIIOService> serv = do_GetService(kIOServiceCID, &rv);
 258     if (NS_FAILED(rv)) return rv;
 259
 260     // create a uri
 261     nsCOMPtr<nsIURI> uri;
 262     rv = NS_NewURI(getter_AddRefs(uri), aURISpec);
 263     if (NS_FAILED(rv)) return rv;
 264
 265     // create a channel
 266     nsCOMPtr<nsIChannel> channel;
 267     rv = serv->NewChannelFromURI(uri, getter_AddRefs(channel));
 268     if (NS_FAILED(rv)) return rv;
 269
 270     Consumer *consumer = new Consumer;
 271     rv = consumer->Init(uri, channel, contextSup);
 272     if (NS_FAILED(rv)) return rv;
 273
 274     // kick off the load
 275     nsCOMPtr<nsIRequest> request;
 276     return channel->AsyncOpen(static_cast<nsIStreamListener*>(consumer), contextSup);
 277 }