From: Simon Glass <s...@chromium.org>

Allow the alignment to be specified when creating a bloblist.

Signed-off-by: Simon Glass <s...@chromium.org>
Signed-off-by: Raymond Mao <raymond....@linaro.org>
---
 common/bloblist.c  |  5 +++--
 include/bloblist.h |  3 ++-
 test/bloblist.c    | 40 ++++++++++++++++++++++------------------
 3 files changed, 27 insertions(+), 21 deletions(-)

diff --git a/common/bloblist.c b/common/bloblist.c
index 55a7e1f591..0d482ecd71 100644
--- a/common/bloblist.c
+++ b/common/bloblist.c
@@ -341,7 +341,7 @@ static u32 bloblist_calc_chksum(struct bloblist_hdr *hdr)
        return chksum;
 }
 
-int bloblist_new(ulong addr, uint size)
+int bloblist_new(ulong addr, uint size, uint align_log2)
 {
        struct bloblist_hdr *hdr;
 
@@ -356,6 +356,7 @@ int bloblist_new(ulong addr, uint size)
        hdr->magic = BLOBLIST_MAGIC;
        hdr->size = size;
        hdr->alloced = hdr->hdr_size;
+       hdr->align_log2 = align_log2 ?: BLOBLIST_BLOB_ALIGN_LOG2;
        hdr->chksum = 0;
        gd->bloblist = hdr;
 
@@ -499,7 +500,7 @@ int bloblist_init(void)
                }
                log_debug("Creating new bloblist size %lx at %lx\n", size,
                          addr);
