Add export function to WebUI certificate viewer.

chrome/browser/ui/webui/certificate_viewer.cc

Index: chrome/browser/ui/webui/certificate_viewer.cc
diff --git a/chrome/browser/ui/webui/certificate_viewer.cc b/chrome/browser/ui/webui/certificate_viewer.cc
index 03f2db10fde07ba13dfa033813e51ec87c5c5a62..e7a0f7cabb6229d655c46179b881053cc161f88a 100644
--- a/chrome/browser/ui/webui/certificate_viewer.cc
+++ b/chrome/browser/ui/webui/certificate_viewer.cc
@@ -2,16 +2,17 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-#include "base/json/json_writer.h"
+#include "base/i18n/time_formatting.h"
 #include "base/utf_string_conversions.h"
 #include "base/string_number_conversions.h"
 #include "chrome/browser/ui/webui/certificate_viewer.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_list.h"
-#include "chrome/browser/ui/views/html_dialog_view.h"
+#include "chrome/browser/ui/gtk/certificate_dialogs.h"
 #include "chrome/browser/ui/browser_dialogs.h"
 #include "chrome/common/net/x509_certificate_model.h"
+#include "content/browser/tab_contents/tab_contents.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "grit/generated_resources.h"
 
@@ -29,27 +30,20 @@ void ShowCertificateViewer(gfx::NativeWindow parent,
   CertificateViewerDialog::ShowDialog(parent, cert);
 }
 
