Quelle  bus.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2021 ARM Ltd.
 */

#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

#include <linux/arm_ffa.h>
#include <linux/device.h>
#include <linux/fs.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/types.h>

#include "common.h"

#define FFA_UEVENT_MODALIAS_FMT "arm_ffa:%04x:%pUb"

static DEFINE_IDA(ffa_bus_id);

static int ffa_device_match(struct device *dev, const struct device_driver *drv)
{
 const struct ffa_device_id *id_table;
 struct ffa_device *ffa_dev;

 id_table = to_ffa_driver(drv)->id_table;
 ffa_dev = to_ffa_dev(dev);

 while (!uuid_is_null(&id_table->uuid)) {
  /*
 * FF-A v1.0 doesn't provide discovery of UUIDs, just the
 * partition IDs, so match it unconditionally here and handle
 * it via the installed bus notifier during driver binding.
 */
  if (uuid_is_null(&ffa_dev->uuid))
   return 1;

  if (uuid_equal(&ffa_dev->uuid, &id_table->uuid))
   return 1;
  id_table++;
 }

 return 0;
}

static int ffa_device_probe(struct device *dev)
{
 struct ffa_driver *ffa_drv = to_ffa_driver(dev->driver);
 struct ffa_device *ffa_dev = to_ffa_dev(dev);

 /* UUID can be still NULL with FF-A v1.0, so just skip probing them */
 if (uuid_is_null(&ffa_dev->uuid))
  return -ENODEV;

 return ffa_drv->probe(ffa_dev);
}

static void ffa_device_remove(struct device *dev)
{
 struct ffa_driver *ffa_drv = to_ffa_driver(dev->driver);

 if (ffa_drv->remove)
  ffa_drv->remove(to_ffa_dev(dev));
}

static int ffa_device_uevent(const struct device *dev, struct kobj_uevent_env *env)
{
 const struct ffa_device *ffa_dev = to_ffa_dev(dev);

 return add_uevent_var(env, "MODALIAS=" FFA_UEVENT_MODALIAS_FMT,
         ffa_dev->vm_id, &ffa_dev->uuid);
}

static ssize_t modalias_show(struct device *dev,
        struct device_attribute *attr, char *buf)
{
 struct ffa_device *ffa_dev = to_ffa_dev(dev);

 return sysfs_emit(buf, FFA_UEVENT_MODALIAS_FMT, ffa_dev->vm_id,
     &ffa_dev->uuid);
}
static DEVICE_ATTR_RO(modalias);

static ssize_t partition_id_show(struct device *dev,
     struct device_attribute *attr, char *buf)
{
 struct ffa_device *ffa_dev = to_ffa_dev(dev);

 return sprintf(buf, "0x%04x\n", ffa_dev->vm_id);
}
static DEVICE_ATTR_RO(partition_id);

static ssize_t uuid_show(struct device *dev, struct device_attribute *attr,
    char *buf)
{
 struct ffa_device *ffa_dev = to_ffa_dev(dev);

 return sprintf(buf, "%pUb\n", &ffa_dev->uuid);
}
static DEVICE_ATTR_RO(uuid);

static struct attribute *ffa_device_attributes_attrs[] = {
 &dev_attr_partition_id.attr,
 &dev_attr_uuid.attr,
 &dev_attr_modalias.attr,
 NULL,
};
ATTRIBUTE_GROUPS(ffa_device_attributes);

const struct bus_type ffa_bus_type = {
 .name  = "arm_ffa",
 .match  = ffa_device_match,
 .probe  = ffa_device_probe,
 .remove  = ffa_device_remove,
 .uevent  = ffa_device_uevent,
 .dev_groups = ffa_device_attributes_groups,
};
EXPORT_SYMBOL_GPL(ffa_bus_type);

