headless/public/util/in_memory_request_job.cc

// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "headless/public/util/in_memory_request_job.h"

#include "base/threading/thread_task_runner_handle.h"
#include "headless/public/util/in_memory_protocol_handler.h"
#include "net/base/io_buffer.h"
#include "net/base/net_errors.h"
#include "net/url_request/url_request_status.h"

namespace headless {

InMemoryRequestJob::InMemoryRequestJob(
    net::URLRequest* request,
    net::NetworkDelegate* network_delegate,
    const InMemoryProtocolHandler::Response* response)
    : net::URLRequestJob(request, network_delegate),
      response_(response),
      data_offset_(0),
      weak_factory_(this) {}

InMemoryRequestJob::~InMemoryRequestJob() {}

void InMemoryRequestJob::Start() {
  // Start reading asynchronously so that all error reporting and data
  // callbacks happen as they would for network requests.
  base::ThreadTaskRunnerHandle::Get()->PostTask(
      FROM_HERE,
      base::Bind(&InMemoryRequestJob::StartAsync, weak_factory_.GetWeakPtr()));
}

void InMemoryRequestJob::StartAsync() {
  if (!response_) {
    NotifyStartError(net::URLRequestStatus(net::URLRequestStatus::FAILED,
                                           net::ERR_FILE_NOT_FOUND));
    return;
  }
  NotifyHeadersComplete();
}

void InMemoryRequestJob::Kill() {
  weak_factory_.InvalidateWeakPtrs();
  URLRequestJob::Kill();
}

int InMemoryRequestJob::ReadRawData(net::IOBuffer* buf, int buf_size) {
  DCHECK(response_);
  DCHECK_LE(data_offset_, response_->data.size());
  int remaining =
      base::checked_cast<int>(response_->data.size() - data_offset_);
  if (buf_size > remaining)
    buf_size = remaining;
  memcpy(buf->data(), response_->data.data() + data_offset_, buf_size);
  data_offset_ += buf_size;
  return buf_size;
}

bool InMemoryRequestJob::GetMimeType(std::string* mime_type) const {
  *mime_type = response_->mime_type;
  return true;
}

}  // namespace headless