/*
 * SPDX-License-Identifier: MIT
 *
 * Copyright © 2018 Intel Corporation
 */

#include "i915_drv.h"
#include "gt/intel_gt.h"
#include "gt/intel_gt_print.h"

#include "../i915_selftest.h"
#include "igt_flush_test.h"
#include "igt_live_test.h"

int igt_live_test_begin(struct igt_live_test *t,
   struct drm_i915_private *i915,
   const char *func,
   const char *name)
{
 struct intel_engine_cs *engine;
 enum intel_engine_id id;
 struct intel_gt *gt;
 unsigned int i;
 int err;

 t->i915 = i915;
 t->func = func;
 t->name = name;

 for_each_gt(gt, i915, i) {

  err = intel_gt_wait_for_idle(gt, MAX_SCHEDULE_TIMEOUT);
  if (err) {
   gt_err(gt, "%s(%s): GT failed to idle before, with err=%d!",
          func, name, err);
   return err;
  }

  for_each_engine(engine, gt, id)
   t->reset_engine[i][id] =
    i915_reset_engine_count(&i915->gpu_error,
       engine);
 }

 t->reset_global = i915_reset_count(&i915->gpu_error);

 return 0;
}

int igt_live_test_end(struct igt_live_test *t)
{
 struct drm_i915_private *i915 = t->i915;
 struct intel_engine_cs *engine;
 enum intel_engine_id id;
 struct intel_gt *gt;
 unsigned int i;

 if (igt_flush_test(i915))
  return -EIO;

 if (t->reset_global != i915_reset_count(&i915->gpu_error)) {
  pr_err("%s(%s): GPU was reset %d times!\n",
         t->func, t->name,
         i915_reset_count(&i915->gpu_error) - t->reset_global);
  return -EIO;
 }

 for_each_gt(gt, i915, i) {
  for_each_engine(engine, gt, id) {
   if (t->reset_engine[i][id] ==
       i915_reset_engine_count(&i915->gpu_error, engine))
    continue;

   gt_err(gt, "%s(%s): engine '%s' was reset %d times!\n",
          t->func, t->name, engine->name,
          i915_reset_engine_count(&i915->gpu_error, engine) -
          t->reset_engine[i][id]);
   return -EIO;
  }
 }

 return 0;
}