drivers/firmware/arm_ffa/driver.c

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