diff mbox

[kvm-unit-tests,v2,06/12] phys_alloc: ensure we account all allocations

Message ID 20180117104005.29211-7-drjones@redhat.com (mailing list archive)
State New, archived
Headers show

Commit Message

Andrew Jones Jan. 17, 2018, 10:39 a.m. UTC
Also throw another assert into phys_alloc_aligned_safe.

Signed-off-by: Andrew Jones <drjones@redhat.com>
---
 lib/alloc_phys.c | 10 ++++++++++
 lib/alloc_phys.h |  6 ++++--
 2 files changed, 14 insertions(+), 2 deletions(-)

Comments

David Hildenbrand Jan. 17, 2018, 12:15 p.m. UTC | #1
On 17.01.2018 11:39, Andrew Jones wrote:
> Also throw another assert into phys_alloc_aligned_safe.
> 
> Signed-off-by: Andrew Jones <drjones@redhat.com>
> ---
>  lib/alloc_phys.c | 10 ++++++++++
>  lib/alloc_phys.h |  6 ++++--
>  2 files changed, 14 insertions(+), 2 deletions(-)
> 
> diff --git a/lib/alloc_phys.c b/lib/alloc_phys.c
> index d09395d1010c..736c3129fc02 100644
> --- a/lib/alloc_phys.c
> +++ b/lib/alloc_phys.c
> @@ -81,6 +81,8 @@ static phys_addr_t phys_alloc_aligned_safe(phys_addr_t size,
>  	if (safe && sizeof(long) == 4)
>  		top_safe = MIN(top_safe, 1ULL << 32);
>  
> +	assert(base < top_safe);
> +
>  	addr = ALIGN(base, align);
>  	size += addr - base;
>  
> @@ -116,6 +118,14 @@ void phys_alloc_get_unused(phys_addr_t *p_base, phys_addr_t *p_top)
>  {
>  	*p_base = base;
>  	*p_top = top;
> +	if (base == top)
> +		return;
> +	spin_lock(&lock);
> +	regions[nr_regions].base = base;
> +	regions[nr_regions].size = top - base;
> +	++nr_regions;
> +	base = top;
> +	spin_unlock(&lock);
>  }
>  
>  static void *early_memalign(size_t alignment, size_t size)
> diff --git a/lib/alloc_phys.h b/lib/alloc_phys.h
> index 848d3db23a1e..7cf59a661180 100644
> --- a/lib/alloc_phys.h
> +++ b/lib/alloc_phys.h
> @@ -37,8 +37,10 @@ extern void phys_alloc_set_minimum_alignment(phys_addr_t align);
>  extern void phys_alloc_show(void);
>  
>  /*
> - * phys_alloc_get_unused returns the addresses for the still-unused part
> - * of the initial free memory region passed to phys_alloc_init.
> + * phys_alloc_get_unused allocates all remaining memory from the region
> + * passed to phys_alloc_init, returning the newly allocated memory's base
> + * and top addresses. phys_allo_get_unused will still return base and top

s/phys_allo_get_unused/phys_alloc_get_unused/

Looks good to me
diff mbox

Patch

diff --git a/lib/alloc_phys.c b/lib/alloc_phys.c
index d09395d1010c..736c3129fc02 100644
--- a/lib/alloc_phys.c
+++ b/lib/alloc_phys.c
@@ -81,6 +81,8 @@  static phys_addr_t phys_alloc_aligned_safe(phys_addr_t size,
 	if (safe && sizeof(long) == 4)
 		top_safe = MIN(top_safe, 1ULL << 32);
 
+	assert(base < top_safe);
+
 	addr = ALIGN(base, align);
 	size += addr - base;
 
@@ -116,6 +118,14 @@  void phys_alloc_get_unused(phys_addr_t *p_base, phys_addr_t *p_top)
 {
 	*p_base = base;
 	*p_top = top;
+	if (base == top)
+		return;
+	spin_lock(&lock);
+	regions[nr_regions].base = base;
+	regions[nr_regions].size = top - base;
+	++nr_regions;
+	base = top;
+	spin_unlock(&lock);
 }
 
 static void *early_memalign(size_t alignment, size_t size)
diff --git a/lib/alloc_phys.h b/lib/alloc_phys.h
index 848d3db23a1e..7cf59a661180 100644
--- a/lib/alloc_phys.h
+++ b/lib/alloc_phys.h
@@ -37,8 +37,10 @@  extern void phys_alloc_set_minimum_alignment(phys_addr_t align);
 extern void phys_alloc_show(void);
 
 /*
- * phys_alloc_get_unused returns the addresses for the still-unused part
- * of the initial free memory region passed to phys_alloc_init.
+ * phys_alloc_get_unused allocates all remaining memory from the region
+ * passed to phys_alloc_init, returning the newly allocated memory's base
+ * and top addresses. phys_allo_get_unused will still return base and top
+ * when no free memory is remaining, but base will equal top.
  */
 extern void phys_alloc_get_unused(phys_addr_t *p_base, phys_addr_t *p_top);