1 /*
2  * Copyright 2019 Red Hat Inc.
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 shall be included in
12  * all copies or substantial portions of the Software.
13  *
14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
17  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
18  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20  * OTHER DEALINGS IN THE SOFTWARE.
21  */
22 #include "priv.h"
23 #include <core/falcon.h>
24 #include <core/firmware.h>
25 #include <nvfw/fw.h>
26 #include <nvfw/ls.h>
27 
28 void
29 nvkm_acr_lsfw_del(struct nvkm_acr_lsfw *lsfw)
30 {
31 	nvkm_blob_dtor(&lsfw->img);
32 	nvkm_firmware_put(lsfw->sig);
33 	list_del(&lsfw->head);
34 	kfree(lsfw);
35 }
36 
37 void
38 nvkm_acr_lsfw_del_all(struct nvkm_acr *acr)
39 {
40 	struct nvkm_acr_lsfw *lsfw, *lsft;
41 	list_for_each_entry_safe(lsfw, lsft, &acr->lsfw, head) {
42 		nvkm_acr_lsfw_del(lsfw);
43 	}
44 }
45 
46 static struct nvkm_acr_lsfw *
47 nvkm_acr_lsfw_get(struct nvkm_acr *acr, enum nvkm_acr_lsf_id id)
48 {
49 	struct nvkm_acr_lsfw *lsfw;
50 	list_for_each_entry(lsfw, &acr->lsfw, head) {
51 		if (lsfw->id == id)
52 			return lsfw;
53 	}
54 	return NULL;
55 }
56 
57 struct nvkm_acr_lsfw *
58 nvkm_acr_lsfw_add(const struct nvkm_acr_lsf_func *func, struct nvkm_acr *acr,
59 		 struct nvkm_falcon *falcon, enum nvkm_acr_lsf_id id)
60 {
61 	struct nvkm_acr_lsfw *lsfw = nvkm_acr_lsfw_get(acr, id);
62 
63 	if (lsfw && lsfw->func) {
64 		nvkm_error(&acr->subdev, "LSFW %d redefined\n", id);
65 		return ERR_PTR(-EEXIST);
66 	}
67 
68 	if (!lsfw) {
69 		if (!(lsfw = kzalloc(sizeof(*lsfw), GFP_KERNEL)))
70 			return ERR_PTR(-ENOMEM);
71 
72 		lsfw->id = id;
73 		list_add_tail(&lsfw->head, &acr->lsfw);
74 	}
75 
76 	lsfw->func = func;
77 	lsfw->falcon = falcon;
78 	return lsfw;
79 }
80 
81 static struct nvkm_acr_lsfw *
82 nvkm_acr_lsfw_load_sig_image_desc_(struct nvkm_subdev *subdev,
83 				   struct nvkm_falcon *falcon,
84 				   enum nvkm_acr_lsf_id id,
85 				   const char *path, int ver,
86 				   const struct nvkm_acr_lsf_func *func,
87 				   const struct firmware **pdesc)
88 {
89 	struct nvkm_acr *acr = subdev->device->acr;
90 	struct nvkm_acr_lsfw *lsfw;
91 	int ret;
92 
93 	if (IS_ERR((lsfw = nvkm_acr_lsfw_add(func, acr, falcon, id))))
94 		return lsfw;
95 
96 	ret = nvkm_firmware_load_name(subdev, path, "sig", ver, &lsfw->sig);
97 	if (ret)
98 		goto done;
99 
100 	ret = nvkm_firmware_load_blob(subdev, path, "image", ver, &lsfw->img);
101 	if (ret)
102 		goto done;
103 
104 	ret = nvkm_firmware_load_name(subdev, path, "desc", ver, pdesc);
105 done:
106 	if (ret) {
107 		nvkm_acr_lsfw_del(lsfw);
108 		return ERR_PTR(ret);
109 	}
110 
111 	return lsfw;
112 }
113 
114 static void
115 nvkm_acr_lsfw_from_desc(const struct nvfw_ls_desc_head *desc,
116 			struct nvkm_acr_lsfw *lsfw)
117 {
118 	lsfw->bootloader_size = ALIGN(desc->bootloader_size, 256);
119 	lsfw->bootloader_imem_offset = desc->bootloader_imem_offset;
120 
121 	lsfw->app_size = ALIGN(desc->app_size, 256);
122 	lsfw->app_start_offset = desc->app_start_offset;
123 	lsfw->app_imem_entry = desc->app_imem_entry;
124 	lsfw->app_resident_code_offset = desc->app_resident_code_offset;
125 	lsfw->app_resident_code_size = desc->app_resident_code_size;
126 	lsfw->app_resident_data_offset = desc->app_resident_data_offset;
127 	lsfw->app_resident_data_size = desc->app_resident_data_size;
128 
129 	lsfw->ucode_size = ALIGN(lsfw->app_resident_data_offset, 256) +
130 			   lsfw->bootloader_size;
131 	lsfw->data_size = lsfw->app_size + lsfw->bootloader_size -
132 			  lsfw->ucode_size;
133 }
134 
135 int
136 nvkm_acr_lsfw_load_sig_image_desc(struct nvkm_subdev *subdev,
137 				  struct nvkm_falcon *falcon,
138 				  enum nvkm_acr_lsf_id id,
139 				  const char *path, int ver,
140 				  const struct nvkm_acr_lsf_func *func)
141 {
142 	const struct firmware *fw;
143 	struct nvkm_acr_lsfw *lsfw;
144 
145 	lsfw = nvkm_acr_lsfw_load_sig_image_desc_(subdev, falcon, id, path, ver,
146 						  func, &fw);
147 	if (IS_ERR(lsfw))
148 		return PTR_ERR(lsfw);
149 
150 	nvkm_acr_lsfw_from_desc(&nvfw_ls_desc(subdev, fw->data)->head, lsfw);
151 	nvkm_firmware_put(fw);
152 	return 0;
153 }
154 
155 int
156 nvkm_acr_lsfw_load_sig_image_desc_v1(struct nvkm_subdev *subdev,
157 				     struct nvkm_falcon *falcon,
158 				     enum nvkm_acr_lsf_id id,
159 				     const char *path, int ver,
160 				     const struct nvkm_acr_lsf_func *func)
161 {
162 	const struct firmware *fw;
163 	struct nvkm_acr_lsfw *lsfw;
164 
165 	lsfw = nvkm_acr_lsfw_load_sig_image_desc_(subdev, falcon, id, path, ver,
166 						  func, &fw);
167 	if (IS_ERR(lsfw))
168 		return PTR_ERR(lsfw);
169 
170 	nvkm_acr_lsfw_from_desc(&nvfw_ls_desc_v1(subdev, fw->data)->head, lsfw);
171 	nvkm_firmware_put(fw);
172 	return 0;
173 }
174 
175 int
176 nvkm_acr_lsfw_load_bl_inst_data_sig(struct nvkm_subdev *subdev,
177 				    struct nvkm_falcon *falcon,
178 				    enum nvkm_acr_lsf_id id,
179 				    const char *path, int ver,
180 				    const struct nvkm_acr_lsf_func *func)
181 {
182 	struct nvkm_acr *acr = subdev->device->acr;
183 	struct nvkm_acr_lsfw *lsfw;
184 	const struct firmware *bl = NULL, *inst = NULL, *data = NULL;
185 	const struct nvfw_bin_hdr *hdr;
186 	const struct nvfw_bl_desc *desc;
187 	u32 *bldata;
188 	int ret;
189 
190 	if (IS_ERR((lsfw = nvkm_acr_lsfw_add(func, acr, falcon, id))))
191 		return PTR_ERR(lsfw);
192 
193 	ret = nvkm_firmware_load_name(subdev, path, "bl", ver, &bl);
194 	if (ret)
195 		goto done;
196 
197 	hdr = nvfw_bin_hdr(subdev, bl->data);
198 	desc = nvfw_bl_desc(subdev, bl->data + hdr->header_offset);
199 	bldata = (void *)(bl->data + hdr->data_offset);
200 
201 	ret = nvkm_firmware_load_name(subdev, path, "inst", ver, &inst);
202 	if (ret)
203 		goto done;
204 
205 	ret = nvkm_firmware_load_name(subdev, path, "data", ver, &data);
206 	if (ret)
207 		goto done;
208 
209 	ret = nvkm_firmware_load_name(subdev, path, "sig", ver, &lsfw->sig);
210 	if (ret)
211 		goto done;
212 
213 	lsfw->bootloader_size = ALIGN(desc->code_size, 256);
214 	lsfw->bootloader_imem_offset = desc->start_tag << 8;
215 
216 	lsfw->app_start_offset = lsfw->bootloader_size;
217 	lsfw->app_imem_entry = 0;
218 	lsfw->app_resident_code_offset = 0;
219 	lsfw->app_resident_code_size = ALIGN(inst->size, 256);
220 	lsfw->app_resident_data_offset = lsfw->app_resident_code_size;
221 	lsfw->app_resident_data_size = ALIGN(data->size, 256);
222 	lsfw->app_size = lsfw->app_resident_code_size +
223 			 lsfw->app_resident_data_size;
224 
225 	lsfw->img.size = lsfw->bootloader_size + lsfw->app_size;
226 	if (!(lsfw->img.data = kzalloc(lsfw->img.size, GFP_KERNEL))) {
227 		ret = -ENOMEM;
228 		goto done;
229 	}
230 
231 	memcpy(lsfw->img.data, bldata, lsfw->bootloader_size);
232 	memcpy(lsfw->img.data + lsfw->app_start_offset +
233 	       lsfw->app_resident_code_offset, inst->data, inst->size);
234 	memcpy(lsfw->img.data + lsfw->app_start_offset +
235 	       lsfw->app_resident_data_offset, data->data, data->size);
236 
237 	lsfw->ucode_size = ALIGN(lsfw->app_resident_data_offset, 256) +
238 			   lsfw->bootloader_size;
239 	lsfw->data_size = lsfw->app_size + lsfw->bootloader_size -
240 			  lsfw->ucode_size;
241 
242 done:
243 	if (ret)
244 		nvkm_acr_lsfw_del(lsfw);
245 	nvkm_firmware_put(data);
246 	nvkm_firmware_put(inst);
247 	nvkm_firmware_put(bl);
248 	return ret;
249 }
250