1 // Copyright 2015 The Android Open Source Project
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 #include <libwebserv/request_utils.h>
16
17 #include <base/bind.h>
18 #include <brillo/streams/memory_stream.h>
19 #include <brillo/streams/stream_utils.h>
20 #include <libwebserv/request.h>
21 #include <libwebserv/response.h>
22
23 namespace libwebserv {
24
25 namespace {
26
27 struct RequestDataContainer {
28 std::unique_ptr<Request> request;
29 std::unique_ptr<Response> response;
30 GetRequestDataSuccessCallback success_callback;
31 GetRequestDataErrorCallback error_callback;
32 std::vector<uint8_t> data;
33 };
34
OnCopySuccess(std::shared_ptr<RequestDataContainer> container,brillo::StreamPtr,brillo::StreamPtr out_stream,uint64_t)35 void OnCopySuccess(std::shared_ptr<RequestDataContainer> container,
36 brillo::StreamPtr /* in_stream */,
37 brillo::StreamPtr out_stream,
38 uint64_t /* size_copied */) {
39 // Close/release the memory stream so we can work with underlying data buffer.
40 out_stream->CloseBlocking(nullptr);
41 out_stream.reset();
42 container->success_callback.Run(std::move(container->request),
43 std::move(container->response),
44 std::move(container->data));
45 }
46
OnCopyError(std::shared_ptr<RequestDataContainer> container,brillo::StreamPtr,brillo::StreamPtr,const brillo::Error * error)47 void OnCopyError(std::shared_ptr<RequestDataContainer> container,
48 brillo::StreamPtr /* in_stream */,
49 brillo::StreamPtr /* out_stream */,
50 const brillo::Error* error) {
51 container->error_callback.Run(std::move(container->request),
52 std::move(container->response), error);
53 }
54
55 } // anonymous namespace
56
GetRequestData(std::unique_ptr<Request> request,std::unique_ptr<Response> response,const GetRequestDataSuccessCallback & success_callback,const GetRequestDataErrorCallback & error_callback)57 void GetRequestData(std::unique_ptr<Request> request,
58 std::unique_ptr<Response> response,
59 const GetRequestDataSuccessCallback& success_callback,
60 const GetRequestDataErrorCallback& error_callback) {
61 auto container = std::make_shared<RequestDataContainer>();
62 auto in_stream = request->GetDataStream();
63 auto out_stream =
64 brillo::MemoryStream::CreateRef(&container->data, nullptr);
65 container->request = std::move(request);
66 container->response = std::move(response);
67 container->success_callback = success_callback;
68 container->error_callback = error_callback;
69 brillo::stream_utils::CopyData(std::move(in_stream), std::move(out_stream),
70 base::Bind(&OnCopySuccess, container),
71 base::Bind(&OnCopyError, container));
72 }
73
74 } // namespace libwebserv
75