libwebserv/request_utils.cc

// Copyright 2015 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include <libwebserv/request_utils.h>

#include <base/bind.h>
#include <brillo/streams/memory_stream.h>
#include <brillo/streams/stream_utils.h>
#include <libwebserv/request.h>
#include <libwebserv/response.h>

namespace libwebserv {

namespace {

struct RequestDataContainer {
  std::unique_ptr<Request> request;
  std::unique_ptr<Response> response;
  GetRequestDataSuccessCallback success_callback;
  GetRequestDataErrorCallback error_callback;
  std::vector<uint8_t> data;
};

void OnCopySuccess(std::shared_ptr<RequestDataContainer> container,
                   brillo::StreamPtr /* in_stream */,
                   brillo::StreamPtr out_stream,
                   uint64_t /* size_copied */) {
  // Close/release the memory stream so we can work with underlying data buffer.
  out_stream->CloseBlocking(nullptr);
  out_stream.reset();
  container->success_callback.Run(std::move(container->request),
                                  std::move(container->response),
                                  std::move(container->data));
}

void OnCopyError(std::shared_ptr<RequestDataContainer> container,
                 brillo::StreamPtr /* in_stream */,
                 brillo::StreamPtr /* out_stream */,
                 const brillo::Error* error) {
  container->error_callback.Run(std::move(container->request),
                                std::move(container->response), error);
}

}  // anonymous namespace

void GetRequestData(std::unique_ptr<Request> request,
                    std::unique_ptr<Response> response,
                    const GetRequestDataSuccessCallback& success_callback,
                    const GetRequestDataErrorCallback& error_callback) {
  auto container = std::make_shared<RequestDataContainer>();
  auto in_stream = request->GetDataStream();
  auto out_stream =
      brillo::MemoryStream::CreateRef(&container->data, nullptr);
  container->request = std::move(request);
  container->response = std::move(response);
  container->success_callback = success_callback;
  container->error_callback = error_callback;
  brillo::stream_utils::CopyData(std::move(in_stream), std::move(out_stream),
                                 base::Bind(&OnCopySuccess, container),
                                 base::Bind(&OnCopyError, container));
}

}  // namespace libwebserv