xref: /aosp_15_r20/external/perfetto/src/trace_processor/importers/archive/gzip_trace_parser.h (revision 6dbdd20afdafa5e3ca9b8809fa73465d530080dc)
1 /*
2  * Copyright (C) 2019 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 #ifndef SRC_TRACE_PROCESSOR_IMPORTERS_ARCHIVE_GZIP_TRACE_PARSER_H_
18 #define SRC_TRACE_PROCESSOR_IMPORTERS_ARCHIVE_GZIP_TRACE_PARSER_H_
19 
20 #include <cstddef>
21 #include <cstdint>
22 #include <memory>
23 
24 #include "perfetto/base/status.h"
25 #include "src/trace_processor/importers/common/chunked_trace_reader.h"
26 #include "src/trace_processor/util/gzip_utils.h"
27 
28 namespace perfetto::trace_processor {
29 
30 class TraceProcessorContext;
31 
32 class GzipTraceParser : public ChunkedTraceReader {
33  public:
34   explicit GzipTraceParser(TraceProcessorContext*);
35   explicit GzipTraceParser(std::unique_ptr<ChunkedTraceReader>);
36   ~GzipTraceParser() override;
37 
38   // ChunkedTraceReader implementation
39   base::Status Parse(TraceBlobView) override;
40   base::Status NotifyEndOfFile() override;
41 
42   base::Status ParseUnowned(const uint8_t*, size_t);
43 
44  private:
45   TraceProcessorContext* const context_;
46   util::GzipDecompressor decompressor_;
47   std::unique_ptr<ChunkedTraceReader> inner_;
48 
49   std::unique_ptr<uint8_t[]> buffer_;
50   size_t bytes_written_ = 0;
51 
52   bool first_chunk_parsed_ = false;
53   enum { kStreamBoundary, kMidStream } output_state_ = kStreamBoundary;
54 };
55 
56 }  // namespace perfetto::trace_processor
57 
58 #endif  // SRC_TRACE_PROCESSOR_IMPORTERS_ARCHIVE_GZIP_TRACE_PARSER_H_
59