OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "chrome/browser/renderer_host/x509_user_cert_resource_handler.h" |
| 6 |
| 7 #include "base/string_util.h" |
| 8 #include "chrome/browser/browser_process.h" |
| 9 #include "chrome/browser/chrome_thread.h" |
| 10 #include "chrome/browser/download/download_file.h" |
| 11 #include "chrome/browser/renderer_host/resource_dispatcher_host.h" |
| 12 #include "chrome/browser/renderer_host/resource_dispatcher_host_request_info.h" |
| 13 #include "chrome/common/url_constants.h" |
| 14 #include "net/base/cert_database.h" |
| 15 #include "net/base/io_buffer.h" |
| 16 #include "net/base/mime_sniffer.h" |
| 17 #include "net/base/mime_util.h" |
| 18 #include "net/http/http_response_headers.h" |
| 19 |
| 20 X509UserCertResourceHandler::X509UserCertResourceHandler( |
| 21 ResourceDispatcherHost *host, URLRequest* request) |
| 22 : host_(host), |
| 23 request_(request), |
| 24 content_length_(0), |
| 25 buffer_(new DownloadBuffer), |
| 26 read_buffer_(NULL), |
| 27 resource_buffer_(NULL) { |
| 28 } |
| 29 |
| 30 bool X509UserCertResourceHandler::OnRequestRedirected(int request_id, |
| 31 const GURL& url, |
| 32 ResourceResponse* resp, |
| 33 bool *defer) { |
| 34 url_ = url; |
| 35 return true; |
| 36 } |
| 37 |
| 38 bool X509UserCertResourceHandler::OnResponseStarted(int request_id, |
| 39 ResourceResponse* resp) { |
| 40 return (resp->response_head.mime_type == "application/x-x509-user-cert"); |
| 41 |
| 42 } |
| 43 |
| 44 bool X509UserCertResourceHandler::OnWillRead(int request_id, |
| 45 net::IOBuffer** buf, |
| 46 int* buf_size, |
| 47 int min_size) { |
| 48 // FIXME(gauravsh): Should we use 'min_size' here? |
| 49 DCHECK(buf && buf_size); |
| 50 if (!read_buffer_) { |
| 51 read_buffer_ = new net::IOBuffer(kReadBufSize); |
| 52 } |
| 53 *buf = read_buffer_.get(); |
| 54 *buf_size = kReadBufSize; |
| 55 |
| 56 return true; |
| 57 } |
| 58 |
| 59 bool X509UserCertResourceHandler::OnReadCompleted(int request_id, |
| 60 int* bytes_read) { |
| 61 if (!*bytes_read) |
| 62 return true; |
| 63 |
| 64 // We have more data to read. |
| 65 DCHECK(read_buffer_); |
| 66 content_length_ += *bytes_read; |
| 67 |
| 68 // Release the ownership of the buffer, and store a reference |
| 69 // to it. A new one will be allocated in OnWillRead(). |
| 70 net::IOBuffer* buffer = NULL; |
| 71 read_buffer_.swap(&buffer); |
| 72 // TODO(gauravsh): Should this be handled by a separate thread? |
| 73 buffer_->contents.push_back(std::make_pair(buffer, *bytes_read)); |
| 74 |
| 75 return true; |
| 76 } |
| 77 |
| 78 bool X509UserCertResourceHandler::OnResponseCompleted( |
| 79 int request_id, |
| 80 const URLRequestStatus& urs, |
| 81 const std::string& sec_info) { |
| 82 // TODO(gauravsh): Verify that 'request_id' was actually a keygen form post |
| 83 // and only then import the certificate |
| 84 scoped_ptr<net::CertDatabase> cert_db(new net::CertDatabase()); |
| 85 AssembleResource(); |
| 86 |
| 87 return cert_db->AddUserCert(resource_buffer_->data(), content_length_); |
| 88 } |
| 89 |
| 90 void X509UserCertResourceHandler::AssembleResource() { |
| 91 size_t bytes_copied = 0; |
| 92 resource_buffer_ = new net::IOBuffer(content_length_); |
| 93 |
| 94 for (size_t i = 0; i < buffer_->contents.size(); ++i) { |
| 95 net::IOBuffer* data = buffer_->contents[i].first; |
| 96 const int data_len = buffer_->contents[i].second; |
| 97 DCHECK(bytes_copied + data_len <= content_length_); |
| 98 memcpy(resource_buffer_->data() + bytes_copied, data->data(), data_len); |
| 99 bytes_copied += data_len; |
| 100 } |
| 101 } |
OLD | NEW |