blob: 769c2d6992fc3fc71940d1a366f8040b410f0521 [file] [log] [blame]
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef NET_BASE_UPLOAD_DATA_STREAM_H_
#define NET_BASE_UPLOAD_DATA_STREAM_H_
#include "base/gtest_prod_util.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_vector.h"
#include "base/memory/weak_ptr.h"
#include "net/base/completion_callback.h"
#include "net/base/net_export.h"
namespace net {
class DrainableIOBuffer;
class IOBuffer;
class UploadElementReader;
// A class to read all elements from an UploadData object.
class NET_EXPORT UploadDataStream {
public:
// An enum used to construct chunked data stream.
enum Chunked { CHUNKED };
// Constructs a non-chunked data stream.
// |element_readers| is cleared by this ctor.
UploadDataStream(ScopedVector<UploadElementReader>* element_readers,
int64 identifier);
// Constructs a chunked data stream.
UploadDataStream(Chunked chunked, int64 identifier);
~UploadDataStream();
// Creates UploadDataStream with a reader.
static UploadDataStream* CreateWithReader(
scoped_ptr<UploadElementReader> reader,
int64 identifier);
// Initializes the stream. This function must be called before calling any
// other method. It is not valid to call any method (other than the
// destructor) if Init() returns a failure. This method can be called multiple
// times. Calling this method after a Init() success results in resetting the
// state.
//
// Does the initialization synchronously and returns the result if possible,
// otherwise returns ERR_IO_PENDING and runs the callback with the result.
//
// Returns OK on success. Returns ERR_UPLOAD_FILE_CHANGED if the expected
// file modification time is set (usually not set, but set for sliced
// files) and the target file is changed.
int Init(const CompletionCallback& callback);
// Initializes the stream synchronously.
// Use this method only if the thread is IO allowed or the data is in-memory.
int InitSync();
// When possible, reads up to |buf_len| bytes synchronously from the upload
// data stream to |buf| and returns the number of bytes read; otherwise,
// returns ERR_IO_PENDING and calls |callback| with the number of bytes read.
// Partial reads are allowed. Zero is returned on a call to Read when there
// are no remaining bytes in the stream, and IsEof() will return true
// hereafter.
//
// If there's less data to read than we initially observed (i.e. the actual
// upload data is smaller than size()), zeros are padded to ensure that
// size() bytes can be read, which can happen for TYPE_FILE payloads.
int Read(IOBuffer* buf, int buf_len, const CompletionCallback& callback);
// Reads data always synchronously.
// Use this method only if the thread is IO allowed or the data is in-memory.
int ReadSync(IOBuffer* buf, int buf_len);
// Identifies a particular upload instance, which is used by the cache to
// formulate a cache key. This value should be unique across browser
// sessions. A value of 0 is used to indicate an unspecified identifier.
int64 identifier() const { return identifier_; }
// Returns the total size of the data stream and the current position.
// size() is not to be used to determine whether the stream has ended
// because it is possible for the stream to end before its size is reached,
// for example, if the file is truncated. When the data is chunked, size()
// always returns zero.
uint64 size() const { return total_size_; }
uint64 position() const { return current_position_; }
bool is_chunked() const { return is_chunked_; }
bool last_chunk_appended() const { return last_chunk_appended_; }
const ScopedVector<UploadElementReader>& element_readers() const {
return element_readers_;
}
// Returns true if all data has been consumed from this upload data
// stream.
bool IsEOF() const;
// Returns true if the upload data in the stream is entirely in memory.
bool IsInMemory() const;
// Adds the given chunk of bytes to be sent with chunked transfer encoding.
void AppendChunk(const char* bytes, int bytes_len, bool is_last_chunk);
private:
friend class SpdyHttpStreamSpdy2Test;
friend class SpdyHttpStreamSpdy3Test;
friend class SpdyNetworkTransactionSpdy2Test;
friend class SpdyNetworkTransactionSpdy3Test;
// Resets this instance to the uninitialized state.
void Reset();
// Runs Init() for all element readers.
// This method is used to implement Init().
int InitInternal(int start_index, const CompletionCallback& callback);
// Resumes initialization and runs callback with the result when necessary.
void ResumePendingInit(int start_index,
const CompletionCallback& callback,
int previous_result);
// Finalizes the initialization process.
// This method is used to implement Init().
void FinalizeInitialization();
// Reads data from the element readers.
// This method is used to implement Read().
int ReadInternal(scoped_refptr<DrainableIOBuffer> buf,
const CompletionCallback& callback);
// Resumes pending read and calls callback with the result when necessary.
void ResumePendingRead(scoped_refptr<DrainableIOBuffer> buf,
const CompletionCallback& callback,
int previous_result);
// These methods are provided only to be used by unit tests.
static void ResetMergeChunks();
static void set_merge_chunks(bool merge) { merge_chunks_ = merge; }
ScopedVector<UploadElementReader> element_readers_;
// Index of the current upload element (i.e. the element currently being
// read). The index is used as a cursor to iterate over elements in
// |upload_data_|.
size_t element_index_;
// Size and current read position within the upload data stream.
// |total_size_| is set to zero when the data is chunked.
uint64 total_size_;
uint64 current_position_;
const int64 identifier_;
const bool is_chunked_;
bool last_chunk_appended_;
// True if the initialization was successful.
bool initialized_successfully_;
// Callback to resume reading chunked data.
base::Closure pending_chunked_read_callback_;
base::WeakPtrFactory<UploadDataStream> weak_ptr_factory_;
// TODO(satish): Remove this once we have a better way to unit test POST
// requests with chunked uploads.
static bool merge_chunks_;
DISALLOW_COPY_AND_ASSIGN(UploadDataStream);
};
} // namespace net
#endif // NET_BASE_UPLOAD_DATA_STREAM_H_