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
17package android.security.net.config;
18
19import android.util.ArraySet;
20import java.security.cert.X509Certificate;
21import java.util.Set;
22
23import com.android.org.conscrypt.TrustedCertificateIndex;
24
25/** @hide */
26public class TestCertificateSource implements CertificateSource {
27
28    private final Set<X509Certificate> mCertificates;
29    private final TrustedCertificateIndex mIndex = new TrustedCertificateIndex();
30    public TestCertificateSource(Set<X509Certificate> certificates) {
31        mCertificates = certificates;
32        for (X509Certificate cert : certificates) {
33            mIndex.index(cert);
34        }
35    }
36
37    @Override
38    public Set<X509Certificate> getCertificates() {
39            return mCertificates;
40    }
41
42    @Override
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
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
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
70    public void handleTrustStorageUpdate() {
71        // Nothing to do.
72    }
73}
74