1 /*
2 * Copyright(c) 2011-2017 Intel Corporation. All rights reserved.
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice (including the next
12 * paragraph) shall be included in all copies or substantial portions of the
13 * Software.
14 *
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
18 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 * SOFTWARE.
22 */
23 #include "i915_drv.h"
24 #include "gvt.h"
25
26 /**
27 * intel_vgpu_find_page_track - find page track rcord of guest page
28 * @vgpu: a vGPU
29 * @gfn: the gfn of guest page
30 *
31 * Returns:
32 * A pointer to struct intel_vgpu_page_track if found, else NULL returned.
33 */
34 struct intel_vgpu_page_track *intel_vgpu_find_page_track(
35 struct intel_vgpu *vgpu, unsigned long gfn)
36 {
37 return radix_tree_lookup(&vgpu->page_track_tree, gfn);
38 }
39
40 /**
41 * intel_vgpu_register_page_track - register a guest page to be tacked
42 * @vgpu: a vGPU
43 * @gfn: the gfn of guest page
44 * @handler: page track handler
45 * @priv: tracker private
46 *
47 * Returns:
48 * zero on success, negative error code if failed.
49 */
50 int intel_vgpu_register_page_track(struct intel_vgpu *vgpu, unsigned long gfn,
51 gvt_page_track_handler_t handler, void *priv)
52 {
53 struct intel_vgpu_page_track *track;
54 int ret;
55
56 track = intel_vgpu_find_page_track(vgpu, gfn);
57 if (track)
58 return -EEXIST;
59
60 track = kzalloc(sizeof(*track), GFP_KERNEL);
61 if (!track)
62 return -ENOMEM;
63
64 track->handler = handler;
65 track->priv_data = priv;
66
67 ret = radix_tree_insert(&vgpu->page_track_tree, gfn, track);
68 if (ret) {
69 kfree(track);
70 return ret;
71 }
72
73 return 0;
74 }
75
76 /**
77 * intel_vgpu_unregister_page_track - unregister the tracked guest page
78 * @vgpu: a vGPU
79 * @gfn: the gfn of guest page
80 *
81 */
82 void intel_vgpu_unregister_page_track(struct intel_vgpu *vgpu,
83 unsigned long gfn)
84 {
85 struct intel_vgpu_page_track *track;
86
87 track = radix_tree_delete(&vgpu->page_track_tree, gfn);
88 if (track) {
89 if (track->tracked)
90 intel_gvt_hypervisor_disable_page_track(vgpu, gfn);
91 kfree(track);
92 }
93 }
94
95 /**
96 * intel_vgpu_enable_page_track - set write-protection on guest page
97 * @vgpu: a vGPU
98 * @gfn: the gfn of guest page
99 *
100 * Returns:
101 * zero on success, negative error code if failed.
102 */
103 int intel_vgpu_enable_page_track(struct intel_vgpu *vgpu, unsigned long gfn)
104 {
105 struct intel_vgpu_page_track *track;
106 int ret;
107
108 track = intel_vgpu_find_page_track(vgpu, gfn);
109 if (!track)
110 return -ENXIO;
111
112 if (track->tracked)
113 return 0;
114
115 ret = intel_gvt_hypervisor_enable_page_track(vgpu, gfn);
116 if (ret)
117 return ret;
118 track->tracked = true;
119 return 0;
120 }
121
122 /**
123 * intel_vgpu_enable_page_track - cancel write-protection on guest page
124 * @vgpu: a vGPU
125 * @gfn: the gfn of guest page
126 *
127 * Returns:
128 * zero on success, negative error code if failed.
129 */
130 int intel_vgpu_disable_page_track(struct intel_vgpu *vgpu, unsigned long gfn)
131 {
132 struct intel_vgpu_page_track *track;
133 int ret;
134
135 track = intel_vgpu_find_page_track(vgpu, gfn);
136 if (!track)
137 return -ENXIO;
138
139 if (!track->tracked)
140 return 0;
141
142 ret = intel_gvt_hypervisor_disable_page_track(vgpu, gfn);
143 if (ret)
144 return ret;
145 track->tracked = false;
146 return 0;
147 }
148
149 /**
150 * intel_vgpu_page_track_handler - called when write to write-protected page
151 * @vgpu: a vGPU
152 * @gpa: the gpa of this write
153 * @data: the writed data
154 * @bytes: the length of this write
155 *
156 * Returns:
157 * zero on success, negative error code if failed.
158 */
159 int intel_vgpu_page_track_handler(struct intel_vgpu *vgpu, u64 gpa,
160 void *data, unsigned int bytes)
161 {
162 struct intel_vgpu_page_track *page_track;
163 int ret = 0;
164
165 mutex_lock(&vgpu->vgpu_lock);
166
167 page_track = intel_vgpu_find_page_track(vgpu, gpa >> PAGE_SHIFT);
168 if (!page_track) {
169 ret = -ENXIO;
170 goto out;
171 }
172
173 if (unlikely(vgpu->failsafe)) {
174 /* Remove write protection to prevent furture traps. */
175 intel_vgpu_disable_page_track(vgpu, gpa >> PAGE_SHIFT);
176 } else {
177 ret = page_track->handler(page_track, gpa, data, bytes);
178 if (ret)
179 gvt_err("guest page write error, gpa %llx\n", gpa);
180 }
181
182 out:
183 mutex_unlock(&vgpu->vgpu_lock);
184 return ret;
185 }