Re: Re: [PATCH v6 08/14] mm/damon: Add debugfs interface

From: SeongJae Park
Date: Tue Mar 10 2020 - 07:57:46 EST


On Tue, 10 Mar 2020 09:02:09 +0000 Jonathan Cameron <Jonathan.Cameron@xxxxxxxxxx> wrote:

> On Mon, 24 Feb 2020 13:30:41 +0100
> SeongJae Park <sjpark@xxxxxxxxxx> wrote:
>
> > From: SeongJae Park <sjpark@xxxxxxxxx>
> >
> > This commit adds a debugfs interface for DAMON.
[...]
> >
> > Signed-off-by: SeongJae Park <sjpark@xxxxxxxxx>
>
> Some of the code in here seems a bit fragile and convoluted.

Indeed, it needs many fixes.

>
> > ---
> > mm/damon.c | 377 ++++++++++++++++++++++++++++++++++++++++++++++++++++-
> > 1 file changed, 376 insertions(+), 1 deletion(-)
> >
> > diff --git a/mm/damon.c b/mm/damon.c
> > index b3e9b9da5720..facb1d7f121b 100644
> > --- a/mm/damon.c
> > +++ b/mm/damon.c
> > @@ -10,6 +10,7 @@
[...]
> >
> > +/*
> > + * debugfs functions
>
> Seems unnecessary when their naming makes this clear.

Agreed, will remove it.

>
[...]
> > +static ssize_t debugfs_pids_write(struct file *file,
> > + const char __user *buf, size_t count, loff_t *ppos)
> > +{
> > + struct damon_ctx *ctx = &damon_user_ctx;
> > + char *kbuf;
> > + unsigned long *targets;
> > + ssize_t nr_targets;
> > + ssize_t ret;
> > +
> > + kbuf = kmalloc_array(count, sizeof(char), GFP_KERNEL);
> > + if (!kbuf)
> > + return -ENOMEM;
> > +
> > + ret = simple_write_to_buffer(kbuf, 512, ppos, buf, count);
>
> Why only 512?

I might lost my mind at that time :'(
Good catch, it should be 'count'.

>
[...]
> > +
> > +static ssize_t debugfs_attrs_write(struct file *file,
> > + const char __user *buf, size_t count, loff_t *ppos)
> > +{
> > + struct damon_ctx *ctx = &damon_user_ctx;
> > + unsigned long s, a, r, minr, maxr;
> > + char *kbuf;
> > + ssize_t ret;
> > +
> > + kbuf = kmalloc_array(count, sizeof(char), GFP_KERNEL);
>
> malloc fine for array of characters. The checks on overflow etc cannot be
> relevant here.

You're right, will use 'kamlloc()' instead.

>
> > + if (!kbuf)
> > + return -ENOMEM;
> > +
> > + ret = simple_write_to_buffer(kbuf, count, ppos, buf, count);
> > + if (ret < 0)
> > + goto out;
> > +
> > + if (sscanf(kbuf, "%lu %lu %lu %lu %lu",
> > + &s, &a, &r, &minr, &maxr) != 5) {
> > + ret = -EINVAL;
> > + goto out;
> > + }
> > +
> > + spin_lock(&ctx->kdamond_lock);
> > + if (ctx->kdamond)
> > + goto monitor_running;
> > +
> > + damon_set_attrs(ctx, s, a, r, minr, maxr);
> > + spin_unlock(&ctx->kdamond_lock);
> > +
> > + goto out;
> > +
> > +monitor_running:
> > + spin_unlock(&ctx->kdamond_lock);
> > + pr_err("%s: kdamond is running. Turn it off first.\n", __func__);
> > + ret = -EINVAL;
>
> This complex exit path is a bad idea from maintainability point of view...
> Just put the pr_err and spin_unlock in the error path above.

Agreed, will do so.

>
> > +out:
> > + kfree(kbuf);
> > + return ret;
> > +}
> > +
> > +static const struct file_operations monitor_on_fops = {
> > + .owner = THIS_MODULE,
> > + .read = debugfs_monitor_on_read,
> > + .write = debugfs_monitor_on_write,
> > +};
> > +
> > +static const struct file_operations pids_fops = {
> > + .owner = THIS_MODULE,
> > + .read = debugfs_pids_read,
> > + .write = debugfs_pids_write,
> > +};
> > +
> > +static const struct file_operations record_fops = {
> > + .owner = THIS_MODULE,
> > + .read = debugfs_record_read,
> > + .write = debugfs_record_write,
> > +};
> > +
> > +static const struct file_operations attrs_fops = {
> > + .owner = THIS_MODULE,
> > + .read = debugfs_attrs_read,
> > + .write = debugfs_attrs_write,
> > +};
> > +
> > +static struct dentry *debugfs_root;
> > +
> > +static int __init debugfs_init(void)
>
> Prefix this function. Chances of sometime getting a header
> that includes debugfs_init feels rather too high!

That's right, I will rename it.

>
> > +{
> > + const char * const file_names[] = {"attrs", "record",
> > + "pids", "monitor_on"};
> > + const struct file_operations *fops[] = {&attrs_fops, &record_fops,
> > + &pids_fops, &monitor_on_fops};
> > + int i;
> > +
> > + debugfs_root = debugfs_create_dir("damon", NULL);
> > + if (!debugfs_root) {
> > + pr_err("failed to create the debugfs dir\n");
> > + return -ENOMEM;
> > + }
> > +
> > + for (i = 0; i < ARRAY_SIZE(file_names); i++) {
> > + if (!debugfs_create_file(file_names[i], 0600, debugfs_root,
> > + NULL, fops[i])) {
> > + pr_err("failed to create %s file\n", file_names[i]);
> > + return -ENOMEM;
> > + }
> > + }
> > +
> > + return 0;
> > +}
> > +
> > +static int __init damon_init_user_ctx(void)
> > +{
> > + int rc;
> > +
> > + struct damon_ctx *ctx = &damon_user_ctx;
> > +
> > + ktime_get_coarse_ts64(&ctx->last_aggregation);
> > + ctx->last_regions_update = ctx->last_aggregation;
> > +
> > + ctx->rbuf_offset = 0;
> > + rc = damon_set_recording(ctx, 1024 * 1024, "/damon.data");
> > + if (rc)
> > + return rc;
> > +
> > + ctx->kdamond = NULL;
> > + ctx->kdamond_stop = false;
> > + spin_lock_init(&ctx->kdamond_lock);
> > +
> > + prandom_seed_state(&ctx->rndseed, 42);
>
> :)

You got the answer ;)

>
> > + INIT_LIST_HEAD(&ctx->tasks_list);
> > +
> > + ctx->sample_cb = NULL;
> > + ctx->aggregate_cb = NULL;
>
> Should already be set to 0.

Oops, right!

>
> > +
> > + return 0;
> > +}
> > +
> > static int __init damon_init(void)
> > {
> > + int rc;
> > +
> > pr_info("init\n");
> >
> > - return 0;
> > + rc = damon_init_user_ctx();
> > + if (rc)
> > + return rc;
> > +
> > + return debugfs_init();
>
> In theory no code should ever be dependent on debugfs succeeding..
> There might be other daemon users so you should just eat the return
> code.

Right! Thank you for catching this!


Thanks,
SeongJae Park

>
>
> > }
> >
> > static void __exit damon_exit(void)
> > {
> > + damon_turn_kdamond(&damon_user_ctx, false);
> > + debugfs_remove_recursive(debugfs_root);
> > +
> > + kfree(damon_user_ctx.rbuf);
> > + kfree(damon_user_ctx.rfile_path);
> > +
> > pr_info("exit\n");
> > }
> >
>