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
|
// 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.
#include "chrome_frame/urlmon_upload_data_stream.h"
#include "net/base/io_buffer.h"
#include "net/base/net_errors.h"
void UrlmonUploadDataStream::Initialize(net::UploadData* upload_data) {
upload_data_ = upload_data;
request_body_stream_.reset(
new net::UploadDataStream(upload_data));
const int result = request_body_stream_->InitSync();
DCHECK_EQ(net::OK, result);
}
STDMETHODIMP UrlmonUploadDataStream::Read(void* pv, ULONG cb, ULONG* read) {
if (pv == NULL) {
NOTREACHED();
return E_POINTER;
}
// Have we already read past the end of the stream?
if (request_body_stream_->IsEOF()) {
if (read) {
*read = 0;
}
return S_FALSE;
}
// The data in request_body_stream_ can be smaller than 'cb' so it's not
// guaranteed that we'll be able to read total_bytes_to_copy bytes.
uint64 total_bytes_to_copy = cb;
uint64 bytes_copied = 0;
char* write_pointer = reinterpret_cast<char*>(pv);
while (bytes_copied < total_bytes_to_copy) {
size_t bytes_to_copy_now = total_bytes_to_copy - bytes_copied;
scoped_refptr<net::IOBufferWithSize> buf(
new net::IOBufferWithSize(bytes_to_copy_now));
int bytes_read = request_body_stream_->ReadSync(buf, buf->size());
if (bytes_read == 0) // Reached the end of the stream.
break;
memcpy(write_pointer, buf->data(), bytes_read);
// Advance our copy tally
bytes_copied += bytes_read;
// Advance our write pointer
write_pointer += bytes_read;
}
DCHECK_LE(bytes_copied, total_bytes_to_copy);
if (read) {
*read = static_cast<ULONG>(bytes_copied);
}
return S_OK;
}
STDMETHODIMP UrlmonUploadDataStream::Seek(LARGE_INTEGER move, DWORD origin,
ULARGE_INTEGER* new_pos) {
// UploadDataStream is really not very seek-able, so for now allow
// STREAM_SEEK_SETs to work with a 0 offset, but fail on everything else.
if (origin == STREAM_SEEK_SET && move.QuadPart == 0) {
if (request_body_stream_->position() != 0) {
request_body_stream_.reset(new net::UploadDataStream(upload_data_));
const int result = request_body_stream_->InitSync();
DCHECK_EQ(net::OK, result);
}
if (new_pos) {
new_pos->QuadPart = 0;
}
return S_OK;
}
DCHECK(false) << __FUNCTION__;
return STG_E_INVALIDFUNCTION;
}
STDMETHODIMP UrlmonUploadDataStream::Stat(STATSTG *stat_stg,
DWORD grf_stat_flag) {
if (stat_stg == NULL)
return E_POINTER;
memset(stat_stg, 0, sizeof(STATSTG));
if (0 == (grf_stat_flag & STATFLAG_NONAME)) {
const wchar_t kStreamBuffer[] = L"PostStream";
stat_stg->pwcsName =
static_cast<wchar_t*>(::CoTaskMemAlloc(sizeof(kStreamBuffer)));
lstrcpy(stat_stg->pwcsName, kStreamBuffer);
}
stat_stg->type = STGTY_STREAM;
stat_stg->cbSize.QuadPart = request_body_stream_->size();
return S_OK;
}
|