-               ret = bloblist_new(addr, size);
+               ret = bloblist_new(addr, size, 0);
        } else {
                log_debug("Found existing bloblist size %lx at %lx\n", size,
                          addr);
diff --git a/include/bloblist.h b/include/bloblist.h
index 57625ea004..c419c0d923 100644
--- a/include/bloblist.h
+++ b/include/bloblist.h
@@ -324,10 +324,11 @@ int bloblist_resize(uint tag, int new_size);
  *
  * @addr: Address of bloblist
  * @size: Initial size for bloblist
+ * @align_log2: Log base 2 of maximum alignment provided by this bloblist
  * Return: 0 if OK, -EFAULT if addr is not aligned correctly, -ENOSPC is the
  * area is not large enough
  */
-int bloblist_new(ulong addr, uint size);
+int bloblist_new(ulong addr, uint size, uint align_log2);
 
 /**
  * bloblist_check() - Check if a bloblist exists
diff --git a/test/bloblist.c b/test/bloblist.c
index be237c6155..b72603ba9a 100644
--- a/test/bloblist.c
+++ b/test/bloblist.c
@@ -72,15 +72,15 @@ static int bloblist_test_init(struct unit_test_state *uts)
        hdr = clear_bloblist();
        ut_asserteq(-ENOENT, bloblist_check(TEST_ADDR, TEST_BLOBLIST_SIZE));
        ut_asserteq_ptr(NULL, bloblist_check_magic(TEST_ADDR));
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        ut_asserteq_ptr(hdr, bloblist_check_magic(TEST_ADDR));
        hdr->version++;
        ut_asserteq(-EPROTONOSUPPORT, bloblist_check(TEST_ADDR,
                                                     TEST_BLOBLIST_SIZE));
 
-       ut_asserteq(-ENOSPC, bloblist_new(TEST_ADDR, 0xc));
-       ut_asserteq(-EFAULT, bloblist_new(1, TEST_BLOBLIST_SIZE));
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_asserteq(-ENOSPC, bloblist_new(TEST_ADDR, 0xc, 0));
+       ut_asserteq(-EFAULT, bloblist_new(1, TEST_BLOBLIST_SIZE, 0));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
 
        ut_asserteq(-EIO, bloblist_check(TEST_ADDR, TEST_BLOBLIST_SIZE));
        ut_assertok(bloblist_finish());
@@ -103,7 +103,7 @@ static int bloblist_test_blob(struct unit_test_state *uts)
        /* At the start there should be no records */
        hdr = clear_bloblist();
        ut_assertnull(bloblist_find(TEST_TAG, TEST_BLOBLIST_SIZE));
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        ut_asserteq(TEST_BLOBLIST_SIZE, bloblist_get_size());
        ut_asserteq(TEST_ADDR, bloblist_get_base());
        ut_asserteq(map_to_sysmem(hdr), TEST_ADDR);
@@ -141,7 +141,7 @@ static int bloblist_test_blob_ensure(struct unit_test_state 
*uts)
 
        /* At the start there should be no records */
        clear_bloblist();
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
 
        /* Test with an empty bloblist */
        size = TEST_SIZE;
@@ -173,7 +173,7 @@ static int bloblist_test_bad_blob(struct unit_test_state 
*uts)
        void *data;
 
        hdr = clear_bloblist();
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        data = hdr + 1;
        data += sizeof(struct bloblist_rec);
        ut_asserteq_addr(data, bloblist_ensure(TEST_TAG, TEST_SIZE));
@@ -189,7 +189,7 @@ static int bloblist_test_checksum(struct unit_test_state 
*uts)
        char *data, *data2;
 
        hdr = clear_bloblist();
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        ut_assertok(bloblist_finish());
        ut_assertok(bloblist_check(TEST_ADDR, TEST_BLOBLIST_SIZE));
 
@@ -206,6 +206,10 @@ static int bloblist_test_checksum(struct unit_test_state 
*uts)
        ut_asserteq(-EIO, bloblist_check(TEST_ADDR, TEST_BLOBLIST_SIZE));
        hdr->chksum--;
 
+       hdr->align_log2++;
+       ut_asserteq(-EIO, bloblist_check(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       hdr->align_log2--;
+
        /* Make sure the checksum changes when we add blobs */
        data = bloblist_add(TEST_TAG, TEST_SIZE, 0);
        ut_asserteq(-EIO, bloblist_check(TEST_ADDR, TEST_BLOBLIST_SIZE));
@@ -251,7 +255,7 @@ static int bloblist_test_cmd_info(struct unit_test_state 
*uts)
        char *data, *data2;
 
        hdr = clear_bloblist();
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        data = bloblist_ensure(TEST_TAG, TEST_SIZE);
        data2 = bloblist_ensure(TEST_TAG2, TEST_SIZE2);
 
@@ -277,7 +281,7 @@ static int bloblist_test_cmd_list(struct unit_test_state 
*uts)
        char *data, *data2;
 
        hdr = clear_bloblist();
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        data = bloblist_ensure(TEST_TAG, TEST_SIZE);
        data2 = bloblist_ensure(TEST_TAG2, TEST_SIZE2);
 
@@ -307,7 +311,7 @@ static int bloblist_test_align(struct unit_test_state *uts)
 
        /* At the start there should be no records */
        hdr = clear_bloblist();
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        ut_assertnull(bloblist_find(TEST_TAG, TEST_BLOBLIST_SIZE));
 
        /* Check the default alignment */
@@ -344,7 +348,7 @@ static int bloblist_test_align(struct unit_test_state *uts)
        memset(hdr, ERASE_BYTE, TEST_BLOBLIST_SIZE);
        memset(hdr, '\0', sizeof(*hdr));
        ut_assertok(bloblist_new(TEST_ADDR + BLOBLIST_ALIGN,
-                                TEST_BLOBLIST_SIZE));
+                                TEST_BLOBLIST_SIZE, 0));
 
        data = bloblist_add(1, 5, BLOBLIST_ALIGN_LOG2 + 1);
        ut_assertnonnull(data);
@@ -365,7 +369,7 @@ static int bloblist_test_reloc(struct unit_test_state *uts)
        ulong new_addr;
        ulong new_size;
 
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        old_ptr = map_sysmem(TEST_ADDR, TEST_BLOBLIST_SIZE);
 
        /* Add one blob and then one that won't fit */
@@ -404,7 +408,7 @@ static int bloblist_test_grow(struct unit_test_state *uts)
        memset(hdr, ERASE_BYTE, TEST_BLOBLIST_SIZE);
 
        /* Create two blobs */
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        blob1 = bloblist_add(TEST_TAG, small_size, 0);
        ut_assertnonnull(blob1);
        ut_assertok(check_zero(blob1, small_size));
@@ -456,7 +460,7 @@ static int bloblist_test_shrink(struct unit_test_state *uts)
        ptr = map_sysmem(TEST_ADDR, TEST_BLOBLIST_SIZE);
 
        /* Create two blobs */
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        blob1 = bloblist_add(TEST_TAG, small_size, 0);
        ut_assertnonnull(blob1);
        strcpy(blob1, test1_str);
@@ -506,7 +510,7 @@ static int bloblist_test_resize_fail(struct unit_test_state 
*uts)
        ptr = map_sysmem(TEST_ADDR, TEST_BLOBLIST_SIZE);
 
        /* Create two blobs */
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        blob1 = bloblist_add(TEST_TAG, small_size, 0);
        ut_assertnonnull(blob1);
 
@@ -543,7 +547,7 @@ static int bloblist_test_resize_last(struct unit_test_state 
*uts)
        hdr = ptr;
 
        /* Create two blobs */
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
        blob1 = bloblist_add(TEST_TAG, small_size, 0);
        ut_assertnonnull(blob1);
 
@@ -588,7 +592,7 @@ static int bloblist_test_blob_maxsize(struct 
unit_test_state *uts)
 
        /* At the start there should be no records */
        clear_bloblist();
-       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE));
+       ut_assertok(bloblist_new(TEST_ADDR, TEST_BLOBLIST_SIZE, 0));
 
        /* Add a blob that takes up all space */
        size = TEST_BLOBLIST_SIZE - sizeof(struct bloblist_hdr) -
-- 
2.25.1

Reply via email to