Reformat.
[bertos.git] / bertos / fs / battfs_test.c
index c1cd89e6e418c454f1c2d52d89d155ad70edc866..019ee48822066b01a45fa48a1bb7e6cedb0c18dc 100644 (file)
@@ -127,6 +127,7 @@ static bool disk_close(struct BattFsSuper *d)
 static void testCheck(BattFsSuper *disk, pgcnt_t *reference)
 {
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
 
        for (int i = 0; i < disk->page_count; i++)
        {
@@ -152,6 +153,7 @@ static void testCheck(BattFsSuper *disk, pgcnt_t *reference)
                        exit(2);
                }
        }
+       ASSERT(battfs_fsck(disk));
        battfs_close(disk);
 }
 
@@ -182,7 +184,7 @@ static void disk1File(BattFsSuper *disk)
 
        for (int i = 0; i < PAGE_COUNT; i++)
        {
-               battfs_writeTestBlock(disk, i, 0, 0, 0, i);
+               battfs_writeTestBlock(disk, i, 0, 0, disk->data_size, i);
                ref[i] = i;
        }
        fclose(fp);
@@ -202,7 +204,7 @@ static void diskHalfFile(BattFsSuper *disk)
 
        for (int i = 0; i < PAGE_COUNT / 2; i++)
        {
-               battfs_writeTestBlock(disk, i, 0, 0, 0, i);
+               battfs_writeTestBlock(disk, i, 0, 0, disk->data_size, i);
                ref[i] = i;
        }
        fseek(fp, FILE_SIZE / 2, SEEK_SET);
@@ -229,9 +231,9 @@ static void oldSeq1(BattFsSuper *disk)
 
        fp = fopen(test_filename, "w+");
        // page, inode, seq, fill, pgoff
-       battfs_writeTestBlock(disk, 0, 0, 0, 0, 0);
-       battfs_writeTestBlock(disk, 1, 0, 0, 0, 1);
-       battfs_writeTestBlock(disk, 2, 0, 1, 0, 1);
+       battfs_writeTestBlock(disk, 0, 0, 0, disk->data_size, 0);
+       battfs_writeTestBlock(disk, 1, 0, 0, disk->data_size, 1);
+       battfs_writeTestBlock(disk, 2, 0, 1, disk->data_size, 1);
        disk->erase(disk, 3);
 
 
@@ -253,9 +255,9 @@ static void oldSeq2(BattFsSuper *disk)
 
        fp = fopen(test_filename, "w+");
        // page, inode, seq, fill, pgoff
-       battfs_writeTestBlock(disk, 0, 0, 0, 0, 0);
-       battfs_writeTestBlock(disk, 1, 0, 1, 0, 1);
-       battfs_writeTestBlock(disk, 2, 0, 0, 0, 1);
+       battfs_writeTestBlock(disk, 0, 0, 0, disk->data_size, 0);
+       battfs_writeTestBlock(disk, 1, 0, 1, disk->data_size, 1);
+       battfs_writeTestBlock(disk, 2, 0, 0, disk->data_size, 1);
        disk->erase(disk, 3);
 
        fclose(fp);
@@ -278,9 +280,9 @@ static void oldSeq3(BattFsSuper *disk)
 
        // page, inode, seq, fill, pgoff
        disk->erase(disk, 0);
-       battfs_writeTestBlock(disk, 1, 0, 0, 0, 0);
-       battfs_writeTestBlock(disk, 2, 0, 1, 0, 1);
-       battfs_writeTestBlock(disk, 3, 0, 0, 0, 1);
+       battfs_writeTestBlock(disk, 1, 0, 0, disk->data_size, 0);
+       battfs_writeTestBlock(disk, 2, 0, 1, disk->data_size, 1);
+       battfs_writeTestBlock(disk, 3, 0, 0, disk->data_size, 1);
 
 
        fclose(fp);
@@ -303,13 +305,13 @@ static void oldSeq2File(BattFsSuper *disk)
 
        // page, inode, seq, fill, pgoff
        disk->erase(disk, 0);
-       battfs_writeTestBlock(disk, 1, 0, 0, 0, 0);
-       battfs_writeTestBlock(disk, 2, 0, 3, 0, 1);
-       battfs_writeTestBlock(disk, 3, 0, 0, 0, 1);
+       battfs_writeTestBlock(disk, 1, 0, 0, disk->data_size, 0);
+       battfs_writeTestBlock(disk, 2, 0, 3, disk->data_size, 1);
+       battfs_writeTestBlock(disk, 3, 0, 0, disk->data_size, 1);
        disk->erase(disk, 4);
