forked from chromium/chromium
-
Notifications
You must be signed in to change notification settings - Fork 0
/
chunked_data_stream_uploader.h
86 lines (65 loc) · 2.89 KB
/
chunked_data_stream_uploader.h
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
// Copyright (c) 2017 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 IOS_NET_CHUNKED_DATA_STREAM_UPLOADER_H_
#define IOS_NET_CHUNKED_DATA_STREAM_UPLOADER_H_
#include <stdint.h>
#include <memory>
#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "net/base/upload_data_stream.h"
namespace net {
class IOBuffer;
// The ChunkedDataStreamUploader is used to support chunked data post for iOS
// NSMutableURLRequest HTTPBodyStream. Called on the network thread. It's
// responsible to coordinate the internal callbacks from network layer with the
// NSInputStream data. Rewind is not supported.
class ChunkedDataStreamUploader : public net::UploadDataStream {
public:
class Delegate {
public:
Delegate() {}
virtual ~Delegate() {}
// Called when the request is ready to read the data for request body.
// Data must be read in this function and put into |buf|. |buf_len| gives
// the length of the provided buffer, and the return values gives the actual
// bytes read. UploadDataStream::Read() currently does not support to return
// failure, so need to handle the stream errors in the callback.
virtual int OnRead(char* buffer, int buffer_length) = 0;
};
ChunkedDataStreamUploader(Delegate* delegate);
~ChunkedDataStreamUploader() override;
// Interface for iOS layer to try to upload data. If there already has a
// internal ReadInternal() callback ready from the network layer, data will be
// writen to buffer immediately. Otherwise, it will do nothing in order to
// wait internal callback. Once it is ready for the network layer to read
// data, the OnRead() callback will be called.
void UploadWhenReady(bool is_final_chunk);
// The uploader interface for iOS layer to use.
base::WeakPtr<ChunkedDataStreamUploader> GetWeakPtr() {
return weak_factory_.GetWeakPtr();
}
private:
// Internal function to implement data upload to network layer.
int Upload();
// net::UploadDataStream implementation:
int InitInternal(const NetLogWithSource& net_log) override;
int ReadInternal(IOBuffer* buffer, int buffer_length) override;
void ResetInternal() override;
Delegate* const delegate_;
// The pointer to the network layer buffer to send and the length of the
// buffer.
net::IOBuffer* pending_read_buffer_;
int pending_read_buffer_length_;
// Flags indicating current upload process has network read callback pending.
bool pending_internal_read_;
// Flags indicating if current block is the last.
bool is_final_chunk_;
// Set to false when a read starts, does not support rewinding
// for stream upload.
bool is_front_of_stream_;
base::WeakPtrFactory<ChunkedDataStreamUploader> weak_factory_;
DISALLOW_COPY_AND_ASSIGN(ChunkedDataStreamUploader);
};
} // namespace net
#endif // IOS_NET_CHUNKED_DATA_STREAM_UPLOADER_H_