mirror of
https://github.com/roytam1/UXP.git
synced 2026-05-26 05:46:58 +00:00
Moonchild
efa2f805fe
This prevents a potential race and simplifies the code a bit by keeping the bytes read separate instead of using mData, which is modified from another thread from OnDataAvailable. Relaxed atomics are fine for these, since they don't guard any memory.
56 lines
1.8 KiB
C++
56 lines
1.8 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef nsIncrementalStreamLoader_h__
|
|
#define nsIncrementalStreamLoader_h__
|
|
|
|
#include "nsIThreadRetargetableStreamListener.h"
|
|
#include "nsIIncrementalStreamLoader.h"
|
|
#include "nsCOMPtr.h"
|
|
#include "mozilla/Attributes.h"
|
|
#include "mozilla/Vector.h"
|
|
|
|
class nsIRequest;
|
|
|
|
class nsIncrementalStreamLoader final : public nsIIncrementalStreamLoader
|
|
, public nsIThreadRetargetableStreamListener
|
|
{
|
|
public:
|
|
NS_DECL_THREADSAFE_ISUPPORTS
|
|
NS_DECL_NSIINCREMENTALSTREAMLOADER
|
|
NS_DECL_NSIREQUESTOBSERVER
|
|
NS_DECL_NSISTREAMLISTENER
|
|
NS_DECL_NSITHREADRETARGETABLESTREAMLISTENER
|
|
|
|
nsIncrementalStreamLoader();
|
|
|
|
static nsresult
|
|
Create(nsISupports *aOuter, REFNSIID aIID, void **aResult);
|
|
|
|
protected:
|
|
~nsIncrementalStreamLoader();
|
|
|
|
static nsresult WriteSegmentFun(nsIInputStream *, void *, const char *,
|
|
uint32_t, uint32_t, uint32_t *);
|
|
|
|
// Utility method to free mData, if present, and update other state to
|
|
// reflect that no data has been allocated.
|
|
void ReleaseData();
|
|
|
|
nsCOMPtr<nsIIncrementalStreamLoaderObserver> mObserver;
|
|
nsCOMPtr<nsISupports> mContext; // the observer's context
|
|
nsCOMPtr<nsIRequest> mRequest;
|
|
|
|
// Buffer to accumulate incoming data. We preallocate if contentSize is
|
|
// available.
|
|
mozilla::Vector<uint8_t, 0> mData;
|
|
|
|
// Number of bytes read, which may differ from the number of bytes in mData,
|
|
// since we incrementally remove from there.
|
|
mozilla::Atomic<uint32_t, mozilla::Relaxed> mBytesRead;
|
|
};
|
|
|
|
#endif // nsIncrementalStreamLoader_h__
|