/* SPDX-License-Identifier: BSD-3-Clause */
/* Copyright(c) 2007-2022 Intel Corporation */
#include <sys/types.h>
#include <sys/sysctl.h>
#include <sys/systm.h>
#include "adf_heartbeat_dbg.h"
#include "adf_common_drv.h"
#include "adf_cfg.h"
#include "adf_heartbeat.h"

#define HB_SYSCTL_ERR(RC)                                                           \
	do {                                                                        \
		if (RC == NULL) {                                                   \
			printf(                                                     \
			    "Memory allocation failed in adf_heartbeat_dbg_add\n"); \
			return ENOMEM;                                              \
		}                                                                   \
	} while (0)

/* Handler for HB status check */
static int qat_dev_hb_read(SYSCTL_HANDLER_ARGS)
{
	enum adf_device_heartbeat_status hb_status = DEV_HB_UNRESPONSIVE;
	struct adf_accel_dev *accel_dev = arg1;
	struct adf_heartbeat *hb;
	int ret = 0;
	if (accel_dev == NULL) {
		return EINVAL;
	}
	hb = accel_dev->heartbeat;

	/* if FW is loaded, proceed else set heartbeat down */
	if (test_bit(ADF_STATUS_AE_UCODE_LOADED, &accel_dev->status)) {
		adf_heartbeat_status(accel_dev, &hb_status);
	}
	if (hb_status == DEV_HB_ALIVE) {
		hb->heartbeat.hb_sysctlvar = 1;
	} else {
		hb->heartbeat.hb_sysctlvar = 0;
	}
	ret = sysctl_handle_int(oidp, &hb->heartbeat.hb_sysctlvar, 0, req);
	return ret;
}

int
adf_heartbeat_dbg_add(struct adf_accel_dev *accel_dev)
{
	struct sysctl_ctx_list *qat_hb_sysctl_ctx;
	struct sysctl_oid *qat_hb_sysctl_tree;
	struct adf_heartbeat *hb;
	struct sysctl_oid *rc = 0;

	if (accel_dev == NULL) {
		return EINVAL;
	}

	if (adf_heartbeat_init(accel_dev))
		return EINVAL;

	hb = accel_dev->heartbeat;
	qat_hb_sysctl_ctx =
	    device_get_sysctl_ctx(accel_dev->accel_pci_dev.pci_dev);
	qat_hb_sysctl_tree =
	    device_get_sysctl_tree(accel_dev->accel_pci_dev.pci_dev);

	rc = SYSCTL_ADD_UINT(qat_hb_sysctl_ctx,
			     SYSCTL_CHILDREN(qat_hb_sysctl_tree),
			     OID_AUTO,
			     "heartbeat_sent",
			     CTLFLAG_RD,
			     &hb->hb_sent_counter,
			     0,
			     "HB sent count");
	HB_SYSCTL_ERR(rc);

	rc = SYSCTL_ADD_UINT(qat_hb_sysctl_ctx,
			     SYSCTL_CHILDREN(qat_hb_sysctl_tree),
			     OID_AUTO,
			     "heartbeat_failed",
			     CTLFLAG_RD,
			     &hb->hb_failed_counter,
			     0,
			     "HB failed count");
	HB_SYSCTL_ERR(rc);

	rc = SYSCTL_ADD_PROC(qat_hb_sysctl_ctx,
			     SYSCTL_CHILDREN(qat_hb_sysctl_tree),
			     OID_AUTO,
			     "heartbeat",
			     CTLTYPE_INT | CTLFLAG_RD,
			     accel_dev,
			     0,
			     qat_dev_hb_read,
			     "IU",
			     "QAT device status");
	HB_SYSCTL_ERR(rc);
	return 0;
}

int
adf_heartbeat_dbg_del(struct adf_accel_dev *accel_dev)
{
	adf_heartbeat_clean(accel_dev);
	return 0;
}