Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/browser/renderer_host/x509_user_cert_resource_handler.h" | 5 #include "content/browser/renderer_host/x509_user_cert_resource_handler.h" |
| 6 | 6 |
| 7 #include "base/string_util.h" | 7 #include "base/string_util.h" |
| 8 #include "chrome/browser/download/download_types.h" | 8 #include "chrome/browser/download/download_types.h" |
| 9 #include "chrome/browser/ssl/ssl_add_cert_handler.h" | 9 #include "chrome/browser/ssl/ssl_add_cert_handler.h" |
| 10 #include "chrome/common/resource_response.h" | 10 #include "chrome/common/resource_response.h" |
| 11 #include "chrome/common/url_constants.h" | 11 #include "chrome/common/url_constants.h" |
| 12 #include "content/browser/renderer_host/resource_dispatcher_host.h" | 12 #include "content/browser/renderer_host/resource_dispatcher_host.h" |
| 13 #include "content/browser/renderer_host/resource_dispatcher_host_request_info.h" | 13 #include "content/browser/renderer_host/resource_dispatcher_host_request_info.h" |
| 14 #include "net/base/io_buffer.h" | 14 #include "net/base/io_buffer.h" |
| 15 #include "net/base/mime_sniffer.h" | 15 #include "net/base/mime_sniffer.h" |
| 16 #include "net/base/mime_util.h" | 16 #include "net/base/mime_util.h" |
| 17 #include "net/base/x509_certificate.h" | 17 #include "net/base/x509_certificate.h" |
| 18 #include "net/http/http_network_session.h" | |
| 18 #include "net/http/http_response_headers.h" | 19 #include "net/http/http_response_headers.h" |
| 20 #include "net/http/http_transaction_factory.h" | |
| 19 #include "net/url_request/url_request.h" | 21 #include "net/url_request/url_request.h" |
| 22 #include "net/url_request/url_request_context.h" | |
| 20 #include "net/url_request/url_request_status.h" | 23 #include "net/url_request/url_request_status.h" |
| 21 | 24 |
| 22 X509UserCertResourceHandler::X509UserCertResourceHandler( | 25 X509UserCertResourceHandler::X509UserCertResourceHandler( |
| 23 ResourceDispatcherHost* host, net::URLRequest* request, | 26 ResourceDispatcherHost* host, net::URLRequest* request, |
| 24 int render_process_host_id, int render_view_id) | 27 int render_process_host_id, int render_view_id) |
| 25 : host_(host), | 28 : host_(host), |
| 26 request_(request), | 29 request_(request), |
| 27 content_length_(0), | 30 content_length_(0), |
| 28 buffer_(new DownloadBuffer), | 31 buffer_(new DownloadBuffer), |
| 29 read_buffer_(NULL), | 32 read_buffer_(NULL), |
| (...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 100 | 103 |
| 101 // TODO(gauravsh): Verify that 'request_id' was actually a keygen form post | 104 // TODO(gauravsh): Verify that 'request_id' was actually a keygen form post |
| 102 // and only then import the certificate. | 105 // and only then import the certificate. |
| 103 AssembleResource(); | 106 AssembleResource(); |
| 104 scoped_refptr<net::X509Certificate> cert( | 107 scoped_refptr<net::X509Certificate> cert( |
| 105 net::X509Certificate::CreateFromBytes(resource_buffer_->data(), | 108 net::X509Certificate::CreateFromBytes(resource_buffer_->data(), |
| 106 content_length_)); | 109 content_length_)); |
| 107 // The handler will run the UI and delete itself when it's finished. | 110 // The handler will run the UI and delete itself when it's finished. |
| 108 new SSLAddCertHandler(request_, cert, render_process_host_id_, | 111 new SSLAddCertHandler(request_, cert, render_process_host_id_, |
| 109 render_view_id_); | 112 render_view_id_); |
| 113 // Force all new SSL connects to renegotiate, and hence use this new | |
| 114 // certificate if necessary. | |
| 115 if (request_->context()->http_transaction_factory()) { | |
| 116 net::HttpNetworkSession* session = | |
| 117 request_->context()->http_transaction_factory()->GetSession(); | |
| 118 session->ResetSSLState(); | |
|
wtc
2011/02/23 00:38:38
I think it's better to call session->ResetSSLState
| |
| 119 } | |
| 110 return true; | 120 return true; |
| 111 } | 121 } |
| 112 | 122 |
| 113 void X509UserCertResourceHandler::OnRequestClosed() { | 123 void X509UserCertResourceHandler::OnRequestClosed() { |
| 114 } | 124 } |
| 115 | 125 |
| 116 X509UserCertResourceHandler::~X509UserCertResourceHandler() { | 126 X509UserCertResourceHandler::~X509UserCertResourceHandler() { |
| 117 } | 127 } |
| 118 | 128 |
| 119 void X509UserCertResourceHandler::AssembleResource() { | 129 void X509UserCertResourceHandler::AssembleResource() { |
| 120 size_t bytes_copied = 0; | 130 size_t bytes_copied = 0; |
| 121 resource_buffer_ = new net::IOBuffer(content_length_); | 131 resource_buffer_ = new net::IOBuffer(content_length_); |
| 122 | 132 |
| 123 for (size_t i = 0; i < buffer_->contents.size(); ++i) { | 133 for (size_t i = 0; i < buffer_->contents.size(); ++i) { |
| 124 net::IOBuffer* data = buffer_->contents[i].first; | 134 net::IOBuffer* data = buffer_->contents[i].first; |
| 125 const int data_len = buffer_->contents[i].second; | 135 const int data_len = buffer_->contents[i].second; |
| 126 DCHECK(bytes_copied + data_len <= content_length_); | 136 DCHECK(bytes_copied + data_len <= content_length_); |
| 127 memcpy(resource_buffer_->data() + bytes_copied, data->data(), data_len); | 137 memcpy(resource_buffer_->data() + bytes_copied, data->data(), data_len); |
| 128 bytes_copied += data_len; | 138 bytes_copied += data_len; |
| 129 } | 139 } |
| 130 } | 140 } |
| OLD | NEW |