On Friday 05 Mar 2021 at 14:39:42 (+0000), Will Deacon wrote: > On Tue, Mar 02, 2021 at 02:59:58PM +0000, Quentin Perret wrote: > > +/** > > + * kvm_pgtable_stage2_idmap_greedy() - Identity-map an Intermediate Physical > > + * Address with a leaf entry at the highest > > + * possible level. > > Not sure it's worth mentioning "highest possible level" here, as > realistically the caller still has to provide a memcache to deal with the > worst case and the structure of the page-table shouldn't matter. Right, we need to pass a range so I suppose that should be enough to say 'this tries to cover large portions of memory'. > > + * @pgt: Page-table structure initialised by kvm_pgtable_*_init(). > > + * @addr: Input address to identity-map. > > + * @prot: Permissions and attributes for the mapping. > > + * @range: Boundaries of the maximum memory region to map. > > + * @mc: Cache of pre-allocated memory from which to allocate page-table > > + * pages. > > + * > > + * This function attempts to install high-level identity-mappings covering @addr > > "high-level"? (again, I think I'd just drop this) > > > + * without overriding existing mappings with incompatible permissions or > > + * attributes. An existing table entry may be coalesced into a block mapping > > + * if and only if it covers @addr and all its leafs are either invalid and/or > > s/leafs/leaf entries/ Ack for both. > > + * have permissions and attributes strictly matching @prot. The mapping is > > + * guaranteed to be contained within the boundaries specified by @range at call > > + * time. If only a subset of the memory specified by @range is mapped (because > > + * of e.g. alignment issues or existing incompatible mappings), @range will be > > + * updated accordingly. > > + * > > + * Return: 0 on success, negative error code on failure. > > + */ > > +int kvm_pgtable_stage2_idmap_greedy(struct kvm_pgtable *pgt, u64 addr, > > + enum kvm_pgtable_prot prot, > > + struct kvm_mem_range *range, > > + void *mc); > > #endif /* __ARM64_KVM_PGTABLE_H__ */ > > diff --git a/arch/arm64/kvm/hyp/pgtable.c b/arch/arm64/kvm/hyp/pgtable.c > > index 8aa01a9e2603..6897d771e2b2 100644 > > --- a/arch/arm64/kvm/hyp/pgtable.c > > +++ b/arch/arm64/kvm/hyp/pgtable.c > > @@ -987,3 +987,122 @@ void kvm_pgtable_stage2_destroy(struct kvm_pgtable *pgt) > > pgt->mm_ops->free_pages_exact(pgt->pgd, pgd_sz); > > pgt->pgd = NULL; > > } > > + > > +struct stage2_reduce_range_data { > > + kvm_pte_t attr; > > + u64 target_addr; > > + u32 start_level; > > + struct kvm_mem_range *range; > > +}; > > + > > +static int __stage2_reduce_range(struct stage2_reduce_range_data *data, u64 addr) > > +{ > > + u32 level = data->start_level; > > + > > + for (; level < KVM_PGTABLE_MAX_LEVELS; level++) { > > + u64 granule = kvm_granule_size(level); > > + u64 start = ALIGN_DOWN(data->target_addr, granule); > > + u64 end = start + granule; > > + > > + /* > > + * The pinned address is in the current range, try one level > > + * deeper. > > + */ > > + if (start == ALIGN_DOWN(addr, granule)) > > + continue; > > + > > + /* > > + * Make sure the current range is a reduction of the existing > > + * range before updating it. > > + */ > > + if (data->range->start <= start && end <= data->range->end) { > > + data->start_level = level; > > + data->range->start = start; > > + data->range->end = end; > > + return 0; > > + } > > + } > > + > > + return -EINVAL; > > +} > > + > > +#define KVM_PTE_LEAF_S2_COMPAT_MASK (KVM_PTE_LEAF_ATTR_S2_PERMS | \ > > + KVM_PTE_LEAF_ATTR_LO_S2_MEMATTR | \ > > + KVM_PTE_LEAF_SW_BIT_PROT_NONE) > > + > > +static int stage2_reduce_range_walker(u64 addr, u64 end, u32 level, > > + kvm_pte_t *ptep, > > + enum kvm_pgtable_walk_flags flag, > > + void * const arg) > > +{ > > + struct stage2_reduce_range_data *data = arg; > > + kvm_pte_t attr; > > + int ret; > > + > > + if (addr < data->range->start || addr >= data->range->end) > > + return 0; > > + > > + attr = *ptep & KVM_PTE_LEAF_S2_COMPAT_MASK; > > + if (!attr || attr == data->attr) > > + return 0; > > + > > + /* > > + * An existing mapping with incompatible protection attributes is > > + * 'pinned', so reduce the range if we hit one. > > + */ > > + ret = __stage2_reduce_range(data, addr); > > + if (ret) > > + return ret; > > + > > + return -EAGAIN; > > +} > > + > > +static int stage2_reduce_range(struct kvm_pgtable *pgt, u64 addr, > > + enum kvm_pgtable_prot prot, > > + struct kvm_mem_range *range) > > +{ > > + struct stage2_reduce_range_data data = { > > + .start_level = pgt->start_level, > > + .range = range, > > + .target_addr = addr, > > + }; > > + struct kvm_pgtable_walker walker = { > > + .cb = stage2_reduce_range_walker, > > + .flags = KVM_PGTABLE_WALK_LEAF, > > + .arg = &data, > > + }; > > + int ret; > > + > > + data.attr = stage2_get_prot_attr(prot) & KVM_PTE_LEAF_S2_COMPAT_MASK; > > + if (!data.attr) > > + return -EINVAL; > > (this will need updating based on the other discussion we had) Ack. > > + /* Reduce the kvm_mem_range to a granule size */ > > + ret = __stage2_reduce_range(&data, range->end); > > + if (ret) > > + return ret; > > + > > + /* Walk the range to check permissions and reduce further if needed */ > > + do { > > + ret = kvm_pgtable_walk(pgt, range->start, range->end, &walker); > > (we spent some time debugging an issue here and you spotted that you're > passing range->end instead of the size ;) Yep, I have the fix applied locally, and ready to fly in v4 :) > > + } while (ret == -EAGAIN); > > I'm a bit nervous about this loop -- what guarantees forward progress here? > Can we return to the host after a few tries instead? -EAGAIN only happens when we've been able to successfully reduce the range to a potentially valid granule size. That can't happen infinitely. We're guaranteed to fail when trying to reduce the range to a granularity smaller than PAGE_SIZE (the -EINVAL case of __stage2_reduce_range), which is indicative of a host memory abort in a page it should not access (because marked PROT_NONE for instance). Cheers, Quentin _______________________________________________ kvmarm mailing list kvmarm@xxxxxxxxxxxxxxxxxxxxx https://lists.cs.columbia.edu/mailman/listinfo/kvmarm