Files
rtorrent/test/rpc/test_object_storage.cc
T
noctuum b1be898ba1 Reject an over-long or empty object_storage key
key_type stores a key of key_size or longer as the empty key.
2026-09-18 16:54:48 +02:00

88 lines
3.6 KiB
C++

#include "config.h"
#include "test/rpc/test_object_storage.h"
#include "helpers/assert.h"
CPPUNIT_TEST_SUITE_REGISTRATION(TestObjectStorage);
void
TestObjectStorage::test_basics() {
rpc::object_storage::iterator itr;
CPPUNIT_ASSERT(m_storage.empty());
itr = m_storage.insert("test_1", torrent::Object("a"), rpc::object_storage::flag_string_type);
CPPUNIT_ASSERT(itr != m_storage.end());
CPPUNIT_ASSERT(&*itr != NULL);
CPPUNIT_ASSERT(itr->first.size() == 6 && std::strcmp(itr->first.data(), "test_1") == 0);
CPPUNIT_ASSERT(itr->second.object.is_string() && itr->second.object.as_string() == "a");
ASSERT_CATCH_INPUT_ERROR( { m_storage.insert("test_1", torrent::Object("a"), rpc::object_storage::flag_string_type); } );
// Test erase.
m_storage.erase(rpc::object_storage::key_type::from_c_str("test_1"));
CPPUNIT_ASSERT(m_storage.empty());
// Test with no type flag.
ASSERT_CATCH_INPUT_ERROR( { m_storage.insert("test_2", torrent::Object("b"), 0); } );
ASSERT_CATCH_INPUT_ERROR( { m_storage.insert("test_3", torrent::Object("c"), ~rpc::object_storage::mask_type); } );
m_storage.clear();
}
void
TestObjectStorage::test_conversions() {
CPPUNIT_ASSERT(m_storage.insert("test_1", torrent::Object("a"), rpc::object_storage::flag_string_type) != m_storage.end());
CPPUNIT_ASSERT(m_storage.insert_str(std::string("test_2"), torrent::Object("a"), rpc::object_storage::flag_string_type) != m_storage.end());
char raw_3[8] = "test_3\x1";
torrent::raw_string raw_string_3(raw_3, 6);
CPPUNIT_ASSERT(m_storage.insert(raw_string_3, torrent::Object("a"), rpc::object_storage::flag_string_type)->first == std::string("test_3"));
m_storage.clear();
}
void
TestObjectStorage::test_validate_keys() {
torrent::raw_string raw_string_4("test_4\0foo", 10);
ASSERT_CATCH_INPUT_ERROR( { m_storage.insert(raw_string_4, torrent::Object("a"), rpc::object_storage::flag_string_type); } );
ASSERT_CATCH_INPUT_ERROR( { m_storage.insert_str("", torrent::Object("a"), rpc::object_storage::flag_string_type); } );
std::string key_max(rpc::object_storage::key_size - 1, 'k');
CPPUNIT_ASSERT(m_storage.insert_str(key_max, torrent::Object("a"), rpc::object_storage::flag_string_type)->first == key_max);
ASSERT_CATCH_INPUT_ERROR( { m_storage.insert_str(key_max + 'k', torrent::Object("a"), rpc::object_storage::flag_string_type); } );
// The over-long key must not have been stored as the empty key.
CPPUNIT_ASSERT(m_storage.find_raw_string(torrent::raw_string::from_c_str("")) == m_storage.end());
m_storage.clear();
}
// And test many other bad/good string combos.
// Test for various conversions of fixed_key_type.
void
TestObjectStorage::test_access() {
m_storage.insert("string_1", torrent::Object("gen_a"), rpc::object_storage::flag_string_type);
m_storage.insert("value_1", int64_t(1), rpc::object_storage::flag_value_type);
CPPUNIT_ASSERT(m_storage.get_c_str("string_1").as_string() == "gen_a");
CPPUNIT_ASSERT(m_storage.set_c_str_string("string_1", "test").as_string() == "test");
// Test value from raw and normal, list, etc.
CPPUNIT_ASSERT(m_storage.get_c_str("value_1").as_value() == 1);
CPPUNIT_ASSERT(m_storage.set_c_str_value("value_1", int64_t(2)).as_value() == 2);
// CPPUNIT_ASSERT(m_storage.set_c_str_value("value_1", "123").as_value() == 123);
// CPPUNIT_ASSERT(m_storage.set_c_str_value("value_1", torrent::raw_string::from_c_str("321")).as_value() == 321);
// CPPUNIT_ASSERT(m_storage.set_c_str_value("value_1", torrent::raw_bencode::from_c_str("i567e")).as_value() == 567);
// ASSERT_CATCH_INPUT_ERROR( { m_storage.set_c_str_value("value_1", "e123"); } );
// Test string from raw and normal, list, etc.
}