int ffa_driver_register(struct ffa_driver *driver, struct module *owner,
   const char *mod_name)
{
 int ret;

 if (!driver->probe)
  return -EINVAL;

 driver->driver.bus = &ffa_bus_type;
 driver->driver.name = driver->name;
 driver->driver.owner = owner;
 driver->driver.mod_name = mod_name;

 ret = driver_register(&driver->driver);
 if (!ret)
  pr_debug("registered new ffa driver %s\n", driver->name);

 return ret;
}
EXPORT_SYMBOL_GPL(ffa_driver_register);

void ffa_driver_unregister(struct ffa_driver *driver)
{
 driver_unregister(&driver->driver);
}
EXPORT_SYMBOL_GPL(ffa_driver_unregister);

static void ffa_release_device(struct device *dev)
{
 struct ffa_device *ffa_dev = to_ffa_dev(dev);

 ida_free(&ffa_bus_id, ffa_dev->id);
 kfree(ffa_dev);
}

static int __ffa_devices_unregister(struct device *dev, void *data)
{
 device_unregister(dev);

 return 0;
}

void ffa_devices_unregister(void)
{
 bus_for_each_dev(&ffa_bus_type, NULL, NULL,
    __ffa_devices_unregister);
}
EXPORT_SYMBOL_GPL(ffa_devices_unregister);

bool ffa_device_is_valid(struct ffa_device *ffa_dev)
{
 bool valid = false;
 struct device *dev = NULL;
 struct ffa_device *tmp_dev;

 do {
  dev = bus_find_next_device(&ffa_bus_type, dev);
  tmp_dev = to_ffa_dev(dev);
  if (tmp_dev == ffa_dev) {
   valid = true;
   break;
  }
  put_device(dev);
 } while (dev);

 put_device(dev);

 return valid;
}

struct ffa_device *
ffa_device_register(const struct ffa_partition_info *part_info,
      const struct ffa_ops *ops)
{
 int id, ret;
 struct device *dev;
 struct ffa_device *ffa_dev;

 if (!part_info)
  return NULL;

 id = ida_alloc_min(&ffa_bus_id, 1, GFP_KERNEL);
 if (id < 0)
  return NULL;

 ffa_dev = kzalloc(sizeof(*ffa_dev), GFP_KERNEL);
 if (!ffa_dev) {
  ida_free(&ffa_bus_id, id);
  return NULL;
 }

 dev = &ffa_dev->dev;
 dev->bus = &ffa_bus_type;
 dev->release = ffa_release_device;
 dev->dma_mask = &dev->coherent_dma_mask;
 dev_set_name(&ffa_dev->dev, "arm-ffa-%d", id);

 ffa_dev->id = id;
 ffa_dev->vm_id = part_info->id;
 ffa_dev->properties = part_info->properties;
 ffa_dev->ops = ops;
 uuid_copy(&ffa_dev->uuid, &part_info->uuid);

 ret = device_register(&ffa_dev->dev);
 if (ret) {
  dev_err(dev, "unable to register device %s err=%d\n",
   dev_name(dev), ret);
  put_device(dev);
  return NULL;
 }

 return ffa_dev;
}
EXPORT_SYMBOL_GPL(ffa_device_register);

void ffa_device_unregister(struct ffa_device *ffa_dev)
{
 if (!ffa_dev)
  return;

 device_unregister(&ffa_dev->dev);
}
EXPORT_SYMBOL_GPL(ffa_device_unregister);

static int __init arm_ffa_bus_init(void)
{
 return bus_register(&ffa_bus_type);
}
subsys_initcall(arm_ffa_bus_init);

static void __exit arm_ffa_bus_exit(void)
{
 ffa_devices_unregister();
 bus_unregister(&ffa_bus_type);
 ida_destroy(&ffa_bus_id);
}
module_exit(arm_ffa_bus_exit);

MODULE_ALIAS("ffa-core");
MODULE_AUTHOR("Sudeep Holla ");
MODULE_DESCRIPTION("ARM FF-A bus");
MODULE_LICENSE("GPL");