mirrors/aria2
1
0
Fork 0
mirror of https://github.com/aria2/aria2.git synced 2025-04-05 21:47:37 +03:00
Code Issues Projects Releases Packages Wiki Activity

Added stream Bencode parser

Browse source
This commit is contained in:
Tatsuhiro Tsujikawa 2012-07-24 23:50:06 +09:00
parent 9620bb0a6d
commit 8311d6ef87
6 changed files with 769 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

200
test/ValueBaseBencodeParserTest.cc Normal file
View file

@ -0,0 +1,200 @@
#include "ValueBaseBencodeParser.h"
#include <cppunit/extensions/HelperMacros.h>
#include "ValueBase.h"
namespace aria2 {
class ValueBaseBencodeParserTest:public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(ValueBaseBencodeParserTest);
CPPUNIT_TEST(testParseUpdate);
CPPUNIT_TEST_SUITE_END();
public:
void testParseUpdate();
};
CPPUNIT_TEST_SUITE_REGISTRATION( ValueBaseBencodeParserTest );
namespace {
void checkDecodeError(const std::string& src)
{
bittorrent::ValueBaseBencodeParser parser;
ssize_t error;
SharedHandle<ValueBase> r = parser.parseFinal(src.c_str(), src.size(),
error);
CPPUNIT_ASSERT(!r);
CPPUNIT_ASSERT(error < 0);
}
} // namespace
void ValueBaseBencodeParserTest::testParseUpdate()
{
bittorrent::ValueBaseBencodeParser parser;
ssize_t error;
{
// empty string
std::string src = "0:";
SharedHandle<ValueBase> s = parser.parseFinal(src.c_str(), src.size(),
error);
CPPUNIT_ASSERT_EQUAL(std::string(""), downcast<String>(s)->s());
}
{
// integer 0
std::string src = "i0e";
SharedHandle<ValueBase> s = parser.parseFinal(src.c_str(), src.size(),
error);
CPPUNIT_ASSERT_EQUAL((int64_t)0, downcast<Integer>(s)->i());
}
{
// empty dict
std::string src = "de";
SharedHandle<ValueBase> d = parser.parseFinal(src.c_str(), src.size(),
error);
CPPUNIT_ASSERT(downcast<Dict>(d)->empty());
}
{
// empty list
std::string src = "le";
SharedHandle<ValueBase> l = parser.parseFinal(src.c_str(), src.size(),
error);
CPPUNIT_ASSERT(downcast<List>(l)->empty());
}
{
// string
std::string src = "3:foo";
SharedHandle<ValueBase> s = parser.parseFinal(src.c_str(), src.size(),
error);
CPPUNIT_ASSERT_EQUAL(std::string("foo"), downcast<String>(s)->s());
}
{
// integer
std::string src = "i9223372036854775807e";
SharedHandle<ValueBase> s = parser.parseFinal(src.c_str(), src.size(),
error);
CPPUNIT_ASSERT_EQUAL((int64_t)9223372036854775807LL,
downcast<Integer>(s)->i());
}
{
// dict, size 1
std::string src = "d3:fooi123ee";
SharedHandle<ValueBase> d = parser.parseFinal(src.c_str(), src.size(),
error);
Dict* dict = downcast<Dict>(d);
CPPUNIT_ASSERT(dict);
CPPUNIT_ASSERT(dict->get("foo"));
CPPUNIT_ASSERT_EQUAL((int64_t)123,
downcast<Integer>(dict->get("foo"))->i());
}
{
// dict, size 2
std::string src = "d3:fooi123e3:bar1:ee";
SharedHandle<ValueBase> d = parser.parseFinal(src.c_str(), src.size(),
error);
Dict* dict = downcast<Dict>(d);
CPPUNIT_ASSERT(dict);
CPPUNIT_ASSERT_EQUAL((size_t)2, dict->size());
CPPUNIT_ASSERT(dict->get("foo"));
CPPUNIT_ASSERT_EQUAL((int64_t)123,
downcast<Integer>(dict->get("foo"))->i());
CPPUNIT_ASSERT(dict->get("bar"));
CPPUNIT_ASSERT_EQUAL(std::string("e"),
downcast<String>(dict->get("bar"))->s());
}
{
// list, size 1
std::string src = "l3:fooe";
SharedHandle<ValueBase> l = parser.parseFinal(src.c_str(), src.size(),
error);
List* list = downcast<List>(l);
CPPUNIT_ASSERT(list);
CPPUNIT_ASSERT_EQUAL((size_t)1, list->size());
CPPUNIT_ASSERT_EQUAL(std::string("foo"),
downcast<String>(list->get(0))->s());
}
{
// list, size 2
std::string src = "l3:fooi123ee";
SharedHandle<ValueBase> l = parser.parseFinal(src.c_str(), src.size(),
error);
List* list = downcast<List>(l);
CPPUNIT_ASSERT(list);
CPPUNIT_ASSERT_EQUAL((size_t)2, list->size());
CPPUNIT_ASSERT_EQUAL(std::string("foo"),
downcast<String>(list->get(0))->s());
CPPUNIT_ASSERT_EQUAL((int64_t)123,
downcast<Integer>(list->get(1))->i());
}
{
// string, integer and list in dict
std::string src = "d4:name5:aria24:sizei12345678900e5:filesl3:bin3:docee";
SharedHandle<ValueBase> r = parser.parseFinal(src.c_str(), src.size(),
error);
const Dict* dict = downcast<Dict>(r);
CPPUNIT_ASSERT(dict);
CPPUNIT_ASSERT_EQUAL(std::string("aria2"),
downcast<String>(dict->get("name"))->s());
CPPUNIT_ASSERT_EQUAL(static_cast<Integer::ValueType>(12345678900LL),
downcast<Integer>(dict->get("size"))->i());
const List* list = downcast<List>(dict->get("files"));
CPPUNIT_ASSERT(list);
CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(2), list->size());
CPPUNIT_ASSERT_EQUAL(std::string("bin"),
downcast<String>(list->get(0))->s());
CPPUNIT_ASSERT_EQUAL(std::string("doc"),
downcast<String>(list->get(1))->s());
}
{
// dict in list
std::string src = "ld1:ki123eee";
SharedHandle<ValueBase> r = parser.parseFinal(src.c_str(), src.size(),
error);
const List* list = downcast<List>(r);
CPPUNIT_ASSERT(list);
CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), list->size());
const Dict* dict = downcast<Dict>(list->get(0));
CPPUNIT_ASSERT(dict);
CPPUNIT_ASSERT_EQUAL(static_cast<Integer::ValueType>(123),
downcast<Integer>(dict->get("k"))->i());
}
{
// empty key is allowed
std::string src = "d0:1:ve";
SharedHandle<ValueBase> s = parser.parseFinal(src.c_str(), src.size(),
error);
}
{
// empty encoded data
std::string src = "";
SharedHandle<ValueBase> s = parser.parseFinal(src.c_str(), src.size(),
error);
CPPUNIT_ASSERT(!s);
}
// integer, without ending 'e'
checkDecodeError("i3");
// dict, without ending 'e'
checkDecodeError("d");
// list, without ending 'e'
checkDecodeError("l");
// string, less than the specified length.
checkDecodeError("3:ab");
// string, but length is invalid
checkDecodeError("x:abc");
// string with minus length
checkDecodeError("-1:a");
// too deep structure
checkDecodeError(std::string(51, 'l')+std::string(51,'e'));
checkDecodeError(std::string(50, 'l')+"d3:fooi100ee"+std::string(50,'e'));
{
// ignore trailing garbage at the end of the input.
std::string src = "5:aria2trail";
SharedHandle<ValueBase> s = parser.parseFinal(src.c_str(), src.size(),
error);
CPPUNIT_ASSERT_EQUAL(std::string("aria2"), downcast<String>(s)->s());
// Get trailing garbage position
CPPUNIT_ASSERT_EQUAL((ssize_t)7, error);
}
}
} // namespace aria2