1 /*
2  * Copyright (C) 2015 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package android.security.net.config;
18 
19 import android.util.ArraySet;
20 import java.security.cert.X509Certificate;
21 import java.util.Set;
22 
23 import com.android.org.conscrypt.TrustedCertificateIndex;
24 
25 /** @hide */
26 public class TestCertificateSource implements CertificateSource {
27 
28     private final Set<X509Certificate> mCertificates;
29     private final TrustedCertificateIndex mIndex = new TrustedCertificateIndex();
TestCertificateSource(Set<X509Certificate> certificates)30     public TestCertificateSource(Set<X509Certificate> certificates) {
31         mCertificates = certificates;
32         for (X509Certificate cert : certificates) {
33             mIndex.index(cert);
34         }
35     }
36 
37     @Override
getCertificates()38     public Set<X509Certificate> getCertificates() {
39             return mCertificates;
40     }
41 
42     @Override
findBySubjectAndPublicKey(X509Certificate cert)43     public X509Certificate findBySubjectAndPublicKey(X509Certificate cert) {
44         java.security.cert.TrustAnchor anchor = mIndex.findBySubjectAndPublicKey(cert);
45         if (anchor == null) {
46             return null;
47         }
48         return anchor.getTrustedCert();
49     }
50 
51     @Override
findByIssuerAndSignature(X509Certificate cert)52     public X509Certificate findByIssuerAndSignature(X509Certificate cert) {
53         java.security.cert.TrustAnchor anchor = mIndex.findByIssuerAndSignature(cert);
54         if (anchor == null) {
55             return null;
56         }
57         return anchor.getTrustedCert();
58     }
59 
60     @Override
findAllByIssuerAndSignature(X509Certificate cert)61     public Set<X509Certificate> findAllByIssuerAndSignature(X509Certificate cert) {
62         Set<X509Certificate> certs = new ArraySet<X509Certificate>();
63         for (java.security.cert.TrustAnchor anchor : mIndex.findAllByIssuerAndSignature(cert)) {
64             certs.add(anchor.getTrustedCert());
65         }
66         return certs;
67     }
68 
69     @Override
handleTrustStorageUpdate()70     public void handleTrustStorageUpdate() {
71         // Nothing to do.
72     }
73 }
74