1181 lines
24 KiB
C
1181 lines
24 KiB
C
|
/*
|
||
|
* 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 <inttypes.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 <ext4.h>
|
||
|
#include "../blockdev/linux/file_dev.h"
|
||
|
#include "../blockdev/windows/file_windows.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 bool verbose = false;
|
||
|
|
||
|
/**@brief Winpart mode*/
|
||
|
static bool winpart = false;
|
||
|
|
||
|
/**@brief Blockdev handle*/
|
||
|
static struct ext4_blockdev *bd;
|
||
|
|
||
|
static bool cache_wb = false;
|
||
|
|
||
|
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)(const 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",
|
||
|
};
|
||
|
|
||
|
static int device_register(const char *p);
|
||
|
static int mount(const char *p);
|
||
|
static int umount(const char *p);
|
||
|
static int mount_point_stats(const char *p);
|
||
|
static int cache_write_back(const char *p);
|
||
|
static int fremove(const char *p);
|
||
|
static int file_open(const char *p);
|
||
|
static int file_close(const char *p);
|
||
|
static int file_read(const char *p);
|
||
|
static int file_write(const char *p);
|
||
|
static int file_seek(const char *p);
|
||
|
static int file_tell(const char *p);
|
||
|
static int file_size(const char *p);
|
||
|
static int dir_rm(const char *p);
|
||
|
static int dir_mk(const char *p);
|
||
|
static int dir_open(const char *p);
|
||
|
static int dir_close(const char *p);
|
||
|
static int dir_close(const char *p);
|
||
|
static int dir_entry_get(const char *p);
|
||
|
|
||
|
static int multi_fcreate(const char *p);
|
||
|
static int multi_fwrite(const char *p);
|
||
|
static int multi_fread(const char *p);
|
||
|
static int multi_fremove(const char *p);
|
||
|
static int multi_dcreate(const char *p);
|
||
|
static int multi_dremove(const char *p);
|
||
|
static int stats_save(const char *p);
|
||
|
static int stats_check(const 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 */
|
||
|
file_open, /*PARAMS(2): fid path flags */
|
||
|
file_close, /*PARAMS(1): fid */
|
||
|
file_read, /*PARAMS(4): fid 0 len 0 */
|
||
|
file_write, /*PARAMS(4): fid 0 len 0 */
|
||
|
file_seek, /*PARAMS(2): fid off origin */
|
||
|
file_tell, /*PARAMS(2): fid exp */
|
||
|
file_size, /*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(const 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", no_argument, 0, 'v'},
|
||
|
{"winpart", no_argument, 0, 'w'},
|
||
|
{"cache_wb", no_argument, 0, 'c'},
|
||
|
{"version", no_argument, 0, 'x'},
|
||
|
{0, 0, 0, 0}};
|
||
|
|
||
|
while (-1 != (c = getopt_long(argc, argv, "i:p:vcwx", long_options,
|
||
|
&option_index))) {
|
||
|
|
||
|
switch (c) {
|
||
|
case 'i':
|
||
|
ext4_fname = optarg;
|
||
|
break;
|
||
|
case 'p':
|
||
|
connection_port = atoi(optarg);
|
||
|
break;
|
||
|
case 'v':
|
||
|
verbose = true;
|
||
|
break;
|
||
|
case 'c':
|
||
|
cache_wb = true;
|
||
|
break;
|
||
|
case 'w':
|
||
|
winpart = true;
|
||
|
break;
|
||
|
case 'x':
|
||
|
puts(VERSION);
|
||
|
exit(0);
|
||
|
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;
|
||
|
}
|
||
|
|
||
|
static int device_register(const 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) {
|
||
|
file_windows_name_set(ext4_fname);
|
||
|
bd = file_windows_dev_get();
|
||
|
|
||
|
} else
|
||
|
#endif
|
||
|
{
|
||
|
file_dev_name_set(ext4_fname);
|
||
|
bd = file_dev_get();
|
||
|
}
|
||
|
|
||
|
ext4_device_unregister_all();
|
||
|
|
||
|
return ext4_device_register(bd, dev_name);
|
||
|
}
|
||
|
|
||
|
static int mount(const 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;
|
||
|
}
|
||
|
|
||
|
if (verbose)
|
||
|
ext4_dmask_set(DEBUG_ALL);
|
||
|
|
||
|
rc = ext4_mount(dev_name, mount_point, false);
|
||
|
if (rc != EOK)
|
||
|
return rc;
|
||
|
|
||
|
rc = ext4_recover(mount_point);
|
||
|
if (rc != EOK && rc != ENOTSUP)
|
||
|
return rc;
|
||
|
|
||
|
rc = ext4_journal_start(mount_point);
|
||
|
if (rc != EOK)
|
||
|
return rc;
|
||
|
|
||
|
if (cache_wb)
|
||
|
ext4_cache_write_back(mount_point, 1);
|
||
|
return rc;
|
||
|
}
|
||
|
|
||
|
static int umount(const char *p)
|
||
|
{
|
||
|
char mount_point[32];
|
||
|
int rc;
|
||
|
|
||
|
if (sscanf(p, "%s", mount_point) != 1) {
|
||
|
printf("Param list error\n");
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
if (cache_wb)
|
||
|
ext4_cache_write_back(mount_point, 0);
|
||
|
|
||
|
rc = ext4_journal_stop(mount_point);
|
||
|
if (rc != EOK)
|
||
|
return rc;
|
||
|
|
||
|
rc = ext4_umount(mount_point);
|
||
|
if (rc != EOK)
|
||
|
return rc;
|
||
|
|
||
|
return rc;
|
||
|
}
|
||
|
|
||
|
static int mount_point_stats(const 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 = %" PRIu32"\n", stats.inodes_count);
|
||
|
printf("\tfree_inodes_count = %" PRIu32"\n",
|
||
|
stats.free_inodes_count);
|
||
|
printf("\tblocks_count = %" PRIu64"\n", stats.blocks_count);
|
||
|
printf("\tfree_blocks_count = %" PRIu64"\n",
|
||
|
stats.free_blocks_count);
|
||
|
printf("\tblock_size = %" PRIu32"\n", stats.block_size);
|
||
|
printf("\tblock_group_count = %" PRIu32"\n",
|
||
|
stats.block_group_count);
|
||
|
printf("\tblocks_per_group = %" PRIu32"\n",
|
||
|
stats.blocks_per_group);
|
||
|
printf("\tinodes_per_group = %" PRIu32"\n",
|
||
|
stats.inodes_per_group);
|
||
|
printf("\tvolume_name = %s\n", stats.volume_name);
|
||
|
}
|
||
|
|
||
|
return rc;
|
||
|
}
|
||
|
|
||
|
static int cache_write_back(const 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);
|
||
|
}
|
||
|
|
||
|
static int fremove(const char *p)
|
||
|
{
|
||
|
char path[255];
|
||
|
|
||
|
if (sscanf(p, "%s", path) != 1) {
|
||
|
printf("Param list error\n");
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
return ext4_fremove(path);
|
||
|
}
|
||
|
|
||
|
static int file_open(const 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;
|
||
|
}
|
||
|
|
||
|
static int file_close(const 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;
|
||
|
}
|
||
|
|
||
|
static int file_read(const 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;
|
||
|
}
|
||
|
|
||
|
static int file_write(const char *p)
|
||
|
{
|
||
|
int fid = MAX_FILES;
|
||
|
int d;
|
||
|
int rc;
|
||
|
|
||
|
int 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;
|
||
|
}
|
||
|
|
||
|
static int file_seek(const 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);
|
||
|
}
|
||
|
|
||
|
static int file_tell(const 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;
|
||
|
}
|
||
|
|
||
|
static int file_size(const 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;
|
||
|
}
|
||
|
|
||
|
static int dir_rm(const char *p)
|
||
|
{
|
||
|
char path[255];
|
||
|
|
||
|
if (sscanf(p, "%s", path) != 1) {
|
||
|
printf("Param list error\n");
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
return ext4_dir_rm(path);
|
||
|
}
|
||
|
|
||
|
static int dir_mk(const char *p)
|
||
|
{
|
||
|
char path[255];
|
||
|
|
||
|
if (sscanf(p, "%s", path) != 1) {
|
||
|
printf("Param list error\n");
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
return ext4_dir_mk(path);
|
||
|
}
|
||
|
|
||
|
static int dir_open(const 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;
|
||
|
}
|
||
|
|
||
|
static int dir_close(const 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;
|
||
|
}
|
||
|
|
||
|
static int dir_entry_get(const 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;
|
||
|
}
|
||
|
|
||
|
static int multi_fcreate(const 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;
|
||
|
}
|
||
|
|
||
|
static int multi_fwrite(const 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;
|
||
|
}
|
||
|
|
||
|
static int multi_fread(const 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;
|
||
|
}
|
||
|
|
||
|
static int multi_fremove(const 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;
|
||
|
}
|
||
|
|
||
|
static int multi_dcreate(const 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;
|
||
|
}
|
||
|
|
||
|
static int multi_dremove(const 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;
|
||
|
|
||
|
static int stats_save(const 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);
|
||
|
}
|
||
|
|
||
|
static int stats_check(const 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 = %" PRIu32"\n",
|
||
|
saved_stats.inodes_count);
|
||
|
printf("\tfree_inodes_count = %" PRIu32"\n",
|
||
|
saved_stats.free_inodes_count);
|
||
|
printf("\tblocks_count = %" PRIu64"\n",
|
||
|
saved_stats.blocks_count);
|
||
|
printf("\tfree_blocks_count = %" PRIu64"\n",
|
||
|
saved_stats.free_blocks_count);
|
||
|
printf("\tblock_size = %" PRIu32"\n",
|
||
|
saved_stats.block_size);
|
||
|
printf("\tblock_group_count = %" PRIu32"\n",
|
||
|
saved_stats.block_group_count);
|
||
|
printf("\tblocks_per_group = %" PRIu32"\n",
|
||
|
saved_stats.blocks_per_group);
|
||
|
printf("\tinodes_per_group = %" PRIu32"\n",
|
||
|
saved_stats.inodes_per_group);
|
||
|
printf("\tvolume_name = %s\n", saved_stats.volume_name);
|
||
|
printf("\tactual_stats:\n");
|
||
|
printf("\tinodes_count = %" PRIu32"\n",
|
||
|
actual_stats.inodes_count);
|
||
|
printf("\tfree_inodes_count = %" PRIu32"\n",
|
||
|
actual_stats.free_inodes_count);
|
||
|
printf("\tblocks_count = %" PRIu64"\n",
|
||
|
actual_stats.blocks_count);
|
||
|
printf("\tfree_blocks_count = %" PRIu64"\n",
|
||
|
actual_stats.free_blocks_count);
|
||
|
printf("\tblock_size = %d\n", actual_stats.block_size);
|
||
|
printf("\tblock_group_count = %" PRIu32"\n",
|
||
|
actual_stats.block_group_count);
|
||
|
printf("\tblocks_per_group = %" PRIu32"\n",
|
||
|
actual_stats.blocks_per_group);
|
||
|
printf("\tinodes_per_group = %" PRIu32"\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_DE_UNKNOWN:
|
||
|
return "[unk] ";
|
||
|
case EXT4_DE_REG_FILE:
|
||
|
return "[fil] ";
|
||
|
case EXT4_DE_DIR:
|
||
|
return "[dir] ";
|
||
|
case EXT4_DE_CHRDEV:
|
||
|
return "[cha] ";
|
||
|
case EXT4_DE_BLKDEV:
|
||
|
return "[blk] ";
|
||
|
case EXT4_DE_FIFO:
|
||
|
return "[fif] ";
|
||
|
case EXT4_DE_SOCK:
|
||
|
return "[soc] ";
|
||
|
case EXT4_DE_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
|
||
|
}
|