/* * Copyright (c) Huawei Technologies Co., Ltd. 2021. All rights reserved. * 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. */ #ifndef TRACE_FILE_READER_H #define TRACE_FILE_READER_H #include "logging.h" #include "nocopyable.h" #include "trace_file_helper.h" #include #include #include #include using google::protobuf::MessageLite; class TraceFileReader { public: TraceFileReader() = default; ~TraceFileReader(); bool Open(const std::string& path); long Read(MessageLite& message); long ReadLen(); bool ReadData(uint8_t buffer[], uint32_t bufferSize); bool ValidateHeader(); TraceFileHeader GetHeader() const; private: std::string path_ {}; std::ifstream stream_ {}; TraceFileHeader header_ {}; TraceFileHelper helper_ {}; DISALLOW_COPY_AND_MOVE(TraceFileReader); }; using TraceFileReaderPtr = STD_PTR(shared, TraceFileReader); #endif // !TRACE_FILE_READER_H