162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright 2011 Analog Devices Inc.
462306a36Sopenharmony_ci */
562306a36Sopenharmony_ci
662306a36Sopenharmony_ci#include <linux/kernel.h>
762306a36Sopenharmony_ci#include <linux/module.h>
862306a36Sopenharmony_ci#include <linux/platform_device.h>
962306a36Sopenharmony_ci#include <linux/slab.h>
1062306a36Sopenharmony_ci#include <linux/list.h>
1162306a36Sopenharmony_ci#include <linux/irq_work.h>
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_ci#include <linux/iio/iio.h>
1462306a36Sopenharmony_ci#include <linux/iio/trigger.h>
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_cistruct iio_sysfs_trig {
1762306a36Sopenharmony_ci	struct iio_trigger *trig;
1862306a36Sopenharmony_ci	struct irq_work work;
1962306a36Sopenharmony_ci	int id;
2062306a36Sopenharmony_ci	struct list_head l;
2162306a36Sopenharmony_ci};
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_cistatic LIST_HEAD(iio_sysfs_trig_list);
2462306a36Sopenharmony_cistatic DEFINE_MUTEX(iio_sysfs_trig_list_mut);
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_cistatic int iio_sysfs_trigger_probe(int id);
2762306a36Sopenharmony_cistatic ssize_t iio_sysfs_trig_add(struct device *dev,
2862306a36Sopenharmony_ci				  struct device_attribute *attr,
2962306a36Sopenharmony_ci				  const char *buf,
3062306a36Sopenharmony_ci				  size_t len)
3162306a36Sopenharmony_ci{
3262306a36Sopenharmony_ci	int ret;
3362306a36Sopenharmony_ci	unsigned long input;
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_ci	ret = kstrtoul(buf, 10, &input);
3662306a36Sopenharmony_ci	if (ret)
3762306a36Sopenharmony_ci		return ret;
3862306a36Sopenharmony_ci	ret = iio_sysfs_trigger_probe(input);
3962306a36Sopenharmony_ci	if (ret)
4062306a36Sopenharmony_ci		return ret;
4162306a36Sopenharmony_ci	return len;
4262306a36Sopenharmony_ci}
4362306a36Sopenharmony_cistatic DEVICE_ATTR(add_trigger, S_IWUSR, NULL, &iio_sysfs_trig_add);
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_cistatic int iio_sysfs_trigger_remove(int id);
4662306a36Sopenharmony_cistatic ssize_t iio_sysfs_trig_remove(struct device *dev,
4762306a36Sopenharmony_ci				     struct device_attribute *attr,
4862306a36Sopenharmony_ci				     const char *buf,
4962306a36Sopenharmony_ci				     size_t len)
5062306a36Sopenharmony_ci{
5162306a36Sopenharmony_ci	int ret;
5262306a36Sopenharmony_ci	unsigned long input;
5362306a36Sopenharmony_ci
5462306a36Sopenharmony_ci	ret = kstrtoul(buf, 10, &input);
5562306a36Sopenharmony_ci	if (ret)
5662306a36Sopenharmony_ci		return ret;
5762306a36Sopenharmony_ci	ret = iio_sysfs_trigger_remove(input);
5862306a36Sopenharmony_ci	if (ret)
5962306a36Sopenharmony_ci		return ret;
6062306a36Sopenharmony_ci	return len;
6162306a36Sopenharmony_ci}
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_cistatic DEVICE_ATTR(remove_trigger, S_IWUSR, NULL, &iio_sysfs_trig_remove);
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_cistatic struct attribute *iio_sysfs_trig_attrs[] = {
6662306a36Sopenharmony_ci	&dev_attr_add_trigger.attr,
6762306a36Sopenharmony_ci	&dev_attr_remove_trigger.attr,
6862306a36Sopenharmony_ci	NULL,
6962306a36Sopenharmony_ci};
7062306a36Sopenharmony_ci
7162306a36Sopenharmony_cistatic const struct attribute_group iio_sysfs_trig_group = {
7262306a36Sopenharmony_ci	.attrs = iio_sysfs_trig_attrs,
7362306a36Sopenharmony_ci};
7462306a36Sopenharmony_ci
7562306a36Sopenharmony_cistatic const struct attribute_group *iio_sysfs_trig_groups[] = {
7662306a36Sopenharmony_ci	&iio_sysfs_trig_group,
7762306a36Sopenharmony_ci	NULL
7862306a36Sopenharmony_ci};
7962306a36Sopenharmony_ci
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci/* Nothing to actually do upon release */
8262306a36Sopenharmony_cistatic void iio_trigger_sysfs_release(struct device *dev)
8362306a36Sopenharmony_ci{
8462306a36Sopenharmony_ci}
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_cistatic struct device iio_sysfs_trig_dev = {
8762306a36Sopenharmony_ci	.bus = &iio_bus_type,
8862306a36Sopenharmony_ci	.groups = iio_sysfs_trig_groups,
8962306a36Sopenharmony_ci	.release = &iio_trigger_sysfs_release,
9062306a36Sopenharmony_ci};
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_cistatic void iio_sysfs_trigger_work(struct irq_work *work)
9362306a36Sopenharmony_ci{
9462306a36Sopenharmony_ci	struct iio_sysfs_trig *trig = container_of(work, struct iio_sysfs_trig,
9562306a36Sopenharmony_ci							work);
9662306a36Sopenharmony_ci
9762306a36Sopenharmony_ci	iio_trigger_poll(trig->trig);
9862306a36Sopenharmony_ci}
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_cistatic ssize_t iio_sysfs_trigger_poll(struct device *dev,
10162306a36Sopenharmony_ci		struct device_attribute *attr, const char *buf, size_t count)
10262306a36Sopenharmony_ci{
10362306a36Sopenharmony_ci	struct iio_trigger *trig = to_iio_trigger(dev);
10462306a36Sopenharmony_ci	struct iio_sysfs_trig *sysfs_trig = iio_trigger_get_drvdata(trig);
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ci	irq_work_queue(&sysfs_trig->work);
10762306a36Sopenharmony_ci
10862306a36Sopenharmony_ci	return count;
10962306a36Sopenharmony_ci}
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_cistatic DEVICE_ATTR(trigger_now, S_IWUSR, NULL, iio_sysfs_trigger_poll);
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_cistatic struct attribute *iio_sysfs_trigger_attrs[] = {
11462306a36Sopenharmony_ci	&dev_attr_trigger_now.attr,
11562306a36Sopenharmony_ci	NULL,
11662306a36Sopenharmony_ci};
11762306a36Sopenharmony_ci
11862306a36Sopenharmony_cistatic const struct attribute_group iio_sysfs_trigger_attr_group = {
11962306a36Sopenharmony_ci	.attrs = iio_sysfs_trigger_attrs,
12062306a36Sopenharmony_ci};
12162306a36Sopenharmony_ci
12262306a36Sopenharmony_cistatic const struct attribute_group *iio_sysfs_trigger_attr_groups[] = {
12362306a36Sopenharmony_ci	&iio_sysfs_trigger_attr_group,
12462306a36Sopenharmony_ci	NULL
12562306a36Sopenharmony_ci};
12662306a36Sopenharmony_ci
12762306a36Sopenharmony_cistatic int iio_sysfs_trigger_probe(int id)
12862306a36Sopenharmony_ci{
12962306a36Sopenharmony_ci	struct iio_sysfs_trig *t;
13062306a36Sopenharmony_ci	int ret;
13162306a36Sopenharmony_ci	bool foundit = false;
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_ci	mutex_lock(&iio_sysfs_trig_list_mut);
13462306a36Sopenharmony_ci	list_for_each_entry(t, &iio_sysfs_trig_list, l)
13562306a36Sopenharmony_ci		if (id == t->id) {
13662306a36Sopenharmony_ci			foundit = true;
13762306a36Sopenharmony_ci			break;
13862306a36Sopenharmony_ci		}
13962306a36Sopenharmony_ci	if (foundit) {
14062306a36Sopenharmony_ci		ret = -EINVAL;
14162306a36Sopenharmony_ci		goto err_unlock;
14262306a36Sopenharmony_ci	}
14362306a36Sopenharmony_ci	t = kmalloc(sizeof(*t), GFP_KERNEL);
14462306a36Sopenharmony_ci	if (t == NULL) {
14562306a36Sopenharmony_ci		ret = -ENOMEM;
14662306a36Sopenharmony_ci		goto err_unlock;
14762306a36Sopenharmony_ci	}
14862306a36Sopenharmony_ci	t->id = id;
14962306a36Sopenharmony_ci	t->trig = iio_trigger_alloc(&iio_sysfs_trig_dev, "sysfstrig%d", id);
15062306a36Sopenharmony_ci	if (!t->trig) {
15162306a36Sopenharmony_ci		ret = -ENOMEM;
15262306a36Sopenharmony_ci		goto err_free_sys_trig;
15362306a36Sopenharmony_ci	}
15462306a36Sopenharmony_ci
15562306a36Sopenharmony_ci	t->trig->dev.groups = iio_sysfs_trigger_attr_groups;
15662306a36Sopenharmony_ci	iio_trigger_set_drvdata(t->trig, t);
15762306a36Sopenharmony_ci
15862306a36Sopenharmony_ci	t->work = IRQ_WORK_INIT_HARD(iio_sysfs_trigger_work);
15962306a36Sopenharmony_ci
16062306a36Sopenharmony_ci	ret = iio_trigger_register(t->trig);
16162306a36Sopenharmony_ci	if (ret)
16262306a36Sopenharmony_ci		goto err_free_trig;
16362306a36Sopenharmony_ci	list_add(&t->l, &iio_sysfs_trig_list);
16462306a36Sopenharmony_ci	__module_get(THIS_MODULE);
16562306a36Sopenharmony_ci	mutex_unlock(&iio_sysfs_trig_list_mut);
16662306a36Sopenharmony_ci	return 0;
16762306a36Sopenharmony_ci
16862306a36Sopenharmony_cierr_free_trig:
16962306a36Sopenharmony_ci	iio_trigger_free(t->trig);
17062306a36Sopenharmony_cierr_free_sys_trig:
17162306a36Sopenharmony_ci	kfree(t);
17262306a36Sopenharmony_cierr_unlock:
17362306a36Sopenharmony_ci	mutex_unlock(&iio_sysfs_trig_list_mut);
17462306a36Sopenharmony_ci	return ret;
17562306a36Sopenharmony_ci}
17662306a36Sopenharmony_ci
17762306a36Sopenharmony_cistatic int iio_sysfs_trigger_remove(int id)
17862306a36Sopenharmony_ci{
17962306a36Sopenharmony_ci	struct iio_sysfs_trig *t = NULL, *iter;
18062306a36Sopenharmony_ci
18162306a36Sopenharmony_ci	mutex_lock(&iio_sysfs_trig_list_mut);
18262306a36Sopenharmony_ci	list_for_each_entry(iter, &iio_sysfs_trig_list, l)
18362306a36Sopenharmony_ci		if (id == iter->id) {
18462306a36Sopenharmony_ci			t = iter;
18562306a36Sopenharmony_ci			break;
18662306a36Sopenharmony_ci		}
18762306a36Sopenharmony_ci	if (!t) {
18862306a36Sopenharmony_ci		mutex_unlock(&iio_sysfs_trig_list_mut);
18962306a36Sopenharmony_ci		return -EINVAL;
19062306a36Sopenharmony_ci	}
19162306a36Sopenharmony_ci
19262306a36Sopenharmony_ci	iio_trigger_unregister(t->trig);
19362306a36Sopenharmony_ci	irq_work_sync(&t->work);
19462306a36Sopenharmony_ci	iio_trigger_free(t->trig);
19562306a36Sopenharmony_ci
19662306a36Sopenharmony_ci	list_del(&t->l);
19762306a36Sopenharmony_ci	kfree(t);
19862306a36Sopenharmony_ci	module_put(THIS_MODULE);
19962306a36Sopenharmony_ci	mutex_unlock(&iio_sysfs_trig_list_mut);
20062306a36Sopenharmony_ci	return 0;
20162306a36Sopenharmony_ci}
20262306a36Sopenharmony_ci
20362306a36Sopenharmony_ci
20462306a36Sopenharmony_cistatic int __init iio_sysfs_trig_init(void)
20562306a36Sopenharmony_ci{
20662306a36Sopenharmony_ci	int ret;
20762306a36Sopenharmony_ci	device_initialize(&iio_sysfs_trig_dev);
20862306a36Sopenharmony_ci	dev_set_name(&iio_sysfs_trig_dev, "iio_sysfs_trigger");
20962306a36Sopenharmony_ci	ret = device_add(&iio_sysfs_trig_dev);
21062306a36Sopenharmony_ci	if (ret)
21162306a36Sopenharmony_ci		put_device(&iio_sysfs_trig_dev);
21262306a36Sopenharmony_ci	return ret;
21362306a36Sopenharmony_ci}
21462306a36Sopenharmony_cimodule_init(iio_sysfs_trig_init);
21562306a36Sopenharmony_ci
21662306a36Sopenharmony_cistatic void __exit iio_sysfs_trig_exit(void)
21762306a36Sopenharmony_ci{
21862306a36Sopenharmony_ci	device_unregister(&iio_sysfs_trig_dev);
21962306a36Sopenharmony_ci}
22062306a36Sopenharmony_cimodule_exit(iio_sysfs_trig_exit);
22162306a36Sopenharmony_ci
22262306a36Sopenharmony_ciMODULE_AUTHOR("Michael Hennerich <michael.hennerich@analog.com>");
22362306a36Sopenharmony_ciMODULE_DESCRIPTION("Sysfs based trigger for the iio subsystem");
22462306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
22562306a36Sopenharmony_ciMODULE_ALIAS("platform:iio-trig-sysfs");
226