tor-browser

The Tor Browser
git clone https://git.dasho.dev/tor-browser.git
Log | Files | Refs | README | LICENSE

NeckoCommon.cpp (2288B)

      1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
      2 /* vim: set sw=2 ts=8 et tw=80 : */
      3 
      4 /* This Source Code Form is subject to the terms of the Mozilla Public
      5 * License, v. 2.0. If a copy of the MPL was not distributed with this
      6 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
      7 
      8 #include "NeckoCommon.h"
      9 
     10 #include "nsIInputStream.h"
     11 #include "nsIMultiPartChannel.h"
     12 #include "nsIParentChannel.h"
     13 #include "nsStringStream.h"
     14 
     15 namespace mozilla::net {
     16 
     17 nsresult ForwardStreamListenerFunctions(nsTArray<StreamListenerFunction> aCalls,
     18                                        nsIStreamListener* aParent) {
     19  nsresult rv = NS_OK;
     20  for (auto& variant : aCalls) {
     21    variant.match(
     22        [&](OnStartRequestParams& aParams) {
     23          rv = aParent->OnStartRequest(aParams.request);
     24          if (NS_FAILED(rv)) {
     25            aParams.request->Cancel(rv);
     26          }
     27        },
     28        [&](OnDataAvailableParams& aParams) {
     29          // Don't deliver OnDataAvailable if we've
     30          // already failed.
     31          if (NS_FAILED(rv)) {
     32            return;
     33          }
     34          nsCOMPtr<nsIInputStream> stringStream;
     35          rv = NS_NewCStringInputStream(getter_AddRefs(stringStream),
     36                                        std::move(aParams.data));
     37          if (NS_SUCCEEDED(rv)) {
     38            rv = aParent->OnDataAvailable(aParams.request, stringStream,
     39                                          aParams.offset, aParams.count);
     40          }
     41          if (NS_FAILED(rv)) {
     42            aParams.request->Cancel(rv);
     43          }
     44        },
     45        [&](OnStopRequestParams& aParams) {
     46          if (NS_SUCCEEDED(rv)) {
     47            aParent->OnStopRequest(aParams.request, aParams.status);
     48          } else {
     49            aParent->OnStopRequest(aParams.request, rv);
     50          }
     51          rv = NS_OK;
     52        },
     53        [&](OnAfterLastPartParams& aParams) {
     54          nsCOMPtr<nsIMultiPartChannelListener> multiListener =
     55              do_QueryInterface(aParent);
     56          if (multiListener) {
     57            if (NS_SUCCEEDED(rv)) {
     58              multiListener->OnAfterLastPart(aParams.status);
     59            } else {
     60              multiListener->OnAfterLastPart(rv);
     61            }
     62          }
     63        });
     64  }
     65  return rv;
     66 }
     67 
     68 }  // namespace mozilla::net