106 lines
3.8 KiB
C++
106 lines
3.8 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
|
|
/* 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 nsInputStreamPump_h__
|
|
#define nsInputStreamPump_h__
|
|
|
|
#include "nsIInputStreamPump.h"
|
|
#include "nsIAsyncInputStream.h"
|
|
#include "nsIThreadRetargetableRequest.h"
|
|
#include "nsCOMPtr.h"
|
|
#include "mozilla/Attributes.h"
|
|
#include "mozilla/ReentrantMonitor.h"
|
|
|
|
class nsIInputStream;
|
|
class nsILoadGroup;
|
|
class nsIStreamListener;
|
|
|
|
class nsInputStreamPump final : public nsIInputStreamPump
|
|
, public nsIInputStreamCallback
|
|
, public nsIThreadRetargetableRequest
|
|
{
|
|
~nsInputStreamPump();
|
|
|
|
public:
|
|
typedef mozilla::ReentrantMonitorAutoEnter ReentrantMonitorAutoEnter;
|
|
NS_DECL_THREADSAFE_ISUPPORTS
|
|
NS_DECL_NSIREQUEST
|
|
NS_DECL_NSIINPUTSTREAMPUMP
|
|
NS_DECL_NSIINPUTSTREAMCALLBACK
|
|
NS_DECL_NSITHREADRETARGETABLEREQUEST
|
|
|
|
nsInputStreamPump();
|
|
|
|
static nsresult
|
|
Create(nsInputStreamPump **result,
|
|
nsIInputStream *stream,
|
|
int64_t streamPos = -1,
|
|
int64_t streamLen = -1,
|
|
uint32_t segsize = 0,
|
|
uint32_t segcount = 0,
|
|
bool closeWhenDone = false);
|
|
|
|
typedef void (*PeekSegmentFun)(void *closure, const uint8_t *buf,
|
|
uint32_t bufLen);
|
|
/**
|
|
* Peek into the first chunk of data that's in the stream. Note that this
|
|
* method will not call the callback when there is no data in the stream.
|
|
* The callback will be called at most once.
|
|
*
|
|
* The data from the stream will not be consumed, i.e. the pump's listener
|
|
* can still read all the data.
|
|
*
|
|
* Do not call before asyncRead. Do not call after onStopRequest.
|
|
*/
|
|
nsresult PeekStream(PeekSegmentFun callback, void *closure);
|
|
|
|
/**
|
|
* Dispatched (to the main thread) by OnStateStop if it's called off main
|
|
* thread. Updates mState based on return value of OnStateStop.
|
|
*/
|
|
nsresult CallOnStateStop();
|
|
|
|
protected:
|
|
|
|
enum {
|
|
STATE_IDLE,
|
|
STATE_START,
|
|
STATE_TRANSFER,
|
|
STATE_STOP
|
|
};
|
|
|
|
nsresult EnsureWaiting();
|
|
uint32_t OnStateStart();
|
|
uint32_t OnStateTransfer();
|
|
uint32_t OnStateStop();
|
|
|
|
uint32_t mState;
|
|
nsCOMPtr<nsILoadGroup> mLoadGroup;
|
|
nsCOMPtr<nsIStreamListener> mListener;
|
|
nsCOMPtr<nsISupports> mListenerContext;
|
|
nsCOMPtr<nsIEventTarget> mTargetThread;
|
|
nsCOMPtr<nsIInputStream> mStream;
|
|
nsCOMPtr<nsIAsyncInputStream> mAsyncStream;
|
|
uint64_t mStreamOffset;
|
|
uint64_t mStreamLength;
|
|
uint32_t mSegSize;
|
|
uint32_t mSegCount;
|
|
nsresult mStatus;
|
|
uint32_t mSuspendCount;
|
|
uint32_t mLoadFlags;
|
|
bool mIsPending;
|
|
// True while in OnInputStreamReady, calling OnStateStart, OnStateTransfer
|
|
// and OnStateStop. Used to prevent calls to AsyncWait during callbacks.
|
|
bool mProcessingCallbacks;
|
|
// True if waiting on the "input stream ready" callback.
|
|
bool mWaitingForInputStreamReady;
|
|
bool mCloseWhenDone;
|
|
bool mRetargeting;
|
|
// Protects state/member var accesses across multiple threads.
|
|
mozilla::ReentrantMonitor mMonitor;
|
|
};
|
|
|
|
#endif // !nsInputStreamChannel_h__
|