Index: chrome/browser/renderer_host/x509_user_cert_resource_handler.cc |
=================================================================== |
--- chrome/browser/renderer_host/x509_user_cert_resource_handler.cc (revision 0) |
+++ chrome/browser/renderer_host/x509_user_cert_resource_handler.cc (revision 0) |
@@ -0,0 +1,101 @@ |
+// Copyright (c) 2009 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 "chrome/browser/renderer_host/x509_user_cert_resource_handler.h" |
+ |
+#include "base/string_util.h" |
+#include "chrome/browser/browser_process.h" |
+#include "chrome/browser/chrome_thread.h" |
+#include "chrome/browser/download/download_file.h" |
+#include "chrome/browser/renderer_host/resource_dispatcher_host.h" |
+#include "chrome/browser/renderer_host/resource_dispatcher_host_request_info.h" |
+#include "chrome/common/url_constants.h" |
+#include "net/base/cert_database.h" |
+#include "net/base/io_buffer.h" |
+#include "net/base/mime_sniffer.h" |
+#include "net/base/mime_util.h" |
+#include "net/http/http_response_headers.h" |
+ |
+X509UserCertResourceHandler::X509UserCertResourceHandler( |
+ ResourceDispatcherHost *host, URLRequest* request) |
+ : host_(host), |
+ request_(request), |
+ content_length_(0), |
+ buffer_(new DownloadBuffer), |
+ read_buffer_(NULL), |
+ resource_buffer_(NULL) { |
+} |
+ |
+bool X509UserCertResourceHandler::OnRequestRedirected(int request_id, |
+ const GURL& url, |
+ ResourceResponse* resp, |
+ bool *defer) { |
+ url_ = url; |
+ return true; |
+} |
+ |
+bool X509UserCertResourceHandler::OnResponseStarted(int request_id, |
+ ResourceResponse* resp) { |
+ return (resp->response_head.mime_type == "application/x-x509-user-cert"); |
+ |
+} |
+ |
+bool X509UserCertResourceHandler::OnWillRead(int request_id, |
+ net::IOBuffer** buf, |
+ int* buf_size, |
+ int min_size) { |
+ // FIXME(gauravsh): Should we use 'min_size' here? |
+ DCHECK(buf && buf_size); |
+ if (!read_buffer_) { |
+ read_buffer_ = new net::IOBuffer(kReadBufSize); |
+ } |
+ *buf = read_buffer_.get(); |
+ *buf_size = kReadBufSize; |
+ |
+ return true; |
+} |
+ |
+bool X509UserCertResourceHandler::OnReadCompleted(int request_id, |
+ int* bytes_read) { |
+ if (!*bytes_read) |
+ return true; |
+ |
+ // We have more data to read. |
+ DCHECK(read_buffer_); |
+ content_length_ += *bytes_read; |
+ |
+ // Release the ownership of the buffer, and store a reference |
+ // to it. A new one will be allocated in OnWillRead(). |
+ net::IOBuffer* buffer = NULL; |
+ read_buffer_.swap(&buffer); |
+ // TODO(gauravsh): Should this be handled by a separate thread? |
+ buffer_->contents.push_back(std::make_pair(buffer, *bytes_read)); |
+ |
+ return true; |
+} |
+ |
+bool X509UserCertResourceHandler::OnResponseCompleted( |
+ int request_id, |
+ const URLRequestStatus& urs, |
+ const std::string& sec_info) { |
+ // TODO(gauravsh): Verify that 'request_id' was actually a keygen form post |
+ // and only then import the certificate |
+ scoped_ptr<net::CertDatabase> cert_db(new net::CertDatabase()); |
+ AssembleResource(); |
+ |
+ return cert_db->AddUserCert(resource_buffer_->data(), content_length_); |
+} |
+ |
+void X509UserCertResourceHandler::AssembleResource() { |
+ size_t bytes_copied = 0; |
+ resource_buffer_ = new net::IOBuffer(content_length_); |
+ |
+ for (size_t i = 0; i < buffer_->contents.size(); ++i) { |
+ net::IOBuffer* data = buffer_->contents[i].first; |
+ const int data_len = buffer_->contents[i].second; |
+ DCHECK(bytes_copied + data_len <= content_length_); |
+ memcpy(resource_buffer_->data() + bytes_copied, data->data(), data_len); |
+ bytes_copied += data_len; |
+ } |
+} |
Property changes on: chrome/browser/renderer_host/x509_user_cert_resource_handler.cc |
___________________________________________________________________ |
Name: svn:eol-style |
+ LF |