Google Git
Sign in
chromium / chromium / src / lkgr / . / net / base / upload_data_stream.cc
blob: 8b4c183659adf6244483c14b09e7a8b85ce9ca56 [file] [log] [blame]
Avi Drissman64595482022-09-14 20:52:29[diff] [blame]1// Copyright 2012 The Chromium Authors
license.botbf09a502008-08-24 00:55:55[diff] [blame]2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
initial.commit586acc5fe2008-07-26 22:42:52[diff] [blame]4
5#include "net/base/upload_data_stream.h"
6
Hans Wennborg0924470b2020-04-27 21:08:05[diff] [blame]7#include "base/check_op.h"
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]8#include "base/values.h"
[email protected]597cf6e2009-05-29 09:43:26[diff] [blame]9#include "net/base/io_buffer.h"
[email protected]9396b252008-09-29 17:29:38[diff] [blame]10#include "net/base/net_errors.h"
mikecirone8b85c432016-09-08 19:11:00[diff] [blame]11#include "net/log/net_log_event_type.h"
[email protected]1eb89e82008-08-15 12:27:03[diff] [blame]12
initial.commit586acc5fe2008-07-26 22:42:52[diff] [blame]13namespace net {
14
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]15namespace {
16
Liam Brady0c651722023-02-24 23:48:20[diff] [blame]17base::Value::Dict NetLogInitEndInfoParams(int result,
18 int total_size,
19 bool is_chunked) {
Matt Menkef523f552022-06-02 17:07:58[diff] [blame]20 base::Value::Dict dict;
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]21
Matt Menkef523f552022-06-02 17:07:58[diff] [blame]22 dict.Set("net_error", result);
23 dict.Set("total_size", total_size);
24 dict.Set("is_chunked", is_chunked);
Liam Brady0c651722023-02-24 23:48:20[diff] [blame]25 return dict;
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]26}
27
Liam Brady0c651722023-02-24 23:48:20[diff] [blame]28base::Value::Dict CreateReadInfoParams(int current_position) {
Matt Menkef523f552022-06-02 17:07:58[diff] [blame]29 base::Value::Dict dict;
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]30
Matt Menkef523f552022-06-02 17:07:58[diff] [blame]31 dict.Set("current_position", current_position);
Liam Brady0c651722023-02-24 23:48:20[diff] [blame]32 return dict;
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]33}
34
35} // namespace
36
wtc69f8ea82015-06-04 00:08:13[diff] [blame]37UploadDataStream::UploadDataStream(bool is_chunked, int64_t identifier)
Yutaka Hiranoc267fb242022-07-05 01:27:17[diff] [blame]38 : UploadDataStream(is_chunked, /*has_null_source=*/false, identifier) {}
39UploadDataStream::UploadDataStream(bool is_chunked,
40 bool has_null_source,
41 int64_t identifier)
42 : identifier_(identifier),
43 is_chunked_(is_chunked),
44 has_null_source_(has_null_source) {}
[email protected]0ab2e242012-02-08 06:07:00[diff] [blame]45
Chris Watkins68b15032017-12-01 03:07:13[diff] [blame]46UploadDataStream::~UploadDataStream() = default;
[email protected]d100e44f2011-01-26 22:47:11[diff] [blame]47
Matt Menkecc1d3a902018-02-05 18:27:33[diff] [blame]48int UploadDataStream::Init(CompletionOnceCallback callback,
tfarina428341112016-09-22 13:38:20[diff] [blame]49 const NetLogWithSource& net_log) {
[email protected]e5d477342012-11-02 15:18:46[diff] [blame]50 Reset();
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]51 DCHECK(!initialized_successfully_);
52 DCHECK(callback_.is_null());
53 DCHECK(!callback.is_null() || IsInMemory());
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]54 net_log_ = net_log;
mikecirone8b85c432016-09-08 19:11:00[diff] [blame]55 net_log_.BeginEvent(NetLogEventType::UPLOAD_DATA_STREAM_INIT);
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]56
57 int result = InitInternal(net_log_);
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]58 if (result == ERR_IO_PENDING) {
59 DCHECK(!IsInMemory());
Matt Menkecc1d3a902018-02-05 18:27:33[diff] [blame]60 callback_ = std::move(callback);
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]61 } else {
62 OnInitCompleted(result);
63 }
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]64
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]65 return result;
[email protected]df7adc62012-09-18 14:01:53[diff] [blame]66}
67
[email protected]5b76814b2012-10-17 10:15:17[diff] [blame]68int UploadDataStream::Read(IOBuffer* buf,
69 int buf_len,
Matt Menkecc1d3a902018-02-05 18:27:33[diff] [blame]70 CompletionOnceCallback callback) {
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]71 DCHECK(!callback.is_null() || IsInMemory());
[email protected]d98961652012-09-11 20:27:21[diff] [blame]72 DCHECK(initialized_successfully_);
Yoichi Osato079c77d62025-06-13 02:27:51[diff] [blame]73 CHECK(buf);
[email protected]5b76814b2012-10-17 10:15:17[diff] [blame]74 DCHECK_GT(buf_len, 0);
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]75
mikecirone8b85c432016-09-08 19:11:00[diff] [blame]76 net_log_.BeginEvent(NetLogEventType::UPLOAD_DATA_STREAM_READ,
Eric Roman06bd9742019-07-13 15:19:13[diff] [blame]77 [&] { return CreateReadInfoParams(current_position_); });
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]78
79 int result = 0;
80 if (!is_eof_)
81 result = ReadInternal(buf, buf_len);
82
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]83 if (result == ERR_IO_PENDING) {
84 DCHECK(!IsInMemory());
Matt Menkecc1d3a902018-02-05 18:27:33[diff] [blame]85 callback_ = std::move(callback);
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]86 } else {
Yoichi Osato490169c2024-06-24 08:46:09[diff] [blame]87 if (result < ERR_IO_PENDING) {
88 LOG(ERROR) << "ReadInternal failed with Error: " << result;
89 }
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]90 OnReadCompleted(result);
91 }
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]92
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]93 return result;
[email protected]5b76814b2012-10-17 10:15:17[diff] [blame]94}
95
[email protected]135e4292012-01-24 17:45:18[diff] [blame]96bool UploadDataStream::IsEOF() const {
[email protected]d98961652012-09-11 20:27:21[diff] [blame]97 DCHECK(initialized_successfully_);
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]98 DCHECK(is_chunked_ || is_eof_ == (current_position_ == total_size_));
99 return is_eof_;
[email protected]b2d26cfd2012-12-11 10:36:06[diff] [blame]100}
101
[email protected]e5d477342012-11-02 15:18:46[diff] [blame]102void UploadDataStream::Reset() {
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]103 // If there's a pending callback, there's a pending init or read call that is
104 // being canceled.
105 if (!callback_.is_null()) {
106 if (!initialized_successfully_) {
107 // If initialization has not yet succeeded, this call is aborting
108 // initialization.
mikecirone8b85c432016-09-08 19:11:00[diff] [blame]109 net_log_.EndEventWithNetErrorCode(
110 NetLogEventType::UPLOAD_DATA_STREAM_INIT, ERR_ABORTED);
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]111 } else {
112 // Otherwise, a read is being aborted.
mikecirone8b85c432016-09-08 19:11:00[diff] [blame]113 net_log_.EndEventWithNetErrorCode(
114 NetLogEventType::UPLOAD_DATA_STREAM_READ, ERR_ABORTED);
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]115 }
116 }
117
[email protected]e5d477342012-11-02 15:18:46[diff] [blame]118 current_position_ = 0;
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]119 initialized_successfully_ = false;
120 is_eof_ = false;
[email protected]e5d477342012-11-02 15:18:46[diff] [blame]121 total_size_ = 0;
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]122 callback_.Reset();
123 ResetInternal();
[email protected]e5d477342012-11-02 15:18:46[diff] [blame]124}
[email protected]df7adc62012-09-18 14:01:53[diff] [blame]125
wtc69f8ea82015-06-04 00:08:13[diff] [blame]126void UploadDataStream::SetSize(uint64_t size) {
[email protected]e5d477342012-11-02 15:18:46[diff] [blame]127 DCHECK(!initialized_successfully_);
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]128 DCHECK(!is_chunked_);
129 total_size_ = size;
[email protected]e5d477342012-11-02 15:18:46[diff] [blame]130}
131
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]132void UploadDataStream::SetIsFinalChunk() {
133 DCHECK(initialized_successfully_);
134 DCHECK(is_chunked_);
135 DCHECK(!is_eof_);
136 is_eof_ = true;
137}
138
139bool UploadDataStream::IsInMemory() const {
140 return false;
141}
142
danakj7f767e62016-04-16 23:20:23[diff] [blame]143const std::vector<std::unique_ptr<UploadElementReader>>*
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]144UploadDataStream::GetElementReaders() const {
Raul Tambre94493c652019-03-11 17:18:35[diff] [blame]145 return nullptr;
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]146}
147
148void UploadDataStream::OnInitCompleted(int result) {
149 DCHECK_NE(ERR_IO_PENDING, result);
[email protected]e5d477342012-11-02 15:18:46[diff] [blame]150 DCHECK(!initialized_successfully_);
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]151 DCHECK_EQ(0u, current_position_);
152 DCHECK(!is_eof_);
[email protected]e5d477342012-11-02 15:18:46[diff] [blame]153
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]154 if (result == OK) {
155 initialized_successfully_ = true;
156 if (!is_chunked_ && total_size_ == 0)
157 is_eof_ = true;
[email protected]e5d477342012-11-02 15:18:46[diff] [blame]158 }
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]159
Eric Roman06bd9742019-07-13 15:19:13[diff] [blame]160 net_log_.EndEvent(NetLogEventType::UPLOAD_DATA_STREAM_INIT, [&] {
161 return NetLogInitEndInfoParams(result, total_size_, is_chunked_);
162 });
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]163
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]164 if (!callback_.is_null())
Matt Menkecc1d3a902018-02-05 18:27:33[diff] [blame]165 std::move(callback_).Run(result);
[email protected]df7adc62012-09-18 14:01:53[diff] [blame]166}
167
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]168void UploadDataStream::OnReadCompleted(int result) {
[email protected]5b76814b2012-10-17 10:15:17[diff] [blame]169 DCHECK(initialized_successfully_);
maksim.sisov52fcdc52016-06-23 06:28:52[diff] [blame]170 DCHECK(result != 0 || is_eof_);
171 DCHECK_NE(ERR_IO_PENDING, result);
[email protected]5b76814b2012-10-17 10:15:17[diff] [blame]172
maksim.sisov52fcdc52016-06-23 06:28:52[diff] [blame]173 if (result > 0) {
174 current_position_ += result;
175 if (!is_chunked_) {
176 DCHECK_LE(current_position_, total_size_);
177 if (current_position_ == total_size_)
178 is_eof_ = true;
179 }
[email protected]fac16e22012-12-29 18:46:31[diff] [blame]180 }
181
mikecirone8b85c432016-09-08 19:11:00[diff] [blame]182 net_log_.EndEventWithNetErrorCode(NetLogEventType::UPLOAD_DATA_STREAM_READ,
maksim.sisov819ba852016-08-17 08:22:52[diff] [blame]183 result);
184
mmenkecbc2b712014-10-09 20:29:07[diff] [blame]185 if (!callback_.is_null())
Matt Menkecc1d3a902018-02-05 18:27:33[diff] [blame]186 std::move(callback_).Run(result);
[email protected]fac16e22012-12-29 18:46:31[diff] [blame]187}
188
shivanishab9a143952016-09-19 17:23:41[diff] [blame]189UploadProgress UploadDataStream::GetUploadProgress() const {
190 // While initialization / rewinding is in progress, return nothing.
191 if (!initialized_successfully_)
192 return UploadProgress();
193
194 return UploadProgress(current_position_, total_size_);
195}
196
Yoichi Osato4c75c0c2020-06-24 08:03:57[diff] [blame]197bool UploadDataStream::AllowHTTP1() const {
198 return true;
199}
200
initial.commit586acc5fe2008-07-26 22:42:52[diff] [blame]201} // namespace net