1 // Copyright 2017 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/winrt_storage_util.h"
6
7 #include <robuffer.h>
8 #include <string.h>
9 #include <wrl/client.h>
10
11 #include "base/strings/string_util.h"
12 #include "base/win/core_winrt_util.h"
13 #include "base/win/scoped_hstring.h"
14
15 namespace base {
16 namespace win {
17
18 using IBuffer = ABI::Windows::Storage::Streams::IBuffer;
19
GetPointerToBufferData(IBuffer * buffer,uint8_t ** out,UINT32 * length)20 HRESULT GetPointerToBufferData(IBuffer* buffer, uint8_t** out, UINT32* length) {
21 *out = nullptr;
22
23 Microsoft::WRL::ComPtr<Windows::Storage::Streams::IBufferByteAccess>
24 buffer_byte_access;
25 HRESULT hr = buffer->QueryInterface(IID_PPV_ARGS(&buffer_byte_access));
26 if (FAILED(hr))
27 return hr;
28
29 hr = buffer->get_Length(length);
30 if (FAILED(hr))
31 return hr;
32
33 // Lifetime of the pointing buffer is controlled by the buffer object.
34 return buffer_byte_access->Buffer(out);
35 }
36
CreateIBufferFromData(const uint8_t * data,UINT32 length,Microsoft::WRL::ComPtr<IBuffer> * buffer)37 HRESULT CreateIBufferFromData(const uint8_t* data,
38 UINT32 length,
39 Microsoft::WRL::ComPtr<IBuffer>* buffer) {
40 *buffer = nullptr;
41
42 Microsoft::WRL::ComPtr<ABI::Windows::Storage::Streams::IBufferFactory>
43 buffer_factory;
44 HRESULT hr = base::win::GetActivationFactory<
45 ABI::Windows::Storage::Streams::IBufferFactory,
46 RuntimeClass_Windows_Storage_Streams_Buffer>(&buffer_factory);
47 if (FAILED(hr))
48 return hr;
49
50 Microsoft::WRL::ComPtr<IBuffer> internal_buffer;
51 hr = buffer_factory->Create(length, &internal_buffer);
52 if (FAILED(hr))
53 return hr;
54
55 hr = internal_buffer->put_Length(length);
56 if (FAILED(hr))
57 return hr;
58
59 uint8_t* p_buffer_data;
60 hr = GetPointerToBufferData(internal_buffer.Get(), &p_buffer_data, &length);
61 if (FAILED(hr))
62 return hr;
63
64 memcpy(p_buffer_data, data, length);
65
66 *buffer = std::move(internal_buffer);
67
68 return S_OK;
69 }
70
71 } // namespace win
72 } // namespace base
73