On Thu, Feb 23, 2023 at 09:59:35AM +1100, Alistair Popple wrote: > > Jason Gunthorpe <jgg@xxxxxxxxxx> writes: > > > On Wed, Feb 22, 2023 at 10:38:25PM +1100, Alistair Popple wrote: > >> When a driver unpins a page we scan the pinners list and assign > >> ownership to the next driver pinning the page by updating memcg_data and > >> removing the vm_account from the list. > > > > I don't see how this works with just the data structure you outlined?? > > Every unique page needs its own list_head in the vm_account, it is > > doable just incredibly costly. > > The idea was every driver already needs to allocate a pages array to > pass to pin_user_pages(), and by necessity drivers have to keep a > reference to the contents of that in one form or another. So > conceptually the equivalent of: > > struct vm_account { > struct list_head possible_pinners; > struct mem_cgroup *memcg; > struct pages **pages; > [...] > }; > > Unpinnig involves finding a new owner by traversing the list of > page->memcg_data->possible_pinners and iterating over *pages[] to figure > out if that vm_account actually has this page pinned or not and could > own it. Oh, you are focusing on Tejun's DOS scenario. The DOS problem is to prevent a pin users in cgroup A from keeping memory charged to cgroup B that it isn't using any more. cgroup B doesn't need to be pinning the memory, it could just be normal VMAs and "isn't using anymore" means it has unmapped all the VMAs. Solving that problem means figuring out when every cgroup stops using the memory - pinning or not. That seems to be very costly. AFAIK this problem also already exists today as the memcg of a page doesn't change while it is pinned. So maybe we don't need to address it. Arguably the pins are not the problem. If we want to treat the pin like allocation then we simply charge the non-owning memcg's for the pin as though it was an allocation. Eg go over every page and if the owning memcg is not the current memcg then charge the current memcg for an allocation of the MAP_SHARED memory. Undoing this is trivial enoug. This doesn't fix the DOS problem but it does sort of harmonize the pin accounting with the memcg by multi-accounting every pin of a MAP_SHARED page. The other drawback is that this isn't the same thing as the current rlimit. The rlimit is largely restricting the creation of unmovable memory. Though, AFAICT memcg seems to bundle unmovable memory (eg GFP_KERNEL) along with movable user pages so it would be self-consistent. I'm unclear if this is OK for libvirt.. > Agree this is costly though. And I don't think all drivers keep the > array around so "iterating over *pages[]" may need to be a callback. I think searching lists of pages is not reasonable. Things like VFIO & KVM use cases effectively pin 90% of all system memory, that is potentially TB of page lists that might need linear searching! Jason