Client Scheme Handler

// Copyright (c) 2009 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 "include/cef_wrapper.h" #include "resource_util.h" #include "scheme_test.h" #include "string_util.h" #include "util.h" #ifdef _WIN32 #include "resource.h" #endif // _WIN32 // Implementation of the schema handler for client:// requests. class ClientSchemeHandler : public CefThreadSafeBase { public: ClientSchemeHandler() : offset_(0) {} // Process the request. All response generation should take place in this // method. If there is no response set |response_length| to zero and // ReadResponse() will not be called. If the response length is not known then // set |response_length| to -1 and ReadResponse() will be called until it // returns false or until the value of |bytes_read| is set to 0. Otherwise, // set |response_length| to a positive value and ReadResponse() will be called // until it returns false, the value of |bytes_read| is set to 0 or the // specified number of bytes have been read. If there is a response set // |mime_type| to the mime type for the response. virtual bool ProcessRequest(CefRefPtr request, CefString& mime_type, int* response_length) { REQUIRE_IO_THREAD(); bool handled = false; Lock(); 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
      std::string dump;
      DumpRequestContents(request, dump);
      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(), "client.png") != NULL) { // Load the response image #if defined(_WIN32) DWORD dwSize; LPBYTE pBytes; if(LoadBinaryResource(IDS_LOGO, dwSize, pBytes)) { data_ = std::string(reinterpret_cast(pBytes), dwSize); handled = true; mime_type = "image/png"; } #elif defined(__APPLE__) if(LoadBinaryResource("logo.png", data_)) { handled = true; mime_type = "image/png"; } #endif } // Set the resulting response length *response_length = data_.length(); Unlock(); return handled; } // Cancel processing of the request. virtual void Cancel() { REQUIRE_IO_THREAD(); } // Copy up to |bytes_to_read| bytes into |data_out|. If the copy succeeds // set |bytes_read| to the number of bytes copied and return true. If the // copy fails return false and ReadResponse() will not be called again. virtual bool ReadResponse(void* data_out, int bytes_to_read, int* bytes_read) { REQUIRE_IO_THREAD(); bool has_data = false; *bytes_read = 0; Lock(); 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; } Unlock(); return has_data; } private: std::string data_; size_t offset_; }; // Implementation of the factory for for creating schema handlers. class ClientSchemeHandlerFactory : public CefThreadSafeBase { public: // Return a new scheme handler instance to handle the request. virtual CefRefPtr Create() { REQUIRE_IO_THREAD(); return new ClientSchemeHandler(); } }; void InitSchemeTest() { CefRegisterScheme("client", "tests", true, new ClientSchemeHandlerFactory()); } void RunSchemeTest(CefRefPtr browser) { browser->GetMainFrame()->LoadURL("client://tests/handler.html"); }