blob: 03f15f40a77cab84a070878e47fada311e754577 [file] [log] [blame]
/*
* Copyright (C) 2018 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "src/tracing/core/packet_stream_validator.h"
#include <string>
#include "gtest/gtest.h"
#include "perfetto/trace/trace_packet.pb.h"
namespace perfetto {
namespace {
TEST(PacketStreamValidatorTest, NullPacket) {
std::string ser_buf;
Slices seq;
EXPECT_TRUE(PacketStreamValidator::Validate(seq));
}
TEST(PacketStreamValidatorTest, SimplePacket) {
protos::TracePacket proto;
proto.mutable_for_testing()->set_str("string field");
std::string ser_buf = proto.SerializeAsString();
Slices seq;
seq.emplace_back(&ser_buf[0], ser_buf.size());
EXPECT_TRUE(PacketStreamValidator::Validate(seq));
}
TEST(PacketStreamValidatorTest, ComplexPacket) {
protos::TracePacket proto;
proto.mutable_for_testing()->set_str("string field");
proto.mutable_ftrace_events()->set_cpu(0);
auto* ft = proto.mutable_ftrace_events()->add_event();
ft->set_pid(42);
ft->mutable_sched_switch()->set_prev_comm("tom");
ft->mutable_sched_switch()->set_prev_pid(123);
ft->mutable_sched_switch()->set_next_comm("jerry");
ft->mutable_sched_switch()->set_next_pid(456);
std::string ser_buf = proto.SerializeAsString();
Slices seq;
seq.emplace_back(&ser_buf[0], ser_buf.size());
EXPECT_TRUE(PacketStreamValidator::Validate(seq));
}
TEST(PacketStreamValidatorTest, SimplePacketWithUid) {
protos::TracePacket proto;
proto.set_trusted_uid(123);
std::string ser_buf = proto.SerializeAsString();
Slices seq;
seq.emplace_back(&ser_buf[0], ser_buf.size());
EXPECT_FALSE(PacketStreamValidator::Validate(seq));
}
TEST(PacketStreamValidatorTest, SimplePacketWithZeroUid) {
protos::TracePacket proto;
proto.set_trusted_uid(0);
std::string ser_buf = proto.SerializeAsString();
Slices seq;
seq.emplace_back(&ser_buf[0], ser_buf.size());
EXPECT_FALSE(PacketStreamValidator::Validate(seq));
}
TEST(PacketStreamValidatorTest, SimplePacketWithNegativeOneUid) {
protos::TracePacket proto;
proto.set_trusted_uid(-1);
std::string ser_buf = proto.SerializeAsString();
Slices seq;
seq.emplace_back(&ser_buf[0], ser_buf.size());
EXPECT_FALSE(PacketStreamValidator::Validate(seq));
}
TEST(PacketStreamValidatorTest, ComplexPacketWithUid) {
protos::TracePacket proto;
proto.mutable_for_testing()->set_str("string field");
proto.mutable_ftrace_events()->set_cpu(0);
auto* ft = proto.mutable_ftrace_events()->add_event();
ft->set_pid(42);
ft->mutable_sched_switch()->set_prev_comm("tom");
ft->mutable_sched_switch()->set_prev_pid(123);
ft->mutable_sched_switch()->set_next_comm("jerry");
ft->mutable_sched_switch()->set_next_pid(456);
proto.set_trusted_uid(123);
std::string ser_buf = proto.SerializeAsString();
Slices seq;
seq.emplace_back(&ser_buf[0], ser_buf.size());
EXPECT_FALSE(PacketStreamValidator::Validate(seq));
}
TEST(PacketStreamValidatorTest, FragmentedPacket) {
protos::TracePacket proto;
proto.mutable_for_testing()->set_str("string field");
proto.mutable_ftrace_events()->set_cpu(0);
auto* ft = proto.mutable_ftrace_events()->add_event();
ft->set_pid(42);
ft->mutable_sched_switch()->set_prev_comm("tom");
ft->mutable_sched_switch()->set_prev_pid(123);
ft->mutable_sched_switch()->set_next_comm("jerry");
ft->mutable_sched_switch()->set_next_pid(456);
std::string ser_buf = proto.SerializeAsString();
for (size_t i = 0; i < ser_buf.size(); i++) {
Slices seq;
seq.emplace_back(&ser_buf[0], i);
seq.emplace_back(&ser_buf[i], ser_buf.size() - i);
EXPECT_TRUE(PacketStreamValidator::Validate(seq));
}
}
TEST(PacketStreamValidatorTest, FragmentedPacketWithUid) {
protos::TracePacket proto;
proto.mutable_for_testing()->set_str("string field");
proto.set_trusted_uid(123);
proto.mutable_ftrace_events()->set_cpu(0);
auto* ft = proto.mutable_ftrace_events()->add_event();
ft->set_pid(42);
ft->mutable_sched_switch()->set_prev_comm("tom");
ft->mutable_sched_switch()->set_prev_pid(123);
ft->mutable_sched_switch()->set_next_comm("jerry");
ft->mutable_sched_switch()->set_next_pid(456);
proto.mutable_for_testing()->set_str("foo");
std::string ser_buf = proto.SerializeAsString();
for (size_t i = 0; i < ser_buf.size(); i++) {
Slices seq;
seq.emplace_back(&ser_buf[0], i);
seq.emplace_back(&ser_buf[i], ser_buf.size() - i);
EXPECT_FALSE(PacketStreamValidator::Validate(seq));
}
}
TEST(PacketStreamValidatorTest, TruncatedPacket) {
protos::TracePacket proto;
proto.mutable_for_testing()->set_str("string field");
std::string ser_buf = proto.SerializeAsString();
for (size_t i = 1; i < ser_buf.size(); i++) {
Slices seq;
seq.emplace_back(&ser_buf[0], i);
EXPECT_FALSE(PacketStreamValidator::Validate(seq));
}
}
TEST(PacketStreamValidatorTest, TrailingGarbage) {
protos::TracePacket proto;
proto.mutable_for_testing()->set_str("string field");
std::string ser_buf = proto.SerializeAsString();
ser_buf += "bike is short for bichael";
Slices seq;
seq.emplace_back(&ser_buf[0], ser_buf.size());
EXPECT_FALSE(PacketStreamValidator::Validate(seq));
}
} // namespace
} // namespace perfetto