Kaladaran/briar
0
0
Fork 0
mirror of https://code.briarproject.org/briar/briar.git synced 2026-02-14 03:39:05 +01:00
Code Issues Projects Releases Wiki Activity

Slightly modified ConnectionWriterImpl to write a full-size frame as

Browse Source 
soon as possible, rather than waiting for the next write.
This commit is contained in:
akwizgran
2011-08-19 20:10:14 +02:00
parent 0132c1eff4
commit 3b32aee6be
2 changed files with 23 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
test/net/sf/briar/transport/ConnectionWriterImplTest.java
View File

@@ -45,21 +45,31 @@ public class ConnectionWriterImplTest extends TransportTest {
}
@Test
public void testFrameIsWrittenAtMaxLength() throws Exception {
public void testWriteByteToMaxLengthWritesFrame() throws Exception {
ByteArrayOutputStream out = new ByteArrayOutputStream();
ConnectionEncrypter e = new NullConnectionEncrypter(out);
ConnectionWriter w = new ConnectionWriterImpl(e, mac);
OutputStream out1 = w.getOutputStream();
// The first maxPayloadLength bytes should be buffered
for(int i = 0; i < maxPayloadLength; i++) out1.write(0);
// The first maxPayloadLength - 1 bytes should be buffered
for(int i = 0; i < maxPayloadLength - 1; i++) out1.write(0);
assertEquals(0, out.size());
// The next byte should trigger the writing of a frame
out1.write(0);
assertEquals(MAX_FRAME_LENGTH, out.size());
// Flushing the stream should write a single-byte frame
out1.flush();
assertEquals(MAX_FRAME_LENGTH + headerLength + 1 + macLength,
out.size());
}
@Test
public void testWriteArrayToMaxLengthWritesFrame() throws Exception {
ByteArrayOutputStream out = new ByteArrayOutputStream();
ConnectionEncrypter e = new NullConnectionEncrypter(out);
ConnectionWriter w = new ConnectionWriterImpl(e, mac);
OutputStream out1 = w.getOutputStream();
// The first maxPayloadLength - 1 bytes should be buffered
out1.write(new byte[maxPayloadLength - 1]);
assertEquals(0, out.size());
// The next maxPayloadLength + 1 bytes should trigger two frames
out1.write(new byte[maxPayloadLength + 1]);
assertEquals(MAX_FRAME_LENGTH * 2, out.size());
}
@Test