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 "src/trace_processor/util/trace_type.h"
20 #include "test/gtest_and_gmock.h"
21
22 namespace perfetto {
23 namespace trace_processor {
24 namespace {
25
TEST(TraceProcessorImplTest,GuessTraceType_Empty)26 TEST(TraceProcessorImplTest, GuessTraceType_Empty) {
27 const uint8_t prefix[] = "";
28 EXPECT_EQ(kUnknownTraceType, GuessTraceType(prefix, 0));
29 }
30
TEST(TraceProcessorImplTest,GuessTraceType_Json)31 TEST(TraceProcessorImplTest, GuessTraceType_Json) {
32 const uint8_t prefix[] = "{\"traceEvents\":[";
33 EXPECT_EQ(kJsonTraceType, GuessTraceType(prefix, sizeof(prefix)));
34 }
35
TEST(TraceProcessorImplTest,GuessTraceType_Ninja)36 TEST(TraceProcessorImplTest, GuessTraceType_Ninja) {
37 const uint8_t prefix[] = "# ninja log v5\n";
38 EXPECT_EQ(kNinjaLogTraceType, GuessTraceType(prefix, sizeof(prefix)));
39 }
40
TEST(TraceProcessorImplTest,GuessTraceType_JsonWithSpaces)41 TEST(TraceProcessorImplTest, GuessTraceType_JsonWithSpaces) {
42 const uint8_t prefix[] = "\n{ \"traceEvents\": [";
43 EXPECT_EQ(kJsonTraceType, GuessTraceType(prefix, sizeof(prefix)));
44 }
45
46 // Some Android build traces do not contain the wrapper. See b/118826940
TEST(TraceProcessorImplTest,GuessTraceType_JsonMissingTraceEvents)47 TEST(TraceProcessorImplTest, GuessTraceType_JsonMissingTraceEvents) {
48 const uint8_t prefix[] = "[{\"";
49 EXPECT_EQ(kJsonTraceType, GuessTraceType(prefix, sizeof(prefix)));
50 }
51
TEST(TraceProcessorImplTest,GuessTraceType_DoctypeHtmlUppercase)52 TEST(TraceProcessorImplTest, GuessTraceType_DoctypeHtmlUppercase) {
53 const uint8_t prefix[] = "<!DOCTYPE HTML>";
54 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
55 }
56
TEST(TraceProcessorImplTest,GuessTraceType_DoctypeHtml)57 TEST(TraceProcessorImplTest, GuessTraceType_DoctypeHtml) {
58 const uint8_t prefix[] = "<!doctype html>";
59 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
60 }
61
TEST(TraceProcessorImplTest,GuessTraceType_DoctypeHtmlMixed)62 TEST(TraceProcessorImplTest, GuessTraceType_DoctypeHtmlMixed) {
63 const uint8_t prefix[] = "<!DoCTyPe HtMl>";
64 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
65 }
66
TEST(TraceProcessorImplTest,GuessTraceType_Html)67 TEST(TraceProcessorImplTest, GuessTraceType_Html) {
68 const uint8_t prefix[] = "<html>";
69 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
70 }
71
TEST(TraceProcessorImplTest,GuessTraceType_HtmlUpper)72 TEST(TraceProcessorImplTest, GuessTraceType_HtmlUpper) {
73 const uint8_t prefix[] = "<HTML>";
74 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
75 }
76
TEST(TraceProcessorImplTest,GuessTraceType_HtmlMixed)77 TEST(TraceProcessorImplTest, GuessTraceType_HtmlMixed) {
78 const uint8_t prefix[] = "<htmL>";
79 EXPECT_EQ(kSystraceTraceType, GuessTraceType(prefix, sizeof(prefix)));
80 }
81
TEST(TraceProcessorImplTest,GuessTraceType_Proto)82 TEST(TraceProcessorImplTest, GuessTraceType_Proto) {
83 const uint8_t prefix[] = {0x0a, 0x00}; // An empty TracePacket.
84 EXPECT_EQ(kProtoTraceType, GuessTraceType(prefix, sizeof(prefix)));
85 }
86
TEST(TraceProcessorImplTest,GuessTraceType_Fuchsia)87 TEST(TraceProcessorImplTest, GuessTraceType_Fuchsia) {
88 const uint8_t prefix[] = {0x10, 0x00, 0x04, 0x46, 0x78, 0x54, 0x16, 0x00};
89 EXPECT_EQ(kFuchsiaTraceType, GuessTraceType(prefix, sizeof(prefix)));
90 }
91
TEST(TraceProcessorImplTest,GuessTraceType_Bmp)92 TEST(TraceProcessorImplTest, GuessTraceType_Bmp) {
93 const uint8_t prefix[] = {0x42, 0x4d, 0x1e, 0x00, 0x00, 0x00, 0x00,
94 0x00, 0x00, 0x00, 0x1a, 0x00, 0x00, 0x00,
95 0x0c, 0x00, 0x00, 0x00, 0x01, 0x00};
96 EXPECT_EQ(kUnknownTraceType, GuessTraceType(prefix, sizeof(prefix)));
97 }
98
99 } // namespace
100 } // namespace trace_processor
101 } // namespace perfetto
102