Re: [PATCH v2 3/7] mm/vmalloc.c: allow vread() to read out vm_map_ram areas

[Lorenzo Stoakes]

On Wed, Jan 04, 2023 at 04:01:36PM +0800, Baoquan He wrote:
> On 12/17/22 at 12:06pm, Lorenzo Stoakes wrote:
> > On Sat, Dec 17, 2022 at 09:54:31AM +0800, Baoquan He wrote:
> > > Currently, vread can read out vmalloc areas which is associated with
> > > a vm_struct. While this doesn't work for areas created by vm_map_ram()
> > > interface because it doesn't have an associated vm_struct. Then in vread(),
> > > these areas will be skipped.
> > >
> > > Here, add a new function vb_vread() to read out areas managed by
> > > vmap_block specifically. Then recognize vm_map_ram areas via vmap->flags
> > > and handle them respectively.
> > >
> > > Signed-off-by: Baoquan He <bhe@xxxxxxxxxx>
> > > ---
> > >  mm/vmalloc.c | 66 ++++++++++++++++++++++++++++++++++++++++++++++------
> > >  1 file changed, 59 insertions(+), 7 deletions(-)
> > >
> > > diff --git a/mm/vmalloc.c b/mm/vmalloc.c
> > > index 190f29bbaaa7..6612914459cf 100644
> > > --- a/mm/vmalloc.c
> > > +++ b/mm/vmalloc.c
> > > @@ -3515,6 +3515,51 @@ static int aligned_vread(char *buf, char *addr, unsigned long count)
> > >  	return copied;
> > >  }
> > >
> > > +static void vb_vread(char *buf, char *addr, int count)
> > > +{
> > > +	char *start;
> > > +	struct vmap_block *vb;
> > > +	unsigned long offset;
> > > +	unsigned int rs, re, n;
> > > +
> > > +	vb = xa_load(&vmap_blocks, addr_to_vb_idx((unsigned long)addr));
> > > +
> > > +	spin_lock(&vb->lock);
> > > +	if (bitmap_empty(vb->used_map, VMAP_BBMAP_BITS)) {
> > > +		spin_unlock(&vb->lock);
> > > +		memset(buf, 0, count);
> > > +		return;
> > > +	}
> > > +	for_each_set_bitrange(rs, re, vb->used_map, VMAP_BBMAP_BITS) {
> > > +		if (!count)
> > > +			break;
> > > +		start = vmap_block_vaddr(vb->va->va_start, rs);
> > > +		if (addr < start) {
> > > +			if (count == 0)
> > > +				break;
> > > +			*buf = '\0';
> > > +			buf++;
> > > +			addr++;
> > > +			count--;
> > > +		}
>
> Very sorry, Lorenzo, I just noticed this mail. It's very weird. Earlier,
> Uladzislau's reply to patch 2/7 got to be seen in my mutt mail client 10
> days later. I am not sure it's my mail client's problem, or a mail server
> delivery issue.
>

Odd, maybe try lei with mutt I find that works well :)

> >
> > I may be missing something here, but is this not essentially 'if the address is
> > below a used region, write a single null byte into the buffer and continue,
> > assuming we are now in a used area?'
>
> Not sure if I got you. for_each_set_bitrange only iterates the used
> regions. So in the for loop, what we do is fill zero into the buffer
> below the used region, then read out the used region. You said
> 'continue', I don't understand what it means.
>
> Assume we have 3 used regions in one vmap block, see below diagram.
>      |_______|______________|________|_____________|_____|_____________|______|
>      |hole 0 |used region 0 |hole 1  |used region 1|hole2|used region2 |hole 3 |
>
> hole 0,1,2 will be set zero when we iterate to the used region above
> them. And the last hole 3 is set at the end of this function. Please
> help point it out if I got it wrong.

Maybe let me rephrase:-

- We want to read `count` bytes from `addr` into `buf`
- We iterate over _used_ blocks, placing the start/end of each block in `rs`, `re`
  respectively.
- If we hit a block whose start address is above the one in which we are interested then:-
  - Place a zero byte in the buffer
  - Increment `addr` by 1 byte
  - Decrement the `count` by 1 byte
  - Carry on

I am seriously confused as to why we do this? Surely we should be checking
whether the range [addr, addr + count) overlaps this block at all, and only then
copying the relevant region?

It's the fact that blocks are at base page granularity but then this condition
is at byte granularity that is confusing to me (again it's _very_ possible I am
just being dumb here and missing something, just really want to understand this
better :)

> > > -		vm = va->vm;
> > > -		vaddr = (char *) vm->addr;
> > > -		if (addr >= vaddr + get_vm_area_size(vm))
> > > +		vaddr = (char *) va->va_start;
> > > +		size = flags ? va_size(va) : get_vm_area_size(vm);
> >
> > For example here, I feel that this ternary should be reversed and based on
> > whether vm is null, unles we expect vm to ever be non-null _and_ flags to be
> > set?
>
> Now only vm_map_ram area sets flags, all other types has vm not null.
> Since those temporary state, e.g vm==NULL, flags==0 case has been
> filtered out. Is below you suggested?
>
> 		size = (!vm&&flags)? va_size(va) : get_vm_area_size(vm);
> 		or
> 		size = (vm&&!flags)? get_vm_area_size(vm):va_size(va);
>

Sorry I didn't phrase this very well, my point is that the key thing you're
relying on here is whether vm exists in order to use it so I simply meant:-

size = vm ? get_vm_area_size(vm) : va_size(va);

This just makes it really explicit that you need vm to be non-NULL, and you've
already done the flags check before so this should suffice.