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
|
// Copyright (c) Microsoft Corporation. All rights reserved. See License.txt in the project root for license information.
using System.Web.Security;
using Xunit;
using Xunit.Extensions;
using Assert = Microsoft.TestCommon.AssertEx;
namespace System.Web.Helpers.AntiXsrf.Test
{
public class MachineKeyCryptoSystemTest
{
private static readonly MachineKeyCryptoSystem _dummyCryptoSystem = new MachineKeyCryptoSystem(HexEncoder, HexDecoder);
[Fact]
public void Base64ToHex()
{
// Arrange
string base64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_0";
string hex = "00108310518720928B30D38F41149351559761969B71D79F8218A39259A7A29AABB2DBAFC31CB3D35DB7E39EBBF3DFBF";
// Act
string retVal = MachineKeyCryptoSystem.Base64ToHex(base64);
// Assert
Assert.Equal(hex, retVal);
}
[Fact]
public void Base64ToHex_HexToBase64_RoundTrips()
{
for (int i = 0; i <= Byte.MaxValue; i++)
{
// Arrange
string hex = String.Format("{0:X2}", i);
// Act
string retVal = MachineKeyCryptoSystem.Base64ToHex(MachineKeyCryptoSystem.HexToBase64(hex));
// Assert
Assert.Equal(hex, retVal);
}
}
[Fact]
public void HexToBase64()
{
// Arrange
string base64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_0";
string hex = "00108310518720928B30D38F41149351559761969B71D79F8218A39259A7A29AABB2DBAFC31CB3D35DB7E39EBBF3DFBF";
// Act
string retVal = MachineKeyCryptoSystem.HexToBase64(hex);
// Assert
Assert.Equal(base64, retVal);
}
[Fact]
public void Protect()
{
// Arrange
byte[] data = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05 };
// Act
string retVal = _dummyCryptoSystem.Protect(data);
// Assert
Assert.Equal("hYfyZgECAwQF0", retVal);
}
[Theory]
[InlineData("hYfyZwECAwQF0")] // bad MagicHeader
[InlineData("hYfy0")] // too short to contain MagicHeader
public void Unprotect_Failure(string protectedData)
{
// Act
byte[] retVal = _dummyCryptoSystem.Unprotect(protectedData);
// Assert
Assert.Null(retVal);
}
[Fact]
public void Unprotect_Success()
{
// Arrange
byte[] expected = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05 };
// Act
byte[] retVal = _dummyCryptoSystem.Unprotect("hYfyZgECAwQF0");
// Assert
Assert.Equal(expected, retVal);
}
[Fact]
public void Protect_Unprotect_RoundTrips()
{
// Arrange
byte[] data = new byte[1024];
new Random().NextBytes(data);
// Act
byte[] roundTripped = _dummyCryptoSystem.Unprotect(_dummyCryptoSystem.Protect(data));
// Assert
Assert.Equal(data, roundTripped);
}
private static string HexEncoder(byte[] data, MachineKeyProtection protection)
{
Assert.Equal(MachineKeyProtection.All, protection);
return HexUtil.HexEncode(data);
}
private static byte[] HexDecoder(string input, MachineKeyProtection protection)
{
Assert.Equal(MachineKeyProtection.All, protection);
return HexUtil.HexDecode(input);
}
}
}
|
