chrome/browser/ui/certificate_dialogs.cc

   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/certificate_dialogs.h"
   6
   7
   8 #include <vector>
   9
  10 #include "base/base64.h"
  11 #include "base/bind.h"
  12 #include "base/file_util.h"
  13 #include "base/logging.h"
  14 #include "base/memory/scoped_ptr.h"
  15 #include "chrome/browser/ui/chrome_select_file_policy.h"
  16 #include "chrome/common/net/x509_certificate_model.h"
  17 #include "content/public/browser/browser_thread.h"
  18 #include "grit/generated_resources.h"
  19 #include "ui/base/dialogs/select_file_dialog.h"
  20 #include "ui/base/l10n/l10n_util.h"
  21
  22 using content::BrowserThread;
  23 using content::WebContents;
  24
  25 namespace {
  26
  27 void WriterCallback(const FilePath& path, const std::string& data) {
  28   int bytes_written = file_util::WriteFile(path, data.data(), data.size());
  29   if (bytes_written != static_cast<ssize_t>(data.size())) {
  30     LOG(ERROR) << "Writing " << path.value() << " ("
  31                << data.size() << "B) returned " << bytes_written;
  32   }
  33 }
  34
  35 void WriteFileOnFileThread(const FilePath& path, const std::string& data) {
  36   BrowserThread::PostTask(
  37       BrowserThread::FILE, FROM_HERE, base::Bind(&WriterCallback, path, data));
  38 }
  39
  40 std::string WrapAt64(const std::string &str) {
  41   std::string result;
  42   for (size_t i = 0; i < str.size(); i += 64) {
  43     result.append(str, i, 64);  // Append clamps the len arg internally.
  44     result.append("\r\n");
  45   }
  46   return result;
  47 }
  48
  49 std::string GetBase64String(net::X509Certificate::OSCertHandle cert) {
  50   std::string base64;
  51   if (!base::Base64Encode(
  52       x509_certificate_model::GetDerString(cert), &base64)) {
  53     LOG(ERROR) << "base64 encoding error";
  54     return "";
  55   }
  56   return "-----BEGIN CERTIFICATE-----\r\n" +
  57       WrapAt64(base64) +
  58       "-----END CERTIFICATE-----\r\n";
  59 }
  60
  61 ////////////////////////////////////////////////////////////////////////////////
  62 // General utility functions.
  63
  64 class Exporter : public ui::SelectFileDialog::Listener {
  65  public:
  66   Exporter(WebContents* web_contents, gfx::NativeWindow parent,
  67            net::X509Certificate::OSCertHandle cert);
  68   ~Exporter();
  69
  70   // SelectFileDialog::Listener implemenation.
  71   virtual void FileSelected(const FilePath& path,
  72                             int index, void* params);
  73   virtual void FileSelectionCanceled(void* params);
  74  private:
  75   scoped_refptr<ui::SelectFileDialog> select_file_dialog_;
  76
  77   // The certificate hierarchy (leaf cert first).
  78   net::X509Certificate::OSCertHandles cert_chain_list_;
  79 };
  80
  81 Exporter::Exporter(WebContents* web_contents,
  82                    gfx::NativeWindow parent,
  83                    net::X509Certificate::OSCertHandle cert)
  84     : select_file_dialog_(ui::SelectFileDialog::Create(
  85         this, new ChromeSelectFilePolicy(web_contents))) {
  86   x509_certificate_model::GetCertChainFromCert(cert, &cert_chain_list_);
  87
  88   // TODO(mattm): should this default to some directory?
  89   // Maybe SavePackage::GetSaveDirPreference? (Except that it's private.)
  90   FilePath suggested_path("certificate");
  91   std::string cert_title = x509_certificate_model::GetTitle(cert);
  92   if (!cert_title.empty())
  93     suggested_path = FilePath(cert_title);
  94
  95   ShowCertSelectFileDialog(select_file_dialog_.get(),
  96                            ui::SelectFileDialog::SELECT_SAVEAS_FILE,
  97                            suggested_path,
  98                            parent,
  99                            NULL);
 100 }
 101
 102 Exporter::~Exporter() {
 103   // There may be pending file dialogs, we need to tell them that we've gone
 104   // away so they don't try and call back to us.
 105   if (select_file_dialog_.get())
 106     select_file_dialog_->ListenerDestroyed();
 107
 108   x509_certificate_model::DestroyCertChain(&cert_chain_list_);
 109 }
 110
 111 void Exporter::FileSelected(const FilePath& path, int index, void* params) {
 112   std::string data;
 113   switch (index) {
 114     case 2:
 115       for (size_t i = 0; i < cert_chain_list_.size(); ++i)
 116         data += GetBase64String(cert_chain_list_[i]);
 117       break;
 118     case 3:
 119       data = x509_certificate_model::GetDerString(cert_chain_list_[0]);
 120       break;
 121     case 4:
 122       data = x509_certificate_model::GetCMSString(cert_chain_list_, 0, 1);
 123       break;
 124     case 5:
 125       data = x509_certificate_model::GetCMSString(
 126           cert_chain_list_, 0, cert_chain_list_.size());
 127       break;
 128     case 1:
 129     default:
 130       data = GetBase64String(cert_chain_list_[0]);
 131       break;
 132   }
 133
 134   if (!data.empty())
 135     WriteFileOnFileThread(path, data);
 136
 137   delete this;
 138 }
 139
 140 void Exporter::FileSelectionCanceled(void* params) {
 141   delete this;
 142 }
 143
 144 } // namespace
 145
 146 void ShowCertSelectFileDialog(ui::SelectFileDialog* select_file_dialog,
 147                               ui::SelectFileDialog::Type type,
 148                               const FilePath& suggested_path,
 149                               gfx::NativeWindow parent,
 150                               void* params) {
 151   ui::SelectFileDialog::FileTypeInfo file_type_info;
 152   file_type_info.extensions.resize(5);
 153   file_type_info.extensions[0].push_back(FILE_PATH_LITERAL("pem"));
 154   file_type_info.extensions[0].push_back(FILE_PATH_LITERAL("crt"));
 155   file_type_info.extension_description_overrides.push_back(
 156       l10n_util::GetStringUTF16(IDS_CERT_EXPORT_TYPE_BASE64));
 157   file_type_info.extensions[1].push_back(FILE_PATH_LITERAL("pem"));
 158   file_type_info.extensions[1].push_back(FILE_PATH_LITERAL("crt"));
 159   file_type_info.extension_description_overrides.push_back(
 160       l10n_util::GetStringUTF16(IDS_CERT_EXPORT_TYPE_BASE64_CHAIN));
 161   file_type_info.extensions[2].push_back(FILE_PATH_LITERAL("der"));
 162   file_type_info.extension_description_overrides.push_back(
 163       l10n_util::GetStringUTF16(IDS_CERT_EXPORT_TYPE_DER));
 164   file_type_info.extensions[3].push_back(FILE_PATH_LITERAL("p7c"));
 165   file_type_info.extension_description_overrides.push_back(
 166       l10n_util::GetStringUTF16(IDS_CERT_EXPORT_TYPE_PKCS7));
 167   file_type_info.extensions[4].push_back(FILE_PATH_LITERAL("p7c"));
 168   file_type_info.extension_description_overrides.push_back(
 169       l10n_util::GetStringUTF16(IDS_CERT_EXPORT_TYPE_PKCS7_CHAIN));
 170   file_type_info.include_all_files = true;
 171   select_file_dialog->SelectFile(
 172       type, string16(),
 173       suggested_path, &file_type_info, 1,
 174       FILE_PATH_LITERAL("crt"),
 175       parent, params);
 176 }
 177
 178 void ShowCertExportDialog(WebContents* web_contents,
 179                           gfx::NativeWindow parent,
 180                           net::X509Certificate::OSCertHandle cert) {
 181   new Exporter(web_contents, parent, cert);
 182 }