ssl_client_certificate_selector.cc revision 5d1f7b1de12d16ceb2c938c56701a3e8bfa558f7 
1// Copyright (c) 2012 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/ui/views/ssl_client_certificate_selector.h"
6
7#include "base/compiler_specific.h"
8#include "base/i18n/time_formatting.h"
9#include "base/logging.h"
10#include "base/strings/utf_string_conversions.h"
11#include "chrome/browser/certificate_viewer.h"
12#include "components/web_modal/web_contents_modal_dialog_host.h"
13#include "components/web_modal/web_contents_modal_dialog_manager.h"
14#include "components/web_modal/web_contents_modal_dialog_manager_delegate.h"
15#include "content/public/browser/browser_thread.h"
16#include "content/public/browser/web_contents.h"
17#include "content/public/browser/web_contents_view.h"
18#include "grit/generated_resources.h"
19#include "net/cert/x509_certificate.h"
20#include "net/ssl/ssl_cert_request_info.h"
21#include "ui/base/l10n/l10n_util.h"
22#include "ui/base/models/table_model.h"
23#include "ui/base/models/table_model_observer.h"
24#include "ui/gfx/native_widget_types.h"
25#include "ui/views/controls/button/label_button.h"
26#include "ui/views/controls/label.h"
27#include "ui/views/controls/table/table_view.h"
28#include "ui/views/layout/grid_layout.h"
29#include "ui/views/layout/layout_constants.h"
30#include "ui/views/widget/widget.h"
31#include "ui/views/window/dialog_client_view.h"
32
33#if defined(USE_NSS)
34#include "chrome/browser/ui/crypto_module_password_dialog_nss.h"
35#endif
36
37using content::BrowserThread;
38using content::WebContents;
39using web_modal::WebContentsModalDialogManager;
40using web_modal::WebContentsModalDialogManagerDelegate;
41
42namespace {
43
44// The dimensions of the certificate selector table view, in pixels.
45static const int kTableViewWidth = 400;
46static const int kTableViewHeight = 100;
47
48}  // namespace
49
50///////////////////////////////////////////////////////////////////////////////
51// CertificateSelectorTableModel:
52
53class CertificateSelectorTableModel : public ui::TableModel {
54 public:
55  explicit CertificateSelectorTableModel(
56      net::SSLCertRequestInfo* cert_request_info);
57
58  // ui::TableModel implementation:
59  virtual int RowCount() OVERRIDE;
60  virtual base::string16 GetText(int index, int column_id) OVERRIDE;
61  virtual void SetObserver(ui::TableModelObserver* observer) OVERRIDE;
62
63 private:
64  std::vector<base::string16> items_;
65
66  DISALLOW_COPY_AND_ASSIGN(CertificateSelectorTableModel);
67};
68
69CertificateSelectorTableModel::CertificateSelectorTableModel(
70    net::SSLCertRequestInfo* cert_request_info) {
71  for (size_t i = 0; i < cert_request_info->client_certs.size(); ++i) {
72    net::X509Certificate* cert = cert_request_info->client_certs[i].get();
73    base::string16 text = l10n_util::GetStringFUTF16(
74        IDS_CERT_SELECTOR_TABLE_CERT_FORMAT,
75        base::UTF8ToUTF16(cert->subject().GetDisplayName()),
76        base::UTF8ToUTF16(cert->issuer().GetDisplayName()));
77    items_.push_back(text);
78  }
79}
80
81int CertificateSelectorTableModel::RowCount() {
82  return items_.size();
83}
84
85base::string16 CertificateSelectorTableModel::GetText(int index,
86                                                      int column_id) {
87  DCHECK_EQ(column_id, 0);
88  DCHECK_GE(index, 0);
89  DCHECK_LT(index, static_cast<int>(items_.size()));
90
91  return items_[index];
92}
93
94void CertificateSelectorTableModel::SetObserver(
95    ui::TableModelObserver* observer) {
96}
97
98///////////////////////////////////////////////////////////////////////////////
99// SSLClientCertificateSelector:
100
101SSLClientCertificateSelector::SSLClientCertificateSelector(
102    WebContents* web_contents,
103    const net::HttpNetworkSession* network_session,
104    net::SSLCertRequestInfo* cert_request_info,
105    const chrome::SelectCertificateCallback& callback)
106    : SSLClientAuthObserver(network_session, cert_request_info, callback),
107      model_(new CertificateSelectorTableModel(cert_request_info)),
108      web_contents_(web_contents),
109      window_(NULL),
110      table_(NULL),
111      view_cert_button_(NULL) {
112  DVLOG(1) << __FUNCTION__;
113}
114
115SSLClientCertificateSelector::~SSLClientCertificateSelector() {
116  table_->SetModel(NULL);
117}
118
119void SSLClientCertificateSelector::Init() {
120  views::GridLayout* layout = views::GridLayout::CreatePanel(this);
121  SetLayoutManager(layout);
122
123  const int column_set_id = 0;
124  views::ColumnSet* column_set = layout->AddColumnSet(column_set_id);
125  column_set->AddColumn(
126      views::GridLayout::FILL, views::GridLayout::FILL,
127      1, views::GridLayout::USE_PREF, 0, 0);
128
129  layout->StartRow(0, column_set_id);
130  base::string16 text = l10n_util::GetStringFUTF16(
131      IDS_CLIENT_CERT_DIALOG_TEXT,
132      base::ASCIIToUTF16(cert_request_info()->host_and_port.ToString()));
133  views::Label* label = new views::Label(text);
134  label->SetMultiLine(true);
135  label->SetHorizontalAlignment(gfx::ALIGN_LEFT);
136  label->SetAllowCharacterBreak(true);
137  layout->AddView(label);
138
139  layout->AddPaddingRow(0, views::kRelatedControlVerticalSpacing);
140
141  CreateCertTable();
142  layout->StartRow(1, column_set_id);
143  layout->AddView(table_->CreateParentIfNecessary(), 1, 1,
144                  views::GridLayout::FILL,
145                  views::GridLayout::FILL, kTableViewWidth, kTableViewHeight);
146
147  layout->AddPaddingRow(0, views::kRelatedControlVerticalSpacing);
148
149  StartObserving();
150
151  WebContentsModalDialogManager* web_contents_modal_dialog_manager =
152      WebContentsModalDialogManager::FromWebContents(web_contents_);
153  WebContentsModalDialogManagerDelegate* modal_delegate =
154      web_contents_modal_dialog_manager->delegate();
155  DCHECK(modal_delegate);
156  window_ = views::Widget::CreateWindowAsFramelessChild(
157      this, modal_delegate->GetWebContentsModalDialogHost()->GetHostView());
158  web_contents_modal_dialog_manager->ShowDialog(window_->GetNativeView());
159
160  // Select the first row automatically.  This must be done after the dialog has
161  // been created.
162  table_->Select(0);
163}
164
165net::X509Certificate* SSLClientCertificateSelector::GetSelectedCert() const {
166  int selected = table_->FirstSelectedRow();
167  if (selected >= 0 &&
168      selected < static_cast<int>(
169          cert_request_info()->client_certs.size()))
170    return cert_request_info()->client_certs[selected].get();
171  return NULL;
172}
173
174///////////////////////////////////////////////////////////////////////////////
175// SSLClientAuthObserver implementation:
176
177void SSLClientCertificateSelector::OnCertSelectedByNotification() {
178  DVLOG(1) << __FUNCTION__;
179  DCHECK(window_);
180  window_->Close();
181}
182
183///////////////////////////////////////////////////////////////////////////////
184// DialogDelegateView implementation:
185
186bool SSLClientCertificateSelector::CanResize() const {
187  return true;
188}
189
190base::string16 SSLClientCertificateSelector::GetWindowTitle() const {
191  return l10n_util::GetStringUTF16(IDS_CLIENT_CERT_DIALOG_TITLE);
192}
193
194void SSLClientCertificateSelector::DeleteDelegate() {
195  DVLOG(1) << __FUNCTION__;
196  delete this;
197}
198
199bool SSLClientCertificateSelector::IsDialogButtonEnabled(
200    ui::DialogButton button) const {
201  if (button == ui::DIALOG_BUTTON_OK)
202    return !!GetSelectedCert();
203  return true;
204}
205
206bool SSLClientCertificateSelector::Cancel() {
207  DVLOG(1) << __FUNCTION__;
208  StopObserving();
209  CertificateSelected(NULL);
210
211  return true;
212}
213
214bool SSLClientCertificateSelector::Accept() {
215  DVLOG(1) << __FUNCTION__;
216  scoped_refptr<net::X509Certificate> cert = GetSelectedCert();
217  if (cert) {
218    // Remove the observer before we try unlocking, otherwise we might act on a
219    // notification while waiting for the unlock dialog, causing us to delete
220    // ourself before the Unlocked callback gets called.
221    StopObserving();
222#if defined(USE_NSS)
223    chrome::UnlockCertSlotIfNecessary(
224        cert,
225        chrome::kCryptoModulePasswordClientAuth,
226        cert_request_info()->host_and_port,
227        window_->GetNativeView(),
228        base::Bind(&SSLClientCertificateSelector::Unlocked,
229                   base::Unretained(this),
230                   cert));
231#else
232    Unlocked(cert);
233#endif
234    return false;  // Unlocked() will close the dialog.
235  }
236
237  return false;
238}
239
240views::View* SSLClientCertificateSelector::GetInitiallyFocusedView() {
241  return table_;
242}
243
244views::View* SSLClientCertificateSelector::CreateExtraView() {
245  DCHECK(!view_cert_button_);
246  view_cert_button_ = new views::LabelButton(this,
247      l10n_util::GetStringUTF16(IDS_PAGEINFO_CERT_INFO_BUTTON));
248  view_cert_button_->SetStyle(views::Button::STYLE_BUTTON);
249  return view_cert_button_;
250}
251
252ui::ModalType SSLClientCertificateSelector::GetModalType() const {
253#if defined(USE_ASH)
254  return ui::MODAL_TYPE_CHILD;
255#else
256  return views::WidgetDelegate::GetModalType();
257#endif
258}
259
260///////////////////////////////////////////////////////////////////////////////
261// views::ButtonListener implementation:
262
263void SSLClientCertificateSelector::ButtonPressed(
264    views::Button* sender, const ui::Event& event) {
265  if (sender == view_cert_button_) {
266    net::X509Certificate* cert = GetSelectedCert();
267    if (cert)
268      ShowCertificateViewer(web_contents_,
269                            web_contents_->GetView()->GetTopLevelNativeWindow(),
270                            cert);
271  }
272}
273
274///////////////////////////////////////////////////////////////////////////////
275// views::TableViewObserver implementation:
276void SSLClientCertificateSelector::OnSelectionChanged() {
277  GetDialogClientView()->ok_button()->SetEnabled(!!GetSelectedCert());
278}
279
280void SSLClientCertificateSelector::OnDoubleClick() {
281  if (Accept())
282    window_->Close();
283}
284
285///////////////////////////////////////////////////////////////////////////////
286// SSLClientCertificateSelector private methods:
287
288void SSLClientCertificateSelector::CreateCertTable() {
289  std::vector<ui::TableColumn> columns;
290  columns.push_back(ui::TableColumn());
291  table_ = new views::TableView(model_.get(),
292                                columns,
293                                views::TEXT_ONLY,
294                                true /* single_selection */);
295  table_->SetObserver(this);
296}
297
298void SSLClientCertificateSelector::Unlocked(net::X509Certificate* cert) {
299  DVLOG(1) << __FUNCTION__;
300  CertificateSelected(cert);
301  window_->Close();
302}
303
304namespace chrome {
305
306void ShowSSLClientCertificateSelector(
307    content::WebContents* contents,
308    const net::HttpNetworkSession* network_session,
309    net::SSLCertRequestInfo* cert_request_info,
310    const chrome::SelectCertificateCallback& callback) {
311  DVLOG(1) << __FUNCTION__ << " " << contents;
312  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
313  (new SSLClientCertificateSelector(
314       contents, network_session, cert_request_info, callback))->Init();
315}
316
317}  // namespace chrome
318