Add export function to WebUI certificate viewer.

chrome/browser/ui/webui/certificate_viewer.h

 // Copyright (c) 2011 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.
 
 #ifndef CHROME_BROWSER_UI_WEBUI_CERTIFICATE_VIEWER_H_
 #define CHROME_BROWSER_UI_WEBUI_CERTIFICATE_VIEWER_H_
 #pragma once
 
+#include "base/values.h"
 #include "chrome/browser/ui/webui/html_dialog_ui.h"
 #include "net/base/x509_certificate.h"
 #include "ui/gfx/native_widget_types.h"
 
 // Displays the WebUI certificate viewer dialog for the passed in certificate.
 void ShowCertificateViewer(gfx::NativeWindow parent,
                            net::X509Certificate*);
 
 // Dialog for displaying detailed certificate information. This is used in linux
 // and chromeos builds to display detailed information in a floating dialog when
@@ skipping 21 matching lines @@
   virtual std::string GetDialogArgs() const OVERRIDE;
   virtual void OnDialogClosed(const std::string& json_retval) OVERRIDE;
   virtual void OnCloseContents(
       TabContents* source, bool* out_close_dialog) OVERRIDE;
   virtual bool ShouldShowDialogTitle() const OVERRIDE;
   virtual bool HandleContextMenu(const ContextMenuParams& params) OVERRIDE;
 
   // The certificate being viewed.
   scoped_refptr<net::X509Certificate> cert_;
 
-  // The argument string for the dialog which passes the certificate pointer.
-  std::string json_args_;
-
   // The title of the certificate viewer dialog, Certificate Viewer: CN.
   string16 title_;
 
   DISALLOW_COPY_AND_ASSIGN(CertificateViewerDialog);
 };
 
+// Dialog handler which handles calls from the JS WebUI code to view certificate
+// details and export the certificate.
+class CertificateViewerDialogHandler : public WebUIMessageHandler {
+ public:
+  explicit CertificateViewerDialogHandler(net::X509Certificate* cert);
+
+  // Overridden from WebUIMessageHandler
+  virtual void RegisterMessages();
+
+ private:
+  // Brings up the export certificate dialog for the chosen certificate in the
+  // chain.
+  //
+  // The input is an integer index to the certificate in the chain to export.
+  void ExportCertificate(const base::ListValue* args);
+
+  // Gets the details for a specific certificate in the certificate chain. Calls
+  // the javascript function cert_viewer.getCertificateFields with a tree
+  // structure containing the fields and values for certain nodes.
+  //
+  // The input is an integer index to the certificate in the chain to view.
+  void RequestCertificateFields(const base::ListValue* args);
+
+  // Extracts the certificate details and returns them to the javascript
+  // function cert_viewer.getCertificateInfo in a dictionary structure.
+  void RequestCertificateInfo(const base::ListValue* args);
+
+  // The certificate being viewed.
+  scoped_refptr<net::X509Certificate> cert_;
+
+  // The certificate chain.
+  net::X509Certificate::OSCertHandles cert_chain_;
+
+  DISALLOW_COPY_AND_ASSIGN(CertificateViewerDialogHandler);
+};
+
 #endif  // CHROME_BROWSER_UI_WEBUI_CERTIFICATE_VIEWER_H_