teec: use multiple of page size for page aligned buffers

When allocating a page aligned buffer, round the size up the next multiple of page size. With this we can guarantee that a part of that page isn't registered already. Signed-off-by: Jens Wiklander <[email protected]>
jenswi-linaro · Sep 28, 2023 · ede2cde · ede2cde
1 parent 8533e0e
commit ede2cde
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/libteec/src/tee_client_api.c b/libteec/src/tee_client_api.c
@@ -79,8 +79,10 @@ static void teec_mutex_unlock(pthread_mutex_t *mu)
 static void *teec_paged_aligned_alloc(size_t sz)
 {
 	void *p = NULL;
+	size_t page_sz = sysconf(_SC_PAGESIZE);
+	size_t aligned_sz = ((sz + page_sz - 1) / page_sz) * page_sz;
 
-	if (!posix_memalign(&p, sysconf(_SC_PAGESIZE), sz))
+	if (!posix_memalign(&p, page_sz, aligned_sz))
 		return p;
 
 	return NULL;