1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "chrome_frame/bind_status_callback_impl.h" 6 7 #include "base/logging.h" 8 #include "base/strings/string_util.h" 9 #include "base/strings/stringprintf.h" 10 #include "base/threading/platform_thread.h" 11 12 BSCBImpl::BSCBImpl() { 13 DVLOG(1) << __FUNCTION__ << me(); 14 } 15 16 BSCBImpl::~BSCBImpl() { 17 DVLOG(1) << __FUNCTION__ << me(); 18 } 19 20 std::string BSCBImpl::me() { 21 return base::StringPrintf(" obj=0x%08X", static_cast<BSCBImpl*>(this)); 22 } 23 24 HRESULT BSCBImpl::DelegateQI(void* obj, REFIID iid, void** ret, DWORD cookie) { 25 BSCBImpl* me = reinterpret_cast<BSCBImpl*>(obj); 26 HRESULT hr = E_NOINTERFACE; 27 if (me->delegate_) 28 hr = me->delegate_.QueryInterface(iid, ret); 29 return hr; 30 } 31 32 void BSCBImpl::Initialize(IBindStatusCallback* original) { 33 DCHECK(!delegate_); 34 delegate_ = original; 35 } 36 37 HRESULT BSCBImpl::AttachToBind(IBindCtx* bind_ctx) { 38 HRESULT hr = S_OK; 39 hr = ::RegisterBindStatusCallback(bind_ctx, this, delegate_.Receive(), 0); 40 if (SUCCEEDED(hr)) { 41 bind_ctx_ = bind_ctx; 42 } 43 44 return hr; 45 } 46 47 HRESULT BSCBImpl::ReleaseBind() { 48 // AddRef ourselves while we release these objects as we might 49 // perish during this operation. 50 AddRef(); 51 52 HRESULT hr = S_OK; 53 if (bind_ctx_) { 54 hr = ::RevokeBindStatusCallback(bind_ctx_, this); 55 } 56 delegate_.Release(); 57 bind_ctx_.Release(); 58 59 Release(); 60 61 return hr; 62 } 63 64 // IServiceProvider 65 HRESULT BSCBImpl::QueryService(REFGUID service, REFIID iid, void** object) { 66 HRESULT hr = E_NOINTERFACE; 67 if (delegate_) { 68 base::win::ScopedComPtr<IServiceProvider> svc; 69 svc.QueryFrom(delegate_); 70 if (svc) { 71 hr = svc->QueryService(service, iid, object); 72 } 73 } 74 return hr; 75 } 76 77 // IBindStatusCallback 78 HRESULT BSCBImpl::OnStartBinding(DWORD reserved, IBinding* binding) { 79 DVLOG(1) << __FUNCTION__ << me() 80 << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); 81 HRESULT hr = S_OK; 82 if (delegate_) 83 hr = delegate_->OnStartBinding(reserved, binding); 84 return hr; 85 } 86 87 HRESULT BSCBImpl::GetPriority(LONG* priority) { 88 DVLOG(1) << __FUNCTION__ << me() 89 << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); 90 HRESULT hr = S_OK; 91 if (delegate_) 92 hr = delegate_->GetPriority(priority); 93 return hr; 94 } 95 96 HRESULT BSCBImpl::OnLowResource(DWORD reserved) { 97 DVLOG(1) << __FUNCTION__ << me() 98 << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); 99 HRESULT hr = S_OK; 100 if (delegate_) 101 hr = delegate_->OnLowResource(reserved); 102 return hr; 103 } 104 105 HRESULT BSCBImpl::OnProgress(ULONG progress, ULONG progress_max, 106 ULONG status_code, LPCWSTR status_text) { 107 DVLOG(1) << __FUNCTION__ << me() 108 << base::StringPrintf(" status=%i tid=%i %ls", status_code, 109 base::PlatformThread::CurrentId(), 110 status_text); 111 HRESULT hr = S_OK; 112 if (delegate_) 113 delegate_->OnProgress(progress, progress_max, status_code, status_text); 114 return hr; 115 } 116 117 HRESULT BSCBImpl::OnStopBinding(HRESULT hresult, LPCWSTR error) { 118 DVLOG(1) << __FUNCTION__ << me() 119 << base::StringPrintf(" hr=0x%08X '%ls' tid=%i", hresult, error, 120 base::PlatformThread::CurrentId()); 121 HRESULT hr = S_OK; 122 if (delegate_) 123 delegate_->OnStopBinding(hresult, error); 124 return hr; 125 } 126 127 HRESULT BSCBImpl::GetBindInfo(DWORD* bindf, BINDINFO* bind_info) { 128 DVLOG(1) << __FUNCTION__ << me() 129 << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); 130 HRESULT hr = S_OK; 131 if (delegate_) 132 delegate_->GetBindInfo(bindf, bind_info); 133 return hr; 134 } 135 136 HRESULT BSCBImpl::OnDataAvailable(DWORD bscf, DWORD size, 137 FORMATETC* format_etc, STGMEDIUM* stgmed) { 138 DVLOG(1) << __FUNCTION__ << me() 139 << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); 140 HRESULT hr = S_OK; 141 if (delegate_) 142 hr = delegate_->OnDataAvailable(bscf, size, format_etc, stgmed); 143 return hr; 144 } 145 146 HRESULT BSCBImpl::OnObjectAvailable(REFIID iid, IUnknown* unk) { 147 DVLOG(1) << __FUNCTION__ << me() 148 << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); 149 HRESULT hr = S_OK; 150 if (delegate_) 151 delegate_->OnObjectAvailable(iid, unk); 152 return hr; 153 } 154 155 // IBindStatusCallbackEx 156 HRESULT BSCBImpl::GetBindInfoEx(DWORD* bindf, BINDINFO* bind_info, 157 DWORD* bindf2, DWORD* reserved) { 158 DVLOG(1) << __FUNCTION__ << me() 159 << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); 160 HRESULT hr = S_OK; 161 if (delegate_) { 162 base::win::ScopedComPtr<IBindStatusCallbackEx> bscbex; 163 bscbex.QueryFrom(delegate_); 164 if (bscbex) 165 hr = bscbex->GetBindInfoEx(bindf, bind_info, bindf2, reserved); 166 } 167 return hr; 168 } 169 170 HRESULT BSCBImpl::BeginningTransaction(LPCWSTR url, LPCWSTR headers, 171 DWORD reserved, 172 LPWSTR* additional_headers) { 173 DVLOG(1) << __FUNCTION__ << me() 174 << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); 175 176 HRESULT hr = S_OK; 177 if (delegate_) { 178 base::win::ScopedComPtr<IHttpNegotiate> http_negotiate; 179 http_negotiate.QueryFrom(delegate_); 180 if (http_negotiate) { 181 hr = http_negotiate->BeginningTransaction(url, headers, reserved, 182 additional_headers); 183 } 184 } 185 186 DLOG_IF(ERROR, FAILED(hr)) << __FUNCTION__; 187 return hr; 188 } 189 190 HRESULT BSCBImpl::OnResponse(DWORD response_code, LPCWSTR response_headers, 191 LPCWSTR request_headers, 192 LPWSTR* additional_headers) { 193 DVLOG(1) << __FUNCTION__ << me() 194 << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); 195 196 HRESULT hr = S_OK; 197 if (delegate_) { 198 base::win::ScopedComPtr<IHttpNegotiate> http_negotiate; 199 http_negotiate.QueryFrom(delegate_); 200 if (http_negotiate) { 201 hr = http_negotiate->OnResponse(response_code, response_headers, 202 request_headers, additional_headers); 203 } 204 } 205 return hr; 206 } 207 208 HRESULT BSCBImpl::GetRootSecurityId(BYTE* security_id, DWORD* security_id_size, 209 DWORD_PTR reserved) { 210 HRESULT hr = S_OK; 211 if (delegate_) { 212 base::win::ScopedComPtr<IHttpNegotiate2> http_negotiate; 213 http_negotiate.QueryFrom(delegate_); 214 if (http_negotiate) { 215 hr = http_negotiate->GetRootSecurityId(security_id, security_id_size, 216 reserved); 217 } 218 } 219 return hr; 220 } 221 222 HRESULT BSCBImpl::GetSerializedClientCertContext(BYTE** cert, 223 DWORD* cert_size) { 224 HRESULT hr = S_OK; 225 if (delegate_) { 226 base::win::ScopedComPtr<IHttpNegotiate3> http_negotiate; 227 http_negotiate.QueryFrom(delegate_); 228 if (http_negotiate) { 229 return http_negotiate->GetSerializedClientCertContext(cert, cert_size); 230 } 231 } 232 return hr; 233 } 234
