DobbyStats.h

/*
* If not stated otherwise in this file or this component's LICENSE file the
* following copyright and licenses apply:
*
* Copyright 2016 Sky UK
*
* 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.
*/
/*
 * File:   DobbyStats.h
 *
 */
#ifndef DOBBYSTATS_H
#define DOBBYSTATS_H
 
#include <ContainerId.h>
#include <IDobbyEnv.h>
#include "IDobbyUtils.h"
 
#include <memory>
#include <string>
 
#if defined(RDK)
#include <json/json.h>
#else
#include <jsoncpp/json.h>
#endif
 
class IDobbyEnv;
 
// -----------------------------------------------------------------------------
class DobbyStats
{
public:
    DobbyStats(const ContainerId &id,
               const std::shared_ptr<IDobbyEnv> &env,
               const std::shared_ptr<IDobbyUtils> &utils);
    ~DobbyStats();
 
public:
    const Json::Value &stats() const;
 
private:
    typedef struct Process
    {
        const pid_t pid;
        const pid_t nsPid;
        const std::string fileName;
        const std::string cmdline;
 
        inline void Serialise(Json::Value &root)
        {
            root["pid"] = pid;
            root["nsPid"] = nsPid;
            root["executable"] = fileName;
            root["cmdline"] = cmdline;
        }
    } Process;
 
private:
    static ssize_t readCgroupFile(const ContainerId &id,
                                  const std::string &cgroupMntPath,
                                  const std::string &cgroupfileName,
                                  char *buf, size_t bufLen);
 
    static Json::Value readSingleCgroupValue(const ContainerId &id,
                                             const std::string &cgroupMntPath,
                                             const std::string &cgroupfileName);
 
    static std::vector<int64_t> readMultipleCgroupValues(const ContainerId &id,
                                                         const std::string &cgroupMntPath,
                                                         const std::string &cgroupfileName);
 
    static Json::Value readMultipleCgroupValuesJson(const ContainerId &id,
                                                    const std::string &cgroupMntPath,
                                                    const std::string &cgroupfileName);
 
    static Json::Value getStats(const ContainerId &id,
                                const std::shared_ptr<IDobbyEnv> &env,
                                const std::shared_ptr<IDobbyUtils> &utils);
 
    static Json::Value getProcessTree(const ContainerId &id,
                                      const std::string &cpuCgroupMntPath,
                                      const std::shared_ptr<IDobbyUtils> &utils);
 
#if defined(RDK)
    static Json::Value readIonCgroupHeaps(const ContainerId &id,
                                          const std::string &ionCgroupPath);
#endif
 
    static Process getProcessInfo(pid_t pid,
                                  const std::shared_ptr<IDobbyUtils> &utils);
 
    static pid_t readNsPidFromProc(pid_t pid);
 
private:
    const Json::Value mStats;
};
 
#endif // !defined(DOBBYSTATS_H)