diff options
Diffstat (limited to 'tests/ubi-tests')
| -rw-r--r-- | tests/ubi-tests/Makefile | 3 | ||||
| -rw-r--r-- | tests/ubi-tests/io_update.c | 89 |
2 files changed, 66 insertions, 26 deletions
diff --git a/tests/ubi-tests/Makefile b/tests/ubi-tests/Makefile index b5c7e2f..0566ebc 100644 --- a/tests/ubi-tests/Makefile +++ b/tests/ubi-tests/Makefile @@ -1,6 +1,7 @@ LIBUBI_PATH=../../ubi-utils/ LIBUBI_SRC_PATH=../../ubi-utils/src LIBUBI_HEADER_PATH=../../ubi-utils/include +UBI_HEADERS_PATH=../../include/ UBIUTILS_PATH=../../ubi-utils/ CC := $(CROSS)gcc @@ -15,7 +16,7 @@ HELPERS=$(addprefix helpers/, $(HELPER_NAMES)) # it removes the. If you want to prevent the removal, uncomment the below #.SECONDARY: $(addsuffix .o, $(TESTS)) $(addsuffix .o, $(HELPERS)) -CFLAGS += -Wall -I$(LIBUBI_HEADER_PATH) -L. -O2 +CFLAGS += -Wall -I$(LIBUBI_HEADER_PATH) -I $(UBI_HEADERS_PATH) -L. -O2 all: ubi-utils libubi $(TESTS) $(HELPERS) diff --git a/tests/ubi-tests/io_update.c b/tests/ubi-tests/io_update.c index d433239..f7a122e 100644 --- a/tests/ubi-tests/io_update.c +++ b/tests/ubi-tests/io_update.c @@ -17,7 +17,7 @@ * * Author: Artem B. Bityutskiy * - * Test UBI volume update. + * Test UBI volume update and atomic LEB change */ #include <stdio.h> @@ -28,7 +28,9 @@ #include <sys/types.h> #include <sys/stat.h> #include <fcntl.h> -#include "libubi.h" + +#include <libubi.h> +#include <mtd/ubi-user.h> #define TESTNAME "io_update" #include "common.h" @@ -65,12 +67,15 @@ const char *node; /* * test_update1 - helper function for test_update(). */ -static int test_update1(struct ubi_vol_info *vol_info) +static int test_update1(struct ubi_vol_info *vol_info, int leb_change) { + long long total_len = leb_change ? vol_info->leb_size + : vol_info->rsvd_bytes; int sequences[SEQ_SZ][3] = SEQUENCES(dev_info.min_io_size, - vol_info->leb_size); + leb_change ? dev_info.min_io_size * 2 + : vol_info->leb_size); char vol_node[strlen(UBI_VOLUME_PATTERN) + 100]; - unsigned char buf[vol_info->rsvd_bytes]; + unsigned char buf[total_len]; int fd, i, j; sprintf(vol_node, UBI_VOLUME_PATTERN, dev_info.dev_num, @@ -86,14 +91,30 @@ static int test_update1(struct ubi_vol_info *vol_info) for (i = 0; i < SEQ_SZ; i++) { int ret, stop = 0, len = 0; off_t off = 0; - unsigned char buf1[vol_info->rsvd_bytes]; - - if (ubi_update_start(libubi, fd, vol_info->rsvd_bytes)) { - failed("ubi_update_start"); - goto close; + long long test_len; + unsigned char buf1[total_len]; + + /* + * test_len is LEB size (if we test atomic LEB change) or + * volume size (if we test update). For better test coverage, + * use a little smaller LEB change/update length. + */ + test_len = total_len - (rand() % (total_len / 10)); + + if (leb_change) { + if (ubi_leb_change_start(libubi, fd, 0, test_len, + UBI_SHORTTERM)) { + failed("ubi_update_start"); + goto close; + } + } else { + if (ubi_update_start(libubi, fd, test_len)) { + failed("ubi_update_start"); + goto close; + } } - for (j = 0; off < vol_info->rsvd_bytes; j++) { + for (j = 0; off < test_len; j++) { int n, rnd_len, l; if (!stop) { @@ -108,13 +129,18 @@ static int test_update1(struct ubi_vol_info *vol_info) * and the other part with 0xFFs to test how UBI * stripes 0xFFs multiple of I/O unit size. */ - if (off + l > vol_info->rsvd_bytes) - l = vol_info->rsvd_bytes - off; + if (off + l > test_len) + l = test_len - off; rnd_len = rand() % (l + 1); for (n = 0; n < rnd_len; n++) buf[off + n] = (unsigned char)rand(); memset(buf + off + rnd_len, 0xFF, l - rnd_len); + /* + * Deliberately pass len instead of l (len may be + * greater then l if this is the last chunk) because + * UBI have to read only l bytes anyway. + */ ret = write(fd, buf + off, len); if (ret < 0) { failed("write"); @@ -133,25 +159,31 @@ static int test_update1(struct ubi_vol_info *vol_info) /* Check data */ if ((ret = lseek(fd, SEEK_SET, 0)) != 0) { - if (ret < 0) - failed("lseek"); + failed("lseek"); err_msg("cannot seek to 0"); goto close; } - memset(buf1, 0x01, vol_info->rsvd_bytes); - ret = read(fd, buf1, vol_info->rsvd_bytes + 1); + + memset(buf1, 0x01, test_len); + + if (vol_info->type == UBI_STATIC_VOLUME) + /* + * Static volume must not let use read more then it + * contains. + */ + ret = read(fd, buf1, test_len + 100); + else + ret = read(fd, buf1, test_len); if (ret < 0) { failed("read"); - err_msg("failed to read %d bytes", - vol_info->rsvd_bytes + 1); + err_msg("failed to read %d bytes", test_len); goto close; } - if (ret != vol_info->rsvd_bytes) { - err_msg("failed to read %d bytes, read %d", - vol_info->rsvd_bytes, ret); + if (ret != test_len) { + err_msg("failed to read %d bytes, read %d", test_len, ret); goto close; } - if (memcmp(buf, buf1, vol_info->rsvd_bytes)) { + if (memcmp(buf, buf1, test_len)) { err_msg("data corruption"); goto close; } @@ -166,7 +198,7 @@ close: } /** - * test_update - check volume update capabilities. + * test_update - check volume update and atomic LEB change capabilities. * * @type volume type (%UBI_DYNAMIC_VOLUME or %UBI_STATIC_VOLUME) * @@ -208,11 +240,18 @@ static int test_update(int type) goto remove; } - if (test_update1(&vol_info)) { + if (test_update1(&vol_info, 0)) { err_msg("alignment = %d", req.alignment); goto remove; } + if (vol_info.type != UBI_STATIC_VOLUME) { + if (test_update1(&vol_info, 1)) { + err_msg("alignment = %d", req.alignment); + goto remove; + } + } + if (ubi_rmvol(libubi, node, req.vol_id)) { failed("ubi_rmvol"); return -1; |