57 lines
1.6 KiB
C++
57 lines
1.6 KiB
C++
// Copyright (c) 2013 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "atom/browser/ui/certificate_trust.h"
|
|
|
|
#include <wincrypt.h>
|
|
#include <windows.h>
|
|
|
|
#include "base/callback.h"
|
|
#include "net/cert/cert_database.h"
|
|
|
|
namespace certificate_trust {
|
|
|
|
void ShowCertificateTrust(atom::NativeWindow* parent_window,
|
|
const scoped_refptr<net::X509Certificate>& cert,
|
|
const std::string& message,
|
|
const ShowTrustCallback& callback) {
|
|
// opening the Trusted Root Certificate store for the current user
|
|
auto hCertStore = CertOpenStore(
|
|
CERT_STORE_PROV_SYSTEM,
|
|
0,
|
|
NULL,
|
|
CERT_SYSTEM_STORE_CURRENT_USER,
|
|
// installing into Trusted Root Certificate Authorities, not Personal
|
|
L"Root");
|
|
|
|
if (hCertStore == NULL) {
|
|
callback.Run();
|
|
return;
|
|
}
|
|
|
|
auto pCertContext = cert->CreateOSCertChainForCert();
|
|
|
|
// This is a blocking call which displays a prompt to the user to confirm
|
|
// they trust this certificate
|
|
auto result = CertAddCertificateContextToStore(
|
|
hCertStore,
|
|
pCertContext,
|
|
CERT_STORE_ADD_REPLACE_EXISTING,
|
|
NULL);
|
|
|
|
if (result) {
|
|
auto cert_db = net::CertDatabase::GetInstance();
|
|
// Force Chromium to reload the certificate since it might be trusted
|
|
// now.
|
|
cert_db->NotifyObserversCertDBChanged(cert.get());
|
|
}
|
|
|
|
CertCloseStore(hCertStore, CERT_CLOSE_STORE_FORCE_FLAG);
|
|
|
|
CertFreeCertificateContext(pCertContext);
|
|
|
|
callback.Run();
|
|
}
|
|
|
|
} // namespace certificate_trust
|