1/*
2 * Copyright (c) Huawei Technologies Co., Ltd. 2021. All rights reserved.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#ifndef HILOG_PLUGIN_H
17#define HILOG_PLUGIN_H
18
19#include "file_cache.h"
20#include "hilog_plugin_config.pb.h"
21#include "hilog_plugin_result.pb.h"
22#include "logging.h"
23#include "plugin_module_api.h"
24
25#include <atomic>
26#include <mutex>
27#include <thread>
28#include <vector>
29
30class HilogPlugin {
31public:
32    HilogPlugin();
33    ~HilogPlugin();
34    int Start(const uint8_t* configData, uint32_t configSize);
35    int Stop();
36
37    int SetWriter(WriterStruct* writer);
38    void Run(void);
39
40private:
41    std::string GetPidCmd();
42    std::string GetlevelCmd();
43    void InitHilogCmd();
44
45    bool OpenLogFile();
46    int GetDateTime(char* psDateTime, uint32_t size);
47
48    template <typename T> void ParseLogLineInfo(const char* data, size_t len, T& hilogLineInfo);
49
50    template <typename T> void ParseLogLineData(const char* data, size_t len, T hilogInfoProto);
51
52    template <typename T> bool SetHilogLineDetails(const char* data, T& hilogLineInfo);
53
54    bool TimeStringToNS(const char* data, struct timespec *ts);
55    bool FindFirstNum(char** p);
56    bool RemoveSpaces(char** p);
57    bool FindFirstSpace(char** p);
58
59    bool StringToL(const char* word, long& value);
60    // for ut
61    void SetConfig(HilogConfig& config)
62    {
63        protoConfig_ = config;
64        return;
65    }
66
67    template <typename T> void FlushData(const T hilogLineProto);
68    template <typename T> void FlushDataOptimize(const T hilogLineProto);
69
70private:
71    HilogConfig protoConfig_;
72    std::vector<char> protoBuffer_;
73    std::vector<char> dataBuffer_;
74    WriterStruct* resultWriter_ = nullptr;
75    std::mutex mutex_;
76    std::thread workThread_;
77    std::atomic<bool> running_ = true;
78    std::vector<std::string> fullCmd_;
79    std::unique_ptr<FILE, std::function<int (FILE*)>> fp_;
80    std::unique_ptr<FileCache> fileCache_ = nullptr;
81    int pipeFds_[2] = {-1, -1};
82    volatile pid_t childPid_ = -1;
83};
84
85#endif // !HILOG_PLUGIN_H