1 /*
2 * SPDX-License-Identifier: MIT
3 *
4 * Copyright © 2018 Intel Corporation
5 */
6
7 #ifndef IGT_WEDGE_ME_H
8 #define IGT_WEDGE_ME_H
9
10 #include <linux/workqueue.h>
11
12 #include "../i915_gem.h"
13
14 struct drm_i915_private;
15
16 struct igt_wedge_me {
17 struct delayed_work work;
18 struct drm_i915_private *i915;
19 const char *name;
20 };
21
__igt_wedge_me(struct work_struct * work)22 static void __igt_wedge_me(struct work_struct *work)
23 {
24 struct igt_wedge_me *w = container_of(work, typeof(*w), work.work);
25
26 pr_err("%s timed out, cancelling test.\n", w->name);
27
28 GEM_TRACE("%s timed out.\n", w->name);
29 GEM_TRACE_DUMP();
30
31 i915_gem_set_wedged(w->i915);
32 }
33
__igt_init_wedge(struct igt_wedge_me * w,struct drm_i915_private * i915,long timeout,const char * name)34 static void __igt_init_wedge(struct igt_wedge_me *w,
35 struct drm_i915_private *i915,
36 long timeout,
37 const char *name)
38 {
39 w->i915 = i915;
40 w->name = name;
41
42 INIT_DELAYED_WORK_ONSTACK(&w->work, __igt_wedge_me);
43 schedule_delayed_work(&w->work, timeout);
44 }
45
__igt_fini_wedge(struct igt_wedge_me * w)46 static void __igt_fini_wedge(struct igt_wedge_me *w)
47 {
48 cancel_delayed_work_sync(&w->work);
49 destroy_delayed_work_on_stack(&w->work);
50 w->i915 = NULL;
51 }
52
53 #define igt_wedge_on_timeout(W, DEV, TIMEOUT) \
54 for (__igt_init_wedge((W), (DEV), (TIMEOUT), __func__); \
55 (W)->i915; \
56 __igt_fini_wedge((W)))
57
58 #endif /* IGT_WEDGE_ME_H */
59