On Wed, Sep 25, 2024 at 12:33 AM Barry Song <baohua@xxxxxxxxxx> wrote: > > On Wed, Sep 25, 2024 at 3:23 PM Huang, Ying <ying.huang@xxxxxxxxx> wrote: > > > > Nhat Pham <nphamcs@xxxxxxxxx> writes: > > > > [snip] > > > > > > > > My understanding now is that there are two for loops. One for loop > > > that checks the entry's states, and one for loop that does the actual > > > incrementing work (or state modification). > > > > > > We can check in the first for loop, if it is safe to proceed: > > > > > > if (!count && !has_cache) { > > > err = -ENOENT; > > > } else if (usage == SWAP_HAS_CACHE) { > > > if (has_cache) > > > err = -EEXIST; > > > } else if ((count & ~COUNT_CONTINUED) > SWAP_MAP_MAX) { > > > err = -EINVAL; > > > } else if (usage == 1 && nr > 1 && (count & ~COUNT_CONTINUED) >= > > > SWAP_MAP_MAX)) { > > > /* the batched variants currently do not support rollback */ > > > err = -ENOMEM; > > > } > > > > > > At this point, IIUC, we have not done any incrementing, so no rollback > > > needed? :) > > > > I think that it's better to add a VM_WARN_ONCE() here. If someone > > enabled 'nr > 1' for __swap_duplicate(), the issue will be more > > explicit. > > ying, i guess you missed this is the exact case Nhat is enabling > 'nr > 1' for __swap_duplicate(). and this warning is functioning. > and he is trying to support the nr>1 case. > > https://lore.kernel.org/linux-mm/20240923231142.4155415-2-nphamcs@xxxxxxxxx/ I'm only supporting the case nr > 1, when there is no need to add swap continuations :) That's the only current use case right now (shmem) :) 1. Keep the non-batched variant: int swap_duplicate(swp_entry_t entry) { int err = 0; while (!err && __swap_duplicate(entry, 1, 1) == -ENOMEM) err = add_swap_count_continuation(entry, GFP_ATOMIC); return err; } 2. Implement the batched variant: int swap_duplicate_nr(swp_entry_t entry, int nr) { swp_entry_t cur_entry; int i, err; if (nr == 1) return swap_duplicate(entry); err = __swap_duplicate(entry, 1, nr); if (err == -ENOMEM) { /* fallback to non-batched version */ for (i = 0; i < nr; i++) { cur_entry = (swp_entry_t){entry.val + i}; if (swap_duplicate(cur_entry)) { /* rollback */ while (--i >= 0) { cur_entry = (swp_entry_t){entry.val + i}; swap_free(cur_entry); } } } } return err; } How does this look? My concern is that there is not really a use for the fallback logic. Basically dead code. I can keep it in if you guys have a use for it soon, but otherwise I lean towards just adding a WARN etc. there, or return -ENOMEM, and WARN at shmem's callsite (because it cannot get -ENOMEM).