1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3 * Arm Firmware Framework for ARMv8-A(FFA) interface driver
4 *
5 * The Arm FFA specification[1] describes a software architecture to
6 * leverages the virtualization extension to isolate software images
7 * provided by an ecosystem of vendors from each other and describes
8 * interfaces that standardize communication between the various software
9 * images including communication between images in the Secure world and
10 * Normal world. Any Hypervisor could use the FFA interfaces to enable
11 * communication between VMs it manages.
12 *
13 * The Hypervisor a.k.a Partition managers in FFA terminology can assign
14 * system resources(Memory regions, Devices, CPU cycles) to the partitions
15 * and manage isolation amongst them.
16 *
17 * [1] https://developer.arm.com/docs/den0077/latest
18 *
19 * Copyright (C) 2021 ARM Ltd.
20 */
21
22 #define DRIVER_NAME "ARM FF-A"
23 #define pr_fmt(fmt) DRIVER_NAME ": " fmt
24
25 #include <linux/acpi.h>
26 #include <linux/arm_ffa.h>
27 #include <linux/bitfield.h>
28 #include <linux/cpuhotplug.h>
29 #include <linux/delay.h>
30 #include <linux/device.h>
31 #include <linux/hashtable.h>
32 #include <linux/interrupt.h>
33 #include <linux/io.h>
34 #include <linux/kernel.h>
35 #include <linux/module.h>
36 #include <linux/mm.h>
37 #include <linux/mutex.h>
38 #include <linux/of_irq.h>
39 #include <linux/scatterlist.h>
40 #include <linux/slab.h>
41 #include <linux/smp.h>
42 #include <linux/uuid.h>
43 #include <linux/xarray.h>
44
45 #include "common.h"
46
47 #define FFA_DRIVER_VERSION FFA_VERSION_1_1
48 #define FFA_MIN_VERSION FFA_VERSION_1_0
49
50 #define SENDER_ID_MASK GENMASK(31, 16)
51 #define RECEIVER_ID_MASK GENMASK(15, 0)
52 #define SENDER_ID(x) ((u16)(FIELD_GET(SENDER_ID_MASK, (x))))
53 #define RECEIVER_ID(x) ((u16)(FIELD_GET(RECEIVER_ID_MASK, (x))))
54 #define PACK_TARGET_INFO(s, r) \
55 (FIELD_PREP(SENDER_ID_MASK, (s)) | FIELD_PREP(RECEIVER_ID_MASK, (r)))
56
57 #define RXTX_MAP_MIN_BUFSZ_MASK GENMASK(1, 0)
58 #define RXTX_MAP_MIN_BUFSZ(x) ((x) & RXTX_MAP_MIN_BUFSZ_MASK)
59
60 #define FFA_MAX_NOTIFICATIONS 64
61
62 static ffa_fn *invoke_ffa_fn;
63
64 static const int ffa_linux_errmap[] = {
65 /* better than switch case as long as return value is continuous */
66 0, /* FFA_RET_SUCCESS */
67 -EOPNOTSUPP, /* FFA_RET_NOT_SUPPORTED */
68 -EINVAL, /* FFA_RET_INVALID_PARAMETERS */
69 -ENOMEM, /* FFA_RET_NO_MEMORY */
70 -EBUSY, /* FFA_RET_BUSY */
71 -EINTR, /* FFA_RET_INTERRUPTED */
72 -EACCES, /* FFA_RET_DENIED */
73 -EAGAIN, /* FFA_RET_RETRY */
74 -ECANCELED, /* FFA_RET_ABORTED */
75 -ENODATA, /* FFA_RET_NO_DATA */
76 -EAGAIN, /* FFA_RET_NOT_READY */
77 };
78
ffa_to_linux_errno(int errno)79 static inline int ffa_to_linux_errno(int errno)
80 {
81 int err_idx = -errno;
82
83 if (err_idx >= 0 && err_idx < ARRAY_SIZE(ffa_linux_errmap))
84 return ffa_linux_errmap[err_idx];
85 return -EINVAL;
86 }
87
88 struct ffa_pcpu_irq {
89 struct ffa_drv_info *info;
90 };
91
92 struct ffa_drv_info {
93 u32 version;
94 u16 vm_id;
95 struct mutex rx_lock; /* lock to protect Rx buffer */
96 struct mutex tx_lock; /* lock to protect Tx buffer */
97 void *rx_buffer;
98 void *tx_buffer;
99 size_t rxtx_bufsz;
100 bool mem_ops_native;
101 bool msg_direct_req2_supp;
102 bool bitmap_created;
103 bool notif_enabled;
104 unsigned int sched_recv_irq;
105 unsigned int notif_pend_irq;
106 unsigned int cpuhp_state;
107 struct ffa_pcpu_irq __percpu *irq_pcpu;
108 struct workqueue_struct *notif_pcpu_wq;
109 struct work_struct notif_pcpu_work;
110 struct work_struct sched_recv_irq_work;
111 struct xarray partition_info;
112 DECLARE_HASHTABLE(notifier_hash, ilog2(FFA_MAX_NOTIFICATIONS));
113 struct mutex notify_lock; /* lock to protect notifier hashtable */
114 };
115
116 static struct ffa_drv_info *drv_info;
117 static void ffa_partitions_cleanup(void);
118
119 /*
120 * The driver must be able to support all the versions from the earliest
121 * supported FFA_MIN_VERSION to the latest supported FFA_DRIVER_VERSION.
122 * The specification states that if firmware supports a FFA implementation
123 * that is incompatible with and at a greater version number than specified
124 * by the caller(FFA_DRIVER_VERSION passed as parameter to FFA_VERSION),
125 * it must return the NOT_SUPPORTED error code.
126 */
ffa_compatible_version_find(u32 version)127 static u32 ffa_compatible_version_find(u32 version)
128 {
129 u16 major = FFA_MAJOR_VERSION(version), minor = FFA_MINOR_VERSION(version);
130 u16 drv_major = FFA_MAJOR_VERSION(FFA_DRIVER_VERSION);
131 u16 drv_minor = FFA_MINOR_VERSION(FFA_DRIVER_VERSION);
132
133 if ((major < drv_major) || (major == drv_major && minor <= drv_minor))
134 return version;
135
136 pr_info("Firmware version higher than driver version, downgrading\n");
137 return FFA_DRIVER_VERSION;
138 }
139
ffa_version_check(u32 * version)140 static int ffa_version_check(u32 *version)
141 {
142 ffa_value_t ver;
143
144 invoke_ffa_fn((ffa_value_t){
145 .a0 = FFA_VERSION, .a1 = FFA_DRIVER_VERSION,
146 }, &ver);
147
148 if (ver.a0 == FFA_RET_NOT_SUPPORTED) {
149 pr_info("FFA_VERSION returned not supported\n");
150 return -EOPNOTSUPP;
151 }
152
153 if (ver.a0 < FFA_MIN_VERSION) {
154 pr_err("Incompatible v%d.%d! Earliest supported v%d.%d\n",
155 FFA_MAJOR_VERSION(ver.a0), FFA_MINOR_VERSION(ver.a0),
156 FFA_MAJOR_VERSION(FFA_MIN_VERSION),
157 FFA_MINOR_VERSION(FFA_MIN_VERSION));
158 return -EINVAL;
159 }
160
161 pr_info("Driver version %d.%d\n", FFA_MAJOR_VERSION(FFA_DRIVER_VERSION),
162 FFA_MINOR_VERSION(FFA_DRIVER_VERSION));
163 pr_info("Firmware version %d.%d found\n", FFA_MAJOR_VERSION(ver.a0),
164 FFA_MINOR_VERSION(ver.a0));
165 *version = ffa_compatible_version_find(ver.a0);
166
167 return 0;
168 }
169
ffa_rx_release(void)170 static int ffa_rx_release(void)
171 {
172 ffa_value_t ret;
173
174 invoke_ffa_fn((ffa_value_t){
175 .a0 = FFA_RX_RELEASE,
176 }, &ret);
177
178 if (ret.a0 == FFA_ERROR)
179 return ffa_to_linux_errno((int)ret.a2);
180
181 /* check for ret.a0 == FFA_RX_RELEASE ? */
182
183 return 0;
184 }
185
ffa_rxtx_map(phys_addr_t tx_buf,phys_addr_t rx_buf,u32 pg_cnt)186 static int ffa_rxtx_map(phys_addr_t tx_buf, phys_addr_t rx_buf, u32 pg_cnt)
187 {
188 ffa_value_t ret;
189
190 invoke_ffa_fn((ffa_value_t){
191 .a0 = FFA_FN_NATIVE(RXTX_MAP),
192 .a1 = tx_buf, .a2 = rx_buf, .a3 = pg_cnt,
193 }, &ret);
194
195 if (ret.a0 == FFA_ERROR)
196 return ffa_to_linux_errno((int)ret.a2);
197
198 return 0;
199 }
200
ffa_rxtx_unmap(u16 vm_id)201 static int ffa_rxtx_unmap(u16 vm_id)
202 {
203 ffa_value_t ret;
204
205 invoke_ffa_fn((ffa_value_t){
206 .a0 = FFA_RXTX_UNMAP, .a1 = PACK_TARGET_INFO(vm_id, 0),
207 }, &ret);
208
209 if (ret.a0 == FFA_ERROR)
210 return ffa_to_linux_errno((int)ret.a2);
211
212 return 0;
213 }
214
ffa_features(u32 func_feat_id,u32 input_props,u32 * if_props_1,u32 * if_props_2)215 static int ffa_features(u32 func_feat_id, u32 input_props,
216 u32 *if_props_1, u32 *if_props_2)
217 {
218 ffa_value_t id;
219
220 if (!ARM_SMCCC_IS_FAST_CALL(func_feat_id) && input_props) {
221 pr_err("%s: Invalid Parameters: %x, %x", __func__,
222 func_feat_id, input_props);
223 return ffa_to_linux_errno(FFA_RET_INVALID_PARAMETERS);
224 }
225
226 invoke_ffa_fn((ffa_value_t){
227 .a0 = FFA_FEATURES, .a1 = func_feat_id, .a2 = input_props,
228 }, &id);
229
230 if (id.a0 == FFA_ERROR)
231 return ffa_to_linux_errno((int)id.a2);
232
233 if (if_props_1)
234 *if_props_1 = id.a2;
235 if (if_props_2)
236 *if_props_2 = id.a3;
237
238 return 0;
239 }
240
241 #define PARTITION_INFO_GET_RETURN_COUNT_ONLY BIT(0)
242
243 /* buffer must be sizeof(struct ffa_partition_info) * num_partitions */
244 static int
__ffa_partition_info_get(u32 uuid0,u32 uuid1,u32 uuid2,u32 uuid3,struct ffa_partition_info * buffer,int num_partitions)245 __ffa_partition_info_get(u32 uuid0, u32 uuid1, u32 uuid2, u32 uuid3,
246 struct ffa_partition_info *buffer, int num_partitions)
247 {
248 int idx, count, flags = 0, sz, buf_sz;
249 ffa_value_t partition_info;
250
251 if (drv_info->version > FFA_VERSION_1_0 &&
252 (!buffer || !num_partitions)) /* Just get the count for now */
253 flags = PARTITION_INFO_GET_RETURN_COUNT_ONLY;
254
255 mutex_lock(&drv_info->rx_lock);
256 invoke_ffa_fn((ffa_value_t){
257 .a0 = FFA_PARTITION_INFO_GET,
258 .a1 = uuid0, .a2 = uuid1, .a3 = uuid2, .a4 = uuid3,
259 .a5 = flags,
260 }, &partition_info);
261
262 if (partition_info.a0 == FFA_ERROR) {
263 mutex_unlock(&drv_info->rx_lock);
264 return ffa_to_linux_errno((int)partition_info.a2);
265 }
266
267 count = partition_info.a2;
268
269 if (drv_info->version > FFA_VERSION_1_0) {
270 buf_sz = sz = partition_info.a3;
271 if (sz > sizeof(*buffer))
272 buf_sz = sizeof(*buffer);
273 } else {
274 /* FFA_VERSION_1_0 lacks size in the response */
275 buf_sz = sz = 8;
276 }
277
278 if (buffer && count <= num_partitions)
279 for (idx = 0; idx < count; idx++)
280 memcpy(buffer + idx, drv_info->rx_buffer + idx * sz,
281 buf_sz);
282
283 ffa_rx_release();
284
285 mutex_unlock(&drv_info->rx_lock);
286
287 return count;
288 }
289
290 #define LAST_INDEX_MASK GENMASK(15, 0)
291 #define CURRENT_INDEX_MASK GENMASK(31, 16)
292 #define UUID_INFO_TAG_MASK GENMASK(47, 32)
293 #define PARTITION_INFO_SZ_MASK GENMASK(63, 48)
294 #define PARTITION_COUNT(x) ((u16)(FIELD_GET(LAST_INDEX_MASK, (x))) + 1)
295 #define CURRENT_INDEX(x) ((u16)(FIELD_GET(CURRENT_INDEX_MASK, (x))))
296 #define UUID_INFO_TAG(x) ((u16)(FIELD_GET(UUID_INFO_TAG_MASK, (x))))
297 #define PARTITION_INFO_SZ(x) ((u16)(FIELD_GET(PARTITION_INFO_SZ_MASK, (x))))
298 static int
__ffa_partition_info_get_regs(u32 uuid0,u32 uuid1,u32 uuid2,u32 uuid3,struct ffa_partition_info * buffer,int num_parts)299 __ffa_partition_info_get_regs(u32 uuid0, u32 uuid1, u32 uuid2, u32 uuid3,
300 struct ffa_partition_info *buffer, int num_parts)
301 {
302 u16 buf_sz, start_idx, cur_idx, count = 0, prev_idx = 0, tag = 0;
303 ffa_value_t partition_info;
304
305 do {
306 start_idx = prev_idx ? prev_idx + 1 : 0;
307
308 invoke_ffa_fn((ffa_value_t){
309 .a0 = FFA_PARTITION_INFO_GET_REGS,
310 .a1 = (u64)uuid1 << 32 | uuid0,
311 .a2 = (u64)uuid3 << 32 | uuid2,
312 .a3 = start_idx | tag << 16,
313 }, &partition_info);
314
315 if (partition_info.a0 == FFA_ERROR)
316 return ffa_to_linux_errno((int)partition_info.a2);
317
318 if (!count)
319 count = PARTITION_COUNT(partition_info.a2);
320 if (!buffer || !num_parts) /* count only */
321 return count;
322
323 cur_idx = CURRENT_INDEX(partition_info.a2);
324 tag = UUID_INFO_TAG(partition_info.a2);
325 buf_sz = PARTITION_INFO_SZ(partition_info.a2);
326 if (buf_sz > sizeof(*buffer))
327 buf_sz = sizeof(*buffer);
328
329 memcpy(buffer + prev_idx * buf_sz, &partition_info.a3,
330 (cur_idx - start_idx + 1) * buf_sz);
331 prev_idx = cur_idx;
332
333 } while (cur_idx < (count - 1));
334
335 return count;
336 }
337
338 /* buffer is allocated and caller must free the same if returned count > 0 */
339 static int
ffa_partition_probe(const uuid_t * uuid,struct ffa_partition_info ** buffer)340 ffa_partition_probe(const uuid_t *uuid, struct ffa_partition_info **buffer)
341 {
342 int count;
343 u32 uuid0_4[4];
344 bool reg_mode = false;
345 struct ffa_partition_info *pbuf;
346
347 if (!ffa_features(FFA_PARTITION_INFO_GET_REGS, 0, NULL, NULL))
348 reg_mode = true;
349
350 export_uuid((u8 *)uuid0_4, uuid);
351 if (reg_mode)
352 count = __ffa_partition_info_get_regs(uuid0_4[0], uuid0_4[1],
353 uuid0_4[2], uuid0_4[3],
354 NULL, 0);
355 else
356 count = __ffa_partition_info_get(uuid0_4[0], uuid0_4[1],
357 uuid0_4[2], uuid0_4[3],
358 NULL, 0);
359 if (count <= 0)
360 return count;
361
362 pbuf = kcalloc(count, sizeof(*pbuf), GFP_KERNEL);
363 if (!pbuf)
364 return -ENOMEM;
365
366 if (reg_mode)
367 count = __ffa_partition_info_get_regs(uuid0_4[0], uuid0_4[1],
368 uuid0_4[2], uuid0_4[3],
369 pbuf, count);
370 else
371 count = __ffa_partition_info_get(uuid0_4[0], uuid0_4[1],
372 uuid0_4[2], uuid0_4[3],
373 pbuf, count);
374 if (count <= 0)
375 kfree(pbuf);
376 else
377 *buffer = pbuf;
378
379 return count;
380 }
381
382 #define VM_ID_MASK GENMASK(15, 0)
ffa_id_get(u16 * vm_id)383 static int ffa_id_get(u16 *vm_id)
384 {
385 ffa_value_t id;
386
387 invoke_ffa_fn((ffa_value_t){
388 .a0 = FFA_ID_GET,
389 }, &id);
390
391 if (id.a0 == FFA_ERROR)
392 return ffa_to_linux_errno((int)id.a2);
393
394 *vm_id = FIELD_GET(VM_ID_MASK, (id.a2));
395
396 return 0;
397 }
398
ffa_msg_send_wait_for_completion(ffa_value_t * ret)399 static inline void ffa_msg_send_wait_for_completion(ffa_value_t *ret)
400 {
401 while (ret->a0 == FFA_INTERRUPT || ret->a0 == FFA_YIELD) {
402 if (ret->a0 == FFA_YIELD)
403 fsleep(1000);
404
405 invoke_ffa_fn((ffa_value_t){
406 .a0 = FFA_RUN, .a1 = ret->a1,
407 }, ret);
408 }
409 }
410
ffa_msg_send_direct_req(u16 src_id,u16 dst_id,bool mode_32bit,struct ffa_send_direct_data * data)411 static int ffa_msg_send_direct_req(u16 src_id, u16 dst_id, bool mode_32bit,
412 struct ffa_send_direct_data *data)
413 {
414 u32 req_id, resp_id, src_dst_ids = PACK_TARGET_INFO(src_id, dst_id);
415 ffa_value_t ret;
416
417 if (mode_32bit) {
418 req_id = FFA_MSG_SEND_DIRECT_REQ;
419 resp_id = FFA_MSG_SEND_DIRECT_RESP;
420 } else {
421 req_id = FFA_FN_NATIVE(MSG_SEND_DIRECT_REQ);
422 resp_id = FFA_FN_NATIVE(MSG_SEND_DIRECT_RESP);
423 }
424
425 invoke_ffa_fn((ffa_value_t){
426 .a0 = req_id, .a1 = src_dst_ids, .a2 = 0,
427 .a3 = data->data0, .a4 = data->data1, .a5 = data->data2,
428 .a6 = data->data3, .a7 = data->data4,
429 }, &ret);
430
431 ffa_msg_send_wait_for_completion(&ret);
432
433 if (ret.a0 == FFA_ERROR)
434 return ffa_to_linux_errno((int)ret.a2);
435
436 if (ret.a0 == resp_id) {
437 data->data0 = ret.a3;
438 data->data1 = ret.a4;
439 data->data2 = ret.a5;
440 data->data3 = ret.a6;
441 data->data4 = ret.a7;
442 return 0;
443 }
444
445 return -EINVAL;
446 }
447
ffa_msg_send2(u16 src_id,u16 dst_id,void * buf,size_t sz)448 static int ffa_msg_send2(u16 src_id, u16 dst_id, void *buf, size_t sz)
449 {
450 u32 src_dst_ids = PACK_TARGET_INFO(src_id, dst_id);
451 struct ffa_indirect_msg_hdr *msg;
452 ffa_value_t ret;
453 int retval = 0;
454
455 if (sz > (drv_info->rxtx_bufsz - sizeof(*msg)))
456 return -ERANGE;
457
458 mutex_lock(&drv_info->tx_lock);
459
460 msg = drv_info->tx_buffer;
461 msg->flags = 0;
462 msg->res0 = 0;
463 msg->offset = sizeof(*msg);
464 msg->send_recv_id = src_dst_ids;
465 msg->size = sz;
466 memcpy((u8 *)msg + msg->offset, buf, sz);
467
468 /* flags = 0, sender VMID = 0 works for both physical/virtual NS */
469 invoke_ffa_fn((ffa_value_t){
470 .a0 = FFA_MSG_SEND2, .a1 = 0, .a2 = 0
471 }, &ret);
472
473 if (ret.a0 == FFA_ERROR)
474 retval = ffa_to_linux_errno((int)ret.a2);
475
476 mutex_unlock(&drv_info->tx_lock);
477 return retval;
478 }
479
ffa_msg_send_direct_req2(u16 src_id,u16 dst_id,const uuid_t * uuid,struct ffa_send_direct_data2 * data)480 static int ffa_msg_send_direct_req2(u16 src_id, u16 dst_id, const uuid_t *uuid,
481 struct ffa_send_direct_data2 *data)
482 {
483 u32 src_dst_ids = PACK_TARGET_INFO(src_id, dst_id);
484 union {
485 uuid_t uuid;
486 __le64 regs[2];
487 } uuid_regs = { .uuid = *uuid };
488 ffa_value_t ret, args = {
489 .a0 = FFA_MSG_SEND_DIRECT_REQ2,
490 .a1 = src_dst_ids,
491 .a2 = le64_to_cpu(uuid_regs.regs[0]),
492 .a3 = le64_to_cpu(uuid_regs.regs[1]),
493 };
494 memcpy((void *)&args + offsetof(ffa_value_t, a4), data, sizeof(*data));
495
496 invoke_ffa_fn(args, &ret);
497
498 ffa_msg_send_wait_for_completion(&ret);
499
500 if (ret.a0 == FFA_ERROR)
501 return ffa_to_linux_errno((int)ret.a2);
502
503 if (ret.a0 == FFA_MSG_SEND_DIRECT_RESP2) {
504 memcpy(data, (void *)&ret + offsetof(ffa_value_t, a4), sizeof(*data));
505 return 0;
506 }
507
508 return -EINVAL;
509 }
510
ffa_mem_first_frag(u32 func_id,phys_addr_t buf,u32 buf_sz,u32 frag_len,u32 len,u64 * handle)511 static int ffa_mem_first_frag(u32 func_id, phys_addr_t buf, u32 buf_sz,
512 u32 frag_len, u32 len, u64 *handle)
513 {
514 ffa_value_t ret;
515
516 invoke_ffa_fn((ffa_value_t){
517 .a0 = func_id, .a1 = len, .a2 = frag_len,
518 .a3 = buf, .a4 = buf_sz,
519 }, &ret);
520
521 while (ret.a0 == FFA_MEM_OP_PAUSE)
522 invoke_ffa_fn((ffa_value_t){
523 .a0 = FFA_MEM_OP_RESUME,
524 .a1 = ret.a1, .a2 = ret.a2,
525 }, &ret);
526
527 if (ret.a0 == FFA_ERROR)
528 return ffa_to_linux_errno((int)ret.a2);
529
530 if (ret.a0 == FFA_SUCCESS) {
531 if (handle)
532 *handle = PACK_HANDLE(ret.a2, ret.a3);
533 } else if (ret.a0 == FFA_MEM_FRAG_RX) {
534 if (handle)
535 *handle = PACK_HANDLE(ret.a1, ret.a2);
536 } else {
537 return -EOPNOTSUPP;
538 }
539
540 return frag_len;
541 }
542
ffa_mem_next_frag(u64 handle,u32 frag_len)543 static int ffa_mem_next_frag(u64 handle, u32 frag_len)
544 {
545 ffa_value_t ret;
546
547 invoke_ffa_fn((ffa_value_t){
548 .a0 = FFA_MEM_FRAG_TX,
549 .a1 = HANDLE_LOW(handle), .a2 = HANDLE_HIGH(handle),
550 .a3 = frag_len,
551 }, &ret);
552
553 while (ret.a0 == FFA_MEM_OP_PAUSE)
554 invoke_ffa_fn((ffa_value_t){
555 .a0 = FFA_MEM_OP_RESUME,
556 .a1 = ret.a1, .a2 = ret.a2,
557 }, &ret);
558
559 if (ret.a0 == FFA_ERROR)
560 return ffa_to_linux_errno((int)ret.a2);
561
562 if (ret.a0 == FFA_MEM_FRAG_RX)
563 return ret.a3;
564 else if (ret.a0 == FFA_SUCCESS)
565 return 0;
566
567 return -EOPNOTSUPP;
568 }
569
570 static int
ffa_transmit_fragment(u32 func_id,phys_addr_t buf,u32 buf_sz,u32 frag_len,u32 len,u64 * handle,bool first)571 ffa_transmit_fragment(u32 func_id, phys_addr_t buf, u32 buf_sz, u32 frag_len,
572 u32 len, u64 *handle, bool first)
573 {
574 if (!first)
575 return ffa_mem_next_frag(*handle, frag_len);
576
577 return ffa_mem_first_frag(func_id, buf, buf_sz, frag_len, len, handle);
578 }
579
ffa_get_num_pages_sg(struct scatterlist * sg)580 static u32 ffa_get_num_pages_sg(struct scatterlist *sg)
581 {
582 u32 num_pages = 0;
583
584 do {
585 num_pages += sg->length / FFA_PAGE_SIZE;
586 } while ((sg = sg_next(sg)));
587
588 return num_pages;
589 }
590
ffa_memory_attributes_get(u32 func_id)591 static u16 ffa_memory_attributes_get(u32 func_id)
592 {
593 /*
594 * For the memory lend or donate operation, if the receiver is a PE or
595 * a proxy endpoint, the owner/sender must not specify the attributes
596 */
597 if (func_id == FFA_FN_NATIVE(MEM_LEND) ||
598 func_id == FFA_MEM_LEND)
599 return 0;
600
601 return FFA_MEM_NORMAL | FFA_MEM_WRITE_BACK | FFA_MEM_INNER_SHAREABLE;
602 }
603
604 static int
ffa_setup_and_transmit(u32 func_id,void * buffer,u32 max_fragsize,struct ffa_mem_ops_args * args)605 ffa_setup_and_transmit(u32 func_id, void *buffer, u32 max_fragsize,
606 struct ffa_mem_ops_args *args)
607 {
608 int rc = 0;
609 bool first = true;
610 u32 composite_offset;
611 phys_addr_t addr = 0;
612 struct ffa_mem_region *mem_region = buffer;
613 struct ffa_composite_mem_region *composite;
614 struct ffa_mem_region_addr_range *constituents;
615 struct ffa_mem_region_attributes *ep_mem_access;
616 u32 idx, frag_len, length, buf_sz = 0, num_entries = sg_nents(args->sg);
617
618 mem_region->tag = args->tag;
619 mem_region->flags = args->flags;
620 mem_region->sender_id = drv_info->vm_id;
621 mem_region->attributes = ffa_memory_attributes_get(func_id);
622 ep_mem_access = buffer +
623 ffa_mem_desc_offset(buffer, 0, drv_info->version);
624 composite_offset = ffa_mem_desc_offset(buffer, args->nattrs,
625 drv_info->version);
626
627 for (idx = 0; idx < args->nattrs; idx++, ep_mem_access++) {
628 ep_mem_access->receiver = args->attrs[idx].receiver;
629 ep_mem_access->attrs = args->attrs[idx].attrs;
630 ep_mem_access->composite_off = composite_offset;
631 ep_mem_access->flag = 0;
632 ep_mem_access->reserved = 0;
633 }
634 mem_region->handle = 0;
635 mem_region->ep_count = args->nattrs;
636 if (drv_info->version <= FFA_VERSION_1_0) {
637 mem_region->ep_mem_size = 0;
638 } else {
639 mem_region->ep_mem_size = sizeof(*ep_mem_access);
640 mem_region->ep_mem_offset = sizeof(*mem_region);
641 memset(mem_region->reserved, 0, 12);
642 }
643
644 composite = buffer + composite_offset;
645 composite->total_pg_cnt = ffa_get_num_pages_sg(args->sg);
646 composite->addr_range_cnt = num_entries;
647 composite->reserved = 0;
648
649 length = composite_offset + CONSTITUENTS_OFFSET(num_entries);
650 frag_len = composite_offset + CONSTITUENTS_OFFSET(0);
651 if (frag_len > max_fragsize)
652 return -ENXIO;
653
654 if (!args->use_txbuf) {
655 addr = virt_to_phys(buffer);
656 buf_sz = max_fragsize / FFA_PAGE_SIZE;
657 }
658
659 constituents = buffer + frag_len;
660 idx = 0;
661 do {
662 if (frag_len == max_fragsize) {
663 rc = ffa_transmit_fragment(func_id, addr, buf_sz,
664 frag_len, length,
665 &args->g_handle, first);
666 if (rc < 0)
667 return -ENXIO;
668
669 first = false;
670 idx = 0;
671 frag_len = 0;
672 constituents = buffer;
673 }
674
675 if ((void *)constituents - buffer > max_fragsize) {
676 pr_err("Memory Region Fragment > Tx Buffer size\n");
677 return -EFAULT;
678 }
679
680 constituents->address = sg_phys(args->sg);
681 constituents->pg_cnt = args->sg->length / FFA_PAGE_SIZE;
682 constituents->reserved = 0;
683 constituents++;
684 frag_len += sizeof(struct ffa_mem_region_addr_range);
685 } while ((args->sg = sg_next(args->sg)));
686
687 return ffa_transmit_fragment(func_id, addr, buf_sz, frag_len,
688 length, &args->g_handle, first);
689 }
690
ffa_memory_ops(u32 func_id,struct ffa_mem_ops_args * args)691 static int ffa_memory_ops(u32 func_id, struct ffa_mem_ops_args *args)
692 {
693 int ret;
694 void *buffer;
695 size_t rxtx_bufsz = drv_info->rxtx_bufsz;
696
697 if (!args->use_txbuf) {
698 buffer = alloc_pages_exact(rxtx_bufsz, GFP_KERNEL);
699 if (!buffer)
700 return -ENOMEM;
701 } else {
702 buffer = drv_info->tx_buffer;
703 mutex_lock(&drv_info->tx_lock);
704 }
705
706 ret = ffa_setup_and_transmit(func_id, buffer, rxtx_bufsz, args);
707
708 if (args->use_txbuf)
709 mutex_unlock(&drv_info->tx_lock);
710 else
711 free_pages_exact(buffer, rxtx_bufsz);
712
713 return ret < 0 ? ret : 0;
714 }
715
ffa_memory_reclaim(u64 g_handle,u32 flags)716 static int ffa_memory_reclaim(u64 g_handle, u32 flags)
717 {
718 ffa_value_t ret;
719
720 invoke_ffa_fn((ffa_value_t){
721 .a0 = FFA_MEM_RECLAIM,
722 .a1 = HANDLE_LOW(g_handle), .a2 = HANDLE_HIGH(g_handle),
723 .a3 = flags,
724 }, &ret);
725
726 if (ret.a0 == FFA_ERROR)
727 return ffa_to_linux_errno((int)ret.a2);
728
729 return 0;
730 }
731
ffa_notification_bitmap_create(void)732 static int ffa_notification_bitmap_create(void)
733 {
734 ffa_value_t ret;
735 u16 vcpu_count = nr_cpu_ids;
736
737 invoke_ffa_fn((ffa_value_t){
738 .a0 = FFA_NOTIFICATION_BITMAP_CREATE,
739 .a1 = drv_info->vm_id, .a2 = vcpu_count,
740 }, &ret);
741
742 if (ret.a0 == FFA_ERROR)
743 return ffa_to_linux_errno((int)ret.a2);
744
745 return 0;
746 }
747
ffa_notification_bitmap_destroy(void)748 static int ffa_notification_bitmap_destroy(void)
749 {
750 ffa_value_t ret;
751
752 invoke_ffa_fn((ffa_value_t){
753 .a0 = FFA_NOTIFICATION_BITMAP_DESTROY,
754 .a1 = drv_info->vm_id,
755 }, &ret);
756
757 if (ret.a0 == FFA_ERROR)
758 return ffa_to_linux_errno((int)ret.a2);
759
760 return 0;
761 }
762
763 #define NOTIFICATION_LOW_MASK GENMASK(31, 0)
764 #define NOTIFICATION_HIGH_MASK GENMASK(63, 32)
765 #define NOTIFICATION_BITMAP_HIGH(x) \
766 ((u32)(FIELD_GET(NOTIFICATION_HIGH_MASK, (x))))
767 #define NOTIFICATION_BITMAP_LOW(x) \
768 ((u32)(FIELD_GET(NOTIFICATION_LOW_MASK, (x))))
769 #define PACK_NOTIFICATION_BITMAP(low, high) \
770 (FIELD_PREP(NOTIFICATION_LOW_MASK, (low)) | \
771 FIELD_PREP(NOTIFICATION_HIGH_MASK, (high)))
772
773 #define RECEIVER_VCPU_MASK GENMASK(31, 16)
774 #define PACK_NOTIFICATION_GET_RECEIVER_INFO(vcpu_r, r) \
775 (FIELD_PREP(RECEIVER_VCPU_MASK, (vcpu_r)) | \
776 FIELD_PREP(RECEIVER_ID_MASK, (r)))
777
778 #define NOTIFICATION_INFO_GET_MORE_PEND_MASK BIT(0)
779 #define NOTIFICATION_INFO_GET_ID_COUNT GENMASK(11, 7)
780 #define ID_LIST_MASK_64 GENMASK(51, 12)
781 #define ID_LIST_MASK_32 GENMASK(31, 12)
782 #define MAX_IDS_64 20
783 #define MAX_IDS_32 10
784
785 #define PER_VCPU_NOTIFICATION_FLAG BIT(0)
786 #define SECURE_PARTITION_BITMAP BIT(0)
787 #define NON_SECURE_VM_BITMAP BIT(1)
788 #define SPM_FRAMEWORK_BITMAP BIT(2)
789 #define NS_HYP_FRAMEWORK_BITMAP BIT(3)
790
ffa_notification_bind_common(u16 dst_id,u64 bitmap,u32 flags,bool is_bind)791 static int ffa_notification_bind_common(u16 dst_id, u64 bitmap,
792 u32 flags, bool is_bind)
793 {
794 ffa_value_t ret;
795 u32 func, src_dst_ids = PACK_TARGET_INFO(dst_id, drv_info->vm_id);
796
797 func = is_bind ? FFA_NOTIFICATION_BIND : FFA_NOTIFICATION_UNBIND;
798
799 invoke_ffa_fn((ffa_value_t){
800 .a0 = func, .a1 = src_dst_ids, .a2 = flags,
801 .a3 = NOTIFICATION_BITMAP_LOW(bitmap),
802 .a4 = NOTIFICATION_BITMAP_HIGH(bitmap),
803 }, &ret);
804
805 if (ret.a0 == FFA_ERROR)
806 return ffa_to_linux_errno((int)ret.a2);
807 else if (ret.a0 != FFA_SUCCESS)
808 return -EINVAL;
809
810 return 0;
811 }
812
813 static
ffa_notification_set(u16 src_id,u16 dst_id,u32 flags,u64 bitmap)814 int ffa_notification_set(u16 src_id, u16 dst_id, u32 flags, u64 bitmap)
815 {
816 ffa_value_t ret;
817 u32 src_dst_ids = PACK_TARGET_INFO(dst_id, src_id);
818
819 invoke_ffa_fn((ffa_value_t) {
820 .a0 = FFA_NOTIFICATION_SET, .a1 = src_dst_ids, .a2 = flags,
821 .a3 = NOTIFICATION_BITMAP_LOW(bitmap),
822 .a4 = NOTIFICATION_BITMAP_HIGH(bitmap),
823 }, &ret);
824
825 if (ret.a0 == FFA_ERROR)
826 return ffa_to_linux_errno((int)ret.a2);
827 else if (ret.a0 != FFA_SUCCESS)
828 return -EINVAL;
829
830 return 0;
831 }
832
833 struct ffa_notify_bitmaps {
834 u64 sp_map;
835 u64 vm_map;
836 u64 arch_map;
837 };
838
ffa_notification_get(u32 flags,struct ffa_notify_bitmaps * notify)839 static int ffa_notification_get(u32 flags, struct ffa_notify_bitmaps *notify)
840 {
841 ffa_value_t ret;
842 u16 src_id = drv_info->vm_id;
843 u16 cpu_id = smp_processor_id();
844 u32 rec_vcpu_ids = PACK_NOTIFICATION_GET_RECEIVER_INFO(cpu_id, src_id);
845
846 invoke_ffa_fn((ffa_value_t){
847 .a0 = FFA_NOTIFICATION_GET, .a1 = rec_vcpu_ids, .a2 = flags,
848 }, &ret);
849
850 if (ret.a0 == FFA_ERROR)
851 return ffa_to_linux_errno((int)ret.a2);
852 else if (ret.a0 != FFA_SUCCESS)
853 return -EINVAL; /* Something else went wrong. */
854
855 notify->sp_map = PACK_NOTIFICATION_BITMAP(ret.a2, ret.a3);
856 notify->vm_map = PACK_NOTIFICATION_BITMAP(ret.a4, ret.a5);
857 notify->arch_map = PACK_NOTIFICATION_BITMAP(ret.a6, ret.a7);
858
859 return 0;
860 }
861
862 struct ffa_dev_part_info {
863 ffa_sched_recv_cb callback;
864 void *cb_data;
865 rwlock_t rw_lock;
866 };
867
__do_sched_recv_cb(u16 part_id,u16 vcpu,bool is_per_vcpu)868 static void __do_sched_recv_cb(u16 part_id, u16 vcpu, bool is_per_vcpu)
869 {
870 struct ffa_dev_part_info *partition;
871 ffa_sched_recv_cb callback;
872 void *cb_data;
873
874 partition = xa_load(&drv_info->partition_info, part_id);
875 if (!partition) {
876 pr_err("%s: Invalid partition ID 0x%x\n", __func__, part_id);
877 return;
878 }
879
880 read_lock(&partition->rw_lock);
881 callback = partition->callback;
882 cb_data = partition->cb_data;
883 read_unlock(&partition->rw_lock);
884
885 if (callback)
886 callback(vcpu, is_per_vcpu, cb_data);
887 }
888
ffa_notification_info_get(void)889 static void ffa_notification_info_get(void)
890 {
891 int idx, list, max_ids, lists_cnt, ids_processed, ids_count[MAX_IDS_64];
892 bool is_64b_resp;
893 ffa_value_t ret;
894 u64 id_list;
895
896 do {
897 invoke_ffa_fn((ffa_value_t){
898 .a0 = FFA_FN_NATIVE(NOTIFICATION_INFO_GET),
899 }, &ret);
900
901 if (ret.a0 != FFA_FN_NATIVE(SUCCESS) && ret.a0 != FFA_SUCCESS) {
902 if (ret.a2 != FFA_RET_NO_DATA)
903 pr_err("Notification Info fetch failed: 0x%lx (0x%lx)",
904 ret.a0, ret.a2);
905 return;
906 }
907
908 is_64b_resp = (ret.a0 == FFA_FN64_SUCCESS);
909
910 ids_processed = 0;
911 lists_cnt = FIELD_GET(NOTIFICATION_INFO_GET_ID_COUNT, ret.a2);
912 if (is_64b_resp) {
913 max_ids = MAX_IDS_64;
914 id_list = FIELD_GET(ID_LIST_MASK_64, ret.a2);
915 } else {
916 max_ids = MAX_IDS_32;
917 id_list = FIELD_GET(ID_LIST_MASK_32, ret.a2);
918 }
919
920 for (idx = 0; idx < lists_cnt; idx++, id_list >>= 2)
921 ids_count[idx] = (id_list & 0x3) + 1;
922
923 /* Process IDs */
924 for (list = 0; list < lists_cnt; list++) {
925 u16 vcpu_id, part_id, *packed_id_list = (u16 *)&ret.a3;
926
927 if (ids_processed >= max_ids - 1)
928 break;
929
930 part_id = packed_id_list[ids_processed++];
931
932 if (ids_count[list] == 1) { /* Global Notification */
933 __do_sched_recv_cb(part_id, 0, false);
934 continue;
935 }
936
937 /* Per vCPU Notification */
938 for (idx = 0; idx < ids_count[list]; idx++) {
939 if (ids_processed >= max_ids - 1)
940 break;
941
942 vcpu_id = packed_id_list[ids_processed++];
943
944 __do_sched_recv_cb(part_id, vcpu_id, true);
945 }
946 }
947 } while (ret.a2 & NOTIFICATION_INFO_GET_MORE_PEND_MASK);
948 }
949
ffa_run(struct ffa_device * dev,u16 vcpu)950 static int ffa_run(struct ffa_device *dev, u16 vcpu)
951 {
952 ffa_value_t ret;
953 u32 target = dev->vm_id << 16 | vcpu;
954
955 invoke_ffa_fn((ffa_value_t){ .a0 = FFA_RUN, .a1 = target, }, &ret);
956
957 while (ret.a0 == FFA_INTERRUPT)
958 invoke_ffa_fn((ffa_value_t){ .a0 = FFA_RUN, .a1 = ret.a1, },
959 &ret);
960
961 if (ret.a0 == FFA_ERROR)
962 return ffa_to_linux_errno((int)ret.a2);
963
964 return 0;
965 }
966
ffa_drvinfo_flags_init(void)967 static void ffa_drvinfo_flags_init(void)
968 {
969 if (!ffa_features(FFA_FN_NATIVE(MEM_LEND), 0, NULL, NULL) ||
970 !ffa_features(FFA_FN_NATIVE(MEM_SHARE), 0, NULL, NULL))
971 drv_info->mem_ops_native = true;
972
973 if (!ffa_features(FFA_MSG_SEND_DIRECT_REQ2, 0, NULL, NULL) ||
974 !ffa_features(FFA_MSG_SEND_DIRECT_RESP2, 0, NULL, NULL))
975 drv_info->msg_direct_req2_supp = true;
976 }
977
ffa_api_version_get(void)978 static u32 ffa_api_version_get(void)
979 {
980 return drv_info->version;
981 }
982
ffa_partition_info_get(const char * uuid_str,struct ffa_partition_info * buffer)983 static int ffa_partition_info_get(const char *uuid_str,
984 struct ffa_partition_info *buffer)
985 {
986 int count;
987 uuid_t uuid;
988 struct ffa_partition_info *pbuf;
989
990 if (uuid_parse(uuid_str, &uuid)) {
991 pr_err("invalid uuid (%s)\n", uuid_str);
992 return -ENODEV;
993 }
994
995 count = ffa_partition_probe(&uuid, &pbuf);
996 if (count <= 0)
997 return -ENOENT;
998
999 memcpy(buffer, pbuf, sizeof(*pbuf) * count);
1000 kfree(pbuf);
1001 return 0;
1002 }
1003
ffa_mode_32bit_set(struct ffa_device * dev)1004 static void ffa_mode_32bit_set(struct ffa_device *dev)
1005 {
1006 dev->mode_32bit = true;
1007 }
1008
ffa_sync_send_receive(struct ffa_device * dev,struct ffa_send_direct_data * data)1009 static int ffa_sync_send_receive(struct ffa_device *dev,
1010 struct ffa_send_direct_data *data)
1011 {
1012 return ffa_msg_send_direct_req(drv_info->vm_id, dev->vm_id,
1013 dev->mode_32bit, data);
1014 }
1015
ffa_indirect_msg_send(struct ffa_device * dev,void * buf,size_t sz)1016 static int ffa_indirect_msg_send(struct ffa_device *dev, void *buf, size_t sz)
1017 {
1018 return ffa_msg_send2(drv_info->vm_id, dev->vm_id, buf, sz);
1019 }
1020
ffa_sync_send_receive2(struct ffa_device * dev,const uuid_t * uuid,struct ffa_send_direct_data2 * data)1021 static int ffa_sync_send_receive2(struct ffa_device *dev, const uuid_t *uuid,
1022 struct ffa_send_direct_data2 *data)
1023 {
1024 if (!drv_info->msg_direct_req2_supp)
1025 return -EOPNOTSUPP;
1026
1027 return ffa_msg_send_direct_req2(drv_info->vm_id, dev->vm_id,
1028 uuid, data);
1029 }
1030
ffa_memory_share(struct ffa_mem_ops_args * args)1031 static int ffa_memory_share(struct ffa_mem_ops_args *args)
1032 {
1033 if (drv_info->mem_ops_native)
1034 return ffa_memory_ops(FFA_FN_NATIVE(MEM_SHARE), args);
1035
1036 return ffa_memory_ops(FFA_MEM_SHARE, args);
1037 }
1038
ffa_memory_lend(struct ffa_mem_ops_args * args)1039 static int ffa_memory_lend(struct ffa_mem_ops_args *args)
1040 {
1041 /* Note that upon a successful MEM_LEND request the caller
1042 * must ensure that the memory region specified is not accessed
1043 * until a successful MEM_RECALIM call has been made.
1044 * On systems with a hypervisor present this will been enforced,
1045 * however on systems without a hypervisor the responsibility
1046 * falls to the calling kernel driver to prevent access.
1047 */
1048 if (drv_info->mem_ops_native)
1049 return ffa_memory_ops(FFA_FN_NATIVE(MEM_LEND), args);
1050
1051 return ffa_memory_ops(FFA_MEM_LEND, args);
1052 }
1053
1054 #define FFA_SECURE_PARTITION_ID_FLAG BIT(15)
1055
1056 #define ffa_notifications_disabled() (!drv_info->notif_enabled)
1057
1058 enum notify_type {
1059 NON_SECURE_VM,
1060 SECURE_PARTITION,
1061 FRAMEWORK,
1062 };
1063
1064 struct notifier_cb_info {
1065 struct hlist_node hnode;
1066 ffa_notifier_cb cb;
1067 void *cb_data;
1068 enum notify_type type;
1069 };
1070
ffa_sched_recv_cb_update(u16 part_id,ffa_sched_recv_cb callback,void * cb_data,bool is_registration)1071 static int ffa_sched_recv_cb_update(u16 part_id, ffa_sched_recv_cb callback,
1072 void *cb_data, bool is_registration)
1073 {
1074 struct ffa_dev_part_info *partition;
1075 bool cb_valid;
1076
1077 if (ffa_notifications_disabled())
1078 return -EOPNOTSUPP;
1079
1080 partition = xa_load(&drv_info->partition_info, part_id);
1081 if (!partition) {
1082 pr_err("%s: Invalid partition ID 0x%x\n", __func__, part_id);
1083 return -EINVAL;
1084 }
1085
1086 write_lock(&partition->rw_lock);
1087
1088 cb_valid = !!partition->callback;
1089 if (!(is_registration ^ cb_valid)) {
1090 write_unlock(&partition->rw_lock);
1091 return -EINVAL;
1092 }
1093
1094 partition->callback = callback;
1095 partition->cb_data = cb_data;
1096
1097 write_unlock(&partition->rw_lock);
1098 return 0;
1099 }
1100
ffa_sched_recv_cb_register(struct ffa_device * dev,ffa_sched_recv_cb cb,void * cb_data)1101 static int ffa_sched_recv_cb_register(struct ffa_device *dev,
1102 ffa_sched_recv_cb cb, void *cb_data)
1103 {
1104 return ffa_sched_recv_cb_update(dev->vm_id, cb, cb_data, true);
1105 }
1106
ffa_sched_recv_cb_unregister(struct ffa_device * dev)1107 static int ffa_sched_recv_cb_unregister(struct ffa_device *dev)
1108 {
1109 return ffa_sched_recv_cb_update(dev->vm_id, NULL, NULL, false);
1110 }
1111
ffa_notification_bind(u16 dst_id,u64 bitmap,u32 flags)1112 static int ffa_notification_bind(u16 dst_id, u64 bitmap, u32 flags)
1113 {
1114 return ffa_notification_bind_common(dst_id, bitmap, flags, true);
1115 }
1116
ffa_notification_unbind(u16 dst_id,u64 bitmap)1117 static int ffa_notification_unbind(u16 dst_id, u64 bitmap)
1118 {
1119 return ffa_notification_bind_common(dst_id, bitmap, 0, false);
1120 }
1121
1122 /* Should be called while the notify_lock is taken */
1123 static struct notifier_cb_info *
notifier_hash_node_get(u16 notify_id,enum notify_type type)1124 notifier_hash_node_get(u16 notify_id, enum notify_type type)
1125 {
1126 struct notifier_cb_info *node;
1127
1128 hash_for_each_possible(drv_info->notifier_hash, node, hnode, notify_id)
1129 if (type == node->type)
1130 return node;
1131
1132 return NULL;
1133 }
1134
1135 static int
update_notifier_cb(int notify_id,enum notify_type type,ffa_notifier_cb cb,void * cb_data,bool is_registration)1136 update_notifier_cb(int notify_id, enum notify_type type, ffa_notifier_cb cb,
1137 void *cb_data, bool is_registration)
1138 {
1139 struct notifier_cb_info *cb_info = NULL;
1140 bool cb_found;
1141
1142 cb_info = notifier_hash_node_get(notify_id, type);
1143 cb_found = !!cb_info;
1144
1145 if (!(is_registration ^ cb_found))
1146 return -EINVAL;
1147
1148 if (is_registration) {
1149 cb_info = kzalloc(sizeof(*cb_info), GFP_KERNEL);
1150 if (!cb_info)
1151 return -ENOMEM;
1152
1153 cb_info->type = type;
1154 cb_info->cb = cb;
1155 cb_info->cb_data = cb_data;
1156
1157 hash_add(drv_info->notifier_hash, &cb_info->hnode, notify_id);
1158 } else {
1159 hash_del(&cb_info->hnode);
1160 }
1161
1162 return 0;
1163 }
1164
ffa_notify_type_get(u16 vm_id)1165 static enum notify_type ffa_notify_type_get(u16 vm_id)
1166 {
1167 if (vm_id & FFA_SECURE_PARTITION_ID_FLAG)
1168 return SECURE_PARTITION;
1169 else
1170 return NON_SECURE_VM;
1171 }
1172
ffa_notify_relinquish(struct ffa_device * dev,int notify_id)1173 static int ffa_notify_relinquish(struct ffa_device *dev, int notify_id)
1174 {
1175 int rc;
1176 enum notify_type type = ffa_notify_type_get(dev->vm_id);
1177
1178 if (ffa_notifications_disabled())
1179 return -EOPNOTSUPP;
1180
1181 if (notify_id >= FFA_MAX_NOTIFICATIONS)
1182 return -EINVAL;
1183
1184 mutex_lock(&drv_info->notify_lock);
1185
1186 rc = update_notifier_cb(notify_id, type, NULL, NULL, false);
1187 if (rc) {
1188 pr_err("Could not unregister notification callback\n");
1189 mutex_unlock(&drv_info->notify_lock);
1190 return rc;
1191 }
1192
1193 rc = ffa_notification_unbind(dev->vm_id, BIT(notify_id));
1194
1195 mutex_unlock(&drv_info->notify_lock);
1196
1197 return rc;
1198 }
1199
ffa_notify_request(struct ffa_device * dev,bool is_per_vcpu,ffa_notifier_cb cb,void * cb_data,int notify_id)1200 static int ffa_notify_request(struct ffa_device *dev, bool is_per_vcpu,
1201 ffa_notifier_cb cb, void *cb_data, int notify_id)
1202 {
1203 int rc;
1204 u32 flags = 0;
1205 enum notify_type type = ffa_notify_type_get(dev->vm_id);
1206
1207 if (ffa_notifications_disabled())
1208 return -EOPNOTSUPP;
1209
1210 if (notify_id >= FFA_MAX_NOTIFICATIONS)
1211 return -EINVAL;
1212
1213 mutex_lock(&drv_info->notify_lock);
1214
1215 if (is_per_vcpu)
1216 flags = PER_VCPU_NOTIFICATION_FLAG;
1217
1218 rc = ffa_notification_bind(dev->vm_id, BIT(notify_id), flags);
1219 if (rc) {
1220 mutex_unlock(&drv_info->notify_lock);
1221 return rc;
1222 }
1223
1224 rc = update_notifier_cb(notify_id, type, cb, cb_data, true);
1225 if (rc) {
1226 pr_err("Failed to register callback for %d - %d\n",
1227 notify_id, rc);
1228 ffa_notification_unbind(dev->vm_id, BIT(notify_id));
1229 }
1230 mutex_unlock(&drv_info->notify_lock);
1231
1232 return rc;
1233 }
1234
ffa_notify_send(struct ffa_device * dev,int notify_id,bool is_per_vcpu,u16 vcpu)1235 static int ffa_notify_send(struct ffa_device *dev, int notify_id,
1236 bool is_per_vcpu, u16 vcpu)
1237 {
1238 u32 flags = 0;
1239
1240 if (ffa_notifications_disabled())
1241 return -EOPNOTSUPP;
1242
1243 if (is_per_vcpu)
1244 flags |= (PER_VCPU_NOTIFICATION_FLAG | vcpu << 16);
1245
1246 return ffa_notification_set(dev->vm_id, drv_info->vm_id, flags,
1247 BIT(notify_id));
1248 }
1249
handle_notif_callbacks(u64 bitmap,enum notify_type type)1250 static void handle_notif_callbacks(u64 bitmap, enum notify_type type)
1251 {
1252 int notify_id;
1253 struct notifier_cb_info *cb_info = NULL;
1254
1255 for (notify_id = 0; notify_id <= FFA_MAX_NOTIFICATIONS && bitmap;
1256 notify_id++, bitmap >>= 1) {
1257 if (!(bitmap & 1))
1258 continue;
1259
1260 mutex_lock(&drv_info->notify_lock);
1261 cb_info = notifier_hash_node_get(notify_id, type);
1262 mutex_unlock(&drv_info->notify_lock);
1263
1264 if (cb_info && cb_info->cb)
1265 cb_info->cb(notify_id, cb_info->cb_data);
1266 }
1267 }
1268
notif_get_and_handle(void * unused)1269 static void notif_get_and_handle(void *unused)
1270 {
1271 int rc;
1272 struct ffa_notify_bitmaps bitmaps;
1273
1274 rc = ffa_notification_get(SECURE_PARTITION_BITMAP |
1275 SPM_FRAMEWORK_BITMAP, &bitmaps);
1276 if (rc) {
1277 pr_err("Failed to retrieve notifications with %d!\n", rc);
1278 return;
1279 }
1280
1281 handle_notif_callbacks(bitmaps.vm_map, NON_SECURE_VM);
1282 handle_notif_callbacks(bitmaps.sp_map, SECURE_PARTITION);
1283 handle_notif_callbacks(bitmaps.arch_map, FRAMEWORK);
1284 }
1285
1286 static void
ffa_self_notif_handle(u16 vcpu,bool is_per_vcpu,void * cb_data)1287 ffa_self_notif_handle(u16 vcpu, bool is_per_vcpu, void *cb_data)
1288 {
1289 struct ffa_drv_info *info = cb_data;
1290
1291 if (!is_per_vcpu)
1292 notif_get_and_handle(info);
1293 else
1294 smp_call_function_single(vcpu, notif_get_and_handle, info, 0);
1295 }
1296
notif_pcpu_irq_work_fn(struct work_struct * work)1297 static void notif_pcpu_irq_work_fn(struct work_struct *work)
1298 {
1299 struct ffa_drv_info *info = container_of(work, struct ffa_drv_info,
1300 notif_pcpu_work);
1301
1302 ffa_self_notif_handle(smp_processor_id(), true, info);
1303 }
1304
1305 static const struct ffa_info_ops ffa_drv_info_ops = {
1306 .api_version_get = ffa_api_version_get,
1307 .partition_info_get = ffa_partition_info_get,
1308 };
1309
1310 static const struct ffa_msg_ops ffa_drv_msg_ops = {
1311 .mode_32bit_set = ffa_mode_32bit_set,
1312 .sync_send_receive = ffa_sync_send_receive,
1313 .indirect_send = ffa_indirect_msg_send,
1314 .sync_send_receive2 = ffa_sync_send_receive2,
1315 };
1316
1317 static const struct ffa_mem_ops ffa_drv_mem_ops = {
1318 .memory_reclaim = ffa_memory_reclaim,
1319 .memory_share = ffa_memory_share,
1320 .memory_lend = ffa_memory_lend,
1321 };
1322
1323 static const struct ffa_cpu_ops ffa_drv_cpu_ops = {
1324 .run = ffa_run,
1325 };
1326
1327 static const struct ffa_notifier_ops ffa_drv_notifier_ops = {
1328 .sched_recv_cb_register = ffa_sched_recv_cb_register,
1329 .sched_recv_cb_unregister = ffa_sched_recv_cb_unregister,
1330 .notify_request = ffa_notify_request,
1331 .notify_relinquish = ffa_notify_relinquish,
1332 .notify_send = ffa_notify_send,
1333 };
1334
1335 static const struct ffa_ops ffa_drv_ops = {
1336 .info_ops = &ffa_drv_info_ops,
1337 .msg_ops = &ffa_drv_msg_ops,
1338 .mem_ops = &ffa_drv_mem_ops,
1339 .cpu_ops = &ffa_drv_cpu_ops,
1340 .notifier_ops = &ffa_drv_notifier_ops,
1341 };
1342
ffa_device_match_uuid(struct ffa_device * ffa_dev,const uuid_t * uuid)1343 void ffa_device_match_uuid(struct ffa_device *ffa_dev, const uuid_t *uuid)
1344 {
1345 int count, idx;
1346 struct ffa_partition_info *pbuf, *tpbuf;
1347
1348 count = ffa_partition_probe(uuid, &pbuf);
1349 if (count <= 0)
1350 return;
1351
1352 for (idx = 0, tpbuf = pbuf; idx < count; idx++, tpbuf++)
1353 if (tpbuf->id == ffa_dev->vm_id)
1354 uuid_copy(&ffa_dev->uuid, uuid);
1355 kfree(pbuf);
1356 }
1357
1358 static int
ffa_bus_notifier(struct notifier_block * nb,unsigned long action,void * data)1359 ffa_bus_notifier(struct notifier_block *nb, unsigned long action, void *data)
1360 {
1361 struct device *dev = data;
1362 struct ffa_device *fdev = to_ffa_dev(dev);
1363
1364 if (action == BUS_NOTIFY_BIND_DRIVER) {
1365 struct ffa_driver *ffa_drv = to_ffa_driver(dev->driver);
1366 const struct ffa_device_id *id_table = ffa_drv->id_table;
1367
1368 /*
1369 * FF-A v1.1 provides UUID for each partition as part of the
1370 * discovery API, the discovered UUID must be populated in the
1371 * device's UUID and there is no need to workaround by copying
1372 * the same from the driver table.
1373 */
1374 if (uuid_is_null(&fdev->uuid))
1375 ffa_device_match_uuid(fdev, &id_table->uuid);
1376
1377 return NOTIFY_OK;
1378 }
1379
1380 return NOTIFY_DONE;
1381 }
1382
1383 static struct notifier_block ffa_bus_nb = {
1384 .notifier_call = ffa_bus_notifier,
1385 };
1386
ffa_setup_partitions(void)1387 static int ffa_setup_partitions(void)
1388 {
1389 int count, idx, ret;
1390 struct ffa_device *ffa_dev;
1391 struct ffa_dev_part_info *info;
1392 struct ffa_partition_info *pbuf, *tpbuf;
1393
1394 if (drv_info->version == FFA_VERSION_1_0) {
1395 ret = bus_register_notifier(&ffa_bus_type, &ffa_bus_nb);
1396 if (ret)
1397 pr_err("Failed to register FF-A bus notifiers\n");
1398 }
1399
1400 count = ffa_partition_probe(&uuid_null, &pbuf);
1401 if (count <= 0) {
1402 pr_info("%s: No partitions found, error %d\n", __func__, count);
1403 return -EINVAL;
1404 }
1405
1406 xa_init(&drv_info->partition_info);
1407 for (idx = 0, tpbuf = pbuf; idx < count; idx++, tpbuf++) {
1408 /* Note that if the UUID will be uuid_null, that will require
1409 * ffa_bus_notifier() to find the UUID of this partition id
1410 * with help of ffa_device_match_uuid(). FF-A v1.1 and above
1411 * provides UUID here for each partition as part of the
1412 * discovery API and the same is passed.
1413 */
1414 ffa_dev = ffa_device_register(tpbuf, &ffa_drv_ops);
1415 if (!ffa_dev) {
1416 pr_err("%s: failed to register partition ID 0x%x\n",
1417 __func__, tpbuf->id);
1418 continue;
1419 }
1420
1421 if (drv_info->version > FFA_VERSION_1_0 &&
1422 !(tpbuf->properties & FFA_PARTITION_AARCH64_EXEC))
1423 ffa_mode_32bit_set(ffa_dev);
1424
1425 info = kzalloc(sizeof(*info), GFP_KERNEL);
1426 if (!info) {
1427 ffa_device_unregister(ffa_dev);
1428 continue;
1429 }
1430 rwlock_init(&info->rw_lock);
1431 ret = xa_insert(&drv_info->partition_info, tpbuf->id,
1432 info, GFP_KERNEL);
1433 if (ret) {
1434 pr_err("%s: failed to save partition ID 0x%x - ret:%d\n",
1435 __func__, tpbuf->id, ret);
1436 ffa_device_unregister(ffa_dev);
1437 kfree(info);
1438 }
1439 }
1440
1441 kfree(pbuf);
1442
1443 /* Allocate for the host */
1444 info = kzalloc(sizeof(*info), GFP_KERNEL);
1445 if (!info) {
1446 /* Already registered devices are freed on bus_exit */
1447 ffa_partitions_cleanup();
1448 return -ENOMEM;
1449 }
1450
1451 rwlock_init(&info->rw_lock);
1452 ret = xa_insert(&drv_info->partition_info, drv_info->vm_id,
1453 info, GFP_KERNEL);
1454 if (ret) {
1455 pr_err("%s: failed to save Host partition ID 0x%x - ret:%d. Abort.\n",
1456 __func__, drv_info->vm_id, ret);
1457 kfree(info);
1458 /* Already registered devices are freed on bus_exit */
1459 ffa_partitions_cleanup();
1460 }
1461
1462 return ret;
1463 }
1464
ffa_partitions_cleanup(void)1465 static void ffa_partitions_cleanup(void)
1466 {
1467 struct ffa_dev_part_info *info;
1468 unsigned long idx;
1469
1470 xa_for_each(&drv_info->partition_info, idx, info) {
1471 xa_erase(&drv_info->partition_info, idx);
1472 kfree(info);
1473 }
1474
1475 xa_destroy(&drv_info->partition_info);
1476 }
1477
1478 /* FFA FEATURE IDs */
1479 #define FFA_FEAT_NOTIFICATION_PENDING_INT (1)
1480 #define FFA_FEAT_SCHEDULE_RECEIVER_INT (2)
1481 #define FFA_FEAT_MANAGED_EXIT_INT (3)
1482
ffa_sched_recv_irq_handler(int irq,void * irq_data)1483 static irqreturn_t ffa_sched_recv_irq_handler(int irq, void *irq_data)
1484 {
1485 struct ffa_pcpu_irq *pcpu = irq_data;
1486 struct ffa_drv_info *info = pcpu->info;
1487
1488 queue_work(info->notif_pcpu_wq, &info->sched_recv_irq_work);
1489
1490 return IRQ_HANDLED;
1491 }
1492
notif_pend_irq_handler(int irq,void * irq_data)1493 static irqreturn_t notif_pend_irq_handler(int irq, void *irq_data)
1494 {
1495 struct ffa_pcpu_irq *pcpu = irq_data;
1496 struct ffa_drv_info *info = pcpu->info;
1497
1498 queue_work_on(smp_processor_id(), info->notif_pcpu_wq,
1499 &info->notif_pcpu_work);
1500
1501 return IRQ_HANDLED;
1502 }
1503
ffa_sched_recv_irq_work_fn(struct work_struct * work)1504 static void ffa_sched_recv_irq_work_fn(struct work_struct *work)
1505 {
1506 ffa_notification_info_get();
1507 }
1508
ffa_irq_map(u32 id)1509 static int ffa_irq_map(u32 id)
1510 {
1511 char *err_str;
1512 int ret, irq, intid;
1513
1514 if (id == FFA_FEAT_NOTIFICATION_PENDING_INT)
1515 err_str = "Notification Pending Interrupt";
1516 else if (id == FFA_FEAT_SCHEDULE_RECEIVER_INT)
1517 err_str = "Schedule Receiver Interrupt";
1518 else
1519 err_str = "Unknown ID";
1520
1521 /* The returned intid is assumed to be SGI donated to NS world */
1522 ret = ffa_features(id, 0, &intid, NULL);
1523 if (ret < 0) {
1524 if (ret != -EOPNOTSUPP)
1525 pr_err("Failed to retrieve FF-A %s %u\n", err_str, id);
1526 return ret;
1527 }
1528
1529 if (acpi_disabled) {
1530 struct of_phandle_args oirq = {};
1531 struct device_node *gic;
1532
1533 /* Only GICv3 supported currently with the device tree */
1534 gic = of_find_compatible_node(NULL, NULL, "arm,gic-v3");
1535 if (!gic)
1536 return -ENXIO;
1537
1538 oirq.np = gic;
1539 oirq.args_count = 1;
1540 oirq.args[0] = intid;
1541 irq = irq_create_of_mapping(&oirq);
1542 of_node_put(gic);
1543 #ifdef CONFIG_ACPI
1544 } else {
1545 irq = acpi_register_gsi(NULL, intid, ACPI_EDGE_SENSITIVE,
1546 ACPI_ACTIVE_HIGH);
1547 #endif
1548 }
1549
1550 if (irq <= 0) {
1551 pr_err("Failed to create IRQ mapping!\n");
1552 return -ENODATA;
1553 }
1554
1555 return irq;
1556 }
1557
ffa_irq_unmap(unsigned int irq)1558 static void ffa_irq_unmap(unsigned int irq)
1559 {
1560 if (!irq)
1561 return;
1562 irq_dispose_mapping(irq);
1563 }
1564
ffa_cpuhp_pcpu_irq_enable(unsigned int cpu)1565 static int ffa_cpuhp_pcpu_irq_enable(unsigned int cpu)
1566 {
1567 if (drv_info->sched_recv_irq)
1568 enable_percpu_irq(drv_info->sched_recv_irq, IRQ_TYPE_NONE);
1569 if (drv_info->notif_pend_irq)
1570 enable_percpu_irq(drv_info->notif_pend_irq, IRQ_TYPE_NONE);
1571 return 0;
1572 }
1573
ffa_cpuhp_pcpu_irq_disable(unsigned int cpu)1574 static int ffa_cpuhp_pcpu_irq_disable(unsigned int cpu)
1575 {
1576 if (drv_info->sched_recv_irq)
1577 disable_percpu_irq(drv_info->sched_recv_irq);
1578 if (drv_info->notif_pend_irq)
1579 disable_percpu_irq(drv_info->notif_pend_irq);
1580 return 0;
1581 }
1582
ffa_uninit_pcpu_irq(void)1583 static void ffa_uninit_pcpu_irq(void)
1584 {
1585 if (drv_info->cpuhp_state) {
1586 cpuhp_remove_state(drv_info->cpuhp_state);
1587 drv_info->cpuhp_state = 0;
1588 }
1589
1590 if (drv_info->notif_pcpu_wq) {
1591 destroy_workqueue(drv_info->notif_pcpu_wq);
1592 drv_info->notif_pcpu_wq = NULL;
1593 }
1594
1595 if (drv_info->sched_recv_irq)
1596 free_percpu_irq(drv_info->sched_recv_irq, drv_info->irq_pcpu);
1597
1598 if (drv_info->notif_pend_irq)
1599 free_percpu_irq(drv_info->notif_pend_irq, drv_info->irq_pcpu);
1600
1601 if (drv_info->irq_pcpu) {
1602 free_percpu(drv_info->irq_pcpu);
1603 drv_info->irq_pcpu = NULL;
1604 }
1605 }
1606
ffa_init_pcpu_irq(void)1607 static int ffa_init_pcpu_irq(void)
1608 {
1609 struct ffa_pcpu_irq __percpu *irq_pcpu;
1610 int ret, cpu;
1611
1612 irq_pcpu = alloc_percpu(struct ffa_pcpu_irq);
1613 if (!irq_pcpu)
1614 return -ENOMEM;
1615
1616 for_each_present_cpu(cpu)
1617 per_cpu_ptr(irq_pcpu, cpu)->info = drv_info;
1618
1619 drv_info->irq_pcpu = irq_pcpu;
1620
1621 if (drv_info->sched_recv_irq) {
1622 ret = request_percpu_irq(drv_info->sched_recv_irq,
1623 ffa_sched_recv_irq_handler,
1624 "ARM-FFA-SRI", irq_pcpu);
1625 if (ret) {
1626 pr_err("Error registering percpu SRI nIRQ %d : %d\n",
1627 drv_info->sched_recv_irq, ret);
1628 drv_info->sched_recv_irq = 0;
1629 return ret;
1630 }
1631 }
1632
1633 if (drv_info->notif_pend_irq) {
1634 ret = request_percpu_irq(drv_info->notif_pend_irq,
1635 notif_pend_irq_handler,
1636 "ARM-FFA-NPI", irq_pcpu);
1637 if (ret) {
1638 pr_err("Error registering percpu NPI nIRQ %d : %d\n",
1639 drv_info->notif_pend_irq, ret);
1640 drv_info->notif_pend_irq = 0;
1641 return ret;
1642 }
1643 }
1644
1645 INIT_WORK(&drv_info->sched_recv_irq_work, ffa_sched_recv_irq_work_fn);
1646 INIT_WORK(&drv_info->notif_pcpu_work, notif_pcpu_irq_work_fn);
1647 drv_info->notif_pcpu_wq = create_workqueue("ffa_pcpu_irq_notification");
1648 if (!drv_info->notif_pcpu_wq)
1649 return -EINVAL;
1650
1651 ret = cpuhp_setup_state(CPUHP_AP_ONLINE_DYN, "ffa/pcpu-irq:starting",
1652 ffa_cpuhp_pcpu_irq_enable,
1653 ffa_cpuhp_pcpu_irq_disable);
1654
1655 if (ret < 0)
1656 return ret;
1657
1658 drv_info->cpuhp_state = ret;
1659 return 0;
1660 }
1661
ffa_notifications_cleanup(void)1662 static void ffa_notifications_cleanup(void)
1663 {
1664 ffa_uninit_pcpu_irq();
1665 ffa_irq_unmap(drv_info->sched_recv_irq);
1666 drv_info->sched_recv_irq = 0;
1667 ffa_irq_unmap(drv_info->notif_pend_irq);
1668 drv_info->notif_pend_irq = 0;
1669
1670 if (drv_info->bitmap_created) {
1671 ffa_notification_bitmap_destroy();
1672 drv_info->bitmap_created = false;
1673 }
1674 drv_info->notif_enabled = false;
1675 }
1676
ffa_notifications_setup(void)1677 static void ffa_notifications_setup(void)
1678 {
1679 int ret;
1680
1681 ret = ffa_features(FFA_NOTIFICATION_BITMAP_CREATE, 0, NULL, NULL);
1682 if (!ret) {
1683 ret = ffa_notification_bitmap_create();
1684 if (ret) {
1685 pr_err("Notification bitmap create error %d\n", ret);
1686 return;
1687 }
1688
1689 drv_info->bitmap_created = true;
1690 }
1691
1692 ret = ffa_irq_map(FFA_FEAT_SCHEDULE_RECEIVER_INT);
1693 if (ret > 0)
1694 drv_info->sched_recv_irq = ret;
1695
1696 ret = ffa_irq_map(FFA_FEAT_NOTIFICATION_PENDING_INT);
1697 if (ret > 0)
1698 drv_info->notif_pend_irq = ret;
1699
1700 if (!drv_info->sched_recv_irq && !drv_info->notif_pend_irq)
1701 goto cleanup;
1702
1703 ret = ffa_init_pcpu_irq();
1704 if (ret)
1705 goto cleanup;
1706
1707 hash_init(drv_info->notifier_hash);
1708 mutex_init(&drv_info->notify_lock);
1709
1710 drv_info->notif_enabled = true;
1711 return;
1712 cleanup:
1713 pr_info("Notification setup failed %d, not enabled\n", ret);
1714 ffa_notifications_cleanup();
1715 }
1716
ffa_init(void)1717 static int __init ffa_init(void)
1718 {
1719 int ret;
1720 u32 buf_sz;
1721 size_t rxtx_bufsz = SZ_4K;
1722
1723 ret = ffa_transport_init(&invoke_ffa_fn);
1724 if (ret)
1725 return ret;
1726
1727 drv_info = kzalloc(sizeof(*drv_info), GFP_KERNEL);
1728 if (!drv_info)
1729 return -ENOMEM;
1730
1731 ret = ffa_version_check(&drv_info->version);
1732 if (ret)
1733 goto free_drv_info;
1734
1735 if (ffa_id_get(&drv_info->vm_id)) {
1736 pr_err("failed to obtain VM id for self\n");
1737 ret = -ENODEV;
1738 goto free_drv_info;
1739 }
1740
1741 ret = ffa_features(FFA_FN_NATIVE(RXTX_MAP), 0, &buf_sz, NULL);
1742 if (!ret) {
1743 if (RXTX_MAP_MIN_BUFSZ(buf_sz) == 1)
1744 rxtx_bufsz = SZ_64K;
1745 else if (RXTX_MAP_MIN_BUFSZ(buf_sz) == 2)
1746 rxtx_bufsz = SZ_16K;
1747 else
1748 rxtx_bufsz = SZ_4K;
1749 }
1750
1751 drv_info->rxtx_bufsz = rxtx_bufsz;
1752 drv_info->rx_buffer = alloc_pages_exact(rxtx_bufsz, GFP_KERNEL);
1753 if (!drv_info->rx_buffer) {
1754 ret = -ENOMEM;
1755 goto free_pages;
1756 }
1757
1758 drv_info->tx_buffer = alloc_pages_exact(rxtx_bufsz, GFP_KERNEL);
1759 if (!drv_info->tx_buffer) {
1760 ret = -ENOMEM;
1761 goto free_pages;
1762 }
1763
1764 ret = ffa_rxtx_map(virt_to_phys(drv_info->tx_buffer),
1765 virt_to_phys(drv_info->rx_buffer),
1766 rxtx_bufsz / FFA_PAGE_SIZE);
1767 if (ret) {
1768 pr_err("failed to register FFA RxTx buffers\n");
1769 goto free_pages;
1770 }
1771
1772 mutex_init(&drv_info->rx_lock);
1773 mutex_init(&drv_info->tx_lock);
1774
1775 ffa_drvinfo_flags_init();
1776
1777 ffa_notifications_setup();
1778
1779 ret = ffa_setup_partitions();
1780 if (ret) {
1781 pr_err("failed to setup partitions\n");
1782 goto cleanup_notifs;
1783 }
1784
1785 ret = ffa_sched_recv_cb_update(drv_info->vm_id, ffa_self_notif_handle,
1786 drv_info, true);
1787 if (ret)
1788 pr_info("Failed to register driver sched callback %d\n", ret);
1789
1790 return 0;
1791
1792 cleanup_notifs:
1793 ffa_notifications_cleanup();
1794 free_pages:
1795 if (drv_info->tx_buffer)
1796 free_pages_exact(drv_info->tx_buffer, rxtx_bufsz);
1797 free_pages_exact(drv_info->rx_buffer, rxtx_bufsz);
1798 free_drv_info:
1799 kfree(drv_info);
1800 return ret;
1801 }
1802 module_init(ffa_init);
1803
ffa_exit(void)1804 static void __exit ffa_exit(void)
1805 {
1806 ffa_notifications_cleanup();
1807 ffa_partitions_cleanup();
1808 ffa_rxtx_unmap(drv_info->vm_id);
1809 free_pages_exact(drv_info->tx_buffer, drv_info->rxtx_bufsz);
1810 free_pages_exact(drv_info->rx_buffer, drv_info->rxtx_bufsz);
1811 kfree(drv_info);
1812 }
1813 module_exit(ffa_exit);
1814
1815 MODULE_ALIAS("arm-ffa");
1816 MODULE_AUTHOR("Sudeep Holla <sudeep.holla@arm.com>");
1817 MODULE_DESCRIPTION("Arm FF-A interface driver");
1818 MODULE_LICENSE("GPL v2");
1819