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
|
package org.bouncycastle.crypto.tls.test;
import java.security.SecureRandom;
import junit.framework.TestCase;
import org.bouncycastle.crypto.tls.DTLSClientProtocol;
import org.bouncycastle.crypto.tls.DTLSServerProtocol;
import org.bouncycastle.crypto.tls.DTLSTransport;
import org.bouncycastle.crypto.tls.DatagramTransport;
import org.bouncycastle.crypto.tls.ProtocolVersion;
import org.bouncycastle.util.Arrays;
public class DTLSTestCase extends TestCase
{
private static void checkDTLSVersion(ProtocolVersion version)
{
if (version != null && !version.isDTLS())
{
throw new IllegalStateException("Non-DTLS version");
}
}
protected final TlsTestConfig config;
public DTLSTestCase(TlsTestConfig config, String name)
{
checkDTLSVersion(config.clientMinimumVersion);
checkDTLSVersion(config.clientOfferVersion);
checkDTLSVersion(config.serverMaximumVersion);
checkDTLSVersion(config.serverMinimumVersion);
this.config = config;
setName(name);
}
protected void runTest() throws Throwable
{
SecureRandom secureRandom = new SecureRandom();
DTLSClientProtocol clientProtocol = new DTLSClientProtocol(secureRandom);
DTLSServerProtocol serverProtocol = new DTLSServerProtocol(secureRandom);
MockDatagramAssociation network = new MockDatagramAssociation(1500);
TlsTestClientImpl clientImpl = new TlsTestClientImpl(config);
TlsTestServerImpl serverImpl = new TlsTestServerImpl(config);
ServerThread serverThread = new ServerThread(serverProtocol, network.getServer(), serverImpl);
serverThread.start();
Exception caught = null;
try
{
DatagramTransport clientTransport = network.getClient();
if (TlsTestConfig.DEBUG)
{
clientTransport = new LoggingDatagramTransport(clientTransport, System.out);
}
DTLSTransport dtlsClient = clientProtocol.connect(clientImpl, clientTransport);
for (int i = 1; i <= 10; ++i)
{
byte[] data = new byte[i];
Arrays.fill(data, (byte)i);
dtlsClient.send(data, 0, data.length);
}
byte[] buf = new byte[dtlsClient.getReceiveLimit()];
while (dtlsClient.receive(buf, 0, buf.length, 100) >= 0)
{
}
dtlsClient.close();
}
catch (Exception e)
{
caught = e;
logException(caught);
}
serverThread.shutdown();
// TODO Add checks that the various streams were closed
assertEquals("Client fatal alert connection end", config.expectFatalAlertConnectionEnd, clientImpl.firstFatalAlertConnectionEnd);
assertEquals("Server fatal alert connection end", config.expectFatalAlertConnectionEnd, serverImpl.firstFatalAlertConnectionEnd);
assertEquals("Client fatal alert description", config.expectFatalAlertDescription, clientImpl.firstFatalAlertDescription);
assertEquals("Server fatal alert description", config.expectFatalAlertDescription, serverImpl.firstFatalAlertDescription);
if (config.expectFatalAlertConnectionEnd == -1)
{
assertNull("Unexpected client exception", caught);
assertNull("Unexpected server exception", serverThread.caught);
}
}
protected void logException(Exception e)
{
if (TlsTestConfig.DEBUG)
{
e.printStackTrace();
}
}
class ServerThread
extends Thread
{
private final DTLSServerProtocol serverProtocol;
private final DatagramTransport serverTransport;
private final TlsTestServerImpl serverImpl;
private volatile boolean isShutdown = false;
Exception caught = null;
ServerThread(DTLSServerProtocol serverProtocol, DatagramTransport serverTransport, TlsTestServerImpl serverImpl)
{
this.serverProtocol = serverProtocol;
this.serverTransport = serverTransport;
this.serverImpl = serverImpl;
}
public void run()
{
try
{
DTLSTransport dtlsServer = serverProtocol.accept(serverImpl, serverTransport);
byte[] buf = new byte[dtlsServer.getReceiveLimit()];
while (!isShutdown)
{
int length = dtlsServer.receive(buf, 0, buf.length, 100);
if (length >= 0)
{
dtlsServer.send(buf, 0, length);
}
}
dtlsServer.close();
}
catch (Exception e)
{
caught = e;
logException(caught);
}
}
void shutdown()
throws InterruptedException
{
if (!isShutdown)
{
isShutdown = true;
this.interrupt();
this.join();
}
}
}
}
|
