1824 lines · c
1// SPDX-License-Identifier: GPL-2.0-only2/*3 * Arm Firmware Framework for ARMv8-A(FFA) interface driver4 *5 * The Arm FFA specification[1] describes a software architecture to6 * leverages the virtualization extension to isolate software images7 * provided by an ecosystem of vendors from each other and describes8 * interfaces that standardize communication between the various software9 * images including communication between images in the Secure world and10 * Normal world. Any Hypervisor could use the FFA interfaces to enable11 * communication between VMs it manages.12 *13 * The Hypervisor a.k.a Partition managers in FFA terminology can assign14 * system resources(Memory regions, Devices, CPU cycles) to the partitions15 * and manage isolation amongst them.16 *17 * [1] https://developer.arm.com/docs/den0077/latest18 *19 * Copyright (C) 2021 ARM Ltd.20 */21 22#define DRIVER_NAME "ARM FF-A"23#define pr_fmt(fmt) DRIVER_NAME ": " fmt24 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_148#define FFA_MIN_VERSION FFA_VERSION_1_049 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 6461 62static ffa_fn *invoke_ffa_fn;63 64static 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 79static 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 88struct ffa_pcpu_irq {89 struct ffa_drv_info *info;90};91 92struct 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 116static struct ffa_drv_info *drv_info;117static void ffa_partitions_cleanup(void);118 119/*120 * The driver must be able to support all the versions from the earliest121 * supported FFA_MIN_VERSION to the latest supported FFA_DRIVER_VERSION.122 * The specification states that if firmware supports a FFA implementation123 * that is incompatible with and at a greater version number than specified124 * by the caller(FFA_DRIVER_VERSION passed as parameter to FFA_VERSION),125 * it must return the NOT_SUPPORTED error code.126 */127static 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 140static 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 170static 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 186static 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 201static 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 215static 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 */244static int245__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))))298static int299__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 */339static int340ffa_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 else356 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 else371 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 else377 *buffer = pbuf;378 379 return count;380}381 382#define VM_ID_MASK GENMASK(15, 0)383static 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 399static 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 411static 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 448static 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 = 0471 }, &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 480static 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 511static 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 543static 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 570static int571ffa_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 580static 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 591static u16 ffa_memory_attributes_get(u32 func_id)592{593 /*594 * For the memory lend or donate operation, if the receiver is a PE or595 * a proxy endpoint, the owner/sender must not specify the attributes596 */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 604static int605ffa_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 691static 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 else711 free_pages_exact(buffer, rxtx_bufsz);712 713 return ret < 0 ? ret : 0;714}715 716static 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 732static 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 748static 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 20783#define MAX_IDS_32 10784 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 791static 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 813static814int 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 833struct ffa_notify_bitmaps {834 u64 sp_map;835 u64 vm_map;836 u64 arch_map;837};838 839static 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 862struct ffa_dev_part_info {863 ffa_sched_recv_cb callback;864 void *cb_data;865 rwlock_t rw_lock;866};867 868static 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 889static 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 950static 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 967static 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 978static u32 ffa_api_version_get(void)979{980 return drv_info->version;981}982 983static 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 1004static void ffa_mode_32bit_set(struct ffa_device *dev)1005{1006 dev->mode_32bit = true;1007}1008 1009static 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 1016static 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 1021static 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 1031static 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 1039static int ffa_memory_lend(struct ffa_mem_ops_args *args)1040{1041 /* Note that upon a successful MEM_LEND request the caller1042 * must ensure that the memory region specified is not accessed1043 * 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 responsibility1046 * 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 1058enum notify_type {1059 NON_SECURE_VM,1060 SECURE_PARTITION,1061 FRAMEWORK,1062};1063 1064struct notifier_cb_info {1065 struct hlist_node hnode;1066 ffa_notifier_cb cb;1067 void *cb_data;1068 enum notify_type type;1069};1070 1071static 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 1101static 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 1107static 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 1112static 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 1117static 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 */1123static struct notifier_cb_info *1124notifier_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 1135static int1136update_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 1165static 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 else1170 return NON_SECURE_VM;1171}1172 1173static 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 1200static 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 1235static 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 1250static 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 1269static 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 1286static void1287ffa_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 else1294 smp_call_function_single(vcpu, notif_get_and_handle, info, 0);1295}1296 1297static 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 1305static 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 1310static 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 1317static 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 1323static const struct ffa_cpu_ops ffa_drv_cpu_ops = {1324 .run = ffa_run,1325};1326 1327static 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 1335static 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 1343void 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 1358static int1359ffa_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 the1370 * discovery API, the discovered UUID must be populated in the1371 * device's UUID and there is no need to workaround by copying1372 * 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 1383static struct notifier_block ffa_bus_nb = {1384 .notifier_call = ffa_bus_notifier,1385};1386 1387static int ffa_setup_partitions(void)1388{1389 int count, idx, ret;1390 uuid_t uuid;1391 struct ffa_device *ffa_dev;1392 struct ffa_dev_part_info *info;1393 struct ffa_partition_info *pbuf, *tpbuf;1394 1395 if (drv_info->version == FFA_VERSION_1_0) {1396 ret = bus_register_notifier(&ffa_bus_type, &ffa_bus_nb);1397 if (ret)1398 pr_err("Failed to register FF-A bus notifiers\n");1399 }1400 1401 count = ffa_partition_probe(&uuid_null, &pbuf);1402 if (count <= 0) {1403 pr_info("%s: No partitions found, error %d\n", __func__, count);1404 return -EINVAL;1405 }1406 1407 xa_init(&drv_info->partition_info);1408 for (idx = 0, tpbuf = pbuf; idx < count; idx++, tpbuf++) {1409 import_uuid(&uuid, (u8 *)tpbuf->uuid);1410 1411 /* Note that if the UUID will be uuid_null, that will require1412 * ffa_bus_notifier() to find the UUID of this partition id1413 * with help of ffa_device_match_uuid(). FF-A v1.1 and above1414 * provides UUID here for each partition as part of the1415 * discovery API and the same is passed.1416 */1417 ffa_dev = ffa_device_register(&uuid, tpbuf->id, &ffa_drv_ops);1418 if (!ffa_dev) {1419 pr_err("%s: failed to register partition ID 0x%x\n",1420 __func__, tpbuf->id);1421 continue;1422 }1423 1424 ffa_dev->properties = tpbuf->properties;1425 1426 if (drv_info->version > FFA_VERSION_1_0 &&1427 !(tpbuf->properties & FFA_PARTITION_AARCH64_EXEC))1428 ffa_mode_32bit_set(ffa_dev);1429 1430 info = kzalloc(sizeof(*info), GFP_KERNEL);1431 if (!info) {1432 ffa_device_unregister(ffa_dev);1433 continue;1434 }1435 rwlock_init(&info->rw_lock);1436 ret = xa_insert(&drv_info->partition_info, tpbuf->id,1437 info, GFP_KERNEL);1438 if (ret) {1439 pr_err("%s: failed to save partition ID 0x%x - ret:%d\n",1440 __func__, tpbuf->id, ret);1441 ffa_device_unregister(ffa_dev);1442 kfree(info);1443 }1444 }1445 1446 kfree(pbuf);1447 1448 /* Allocate for the host */1449 info = kzalloc(sizeof(*info), GFP_KERNEL);1450 if (!info) {1451 /* Already registered devices are freed on bus_exit */1452 ffa_partitions_cleanup();1453 return -ENOMEM;1454 }1455 1456 rwlock_init(&info->rw_lock);1457 ret = xa_insert(&drv_info->partition_info, drv_info->vm_id,1458 info, GFP_KERNEL);1459 if (ret) {1460 pr_err("%s: failed to save Host partition ID 0x%x - ret:%d. Abort.\n",1461 __func__, drv_info->vm_id, ret);1462 kfree(info);1463 /* Already registered devices are freed on bus_exit */1464 ffa_partitions_cleanup();1465 }1466 1467 return ret;1468}1469 1470static void ffa_partitions_cleanup(void)1471{1472 struct ffa_dev_part_info *info;1473 unsigned long idx;1474 1475 xa_for_each(&drv_info->partition_info, idx, info) {1476 xa_erase(&drv_info->partition_info, idx);1477 kfree(info);1478 }1479 1480 xa_destroy(&drv_info->partition_info);1481}1482 1483/* FFA FEATURE IDs */1484#define FFA_FEAT_NOTIFICATION_PENDING_INT (1)1485#define FFA_FEAT_SCHEDULE_RECEIVER_INT (2)1486#define FFA_FEAT_MANAGED_EXIT_INT (3)1487 1488static irqreturn_t ffa_sched_recv_irq_handler(int irq, void *irq_data)1489{1490 struct ffa_pcpu_irq *pcpu = irq_data;1491 struct ffa_drv_info *info = pcpu->info;1492 1493 queue_work(info->notif_pcpu_wq, &info->sched_recv_irq_work);1494 1495 return IRQ_HANDLED;1496}1497 1498static irqreturn_t notif_pend_irq_handler(int irq, void *irq_data)1499{1500 struct ffa_pcpu_irq *pcpu = irq_data;1501 struct ffa_drv_info *info = pcpu->info;1502 1503 queue_work_on(smp_processor_id(), info->notif_pcpu_wq,1504 &info->notif_pcpu_work);1505 1506 return IRQ_HANDLED;1507}1508 1509static void ffa_sched_recv_irq_work_fn(struct work_struct *work)1510{1511 ffa_notification_info_get();1512}1513 1514static int ffa_irq_map(u32 id)1515{1516 char *err_str;1517 int ret, irq, intid;1518 1519 if (id == FFA_FEAT_NOTIFICATION_PENDING_INT)1520 err_str = "Notification Pending Interrupt";1521 else if (id == FFA_FEAT_SCHEDULE_RECEIVER_INT)1522 err_str = "Schedule Receiver Interrupt";1523 else1524 err_str = "Unknown ID";1525 1526 /* The returned intid is assumed to be SGI donated to NS world */1527 ret = ffa_features(id, 0, &intid, NULL);1528 if (ret < 0) {1529 if (ret != -EOPNOTSUPP)1530 pr_err("Failed to retrieve FF-A %s %u\n", err_str, id);1531 return ret;1532 }1533 1534 if (acpi_disabled) {1535 struct of_phandle_args oirq = {};1536 struct device_node *gic;1537 1538 /* Only GICv3 supported currently with the device tree */1539 gic = of_find_compatible_node(NULL, NULL, "arm,gic-v3");1540 if (!gic)1541 return -ENXIO;1542 1543 oirq.np = gic;1544 oirq.args_count = 1;1545 oirq.args[0] = intid;1546 irq = irq_create_of_mapping(&oirq);1547 of_node_put(gic);1548#ifdef CONFIG_ACPI1549 } else {1550 irq = acpi_register_gsi(NULL, intid, ACPI_EDGE_SENSITIVE,1551 ACPI_ACTIVE_HIGH);1552#endif1553 }1554 1555 if (irq <= 0) {1556 pr_err("Failed to create IRQ mapping!\n");1557 return -ENODATA;1558 }1559 1560 return irq;1561}1562 1563static void ffa_irq_unmap(unsigned int irq)1564{1565 if (!irq)1566 return;1567 irq_dispose_mapping(irq);1568}1569 1570static int ffa_cpuhp_pcpu_irq_enable(unsigned int cpu)1571{1572 if (drv_info->sched_recv_irq)1573 enable_percpu_irq(drv_info->sched_recv_irq, IRQ_TYPE_NONE);1574 if (drv_info->notif_pend_irq)1575 enable_percpu_irq(drv_info->notif_pend_irq, IRQ_TYPE_NONE);1576 return 0;1577}1578 1579static int ffa_cpuhp_pcpu_irq_disable(unsigned int cpu)1580{1581 if (drv_info->sched_recv_irq)1582 disable_percpu_irq(drv_info->sched_recv_irq);1583 if (drv_info->notif_pend_irq)1584 disable_percpu_irq(drv_info->notif_pend_irq);1585 return 0;1586}1587 1588static void ffa_uninit_pcpu_irq(void)1589{1590 if (drv_info->cpuhp_state) {1591 cpuhp_remove_state(drv_info->cpuhp_state);1592 drv_info->cpuhp_state = 0;1593 }1594 1595 if (drv_info->notif_pcpu_wq) {1596 destroy_workqueue(drv_info->notif_pcpu_wq);1597 drv_info->notif_pcpu_wq = NULL;1598 }1599 1600 if (drv_info->sched_recv_irq)1601 free_percpu_irq(drv_info->sched_recv_irq, drv_info->irq_pcpu);1602 1603 if (drv_info->notif_pend_irq)1604 free_percpu_irq(drv_info->notif_pend_irq, drv_info->irq_pcpu);1605 1606 if (drv_info->irq_pcpu) {1607 free_percpu(drv_info->irq_pcpu);1608 drv_info->irq_pcpu = NULL;1609 }1610}1611 1612static int ffa_init_pcpu_irq(void)1613{1614 struct ffa_pcpu_irq __percpu *irq_pcpu;1615 int ret, cpu;1616 1617 irq_pcpu = alloc_percpu(struct ffa_pcpu_irq);1618 if (!irq_pcpu)1619 return -ENOMEM;1620 1621 for_each_present_cpu(cpu)1622 per_cpu_ptr(irq_pcpu, cpu)->info = drv_info;1623 1624 drv_info->irq_pcpu = irq_pcpu;1625 1626 if (drv_info->sched_recv_irq) {1627 ret = request_percpu_irq(drv_info->sched_recv_irq,1628 ffa_sched_recv_irq_handler,1629 "ARM-FFA-SRI", irq_pcpu);1630 if (ret) {1631 pr_err("Error registering percpu SRI nIRQ %d : %d\n",1632 drv_info->sched_recv_irq, ret);1633 drv_info->sched_recv_irq = 0;1634 return ret;1635 }1636 }1637 1638 if (drv_info->notif_pend_irq) {1639 ret = request_percpu_irq(drv_info->notif_pend_irq,1640 notif_pend_irq_handler,1641 "ARM-FFA-NPI", irq_pcpu);1642 if (ret) {1643 pr_err("Error registering percpu NPI nIRQ %d : %d\n",1644 drv_info->notif_pend_irq, ret);1645 drv_info->notif_pend_irq = 0;1646 return ret;1647 }1648 }1649 1650 INIT_WORK(&drv_info->sched_recv_irq_work, ffa_sched_recv_irq_work_fn);1651 INIT_WORK(&drv_info->notif_pcpu_work, notif_pcpu_irq_work_fn);1652 drv_info->notif_pcpu_wq = create_workqueue("ffa_pcpu_irq_notification");1653 if (!drv_info->notif_pcpu_wq)1654 return -EINVAL;1655 1656 ret = cpuhp_setup_state(CPUHP_AP_ONLINE_DYN, "ffa/pcpu-irq:starting",1657 ffa_cpuhp_pcpu_irq_enable,1658 ffa_cpuhp_pcpu_irq_disable);1659 1660 if (ret < 0)1661 return ret;1662 1663 drv_info->cpuhp_state = ret;1664 return 0;1665}1666 1667static void ffa_notifications_cleanup(void)1668{1669 ffa_uninit_pcpu_irq();1670 ffa_irq_unmap(drv_info->sched_recv_irq);1671 drv_info->sched_recv_irq = 0;1672 ffa_irq_unmap(drv_info->notif_pend_irq);1673 drv_info->notif_pend_irq = 0;1674 1675 if (drv_info->bitmap_created) {1676 ffa_notification_bitmap_destroy();1677 drv_info->bitmap_created = false;1678 }1679 drv_info->notif_enabled = false;1680}1681 1682static void ffa_notifications_setup(void)1683{1684 int ret;1685 1686 ret = ffa_features(FFA_NOTIFICATION_BITMAP_CREATE, 0, NULL, NULL);1687 if (!ret) {1688 ret = ffa_notification_bitmap_create();1689 if (ret) {1690 pr_err("Notification bitmap create error %d\n", ret);1691 return;1692 }1693 1694 drv_info->bitmap_created = true;1695 }1696 1697 ret = ffa_irq_map(FFA_FEAT_SCHEDULE_RECEIVER_INT);1698 if (ret > 0)1699 drv_info->sched_recv_irq = ret;1700 1701 ret = ffa_irq_map(FFA_FEAT_NOTIFICATION_PENDING_INT);1702 if (ret > 0)1703 drv_info->notif_pend_irq = ret;1704 1705 if (!drv_info->sched_recv_irq && !drv_info->notif_pend_irq)1706 goto cleanup;1707 1708 ret = ffa_init_pcpu_irq();1709 if (ret)1710 goto cleanup;1711 1712 hash_init(drv_info->notifier_hash);1713 mutex_init(&drv_info->notify_lock);1714 1715 drv_info->notif_enabled = true;1716 return;1717cleanup:1718 pr_info("Notification setup failed %d, not enabled\n", ret);1719 ffa_notifications_cleanup();1720}1721 1722static int __init ffa_init(void)1723{1724 int ret;1725 u32 buf_sz;1726 size_t rxtx_bufsz = SZ_4K;1727 1728 ret = ffa_transport_init(&invoke_ffa_fn);1729 if (ret)1730 return ret;1731 1732 drv_info = kzalloc(sizeof(*drv_info), GFP_KERNEL);1733 if (!drv_info)1734 return -ENOMEM;1735 1736 ret = ffa_version_check(&drv_info->version);1737 if (ret)1738 goto free_drv_info;1739 1740 if (ffa_id_get(&drv_info->vm_id)) {1741 pr_err("failed to obtain VM id for self\n");1742 ret = -ENODEV;1743 goto free_drv_info;1744 }1745 1746 ret = ffa_features(FFA_FN_NATIVE(RXTX_MAP), 0, &buf_sz, NULL);1747 if (!ret) {1748 if (RXTX_MAP_MIN_BUFSZ(buf_sz) == 1)1749 rxtx_bufsz = SZ_64K;1750 else if (RXTX_MAP_MIN_BUFSZ(buf_sz) == 2)1751 rxtx_bufsz = SZ_16K;1752 else1753 rxtx_bufsz = SZ_4K;1754 }1755 1756 drv_info->rxtx_bufsz = rxtx_bufsz;1757 drv_info->rx_buffer = alloc_pages_exact(rxtx_bufsz, GFP_KERNEL);1758 if (!drv_info->rx_buffer) {1759 ret = -ENOMEM;1760 goto free_pages;1761 }1762 1763 drv_info->tx_buffer = alloc_pages_exact(rxtx_bufsz, GFP_KERNEL);1764 if (!drv_info->tx_buffer) {1765 ret = -ENOMEM;1766 goto free_pages;1767 }1768 1769 ret = ffa_rxtx_map(virt_to_phys(drv_info->tx_buffer),1770 virt_to_phys(drv_info->rx_buffer),1771 rxtx_bufsz / FFA_PAGE_SIZE);1772 if (ret) {1773 pr_err("failed to register FFA RxTx buffers\n");1774 goto free_pages;1775 }1776 1777 mutex_init(&drv_info->rx_lock);1778 mutex_init(&drv_info->tx_lock);1779 1780 ffa_drvinfo_flags_init();1781 1782 ffa_notifications_setup();1783 1784 ret = ffa_setup_partitions();1785 if (ret) {1786 pr_err("failed to setup partitions\n");1787 goto cleanup_notifs;1788 }1789 1790 ret = ffa_sched_recv_cb_update(drv_info->vm_id, ffa_self_notif_handle,1791 drv_info, true);1792 if (ret)1793 pr_info("Failed to register driver sched callback %d\n", ret);1794 1795 return 0;1796 1797cleanup_notifs:1798 ffa_notifications_cleanup();1799free_pages:1800 if (drv_info->tx_buffer)1801 free_pages_exact(drv_info->tx_buffer, rxtx_bufsz);1802 free_pages_exact(drv_info->rx_buffer, rxtx_bufsz);1803free_drv_info:1804 kfree(drv_info);1805 return ret;1806}1807module_init(ffa_init);1808 1809static void __exit ffa_exit(void)1810{1811 ffa_notifications_cleanup();1812 ffa_partitions_cleanup();1813 ffa_rxtx_unmap(drv_info->vm_id);1814 free_pages_exact(drv_info->tx_buffer, drv_info->rxtx_bufsz);1815 free_pages_exact(drv_info->rx_buffer, drv_info->rxtx_bufsz);1816 kfree(drv_info);1817}1818module_exit(ffa_exit);1819 1820MODULE_ALIAS("arm-ffa");1821MODULE_AUTHOR("Sudeep Holla <sudeep.holla@arm.com>");1822MODULE_DESCRIPTION("Arm FF-A interface driver");1823MODULE_LICENSE("GPL v2");1824