205 lines
5.2 KiB
C++
205 lines
5.2 KiB
C++
|
// 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 "libcef/common/response_impl.h"
|
||
|
|
||
|
#include <string>
|
||
|
|
||
|
#include "base/logging.h"
|
||
|
#include "base/strings/stringprintf.h"
|
||
|
#include "net/http/http_request_headers.h"
|
||
|
#include "net/http/http_response_headers.h"
|
||
|
#include "third_party/WebKit/public/platform/WebHTTPHeaderVisitor.h"
|
||
|
#include "third_party/WebKit/public/platform/WebString.h"
|
||
|
#include "third_party/WebKit/public/platform/WebURLResponse.h"
|
||
|
|
||
|
|
||
|
#define CHECK_READONLY_RETURN_VOID() \
|
||
|
if (read_only_) { \
|
||
|
NOTREACHED() << "object is read only"; \
|
||
|
return; \
|
||
|
}
|
||
|
|
||
|
|
||
|
// CefResponse ----------------------------------------------------------------
|
||
|
|
||
|
// static
|
||
|
CefRefPtr<CefResponse> CefResponse::Create() {
|
||
|
CefRefPtr<CefResponse> response(new CefResponseImpl());
|
||
|
return response;
|
||
|
}
|
||
|
|
||
|
|
||
|
// CefResponseImpl ------------------------------------------------------------
|
||
|
|
||
|
CefResponseImpl::CefResponseImpl()
|
||
|
: status_code_(0),
|
||
|
read_only_(false) {
|
||
|
}
|
||
|
|
||
|
bool CefResponseImpl::IsReadOnly() {
|
||
|
AutoLock lock_scope(this);
|
||
|
return read_only_;
|
||
|
}
|
||
|
|
||
|
int CefResponseImpl::GetStatus() {
|
||
|
AutoLock lock_scope(this);
|
||
|
return status_code_;
|
||
|
}
|
||
|
|
||
|
void CefResponseImpl::SetStatus(int status) {
|
||
|
AutoLock lock_scope(this);
|
||
|
CHECK_READONLY_RETURN_VOID();
|
||
|
status_code_ = status;
|
||
|
}
|
||
|
|
||
|
CefString CefResponseImpl::GetStatusText() {
|
||
|
AutoLock lock_scope(this);
|
||
|
return status_text_;
|
||
|
}
|
||
|
|
||
|
void CefResponseImpl::SetStatusText(const CefString& statusText) {
|
||
|
AutoLock lock_scope(this);
|
||
|
CHECK_READONLY_RETURN_VOID();
|
||
|
status_text_ = statusText;
|
||
|
}
|
||
|
|
||
|
CefString CefResponseImpl::GetMimeType() {
|
||
|
AutoLock lock_scope(this);
|
||
|
return mime_type_;
|
||
|
}
|
||
|
|
||
|
void CefResponseImpl::SetMimeType(const CefString& mimeType) {
|
||
|
AutoLock lock_scope(this);
|
||
|
CHECK_READONLY_RETURN_VOID();
|
||
|
mime_type_ = mimeType;
|
||
|
}
|
||
|
|
||
|
CefString CefResponseImpl::GetHeader(const CefString& name) {
|
||
|
AutoLock lock_scope(this);
|
||
|
|
||
|
CefString value;
|
||
|
|
||
|
HeaderMap::const_iterator it = header_map_.find(name);
|
||
|
if (it != header_map_.end())
|
||
|
value = it->second;
|
||
|
|
||
|
return value;
|
||
|
}
|
||
|
|
||
|
void CefResponseImpl::GetHeaderMap(HeaderMap& map) {
|
||
|
AutoLock lock_scope(this);
|
||
|
map = header_map_;
|
||
|
}
|
||
|
|
||
|
void CefResponseImpl::SetHeaderMap(const HeaderMap& headerMap) {
|
||
|
AutoLock lock_scope(this);
|
||
|
CHECK_READONLY_RETURN_VOID();
|
||
|
header_map_ = headerMap;
|
||
|
}
|
||
|
|
||
|
net::HttpResponseHeaders* CefResponseImpl::GetResponseHeaders() {
|
||
|
AutoLock lock_scope(this);
|
||
|
|
||
|
std::string response;
|
||
|
std::string status_text;
|
||
|
bool has_content_type_header = false;
|
||
|
|
||
|
if (!status_text_.empty())
|
||
|
status_text = status_text_;
|
||
|
else
|
||
|
status_text = (status_code_ == 200)?"OK":"ERROR";
|
||
|
|
||
|
base::SStringPrintf(&response, "HTTP/1.1 %d %s", status_code_,
|
||
|
status_text.c_str());
|
||
|
if (header_map_.size() > 0) {
|
||
|
for (HeaderMap::const_iterator header = header_map_.begin();
|
||
|
header != header_map_.end();
|
||
|
++header) {
|
||
|
const CefString& key = header->first;
|
||
|
const CefString& value = header->second;
|
||
|
|
||
|
if (!key.empty()) {
|
||
|
// Delimit with "\0" as required by net::HttpResponseHeaders.
|
||
|
std::string key_str(key);
|
||
|
std::string value_str(value);
|
||
|
base::StringAppendF(&response, "%c%s: %s", '\0', key_str.c_str(),
|
||
|
value_str.c_str());
|
||
|
|
||
|
if (!has_content_type_header &&
|
||
|
key_str == net::HttpRequestHeaders::kContentType) {
|
||
|
has_content_type_header = true;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!has_content_type_header) {
|
||
|
std::string mime_type;
|
||
|
if (!mime_type_.empty())
|
||
|
mime_type = mime_type_;
|
||
|
else
|
||
|
mime_type = "text/html";
|
||
|
|
||
|
base::StringAppendF(&response, "%c%s: %s", '\0',
|
||
|
net::HttpRequestHeaders::kContentType, mime_type.c_str());
|
||
|
}
|
||
|
|
||
|
return new net::HttpResponseHeaders(response);
|
||
|
}
|
||
|
|
||
|
void CefResponseImpl::SetResponseHeaders(
|
||
|
const net::HttpResponseHeaders& headers) {
|
||
|
AutoLock lock_scope(this);
|
||
|
|
||
|
header_map_.empty();
|
||
|
|
||
|
void* iter = NULL;
|
||
|
std::string name, value;
|
||
|
while (headers.EnumerateHeaderLines(&iter, &name, &value))
|
||
|
header_map_.insert(std::make_pair(name, value));
|
||
|
|
||
|
status_code_ = headers.response_code();
|
||
|
status_text_ = headers.GetStatusText();
|
||
|
|
||
|
std::string mime_type;
|
||
|
if (headers.GetMimeType(&mime_type))
|
||
|
mime_type_ = mime_type;
|
||
|
}
|
||
|
|
||
|
void CefResponseImpl::Set(const WebKit::WebURLResponse& response) {
|
||
|
DCHECK(!response.isNull());
|
||
|
|
||
|
AutoLock lock_scope(this);
|
||
|
CHECK_READONLY_RETURN_VOID();
|
||
|
|
||
|
WebKit::WebString str;
|
||
|
status_code_ = response.httpStatusCode();
|
||
|
str = response.httpStatusText();
|
||
|
status_text_ = CefString(str);
|
||
|
str = response.mimeType();
|
||
|
mime_type_ = CefString(str);
|
||
|
|
||
|
class HeaderVisitor : public WebKit::WebHTTPHeaderVisitor {
|
||
|
public:
|
||
|
explicit HeaderVisitor(HeaderMap* map) : map_(map) {}
|
||
|
|
||
|
virtual void visitHeader(const WebKit::WebString& name,
|
||
|
const WebKit::WebString& value) {
|
||
|
map_->insert(std::make_pair(string16(name), string16(value)));
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
HeaderMap* map_;
|
||
|
};
|
||
|
|
||
|
HeaderVisitor visitor(&header_map_);
|
||
|
response.visitHTTPHeaderFields(&visitor);
|
||
|
}
|
||
|
|
||
|
void CefResponseImpl::SetReadOnly(bool read_only) {
|
||
|
AutoLock lock_scope(this);
|
||
|
read_only_ = read_only;
|
||
|
}
|