| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2010 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 "base/win/scoped_bstr.h" | |
| 6 | |
| 7 #include "base/logging.h" | |
| 8 | |
| 9 namespace base { | |
| 10 namespace win { | |
| 11 | |
| 12 ScopedBstr::ScopedBstr(const char16* non_bstr) | |
| 13 : bstr_(SysAllocString(non_bstr)) { | |
| 14 } | |
| 15 | |
| 16 ScopedBstr::~ScopedBstr() { | |
| 17 COMPILE_ASSERT(sizeof(ScopedBstr) == sizeof(BSTR), ScopedBstrSize); | |
| 18 SysFreeString(bstr_); | |
| 19 } | |
| 20 | |
| 21 void ScopedBstr::Reset(BSTR bstr) { | |
| 22 if (bstr != bstr_) { | |
| 23 // if |bstr_| is NULL, SysFreeString does nothing. | |
| 24 SysFreeString(bstr_); | |
| 25 bstr_ = bstr; | |
| 26 } | |
| 27 } | |
| 28 | |
| 29 BSTR ScopedBstr::Release() { | |
| 30 BSTR bstr = bstr_; | |
| 31 bstr_ = NULL; | |
| 32 return bstr; | |
| 33 } | |
| 34 | |
| 35 void ScopedBstr::Swap(ScopedBstr& bstr2) { | |
| 36 BSTR tmp = bstr_; | |
| 37 bstr_ = bstr2.bstr_; | |
| 38 bstr2.bstr_ = tmp; | |
| 39 } | |
| 40 | |
| 41 BSTR* ScopedBstr::Receive() { | |
| 42 DCHECK(!bstr_) << "BSTR leak."; | |
| 43 return &bstr_; | |
| 44 } | |
| 45 | |
| 46 BSTR ScopedBstr::Allocate(const char16* str) { | |
| 47 Reset(SysAllocString(str)); | |
| 48 return bstr_; | |
| 49 } | |
| 50 | |
| 51 BSTR ScopedBstr::AllocateBytes(size_t bytes) { | |
| 52 Reset(SysAllocStringByteLen(NULL, static_cast<UINT>(bytes))); | |
| 53 return bstr_; | |
| 54 } | |
| 55 | |
| 56 void ScopedBstr::SetByteLen(size_t bytes) { | |
| 57 DCHECK(bstr_ != NULL) << "attempting to modify a NULL bstr"; | |
| 58 uint32* data = reinterpret_cast<uint32*>(bstr_); | |
| 59 data[-1] = static_cast<uint32>(bytes); | |
| 60 } | |
| 61 | |
| 62 size_t ScopedBstr::Length() const { | |
| 63 return SysStringLen(bstr_); | |
| 64 } | |
| 65 | |
| 66 size_t ScopedBstr::ByteLength() const { | |
| 67 return SysStringByteLen(bstr_); | |
| 68 } | |
| 69 | |
| 70 } // namespace win | |
| 71 } // namespace base | |
| OLD | NEW |