-/*\r
- * Copyright (c) 2014 Grzegorz Kostka (kostka.grzegorz@gmail.com)\r
- * All rights reserved.\r
- *\r
- * Redistribution and use in source and binary forms, with or without\r
- * modification, are permitted provided that the following conditions\r
- * are met:\r
- *\r
- * - Redistributions of source code must retain the above copyright\r
- * notice, this list of conditions and the following disclaimer.\r
- * - Redistributions in binary form must reproduce the above copyright\r
- * notice, this list of conditions and the following disclaimer in the\r
- * documentation and/or other materials provided with the distribution.\r
- * - The name of the author may not be used to endorse or promote products\r
- * derived from this software without specific prior written permission.\r
- *\r
- * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR\r
- * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES\r
- * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.\r
- * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,\r
- * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT\r
- * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\r
- * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\r
- * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\r
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF\r
- * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\r
- */\r
-\r
-#include <errno.h>\r
-#include <stdio.h>\r
-#include <stdlib.h>\r
-#include <unistd.h>\r
-#include <string.h>\r
-#include <stdint.h>\r
-#include <stdbool.h>\r
-#include <getopt.h>\r
-#include <time.h>\r
-\r
-#ifdef WIN32\r
-#include <winsock2.h>\r
-#include <ws2tcpip.h>\r
-#include <windows.h>\r
-#else\r
-#include <sys/socket.h>\r
-#include <netinet/in.h>\r
-#include <arpa/inet.h>\r
-#include <sys/types.h>\r
-#endif\r
-\r
-\r
-#include <ext4_filedev.h>\r
-#include <io_raw.h>\r
-\r
-#include <ext4.h>\r
-\r
-static int winsock_init(void);\r
-static void winsock_fini(void);\r
-static char* entry_to_str(uint8_t type);\r
-\r
-#define MAX_FILES 64\r
-#define MAX_DIRS 64\r
-\r
-#define MAX_RW_BUFFER (1024 * 1024)\r
-#define RW_BUFFER_PATERN ('x')\r
-\r
-\r
-/**@brief Default connection port*/\r
-static int connection_port = 1234;\r
-\r
-/**@brief Default filesystem filename.*/\r
-static char *ext4_fname = "ext2";\r
-\r
-/**@brief Verbose mode*/\r
-static int verbose = 0;\r
-\r
-/**@brief Winpart mode*/\r
-static int winpart = 0;\r
-\r
-/**@brief Blockdev handle*/\r
-static struct ext4_blockdev *bd;\r
-\r
-static int cache_wb = 0;\r
-\r
-static char read_buffer[MAX_RW_BUFFER];\r
-static char write_buffer[MAX_RW_BUFFER];\r
-\r
-\r
-static const char *usage = " \n\\r
-Welcome in lwext4_server. \n\\r
-Copyright (c) 2013 Grzegorz Kostka (kostka.grzegorz@gmail.com) \n\\r
-Usage: \n\\r
- --image (-i) - ext2/3/4 image file \n\\r
- --port (-p) - server port \n\\r
- --verbose (-v) - verbose mode \n\\r
- --winpart (-w) - windows_partition mode \n\\r
- --cache_wb (-c) - cache writeback_mode \n\\r
-\n";\r
-\r
-\r
-\r
-/**@brief Open file instance descriptor.*/\r
-struct lwext4_files {\r
- char name[255];\r
- ext4_file fd;\r
-};\r
-\r
-/**@brief Open directory instance descriptor.*/\r
-struct lwext4_dirs {\r
- char name[255];\r
- ext4_dir fd;\r
-};\r
-\r
-/**@brief Library call opcode.*/\r
-struct lwext4_op_codes {\r
- char *func;\r
-};\r
-\r
-/**@brief Library call wraper.*/\r
-struct lwext4_call {\r
- int (*lwext4_call)(char *p);\r
-};\r
-\r
-/**@brief */\r
-static struct lwext4_files file_tab[MAX_FILES];\r
-\r
-/**@brief */\r
-static struct lwext4_dirs dir_tab[MAX_DIRS];\r
-\r
-/**@brief */\r
-static struct lwext4_op_codes op_codes[] = {\r
- "device_register",\r
- "mount",\r
- "umount",\r
- "mount_point_stats",\r
- "cache_write_back",\r
- "fremove",\r
- "fopen",\r
- "fclose",\r
- "fread",\r
- "fwrite",\r
- "fseek",\r
- "ftell",\r
- "fsize",\r
- "dir_rm",\r
- "dir_mk",\r
- "dir_open",\r
- "dir_close",\r
- "dir_entry_get",\r
-\r
- "multi_fcreate",\r
- "multi_fwrite",\r
- "multi_fread",\r
- "multi_fremove",\r
- "multi_dcreate",\r
- "multi_dremove",\r
- "stats_save",\r
- "stats_check",\r
-};\r
-\r
-int _device_register(char *p);\r
-int _mount(char *p);\r
-int _umount(char *p);\r
-int _mount_point_stats(char *p);\r
-int _cache_write_back(char *p);\r
-int _fremove(char *p);\r
-int _fopen(char *p);\r
-int _fclose(char *p);\r
-int _fread(char *p);\r
-int _fwrite(char *p);\r
-int _fseek(char *p);\r
-int _ftell(char *p);\r
-int _fsize(char *p);\r
-int _dir_rm(char *p);\r
-int _dir_mk(char *p);\r
-int _dir_open(char *p);\r
-int _dir_close(char *p);\r
-int _dir_close(char *p);\r
-int _dir_entry_get(char *p);\r
-\r
-int _multi_fcreate(char *p);\r
-int _multi_fwrite(char *p);\r
-int _multi_fread(char *p);\r
-int _multi_fremove(char *p);\r
-int _multi_dcreate(char *p);\r
-int _multi_dremove(char *p);\r
-int _stats_save(char *p);\r
-int _stats_check(char *p);\r
-\r
-/**@brief */\r
-static struct lwext4_call op_call[] = {\r
- _device_register, /*PARAMS(3): 0 cache_mode dev_name */\r
- _mount, /*PARAMS(2): dev_name mount_point */\r
- _umount, /*PARAMS(1): mount_point */\r
- _mount_point_stats,/*PARAMS(2): mount_point, 0 */\r
- _cache_write_back, /*PARAMS(2): mount_point, en */\r
- _fremove, /*PARAMS(1): path */\r
- _fopen, /*PARAMS(2): fid path flags */\r
- _fclose, /*PARAMS(1): fid */\r
- _fread, /*PARAMS(4): fid 0 len 0 */\r
- _fwrite, /*PARAMS(4): fid 0 len 0 */\r
- _fseek, /*PARAMS(2): fid off origin */\r
- _ftell, /*PARAMS(2): fid exp */\r
- _fsize, /*PARAMS(2): fid exp */\r
- _dir_rm, /*PARAMS(1): path */\r
- _dir_mk, /*PARAMS(1): path */\r
- _dir_open, /*PARAMS(2): did, path */\r
- _dir_close, /*PARAMS(1): did */\r
- _dir_entry_get, /*PARAMS(2): did, exp */\r
-\r
- _multi_fcreate, /*PARAMS(3): path prefix cnt */\r
- _multi_fwrite, /*PARAMS(4): path prefix cnt size */\r
- _multi_fread, /*PARAMS(4): path prefix cnt size */\r
- _multi_fremove, /*PARAMS(2): path prefix cnt */\r
- _multi_dcreate, /*PARAMS(3): path prefix cnt */\r
- _multi_dremove, /*PARAMS(2): path prefix */\r
- _stats_save, /*PARAMS(1): path */\r
- _stats_check, /*PARAMS(1): path */\r
-};\r
-\r
-static clock_t get_ms(void)\r
-{\r
- struct timeval t;\r
- gettimeofday(&t, NULL);\r
- return (t.tv_sec * 1000) + (t.tv_usec / 1000);\r
-}\r
-\r
-/**@brief */\r
-static int exec_op_code(char *opcode)\r
-{\r
- int i;\r
- int r = -1;\r
-\r
- for (i = 0; i < sizeof(op_codes) / sizeof(op_codes[0]); ++i) {\r
-\r
- if(strncmp(op_codes[i].func, opcode, strlen(op_codes[i].func)))\r
- continue;\r
-\r
- if(opcode[strlen(op_codes[i].func)] != ' ')\r
- continue;\r
-\r
- printf("%s\n", opcode);\r
- opcode += strlen(op_codes[i].func);\r
- /*Call*/\r
-\r
- clock_t t = get_ms();\r
- r = op_call[i].lwext4_call(opcode);\r
-\r
- printf("rc: %d, time: %ums\n", r, (unsigned int)(get_ms() - t));\r
-\r
- break;\r
- }\r
-\r
- return r;\r
-}\r
-\r
-\r
-static int server_open(void)\r
-{\r
- int fd = 0;\r
- struct sockaddr_in serv_addr;\r
-\r
- memset(&serv_addr, 0, sizeof(serv_addr));\r
-\r
- if(winsock_init() < 0) {\r
- printf("winsock_init() error\n");\r
- exit(-1);\r
- }\r
-\r
- fd = socket(AF_INET, SOCK_STREAM, 0);\r
- if(fd < 0) {\r
- printf("socket() error: %s\n", strerror(errno));\r
- exit(-1);\r
- }\r
-\r
- int yes = 1;\r
- if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (void *)&yes, sizeof(int))) {\r
- printf("setsockopt() error: %s\n", strerror(errno));\r
- exit(-1);\r
- }\r
-\r
- serv_addr.sin_family = AF_INET;\r
- serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);\r
- serv_addr.sin_port = htons(connection_port);\r
-\r
- if(bind(fd, (struct sockaddr*)&serv_addr, sizeof(serv_addr))){\r
- printf("bind() error: %s\n", strerror(errno));\r
- exit(-1);\r
- }\r
-\r
- if(listen(fd, 1)){\r
- printf("listen() error: %s\n", strerror(errno));\r
- exit(-1);\r
- }\r
-\r
- return fd;\r
-}\r
-\r
-static bool parse_opt(int argc, char **argv)\r
-{\r
- int option_index = 0;\r
- int c;\r
-\r
- static struct option long_options[] = {\r
- {"image", required_argument, 0, 'i'},\r
- {"port", required_argument, 0, 'p'},\r
- {"verbose", required_argument, 0, 'v'},\r
- {"winpart", required_argument, 0, 'w'},\r
- {"cache_wb",required_argument, 0, 'c'},\r
- {0, 0, 0, 0}\r
- };\r
-\r
- while(-1 != (c = getopt_long (argc, argv, "c:i:p:v:w:", long_options, &option_index))) {\r
-\r
- switch(c){\r
- case 'i':\r
- ext4_fname = optarg;\r
- break;\r
- case 'p':\r
- connection_port = atoi(optarg);\r
- break;\r
- case 'v':\r
- verbose = atoi(optarg);\r
- break;\r
- case 'c':\r
- cache_wb = atoi(optarg);\r
- break;\r
- case 'w':\r
- winpart = atoi(optarg);\r
- break;\r
- default:\r
- printf("%s", usage);\r
- return false;\r
-\r
- }\r
- }\r
- return true;\r
-}\r
-\r
-int main(int argc, char *argv[])\r
-{\r
- int n;\r
- int listenfd;\r
- int connfd;\r
- char op_code[128];\r
-\r
- if(!parse_opt(argc, argv))\r
- return -1;\r
-\r
- listenfd = server_open();\r
-\r
- printf("lwext4_server: listening on port: %d\n", connection_port);\r
-\r
- memset(write_buffer, RW_BUFFER_PATERN, MAX_RW_BUFFER);\r
- while(1)\r
- {\r
- connfd = accept(listenfd, (struct sockaddr*)NULL, NULL);\r
-\r
- n = recv(connfd, op_code, sizeof(op_code), 0);\r
-\r
- if(n < 0) {\r
- printf("recv() error: %s fd = %d\n", strerror(errno), connfd);\r
- break;\r
- }\r
-\r
- op_code[n] = 0;\r
-\r
- int r = exec_op_code(op_code);\r
-\r
- n = send(connfd, (void *)&r, sizeof(r), 0);\r
- if(n < 0) {\r
- printf("send() error: %s fd = %d\n", strerror(errno), connfd);\r
- break;\r
- }\r
-\r
- close(connfd);\r
- }\r
-\r
- winsock_fini();\r
- return 0;\r
-}\r
-\r
-\r
-int _device_register(char *p)\r
-{\r
- int dev;\r
- int cache_mode;\r
- char dev_name[32];\r
-\r
- if(sscanf(p, "%d %d %s", &dev, &cache_mode, dev_name) != 3){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
-#ifdef WIN32\r
- if(winpart){\r
- ext4_io_raw_filename(ext4_fname);\r
- bd = ext4_io_raw_dev_get();\r
-\r
- }\r
- else\r
-#endif\r
- {\r
- ext4_filedev_filename(ext4_fname);\r
- bd = ext4_filedev_get();\r
- }\r
- return ext4_device_register(bd, 0, dev_name);\r
-}\r
-\r
-int _mount(char *p)\r
-{\r
- char dev_name[32];\r
- char mount_point[32];\r
- int rc;\r
-\r
- if(sscanf(p, "%s %s", dev_name, mount_point) != 2){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- rc = ext4_mount(dev_name, mount_point);\r
- if(cache_wb)\r
- ext4_cache_write_back(mount_point, 1);\r
- return rc;\r
-}\r
-\r
-int _umount(char *p)\r
-{\r
- char mount_point[32];\r
-\r
- if(sscanf(p, "%s", mount_point) != 1){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- if(cache_wb)\r
- ext4_cache_write_back(mount_point, 0);\r
-\r
- return ext4_umount(mount_point);\r
-}\r
-\r
-int _mount_point_stats(char *p)\r
-{\r
- char mount_point[32];\r
- int d;\r
- int rc;\r
- struct ext4_mount_stats stats;\r
-\r
- if(sscanf(p, "%s %d", mount_point, &d) != 2){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- rc = ext4_mount_point_stats(mount_point, &stats);\r
-\r
- if(rc != EOK)\r
- return;\r
-\r
- if(verbose){\r
- printf("\tinodes_count = %d\n", stats.inodes_count);\r
- printf("\tfree_inodes_count = %d\n", stats.free_inodes_count);\r
- printf("\tblocks_count = %llu\n", stats.blocks_count);\r
- printf("\tfree_blocks_count = %llu\n", stats.free_blocks_count);\r
-\r
- printf("\tblock_size = %d\n", stats.block_size);\r
- printf("\tblock_group_count = %d\n", stats.block_group_count);\r
- printf("\tblocks_per_group = %d\n", stats.blocks_per_group);\r
- printf("\tinodes_per_group = %d\n", stats.inodes_per_group);\r
-\r
- printf("\tvolume_name = %s\n", stats.volume_name);\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-int _cache_write_back(char *p)\r
-{\r
- char mount_point[32];\r
- int en;\r
-\r
- if(sscanf(p, "%s %d", mount_point, &en) != 2){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- return ext4_cache_write_back(mount_point, en);\r
-}\r
-\r
-\r
-int _fremove(char *p)\r
-{\r
- char path[255];\r
-\r
- if(sscanf(p, "%s", path) != 1){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- return ext4_fremove(path);\r
-}\r
-\r
-int _fopen(char *p)\r
-{\r
- int fid = MAX_FILES;\r
- char path[256];\r
- char flags[8];\r
- int rc;\r
-\r
- if(sscanf(p, "%d %s %s", &fid, path, flags) != 3){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- if(!(fid < MAX_FILES)){\r
- printf("File id too big\n");\r
- return -1;\r
- }\r
-\r
- rc = ext4_fopen(&file_tab[fid].fd, path, flags);\r
-\r
- if(rc == EOK)\r
- strcpy(file_tab[fid].name, path);\r
-\r
- return rc;\r
-}\r
-\r
-int _fclose(char *p)\r
-{\r
- int fid = MAX_FILES;\r
- int rc;\r
-\r
- if(sscanf(p, "%d", &fid) != 1){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- if(!(fid < MAX_FILES)){\r
- printf("File id too big\n");\r
- return -1;\r
- }\r
-\r
- if(file_tab[fid].name[0] == 0){\r
- printf("File id empty\n");\r
- return -1;\r
- }\r
-\r
- rc = ext4_fclose(&file_tab[fid].fd);\r
-\r
- if(rc == EOK)\r
- file_tab[fid].name[0] = 0;\r
-\r
- return rc;\r
-}\r
-\r
-int _fread(char *p)\r
-{\r
- int fid = MAX_FILES;\r
- int len;\r
- int d;\r
- int rc;\r
- int rb;\r
-\r
- if(sscanf(p, "%d %d %d %d", &fid, &d, &len, &d) != 4){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- if(!(fid < MAX_FILES)){\r
- printf("File id too big\n");\r
- return -1;\r
- }\r
-\r
- if(file_tab[fid].name[0] == 0){\r
- printf("File id empty\n");\r
- return -1;\r
- }\r
-\r
- while(len){\r
- d = len > MAX_RW_BUFFER ? MAX_RW_BUFFER : len;\r
-\r
- memset(read_buffer, 0, MAX_RW_BUFFER);\r
- rc = ext4_fread(&file_tab[fid].fd, read_buffer, d, &rb);\r
-\r
- if(rc != EOK)\r
- break;\r
-\r
- if(rb != d){\r
- printf("Read count error\n");\r
- return -1;\r
- }\r
-\r
- if(memcmp(read_buffer, write_buffer, d)){\r
- printf("Read compare error\n");\r
- return -1;\r
- }\r
-\r
- len -= d;\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-int _fwrite(char *p)\r
-{\r
- int fid = MAX_FILES;\r
- int len;\r
- int d;\r
- int rc;\r
- int wb;\r
-\r
- if(sscanf(p, "%d %d %d %d", &fid, &d, &len, &d) != 4){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- if(!(fid < MAX_FILES)){\r
- printf("File id too big\n");\r
- return -1;\r
- }\r
-\r
- if(file_tab[fid].name[0] == 0){\r
- printf("File id empty\n");\r
- return -1;\r
- }\r
-\r
- while(len){\r
- d = len > MAX_RW_BUFFER ? MAX_RW_BUFFER : len;\r
- rc = ext4_fwrite(&file_tab[fid].fd, write_buffer, d, &wb);\r
-\r
- if(rc != EOK)\r
- break;\r
-\r
- if(wb != d){\r
- printf("Write count error\n");\r
- return -1;\r
- }\r
-\r
- len -= d;\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-int _fseek(char *p)\r
-{\r
- int fid = MAX_FILES;\r
- int off;\r
- int origin;\r
-\r
- if(sscanf(p, "%d %d %d", &fid, &off, &origin) != 3){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- if(!(fid < MAX_FILES)){\r
- printf("File id too big\n");\r
- return -1;\r
- }\r
-\r
- if(file_tab[fid].name[0] == 0){\r
- printf("File id empty\n");\r
- return -1;\r
- }\r
-\r
- return ext4_fseek(&file_tab[fid].fd, off, origin);\r
-}\r
-\r
-int _ftell(char *p)\r
-{\r
- int fid = MAX_FILES;\r
- uint32_t exp_pos;\r
-\r
- if(sscanf(p, "%d %u", &fid, &exp_pos) != 2){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- if(!(fid < MAX_FILES)){\r
- printf("File id too big\n");\r
- return -1;\r
- }\r
-\r
- if(file_tab[fid].name[0] == 0){\r
- printf("File id empty\n");\r
- return -1;\r
- }\r
-\r
-\r
- if(exp_pos != ext4_ftell(&file_tab[fid].fd)){\r
- printf("Expected filepos error\n");\r
- return -1;\r
- }\r
-\r
- return EOK;\r
-}\r
-\r
-int _fsize(char *p)\r
-{\r
- int fid = MAX_FILES;\r
- uint32_t exp_size;\r
-\r
- if(sscanf(p, "%d %u", &fid, &exp_size) != 2){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- if(!(fid < MAX_FILES)){\r
- printf("File id too big\n");\r
- return -1;\r
- }\r
-\r
- if(file_tab[fid].name[0] == 0){\r
- printf("File id empty\n");\r
- return -1;\r
- }\r
-\r
- if(exp_size != ext4_fsize(&file_tab[fid].fd)){\r
- printf("Expected filesize error\n");\r
- return -1;\r
- }\r
-\r
- return EOK;\r
-}\r
-\r
-int _dir_rm(char *p)\r
-{\r
- char path[255];\r
-\r
- if(sscanf(p, "%s", path) != 1){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- return ext4_dir_rm(path);\r
-}\r
-\r
-int _dir_mk(char *p)\r
-{\r
- char path[255];\r
-\r
- if(sscanf(p, "%s", path) != 1){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- return ext4_dir_mk(path);\r
-}\r
-\r
-int _dir_open(char *p)\r
-{\r
- int did = MAX_DIRS;\r
- char path[255];\r
- int rc;\r
-\r
- if(sscanf(p, "%d %s", &did, path) != 2){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- if(!(did < MAX_DIRS)){\r
- printf("Dir id too big\n");\r
- return -1;\r
- }\r
-\r
- rc = ext4_dir_open(&dir_tab[did].fd, path);\r
-\r
- if(rc == EOK)\r
- strcpy(dir_tab[did].name, path);\r
-\r
- return rc;\r
-}\r
-\r
-int _dir_close(char *p)\r
-{\r
- int did = MAX_DIRS;\r
- int rc;\r
-\r
- if(sscanf(p, "%d", &did) != 1){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- if(!(did < MAX_DIRS)){\r
- printf("Dir id too big\n");\r
- return -1;\r
- }\r
-\r
- if(dir_tab[did].name[0] == 0){\r
- printf("Dir id empty\n");\r
- return -1;\r
- }\r
-\r
- rc = ext4_dir_close(&dir_tab[did].fd);\r
-\r
- if(rc == EOK)\r
- dir_tab[did].name[0] = 0;\r
-\r
- return rc;\r
-}\r
-\r
-int _dir_entry_get(char *p)\r
-{\r
- int did = MAX_DIRS;\r
- int exp;\r
- char name[256];\r
-\r
- if(sscanf(p, "%d %d", &did, &exp) != 2){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- if(!(did < MAX_DIRS)){\r
- printf("Dir id too big\n");\r
- return -1;\r
- }\r
-\r
- if(dir_tab[did].name[0] == 0){\r
- printf("Dir id empty\n");\r
- return -1;\r
- }\r
-\r
-\r
- int idx = 0;\r
- ext4_direntry *d;\r
-\r
- while(d = ext4_dir_entry_get(&dir_tab[did].fd, idx++)){\r
-\r
- memcpy(name, d->name, d->name_length);\r
- name[d->name_length] = 0;\r
- if(verbose){\r
- printf("\t%s %s\n", entry_to_str(d->inode_type), name);\r
- }\r
- }\r
-\r
- idx--;\r
-\r
- if(idx < 2){\r
- printf("Minumum dir entry error\n");\r
- return -1;\r
- }\r
-\r
- if((idx - 2) != exp){\r
- printf("Expected dir entry error\n");\r
- return -1;\r
- }\r
-\r
- return EOK;\r
-}\r
-\r
-int _multi_fcreate(char *p)\r
-{\r
- char path[256];\r
- char path1[256];\r
- char prefix[32];\r
- int cnt;\r
- int rc;\r
- int i;\r
- ext4_file fd;\r
-\r
- if(sscanf(p, "%s %s %d", path, prefix, &cnt) != 3){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- for (i = 0; i < cnt; ++i) {\r
- sprintf(path1, "%s%s%d", path, prefix, i);\r
- rc = ext4_fopen(&fd, path1, "wb+");\r
-\r
- if(rc != EOK)\r
- break;\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-int _multi_fwrite(char *p)\r
-{\r
- char path[256];\r
- char path1[256];\r
- char prefix[32];\r
- int cnt;\r
- int len, ll;\r
- int rc;\r
- int i, d, wb;\r
- ext4_file fd;\r
-\r
- if(sscanf(p, "%s %s %d %d", path, prefix, &cnt, &ll) != 4){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- for (i = 0; i < cnt; ++i) {\r
- sprintf(path1, "%s%s%d", path, prefix, i);\r
- rc = ext4_fopen(&fd, path1, "rb+");\r
-\r
- if(rc != EOK)\r
- break;\r
-\r
- len = ll;\r
- while(len){\r
- d = len > MAX_RW_BUFFER ? MAX_RW_BUFFER : len;\r
- rc = ext4_fwrite(&fd, write_buffer, d, &wb);\r
-\r
- if(rc != EOK)\r
- break;\r
-\r
- if(wb != d){\r
- printf("Write count error\n");\r
- return -1;\r
- }\r
-\r
- len -= d;\r
- }\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-int _multi_fread(char *p)\r
-{\r
- char path[256];\r
- char path1[256];\r
- char prefix[32];\r
- int cnt;\r
- int len, ll;\r
- int rc;\r
- int i, d, rb;\r
- ext4_file fd;\r
-\r
- if(sscanf(p, "%s %s %d %d", path, prefix, &cnt, &ll) != 4){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- for (i = 0; i < cnt; ++i) {\r
- sprintf(path1, "%s%s%d", path, prefix, i);\r
- rc = ext4_fopen(&fd, path1, "rb+");\r
-\r
- if(rc != EOK)\r
- break;\r
-\r
- len = ll;\r
- while(len){\r
- d = len > MAX_RW_BUFFER ? MAX_RW_BUFFER : len;\r
-\r
- memset(read_buffer, 0, MAX_RW_BUFFER);\r
- rc = ext4_fread(&fd, read_buffer, d, &rb);\r
-\r
- if(rc != EOK)\r
- break;\r
-\r
- if(rb != d){\r
- printf("Read count error\n");\r
- return -1;\r
- }\r
-\r
- if(memcmp(read_buffer, write_buffer, d)){\r
- printf("Read compare error\n");\r
- return -1;\r
- }\r
-\r
- len -= d;\r
- }\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-int _multi_fremove(char *p)\r
-{\r
- char path[256];\r
- char path1[256];\r
- char prefix[32];\r
- int cnt, i, rc;\r
-\r
- if(sscanf(p, "%s %s %d", path, prefix, &cnt) != 3){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- for (i = 0; i < cnt; ++i) {\r
- sprintf(path1, "%s%s%d", path, prefix, i);\r
- rc = ext4_fremove(path1);\r
- if(rc != EOK)\r
- break;\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-int _multi_dcreate(char *p)\r
-{\r
- char path[256];\r
- char path1[256];\r
- char prefix[32];\r
- int cnt, i, rc;\r
-\r
- if(sscanf(p, "%s %s %d", path, prefix, &cnt) != 3){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- for (i = 0; i < cnt; ++i) {\r
- sprintf(path1, "%s%s%d", path, prefix, i);\r
- rc = ext4_dir_mk(path1);\r
- if(rc != EOK)\r
- break;\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-int _multi_dremove(char *p)\r
-{\r
- char path[256];\r
- char path1[256];\r
- char prefix[32];\r
- int cnt, i, rc;\r
-\r
- if(sscanf(p, "%s %s %d", path, prefix, &cnt) != 3){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- for (i = 0; i < cnt; ++i) {\r
- sprintf(path1, "%s%s%d", path, prefix, i);\r
- rc = ext4_dir_rm(path1);\r
- if(rc != EOK)\r
- break;\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-struct ext4_mount_stats saved_stats;\r
-\r
-int _stats_save(char *p)\r
-{\r
- char path[256];\r
-\r
- if(sscanf(p, "%s", path) != 1){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- return ext4_mount_point_stats(path, &saved_stats);\r
-}\r
-\r
-int _stats_check(char *p)\r
-{\r
- char path[256];\r
- int rc;\r
-\r
- struct ext4_mount_stats actual_stats;\r
-\r
- if(sscanf(p, "%s", path) != 1){\r
- printf("Param list error\n");\r
- return -1;\r
- }\r
-\r
- rc = ext4_mount_point_stats(path, &actual_stats);\r
-\r
- if(rc != EOK)\r
- return rc;\r
-\r
- if(memcmp(&saved_stats, &actual_stats, sizeof(struct ext4_mount_stats))){\r
- if(verbose){\r
- printf("\tMount point stats error:\n");\r
- printf("\tsaved_stats:\n");\r
- printf("\tinodes_count = %d\n", saved_stats.inodes_count);\r
- printf("\tfree_inodes_count = %d\n", saved_stats.free_inodes_count);\r
- printf("\tblocks_count = %llu\n", saved_stats.blocks_count);\r
- printf("\tfree_blocks_count = %llu\n", saved_stats.free_blocks_count);\r
- printf("\tblock_size = %d\n", saved_stats.block_size);\r
- printf("\tblock_group_count = %d\n", saved_stats.block_group_count);\r
- printf("\tblocks_per_group = %d\n", saved_stats.blocks_per_group);\r
- printf("\tinodes_per_group = %d\n", saved_stats.inodes_per_group);\r
- printf("\tvolume_name = %s\n", saved_stats.volume_name);\r
- printf("\tactual_stats:\n");\r
- printf("\tinodes_count = %d\n", actual_stats.inodes_count);\r
- printf("\tfree_inodes_count = %d\n", actual_stats.free_inodes_count);\r
- printf("\tblocks_count = %llu\n", actual_stats.blocks_count);\r
- printf("\tfree_blocks_count = %llu\n", actual_stats.free_blocks_count);\r
- printf("\tblock_size = %d\n", actual_stats.block_size);\r
- printf("\tblock_group_count = %d\n", actual_stats.block_group_count);\r
- printf("\tblocks_per_group = %d\n", actual_stats.blocks_per_group);\r
- printf("\tinodes_per_group = %d\n", actual_stats.inodes_per_group);\r
- printf("\tvolume_name = %s\n", actual_stats.volume_name);\r
- }\r
- return -1;\r
- }\r
-\r
-\r
- return rc;\r
-}\r
-\r
-\r
-static char* entry_to_str(uint8_t type)\r
-{\r
- switch(type){\r
- case EXT4_DIRENTRY_UNKNOWN:\r
- return "[UNK] ";\r
- case EXT4_DIRENTRY_REG_FILE:\r
- return "[FIL] ";\r
- case EXT4_DIRENTRY_DIR:\r
- return "[DIR] ";\r
- case EXT4_DIRENTRY_CHRDEV:\r
- return "[CHA] ";\r
- case EXT4_DIRENTRY_BLKDEV:\r
- return "[BLK] ";\r
- case EXT4_DIRENTRY_FIFO:\r
- return "[FIF] ";\r
- case EXT4_DIRENTRY_SOCK:\r
- return "[SOC] ";\r
- case EXT4_DIRENTRY_SYMLINK:\r
- return "[SYM] ";\r
- default:\r
- break;\r
- }\r
- return "[???]";\r
-}\r
-\r
-static int winsock_init(void)\r
-{\r
-#if WIN32\r
- int rc;\r
- static WSADATA wsaData;\r
- rc = WSAStartup(MAKEWORD(2,2), &wsaData);\r
- if (rc != 0) {\r
- return -1;\r
- }\r
-#endif\r
- return 0;\r
-}\r
-\r
-static void winsock_fini(void)\r
-{\r
-#if WIN32\r
- WSACleanup();\r
-#endif\r
-}\r
-\r
+/*
+ * Copyright (c) 2014 Grzegorz Kostka (kostka.grzegorz@gmail.com)
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * - Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * - Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ * - The name of the author may not be used to endorse or promote products
+ * derived from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include <stdint.h>
+#include <stdbool.h>
+#include <getopt.h>
+#include <time.h>
+#include <sys/time.h>
+
+#ifdef WIN32
+#include <winsock2.h>
+#include <ws2tcpip.h>
+#include <windows.h>
+#else
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <sys/types.h>
+#endif
+
+#include "../blockdev/linux/ext4_filedev.h"
+#include "../blockdev/windows/io_raw.h"
+
+#include <ext4.h>
+#include <errno.h>
+
+static int winsock_init(void);
+static void winsock_fini(void);
+static char *entry_to_str(uint8_t type);
+
+#define MAX_FILES 64
+#define MAX_DIRS 64
+
+#define MAX_RW_BUFFER (1024 * 1024)
+#define RW_BUFFER_PATERN ('x')
+
+/**@brief Default connection port*/
+static int connection_port = 1234;
+
+/**@brief Default filesystem filename.*/
+static char *ext4_fname = "ext2";
+
+/**@brief Verbose mode*/
+static int verbose = 0;
+
+/**@brief Winpart mode*/
+static int winpart = 0;
+
+/**@brief Blockdev handle*/
+static struct ext4_blockdev *bd;
+
+static int cache_wb = 0;
+
+static char read_buffer[MAX_RW_BUFFER];
+static char write_buffer[MAX_RW_BUFFER];
+
+static const char *usage = " \n\
+Welcome in lwext4_server. \n\
+Copyright (c) 2013 Grzegorz Kostka (kostka.grzegorz@gmail.com) \n\
+Usage: \n\
+ --image (-i) - ext2/3/4 image file \n\
+ --port (-p) - server port \n\
+ --verbose (-v) - verbose mode \n\
+ --winpart (-w) - windows_partition mode \n\
+ --cache_wb (-c) - cache writeback_mode \n\
+\n";
+
+/**@brief Open file instance descriptor.*/
+struct lwext4_files {
+ char name[255];
+ ext4_file fd;
+};
+
+/**@brief Open directory instance descriptor.*/
+struct lwext4_dirs {
+ char name[255];
+ ext4_dir fd;
+};
+
+/**@brief Library call opcode.*/
+struct lwext4_op_codes {
+ char *func;
+};
+
+/**@brief Library call wraper.*/
+struct lwext4_call {
+ int (*lwext4_call)(char *p);
+};
+
+/**@brief */
+static struct lwext4_files file_tab[MAX_FILES];
+
+/**@brief */
+static struct lwext4_dirs dir_tab[MAX_DIRS];
+
+/**@brief */
+static struct lwext4_op_codes op_codes[] = {
+ "device_register",
+ "mount",
+ "umount",
+ "mount_point_stats",
+ "cache_write_back",
+ "fremove",
+ "fopen",
+ "fclose",
+ "fread",
+ "fwrite",
+ "fseek",
+ "ftell",
+ "fsize",
+ "dir_rm",
+ "dir_mk",
+ "dir_open",
+ "dir_close",
+ "dir_entry_get",
+
+ "multi_fcreate",
+ "multi_fwrite",
+ "multi_fread",
+ "multi_fremove",
+ "multi_dcreate",
+ "multi_dremove",
+ "stats_save",
+ "stats_check",
+};
+
+int _device_register(char *p);
+int _mount(char *p);
+int _umount(char *p);
+int _mount_point_stats(char *p);
+int _cache_write_back(char *p);
+int _fremove(char *p);
+int _fopen(char *p);
+int _fclose(char *p);
+int _fread(char *p);
+int _fwrite(char *p);
+int _fseek(char *p);
+int _ftell(char *p);
+int _fsize(char *p);
+int _dir_rm(char *p);
+int _dir_mk(char *p);
+int _dir_open(char *p);
+int _dir_close(char *p);
+int _dir_close(char *p);
+int _dir_entry_get(char *p);
+
+int _multi_fcreate(char *p);
+int _multi_fwrite(char *p);
+int _multi_fread(char *p);
+int _multi_fremove(char *p);
+int _multi_dcreate(char *p);
+int _multi_dremove(char *p);
+int _stats_save(char *p);
+int _stats_check(char *p);
+
+/**@brief */
+static struct lwext4_call op_call[] = {
+ _device_register, /*PARAMS(3): 0 cache_mode dev_name */
+ _mount, /*PARAMS(2): dev_name mount_point */
+ _umount, /*PARAMS(1): mount_point */
+ _mount_point_stats, /*PARAMS(2): mount_point, 0 */
+ _cache_write_back, /*PARAMS(2): mount_point, en */
+ _fremove, /*PARAMS(1): path */
+ _fopen, /*PARAMS(2): fid path flags */
+ _fclose, /*PARAMS(1): fid */
+ _fread, /*PARAMS(4): fid 0 len 0 */
+ _fwrite, /*PARAMS(4): fid 0 len 0 */
+ _fseek, /*PARAMS(2): fid off origin */
+ _ftell, /*PARAMS(2): fid exp */
+ _fsize, /*PARAMS(2): fid exp */
+ _dir_rm, /*PARAMS(1): path */
+ _dir_mk, /*PARAMS(1): path */
+ _dir_open, /*PARAMS(2): did, path */
+ _dir_close, /*PARAMS(1): did */
+ _dir_entry_get, /*PARAMS(2): did, exp */
+
+ _multi_fcreate, /*PARAMS(3): path prefix cnt */
+ _multi_fwrite, /*PARAMS(4): path prefix cnt size */
+ _multi_fread, /*PARAMS(4): path prefix cnt size */
+ _multi_fremove, /*PARAMS(2): path prefix cnt */
+ _multi_dcreate, /*PARAMS(3): path prefix cnt */
+ _multi_dremove, /*PARAMS(2): path prefix */
+ _stats_save, /*PARAMS(1): path */
+ _stats_check, /*PARAMS(1): path */
+};
+
+static clock_t get_ms(void)
+{
+ struct timeval t;
+ gettimeofday(&t, NULL);
+ return (t.tv_sec * 1000) + (t.tv_usec / 1000);
+}
+
+/**@brief */
+static int exec_op_code(char *opcode)
+{
+ int i;
+ int r = -1;
+
+ for (i = 0; i < sizeof(op_codes) / sizeof(op_codes[0]); ++i) {
+
+ if (strncmp(op_codes[i].func, opcode, strlen(op_codes[i].func)))
+ continue;
+
+ if (opcode[strlen(op_codes[i].func)] != ' ')
+ continue;
+
+ printf("%s\n", opcode);
+ opcode += strlen(op_codes[i].func);
+ /*Call*/
+
+ clock_t t = get_ms();
+ r = op_call[i].lwext4_call(opcode);
+
+ printf("rc: %d, time: %ums\n", r, (unsigned int)(get_ms() - t));
+
+ break;
+ }
+
+ return r;
+}
+
+static int server_open(void)
+{
+ int fd = 0;
+ struct sockaddr_in serv_addr;
+
+ memset(&serv_addr, 0, sizeof(serv_addr));
+
+ if (winsock_init() < 0) {
+ printf("winsock_init() error\n");
+ exit(-1);
+ }
+
+ fd = socket(AF_INET, SOCK_STREAM, 0);
+ if (fd < 0) {
+ printf("socket() error: %s\n", strerror(errno));
+ exit(-1);
+ }
+
+ int yes = 1;
+ if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (void *)&yes,
+ sizeof(int))) {
+ printf("setsockopt() error: %s\n", strerror(errno));
+ exit(-1);
+ }
+
+ serv_addr.sin_family = AF_INET;
+ serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
+ serv_addr.sin_port = htons(connection_port);
+
+ if (bind(fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr))) {
+ printf("bind() error: %s\n", strerror(errno));
+ exit(-1);
+ }
+
+ if (listen(fd, 1)) {
+ printf("listen() error: %s\n", strerror(errno));
+ exit(-1);
+ }
+
+ return fd;
+}
+
+static bool parse_opt(int argc, char **argv)
+{
+ int option_index = 0;
+ int c;
+
+ static struct option long_options[] = {
+ {"image", required_argument, 0, 'i'},
+ {"port", required_argument, 0, 'p'},
+ {"verbose", required_argument, 0, 'v'},
+ {"winpart", required_argument, 0, 'w'},
+ {"cache_wb", required_argument, 0, 'c'},
+ {0, 0, 0, 0}};
+
+ while (-1 != (c = getopt_long(argc, argv, "c:i:p:v:w:", long_options,
+ &option_index))) {
+
+ switch (c) {
+ case 'i':
+ ext4_fname = optarg;
+ break;
+ case 'p':
+ connection_port = atoi(optarg);
+ break;
+ case 'v':
+ verbose = atoi(optarg);
+ break;
+ case 'c':
+ cache_wb = atoi(optarg);
+ break;
+ case 'w':
+ winpart = atoi(optarg);
+ break;
+ default:
+ printf("%s", usage);
+ return false;
+ }
+ }
+ return true;
+}
+
+int main(int argc, char *argv[])
+{
+ int n;
+ int listenfd;
+ int connfd;
+ char op_code[128];
+
+ if (!parse_opt(argc, argv))
+ return -1;
+
+ listenfd = server_open();
+
+ printf("lwext4_server: listening on port: %d\n", connection_port);
+
+ memset(write_buffer, RW_BUFFER_PATERN, MAX_RW_BUFFER);
+ while (1) {
+ connfd = accept(listenfd, (struct sockaddr *)NULL, NULL);
+
+ n = recv(connfd, op_code, sizeof(op_code), 0);
+
+ if (n < 0) {
+ printf("recv() error: %s fd = %d\n", strerror(errno),
+ connfd);
+ break;
+ }
+
+ op_code[n] = 0;
+
+ int r = exec_op_code(op_code);
+
+ n = send(connfd, (void *)&r, sizeof(r), 0);
+ if (n < 0) {
+ printf("send() error: %s fd = %d\n", strerror(errno),
+ connfd);
+ break;
+ }
+
+ close(connfd);
+ }
+
+ winsock_fini();
+ return 0;
+}
+
+int _device_register(char *p)
+{
+ int dev;
+ int cache_mode;
+ char dev_name[32];
+
+ if (sscanf(p, "%d %d %s", &dev, &cache_mode, dev_name) != 3) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+#ifdef WIN32
+ if (winpart) {
+ ext4_io_raw_filename(ext4_fname);
+ bd = ext4_io_raw_dev_get();
+
+ } else
+#endif
+ {
+ ext4_filedev_filename(ext4_fname);
+ bd = ext4_filedev_get();
+ }
+ return ext4_device_register(bd, 0, dev_name);
+}
+
+int _mount(char *p)
+{
+ char dev_name[32];
+ char mount_point[32];
+ int rc;
+
+ if (sscanf(p, "%s %s", dev_name, mount_point) != 2) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ rc = ext4_mount(dev_name, mount_point);
+ if (cache_wb)
+ ext4_cache_write_back(mount_point, 1);
+ return rc;
+}
+
+int _umount(char *p)
+{
+ char mount_point[32];
+
+ if (sscanf(p, "%s", mount_point) != 1) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ if (cache_wb)
+ ext4_cache_write_back(mount_point, 0);
+
+ return ext4_umount(mount_point);
+}
+
+int _mount_point_stats(char *p)
+{
+ char mount_point[32];
+ int d;
+ int rc;
+ struct ext4_mount_stats stats;
+
+ if (sscanf(p, "%s %d", mount_point, &d) != 2) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ rc = ext4_mount_point_stats(mount_point, &stats);
+
+ if (rc != EOK)
+ return rc;
+
+ if (verbose) {
+ printf("\tinodes_count = %d\n", stats.inodes_count);
+ printf("\tfree_inodes_count = %d\n", stats.free_inodes_count);
+ printf("\tblocks_count = %llu\n", stats.blocks_count);
+ printf("\tfree_blocks_count = %llu\n", stats.free_blocks_count);
+
+ printf("\tblock_size = %d\n", stats.block_size);
+ printf("\tblock_group_count = %d\n", stats.block_group_count);
+ printf("\tblocks_per_group = %d\n", stats.blocks_per_group);
+ printf("\tinodes_per_group = %d\n", stats.inodes_per_group);
+
+ printf("\tvolume_name = %s\n", stats.volume_name);
+ }
+
+ return rc;
+}
+
+int _cache_write_back(char *p)
+{
+ char mount_point[32];
+ int en;
+
+ if (sscanf(p, "%s %d", mount_point, &en) != 2) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ return ext4_cache_write_back(mount_point, en);
+}
+
+int _fremove(char *p)
+{
+ char path[255];
+
+ if (sscanf(p, "%s", path) != 1) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ return ext4_fremove(path);
+}
+
+int _fopen(char *p)
+{
+ int fid = MAX_FILES;
+ char path[256];
+ char flags[8];
+ int rc;
+
+ if (sscanf(p, "%d %s %s", &fid, path, flags) != 3) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ if (!(fid < MAX_FILES)) {
+ printf("File id too big\n");
+ return -1;
+ }
+
+ rc = ext4_fopen(&file_tab[fid].fd, path, flags);
+
+ if (rc == EOK)
+ strcpy(file_tab[fid].name, path);
+
+ return rc;
+}
+
+int _fclose(char *p)
+{
+ int fid = MAX_FILES;
+ int rc;
+
+ if (sscanf(p, "%d", &fid) != 1) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ if (!(fid < MAX_FILES)) {
+ printf("File id too big\n");
+ return -1;
+ }
+
+ if (file_tab[fid].name[0] == 0) {
+ printf("File id empty\n");
+ return -1;
+ }
+
+ rc = ext4_fclose(&file_tab[fid].fd);
+
+ if (rc == EOK)
+ file_tab[fid].name[0] = 0;
+
+ return rc;
+}
+
+int _fread(char *p)
+{
+ int fid = MAX_FILES;
+ int len;
+ int d;
+ int rc;
+ size_t rb;
+
+ if (sscanf(p, "%d %d %d %d", &fid, &d, &len, &d) != 4) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ if (!(fid < MAX_FILES)) {
+ printf("File id too big\n");
+ return -1;
+ }
+
+ if (file_tab[fid].name[0] == 0) {
+ printf("File id empty\n");
+ return -1;
+ }
+
+ while (len) {
+ d = len > MAX_RW_BUFFER ? MAX_RW_BUFFER : len;
+
+ memset(read_buffer, 0, MAX_RW_BUFFER);
+ rc = ext4_fread(&file_tab[fid].fd, read_buffer, d, &rb);
+
+ if (rc != EOK)
+ break;
+
+ if (rb != d) {
+ printf("Read count error\n");
+ return -1;
+ }
+
+ if (memcmp(read_buffer, write_buffer, d)) {
+ printf("Read compare error\n");
+ return -1;
+ }
+
+ len -= d;
+ }
+
+ return rc;
+}
+
+int _fwrite(char *p)
+{
+ int fid = MAX_FILES;
+ int d;
+ int rc;
+
+ size_t len;
+ size_t wb;
+
+ if (sscanf(p, "%d %d %d %d", &fid, &d, &len, &d) != 4) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ if (!(fid < MAX_FILES)) {
+ printf("File id too big\n");
+ return -1;
+ }
+
+ if (file_tab[fid].name[0] == 0) {
+ printf("File id empty\n");
+ return -1;
+ }
+
+ while (len) {
+ d = len > MAX_RW_BUFFER ? MAX_RW_BUFFER : len;
+ rc = ext4_fwrite(&file_tab[fid].fd, write_buffer, d, &wb);
+
+ if (rc != EOK)
+ break;
+
+ if (wb != d) {
+ printf("Write count error\n");
+ return -1;
+ }
+
+ len -= d;
+ }
+
+ return rc;
+}
+
+int _fseek(char *p)
+{
+ int fid = MAX_FILES;
+ int off;
+ int origin;
+
+ if (sscanf(p, "%d %d %d", &fid, &off, &origin) != 3) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ if (!(fid < MAX_FILES)) {
+ printf("File id too big\n");
+ return -1;
+ }
+
+ if (file_tab[fid].name[0] == 0) {
+ printf("File id empty\n");
+ return -1;
+ }
+
+ return ext4_fseek(&file_tab[fid].fd, off, origin);
+}
+
+int _ftell(char *p)
+{
+ int fid = MAX_FILES;
+ uint32_t exp_pos;
+
+ if (sscanf(p, "%d %u", &fid, &exp_pos) != 2) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ if (!(fid < MAX_FILES)) {
+ printf("File id too big\n");
+ return -1;
+ }
+
+ if (file_tab[fid].name[0] == 0) {
+ printf("File id empty\n");
+ return -1;
+ }
+
+ if (exp_pos != ext4_ftell(&file_tab[fid].fd)) {
+ printf("Expected filepos error\n");
+ return -1;
+ }
+
+ return EOK;
+}
+
+int _fsize(char *p)
+{
+ int fid = MAX_FILES;
+ uint32_t exp_size;
+
+ if (sscanf(p, "%d %u", &fid, &exp_size) != 2) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ if (!(fid < MAX_FILES)) {
+ printf("File id too big\n");
+ return -1;
+ }
+
+ if (file_tab[fid].name[0] == 0) {
+ printf("File id empty\n");
+ return -1;
+ }
+
+ if (exp_size != ext4_fsize(&file_tab[fid].fd)) {
+ printf("Expected filesize error\n");
+ return -1;
+ }
+
+ return EOK;
+}
+
+int _dir_rm(char *p)
+{
+ char path[255];
+
+ if (sscanf(p, "%s", path) != 1) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ return ext4_dir_rm(path);
+}
+
+int _dir_mk(char *p)
+{
+ char path[255];
+
+ if (sscanf(p, "%s", path) != 1) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ return ext4_dir_mk(path);
+}
+
+int _dir_open(char *p)
+{
+ int did = MAX_DIRS;
+ char path[255];
+ int rc;
+
+ if (sscanf(p, "%d %s", &did, path) != 2) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ if (!(did < MAX_DIRS)) {
+ printf("Dir id too big\n");
+ return -1;
+ }
+
+ rc = ext4_dir_open(&dir_tab[did].fd, path);
+
+ if (rc == EOK)
+ strcpy(dir_tab[did].name, path);
+
+ return rc;
+}
+
+int _dir_close(char *p)
+{
+ int did = MAX_DIRS;
+ int rc;
+
+ if (sscanf(p, "%d", &did) != 1) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ if (!(did < MAX_DIRS)) {
+ printf("Dir id too big\n");
+ return -1;
+ }
+
+ if (dir_tab[did].name[0] == 0) {
+ printf("Dir id empty\n");
+ return -1;
+ }
+
+ rc = ext4_dir_close(&dir_tab[did].fd);
+
+ if (rc == EOK)
+ dir_tab[did].name[0] = 0;
+
+ return rc;
+}
+
+int _dir_entry_get(char *p)
+{
+ int did = MAX_DIRS;
+ int exp;
+ char name[256];
+
+ if (sscanf(p, "%d %d", &did, &exp) != 2) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ if (!(did < MAX_DIRS)) {
+ printf("Dir id too big\n");
+ return -1;
+ }
+
+ if (dir_tab[did].name[0] == 0) {
+ printf("Dir id empty\n");
+ return -1;
+ }
+
+ int idx = 0;
+ const ext4_direntry *d;
+
+ while ((d = ext4_dir_entry_next(&dir_tab[did].fd)) != NULL) {
+
+ idx++;
+ memcpy(name, d->name, d->name_length);
+ name[d->name_length] = 0;
+ if (verbose) {
+ printf("\t%s %s\n", entry_to_str(d->inode_type), name);
+ }
+ }
+
+ if (idx < 2) {
+ printf("Minumum dir entry error\n");
+ return -1;
+ }
+
+ if ((idx - 2) != exp) {
+ printf("Expected dir entry error\n");
+ return -1;
+ }
+
+ return EOK;
+}
+
+int _multi_fcreate(char *p)
+{
+ char path[256];
+ char path1[256];
+ char prefix[32];
+ int cnt;
+ int rc;
+ int i;
+ ext4_file fd;
+
+ if (sscanf(p, "%s %s %d", path, prefix, &cnt) != 3) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ for (i = 0; i < cnt; ++i) {
+ sprintf(path1, "%s%s%d", path, prefix, i);
+ rc = ext4_fopen(&fd, path1, "wb+");
+
+ if (rc != EOK)
+ break;
+ }
+
+ return rc;
+}
+
+int _multi_fwrite(char *p)
+{
+ char path[256];
+ char path1[256];
+ char prefix[32];
+ int cnt, i;
+ int len, ll;
+ int rc;
+ size_t d, wb;
+ ext4_file fd;
+
+ if (sscanf(p, "%s %s %d %d", path, prefix, &cnt, &ll) != 4) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ for (i = 0; i < cnt; ++i) {
+ sprintf(path1, "%s%s%d", path, prefix, i);
+ rc = ext4_fopen(&fd, path1, "rb+");
+
+ if (rc != EOK)
+ break;
+
+ len = ll;
+ while (len) {
+ d = len > MAX_RW_BUFFER ? MAX_RW_BUFFER : len;
+ rc = ext4_fwrite(&fd, write_buffer, d, &wb);
+
+ if (rc != EOK)
+ break;
+
+ if (wb != d) {
+ printf("Write count error\n");
+ return -1;
+ }
+
+ len -= d;
+ }
+ }
+
+ return rc;
+}
+
+int _multi_fread(char *p)
+{
+ char path[256];
+ char path1[256];
+ char prefix[32];
+ int cnt;
+ int len, ll;
+ int rc ,i, d;
+ size_t rb;
+ ext4_file fd;
+
+ if (sscanf(p, "%s %s %d %d", path, prefix, &cnt, &ll) != 4) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ for (i = 0; i < cnt; ++i) {
+ sprintf(path1, "%s%s%d", path, prefix, i);
+ rc = ext4_fopen(&fd, path1, "rb+");
+
+ if (rc != EOK)
+ break;
+
+ len = ll;
+ while (len) {
+ d = len > MAX_RW_BUFFER ? MAX_RW_BUFFER : len;
+
+ memset(read_buffer, 0, MAX_RW_BUFFER);
+ rc = ext4_fread(&fd, read_buffer, d, &rb);
+
+ if (rc != EOK)
+ break;
+
+ if (rb != d) {
+ printf("Read count error\n");
+ return -1;
+ }
+
+ if (memcmp(read_buffer, write_buffer, d)) {
+ printf("Read compare error\n");
+ return -1;
+ }
+
+ len -= d;
+ }
+ }
+
+ return rc;
+}
+
+int _multi_fremove(char *p)
+{
+ char path[256];
+ char path1[256];
+ char prefix[32];
+ int cnt, i, rc;
+
+ if (sscanf(p, "%s %s %d", path, prefix, &cnt) != 3) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ for (i = 0; i < cnt; ++i) {
+ sprintf(path1, "%s%s%d", path, prefix, i);
+ rc = ext4_fremove(path1);
+ if (rc != EOK)
+ break;
+ }
+
+ return rc;
+}
+
+int _multi_dcreate(char *p)
+{
+ char path[256];
+ char path1[256];
+ char prefix[32];
+ int cnt, i, rc;
+
+ if (sscanf(p, "%s %s %d", path, prefix, &cnt) != 3) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ for (i = 0; i < cnt; ++i) {
+ sprintf(path1, "%s%s%d", path, prefix, i);
+ rc = ext4_dir_mk(path1);
+ if (rc != EOK)
+ break;
+ }
+
+ return rc;
+}
+
+int _multi_dremove(char *p)
+{
+ char path[256];
+ char path1[256];
+ char prefix[32];
+ int cnt, i, rc;
+
+ if (sscanf(p, "%s %s %d", path, prefix, &cnt) != 3) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ for (i = 0; i < cnt; ++i) {
+ sprintf(path1, "%s%s%d", path, prefix, i);
+ rc = ext4_dir_rm(path1);
+ if (rc != EOK)
+ break;
+ }
+
+ return rc;
+}
+
+struct ext4_mount_stats saved_stats;
+
+int _stats_save(char *p)
+{
+ char path[256];
+
+ if (sscanf(p, "%s", path) != 1) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ return ext4_mount_point_stats(path, &saved_stats);
+}
+
+int _stats_check(char *p)
+{
+ char path[256];
+ int rc;
+
+ struct ext4_mount_stats actual_stats;
+
+ if (sscanf(p, "%s", path) != 1) {
+ printf("Param list error\n");
+ return -1;
+ }
+
+ rc = ext4_mount_point_stats(path, &actual_stats);
+
+ if (rc != EOK)
+ return rc;
+
+ if (memcmp(&saved_stats, &actual_stats,
+ sizeof(struct ext4_mount_stats))) {
+ if (verbose) {
+ printf("\tMount point stats error:\n");
+ printf("\tsaved_stats:\n");
+ printf("\tinodes_count = %d\n",
+ saved_stats.inodes_count);
+ printf("\tfree_inodes_count = %d\n",
+ saved_stats.free_inodes_count);
+ printf("\tblocks_count = %llu\n",
+ saved_stats.blocks_count);
+ printf("\tfree_blocks_count = %llu\n",
+ saved_stats.free_blocks_count);
+ printf("\tblock_size = %d\n", saved_stats.block_size);
+ printf("\tblock_group_count = %d\n",
+ saved_stats.block_group_count);
+ printf("\tblocks_per_group = %d\n",
+ saved_stats.blocks_per_group);
+ printf("\tinodes_per_group = %d\n",
+ saved_stats.inodes_per_group);
+ printf("\tvolume_name = %s\n", saved_stats.volume_name);
+ printf("\tactual_stats:\n");
+ printf("\tinodes_count = %d\n",
+ actual_stats.inodes_count);
+ printf("\tfree_inodes_count = %d\n",
+ actual_stats.free_inodes_count);
+ printf("\tblocks_count = %llu\n",
+ actual_stats.blocks_count);
+ printf("\tfree_blocks_count = %llu\n",
+ actual_stats.free_blocks_count);
+ printf("\tblock_size = %d\n", actual_stats.block_size);
+ printf("\tblock_group_count = %d\n",
+ actual_stats.block_group_count);
+ printf("\tblocks_per_group = %d\n",
+ actual_stats.blocks_per_group);
+ printf("\tinodes_per_group = %d\n",
+ actual_stats.inodes_per_group);
+ printf("\tvolume_name = %s\n",
+ actual_stats.volume_name);
+ }
+ return -1;
+ }
+
+ return rc;
+}
+
+static char *entry_to_str(uint8_t type)
+{
+ switch (type) {
+ case EXT4_DIRENTRY_UNKNOWN:
+ return "[UNK] ";
+ case EXT4_DIRENTRY_REG_FILE:
+ return "[FIL] ";
+ case EXT4_DIRENTRY_DIR:
+ return "[DIR] ";
+ case EXT4_DIRENTRY_CHRDEV:
+ return "[CHA] ";
+ case EXT4_DIRENTRY_BLKDEV:
+ return "[BLK] ";
+ case EXT4_DIRENTRY_FIFO:
+ return "[FIF] ";
+ case EXT4_DIRENTRY_SOCK:
+ return "[SOC] ";
+ case EXT4_DIRENTRY_SYMLINK:
+ return "[SYM] ";
+ default:
+ break;
+ }
+ return "[???]";
+}
+
+static int winsock_init(void)
+{
+#if WIN32
+ int rc;
+ static WSADATA wsaData;
+ rc = WSAStartup(MAKEWORD(2, 2), &wsaData);
+ if (rc != 0) {
+ return -1;
+ }
+#endif
+ return 0;
+}
+
+static void winsock_fini(void)
+{
+#if WIN32
+ WSACleanup();
+#endif
+}