-void CertificateViewerDialog::ShowDialog(gfx::NativeWindow owning_window,
+////////////////////////////////////////////////////////////////////////////////
+// CertificateViewerDialog
+
+void CertificateViewerDialog::ShowDialog(gfx::NativeWindow parent,
                                          net::X509Certificate* cert) {
   Browser* browser = BrowserList::GetLastActive();
   DCHECK(browser);
-  browser->BrowserShowHtmlDialog(new CertificateViewerDialog(cert),
-                                 owning_window);
-}
-
-// TODO(flackr): This is duplicated from cookies_view_handler.cc
-// Encodes a pointer value into a hex string.
-std::string PointerToHexString(const void* pointer) {
-  return base::HexEncode(&pointer, sizeof(pointer));
+  browser->BrowserShowHtmlDialog(new CertificateViewerDialog(parent, cert),
+                                 parent);
 }
 
-CertificateViewerDialog::CertificateViewerDialog(net::X509Certificate* cert)
-    : cert_(cert) {
-  // Construct the JSON string with a pointer to the stored certificate.
-  DictionaryValue args;
-  args.SetString("cert", PointerToHexString(cert_));
-  base::JSONWriter::Write(&args, false, &json_args_);
-
+CertificateViewerDialog::CertificateViewerDialog(gfx::NativeWindow parent,
+                                                 net::X509Certificate* cert)
+    : cert_(cert), parent_(parent) {
   // Construct the dialog title from the certificate.
   net::X509Certificate::OSCertHandles cert_chain;
   x509_certificate_model::GetCertChainFromCert(cert_->os_cert_handle(),
@@ -59,7 +53,7 @@ CertificateViewerDialog::CertificateViewerDialog(net::X509Certificate* cert)
 }
 
 bool CertificateViewerDialog::IsDialogModal() const {
-  return false;
+  return true;
 }
 
 string16 CertificateViewerDialog::GetDialogTitle() const {
@@ -72,6 +66,7 @@ GURL CertificateViewerDialog::GetDialogContentURL() const {
 
 void CertificateViewerDialog::GetWebUIMessageHandlers(
     std::vector<WebUIMessageHandler*>* handlers) const {
+  handlers->push_back(new CertificateViewerDialogHandler(parent_, cert_));
 }
 
 void CertificateViewerDialog::GetDialogSize(gfx::Size* size) const {
@@ -79,7 +74,7 @@ void CertificateViewerDialog::GetDialogSize(gfx::Size* size) const {
 }
 
 std::string CertificateViewerDialog::GetDialogArgs() const {
-  return json_args_;
+  return std::string();
 }
 
 void CertificateViewerDialog::OnDialogClosed(const std::string& json_retval) {
@@ -100,3 +95,292 @@ bool CertificateViewerDialog::HandleContextMenu(
     const ContextMenuParams& params) {
   return true;
 }
+
+////////////////////////////////////////////////////////////////////////////////
+// CertificateViewerDialogHandler
+
+CertificateViewerDialogHandler::CertificateViewerDialogHandler(
+    gfx::NativeWindow parent,
+    net::X509Certificate* cert) : cert_(cert), parent_(parent) {
+  x509_certificate_model::GetCertChainFromCert(cert_->os_cert_handle(),
+      &cert_chain_);
+}
+
+void CertificateViewerDialogHandler::RegisterMessages() {
+  web_ui_->RegisterMessageCallback("exportCertificate",
+      NewCallback(this,
+          &CertificateViewerDialogHandler::ExportCertificate));
+  web_ui_->RegisterMessageCallback("requestCertificateInfo",
+      NewCallback(this,
+          &CertificateViewerDialogHandler::RequestCertificateInfo));
+  web_ui_->RegisterMessageCallback("requestCertificateFields",
+      NewCallback(this,
+          &CertificateViewerDialogHandler::RequestCertificateFields));
+}
+
+void CertificateViewerDialogHandler::ExportCertificate(
+    const base::ListValue* args) {
+  int cert_index;
+  double val;
+  if (!(args->GetDouble(0, &val)))
+    return;
+  cert_index = (int)val;
+  if (cert_index < 0 || cert_index >= (int)cert_chain_.size())
+    return;
+
+  // TODO(flackr): We should be able to use the current dialog's container
+  //     window as a parent for the export dialog however chromeos builds fail
+  //     to locate the browser for this window. HtmlDialog should probably
+  //     behave similar to a window.open popup with an associated Browser
+  //     object.
+  ShowCertExportDialog(web_ui_->tab_contents(),
+                       parent_,
+                       cert_chain_[cert_index]);
+}
+
+void CertificateViewerDialogHandler::RequestCertificateInfo(
+    const base::ListValue* args) {
+  // Certificate information. The keys in this dictionary's general key
+  // correspond to the IDs in the Html page.
+  DictionaryValue cert_info;
+  net::X509Certificate::OSCertHandle cert_hnd = cert_->os_cert_handle();
+
+  // Get the certificate chain.
+  net::X509Certificate::OSCertHandles cert_chain;
+  x509_certificate_model::GetCertChainFromCert(cert_hnd, &cert_chain);
+
+  // Certificate usage.
+  std::vector<std::string> usages;
+  x509_certificate_model::GetUsageStrings(cert_hnd, &usages);
+  std::string usagestr;
+  for (std::vector<std::string>::iterator it = usages.begin();
+      it != usages.end(); ++it) {
+    if (usagestr.length() > 0) {
+      usagestr += "\n";
+    }
+    usagestr += *it;
+  }
+  cert_info.SetString("general.usages", usagestr);
+
+  // Standard certificate details.
+  const std::string alternative_text =
+      l10n_util::GetStringUTF8(IDS_CERT_INFO_FIELD_NOT_PRESENT);
+  cert_info.SetString("general.title", l10n_util::GetStringFUTF8(
+      IDS_CERT_INFO_DIALOG_TITLE, UTF8ToUTF16(x509_certificate_model::GetTitle(
+          cert_chain.front()))));
+
+  // Issued to information.
+  cert_info.SetString("general.issued-cn",
+      x509_certificate_model::GetSubjectCommonName(cert_hnd, alternative_text));
+  cert_info.SetString("general.issued-o",
+      x509_certificate_model::GetSubjectOrgName(cert_hnd, alternative_text));
+  cert_info.SetString("general.issued-ou",
+      x509_certificate_model::GetSubjectOrgUnitName(cert_hnd,
+                                                    alternative_text));
+  cert_info.SetString("general.issued-sn",
+      x509_certificate_model::GetSerialNumberHexified(cert_hnd,
+                                                      alternative_text));
+
+  // Issuer information.
+  cert_info.SetString("general.issuer-cn",
+      x509_certificate_model::GetIssuerCommonName(cert_hnd, alternative_text));
+  cert_info.SetString("general.issuer-o",
+      x509_certificate_model::GetIssuerOrgName(cert_hnd, alternative_text));
+  cert_info.SetString("general.issuer-ou",
+      x509_certificate_model::GetIssuerOrgUnitName(cert_hnd, alternative_text));
+
+  // Validity period.
+  base::Time issued, expires;
+  std::string issued_str, expires_str;
+  if (x509_certificate_model::GetTimes(cert_hnd, &issued, &expires)) {
+    issued_str = UTF16ToUTF8(
+        base::TimeFormatShortDateNumeric(issued));
+    expires_str = UTF16ToUTF8(
+        base::TimeFormatShortDateNumeric(expires));
+  } else {
+    issued_str = alternative_text;
+    expires_str = alternative_text;
+  }
+  cert_info.SetString("general.issue-date", issued_str);
+  cert_info.SetString("general.expiry-date", expires_str);
+
+  cert_info.SetString("general.sha256",
+      x509_certificate_model::HashCertSHA256(cert_hnd));
+  cert_info.SetString("general.sha1",
+      x509_certificate_model::HashCertSHA1(cert_hnd));
+
+  // Certificate hierarchy is constructed from bottom up.
+  ListValue* children = NULL;
+  int index = 0;
+  for (net::X509Certificate::OSCertHandles::const_iterator i =
+      cert_chain.begin(); i != cert_chain.end(); ++i, ++index) {
+    DictionaryValue* cert_node = new DictionaryValue();
+    ListValue cert_details;
+    cert_node->SetString("label", x509_certificate_model::GetTitle(*i).c_str());
+    cert_node->SetDouble("payload.index", index);
+    // Add the child from the previous iteration.
+    if (children)
+      cert_node->Set("children", children);
+
+    // Add this node to the children list for the next iteration.
+    children = new ListValue();
+    children->Append(cert_node);
+  }
+  // Set the last node as the top of the certificate hierarchy.
+  cert_info.Set("hierarchy", children);
+
+  // Send certificate information to javascript.
+  web_ui_->CallJavascriptFunction("cert_viewer.getCertificateInfo", cert_info);
+}
+
+void CertificateViewerDialogHandler::RequestCertificateFields(
+    const base::ListValue* args) {
+  int cert_index;
+  double val;
+  if (!(args->GetDouble(0, &val)))
+    return;
+  cert_index = (int)val;
+  if (cert_index < 0 || cert_index >= (int)cert_chain_.size())
+    return;
+  net::X509Certificate::OSCertHandle cert = cert_chain_[cert_index];
+
+  ListValue root_list;
+  DictionaryValue* node_details;
+  DictionaryValue* alt_node_details;
+  ListValue* cert_sub_fields;
+  root_list.Append(node_details = new DictionaryValue());
+  node_details->SetString("label", x509_certificate_model::GetTitle(cert));
+
+  ListValue* cert_fields;
+  node_details->Set("children", cert_fields = new ListValue());
+  cert_fields->Append(node_details = new DictionaryValue());
+
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_CERTIFICATE));
+  node_details->Set("children", cert_fields = new ListValue());
+
+  // Main certificate fields.
+  cert_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_VERSION));
+  std::string version = x509_certificate_model::GetVersion(cert);
+  if (!version.empty())
+    node_details->SetString("payload.val",
+        l10n_util::GetStringFUTF8(IDS_CERT_DETAILS_VERSION_FORMAT,
+                                  UTF8ToUTF16(version)));
+
+  cert_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_SERIAL_NUMBER));
+  node_details->SetString("payload.val",
+      x509_certificate_model::GetSerialNumberHexified(cert,
+          l10n_util::GetStringUTF8(IDS_CERT_INFO_FIELD_NOT_PRESENT)));
+
+  cert_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_CERTIFICATE_SIG_ALG));
+  node_details->SetString("payload.val",
+      x509_certificate_model::ProcessSecAlgorithmSignature(cert));
+
+  cert_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_ISSUER));
+  node_details->SetString("payload.val",
+      x509_certificate_model::GetIssuerName(cert));
+
+  // Validity period.
+  cert_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_VALIDITY));
+
+  node_details->Set("children", cert_sub_fields = new ListValue());
+  cert_sub_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_NOT_BEFORE));
+  cert_sub_fields->Append(alt_node_details = new DictionaryValue());
+  alt_node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_NOT_AFTER));
+  base::Time issued, expires;
+  if (x509_certificate_model::GetTimes(cert, &issued, &expires)) {
+    node_details->SetString("payload.val",
+        UTF16ToUTF8(base::TimeFormatShortDateAndTime(issued)));
+    alt_node_details->SetString("payload.val",
+        UTF16ToUTF8(base::TimeFormatShortDateAndTime(expires)));
+  }
+
+  cert_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_SUBJECT));
+  node_details->SetString("payload.val",
+      x509_certificate_model::GetSubjectName(cert));
+
+  // Subject key information.
+  cert_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_SUBJECT_KEY_INFO));
+
+  node_details->Set("children", cert_sub_fields = new ListValue());
+  cert_sub_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_SUBJECT_KEY_ALG));
+  node_details->SetString("payload.val",
+      x509_certificate_model::ProcessSecAlgorithmSubjectPublicKey(cert));
+  cert_sub_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_SUBJECT_KEY));
+  node_details->SetString("payload.val",
+      x509_certificate_model::ProcessSubjectPublicKeyInfo(cert));
+
+  // Extensions.
+  x509_certificate_model::Extensions extensions;
+  x509_certificate_model::GetExtensions(
+      l10n_util::GetStringUTF8(IDS_CERT_EXTENSION_CRITICAL),
+      l10n_util::GetStringUTF8(IDS_CERT_EXTENSION_NON_CRITICAL),
+      cert, &extensions);
+
+  if (!extensions.empty()) {
+    cert_fields->Append(node_details = new DictionaryValue());
+    node_details->SetString("label",
+        l10n_util::GetStringUTF8(IDS_CERT_DETAILS_EXTENSIONS));
+
+    node_details->Set("children", cert_sub_fields = new ListValue());
+    for (x509_certificate_model::Extensions::const_iterator i =
+         extensions.begin(); i != extensions.end(); ++i) {
+      cert_sub_fields->Append(node_details = new DictionaryValue());
+      node_details->SetString("label", i->name);
+      node_details->SetString("payload.val", i->value);
+    }
+  }
+
+  cert_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_CERTIFICATE_SIG_ALG));
+  node_details->SetString("payload.val",
+      x509_certificate_model::ProcessSecAlgorithmSignatureWrap(cert));
+
+  cert_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_DETAILS_CERTIFICATE_SIG_VALUE));
+  node_details->SetString("payload.val",
+      x509_certificate_model::ProcessRawBitsSignatureWrap(cert));
+
+  cert_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_INFO_FINGERPRINTS_GROUP));
+  node_details->Set("children", cert_sub_fields = new ListValue());
+
+  cert_sub_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_INFO_SHA256_FINGERPRINT_LABEL));
+  node_details->SetString("payload.val",
+      x509_certificate_model::HashCertSHA256(cert));
+  cert_sub_fields->Append(node_details = new DictionaryValue());
+  node_details->SetString("label",
+      l10n_util::GetStringUTF8(IDS_CERT_INFO_SHA1_FINGERPRINT_LABEL));
+  node_details->SetString("payload.val",
+      x509_certificate_model::HashCertSHA1(cert));
+
+  // Send certificate information to javascript.
+  web_ui_->CallJavascriptFunction("cert_viewer.getCertificateFields",
+      root_list);
+}