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 |