// Copyright (c) 2012 The Chromium Embedded Framework 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 "tests/cefclient/browser/scheme_test.h" #include #include #include "include/cef_browser.h" #include "include/cef_callback.h" #include "include/cef_frame.h" #include "include/cef_request.h" #include "include/cef_resource_handler.h" #include "include/cef_response.h" #include "include/cef_scheme.h" #include "include/wrapper/cef_helpers.h" #include "tests/cefclient/browser/test_runner.h" #include "tests/shared/browser/resource_util.h" namespace client { namespace scheme_test { namespace { // Implementation of the schema handler for client:// requests. class ClientSchemeHandler : public CefResourceHandler { public: ClientSchemeHandler() : offset_(0) {} virtual bool ProcessRequest(CefRefPtr request, CefRefPtr callback) OVERRIDE { CEF_REQUIRE_IO_THREAD(); bool handled = false; std::string url = request->GetURL(); if (strstr(url.c_str(), "handler.html") != NULL) { // Build the response html data_ = "Client Scheme Handler" "" "This contents of this page page are served by the " "ClientSchemeHandler class handling the client:// protocol." "
You should see an image:" "
";

      // Output a string representation of the request
      const std::string& dump = test_runner::DumpRequestContents(request);
      data_.append(dump);

      data_.append(
          "

Try the test form:" "
" "" "" "" "
"); handled = true; // Set the resulting mime type mime_type_ = "text/html"; } else if (strstr(url.c_str(), "logo.png") != NULL) { // Load the response image if (LoadBinaryResource("logo.png", data_)) { handled = true; // Set the resulting mime type mime_type_ = "image/png"; } } if (handled) { // Indicate the headers are available. callback->Continue(); return true; } return false; } virtual void GetResponseHeaders(CefRefPtr response, int64& response_length, CefString& redirectUrl) OVERRIDE { CEF_REQUIRE_IO_THREAD(); DCHECK(!data_.empty()); response->SetMimeType(mime_type_); response->SetStatus(200); // Set the resulting response length response_length = data_.length(); } virtual void Cancel() OVERRIDE { CEF_REQUIRE_IO_THREAD(); } virtual bool ReadResponse(void* data_out, int bytes_to_read, int& bytes_read, CefRefPtr callback) OVERRIDE { CEF_REQUIRE_IO_THREAD(); bool has_data = false; bytes_read = 0; if (offset_ < data_.length()) { // Copy the next block of data into the buffer. int transfer_size = std::min(bytes_to_read, static_cast(data_.length() - offset_)); memcpy(data_out, data_.c_str() + offset_, transfer_size); offset_ += transfer_size; bytes_read = transfer_size; has_data = true; } return has_data; } private: std::string data_; std::string mime_type_; size_t offset_; IMPLEMENT_REFCOUNTING(ClientSchemeHandler); }; // Implementation of the factory for for creating schema handlers. class ClientSchemeHandlerFactory : public CefSchemeHandlerFactory { public: // Return a new scheme handler instance to handle the request. virtual CefRefPtr Create( CefRefPtr browser, CefRefPtr frame, const CefString& scheme_name, CefRefPtr request) OVERRIDE { CEF_REQUIRE_IO_THREAD(); return new ClientSchemeHandler(); } IMPLEMENT_REFCOUNTING(ClientSchemeHandlerFactory); }; } // namespace void RegisterSchemeHandlers() { CefRegisterSchemeHandlerFactory("client", "tests", new ClientSchemeHandlerFactory()); } } // namespace scheme_test } // namespace client