Skip to content

Commit

Permalink
Merge git://git.kvack.org/~bcrl/aio-next
Browse files Browse the repository at this point in the history
Pull aio fix from Ben LaHaise:
 "Improve aio-nr counting on large SMP systems.

  It has been in linux-next for quite some time"

* git://git.kvack.org/~bcrl/aio-next:
  fs: aio: fix the increment of aio-nr and counting against aio-max-nr
  • Loading branch information
torvalds committed Sep 7, 2017
2 parents ae8ac6b + 2a8a986 commit 2500e28
Showing 1 changed file with 12 additions and 7 deletions.
19 changes: 12 additions & 7 deletions fs/aio.c
Original file line number Diff line number Diff line change
Expand Up @@ -441,10 +441,9 @@ static const struct address_space_operations aio_ctx_aops = {
#endif
};

static int aio_setup_ring(struct kioctx *ctx)
static int aio_setup_ring(struct kioctx *ctx, unsigned int nr_events)
{
struct aio_ring *ring;
unsigned nr_events = ctx->max_reqs;
struct mm_struct *mm = current->mm;
unsigned long size, unused;
int nr_pages;
Expand Down Expand Up @@ -706,6 +705,12 @@ static struct kioctx *ioctx_alloc(unsigned nr_events)
struct kioctx *ctx;
int err = -ENOMEM;

/*
* Store the original nr_events -- what userspace passed to io_setup(),
* for counting against the global limit -- before it changes.
*/
unsigned int max_reqs = nr_events;

/*
* We keep track of the number of available ringbuffer slots, to prevent
* overflow (reqs_available), and we also use percpu counters for this.
Expand All @@ -724,14 +729,14 @@ static struct kioctx *ioctx_alloc(unsigned nr_events)
return ERR_PTR(-EINVAL);
}

if (!nr_events || (unsigned long)nr_events > (aio_max_nr * 2UL))
if (!nr_events || (unsigned long)max_reqs > aio_max_nr)
return ERR_PTR(-EAGAIN);

ctx = kmem_cache_zalloc(kioctx_cachep, GFP_KERNEL);
if (!ctx)
return ERR_PTR(-ENOMEM);

ctx->max_reqs = nr_events;
ctx->max_reqs = max_reqs;

spin_lock_init(&ctx->ctx_lock);
spin_lock_init(&ctx->completion_lock);
Expand All @@ -753,7 +758,7 @@ static struct kioctx *ioctx_alloc(unsigned nr_events)
if (!ctx->cpu)
goto err;

err = aio_setup_ring(ctx);
err = aio_setup_ring(ctx, nr_events);
if (err < 0)
goto err;

Expand All @@ -764,8 +769,8 @@ static struct kioctx *ioctx_alloc(unsigned nr_events)

/* limit the number of system wide aios */
spin_lock(&aio_nr_lock);
if (aio_nr + nr_events > (aio_max_nr * 2UL) ||
aio_nr + nr_events < aio_nr) {
if (aio_nr + ctx->max_reqs > aio_max_nr ||
aio_nr + ctx->max_reqs < aio_nr) {
spin_unlock(&aio_nr_lock);
err = -EAGAIN;
goto err_ctx;
Expand Down

0 comments on commit 2500e28

Please sign in to comment.