Unverified Commit 7ddc52ba authored by openeuler-ci-bot's avatar openeuler-ci-bot Committed by Gitee
Browse files

!15768 gpio: aggregator: protect driver attr handlers against module unload 

Merge Pull Request from: @ci-robot 
 
PR sync from: Chen Zhongjin <chenzhongjin@huawei.com>
https://mailweb.openeuler.org/archives/list/kernel@openeuler.org/message/M7GPKILMPLWQTINA5FOQSAKL3YMJEH5E/ 
 
https://gitee.com/openeuler/kernel/issues/IBY441 
 
Link:https://gitee.com/openeuler/kernel/pulls/15768

 

Reviewed-by: default avatarYe Weihua <yeweihua4@huawei.com>
Signed-off-by: default avatarZhang Peng <zhangpeng362@huawei.com>
parents 77e7b353 8d1e33b8

drivers/gpio/gpio-aggregator.c

+17 −3
Original line number Diff line number Diff line
@@ -121,10 +121,15 @@ static ssize_t new_device_store(struct device_driver *driver, const char *buf, 
	struct platform_device *pdev;

	int res, id;



	if (!try_module_get(THIS_MODULE))

		return -ENOENT;



	/* kernfs guarantees string termination, so count + 1 is safe */

	aggr = kzalloc(sizeof(*aggr) + count + 1, GFP_KERNEL);

	if (!aggr)

		return -ENOMEM;

	if (!aggr) {

		res = -ENOMEM;

		goto put_module;

	}



	memcpy(aggr->args, buf, count + 1);
@@ -163,6 +168,7 @@ static ssize_t new_device_store(struct device_driver *driver, const char *buf, 
	}



	aggr->pdev = pdev;

	module_put(THIS_MODULE);

	return count;



remove_table:
@@ -177,6 +183,8 @@ static ssize_t new_device_store(struct device_driver *driver, const char *buf, 
	kfree(aggr->lookups);

free_ga:

	kfree(aggr);

put_module:

	module_put(THIS_MODULE);

	return res;

}
@@ -205,13 +213,19 @@ static ssize_t delete_device_store(struct device_driver *driver, 
	if (error)

		return error;



	if (!try_module_get(THIS_MODULE))

		return -ENOENT;



	mutex_lock(&gpio_aggregator_lock);

	aggr = idr_remove(&gpio_aggregator_idr, id);

	mutex_unlock(&gpio_aggregator_lock);

	if (!aggr)

	if (!aggr) {

		module_put(THIS_MODULE);

		return -ENOENT;

	}



	gpio_aggregator_free(aggr);

	module_put(THIS_MODULE);

	return count;

}

static DRIVER_ATTR_WO(delete_device);