On 10/14/20 8:05 AM, Cristian Marussi wrote: > Extend common protocol registration routines and provide some new generic > protocols' init/deinit helpers that tracks protocols' users and automatically > perform the proper initialization/de-initialization on demand. > > Convert all protocols to use new registration schema while modifying only Base > protocol to use also the new initialization helpers. > > All other standard protocols' initialization is still umodified and bound to > SCMI devices probing. > > Signed-off-by: Cristian Marussi <cristian.maru...@arm.com> > ---
[snip] > +static struct scmi_protocol scmi_base = { > + .id = SCMI_PROTOCOL_BASE, > + .init = &scmi_base_protocol_init, > + .ops = NULL, > +}; This could be const I believe. > + > +DEFINE_SCMI_PROTOCOL_REGISTER_UNREGISTER(base, scmi_base) > diff --git a/drivers/firmware/arm_scmi/bus.c b/drivers/firmware/arm_scmi/bus.c > index 1377ec76a45d..afa2e4818a2b 100644 > --- a/drivers/firmware/arm_scmi/bus.c > +++ b/drivers/firmware/arm_scmi/bus.c > @@ -16,7 +16,7 @@ > #include "common.h" > > static DEFINE_IDA(scmi_bus_id); > -static DEFINE_IDR(scmi_protocols); > +static DEFINE_IDR(scmi_available_protocols); > static DEFINE_SPINLOCK(protocol_lock); > > static const struct scmi_device_id * > @@ -51,13 +51,29 @@ static int scmi_dev_match(struct device *dev, struct > device_driver *drv) > return 0; > } > > +const struct scmi_protocol *scmi_get_protocol(int protocol_id) > +{ > + const struct scmi_protocol *proto; > + > + proto = idr_find(&scmi_available_protocols, protocol_id); > + if (!proto) { > + pr_warn("SCMI Protocol 0x%x not found!\n", protocol_id); > + return NULL; > + } > + > + pr_debug("GOT SCMI Protocol 0x%x\n", protocol_id); > + > + return proto; > +} > + > static int scmi_protocol_init(int protocol_id, struct scmi_handle *handle) > { > - scmi_prot_init_fn_t fn = idr_find(&scmi_protocols, protocol_id); > + const struct scmi_protocol *proto; > > - if (unlikely(!fn)) > + proto = idr_find(&scmi_available_protocols, protocol_id); > + if (!proto) > return -EINVAL; > - return fn(handle); > + return proto->init(handle); > } > > static int scmi_protocol_dummy_init(struct scmi_handle *handle) > @@ -84,7 +100,7 @@ static int scmi_dev_probe(struct device *dev) > return ret; > > /* Skip protocol initialisation for additional devices */ > - idr_replace(&scmi_protocols, &scmi_protocol_dummy_init, > + idr_replace(&scmi_available_protocols, &scmi_protocol_dummy_init, > scmi_dev->protocol_id); > > return scmi_drv->probe(scmi_dev); > @@ -194,26 +210,45 @@ void scmi_set_handle(struct scmi_device *scmi_dev) > scmi_dev->handle = scmi_handle_get(&scmi_dev->dev); > } > > -int scmi_protocol_register(int protocol_id, scmi_prot_init_fn_t fn) > +int scmi_protocol_register(struct scmi_protocol *proto) And this could probably take a const struct scmi_protocol here too as you do not appear to be modifying proto, idr_alloc() may complain. [snip] > > -DEFINE_SCMI_PROTOCOL_REGISTER_UNREGISTER(SCMI_PROTOCOL_CLOCK, clock) > +static struct scmi_protocol scmi_clock = { static const here too and throughout your submission. [snip] > + struct scmi_protocol_instance *protocols[SCMI_MAX_PROTO]; Humm that would be 2048 bytes on a 64-bit platform and 1024 bytes on a 32-bit platform, this is not so bad, but it is a bit wasteful given that 6-7 standard protocols are typically found, and most often no proprietary protocols are registered. Not necessarily to be addressed right now. > + /* Ensure mutual exclusive access to protocols instance array */ > + struct mutex protocols_mtx; > u8 *protocols_imp; > struct list_head node; > int users; > @@ -519,6 +542,132 @@ int scmi_version_get(const struct scmi_handle *handle, > u8 protocol, > return ret; > } > > +/** > + * scmi_get_protocol_instance - Protocol initialization helper. > + * @handle: A reference to the SCMI platform instance. > + * @protocol_id: The protocol being requested. > + * > + * In case the required protocol has never been requested before for this > + * instance, allocate and initialize all the needed structures while handling > + * resource allocation with a dedicated per-protocol devres subgroup. > + * > + * Return: A reference to an initialized protocol instance or error on > failure. > + */ > +static struct scmi_protocol_instance * __must_check > +scmi_get_protocol_instance(struct scmi_handle *handle, u8 protocol_id) > +{ > + int ret = -ENOMEM; > + void *gid; > + struct scmi_protocol_instance *pi; > + struct scmi_info *info = handle_to_scmi_info(handle); > + > + mutex_lock(&info->protocols_mtx); > + /* Ensure protocols has been updated */ > + smp_rmb(); > + pi = info->protocols[protocol_id]; > + > + if (!pi) { > + const struct scmi_protocol *proto; > + > + /* Fail if protocol not registered on bus */ > + proto = scmi_get_protocol(protocol_id); > + if (!proto) { > + ret = -EINVAL; You could return -ENODEV here and propagate that error code for scmi_probe() to use as -is. > + goto out; > + } > + > + /* Protocol specific devres group */ > + gid = devres_open_group(handle->dev, NULL, GFP_KERNEL); > + if (!gid) > + goto out; > + > + pi = devm_kzalloc(handle->dev, sizeof(*pi), GFP_KERNEL); > + if (!pi) > + goto clean; > + > + pi->gid = gid; > + pi->proto = proto; > + refcount_set(&pi->users, 1); > + /* proto->init is assured NON NULL by scmi_protocol_register */ > + ret = pi->proto->init(handle); > + if (ret) > + goto clean; > + > + info->protocols[protocol_id] = pi; > + /* Ensure initialized protocol is visible */ > + smp_wmb(); > + > + devres_close_group(handle->dev, pi->gid); > + dev_dbg(handle->dev, "Initialized protocol: 0x%X\n", > + protocol_id); > + } else { > + refcount_inc(&pi->users); > + } You may be able to re-arrange the indentation and do: if (pi) { refcount_inc(&pi->users); mutex_unlock(&info->protocols_mtx); return pi; } to reduce the indentation level, and this would also have the fast path tested first. > + mutex_unlock(&info->protocols_mtx); > + > + return pi; > + > +clean: > + devres_release_group(handle->dev, gid); > +out: > + mutex_unlock(&info->protocols_mtx); > + return ERR_PTR(ret); > +} > + > +/** > + * scmi_acquire_protocol - Protocol acquire > + * @handle: A reference to the SCMI platform instance. > + * @protocol_id: The protocol being requested. > + * > + * Register a new user for the requested protocol on the specified SCMI > + * platform instance, possibly triggering its initialization on first user. > + * > + * Return: 0 if protocol was acquired successfully. > + */ > +int scmi_acquire_protocol(struct scmi_handle *handle, u8 protocol_id) > +{ > + return IS_ERR(scmi_get_protocol_instance(handle, protocol_id)); > +} > + > +/** > + * scmi_release_protocol - Protocol de-initialization helper. > + * @handle: A reference to the SCMI platform instance. > + * @protocol_id: The protocol being requested. > + * > + * Remove one user for the specified protocol and triggers de-initialization > + * and resources de-allocation once the last user has gone. > + */ > +void scmi_release_protocol(struct scmi_handle *handle, u8 protocol_id) > +{ > + struct scmi_info *info = handle_to_scmi_info(handle); > + struct scmi_protocol_instance *pi; > + > + mutex_lock(&info->protocols_mtx); > + /* Ensure protocols has been updated */ > + smp_rmb(); > + pi = info->protocols[protocol_id]; > + if (WARN_ON(!pi)) { > + mutex_unlock(&info->protocols_mtx); > + return; Maybe define an "out" label just to avoid the repetition? > + } > + > + if (refcount_dec_and_test(&pi->users)) { > + void *gid = pi->gid; > + > + if (pi->proto->deinit) > + pi->proto->deinit(handle); > + > + info->protocols[protocol_id] = NULL; > + /* Ensure deinitialized protocol is visible */ > + smp_wmb(); > + > + devres_release_group(handle->dev, gid); > + dev_dbg(handle->dev, "De-Initialized protocol: 0x%X\n", > + protocol_id); > + } > + mutex_unlock(&info->protocols_mtx); > +} > + > void scmi_setup_protocol_implemented(const struct scmi_handle *handle, > u8 *prot_imp) > { > @@ -785,6 +934,7 @@ static int scmi_probe(struct platform_device *pdev) > info->dev = dev; > info->desc = desc; > INIT_LIST_HEAD(&info->node); > + mutex_init(&info->protocols_mtx); > > platform_set_drvdata(pdev, info); > idr_init(&info->tx_idr); > @@ -805,10 +955,14 @@ static int scmi_probe(struct platform_device *pdev) > if (scmi_notification_init(handle)) > dev_err(dev, "SCMI Notifications NOT available.\n"); > > - ret = scmi_base_protocol_init(handle); > - if (ret) { > - dev_err(dev, "unable to communicate with SCMI(%d)\n", ret); > - return ret; > + /* > + * Trigger SCMI Base protocol initialization. > + * It's mandatory and won't be ever released/deinit until the > + * SCMI stack is shutdown/unloaded as a whole. > + */ > + if (scmi_acquire_protocol(handle, SCMI_PROTOCOL_BASE)) { > + dev_err(dev, "unable to communicate with SCMI\n"); > + return -ENODEV; and you could do: ret = scmi_acquire_protocol(.., ...); if (ret) { dev_err(dev, "unable to communicate with SCMI\n"); return ret; } Everything else looked good to me, thanks! -- Florian