-       battfs_writeTestBlock(disk, 5, 4, 0, 0, 0);
-       battfs_writeTestBlock(disk, 6, 4, 1, 0, 1);
-       battfs_writeTestBlock(disk, 7, 4, 0, 0, 1);
+       battfs_writeTestBlock(disk, 5, 4, 0, disk->data_size, 0);
+       battfs_writeTestBlock(disk, 6, 4, 1, disk->data_size, 1);
+       battfs_writeTestBlock(disk, 7, 4, 0, disk->data_size, 1);
 
 
        fclose(fp);
@@ -353,6 +355,7 @@ static void openFile(BattFsSuper *disk)
        fclose(fp);
 
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
        ASSERT(!battfs_fileExists(disk, INEXISTENT_INODE));
 
        ASSERT(battfs_fileExists(disk, INODE));
@@ -387,6 +390,7 @@ static void openFile(BattFsSuper *disk)
        ASSERT(kfile_close(&fd1.fd) == 0);
        ASSERT(kfile_close(&fd2.fd) == 0);
        ASSERT(LIST_EMPTY(&disk->file_opened_list));
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_close(disk));
 
        TRACEMSG("10: passed\n");
@@ -418,6 +422,7 @@ static void readFile(BattFsSuper *disk)
        fclose(fp);
 
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_fileopen(disk, &fd1, INODE, MODE));
        ASSERT(kfile_read(&fd1.fd, buf, sizeof(buf)) == sizeof(buf));
        ASSERT(fd1.fd.seek_pos == sizeof(buf));
@@ -425,6 +430,7 @@ static void readFile(BattFsSuper *disk)
                ASSERT(buf[i] == 0xff);
 
        ASSERT(kfile_close(&fd1.fd) == 0);
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_close(disk));
 
        TRACEMSG("11: passed\n");
@@ -455,6 +461,7 @@ static void readAcross(BattFsSuper *disk)
        fclose(fp);
 
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_fileopen(disk, &fd1, INODE, MODE));
 
        ASSERT(kfile_read(&fd1.fd, buf, sizeof(buf)) == sizeof(buf));
@@ -484,6 +491,7 @@ static void readAcross(BattFsSuper *disk)
        ASSERT(fd1.fd.seek_pos = (kfile_off_t)fd1.fd.size);
 
        ASSERT(kfile_close(&fd1.fd) == 0);
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_close(disk));
 
        TRACEMSG("12: passed\n");
@@ -519,6 +527,7 @@ static void writeFile(BattFsSuper *disk)
                buf[i] = i;
 
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_fileopen(disk, &fd1, INODE, MODE));
        ASSERT(kfile_write(&fd1.fd, buf, sizeof(buf)) == sizeof(buf));
        ASSERT(fd1.fd.seek_pos == sizeof(buf));
@@ -531,6 +540,7 @@ static void writeFile(BattFsSuper *disk)
                ASSERT(buf[i] == i);
 
        ASSERT(kfile_close(&fd1.fd) == 0);
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_close(disk));
 
        TRACEMSG("13: passed\n");
@@ -561,6 +571,7 @@ static void writeAcross(BattFsSuper *disk)
        fclose(fp);
 
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_fileopen(disk, &fd1, INODE, MODE));
 
        uint8_t val = 0;
@@ -601,6 +612,7 @@ static void writeAcross(BattFsSuper *disk)
        ASSERT(fd1.fd.seek_pos == (kfile_off_t)sizeof(buf) * 3);
 
        ASSERT(kfile_close(&fd1.fd) == 0);
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_close(disk));
 
        TRACEMSG("14: passed\n");
@@ -621,6 +633,7 @@ static void createFile(BattFsSuper *disk)
        unsigned int MODE = BATTFS_CREATE;
 
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_fileopen(disk, &fd1, INODE, MODE));
        for (int i = 0; i < FILE_SIZE / 2; i++)
                ASSERT(kfile_putc(i, &fd1.fd) != EOF);
@@ -628,9 +641,11 @@ static void createFile(BattFsSuper *disk)
        ASSERT(fd1.fd.seek_pos == FILE_SIZE / 2);
        ASSERT(fd1.fd.size == FILE_SIZE / 2);
        ASSERT(kfile_close(&fd1.fd) == 0);
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_close(disk));
 
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_fileopen(disk, &fd1, INODE, 0));
        ASSERT(fd1.fd.size == FILE_SIZE / 2);
        ASSERT(fd1.fd.seek_pos == 0);
