demos/generic/main.c

   1 /*\r
   2  * Copyright (c) 2013 Grzegorz Kostka (kostka.grzegorz@gmail.com)\r
   3  * All rights reserved.\r
   4  *\r
   5  * Redistribution and use in source and binary forms, with or without\r
   6  * modification, are permitted provided that the following conditions\r
   7  * are met:\r
   8  *\r
   9  * - Redistributions of source code must retain the above copyright\r
  10  *   notice, this list of conditions and the following disclaimer.\r
  11  * - Redistributions in binary form must reproduce the above copyright\r
  12  *   notice, this list of conditions and the following disclaimer in the\r
  13  *   documentation and/or other materials provided with the distribution.\r
  14  * - The name of the author may not be used to endorse or promote products\r
  15  *   derived from this software without specific prior written permission.\r
  16  *\r
  17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR\r
  18  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES\r
  19  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.\r
  20  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,\r
  21  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT\r
  22  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\r
  23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\r
  24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\r
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF\r
  26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\r
  27  */\r
  28 \r
  29 #include <stdio.h>\r
  30 #include <stdlib.h>\r
  31 #include <string.h>\r
  32 #include <unistd.h>\r
  33 #include <getopt.h>\r
  34 #include <stdbool.h>\r
  35 #include <time.h>\r
  36 \r
  37 #include <ext4_filedev.h>\r
  38 #include <io_raw.h>\r
  39 #include <ext4.h>\r
  40 \r
  41 /**@brief   Input stream name.*/\r
  42 char input_name[128] = "ext2";\r
  43 \r
  44 /**@brief       Read-write size*/\r
  45 static int rw_szie  = 1024 * 1024;\r
  46 \r
  47 /**@brief       Read-write size*/\r
  48 static int rw_count = 10;\r
  49 \r
  50 /**@brief   Directory test count*/\r
  51 static int dir_cnt  = 0;\r
  52 \r
  53 /**@brief   Static or dynamic cache mode*/\r
  54 static bool cache_mode = true;\r
  55 \r
  56 /**@brief   Cleanup after test.*/\r
  57 static bool cleanup_flag = false;\r
  58 \r
  59 /**@brief   Block device stats.*/\r
  60 static bool bstat = false;\r
  61 \r
  62 /**@brief   Superblock stats.*/\r
  63 static bool sbstat = false;\r
  64 \r
  65 /**@brief   Indicates that input is windows partition.*/\r
  66 static bool winpart = false;\r
  67 \r
  68 /**@brief       File write buffer*/\r
  69 static uint8_t  *wr_buff;\r
  70 \r
  71 /**@brief       File read buffer.*/\r
  72 static uint8_t  *rd_buff;\r
  73 \r
  74 /**@brief       Block device handle.*/\r
  75 static struct ext4_blockdev *bd;\r
  76 \r
  77 /**@brief       Block cache handle.*/\r
  78 static struct ext4_bcache   *bc;\r
  79 \r
  80 static const char *usage = "                                                                    \n\\r
  81 Welcome in ext4 generic demo.                                                                   \n\\r
  82 Copyright (c) 2013 Grzegorz Kostka (kostka.grzegorz@gmail.com)  \n\\r
  83 Usage:                                                                                                                  \n\\r
  84     --i   - input file             (default = ext2)                             \n\\r
  85     --rws - single R/W size        (default = 1024)                             \n\\r
  86     --rwc - R/W count              (default = 10000)                    \n\\r
  87     --cache  - 0 static, 1 dynamic  (default = 1)                \n\\r
  88     --dirs   - directory test count (default = 0)                \n\\r
  89     --clean  - clean up after test                               \n\\r
  90     --bstat  - block device stats                                \n\\r
  91     --sbstat - superblock stats                                  \n\\r
  92     --wpart  - windows partition mode                            \n\\r
  93 \n";\r
  94 \r
  95 static char* entry_to_str(uint8_t type)\r
  96 {\r
  97         switch(type){\r
  98         case EXT4_DIRENTRY_UNKNOWN:\r
  99                 return "[UNK] ";\r
 100         case EXT4_DIRENTRY_REG_FILE:\r
 101                 return "[FIL] ";\r
 102         case EXT4_DIRENTRY_DIR:\r
 103                 return "[DIR] ";\r
 104         case EXT4_DIRENTRY_CHRDEV:\r
 105                 return "[CHA] ";\r
 106         case EXT4_DIRENTRY_BLKDEV:\r
 107                 return "[BLK] ";\r
 108         case EXT4_DIRENTRY_FIFO:\r
 109                 return "[FIF] ";\r
 110         case EXT4_DIRENTRY_SOCK:\r
 111                 return "[SOC] ";\r
 112         case EXT4_DIRENTRY_SYMLINK:\r
 113                 return "[SYM] ";\r
 114         default:\r
 115                 break;\r
 116         }\r
 117         return "[???]";\r
 118 }\r
 119 \r
 120 static void dir_ls(const char *path)\r
 121 {\r
 122         int j = 0;\r
 123         char sss[255];\r
 124         ext4_dir d;\r
 125         ext4_direntry *de;\r
 126 \r
 127         printf("**********************************************\n");\r
 128 \r
 129         ext4_dir_open(&d, path);\r
 130         de = ext4_dir_entry_get(&d, j++);\r
 131         printf("ls %s\n", path);\r
 132 \r
 133         while(de){\r
 134                 memcpy(sss, de->name, de->name_length);\r
 135                 sss[de->name_length] = 0;\r
 136                 printf("%s", entry_to_str(de->inode_type));\r
 137                 printf("%s", sss);\r
 138                 printf("\n");\r
 139                 de = ext4_dir_entry_get(&d, j++);\r
 140         }\r
 141         printf("**********************************************\n");\r
 142         ext4_dir_close(&d);\r
 143 }\r
 144 \r
 145 static void mp_stats(void)\r
 146 {\r
 147     struct ext4_mount_stats stats;\r
 148     ext4_mount_point_stats("/mp/", &stats);\r
 149 \r
 150     printf("**********************************************\n");\r
 151     printf("ext4_mount_point_stats\n");\r
 152     printf("inodes_count        = %u\n", stats.inodes_count);\r
 153     printf("free_inodes_count   = %u\n", stats.free_inodes_count);\r
 154     printf("blocks_count        = %u\n", (uint32_t)stats.blocks_count);\r
 155     printf("free_blocks_count   = %u\n", (uint32_t)stats.free_blocks_count);\r
 156     printf("block_size          = %u\n", stats.block_size);\r
 157     printf("block_group_count   = %u\n", stats.block_group_count);\r
 158     printf("blocks_per_group    = %u\n", stats.blocks_per_group);\r
 159     printf("inodes_per_group    = %u\n", stats.inodes_per_group);\r
 160     printf("volume_name         = %s\n", stats.volume_name);\r
 161 \r
 162     printf("**********************************************\n");\r
 163 \r
 164 }\r
 165 \r
 166 static void block_stats(void)\r
 167 {\r
 168     uint32_t i;\r
 169 \r
 170     printf("**********************************************\n");\r
 171     printf("ext4 blockdev stats\n");\r
 172     printf("bdev->bread_ctr          = %u\n", bd->bread_ctr);\r
 173     printf("bdev->bwrite_ctr         = %u\n", bd->bwrite_ctr);\r
 174 \r
 175 \r
 176     printf("bcache->ref_blocks       = %u\n", bc->ref_blocks);\r
 177     printf("bcache->max_ref_blocks   = %u\n", bc->max_ref_blocks);\r
 178     printf("bcache->lru_ctr          = %u\n", bc->lru_ctr);\r
 179 \r
 180     printf("\n");\r
 181     for (i = 0; i < bc->cnt; ++i) {\r
 182         printf("bcache->refctr[%d]     = %u\n", i, bc->refctr[i]);\r
 183     }\r
 184 \r
 185     printf("\n");\r
 186     for (i = 0; i < bc->cnt; ++i) {\r
 187         printf("bcache->lru_id[%d]     = %u\n", i, bc->lru_id[i]);\r
 188     }\r
 189 \r
 190     printf("\n");\r
 191     for (i = 0; i < bc->cnt; ++i) {\r
 192         printf("bcache->free_delay[%d] = %d\n", i, bc->free_delay[i]);\r
 193     }\r
 194 \r
 195     printf("\n");\r
 196     for (i = 0; i < bc->cnt; ++i) {\r
 197         printf("bcache->lba[%d]        = %u\n", i, (uint32_t)bc->lba[i]);\r
 198     }\r
 199 \r
 200     printf("**********************************************\n");\r
 201 }\r
 202 \r
 203 static clock_t get_ms(void)\r
 204 {\r
 205     return (clock() * 1000) / (CLOCKS_PER_SEC);\r
 206 }\r
 207 \r
 208 static bool dir_test(int len)\r
 209 {\r
 210     ext4_file f;\r
 211     int       r;\r
 212     int       i;\r
 213     char path[64];\r
 214     clock_t diff;\r
 215     clock_t stop;\r
 216     clock_t start;\r
 217     start = get_ms();\r
 218 \r
 219     printf("Directory create: /mp/dir1\n");\r
 220     r = ext4_dir_mk("/mp/dir1");\r
 221     if(r != EOK){\r
 222         printf("Unable to create directory: /mp/dir1\n");\r
 223         return false;\r
 224     }\r
 225 \r
 226 \r
 227     printf("Add files to: /mp/dir1\n");\r
 228     for (i = 0; i < len; ++i) {\r
 229         sprintf(path, "/mp/dir1/f%d", i);\r
 230         r = ext4_fopen(&f, path, "wb");\r
 231         if(r != EOK){\r
 232             printf("Unable to create file in directory: /mp/dir1\n");\r
 233             return false;\r
 234         }\r
 235     }\r
 236 \r
 237     stop =  get_ms();\r
 238     diff = stop - start;\r
 239     dir_ls("/mp/dir1");\r
 240     printf("dir_test time: %d ms\n", (int)diff);\r
 241     return true;\r
 242 }\r
 243 \r
 244 \r
 245 static bool file_test(void)\r
 246 {\r
 247     int r;\r
 248     uint32_t  size;\r
 249     ext4_file f;\r
 250     int i;\r
 251     clock_t start;\r
 252     clock_t stop;\r
 253     clock_t diff;\r
 254     uint32_t kbps;\r
 255     uint64_t size_bytes;\r
 256     /*Add hello world file.*/\r
 257     r = ext4_fopen(&f, "/mp/hello.txt", "wb");\r
 258     r = ext4_fwrite(&f, "Hello World !\n", strlen("Hello World !\n"), 0);\r
 259     r = ext4_fclose(&f);\r
 260 \r
 261 \r
 262     printf("ext4_fopen: test1\n");\r
 263 \r
 264     start = get_ms();\r
 265     r = ext4_fopen(&f, "/mp/test1", "wb");\r
 266     if(r != EOK){\r
 267         printf("ext4_fopen ERROR = %d\n", r);\r
 268         return false;\r
 269     }\r
 270 \r
 271     printf("ext4_write: %d * %d ..." , rw_szie, rw_count);\r
 272     for (i = 0; i < rw_count; ++i) {\r
 273 \r
 274         memset(wr_buff, i % 10 + '0', rw_szie);\r
 275 \r
 276         r = ext4_fwrite(&f, wr_buff, rw_szie, &size);\r
 277 \r
 278         if((r != EOK) || (size != rw_szie))\r
 279             break;\r
 280     }\r
 281 \r
 282     if(i != rw_count){\r
 283         printf("ERROR: rw_count = %d\n", i);\r
 284         return false;\r
 285     }\r
 286 \r
 287     printf("OK\n");\r
 288     stop = get_ms();\r
 289     diff = stop - start;\r
 290     size_bytes = rw_szie * rw_count;\r
 291     size_bytes = (size_bytes * 1000) / 1024;\r
 292     kbps = (size_bytes) / (diff + 1);\r
 293     printf("file_test write time: %d ms\n", (int)diff);\r
 294     printf("file_test write speed: %d KB/s\n", kbps);\r
 295     r = ext4_fclose(&f);\r
 296     printf("ext4_fopen: test1\n");\r
 297 \r
 298 \r
 299     start = get_ms();\r
 300     r = ext4_fopen(&f, "/mp/test1", "r+");\r
 301     if(r != EOK){\r
 302         printf("ext4_fopen ERROR = %d\n", r);\r
 303         return false;\r
 304     }\r
 305 \r
 306     printf("ext4_read: %d * %d ..." , rw_szie, rw_count);\r
 307 \r
 308     for (i = 0; i < rw_count; ++i) {\r
 309         memset(wr_buff, i % 10 + '0', rw_szie);\r
 310         r = ext4_fread(&f, rd_buff, rw_szie, &size);\r
 311 \r
 312         if((r != EOK) || (size != rw_szie))\r
 313             break;\r
 314 \r
 315         if(memcmp(rd_buff, wr_buff, rw_szie)){\r
 316             break;\r
 317         }\r
 318     }\r
 319     if(i != rw_count){\r
 320         printf("ERROR: rw_count = %d\n", i);\r
 321         return false;\r
 322     }\r
 323     printf("OK\n");\r
 324     stop = get_ms();\r
 325     diff = stop - start;\r
 326     size_bytes = rw_szie * rw_count;\r
 327     size_bytes = (size_bytes * 1000) / 1024;\r
 328     kbps = (size_bytes) / (diff + 1);\r
 329     printf("file_test read time: %d ms\n", (int)diff);\r
 330     printf("file_test read speed: %d KB/s\n", kbps);\r
 331     r = ext4_fclose(&f);\r
 332 \r
 333     return true;\r
 334 \r
 335 }\r
 336 static void cleanup(void)\r
 337 {\r
 338     clock_t start;\r
 339     clock_t stop;\r
 340     clock_t diff;\r
 341 \r
 342     ext4_fremove("/mp/hello.txt");\r
 343 \r
 344     printf("cleanup: remove /mp/test1\n");\r
 345     start = get_ms();\r
 346     ext4_fremove("/mp/test1");\r
 347     stop = get_ms();\r
 348     diff = stop - start;\r
 349     printf("cleanup: time: %d ms\n", (int)diff);\r
 350 \r
 351 \r
 352     printf("cleanup: remove /mp/test1\n");\r
 353     start =get_ms();\r
 354     ext4_dir_rm("/mp/dir1");\r
 355     stop = get_ms();\r
 356     diff = stop - start;\r
 357     printf("cleanup: time: %d ms\n", (int)diff);\r
 358 }\r
 359 \r
 360 static bool open_filedev(void)\r
 361 {\r
 362     ext4_filedev_filename(input_name);\r
 363     bd = ext4_filedev_get();\r
 364     bc = ext4_filecache_get();\r
 365     if(!bd || !bc){\r
 366         printf("Block device ERROR\n");\r
 367         return false;\r
 368     }\r
 369     return true;\r
 370 }\r
 371 \r
 372 static bool open_winpartition(void)\r
 373 {\r
 374 #ifdef WIN32\r
 375     ext4_io_raw_filename(input_name);\r
 376     bd = ext4_io_raw_dev_get();\r
 377     bc = ext4_io_raw_cache_get();\r
 378     if(!bd || !bc){\r
 379         printf("Block device ERROR\n");\r
 380         return false;\r
 381     }\r
 382     return true;\r
 383 #else\r
 384     printf("open_winpartition: this mode shouls be used only under windows !\n");\r
 385     return false;\r
 386 #endif\r
 387 }\r
 388 \r
 389 static bool mount(void)\r
 390 {\r
 391     int r;\r
 392     if(winpart){\r
 393          if(!open_winpartition())\r
 394              return false;\r
 395     }else{\r
 396         if(!open_filedev())\r
 397             return false;\r
 398 \r
 399     }\r
 400     wr_buff = malloc(rw_szie);\r
 401     rd_buff = malloc(rw_szie);\r
 402 \r
 403     if(!wr_buff || !rd_buff){\r
 404         printf("Read-Write allocation ERROR\n");\r
 405         return EXIT_FAILURE;\r
 406     }\r
 407 \r
 408     ext4_dmask_set(EXT4_DEBUG_ALL);\r
 409 \r
 410     r = ext4_device_register(bd, cache_mode ? 0 : bc, "ext4_filesim");\r
 411     if(r != EOK){\r
 412         printf("ext4_device_register ERROR = %d\n", r);\r
 413         return false;\r
 414     }\r
 415 \r
 416     r = ext4_mount("ext4_filesim", "/mp/");\r
 417     if(r != EOK){\r
 418         printf("ext4_mount ERROR = %d\n", r);\r
 419         return false;\r
 420     }\r
 421 \r
 422     return true;\r
 423 }\r
 424 \r
 425 static bool umount(void)\r
 426 {\r
 427     int r = ext4_umount("/mp/");\r
 428     if(r != EOK){\r
 429         printf("ext4_umount: FAIL %d", r);\r
 430         return false;\r
 431     }\r
 432     return true;\r
 433 }\r
 434 \r
 435 static bool parse_opt(int argc, char **argv)\r
 436 {\r
 437     int option_index = 0;\r
 438     int c;\r
 439 \r
 440     static struct option long_options[] =\r
 441       {\r
 442         {"in",      required_argument, 0, 'a'},\r
 443         {"rws",     required_argument, 0, 'b'},\r
 444         {"rwc",     required_argument, 0, 'c'},\r
 445         {"cache",   required_argument, 0, 'd'},\r
 446         {"dirs",    required_argument, 0, 'e'},\r
 447         {"clean",   no_argument,       0, 'f'},\r
 448         {"bstat",   no_argument,       0, 'g'},\r
 449         {"sbstat",  no_argument,       0, 'h'},\r
 450         {"wpart",   no_argument,       0, 'i'},\r
 451         {0, 0, 0, 0}\r
 452       };\r
 453 \r
 454     while(-1 != (c = getopt_long (argc, argv, "a:b:c:d:e:fghi", long_options, &option_index))) {\r
 455 \r
 456         switch(c){\r
 457             case 'a':\r
 458                 strcpy(input_name, optarg);\r
 459                 break;\r
 460             case 'b':\r
 461                 rw_szie = atoi(optarg);\r
 462                 break;\r
 463             case 'c':\r
 464                 rw_count = atoi(optarg);\r
 465                 break;\r
 466             case 'd':\r
 467                 cache_mode = atoi(optarg);\r
 468                 break;\r
 469             case 'e':\r
 470                 dir_cnt = atoi(optarg);\r
 471                 break;\r
 472             case 'f':\r
 473                 cleanup_flag = true;\r
 474                 break;\r
 475             case 'g':\r
 476                 bstat = true;\r
 477                 break;\r
 478             case 'h':\r
 479                 sbstat = true;\r
 480                 break;\r
 481             case 'i':\r
 482                 winpart = true;\r
 483                 break;\r
 484             default:\r
 485                 printf("%s", usage);\r
 486                 return false;\r
 487 \r
 488         }\r
 489     }\r
 490     return true;\r
 491 }\r
 492 \r
 493 int main(int argc, char **argv)\r
 494 {\r
 495     if(!parse_opt(argc, argv))\r
 496         return EXIT_FAILURE;\r
 497 \r
 498     printf("Test conditions:\n");\r
 499     printf("Imput name: %s\n", input_name);\r
 500     printf("RW size: %d\n",  rw_szie);\r
 501     printf("RW count: %d\n", rw_count);\r
 502     printf("Cache mode: %s\n", cache_mode ? "dynamic" : "static");\r
 503 \r
 504     if(!mount())\r
 505         return EXIT_FAILURE;\r
 506 \r
 507 \r
 508     cleanup();\r
 509 \r
 510     if(sbstat)\r
 511         mp_stats();\r
 512 \r
 513     dir_ls("/mp/");\r
 514     fflush(stdout);\r
 515     if(!dir_test(dir_cnt))\r
 516             return EXIT_FAILURE;\r
 517 \r
 518     fflush(stdout);\r
 519         if(!file_test())\r
 520             return EXIT_FAILURE;\r
 521 \r
 522         fflush(stdout);\r
 523         dir_ls("/mp/");\r
 524 \r
 525         if(sbstat)\r
 526             mp_stats();\r
 527 \r
 528         if(cleanup_flag)\r
 529             cleanup();\r
 530 \r
 531     if(bstat)\r
 532         block_stats();\r
 533 \r
 534     if(!umount())\r
 535         return EXIT_FAILURE;\r
 536 \r
 537     printf("Test finish: OK\n");\r
 538     return EXIT_SUCCESS;\r
 539 \r
 540 }\r