/*
 * eset_rtp (ESET Real-time file system protection module)
 * Copyright (C) 1992-2025 ESET, spol. s r.o.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 *
 * In case of any questions, you can contact us at ESET, spol. s r.o., Einsteinova 24, 851 01 Bratislava, Slovakia.
 */

#include "ertp_qos_event_data.h"

#include <linux/ioprio.h>

#include "ertp_event.h"

#define ERTP_EVENT_QOS_READ_DEADLINE_MSECS 125
#define ERTP_EVENT_QOS_WRITE_DEADLINE_MSECS 500

static inline unsigned long get_deadline(enum ertp_qos_event_type event_type) {
  switch (event_type) {
    case ERTP_QOS_EVENT_TYPE_READ:
      return msecs_to_jiffies(ERTP_EVENT_QOS_READ_DEADLINE_MSECS);

    case ERTP_QOS_EVENT_TYPE_WRITE:
      return msecs_to_jiffies(ERTP_EVENT_QOS_WRITE_DEADLINE_MSECS);
  }
  BUG();
}

static inline int ertp_get_current_ioprio(void) {
#if LINUX_VERSION_CODE < KERNEL_VERSION(5, 0, 0)
  struct io_context* ioc = current->io_context;
  if (ioc) return ioc->ioprio;
  return IOPRIO_PRIO_VALUE(IOPRIO_CLASS_NONE, 0);
#else
  return get_current_ioprio();
#endif
}

void ertp_qos_event_data_fill(struct ertp_qos_event_data* data) {
  data->io_priority_class = IOPRIO_PRIO_CLASS(ertp_get_current_ioprio());
  data->current_deadline = jiffies + get_deadline(data->type);
}

const char* ertp_qos_event_data_debug_get_prio_str(
    const struct ertp_qos_event_data* data) {
  switch (data->io_priority_class) {
    case IOPRIO_CLASS_RT:
      return "RT";
    case IOPRIO_CLASS_BE:
      return "BE";
    case IOPRIO_CLASS_IDLE:
      return "IDLE";
    case IOPRIO_CLASS_NONE:
      return "NONE";
  }
  return "-";
}

const char* ertp_qos_event_data_debug_get_type_str(
    const struct ertp_qos_event_data* data) {
  switch (data->type) {
    case ERTP_QOS_EVENT_TYPE_READ:
      return "R";
    case ERTP_QOS_EVENT_TYPE_WRITE:
      return "W";
  }
  return "-";
}