-
Notifications
You must be signed in to change notification settings - Fork 4.6k
/
X509StoreMutableTests.cs
184 lines (153 loc) · 7.57 KB
/
X509StoreMutableTests.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using Xunit;
namespace System.Security.Cryptography.X509Certificates.Tests
{
[OuterLoop("Modifies system state")]
[SkipOnPlatform(TestPlatforms.Browser, "Browser doesn't support X509Store")]
public static partial class X509StoreMutableTests
{
[ConditionalFact(nameof(PermissionsAllowStoreWrite))]
public static void AddToStoreTwice()
{
using (var store = new X509Store(StoreName.My, StoreLocation.CurrentUser))
using (var cert = new X509Certificate2(TestData.PfxData, TestData.PfxDataPassword, X509KeyStorageFlags.Exportable))
using (var certOnly = new X509Certificate2(cert.RawData))
{
store.Open(OpenFlags.ReadWrite);
// Defensive removal.
store.Remove(certOnly);
Assert.False(IsCertInStore(cert, store), "PfxData certificate was found on pre-condition");
store.Add(cert);
Assert.True(IsCertInStore(certOnly, store), "PfxData certificate was found after add");
// No exception for duplicate item.
store.Add(cert);
// Cleanup
store.Remove(certOnly);
}
}
[ConditionalFact(nameof(PermissionsAllowStoreWrite))]
public static void AddPrivateAfterPublic()
{
using (var store = new X509Store(StoreName.My, StoreLocation.CurrentUser))
using (var cert = new X509Certificate2(TestData.PfxData, TestData.PfxDataPassword, X509KeyStorageFlags.Exportable))
using (var certOnly = new X509Certificate2(cert.RawData))
{
store.Open(OpenFlags.ReadWrite);
// Defensive removal.
store.Remove(certOnly);
Assert.False(IsCertInStore(cert, store), "PfxData certificate was found on pre-condition");
store.Add(certOnly);
Assert.True(IsCertInStore(certOnly, store), "PfxData certificate was found after add");
Assert.False(StoreHasPrivateKey(store, certOnly), "Store has a private key for PfxData after public-only add");
// Add the private key
store.Add(cert);
Assert.True(StoreHasPrivateKey(store, certOnly), "Store has a private key for PfxData after PFX add");
// Cleanup
store.Remove(certOnly);
}
}
[ConditionalFact(nameof(PermissionsAllowStoreWrite))]
public static void AddPublicAfterPrivate()
{
using (var store = new X509Store(StoreName.My, StoreLocation.CurrentUser))
using (var cert = new X509Certificate2(TestData.PfxData, TestData.PfxDataPassword, X509KeyStorageFlags.Exportable))
using (var certOnly = new X509Certificate2(cert.RawData))
{
store.Open(OpenFlags.ReadWrite);
// Defensive removal.
store.Remove(certOnly);
Assert.False(IsCertInStore(cert, store), "PfxData certificate was found on pre-condition");
// Add the private key
store.Add(cert);
Assert.True(IsCertInStore(certOnly, store), "PfxData certificate was found after add");
Assert.True(StoreHasPrivateKey(store, certOnly), "Store has a private key for PfxData after PFX add");
// Add the public key with no private key
store.Add(certOnly);
Assert.True(StoreHasPrivateKey(store, certOnly), "Store has a private key for PfxData after public-only add");
// Cleanup
store.Remove(certOnly);
}
}
[ConditionalTheory(nameof(PermissionsAllowStoreWrite))]
[InlineData(true)]
[InlineData(false)]
public static void VerifyRemove(bool withPrivateKey)
{
using (var store = new X509Store(StoreName.My, StoreLocation.CurrentUser))
using (var certWithPrivateKey = new X509Certificate2(TestData.PfxData, TestData.PfxDataPassword, X509KeyStorageFlags.Exportable))
using (var certOnly = new X509Certificate2(certWithPrivateKey.RawData))
{
X509Certificate2 cert = withPrivateKey ? certWithPrivateKey : certOnly;
store.Open(OpenFlags.ReadWrite);
// Defensive removal. Sort of circular, but it's the best we can do.
store.Remove(cert);
Assert.False(IsCertInStore(cert, store), "PfxData certificate was found on pre-condition");
store.Add(cert);
Assert.True(IsCertInStore(cert, store), "PfxData certificate was found after add");
store.Remove(cert);
Assert.False(IsCertInStore(cert, store), "PfxData certificate was found after remove");
}
}
[ConditionalFact(nameof(PermissionsAllowStoreWrite))]
public static void RemovePublicDeletesPrivateKey()
{
using (var store = new X509Store(StoreName.My, StoreLocation.CurrentUser))
using (var cert = new X509Certificate2(TestData.PfxData, TestData.PfxDataPassword, X509KeyStorageFlags.Exportable))
using (var certOnly = new X509Certificate2(cert.RawData))
{
store.Open(OpenFlags.ReadWrite);
// Defensive removal.
store.Remove(cert);
Assert.False(IsCertInStore(cert, store), "PfxData certificate was found on pre-condition");
// Add the private key
store.Add(cert);
Assert.True(IsCertInStore(cert, store), "PfxData certificate was found after add");
store.Remove(certOnly);
Assert.False(IsCertInStore(cert, store), "PfxData certificate was found after remove");
// Add back the public key only
store.Add(certOnly);
Assert.True(IsCertInStore(cert, store), "PfxData certificate was found after public-only add");
Assert.False(StoreHasPrivateKey(store, cert), "Store has a private key for cert after public-only add");
// Cleanup
store.Remove(certOnly);
}
}
private static bool StoreHasPrivateKey(X509Store store, X509Certificate2 forCert)
{
using (ImportedCollection coll = new ImportedCollection(store.Certificates))
{
foreach (X509Certificate2 storeCert in coll.Collection)
{
if (forCert.Equals(storeCert))
{
return storeCert.HasPrivateKey;
}
}
}
Assert.Fail($"Certificate ({forCert.Subject}) exists in the store");
return false;
}
private static bool IsCertInStore(X509Certificate2 cert, X509Store store)
{
using (ImportedCollection coll = new ImportedCollection(store.Certificates))
{
foreach (X509Certificate2 storeCert in coll.Collection)
{
if (cert.Equals(storeCert))
{
return true;
}
}
}
return false;
}
private static int GetStoreCertificateCount(X509Store store)
{
using (var coll = new ImportedCollection(store.Certificates))
{
return coll.Collection.Count;
}
}
}
}