/********************************************************************************
 *
 * Copyright (c) 2018 ROCm Developer Tools
 *
 * MIT LICENSE:
 * Permission is hereby granted, free of charge, to any person obtaining a copy of
 * this software and associated documentation files (the "Software"), to deal in
 * the Software without restriction, including without limitation the rights to
 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
 * of the Software, and to permit persons to whom the Software is furnished to do
 * so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 *******************************************************************************/
#include "include/rvs_module.h"

#include <pci/pci.h>
#include <unistd.h>
#include <iostream>

#include "include/gpu_util.h"
#include "include/rvsloglp.h"
#include "include/worker.h"
#include "include/action.h"

/**
 * @defgroup PESM PESM Module
 *
 * @brief PCIe State Monitoring module
 *
 * The PCIe State Monitor tool is used to actively monitor the PCIe interconnect between the host
 * platform and the GPU. The module will register a “listener” on a target GPU’s PCIe
 * interconnect, and log a message whenever it detects a state change. The PESM will be able to
 * detect the following state changes:
 *   - 1.2.PCIe link speed changes
 *   - GPU power state changes
 */

Worker* pworker;

extern "C" int rvs_module_has_interface(int iid) {
  int sts = 0;
  switch (iid) {
  case 0:
  case 1:
    sts = 1;
  }
  return sts;
}

extern "C" const char* rvs_module_get_description(void) {
  return "ROCm Validation Suite PESM module";
}

extern "C" const char* rvs_module_get_config(void) {
  return "monitor (bool)";
}

extern "C" const char* rvs_module_get_output(void) {
  return "state (string)";
}

extern "C" int   rvs_module_init(void* pMi) {
  pworker = nullptr;
  rvs::lp::Initialize(static_cast<T_MODULE_INIT*>(pMi));
  rvs::gpulist::Initialize();
  return 0;
}

extern "C" int   rvs_module_terminate(void) {
  rvs::lp::Log("[module_terminate] pesm rvs_module_terminate() - entered",
               rvs::logtrace);
  if (pworker) {
    rvs::lp::Log(
      "[module_terminate] pesm rvs_module_terminate() - pworker exists",
                 rvs::logtrace);
    pworker->set_stop_name("module_terminate");
    pworker->stop();
    delete pworker;
    pworker = nullptr;
    rvs::lp::Log(
      "[module_terminate] pesm rvs_module_terminate() - monitoring stopped",
                 rvs::logtrace);
  }
  return 0;
}

extern "C" void* rvs_module_action_create(void) {
  return static_cast<void*>(new pesm_action);
}

extern "C" int   rvs_module_action_destroy(void* pAction) {
  delete static_cast<rvs::actionbase*>(pAction);
  return 0;
}

extern "C" int rvs_module_action_property_set(
  void* pAction, const char* Key, const char* Val) {
  return static_cast<rvs::actionbase*>(pAction)->property_set(Key, Val);
}

extern "C" int rvs_module_action_run(void* pAction) {
  return static_cast<rvs::actionbase*>(pAction)->run();
}