netwerk/test/TestUpload.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 "TestCommon.h"
   7 #include <algorithm>
   8 #ifdef WIN32
   9 #include <windows.h>
  10 #endif
  11
  12 #include "nsIComponentRegistrar.h"
  13 #include "nsIIOService.h"
  14 #include "nsIServiceManager.h"
  15 #include "nsNetUtil.h"
  16
  17 #include "nsIUploadChannel.h"
  18
  19 static NS_DEFINE_CID(kIOServiceCID, NS_IOSERVICE_CID);
  20
  21 #include "prlog.h"
  22 #if defined(PR_LOGGING)
  23 //
  24 // set NSPR_LOG_MODULES=Test:5
  25 //
  26 static PRLogModuleInfo *gTestLog = nullptr;
  27 #endif
  28 #define LOG(args) PR_LOG(gTestLog, PR_LOG_DEBUG, args)
  29
  30 //-----------------------------------------------------------------------------
  31 // InputTestConsumer
  32 //-----------------------------------------------------------------------------
  33
  34 class InputTestConsumer : public nsIStreamListener
  35 {
  36   virtual ~InputTestConsumer();
  37
  38 public:
  39
  40   InputTestConsumer();
  41
  42   NS_DECL_ISUPPORTS
  43   NS_DECL_NSIREQUESTOBSERVER
  44   NS_DECL_NSISTREAMLISTENER
  45 };
  46
  47 InputTestConsumer::InputTestConsumer()
  48 {
  49 }
  50
  51 InputTestConsumer::~InputTestConsumer()
  52 {
  53 }
  54
  55 NS_IMPL_ISUPPORTS(InputTestConsumer,
  56                   nsIStreamListener,
  57                   nsIRequestObserver)
  58
  59 NS_IMETHODIMP
  60 InputTestConsumer::OnStartRequest(nsIRequest *request, nsISupports* context)
  61 {
  62   LOG(("InputTestConsumer::OnStartRequest\n"));
  63   return NS_OK;
  64 }
  65
  66 NS_IMETHODIMP
  67 InputTestConsumer::OnDataAvailable(nsIRequest *request,
  68                                    nsISupports* context,
  69                                    nsIInputStream *aIStream,
  70                                    uint64_t aSourceOffset,
  71                                    uint32_t aLength)
  72 {
  73   char buf[1025];
  74   uint32_t amt, size;
  75   nsresult rv;
  76
  77   while (aLength) {
  78     size = std::min<uint32_t>(aLength, sizeof(buf));
  79     rv = aIStream->Read(buf, size, &amt);
  80     if (NS_FAILED(rv)) {
  81       NS_ASSERTION((NS_BASE_STREAM_WOULD_BLOCK != rv),
  82                    "The stream should never block.");
  83       return rv;
  84     }
  85     aLength -= amt;
  86   }
  87   return NS_OK;
  88 }
  89
  90
  91 NS_IMETHODIMP
  92 InputTestConsumer::OnStopRequest(nsIRequest *request, nsISupports* context,
  93                                  nsresult aStatus)
  94 {
  95     LOG(("InputTestConsumer::OnStopRequest [status=%x]\n", aStatus));
  96     QuitPumpingEvents();
  97     return NS_OK;
  98 }
  99
 100
 101 int
 102 main(int argc, char* argv[])
 103 {
 104     if (test_common_init(&argc, &argv) != 0)
 105         return -1;
 106
 107     nsresult rv;
 108
 109     if (argc < 2) {
 110         printf("usage: %s <url> <file-to-upload>\n", argv[0]);
 111         return -1;
 112     }
 113     char* uriSpec  = argv[1];
 114     char* fileName = argv[2];
 115
 116 #if defined(PR_LOGGING)
 117     gTestLog = PR_NewLogModule("Test");
 118 #endif
 119
 120     {
 121         nsCOMPtr<nsIServiceManager> servMan;
 122         NS_InitXPCOM2(getter_AddRefs(servMan), nullptr, nullptr);
 123
 124         // first thing to do is create ourselves a stream that
 125         // is to be uploaded.
 126         nsCOMPtr<nsIInputStream> uploadStream;
 127         rv = NS_NewPostDataStream(getter_AddRefs(uploadStream),
 128                                   true,
 129                                   nsDependentCString(fileName)); // XXX UTF-8
 130         if (NS_FAILED(rv)) return -1;
 131
 132         nsCOMPtr<nsIIOService> ioService(do_GetService(kIOServiceCID, &rv));
 133
 134         // create our url.
 135         nsCOMPtr<nsIURI> uri;
 136         rv = NS_NewURI(getter_AddRefs(uri), uriSpec);
 137         if (NS_FAILED(rv)) return -1;
 138
 139         nsCOMPtr<nsIChannel> channel;
 140         rv = ioService->NewChannelFromURI(uri, getter_AddRefs(channel));
 141         if (NS_FAILED(rv)) return -1;
 142
 143         // QI and set the upload stream
 144         nsCOMPtr<nsIUploadChannel> uploadChannel(do_QueryInterface(channel));
 145         uploadChannel->SetUploadStream(uploadStream, EmptyCString(), -1);
 146
 147         // create a dummy listener
 148         InputTestConsumer* listener;
 149
 150         listener = new InputTestConsumer;
 151         if (!listener) {
 152             NS_ERROR("Failed to create a new stream listener!");
 153             return -1;
 154         }
 155         NS_ADDREF(listener);
 156
 157         channel->AsyncOpen(listener, nullptr);
 158
 159         PumpEvents();
 160     } // this scopes the nsCOMPtrs
 161     // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
 162     rv = NS_ShutdownXPCOM(nullptr);
 163     NS_ASSERTION(NS_SUCCEEDED(rv), "NS_ShutdownXPCOM failed");
 164
 165     return 0;
 166 }
 167