1*598139dcSAndroid Build Coastguard Worker /*
2*598139dcSAndroid Build Coastguard Worker * Copyright (C) 2020 The Android Open Source Project
3*598139dcSAndroid Build Coastguard Worker *
4*598139dcSAndroid Build Coastguard Worker * Licensed under the Apache License, Version 2.0 (the "License");
5*598139dcSAndroid Build Coastguard Worker * you may not use this file except in compliance with the License.
6*598139dcSAndroid Build Coastguard Worker * You may obtain a copy of the License at
7*598139dcSAndroid Build Coastguard Worker *
8*598139dcSAndroid Build Coastguard Worker * http://www.apache.org/licenses/LICENSE-2.0
9*598139dcSAndroid Build Coastguard Worker *
10*598139dcSAndroid Build Coastguard Worker * Unless required by applicable law or agreed to in writing, software
11*598139dcSAndroid Build Coastguard Worker * distributed under the License is distributed on an "AS IS" BASIS,
12*598139dcSAndroid Build Coastguard Worker * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*598139dcSAndroid Build Coastguard Worker * See the License for the specific language governing permissions and
14*598139dcSAndroid Build Coastguard Worker * limitations under the License.
15*598139dcSAndroid Build Coastguard Worker */
16*598139dcSAndroid Build Coastguard Worker
17*598139dcSAndroid Build Coastguard Worker #include "SerializedLogChunk.h"
18*598139dcSAndroid Build Coastguard Worker
19*598139dcSAndroid Build Coastguard Worker #include <limits>
20*598139dcSAndroid Build Coastguard Worker
21*598139dcSAndroid Build Coastguard Worker #include <android-base/silent_death_test.h>
22*598139dcSAndroid Build Coastguard Worker #include <android-base/stringprintf.h>
23*598139dcSAndroid Build Coastguard Worker #include <android/log.h>
24*598139dcSAndroid Build Coastguard Worker #include <gtest/gtest.h>
25*598139dcSAndroid Build Coastguard Worker
26*598139dcSAndroid Build Coastguard Worker using SerializedLogChunk_DeathTest = SilentDeathTest;
27*598139dcSAndroid Build Coastguard Worker
28*598139dcSAndroid Build Coastguard Worker using android::base::StringPrintf;
29*598139dcSAndroid Build Coastguard Worker
TEST(SerializedLogChunk,smoke)30*598139dcSAndroid Build Coastguard Worker TEST(SerializedLogChunk, smoke) {
31*598139dcSAndroid Build Coastguard Worker size_t chunk_size = 10 * 4096;
32*598139dcSAndroid Build Coastguard Worker auto chunk = SerializedLogChunk{chunk_size};
33*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(chunk_size + sizeof(SerializedLogChunk), chunk.PruneSize());
34*598139dcSAndroid Build Coastguard Worker
35*598139dcSAndroid Build Coastguard Worker static const char log_message[] = "log message";
36*598139dcSAndroid Build Coastguard Worker size_t expected_total_len = sizeof(SerializedLogEntry) + sizeof(log_message);
37*598139dcSAndroid Build Coastguard Worker ASSERT_TRUE(chunk.CanLog(expected_total_len));
38*598139dcSAndroid Build Coastguard Worker EXPECT_TRUE(chunk.CanLog(chunk_size));
39*598139dcSAndroid Build Coastguard Worker EXPECT_FALSE(chunk.CanLog(chunk_size + 1));
40*598139dcSAndroid Build Coastguard Worker
41*598139dcSAndroid Build Coastguard Worker log_time time(CLOCK_REALTIME);
42*598139dcSAndroid Build Coastguard Worker auto* entry = chunk.Log(1234, time, 0, 1, 2, log_message, sizeof(log_message));
43*598139dcSAndroid Build Coastguard Worker ASSERT_NE(nullptr, entry);
44*598139dcSAndroid Build Coastguard Worker
45*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(1234U, entry->sequence());
46*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(time, entry->realtime());
47*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(0U, entry->uid());
48*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(1, entry->pid());
49*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(2, entry->tid());
50*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(sizeof(log_message), entry->msg_len());
51*598139dcSAndroid Build Coastguard Worker EXPECT_STREQ(log_message, entry->msg());
52*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(expected_total_len, entry->total_len());
53*598139dcSAndroid Build Coastguard Worker
54*598139dcSAndroid Build Coastguard Worker EXPECT_FALSE(chunk.CanLog(chunk_size));
55*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(static_cast<int>(expected_total_len), chunk.write_offset());
56*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(1234U, chunk.highest_sequence_number());
57*598139dcSAndroid Build Coastguard Worker }
58*598139dcSAndroid Build Coastguard Worker
TEST(SerializedLogChunk,fill_log_exactly)59*598139dcSAndroid Build Coastguard Worker TEST(SerializedLogChunk, fill_log_exactly) {
60*598139dcSAndroid Build Coastguard Worker static const char log_message[] = "this is a log message";
61*598139dcSAndroid Build Coastguard Worker size_t individual_message_size = sizeof(SerializedLogEntry) + sizeof(log_message);
62*598139dcSAndroid Build Coastguard Worker size_t chunk_size = individual_message_size * 3;
63*598139dcSAndroid Build Coastguard Worker auto chunk = SerializedLogChunk{chunk_size};
64*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(chunk_size + sizeof(SerializedLogChunk), chunk.PruneSize());
65*598139dcSAndroid Build Coastguard Worker
66*598139dcSAndroid Build Coastguard Worker ASSERT_TRUE(chunk.CanLog(individual_message_size));
67*598139dcSAndroid Build Coastguard Worker EXPECT_NE(nullptr, chunk.Log(1, log_time(), 1000, 1, 1, log_message, sizeof(log_message)));
68*598139dcSAndroid Build Coastguard Worker
69*598139dcSAndroid Build Coastguard Worker ASSERT_TRUE(chunk.CanLog(individual_message_size));
70*598139dcSAndroid Build Coastguard Worker EXPECT_NE(nullptr, chunk.Log(2, log_time(), 1000, 2, 1, log_message, sizeof(log_message)));
71*598139dcSAndroid Build Coastguard Worker
72*598139dcSAndroid Build Coastguard Worker ASSERT_TRUE(chunk.CanLog(individual_message_size));
73*598139dcSAndroid Build Coastguard Worker EXPECT_NE(nullptr, chunk.Log(3, log_time(), 1000, 3, 1, log_message, sizeof(log_message)));
74*598139dcSAndroid Build Coastguard Worker
75*598139dcSAndroid Build Coastguard Worker EXPECT_FALSE(chunk.CanLog(1));
76*598139dcSAndroid Build Coastguard Worker }
77*598139dcSAndroid Build Coastguard Worker
TEST(SerializedLogChunk,three_logs)78*598139dcSAndroid Build Coastguard Worker TEST(SerializedLogChunk, three_logs) {
79*598139dcSAndroid Build Coastguard Worker size_t chunk_size = 10 * 4096;
80*598139dcSAndroid Build Coastguard Worker auto chunk = SerializedLogChunk{chunk_size};
81*598139dcSAndroid Build Coastguard Worker
82*598139dcSAndroid Build Coastguard Worker chunk.Log(2, log_time(0x1234, 0x5678), 0x111, 0x222, 0x333, "initial message",
83*598139dcSAndroid Build Coastguard Worker strlen("initial message"));
84*598139dcSAndroid Build Coastguard Worker chunk.Log(3, log_time(0x2345, 0x6789), 0x444, 0x555, 0x666, "second message",
85*598139dcSAndroid Build Coastguard Worker strlen("second message"));
86*598139dcSAndroid Build Coastguard Worker auto uint64_t_max = std::numeric_limits<uint64_t>::max();
87*598139dcSAndroid Build Coastguard Worker auto uint32_t_max = std::numeric_limits<uint32_t>::max();
88*598139dcSAndroid Build Coastguard Worker chunk.Log(uint64_t_max, log_time(uint32_t_max, uint32_t_max), uint32_t_max, uint32_t_max,
89*598139dcSAndroid Build Coastguard Worker uint32_t_max, "last message", strlen("last message"));
90*598139dcSAndroid Build Coastguard Worker
91*598139dcSAndroid Build Coastguard Worker static const char expected_buffer_data[] =
92*598139dcSAndroid Build Coastguard Worker "\x11\x01\x00\x00\x22\x02\x00\x00\x33\x03\x00\x00" // UID PID TID
93*598139dcSAndroid Build Coastguard Worker "\x02\x00\x00\x00\x00\x00\x00\x00" // Sequence
94*598139dcSAndroid Build Coastguard Worker "\x34\x12\x00\x00\x78\x56\x00\x00" // Timestamp
95*598139dcSAndroid Build Coastguard Worker "\x0F\x00initial message" // msg_len + message
96*598139dcSAndroid Build Coastguard Worker "\x44\x04\x00\x00\x55\x05\x00\x00\x66\x06\x00\x00" // UID PID TID
97*598139dcSAndroid Build Coastguard Worker "\x03\x00\x00\x00\x00\x00\x00\x00" // Sequence
98*598139dcSAndroid Build Coastguard Worker "\x45\x23\x00\x00\x89\x67\x00\x00" // Timestamp
99*598139dcSAndroid Build Coastguard Worker "\x0E\x00second message" // msg_len + message
100*598139dcSAndroid Build Coastguard Worker "\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF" // UID PID TID
101*598139dcSAndroid Build Coastguard Worker "\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF" // Sequence
102*598139dcSAndroid Build Coastguard Worker "\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF" // Timestamp
103*598139dcSAndroid Build Coastguard Worker "\x0C\x00last message"; // msg_len + message
104*598139dcSAndroid Build Coastguard Worker
105*598139dcSAndroid Build Coastguard Worker for (size_t i = 0; i < sizeof(expected_buffer_data) - 1; ++i) {
106*598139dcSAndroid Build Coastguard Worker EXPECT_EQ(static_cast<uint8_t>(expected_buffer_data[i]), chunk.data()[i])
107*598139dcSAndroid Build Coastguard Worker << "position: " << i;
108*598139dcSAndroid Build Coastguard Worker }
109*598139dcSAndroid Build Coastguard Worker }
110*598139dcSAndroid Build Coastguard Worker
111*598139dcSAndroid Build Coastguard Worker // Check that the CHECK() in DecReaderRefCount() if the ref count goes bad is caught.
TEST_F(SerializedLogChunk_DeathTest,catch_DecCompressedRef_CHECK)112*598139dcSAndroid Build Coastguard Worker TEST_F(SerializedLogChunk_DeathTest, catch_DecCompressedRef_CHECK) {
113*598139dcSAndroid Build Coastguard Worker size_t chunk_size = 10 * 4096;
114*598139dcSAndroid Build Coastguard Worker auto chunk = SerializedLogChunk{chunk_size};
115*598139dcSAndroid Build Coastguard Worker EXPECT_DEATH({ chunk.DecReaderRefCount(); }, "");
116*598139dcSAndroid Build Coastguard Worker }
117*598139dcSAndroid Build Coastguard Worker
118