DXR is a code search and navigation tool aimed at making sense of large projects. It supports full-text and regex searches as well as structural queries.

Mercurial (d38398e5144e)

VCS Links

nsAHttpSegmentReader

nsAHttpSegmentWriter

nsAHttpTransaction

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256
/* 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 nsAHttpTransaction_h__
#define nsAHttpTransaction_h__

#include "nsISupports.h"
#include "nsTArray.h"
#include "nsWeakReference.h"

class nsIInterfaceRequestor;
class nsITransport;
class nsIRequestContext;

namespace mozilla { namespace net {

class nsAHttpConnection;
class nsAHttpSegmentReader;
class nsAHttpSegmentWriter;
class nsHttpTransaction;
class nsHttpRequestHead;
class nsHttpConnectionInfo;
class NullHttpTransaction;
class SpdyConnectTransaction;

//----------------------------------------------------------------------------
// Abstract base class for a HTTP transaction:
//
// A transaction is a "sink" for the response data.  The connection pushes
// data to the transaction by writing to it.  The transaction supports
// WriteSegments and may refuse to accept data if its buffers are full (its
// write function returns NS_BASE_STREAM_WOULD_BLOCK in this case).
//----------------------------------------------------------------------------

// 2af6d634-13e3-494c-8903-c9dce5c22fc0
#define NS_AHTTPTRANSACTION_IID \
{ 0x2af6d634, 0x13e3, 0x494c, {0x89, 0x03, 0xc9, 0xdc, 0xe5, 0xc2, 0x2f, 0xc0 }}

class nsAHttpTransaction : public nsSupportsWeakReference
{
public:
    NS_DECLARE_STATIC_IID_ACCESSOR(NS_AHTTPTRANSACTION_IID)

    // called by the connection when it takes ownership of the transaction.
    virtual void SetConnection(nsAHttpConnection *) = 0;

    // used to obtain the connection associated with this transaction
    virtual nsAHttpConnection *Connection() = 0;

    // called by the connection to get security callbacks to set on the
    // socket transport.
    virtual void GetSecurityCallbacks(nsIInterfaceRequestor **) = 0;

    // called to report socket status (see nsITransportEventSink)
    virtual void OnTransportStatus(nsITransport* transport,
                                   nsresult status, int64_t progress) = 0;

    // called to check the transaction status.
    virtual bool     IsDone() = 0;
    virtual nsresult Status() = 0;
    virtual uint32_t Caps() = 0;

    // called to notify that a requested DNS cache entry was refreshed.
    virtual void     SetDNSWasRefreshed() = 0;

    // called to find out how much request data is available for writing.
    virtual uint64_t Available() = 0;

    // called to read request data from the transaction.
    virtual nsresult ReadSegments(nsAHttpSegmentReader *reader,
                                  uint32_t count, uint32_t *countRead) = 0;

    // called to write response data to the transaction.
    virtual nsresult WriteSegments(nsAHttpSegmentWriter *writer,
                                   uint32_t count, uint32_t *countWritten) = 0;

    // These versions of the functions allow the overloader to specify whether or
    // not it is safe to call *Segments() in a loop while they return OK.
    // The callee should turn again to false if it is not, otherwise leave untouched
    virtual nsresult ReadSegmentsAgain(nsAHttpSegmentReader *reader,
                                       uint32_t count, uint32_t *countRead, bool *again)
    {
        return ReadSegments(reader, count, countRead);
    }
    virtual nsresult WriteSegmentsAgain(nsAHttpSegmentWriter *writer,
                                   uint32_t count, uint32_t *countWritten, bool *again)
    {
        return WriteSegments(writer, count, countWritten);
    }

    // called to close the transaction
    virtual void Close(nsresult reason) = 0;

    // called to indicate a failure with proxy CONNECT
    virtual void SetProxyConnectFailed() = 0;

    // called to retrieve the request headers of the transaction
    virtual nsHttpRequestHead *RequestHead() = 0;

    // determine the number of real http/1.x transactions on this
    // abstract object. Pipelines had multiple, SPDY has 0,
    // normal http transactions have 1.
    virtual uint32_t Http1xTransactionCount() = 0;

    // called to remove the unused sub transactions from an object that can
    // handle multiple transactions simultaneously (i.e. h2).
    //
    // Returns NS_ERROR_NOT_IMPLEMENTED if the object does not implement
    // sub-transactions.
    //
    // Returns NS_ERROR_ALREADY_OPENED if the subtransactions have been
    // at least partially written and cannot be moved.
    //
    virtual nsresult TakeSubTransactions(
        nsTArray<RefPtr<nsAHttpTransaction> > &outTransactions) = 0;

    // Occasionally the abstract interface has to give way to base implementations
    // to respect differences between spdy, h2, etc..
    // These Query* (and IsNullTransaction()) functions provide a way to do
    // that without using xpcom or rtti. Any calling code that can't deal with
    // a null response from one of them probably shouldn't be using nsAHttpTransaction

    // equivalent to !!dynamic_cast<NullHttpTransaction *>(this)
    // A null transaction is expected to return BASE_STREAM_CLOSED on all of
    // its IO functions all the time.
    virtual bool IsNullTransaction() { return false; }
    virtual NullHttpTransaction *QueryNullTransaction() { return nullptr; }

    // If we used rtti this would be the result of doing
    // dynamic_cast<nsHttpTransaction *>(this).. i.e. it can be nullptr for
    // non nsHttpTransaction implementations of nsAHttpTransaction
    virtual nsHttpTransaction *QueryHttpTransaction() { return nullptr; }

    // If we used rtti this would be the result of doing
    // dynamic_cast<SpdyConnectTransaction *>(this).. i.e. it can be nullptr for
    // other types
    virtual SpdyConnectTransaction *QuerySpdyConnectTransaction() { return nullptr; }

    // return the request context associated with the transaction
    virtual nsIRequestContext *RequestContext() { return nullptr; }

    // return the connection information associated with the transaction
    virtual nsHttpConnectionInfo *ConnectionInfo() = 0;

    // The base definition of these is done in nsHttpTransaction.cpp
    virtual bool ResponseTimeoutEnabled() const;
    virtual PRIntervalTime ResponseTimeout();

    // conceptually the security info is part of the connection, but sometimes
    // in the case of TLS tunneled within TLS the transaction might present
    // a more specific security info that cannot be represented as a layer in
    // the connection due to multiplexing. This interface represents such an
    // overload. If it returns NS_FAILURE the connection should be considered
    // authoritative.
    virtual nsresult GetTransactionSecurityInfo(nsISupports **)
    {
        return NS_ERROR_NOT_IMPLEMENTED;
    }

    virtual void DisableSpdy() { }
    virtual void ReuseConnectionOnRestartOK(bool) { }

    // Returns true if early-data is possible.
    virtual bool Do0RTT() {
        return false;
    }
    // This function will be called when a tls handshake has been finished and
    // we know whether early-data that was sent has been accepted or not, e.g.
    // do we need to restart a transaction. This will be called only if Do0RTT
    // returns true.
    // If aRestart parameter is true we need to restart the transaction,
    // otherwise the erly-data has been accepted and we can continue the
    // transaction.
    // If aAlpnChanged is true (and we were assuming http/2), we'll need to take
    // the transactions out of the session, rewind them all, and start them back
    // over as http/1 transactions
    // The function will return success or failure of the transaction restart.
    virtual nsresult Finish0RTT(bool aRestart, bool aAlpnChanged) {
        return NS_ERROR_NOT_IMPLEMENTED;
    }
};

NS_DEFINE_STATIC_IID_ACCESSOR(nsAHttpTransaction, NS_AHTTPTRANSACTION_IID)

#define NS_DECL_NSAHTTPTRANSACTION \
    void SetConnection(nsAHttpConnection *) override; \
    nsAHttpConnection *Connection() override; \
    void GetSecurityCallbacks(nsIInterfaceRequestor **) override;       \
    void OnTransportStatus(nsITransport* transport, \
                           nsresult status, int64_t progress) override; \
    bool     IsDone() override; \
    nsresult Status() override; \
    uint32_t Caps() override;   \
    void     SetDNSWasRefreshed() override; \
    uint64_t Available() override; \
    virtual nsresult ReadSegments(nsAHttpSegmentReader *, uint32_t, uint32_t *) override; \
    virtual nsresult WriteSegments(nsAHttpSegmentWriter *, uint32_t, uint32_t *) override; \
    virtual void Close(nsresult reason) override;                                \
    nsHttpConnectionInfo *ConnectionInfo() override;                             \
    void     SetProxyConnectFailed() override;                                   \
    virtual nsHttpRequestHead *RequestHead() override;                                   \
    uint32_t Http1xTransactionCount() override;                                  \
    nsresult TakeSubTransactions(nsTArray<RefPtr<nsAHttpTransaction> > &outTransactions) override;

//-----------------------------------------------------------------------------
// nsAHttpSegmentReader
//-----------------------------------------------------------------------------

class nsAHttpSegmentReader
{
public:
    // any returned failure code stops segment iteration
    virtual nsresult OnReadSegment(const char *segment,
                                   uint32_t count,
                                   uint32_t *countRead) = 0;

    // Ask the segment reader to commit to accepting size bytes of
    // data from subsequent OnReadSegment() calls or throw hard
    // (i.e. not wouldblock) exceptions. Implementations
    // can return NS_ERROR_FAILURE if they never make commitments of that size
    // (the default), NS_OK if they make the commitment, or
    // NS_BASE_STREAM_WOULD_BLOCK if they cannot make the
    // commitment now but might in the future and forceCommitment is not true .
    // (forceCommitment requires a hard failure or OK at this moment.)
    //
    // SpdySession uses this to make sure frames are atomic.
    virtual nsresult CommitToSegmentSize(uint32_t size, bool forceCommitment)
    {
        return NS_ERROR_FAILURE;
    }
};

#define NS_DECL_NSAHTTPSEGMENTREADER \
    nsresult OnReadSegment(const char *, uint32_t, uint32_t *) override;

//-----------------------------------------------------------------------------
// nsAHttpSegmentWriter
//-----------------------------------------------------------------------------

class nsAHttpSegmentWriter
{
public:
    // any returned failure code stops segment iteration
    virtual nsresult OnWriteSegment(char *segment,
                                    uint32_t count,
                                    uint32_t *countWritten) = 0;
};

#define NS_DECL_NSAHTTPSEGMENTWRITER \
    nsresult OnWriteSegment(char *, uint32_t, uint32_t *) override;

} // namespace net
} // namespace mozilla

#endif // nsAHttpTransaction_h__