mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
91 lines
3.3 KiB
C++
91 lines
3.3 KiB
C++
// Protocol Buffers - Google's data interchange format
|
|
// Copyright 2008 Google Inc. All rights reserved.
|
|
// https://developers.google.com/protocol-buffers/
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions are
|
|
// met:
|
|
//
|
|
// * Redistributions of source code must retain the above copyright
|
|
// notice, this list of conditions and the following disclaimer.
|
|
// * Redistributions in binary form must reproduce the above
|
|
// copyright notice, this list of conditions and the following disclaimer
|
|
// in the documentation and/or other materials provided with the
|
|
// distribution.
|
|
// * Neither the name of Google Inc. nor the names of its
|
|
// contributors may be used to endorse or promote products derived from
|
|
// this software without specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
#include <thirdparty/protobuf/arena_test_util.h>
|
|
#include <thirdparty/protobuf/map_lite_test_util.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
namespace google {
|
|
namespace protobuf {
|
|
namespace {
|
|
|
|
class LiteArenaTest : public testing::Test {
|
|
protected:
|
|
LiteArenaTest() {
|
|
ArenaOptions options;
|
|
options.start_block_size = 128 * 1024;
|
|
options.max_block_size = 128 * 1024;
|
|
arena_.reset(new Arena(options));
|
|
// Trigger the allocation of the first arena block, so that further use of
|
|
// the arena will not require any heap allocations.
|
|
Arena::CreateArray<char>(arena_.get(), 1);
|
|
}
|
|
|
|
std::unique_ptr<Arena> arena_;
|
|
};
|
|
|
|
TEST_F(LiteArenaTest, MapNoHeapAllocation) {
|
|
std::string data;
|
|
data.reserve(128 * 1024);
|
|
|
|
{
|
|
// TODO(teboring): Enable no heap check when ArenaStringPtr is used in
|
|
// Map.
|
|
// internal::NoHeapChecker no_heap;
|
|
|
|
protobuf_unittest::TestArenaMapLite* from =
|
|
Arena::CreateMessage<protobuf_unittest::TestArenaMapLite>(arena_.get());
|
|
MapLiteTestUtil::SetArenaMapFields(from);
|
|
from->SerializeToString(&data);
|
|
|
|
protobuf_unittest::TestArenaMapLite* to =
|
|
Arena::CreateMessage<protobuf_unittest::TestArenaMapLite>(arena_.get());
|
|
to->ParseFromString(data);
|
|
MapLiteTestUtil::ExpectArenaMapFieldsSet(*to);
|
|
}
|
|
}
|
|
|
|
TEST_F(LiteArenaTest, UnknownFieldMemLeak) {
|
|
protobuf_unittest::ForeignMessageArenaLite* message =
|
|
Arena::CreateMessage<protobuf_unittest::ForeignMessageArenaLite>(
|
|
arena_.get());
|
|
std::string data = "\012\000";
|
|
int original_capacity = data.capacity();
|
|
while (data.capacity() <= original_capacity) {
|
|
data.append("a");
|
|
}
|
|
data[1] = data.size() - 2;
|
|
message->ParseFromString(data);
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace protobuf
|
|
} // namespace google
|