github
/
aria2
mirror of https://github.com/aria2/aria2
1
0
Fork 0
Code Issues Releases Wiki Activity
aria2/test/PeerConnectionTest.cc

40 lines
997 B
C++
Raw Normal View History

Reserve PeerConnection's buffer capacity according to number of pieces. If the number of pieces gets bigger, the length of Bitfield message payload exceeds the initial buffer capacity of PeerConnection, which is MAX_PAYLOAD_LEN. We expand buffer as necessary so that PeerConnection can receive the Bitfield message.
2012-02-08 16:51:42 +00:00
#include "PeerConnection.h"
#include <cstring>
#include <cppunit/extensions/HelperMacros.h>
#include "Peer.h"
#include "SocketCore.h"
namespace aria2 {
make clang-format using clang-format-3.6
2015-12-27 09:39:47 +00:00
class PeerConnectionTest : public CppUnit::TestFixture {
Reserve PeerConnection's buffer capacity according to number of pieces. If the number of pieces gets bigger, the length of Bitfield message payload exceeds the initial buffer capacity of PeerConnection, which is MAX_PAYLOAD_LEN. We expand buffer as necessary so that PeerConnection can receive the Bitfield message.
2012-02-08 16:51:42 +00:00
CPPUNIT_TEST_SUITE(PeerConnectionTest);
CPPUNIT_TEST(testReserveBuffer);
CPPUNIT_TEST_SUITE_END();
make clang-format using clang-format-3.6
2015-12-27 09:39:47 +00:00
Reserve PeerConnection's buffer capacity according to number of pieces. If the number of pieces gets bigger, the length of Bitfield message payload exceeds the initial buffer capacity of PeerConnection, which is MAX_PAYLOAD_LEN. We expand buffer as necessary so that PeerConnection can receive the Bitfield message.
2012-02-08 16:51:42 +00:00
public:
void testReserveBuffer();
};
CPPUNIT_TEST_SUITE_REGISTRATION(PeerConnectionTest);
make clang-format using clang-format-3.6
2015-12-27 09:39:47 +00:00
void PeerConnectionTest::testReserveBuffer()
{
Require -std=c++11 and use std::shared_ptr instead of SharedHandle
2013-06-21 16:10:38 +00:00
PeerConnection con(1, std::shared_ptr<Peer>(), std::shared_ptr<SocketCore>());
Reserve PeerConnection's buffer capacity according to number of pieces. If the number of pieces gets bigger, the length of Bitfield message payload exceeds the initial buffer capacity of PeerConnection, which is MAX_PAYLOAD_LEN. We expand buffer as necessary so that PeerConnection can receive the Bitfield message.
2012-02-08 16:51:42 +00:00
con.presetBuffer((unsigned char*)"foo", 3);
Rewritten PeerConnection::receiveMessage() The old implementation calls at least 2 read(2) (4bytes length and payload) to receive the message. This change will read as many bytes as possible in one read(2) call. BtPieceMessage::data_ is now just a const pointer to the internal buffer of PeerConnection.
2012-06-25 12:29:48 +00:00
CPPUNIT_ASSERT_EQUAL((size_t)MAX_BUFFER_CAPACITY, con.getBufferCapacity());
Reserve PeerConnection's buffer capacity according to number of pieces. If the number of pieces gets bigger, the length of Bitfield message payload exceeds the initial buffer capacity of PeerConnection, which is MAX_PAYLOAD_LEN. We expand buffer as necessary so that PeerConnection can receive the Bitfield message.
2012-02-08 16:51:42 +00:00
CPPUNIT_ASSERT_EQUAL((size_t)3, con.getBufferLength());
Extend maximum BitTorrent incoming payload size to 64KiB
2016-07-20 13:42:29 +00:00
constexpr size_t newLength = 128_k;
Reserve PeerConnection's buffer capacity according to number of pieces. If the number of pieces gets bigger, the length of Bitfield message payload exceeds the initial buffer capacity of PeerConnection, which is MAX_PAYLOAD_LEN. We expand buffer as necessary so that PeerConnection can receive the Bitfield message.
2012-02-08 16:51:42 +00:00
con.reserveBuffer(newLength);
Rewritten PeerConnection::receiveMessage() The old implementation calls at least 2 read(2) (4bytes length and payload) to receive the message. This change will read as many bytes as possible in one read(2) call. BtPieceMessage::data_ is now just a const pointer to the internal buffer of PeerConnection.
2012-06-25 12:29:48 +00:00
CPPUNIT_ASSERT_EQUAL(newLength, con.getBufferCapacity());
Reserve PeerConnection's buffer capacity according to number of pieces. If the number of pieces gets bigger, the length of Bitfield message payload exceeds the initial buffer capacity of PeerConnection, which is MAX_PAYLOAD_LEN. We expand buffer as necessary so that PeerConnection can receive the Bitfield message.
2012-02-08 16:51:42 +00:00
CPPUNIT_ASSERT_EQUAL((size_t)3, con.getBufferLength());
CPPUNIT_ASSERT(memcmp("foo", con.getBuffer(), 3) == 0);
}
} // namespace aria2