@@ -644,6 +659,7 @@ static void createFile(BattFsSuper *disk)
 
        ASSERT(fd1.fd.seek_pos == FILE_SIZE / 2);
        ASSERT(kfile_close(&fd1.fd) == 0);
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_close(disk));
 
 
@@ -666,12 +682,13 @@ static void multipleWrite(BattFsSuper *disk)
        uint8_t buf[1000];
 
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_fileopen(disk, &fd1, INODE, MODE));
 
        int j;
        for (j = 1; j < 1013; j++)
        {
-               for (int i = 0; i < sizeof(buf); i++)
+               for (unsigned i = 0; i < sizeof(buf); i++)
                        buf[i] = j+i;
 
                ASSERT(kfile_write(&fd1.fd, buf, sizeof(buf)) == sizeof(buf));
@@ -681,23 +698,26 @@ static void multipleWrite(BattFsSuper *disk)
                memset(buf, 0, sizeof(buf));
                ASSERT(kfile_read(&fd1.fd, buf, sizeof(buf)) == sizeof(buf));
                ASSERT(fd1.fd.seek_pos == sizeof(buf));
-               for (int i = 0; i < sizeof(buf); i++)
+               for (unsigned i = 0; i < sizeof(buf); i++)
                        ASSERT(buf[i] == ((j+i) & 0xff));
                ASSERT(kfile_seek(&fd1.fd, 0, KSM_SEEK_SET) == 0);
                ASSERT(disk->free_bytes == disk->disk_size - sizeof(buf));
        }
        ASSERT(kfile_close(&fd1.fd) == 0);
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_close(disk));
 
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
        ASSERT(disk->free_bytes == disk->disk_size - sizeof(buf));
        ASSERT(battfs_fileopen(disk, &fd1, INODE, 0));
        ASSERT(fd1.fd.size == sizeof(buf));
        memset(buf, 0, sizeof(buf));
        ASSERT(kfile_read(&fd1.fd, buf, sizeof(buf)) == sizeof(buf));
-       for (int i = 0; i < sizeof(buf); i++)
+       for (unsigned i = 0; i < sizeof(buf); i++)
                        ASSERT(buf[i] == ((j-1+i) & 0xff));
        ASSERT(kfile_close(&fd1.fd) == 0);
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_close(disk));
 
 
@@ -720,29 +740,31 @@ static void increaseFile(BattFsSuper *disk)
        uint8_t buf[1000];
 
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_fileopen(disk, &fd1, INODE1, MODE));
        ASSERT(battfs_fileopen(disk, &fd2, INODE2, MODE));
-       for (int i = 0; i < sizeof(buf); i++)
+       for (unsigned i = 0; i < sizeof(buf); i++)
                ASSERT(kfile_putc(i, &fd2.fd) != EOF);
        ASSERT(kfile_seek(&fd2.fd, 0, KSM_SEEK_SET) == 0);
        memset(buf, 0, sizeof(buf));
        ASSERT(kfile_read(&fd2.fd, buf, sizeof(buf)) == sizeof(buf));
 
-       for (int i = 0; i < sizeof(buf); i++)
+       for (unsigned i = 0; i < sizeof(buf); i++)
                ASSERT(buf[i] == (i & 0xff));
        ASSERT(kfile_seek(&fd2.fd, 0, KSM_SEEK_SET) == 0);
 
-       for (int i = 0; i < sizeof(buf); i++)
+       for (unsigned i = 0; i < sizeof(buf); i++)
                ASSERT(kfile_putc(i, &fd1.fd) != EOF);
 
        memset(buf, 0, sizeof(buf));
        ASSERT(kfile_read(&fd2.fd, buf, sizeof(buf)) == sizeof(buf));
 
-       for (int i = 0; i < sizeof(buf); i++)
+       for (unsigned i = 0; i < sizeof(buf); i++)
                ASSERT(buf[i] == (i & 0xff));
 
        ASSERT(kfile_close(&fd1.fd) == 0);
        ASSERT(kfile_close(&fd2.fd) == 0);
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_close(disk));
 
        TRACEMSG("17: passed\n");
@@ -774,22 +796,217 @@ static void readEOF(BattFsSuper *disk)
        fclose(fp);
 
        ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_fileopen(disk, &fd1, INODE, MODE));
        ASSERT(kfile_seek(&fd1.fd, fd1.fd.size + 10, SEEK_SET) == fd1.fd.size + 10);
        ASSERT(fd1.fd.seek_pos == fd1.fd.size + 10);
        ASSERT(kfile_read(&fd1.fd, buf, sizeof(buf)) == 0);
 
        ASSERT(kfile_close(&fd1.fd) == 0);
