1 /*
2 * Copyright (C) 2018 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "src/trace_processor/forwarding_trace_parser.h"
18
19 #include "test/gtest_and_gmock.h"
20
21 namespace perfetto {
22 namespace trace_processor {
23 namespace {
24
TEST(TraceProcessorImplTest,GuessTraceType_Empty)25 TEST(TraceProcessorImplTest, GuessTraceType_Empty) {
26 const uint8_t prefix[] = "";
27 EXPECT_EQ(kUnknownTraceType, GuessTraceType(prefix, 0));
28 }
29
TEST(TraceProcessorImplTest,GuessTraceType_Json)30 TEST(TraceProcessorImplTest, GuessTraceType_Json) {
31 const uint8_t prefix[] = "{\"traceEvents\":[";
32 EXPECT_EQ(kJsonTraceType, GuessTraceType(prefix, sizeof(prefix)));
33 }
34
TEST(TraceProcessorImplTest,GuessTraceType_Ninja)35 TEST(TraceProcessorImplTest, GuessTraceType_Ninja) {
36 const uint8_t prefix[] = "# ninja log v5\n";
37 EXPECT_EQ(kNinjaLogTraceType, GuessTraceType(prefix, sizeof(prefix)));
38 }
39
TEST(TraceProcessorImplTest,GuessTraceType_JsonWithSpaces)40 TEST(TraceProcessorImplTest, GuessTraceType_JsonWithSpaces) {
41 const uint8_t prefix[] = "\n{ \"traceEvents\": [";
42 EXPECT_EQ(kJsonTraceType, GuessTraceType(prefix, sizeof(prefix)));
43 }
44
45 // Some Android build traces do not contain the wrapper. See b/118826940
TEST(TraceProcessorImplTest,GuessTraceType_JsonMissingTraceEvents)46 TEST(TraceProcessorImplTest, GuessTraceType_JsonMissingTraceEvents) {
47 const uint8_t prefix[] = "[{\"";
48 EXPECT_EQ(kJsonTraceType, GuessTraceType(prefix, sizeof(prefix)));
49 }
50
TEST(TraceProcessorImplTest,GuessTraceType_DoctypeHtmlUppercase)51 TEST(TraceProcessorImplTest, GuessTraceType_DoctypeHtmlUppercase) {
52 const uint8_t prefix[] = "<!DOCTYPE HTML>";
53 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
54 }
55
TEST(TraceProcessorImplTest,GuessTraceType_DoctypeHtml)56 TEST(TraceProcessorImplTest, GuessTraceType_DoctypeHtml) {
57 const uint8_t prefix[] = "<!doctype html>";
58 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
59 }
60
TEST(TraceProcessorImplTest,GuessTraceType_DoctypeHtmlMixed)61 TEST(TraceProcessorImplTest, GuessTraceType_DoctypeHtmlMixed) {
62 const uint8_t prefix[] = "<!DoCTyPe HtMl>";
63 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
64 }
65
TEST(TraceProcessorImplTest,GuessTraceType_Html)66 TEST(TraceProcessorImplTest, GuessTraceType_Html) {
67 const uint8_t prefix[] = "<html>";
68 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
69 }
70
TEST(TraceProcessorImplTest,GuessTraceType_HtmlUpper)71 TEST(TraceProcessorImplTest, GuessTraceType_HtmlUpper) {
72 const uint8_t prefix[] = "<HTML>";
73 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
74 }
75
TEST(TraceProcessorImplTest,GuessTraceType_HtmlMixed)76 TEST(TraceProcessorImplTest, GuessTraceType_HtmlMixed) {
77 const uint8_t prefix[] = "<htmL>";
78 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
79 }
80
TEST(TraceProcessorImplTest,GuessTraceType_Proto)81 TEST(TraceProcessorImplTest, GuessTraceType_Proto) {
82 const uint8_t prefix[] = {0x0a, 0x00}; // An empty TracePacket.
83 EXPECT_EQ(kProtoTraceType, GuessTraceType(prefix, sizeof(prefix)));
84 }
85
TEST(TraceProcessorImplTest,GuessTraceType_Fuchsia)86 TEST(TraceProcessorImplTest, GuessTraceType_Fuchsia) {
87 const uint8_t prefix[] = {0x10, 0x00, 0x04, 0x46, 0x78, 0x54, 0x16, 0x00};
88 EXPECT_EQ(kFuchsiaTraceType, GuessTraceType(prefix, sizeof(prefix)));
89 }
90
TEST(TraceProcessorImplTest,GuessTraceType_Bmp)91 TEST(TraceProcessorImplTest, GuessTraceType_Bmp) {
92 const uint8_t prefix[] = {0x42, 0x4d, 0x1e, 0x00, 0x00, 0x00, 0x00,
93 0x00, 0x00, 0x00, 0x1a, 0x00, 0x00, 0x00,
94 0x0c, 0x00, 0x00, 0x00, 0x01, 0x00};
95 EXPECT_EQ(kUnknownTraceType, GuessTraceType(prefix, sizeof(prefix)));
96 }
97
98 } // namespace
99 } // namespace trace_processor
100 } // namespace perfetto
101