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
|
// Copyright (c) 2006-2008 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.
#include "net/base/upload_data_stream.h"
#include "base/logging.h"
namespace net {
UploadDataStream::UploadDataStream(const UploadData* data)
: data_(data),
next_element_handle_(INVALID_HANDLE_VALUE),
total_size_(data->GetContentLength()) {
Reset();
FillBuf();
}
UploadDataStream::~UploadDataStream() {
if (next_element_handle_ != INVALID_HANDLE_VALUE)
CloseHandle(next_element_handle_);
}
void UploadDataStream::DidConsume(size_t num_bytes) {
DCHECK(num_bytes <= buf_len_);
buf_len_ -= num_bytes;
if (buf_len_)
memmove(buf_, buf_ + num_bytes, buf_len_);
FillBuf();
current_position_ += num_bytes;
}
void UploadDataStream::Reset() {
if (next_element_handle_ != INVALID_HANDLE_VALUE) {
CloseHandle(next_element_handle_);
next_element_handle_ = INVALID_HANDLE_VALUE;
}
buf_len_ = 0;
next_element_ = data_->elements().begin();
next_element_offset_ = 0;
next_element_remaining_ = 0;
current_position_ = 0;
}
void UploadDataStream::FillBuf() {
std::vector<UploadData::Element>::const_iterator end =
data_->elements().end();
while (buf_len_ < kBufSize && next_element_ != end) {
bool advance_to_next_element = false;
size_t size_remaining = kBufSize - buf_len_;
if ((*next_element_).type() == UploadData::TYPE_BYTES) {
const std::vector<char>& d = (*next_element_).bytes();
size_t count = d.size() - next_element_offset_;
size_t bytes_copied = std::min(count, size_remaining);
memcpy(buf_ + buf_len_, &d[next_element_offset_], bytes_copied);
buf_len_ += bytes_copied;
if (bytes_copied == count) {
advance_to_next_element = true;
} else {
next_element_offset_ += bytes_copied;
}
} else {
DCHECK((*next_element_).type() == UploadData::TYPE_FILE);
if (next_element_handle_ == INVALID_HANDLE_VALUE) {
next_element_handle_ = CreateFile((*next_element_).file_path().c_str(),
GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL, OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL, NULL);
// If the file does not exist, that's technically okay.. we'll just
// upload an empty file. This is for consistency with Mozilla.
DLOG_IF(WARNING, next_element_handle_ == INVALID_HANDLE_VALUE) <<
"Unable to open file \"" << (*next_element_).file_path() <<
"\" for reading: " << GetLastError();
next_element_remaining_ = (*next_element_).file_range_length();
if ((*next_element_).file_range_offset()) {
LARGE_INTEGER offset;
offset.QuadPart = (*next_element_).file_range_offset();
if (!SetFilePointerEx(next_element_handle_, offset,
NULL, FILE_BEGIN)) {
DLOG(WARNING) <<
"Unable to set file position for file \"" <<
(*next_element_).file_path() << "\": " << GetLastError();
next_element_remaining_ = 0;
}
}
}
// ReadFile will happily fail if given an invalid handle.
BOOL ok = FALSE;
DWORD bytes_read = 0;
uint64 amount_to_read = std::min(static_cast<uint64>(size_remaining),
next_element_remaining_);
if ((amount_to_read > 0) &&
(ok = ReadFile(next_element_handle_, buf_ + buf_len_,
static_cast<DWORD>(amount_to_read), &bytes_read,
NULL))) {
buf_len_ += bytes_read;
next_element_remaining_ -= bytes_read;
}
if (!ok || bytes_read == 0)
advance_to_next_element = true;
}
if (advance_to_next_element) {
++next_element_;
next_element_offset_ = 0;
if (next_element_handle_ != INVALID_HANDLE_VALUE) {
CloseHandle(next_element_handle_);
next_element_handle_ = INVALID_HANDLE_VALUE;
}
}
}
}
} // namespace net
|