+       ASSERT(battfs_fsck(disk));
        ASSERT(battfs_close(disk));
 
        TRACEMSG("18: passed\n");
 }
 
+static void writeEOF(BattFsSuper *disk)
+{
+       TRACEMSG("19: writing over EOF test\n");
+
+       FILE *fpt = fopen(test_filename, "w+");
+
+       for (int i = 0; i < FILE_SIZE / 5; i++)
+               fputc(0xff, fpt);
+       fclose(fpt);
+
+       BattFs fd1;
+       inode_t INODE = 0;
+       unsigned int MODE = BATTFS_CREATE;
+       uint8_t buf[FILE_SIZE / 13];
+
+       for (int i = 0; i < 2; i++)
+               buf[i] = i;
+
+       ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
+       disk_size_t prev_free = disk->free_bytes;
+       ASSERT(battfs_fileopen(disk, &fd1, INODE, MODE));
+       ASSERT(fd1.fd.size == 0);
+
+       ASSERT(kfile_seek(&fd1.fd, 2, KSM_SEEK_END) == 2);
+       ASSERT(kfile_write(&fd1.fd, buf, 2));
+       ASSERT(fd1.fd.seek_pos == 4);
+       ASSERT(fd1.fd.size == 4);
+       ASSERT(disk->free_bytes == prev_free - 4);
+       ASSERT(kfile_seek(&fd1.fd, 0, KSM_SEEK_SET) == 0);
+       memset(buf, 0, 2);
+       ASSERT(kfile_read(&fd1.fd, buf, 2) == 2);
+       for (int i = 0; i < 2; i++)
+               ASSERT(buf[i] == 0);
+
+       memset(buf, 0, 2);
+       ASSERT(kfile_read(&fd1.fd, buf, 2) == 2);
+       for (int i = 0; i < 2; i++)
+               ASSERT(buf[i] == (i & 0xff));
+
+       ASSERT(kfile_seek(&fd1.fd, sizeof(buf), KSM_SEEK_END) == sizeof(buf) + 4);
+       for (unsigned i = 0; i < sizeof(buf); i++)
+               buf[i] = i;
+       ASSERT(kfile_write(&fd1.fd, buf, sizeof(buf)));
+       ASSERT(fd1.fd.seek_pos == sizeof(buf) * 2 + 4);
+       ASSERT(fd1.fd.size == sizeof(buf) * 2 + 4);
+       ASSERT(disk->free_bytes == prev_free - sizeof(buf) * 2 - 4);
+
+       ASSERT(kfile_seek(&fd1.fd, 0, KSM_SEEK_SET) == 0);
+
+       memset(buf, 0, 2);
+       ASSERT(kfile_read(&fd1.fd, buf, 2) == 2);
+       ASSERT(fd1.fd.seek_pos == 2);
+       for (int i = 0; i < 2; i++)
+               ASSERT(buf[i] == 0);
+
+       memset(buf, 0, 2);
+       ASSERT(kfile_read(&fd1.fd, buf, 2) == 2);
+       ASSERT(fd1.fd.seek_pos == 4);
+       for (int i = 0; i < 2; i++)
+               ASSERT(buf[i] == (i & 0xff));
+
+       memset(buf, 0, 4);
+       ASSERT(kfile_read(&fd1.fd, buf, sizeof(buf)) == sizeof(buf));
+       ASSERT(fd1.fd.seek_pos == sizeof(buf) + 4);
+       for (unsigned i = 0; i < sizeof(buf); i++)
+               ASSERT(buf[i] == 0);
+
+       memset(buf, 0, sizeof(buf));
+       ASSERT(kfile_read(&fd1.fd, buf, sizeof(buf)) == sizeof(buf));
+       for (unsigned i = 0; i < sizeof(buf); i++)
+               ASSERT(buf[i] == (i & 0xff));
+
+       ASSERT(kfile_close(&fd1.fd) == 0);
+       ASSERT(battfs_fsck(disk));
+       ASSERT(battfs_close(disk));
+
+       TRACEMSG("19: passed\n");
+
+}
+
+static void endOfSpace(BattFsSuper *disk)
+{
+       TRACEMSG("20: what happens when disk space is over?\n");
+       BattFs fd1;
+       uint8_t buf[(PAGE_SIZE - BATTFS_HEADER_LEN) * 5];
+
+       fp = fopen(test_filename, "w+");
+
+       unsigned int PAGE_FILL = PAGE_SIZE - BATTFS_HEADER_LEN;
+       inode_t INODE = 0;
+       unsigned int MODE = BATTFS_CREATE;
+
+       disk->erase(disk, 0);
+       disk->erase(disk, 1);
+       disk->erase(disk, 2);
+       disk->erase(disk, 3);
+       fclose(fp);
+
+       ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
+       ASSERT(battfs_fileopen(disk, &fd1, INODE, MODE));
+       ASSERT(kfile_write(&fd1.fd, buf, sizeof(buf)) == PAGE_FILL * 4);
+       ASSERT(fd1.fd.size == PAGE_FILL * 4);
+       ASSERT(fd1.fd.seek_pos == PAGE_FILL * 4);
+       ASSERT(disk->free_bytes == 0);
+
+       ASSERT(kfile_close(&fd1.fd) == 0);
+       ASSERT(battfs_fsck(disk));
+       ASSERT(battfs_close(disk));
+
+       TRACEMSG("20: passed\n");
+}
+
+
+static void multipleFilesRW(BattFsSuper *disk)
+{
+       TRACEMSG("21: multiple files read/write test\n");
+
+       FILE *fpt = fopen(test_filename, "w+");
+
+       for (int i = 0; i < FILE_SIZE; i++)
+               fputc(0xff, fpt);
+       fclose(fpt);
+
+       #define N_FILES 10
+       BattFs fd[N_FILES];
+       inode_t INODE = 0;
+       unsigned int MODE = BATTFS_CREATE;
+       uint32_t buf[FILE_SIZE / (4 * N_FILES * sizeof(uint32_t))];
+
+       ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
+       for (inode_t i = 0; i < N_FILES; i++)
+               ASSERT(battfs_fileopen(disk, &fd[i], i, MODE));
+
+       for (int i = N_FILES - 1; i >= 0; i--)
+       {
+               for (uint32_t j = 0; j < countof(buf); j++)
+                       buf[j] = j+i;
+
+               ASSERT(kfile_write(&fd[i], buf, sizeof(buf)) == sizeof(buf));
+               ASSERT(fd[i].fd.size == sizeof(buf));
+               ASSERT(fd[i].fd.seek_pos == sizeof(buf));
+               ASSERT(kfile_seek(&fd[i].fd, 0, SEEK_SET) == 0);
+       }
+
+       for (inode_t i = 0; i < N_FILES; i++)
+       {
+               memset(buf, 0, sizeof(buf));
+               ASSERT(kfile_read(&fd[i], buf, sizeof(buf)) == sizeof(buf));
+
+               for (uint32_t j = 0; j < countof(buf); j++)
+                       ASSERT(buf[j] == j+i);
+
+               ASSERT(fd[i].fd.size == sizeof(buf));
+               ASSERT(fd[i].fd.seek_pos == sizeof(buf));
+               ASSERT(kfile_seek(&fd[i].fd, 0, SEEK_SET) == 0);
+       }
+
+       for (inode_t i = 0; i < N_FILES; i++)
+               ASSERT(kfile_close(&fd[i].fd) == 0);
+
+       ASSERT(battfs_fsck(disk));
+       ASSERT(battfs_close(disk));
+
+       ASSERT(battfs_init(disk));
+       ASSERT(battfs_fsck(disk));
+
+       for (inode_t i = 0; i < N_FILES; i++)
+               ASSERT(battfs_fileopen(disk, &fd[i], i, 0));
+
+       for (inode_t i = 0; i < N_FILES; i++)
+       {
+               memset(buf, 0, sizeof(buf));
+               ASSERT(kfile_read(&fd[i], buf, sizeof(buf)) == sizeof(buf));
+
+               for (uint32_t j = 0; j < countof(buf); j++)
+                       ASSERT(buf[j] == j+i);
+
+               ASSERT(fd[i].fd.size == sizeof(buf));
+               ASSERT(fd[i].fd.seek_pos == sizeof(buf));
+               ASSERT(kfile_seek(&fd[i].fd, 0, SEEK_SET) == 0);
+       }
+
+       for (inode_t i = 0; i < N_FILES; i++)
+               ASSERT(kfile_close(&fd[i].fd) == 0);
+
+       ASSERT(battfs_close(disk));
+       TRACEMSG("21: passed\n");
+}
+
 
 int battfs_testRun(void)
 {
        BattFsSuper disk;
 
+       disk.page_size = PAGE_SIZE;
        disk.open = disk_open;
        disk.read = disk_page_read;
        disk.load = disk_page_load;
@@ -815,6 +1032,9 @@ int battfs_testRun(void)
        multipleWrite(&disk);
        increaseFile(&disk);
        readEOF(&disk);
+       writeEOF(&disk);
+       endOfSpace(&disk);
+       multipleFilesRW(&disk);
 
        kprintf("All tests passed!\n");